/*
 * Copyright (c) Contributors to the Open 3D Engine Project
 * 
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */
#include <PhysX_precompiled.h>

#include <AzCore/Component/TransformBus.h>
#include <AzToolsFramework/Manipulators/ManipulatorManager.h>
#include <AzToolsFramework/Manipulators/ManipulatorView.h>

#include <Editor/EditorSubComponentModeVec3.h>
#include <PhysX/EditorJointBus.h>
#include <Source/Utils.h>

namespace PhysX
{
    EditorSubComponentModeVec3::EditorSubComponentModeVec3(
        const AZ::EntityComponentIdPair& entityComponentIdPair
        , const AZ::Uuid& componentType
        , const AZStd::string& name)
        : EditorSubComponentModeBase(entityComponentIdPair, componentType, name)
        , m_translationManipulators(AzToolsFramework::TranslationManipulators::Dimensions::Three,
            AZ::Transform::Identity(), AZ::Vector3::CreateOne())
    {
        AZ::Transform worldTransform = PhysX::Utils::GetEntityWorldTransformWithoutScale(m_entityComponentId.GetEntityId());

        AZ::Vector3 localTranslation;
        EditorJointRequestBus::EventResult(
            localTranslation, m_entityComponentId
            , &EditorJointRequests::GetVector3Value
            , m_name);

        m_translationManipulators.SetSpace(worldTransform);
        m_translationManipulators.SetLocalPosition(localTranslation);
        m_translationManipulators.AddEntityComponentIdPair(m_entityComponentId);

        m_translationManipulators.Register(AzToolsFramework::g_mainManipulatorManagerId);

        AzToolsFramework::ConfigureTranslationManipulatorAppearance3d(&m_translationManipulators);

        m_translationManipulators.InstallLinearManipulatorMouseMoveCallback([this](
            const AzToolsFramework::LinearManipulator::Action& action)
        {
            OnManipulatorMoved(action.LocalPosition());
        });

        m_translationManipulators.InstallPlanarManipulatorMouseMoveCallback([this](
            const AzToolsFramework::PlanarManipulator::Action& action)
        {
            OnManipulatorMoved(action.LocalPosition());
        });

        m_translationManipulators.InstallSurfaceManipulatorMouseMoveCallback([this](
            const AzToolsFramework::SurfaceManipulator::Action& action)
        {
            OnManipulatorMoved(action.LocalPosition());
        });
    }

    EditorSubComponentModeVec3::~EditorSubComponentModeVec3()
    {
        m_translationManipulators.Unregister();
    }

    void EditorSubComponentModeVec3::Refresh()
    {
        AZ::Transform worldTransform = PhysX::Utils::GetEntityWorldTransformWithoutScale(m_entityComponentId.GetEntityId());

        AZ::Vector3 localTranslation;
        EditorJointRequestBus::EventResult(
            localTranslation, m_entityComponentId
            , &EditorJointRequests::GetVector3Value
            , m_name);

        m_translationManipulators.SetSpace(worldTransform);
        m_translationManipulators.SetLocalPosition(localTranslation);
        m_translationManipulators.SetBoundsDirty();
    }

    void EditorSubComponentModeVec3::OnManipulatorMoved(const AZ::Vector3& position)
    {
        m_translationManipulators.SetLocalPosition(position);
        EditorJointRequestBus::Event(
            m_entityComponentId
            , &EditorJointRequests::SetVector3Value
            , m_name
            , position);
    }
} // namespace PhysX