//
//  LaserPointer.cpp
//  interface/src/raypick
//
//  Created by Sam Gondelman 7/11/2017
//  Copyright 2017 High Fidelity, Inc.
//
//  Distributed under the Apache License, Version 2.0.
//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//
#include "LaserPointer.h"

#include "JointRayPick.h"
#include "StaticRayPick.h"

#include "Application.h"
#include "avatar/AvatarManager.h"

LaserPointer::LaserPointer(const QString& jointName, const glm::vec3& posOffset, const glm::vec3& dirOffset, const uint16_t filter, const float maxDistance,
    const QHash<QString, RenderState>& renderStates, const bool faceAvatar, const bool centerEndY, const bool enabled) :
    _renderingEnabled(enabled),
    _renderStates(renderStates),
    _faceAvatar(faceAvatar),
    _centerEndY(centerEndY)
{
    _rayPickUID = RayPickManager::getInstance().addRayPick(std::make_shared<JointRayPick>(jointName, posOffset, dirOffset, filter, maxDistance, enabled));

    for (auto& state : _renderStates.keys()) {
        if (!enabled || state != _currentRenderState) {
            disableRenderState(state);
        }
    }
}

LaserPointer::LaserPointer(const glm::vec3& position, const glm::vec3& direction, const uint16_t filter, const float maxDistance,
        const QHash<QString, RenderState>& renderStates, const bool faceAvatar, const bool centerEndY, const bool enabled) :
    _renderingEnabled(enabled),
    _renderStates(renderStates),
    _faceAvatar(faceAvatar),
    _centerEndY(centerEndY)
{
    _rayPickUID = RayPickManager::getInstance().addRayPick(std::make_shared<StaticRayPick>(position, direction, filter, maxDistance, enabled));

    for (auto& state : _renderStates.keys()) {
        if (!enabled || state != _currentRenderState) {
            disableRenderState(state);
        }
    }
}

LaserPointer::~LaserPointer() {
    RayPickManager::getInstance().removeRayPick(_rayPickUID);
    for (RenderState& renderState : _renderStates) {
        if (!renderState.getStartID().isNull()) {
            qApp->getOverlays().deleteOverlay(renderState.getStartID());
        }
        if (!renderState.getPathID().isNull()) {
            qApp->getOverlays().deleteOverlay(renderState.getPathID());
        }
        if (!renderState.getEndID().isNull()) {
            qApp->getOverlays().deleteOverlay(renderState.getEndID());
        }
    }
}

void LaserPointer::enable() {
    RayPickManager::getInstance().enableRayPick(_rayPickUID);
    _renderingEnabled = true;
}

void LaserPointer::disable() {
    RayPickManager::getInstance().disableRayPick(_rayPickUID);
    _renderingEnabled = false;
    if (!_currentRenderState.isEmpty() && _renderStates.contains(_currentRenderState)) {
        disableRenderState(_currentRenderState);
    }
}

void LaserPointer::setRenderState(const QString& state) {
    if (!_currentRenderState.isEmpty() && _renderStates.contains(_currentRenderState)) {
        disableRenderState(_currentRenderState);
    }
    _currentRenderState = state;
}

void LaserPointer::disableRenderState(const QString& renderState) {
    if (!_renderStates[renderState].getStartID().isNull()) {
        QVariantMap startProps;
        startProps.insert("visible", false);
        startProps.insert("ignoreRayIntersection", true);
        qApp->getOverlays().editOverlay(_renderStates[renderState].getStartID(), startProps);
    }
    if (!_renderStates[renderState].getPathID().isNull()) {
        QVariantMap pathProps;
        pathProps.insert("visible", false);
        pathProps.insert("ignoreRayIntersection", true);
        qApp->getOverlays().editOverlay(_renderStates[renderState].getPathID(), pathProps);
    }
    if (!_renderStates[renderState].getEndID().isNull()) {
        QVariantMap endProps;
        endProps.insert("visible", false);
        endProps.insert("ignoreRayIntersection", true);
        qApp->getOverlays().editOverlay(_renderStates[renderState].getEndID(), endProps);
    }
}

void LaserPointer::update() {
    RayPickResult prevRayPickResult = RayPickManager::getInstance().getPrevRayPickResult(_rayPickUID);
    if (_renderingEnabled && !_currentRenderState.isEmpty() && _renderStates.contains(_currentRenderState) && prevRayPickResult.type != IntersectionType::NONE) {
        PickRay pickRay = RayPickManager::getInstance().getPickRay(_rayPickUID);
        if (!_renderStates[_currentRenderState].getStartID().isNull()) {
            QVariantMap startProps;
            startProps.insert("position", vec3toVariant(pickRay.origin));
            startProps.insert("visible", true);
            startProps.insert("ignoreRayIntersection", _renderStates[_currentRenderState].doesStartIgnoreRays());
            qApp->getOverlays().editOverlay(_renderStates[_currentRenderState].getStartID(), startProps);
        }
        glm::vec3 endVec = pickRay.origin + pickRay.direction * prevRayPickResult.distance;
        QVariant end = vec3toVariant(endVec);
        if (!_renderStates[_currentRenderState].getPathID().isNull()) {
            QVariantMap pathProps;
            pathProps.insert("start", vec3toVariant(pickRay.origin));
            pathProps.insert("end", end);
            pathProps.insert("visible", true);
            pathProps.insert("ignoreRayIntersection", _renderStates[_currentRenderState].doesPathIgnoreRays());
            qApp->getOverlays().editOverlay(_renderStates[_currentRenderState].getPathID(), pathProps);
        }
        if (!_renderStates[_currentRenderState].getEndID().isNull()) {
            QVariantMap endProps;
            if (_centerEndY) {
                endProps.insert("position", end);
            } else {
                glm::vec3 dim = vec3FromVariant(qApp->getOverlays().getProperty(_renderStates[_currentRenderState].getEndID(), "dimensions").value);
                endProps.insert("position", vec3toVariant(endVec + glm::vec3(0, 0.5f * dim.y, 0)));
            }
            if (_faceAvatar) {
                glm::quat rotation = glm::inverse(glm::quat_cast(glm::lookAt(endVec, DependencyManager::get<AvatarManager>()->getMyAvatar()->getPosition(), Vectors::UP)));
                endProps.insert("rotation", quatToVariant(glm::quat(glm::radians(glm::vec3(0, glm::degrees(safeEulerAngles(rotation)).y, 0)))));
            }
            endProps.insert("visible", true);
            endProps.insert("ignoreRayIntersection", _renderStates[_currentRenderState].doesEndIgnoreRays());
            qApp->getOverlays().editOverlay(_renderStates[_currentRenderState].getEndID(), endProps);
        }
    } else {
        disableRenderState(_currentRenderState);
    }
}

RenderState::RenderState(const OverlayID& startID, const OverlayID& pathID, const OverlayID& endID) :
    _startID(startID), _pathID(pathID), _endID(endID)
{
    if (!_startID.isNull()) {
        _startIgnoreRays = qApp->getOverlays().getOverlay(_startID)->getProperty("ignoreRayIntersection").toBool();
    }
    if (!_pathID.isNull()) {
        _pathIgnoreRays = qApp->getOverlays().getOverlay(_pathID)->getProperty("ignoreRayIntersection").toBool();
    }
    if (!_endID.isNull()) {
        _endIgnoreRays = qApp->getOverlays().getOverlay(_endID)->getProperty("ignoreRayIntersection").toBool();
    }
}