UserFrame.h

#ifndef NUITRACK_USERFRAME_H_
#define NUITRACK_USERFRAME_H_

#include <vector>

#include "nuitrack/types/Frame.h"
#include "nuitrack/types/User.h"
#include "nuitrack/capi/UserTracker_CAPI.h"

namespace tdv
{
namespace nuitrack
{

class UserFrame : public Frame<UserFrame, uint16_t>
{
public:
    UserFrame(UserTrackerData* pimpl)
    {
        _pimpl = pimpl;
        nuitrack_AddUserTrackerDataRef(pimpl);
    }

    virtual ~UserFrame()
    {
        nuitrack_DestroyUserTrackerData(_pimpl);
    }

    const std::vector<User> getUsers() const
    {
        UserTrackerData* dataHolder = _pimpl;

        std::vector<User> usersVector;

        size_t numUsers = nuitrack_GetUserFrameNumUsers(dataHolder);

        for(size_t i = 0; i < numUsers; i++)
        {
            User user;
            nuitrack_GetUserFrameUser(dataHolder, i, &user);
            usersVector.push_back(user);
        }

        return usersVector;
    }

    int getRows() const
    {
        return nuitrack_GetUserFrameRows(_pimpl);
    }

    int getCols() const
    {
        return nuitrack_GetUserFrameCols(_pimpl);
    }

    uint64_t getID() const
    {
        return nuitrack_GetUserFrameID(_pimpl);
    }

    const DataType* getData() const
    {
        return nuitrack_GetUserFrameData(_pimpl);
    }

    uint64_t getTimestamp() const
    {
        return nuitrack_GetUserFrameTimestamp(_pimpl);
    }

    const Vector3 getFloor() const
    {
        Vector3 floor;
        nuitrack_GetUserFrameFloor(_pimpl, &floor);
        return floor;
    }

    const Vector3 getFloorNormal() const
    {
        Vector3 floorNormal;
        nuitrack_GetUserFrameFloorNormal(_pimpl, &floorNormal);
        return floorNormal;
    }

private:
    UserTrackerData* _pimpl;

};


}
}

#endif /* NUITRACK_USERFRAME_H_ */