/* Homepage         http://facetracknoir.sourceforge.net/home/default.htm        *
 *                                                                               *
 * ISC License (ISC)                                                             *
 *                                                                               *
 * Copyright (c) 2015, Wim Vriend
 * Copyright (c) 2014, Stanislaw Halik <sthalik@misaki.pl>
 *                                                                               *
 * Permission to use, copy, modify, and/or distribute this software for any      *
 * purpose with or without fee is hereby granted, provided that the above        *
 * copyright notice and this permission notice appear in all copies.             *
 */
#include "ftnoir_protocol_sc.h"
#include "api/plugin-api.hpp"
#include "opentrack-library-path.h"
#include "compat/timer.hpp"

simconnect::simconnect() : hSimConnect(nullptr), should_reconnect(false)
{
}

simconnect::~simconnect()
{
    requestInterruption();
    wait();
}

void simconnect::run()
{
    HANDLE event = CreateEvent(NULL, FALSE, FALSE, nullptr);

    if (event == nullptr)
    {
        qDebug() << "simconnect: event create" << GetLastError();
        return;
    }

    while (!isInterruptionRequested())
    {
        HRESULT hr;

        qDebug() << "simconnect: starting";

        if (SUCCEEDED(hr = simconnect_open(&hSimConnect, "opentrack", nullptr, 0, event, 0)))
        {
            if (!SUCCEEDED(hr = simconnect_subscribetosystemevent(hSimConnect, 0, "Frame")))
            {
                qDebug() << "simconnect: can't subscribe to frame event:" << hr;

                should_reconnect = true; // will free simconnect handle and continue the loop
            }

            Timer tm;
            int idle_seconds = 0;

            while (!isInterruptionRequested())
            {
                if (should_reconnect)
                {
                    should_reconnect = false;
                    break;
                }

                if (WaitForSingleObject(event, 10) == WAIT_OBJECT_0)
                {
                    tm.start();
                    idle_seconds = 0;

                    if (!SUCCEEDED(hr = simconnect_calldispatch(hSimConnect, processNextSimconnectEvent, reinterpret_cast<void*>(this))))
                    {
                        qDebug() << "simconnect: calldispatch failed:" << hr;
                        break;
                    }
                }
                else
                {
                    if (int(tm.elapsed_seconds()) > idle_seconds)
                    {
                        idle_seconds++;
                        qDebug() << "simconnect: can't process event for" << int(tm.elapsed_seconds()) << "seconds";
                    }
                }
            }

            (void) simconnect_close(hSimConnect);
        }
        else
        {
            qDebug() << "simconnect: can't open handle:" << hr;
        }

        if (!isInterruptionRequested())
            Sleep(100);
    }

    CloseHandle(event);
}

void simconnect::pose( const double *headpose )
{
    // euler degrees
    virtSCRotX = float(-headpose[Pitch]);
    virtSCRotY = float(headpose[Yaw]);
    virtSCRotZ = float(headpose[Roll]);

    // cm to meters
    virtSCPosX = float(-headpose[TX]/100);
    virtSCPosY = float(headpose[TY]/100);
    virtSCPosZ = float(-headpose[TZ]/100);
}

#ifdef __GNUC__
#   pragma GCC diagnostic ignored "-Wmissing-field-initializers"
#endif

class ActivationContext
{
public:
    ActivationContext(const int resid) :
        ok(false)
    {
        hactctx = INVALID_HANDLE_VALUE;
        actctx_cookie = 0;
        ACTCTXA actx = {0};
        actx.cbSize = sizeof(ACTCTXA);
        actx.lpResourceName = MAKEINTRESOURCEA(resid);
        actx.dwFlags = ACTCTX_FLAG_RESOURCE_NAME_VALID;
#ifdef _MSC_VER
#	define PREFIX ""
#else
#	define PREFIX "lib"
#endif
        static const QString prefix = OPENTRACK_BASE_PATH + OPENTRACK_LIBRARY_PATH;
        QString path = prefix + PREFIX "opentrack-proto-simconnect.dll";
        QByteArray name = QFile::encodeName(path);
        actx.lpSource = name.constData();
        hactctx = CreateActCtxA(&actx);
        actctx_cookie = 0;
        if (hactctx != INVALID_HANDLE_VALUE)
        {
            if (!ActivateActCtx(hactctx, &actctx_cookie))
            {
                qDebug() << "simconnect: can't set win32 activation context" << GetLastError();
                ReleaseActCtx(hactctx);
                hactctx = INVALID_HANDLE_VALUE;
            }
            else
                ok = true;
        } else {
            qDebug() << "simconnect: can't create win32 activation context" << GetLastError();
        }
    }
    ~ActivationContext() {
        if (hactctx != INVALID_HANDLE_VALUE)
        {
            DeactivateActCtx(0, actctx_cookie);
            ReleaseActCtx(hactctx);
        }
    }
    bool is_ok() const { return ok; }

private:
    ULONG_PTR actctx_cookie;
    HANDLE hactctx;
    bool ok;
};

bool simconnect::correct()
{
    if (!SCClientLib.isLoaded())
    {
        ActivationContext ctx(142 + static_cast<int>(s.sxs_manifest));

        if (ctx.is_ok())
        {
            SCClientLib.setFileName("SimConnect.dll");
            if (!SCClientLib.load()) {
                qDebug() << "simconnect: dll load failed --" << SCClientLib.errorString();
                return false;
            }
        }
        else
            return false;
    }

    simconnect_open = (importSimConnect_Open) SCClientLib.resolve("SimConnect_Open");
    if (simconnect_open == NULL) {
        qDebug() << "simconnect: SimConnect_Open function not found in DLL!";
        return false;
    }
    simconnect_set6DOF = (importSimConnect_CameraSetRelative6DOF) SCClientLib.resolve("SimConnect_CameraSetRelative6DOF");
    if (simconnect_set6DOF == NULL) {
        qDebug() << "simconnect: SimConnect_CameraSetRelative6DOF function not found in DLL!";
        return false;
    }
    simconnect_close = (importSimConnect_Close) SCClientLib.resolve("SimConnect_Close");
    if (simconnect_close == NULL) {
        qDebug() << "simconnect: SimConnect_Close function not found in DLL!";
        return false;
    }

    simconnect_calldispatch = (importSimConnect_CallDispatch) SCClientLib.resolve("SimConnect_CallDispatch");
    if (simconnect_calldispatch == NULL) {
        qDebug() << "simconnect: SimConnect_CallDispatch function not found in DLL!";
        return false;
    }

    simconnect_subscribetosystemevent = (importSimConnect_SubscribeToSystemEvent) SCClientLib.resolve("SimConnect_SubscribeToSystemEvent");
    if (simconnect_subscribetosystemevent == NULL) {
        qDebug() << "simconnect: SimConnect_SubscribeToSystemEvent function not found in DLL!";
        return false;
    }

    start();

    return true;
}

void simconnect::handle()
{
    (void) simconnect_set6DOF(hSimConnect, virtSCPosX, virtSCPosY, virtSCPosZ, virtSCRotX, virtSCRotZ, virtSCRotY);
}

void CALLBACK simconnect::processNextSimconnectEvent(SIMCONNECT_RECV* pData, DWORD, void *self_)
{
    simconnect& self = *reinterpret_cast<simconnect*>(self_);

    switch(pData->dwID)
    {
    default:
        break;
    case SIMCONNECT_RECV_ID_EXCEPTION:
    case SIMCONNECT_RECV_ID_QUIT:
        self.should_reconnect = true;
        qDebug() << "simconnect: received ID_QUIT event";
        break;
    case SIMCONNECT_RECV_ID_EVENT_FRAME:
        self.handle();
        break;
    }
}

OPENTRACK_DECLARE_PROTOCOL(simconnect, SCControls, simconnectDll)