/*
 * 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.
 *
 */

#pragma once

#include "EditorWhiteBoxComponentModeTypes.h"
#include "SubComponentModes/EditorWhiteBoxDefaultModeBus.h"

#include <EditorWhiteBoxEdgeModifierBus.h>
#include <EditorWhiteBoxPolygonModifierBus.h>

namespace AZ
{
    class EntityComponentIdPair;
}

namespace AzFramework
{
    struct ViewportInfo;
}

namespace AzToolsFramework
{
    namespace ViewportInteraction
    {
        struct MouseInteractionEvent;
    }

    struct ActionOverride;
} // namespace AzToolsFramework

namespace WhiteBox
{
    class EdgeTranslationModifier;
    class PolygonTranslationModifier;
    class PolygonScaleModifier;
    class EdgeScaleModifier;
    class VertexTranslationModifier;

    //! The default mode of the EditorWhiteBoxComponentMode - this state allows immediate
    //! interaction of polygons and edges.
    class DefaultMode
        : private EditorWhiteBoxDefaultModeRequestBus::Handler
        , private EditorWhiteBoxPolygonModifierNotificationBus::Handler
        , private EditorWhiteBoxEdgeModifierNotificationBus::Handler
    {
    public:
        AZ_CLASS_ALLOCATOR_DECL

        //! A variant to hold either a selected polygon translation, edge translation,
        //! or a vertex selection modifier - default is empty (monostate).
        using SelectedTranslationModifier = AZStd::variant<
            AZStd::monostate, AZStd::unique_ptr<PolygonTranslationModifier>, AZStd::unique_ptr<EdgeTranslationModifier>,
            AZStd::unique_ptr<VertexTranslationModifier>>;

        explicit DefaultMode(const AZ::EntityComponentIdPair& entityComponentIdPair);
        DefaultMode(DefaultMode&&) = default;
        DefaultMode& operator=(DefaultMode&&) = default;
        ~DefaultMode();

        // Interface
        void Refresh();
        AZStd::vector<AzToolsFramework::ActionOverride> PopulateActions(
            const AZ::EntityComponentIdPair& entityComponentIdPair);
        void Display(
            const AZ::EntityComponentIdPair& entityComponentIdPair, const AZ::Transform& worldFromLocal,
            const IntersectionAndRenderData& renderData, const AzFramework::ViewportInfo& viewportInfo,
            AzFramework::DebugDisplayRequests& debugDisplay);
        bool HandleMouseInteraction(
            const AzToolsFramework::ViewportInteraction::MouseInteractionEvent& mouseInteraction,
            const AZ::EntityComponentIdPair& entityComponentIdPair,
            const AZStd::optional<EdgeIntersection>& edgeIntersection,
            const AZStd::optional<PolygonIntersection>& polygonIntersection,
            const AZStd::optional<VertexIntersection>& vertexIntersection);

    private:
        // EditorWhiteBoxDefaultModeRequestBus ...
        void CreatePolygonScaleModifier(const Api::PolygonHandle& polygonHandle) override;
        void CreateEdgeScaleModifier(Api::EdgeHandle edgeHandle) override;
        void AssignSelectedPolygonTranslationModifier() override;
        void AssignSelectedEdgeTranslationModifier() override;
        void AssignSelectedVertexSelectionModifier() override;
        void RefreshPolygonScaleModifier() override;
        void RefreshEdgeScaleModifier() override;
        void RefreshPolygonTranslationModifier() override;
        void RefreshEdgeTranslationModifier() override;
        void RefreshVertexSelectionModifier() override;

        // EditorWhiteBoxPolygonModifierNotificationBus ...
        void OnPolygonModifierUpdatedPolygonHandle(
            const Api::PolygonHandle& previousPolygonHandle, const Api::PolygonHandle& nextPolygonHandle) override;

        // EditorWhiteBoxEdgeModifierNotificationBus ...
        void OnEdgeModifierUpdatedEdgeHandle(
            Api::EdgeHandle previousEdgeHandle, Api::EdgeHandle nextEdgeHandle) override;

        //! Find all potentially interactive edge handles the user can select and manipulate.
        Api::EdgeHandles FindInteractiveEdgeHandles(const WhiteBoxMesh& whiteBox);

        //! The entity component id of the component mode this sub-mode is associated with.
        AZ::EntityComponentIdPair m_entityComponentIdPair;

        AZStd::unique_ptr<PolygonTranslationModifier>
            m_polygonTranslationModifier; //!< The hovered polygon translation modifier.
        AZStd::unique_ptr<EdgeTranslationModifier>
            m_edgeTranslationModifier; //!< The currently instantiated edge translation modifier.
        AZStd::unique_ptr<PolygonScaleModifier>
            m_polygonScaleModifier; //!< The currently instantiated polygon scale modifier.
        AZStd::unique_ptr<EdgeScaleModifier> m_edgeScaleModifier; //!< The currently instantiated edge scale modifier.
        AZStd::unique_ptr<VertexTranslationModifier>
            m_vertexTranslationModifier; //!< The currently instantiated vertex selection modifier.
        SelectedTranslationModifier m_selectedModifier; //!< The type of selected translation modifier.
    };
} // namespace WhiteBox