Nuitrack.h

#ifndef NUITRACK_HEADER_ONLY_API_H_
#define NUITRACK_HEADER_ONLY_API_H_

#include <string>
#include <memory>
#include <functional>

#include <stdio.h>

#include "nuitrack/types/Export.h"
#include "nuitrack/types/Error.h"

#include "nuitrack/modules/NuitrackModule.h"
#include "nuitrack/modules/DepthSensor.h"
#include "nuitrack/modules/ColorSensor.h"
#include "nuitrack/modules/UserTracker.h"
#include "nuitrack/modules/SkeletonTracker.h"
#include "nuitrack/modules/GestureRecognizer.h"
#include "nuitrack/modules/HandTracker.h"
#include "nuitrack/types/IssuesData.h"
#include "nuitrack/types/NuitrackDevice.h"

#include "nuitrack/capi/Public_Nuitrack_CAPI.h"
#include "nuitrack/capi/IssueTracker_CAPI.h"
#include "nuitrack/capi/NuitrackDevice_CAPI.h"

namespace tdv
{
namespace nuitrack
{

class NUITRACK_LOCAL Nuitrack
{
public:
    static void init(const std::string& config = "")
    {
        nuitrack_error* e = nullptr;
        nuitrack_InitializeFromConfig_E(config.c_str(), &e);
        ExceptionTranslator::handle(e);
    }

    static void run()
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_Run());
    }

    static void update()
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_Update());
    }

    static void update(std::shared_ptr<HeaderOnlyAPI_Module> module)
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_SyncUpdatePublic(module.get()));
    }

    static void waitUpdate(std::shared_ptr<HeaderOnlyAPI_Module> module)
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_WaitSyncUpdatePublic(module.get()));
    }

    static void release()
    {
        CallbackStruct<IssuesData::Ptr>* callbackStruct =
                (CallbackStruct<IssuesData::Ptr>*)nuitrack_getIssuesCallbackStruct();
        if(callbackStruct != NULL)
        {
            nuitrack_setIssuesCallbackStruct(NULL);
            delete callbackStruct;
        }
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_Release());
    }

    static void setConfigValue(const std::string& key, const std::string& value)
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_SetConfigValue(key.c_str(), value.c_str()));
    }

    static std::string getConfigValue(const std::string& key)
    {
        const int bufferSize = 5000;
        std::string result;
        result.resize(bufferSize);
        ExceptionTranslator::generateExceptionByErrorCode(
            nuitrack_GetConfigValue(key.c_str(), (char *)result.c_str(), bufferSize));

        result.resize(strlen(result.c_str()));
        return result;
    }

    static std::string getInstancesJson()
    {
        int size = 0;
        nuitrack_error* e = nullptr;
        nuitrack_GetInstancesJsonSize(&size, &e);
        ExceptionTranslator::handle(e);

        std::string result;
        result.resize(size);
        nuitrack_GetInstancesJsonData((char *)result.c_str(), &e);
        ExceptionTranslator::handle(e);

        return result;
    }

    static int getVersion()
    {
        int version;
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_GetVersion(&version));

        return version;
    }

    typedef std::function<void (IssuesData::Ptr)> OnIssueUpdate;

    static uint64_t connectOnIssuesUpdate(const OnIssueUpdate& callback)
    {
        IssueTracker* issueTracker = NULL;
        nuitrack_getIssueTracker(&issueTracker);
        if(issueTracker == NULL)
            return 0;

        if(nuitrack_getIssuesCallbackStruct() == NULL)
        {
            CallbackStruct<IssuesData::Ptr>* ptr = new CallbackStruct<IssuesData::Ptr>();
            nuitrack_setIssuesCallbackStruct(ptr);

            IssueTrackerCallbackWrapper* callbackWrapper = new IssueTrackerCallbackWrapper();
            callbackWrapper->setIssueTracker(issueTracker);
            callbackWrapper->setFunctionAddress(&onIssuesUpdateCallback);
            nuitrack_registerIssuesTrackerCallback(callbackWrapper);
        }

        return ((CallbackStruct<IssuesData::Ptr>*)(nuitrack_getIssuesCallbackStruct()))->addCallback(callback);
    }

    static void onIssuesUpdateCallback(IssueTrackerData* data)
    {
        IssuesData::Ptr newData = IssuesData::Ptr(new IssuesData(data));
        CallbackStruct<IssuesData::Ptr>* callbackStruct =
                (CallbackStruct<IssuesData::Ptr>*)nuitrack_getIssuesCallbackStruct();
        if(callbackStruct == NULL)
            return;
        callbackStruct->executeAllCallbacks(newData);
    }

    static void disconnectOnIssuesUpdate(uint64_t handler)
    {
        CallbackStruct<IssuesData::Ptr>* callbackStruct =
                (CallbackStruct<IssuesData::Ptr>*)nuitrack_getIssuesCallbackStruct();
        if(callbackStruct == NULL)
            return;
        callbackStruct->deleteCallback(handler);
    }

    static std::vector<device::NuitrackDevice::Ptr> getDeviceList()
    {
        std::vector<device::NuitrackDevice::Ptr> result;
        int device_limit = 0;
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_nuitrackDevice_getDeviceListLimitConst(device_limit));
        std::vector<NuitrackDeviceDataPtr> bufferOfDevices(device_limit);
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_GetDeviceList(bufferOfDevices.data(), device_limit));

        for(int i = 0; i < device_limit; i++)
        {
            if(bufferOfDevices[i] == NULL)
                break;
            result.emplace_back(new device::NuitrackDevice(bufferOfDevices[i]));
        }

        return result;
    }

    static void setDevice(device::NuitrackDevice::Ptr dev)
    {
        ExceptionTranslator::generateExceptionByErrorCode(nuitrack_SetDevice(dev->_pimpl));
    }

    static std::string getLicense()
    {
        const int bufferSize = 5000;
        std::string result;
        result.resize(bufferSize);
        ExceptionTranslator::generateExceptionByErrorCode(
                    nuitrack_GetLicense((char *)result.c_str(), bufferSize));

        result.resize(strlen(result.c_str()));
        return result;
    }
};

} /* namespace nuitrack */
} /* namespace tdv */

#endif /* NUITRACK_HEADER_ONLY_API_H_ */