/* Copyright (c) 2013 mm0zct
 *
 * 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.
 *
 * 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.
 */

#include "rift-140.hpp"
#include "api/plugin-api.hpp"
#include <Extras/OVR_Math.h>

#include <QString>

using namespace OVR;

rift_tracker_140::rift_tracker_140() = default;

rift_tracker_140::~rift_tracker_140()
{
    if (hmd)
    {
        ovr_Destroy(hmd);
        ovr_Shutdown();
    }
}

module_status rift_tracker_140::start_tracker(QFrame*)
{
    if (OVR_FAILURE(ovr_Initialize(nullptr)))
        goto error;

    if (OVR_FAILURE(ovr_Create(&hmd, &luid)))
        goto error;

    if (OVR_FAILURE(ovr_SetTrackingOriginType(hmd, ovrTrackingOrigin_EyeLevel)))
        goto error;

    return status_ok();
error:
    hmd = nullptr;

    ovrErrorInfo err;
    ovr_GetLastErrorInfo(&err);

    QString error_string(err.ErrorString);
    if (error_string.size() == 0)
        error_string = QStringLiteral("Unknown reason #%1").arg(err.Result);

    ovr_Shutdown();

    return error(error_string);
}

void rift_tracker_140::data(double *data)
{
    if (hmd)
    {
        ovrTrackingState ss = ovr_GetTrackingState(hmd, 0, false);
        if (ss.StatusFlags & ovrStatus_OrientationTracked)
        {
            constexpr float c_mult = 8;
            constexpr float c_div = 1/c_mult;

            Vector3f axis;
            float angle;

            const Posef pose(ss.HeadPose.ThePose);
            pose.Rotation.GetAxisAngle(&axis, &angle);
            angle *= c_div;

            float yaw, pitch, roll;
            Quatf(axis, angle).GetYawPitchRoll(&yaw, &pitch, &roll);

            yaw *= c_mult;
            pitch *= c_mult;
            roll *= c_mult;

            constexpr double d2r = 180 / M_PI;
            data[Yaw] =     double(yaw)                 * -d2r;
            data[Pitch] =   double(pitch)               *  d2r;
            data[Roll] =    double(roll)                *  d2r;
            data[TX] = double(pose.Translation.x)   * -1e2;
            data[TY] = double(pose.Translation.y)   *  1e2;
            data[TZ] = double(pose.Translation.z)   *  1e2;
        }
    }
}

bool rift_tracker_140::center()
{
    (void)ovr_RecenterTrackingOrigin(hmd);
    return false;
}

OPENTRACK_DECLARE_TRACKER(rift_tracker_140, dialog_rift_140, rift_140Dll)