/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#pragma once

#include <AzCore/EBus/EBus.h>
#include <AzFramework/Viewport/ViewportId.h>
#include <AzFramework/Viewport/MultiViewportController.h>
#include <AzCore/Math/Vector3.h>

#include <Atom/RPI.Public/Base.h>

#include <QtCore/qnamespace.h>
#include <QPoint>

namespace AzFramework
{
    struct ScreenPoint;
}

namespace SandboxEditor
{
    class OrbitCameraControls
    : public AZ::EBusTraits
    {
    public:
        //////////////////////////////////////////////////////////////////////////
        // EBusTraits overrides
        static const AZ::EBusHandlerPolicy HandlerPolicy = AZ::EBusHandlerPolicy::Multiple;
        static const AZ::EBusAddressPolicy AddressPolicy = AZ::EBusAddressPolicy::ById;
        using BusIdType = AzFramework::ViewportId;
        //////////////////////////////////////////////////////////////////////////

        virtual void SetOrbitDistance(float orbitDistance [[maybe_unused]]) {;}
    };
    using OrbitCameraControlsBus = AZ::EBus<OrbitCameraControls>;

    class LegacyViewportCameraControllerInstance;
    using LegacyViewportCameraController = AzFramework::MultiViewportController<LegacyViewportCameraControllerInstance>;

    class LegacyViewportCameraControllerInstance final
        : public AzFramework::MultiViewportControllerInstanceInterface<LegacyViewportCameraController>
        , public OrbitCameraControlsBus::Handler
    {
    public:
        LegacyViewportCameraControllerInstance(AzFramework::ViewportId viewport, LegacyViewportCameraController* controller);
        ~LegacyViewportCameraControllerInstance();

        bool HandleInputChannelEvent(const AzFramework::ViewportControllerInputEvent& event) override;
        void ResetInputChannels() override;
        void UpdateViewport(const AzFramework::ViewportControllerUpdateEvent& event) override;

        void SetOrbitDistance(float orbitDistance) override;

    private:
        bool JustAltHeld() const;
        bool NoModifierHeld() const;
        bool AllowDolly() const;
        bool AllowOrbit() const;
        bool AllowPan() const;
        bool InvertPan() const;

        static Qt::KeyboardModifier GetKeyboardModifier(const AzFramework::InputChannel& inputChannel);
        static Qt::Key GetKeyboardKey(const AzFramework::InputChannel& inputChannel);

        AZ::RPI::ViewportContextPtr GetViewportContext();

        bool HandleMouseMove(int dx, int dy);
        bool HandleMouseWheel(float zDelta);
        bool IsKeyDown(Qt::Key key) const;
        void UpdateCursorCapture(bool shouldCaptureCursor);

        bool m_inRotateMode = false;
        bool m_inMoveMode = false;
        bool m_inOrbitMode = false;
        bool m_inZoomMode = false;
        int m_totalMouseMoveDelta = 0;
        float m_orbitDistance = 10.f;
        float m_moveSpeed = 1.f;
        AZ::Vector3 m_orbitTarget = {};
        unsigned int m_modifiers = {};
        AZStd::unordered_set<Qt::Key> m_pressedKeys;
        bool m_capturingCursor = false;
    };

} //namespace SandboxEditor