o3de/Gems/GraphCanvas/Code/Source/Components/NodePropertyDisplays/ComboBoxNodePropertyDisplay.h

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

class QEvent;

#include <AzCore/Component/EntityBus.h>

#include <AzToolsFramework/UI/PropertyEditor/PropertyEntityIdCtrl.hxx>

#include <GraphCanvas/Components/NodePropertyDisplay/NodePropertyDisplay.h>
#include <GraphCanvas/Components/NodePropertyDisplay/ComboBoxDataInterface.h>
#include <GraphCanvas/Components/MimeDataHandlerBus.h>
#include <Widgets/GraphCanvasComboBox.h>

namespace GraphCanvas
{
    class GraphCanvasLabel;
    class ComboBoxGraphicsEventFilter;

    class ComboBoxNodePropertyDisplay
        : public NodePropertyDisplay
        , public GeometryNotificationBus::Handler
        , public ViewNotificationBus::Handler
    {
        friend class ComboBoxGraphicsEventFilter;

    public:
        AZ_CLASS_ALLOCATOR(ComboBoxNodePropertyDisplay, AZ::SystemAllocator, 0);
        ComboBoxNodePropertyDisplay(ComboBoxDataInterface* dataInterface);
        virtual ~ComboBoxNodePropertyDisplay();

        // Will color the outline of the label with the data type of the contained
        // type when enabled.
        void SetDataTypeOutlineEnabled(bool enableOutline);

        // NodePropertyDisplay
        void RefreshStyle() override;
        void UpdateDisplay() override;

        QGraphicsLayoutItem* GetDisabledGraphicsLayoutItem() override;
        QGraphicsLayoutItem* GetDisplayGraphicsLayoutItem() override;
        QGraphicsLayoutItem* GetEditableGraphicsLayoutItem() override;
        ////

        // GeometryNotificationBus
        void OnPositionChanged(const AZ::EntityId& targetEntity, const AZ::Vector2& position) override;
        ////

        // ViewNotificationBus
        void OnZoomChanged(qreal zoomLevel) override;
        ////

        // DataSlotNotifications
        void OnDisplayTypeChanged(const AZ::Uuid& dataTypes, const AZStd::vector<AZ::Uuid>& containerTypes) override;
        void OnDragDropStateStateChanged(const DragDropState& dragState) override;
        ////

    protected:

        void OnSlotIdSet() override;

    private:

        void UpdateOutlineColor();
        void ShowContextMenu(const QPoint&);

        void EditStart();
        void EditFinished();
        void SubmitValue();
        void SetupProxyWidget();
        void CleanupProxyWidget();

        void OnMenuAboutToDisplay();
        void UpdateMenuDisplay(const ViewId& viewId, bool forceUpdate = false);

        bool m_valueDirty;
        bool m_menuDisplayDirty;

        ComboBoxDataInterface*          m_dataInterface;
        ComboBoxGraphicsEventFilter*    m_eventFilter;

        GraphCanvasLabel*               m_disabledLabel;
        GraphCanvasComboBox*            m_comboBox;
        QGraphicsProxyWidget*           m_proxyWidget;
        GraphCanvasLabel*               m_displayLabel;

        bool                            m_dataTypeOutlineEnabled;
    };
}