/*
MIT License

Copyright (c) 2016 Benjamin "Nefarius" Höglinger

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
*/


// ReSharper disable CppMissingIncludeGuard
#ifdef _MSC_VER
// ReSharper restore CppMissingIncludeGuard
#pragma once
#endif

#include <initguid.h>
#include <ViGEmBusShared.h>

#ifdef VIGEM_EXPORTS
#define VIGEM_API __declspec(dllexport)
#else
#define VIGEM_API __declspec(dllimport)
#endif

#define VIGEM_TARGETS_MAX   0xFF

typedef enum _VIGEM_ERRORS
{
    VIGEM_ERROR_NONE = 0x20000000,
    VIGEM_ERROR_BUS_NOT_FOUND = 0xE0000001,
    VIGEM_ERROR_NO_FREE_SLOT = 0xE0000002,
    VIGEM_ERROR_INVALID_TARGET = 0xE0000003,
    VIGEM_ERROR_REMOVAL_FAILED = 0xE0000004,
    VIGEM_ERROR_ALREADY_CONNECTED = 0xE0000005,
    VIGEM_ERROR_TARGET_UNINITIALIZED = 0xE0000006,
    VIGEM_ERROR_TARGET_NOT_PLUGGED_IN = 0xE0000007,
    VIGEM_ERROR_BUS_VERSION_MISMATCH = 0xE0000008,
    VIGEM_ERROR_BUS_ACCESS_FAILED = 0xE0000009
} VIGEM_ERROR;

#define VIGEM_SUCCESS(_val_) (_val_ == VIGEM_ERROR_NONE)

typedef enum _VIGEM_TARGET_STATE
{
    VIGEM_TARGET_NEW,
    VIGEM_TARGET_INITIALIZED,
    VIGEM_TARGET_CONNECTED,
    VIGEM_TARGET_DISCONNECTED
} VIGEM_TARGET_STATE, *PVIGEM_TARGET_STATE;

//
// Represents a virtual gamepad object.
// 
typedef struct _VIGEM_TARGET
{
    IN ULONG Size;
    IN ULONG SerialNo;
    IN VIGEM_TARGET_STATE State;
    IN USHORT VendorId;
    IN USHORT ProductId;
} VIGEM_TARGET, *PVIGEM_TARGET;

//
// Initializes a virtual gamepad object.
// 
VOID FORCEINLINE VIGEM_TARGET_INIT(
    _Out_ PVIGEM_TARGET Target
)
{
    RtlZeroMemory(Target, sizeof(VIGEM_TARGET));

    Target->Size = sizeof(VIGEM_TARGET);
    Target->State = VIGEM_TARGET_INITIALIZED;
}

typedef VOID(CALLBACK* PVIGEM_XUSB_NOTIFICATION)(
    VIGEM_TARGET Target,
    UCHAR LargeMotor,
    UCHAR SmallMotor,
    UCHAR LedNumber);

typedef VOID(CALLBACK* PVIGEM_DS4_NOTIFICATION)(
    VIGEM_TARGET Target,
    UCHAR LargeMotor,
    UCHAR SmallMotor,
    DS4_LIGHTBAR_COLOR LightbarColor);

#ifdef __cplusplus
extern "C"
{ // only need to export C interface if
  // used by C++ source code
#endif

    VIGEM_API VIGEM_ERROR vigem_init();

    VIGEM_API VOID vigem_shutdown();

    VIGEM_API VIGEM_ERROR vigem_target_plugin(
        _In_ VIGEM_TARGET_TYPE Type,
        _Out_ PVIGEM_TARGET Target);

    VIGEM_API VIGEM_ERROR vigem_target_unplug(
        _Out_ PVIGEM_TARGET Target);

    VIGEM_API VIGEM_ERROR vigem_register_xusb_notification(
        _In_ PVIGEM_XUSB_NOTIFICATION Notification,
        _In_ VIGEM_TARGET Target);

    VIGEM_API VIGEM_ERROR vigem_register_ds4_notification(
        _In_ PVIGEM_DS4_NOTIFICATION Notification,
        _In_ VIGEM_TARGET Target);

    VIGEM_API VIGEM_ERROR vigem_xusb_submit_report(
        _In_ VIGEM_TARGET Target,
        _In_ XUSB_REPORT Report);

    VIGEM_API VIGEM_ERROR vigem_ds4_submit_report(
        _In_ VIGEM_TARGET Target,
        _In_ DS4_REPORT Report);

    VIGEM_API VIGEM_ERROR vigem_xgip_submit_report(
        _In_ VIGEM_TARGET Target,
        _In_ XGIP_REPORT Report);

    VIGEM_API VOID vigem_target_set_vid(
        _Out_ PVIGEM_TARGET Target,
        _In_ USHORT VendorId);

    VIGEM_API VOID vigem_target_set_pid(
        _Out_ PVIGEM_TARGET Target,
        _In_ USHORT ProductId);

#ifdef __cplusplus
}
#endif