//

// Copyright (C) 2014-2015 Stéphane Lenclud.

//

// This file is part of SharpLibHid.

//

// SharpDisplayManager is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

//

// SharpDisplayManager is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

// GNU General Public License for more details.

//

// You should have received a copy of the GNU General Public License

// along with SharpDisplayManager.  If not, see <http://www.gnu.org/licenses/>.

//

using System;

using System.Windows.Forms;

using System.Runtime.InteropServices;

using System.Diagnostics;

using System.Text;

using Microsoft.Win32.SafeHandles;

using SharpLib.Win32;

using System.Collections.Generic;

namespace SharpLib.Hid

{

    /// <summary>

    /// Our HID handler manages raw input registrations, processes WM_INPUT messages and broadcasts HID events in return.

    /// </summary>

    public class Handler : IDisposable

    {

        public delegate void HidEventHandler(object aSender, Event aHidEvent);

        public event HidEventHandler OnHidEvent;

        List<Event> iHidEvents;

        RAWINPUTDEVICE[] iRawInputDevices;

        public bool IsRegistered { get; private set; }

        public bool ManageRepeats { get; private set; }

        public Handler(RAWINPUTDEVICE[] aRawInputDevices, bool aManageRepeats=false)

        {

            iRawInputDevices = aRawInputDevices;

            iHidEvents = new List<Event>();

            IsRegistered = Function.RegisterRawInputDevices(iRawInputDevices, (uint)iRawInputDevices.Length, (uint)Marshal.SizeOf(iRawInputDevices[0]));

            ManageRepeats = aManageRepeats;

        }

        /// <summary>

        /// Will de-register devices.

        /// </summary>

        public void Dispose()

        {

            //Setup device removal

            for (int i=0; i<iRawInputDevices.Length; i++)

            {

                iRawInputDevices[i].dwFlags = Const.RIDEV_REMOVE;

                iRawInputDevices[i].hwndTarget = IntPtr.Zero;

            }

            //De-register

            Function.RegisterRawInputDevices(iRawInputDevices, (uint)iRawInputDevices.Length, (uint)Marshal.SizeOf(iRawInputDevices[0]));

            IsRegistered = false;

        }

        /// <summary>

        /// Process a WM_INPUT message.

        /// </summary>

        /// <param name="aMessage"></param>

        public void ProcessInput(ref Message aMessage)

        {

            if (aMessage.Msg != Const.WM_INPUT)

            {

                //We only process WM_INPUT messages

                return;

            }

            Event hidEvent = new Event(aMessage, OnHidEventRepeat, ManageRepeats);

            hidEvent.DebugWrite();

            if (!hidEvent.IsValid || !hidEvent.IsGeneric)

            {

                Debug.WriteLine("Skipping HID message.");

                return;

            }

            //

            if (ManageRepeats)

            {

                if (hidEvent.IsButtonUp)

                {

                    //This is a key up event

                    //We need to discard any events belonging to the same page and collection

                    for (int i = (iHidEvents.Count - 1); i >= 0; i--)

                    {

                        if (iHidEvents[i].UsageId == hidEvent.UsageId)

                        {

                            iHidEvents[i].Dispose();

                            iHidEvents.RemoveAt(i);

                        }

                    }

                }

                else

                {

                    //Keep that event until we get a key up message

                    iHidEvents.Add(hidEvent);

                }

            }

            //Broadcast our events

            OnHidEvent(this, hidEvent);    

        }

        public void OnHidEventRepeat(Event aHidEvent)

        {

            //Broadcast our events

            OnHidEvent(this, aHidEvent);    

        }

    }

}