/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include "LmbrCentral_precompiled.h"
#include "EditorSplineComponentMode.h"

#include <AzToolsFramework/Manipulators/SplineHoverSelection.h>
#include <AzToolsFramework/Manipulators/ManipulatorManager.h>

namespace LmbrCentral
{
    AZ_CLASS_ALLOCATOR_IMPL(EditorSplineComponentMode, AZ::SystemAllocator, 0)

    EditorSplineComponentMode::EditorSplineComponentMode(
        const AZ::EntityComponentIdPair& entityComponentIdPair, const AZ::Uuid componentType)
        : EditorBaseComponentMode(entityComponentIdPair, componentType)
    {
        AZ::TransformNotificationBus::Handler::BusConnect(entityComponentIdPair.GetEntityId());
        SplineComponentNotificationBus::Handler::BusConnect(entityComponentIdPair.GetEntityId());
        EditorSplineComponentNotificationBus::Handler::BusConnect(entityComponentIdPair.GetEntityId());

        CreateManipulators();
    }

    EditorSplineComponentMode::~EditorSplineComponentMode()
    {
        EditorSplineComponentNotificationBus::Handler::BusDisconnect();
        SplineComponentNotificationBus::Handler::BusDisconnect();
        AZ::TransformNotificationBus::Handler::BusDisconnect();

        m_vertexSelection.Destroy();
    }

    void EditorSplineComponentMode::Refresh()
    {
        ContainerChanged();
    }

    AZStd::vector<AzToolsFramework::ActionOverride> EditorSplineComponentMode::PopulateActionsImpl()
    {
        return m_vertexSelection.ActionOverrides();
    }

    bool EditorSplineComponentMode::HandleMouseInteraction(
        const AzToolsFramework::ViewportInteraction::MouseInteractionEvent& mouseInteraction)
    {
        return m_vertexSelection.HandleMouse(mouseInteraction);
    }

    void EditorSplineComponentMode::CreateManipulators()
    {
        using namespace AzToolsFramework;

        bool empty = false;
        SplineComponentRequestBus::EventResult(
            empty, GetEntityId(), &SplineComponentRequests::Empty);

        // if we have no vertices, do not attempt to create any manipulators
        if (empty)
        {
            return;
        }

        AZ::SplinePtr spline = nullptr;
        SplineComponentRequestBus::EventResult(
            spline, GetEntityId(), &SplineComponentRequests::GetSpline);

        m_vertexSelection.Create(
            GetEntityComponentIdPair(), g_mainManipulatorManagerId,
            AZStd::make_unique<SplineHoverSelection>(
                GetEntityComponentIdPair(), g_mainManipulatorManagerId, spline),
            TranslationManipulators::Dimensions::Three, ConfigureTranslationManipulatorAppearance3d);
    }

    void EditorSplineComponentMode::ContainerChanged()
    {
        // destroy and recreate manipulators when container is modified (vertices are added or removed)
        m_vertexSelection.Destroy();
        CreateManipulators();
    }

    void EditorSplineComponentMode::OnSplineChanged()
    {
        m_vertexSelection.RefreshLocal();
    }

    void EditorSplineComponentMode::OnVertexAdded(size_t index)
    {
        ContainerChanged();

        AZ::SplinePtr spline = nullptr;
        SplineComponentRequestBus::EventResult(
            spline, GetEntityId(), &SplineComponentRequests::GetSpline);

        m_vertexSelection.CreateTranslationManipulator(
            GetEntityComponentIdPair(), AzToolsFramework::g_mainManipulatorManagerId,
            spline->m_vertexContainer.GetVertices()[index], index);
    }

    void EditorSplineComponentMode::OnVertexRemoved(size_t /*index*/)
    {
        ContainerChanged();
    }

    void EditorSplineComponentMode::OnVertexUpdated(size_t /*index*/)
    {
        m_vertexSelection.RefreshLocal();
    }

    void EditorSplineComponentMode::OnVerticesSet(const AZStd::vector<AZ::Vector3>& /*vertices*/)
    {
        ContainerChanged();
    }

    void EditorSplineComponentMode::OnVerticesCleared()
    {
        ContainerChanged();
    }

    void EditorSplineComponentMode::OnTransformChanged(
        const AZ::Transform& /*local*/, const AZ::Transform& world)
    {
        // update the space manipulators are in after the entity has moved
        m_vertexSelection.RefreshSpace(world);
    }

    void EditorSplineComponentMode::OnSplineTypeChanged()
    {
        ContainerChanged();
    }
} // namespace LmbrCentral