o3de/Gems/GraphCanvas/Code/Source/Components/Slots/Data/DataSlotComponent.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

#include <Components/Slots/SlotComponent.h>

#include <GraphCanvas/Components/Slots/Data/DataSlotBus.h>
#include <GraphCanvas/Components/Nodes/NodeBus.h>

namespace GraphCanvas
{
    class DataSlotComponent
        : public SlotComponent
        , public DataSlotRequestBus::Handler
    {
    public:
        AZ_COMPONENT(DataSlotComponent, "{DB13C73D-2453-44F8-BB38-316C90264B73}", SlotComponent);
        static void Reflect(AZ::ReflectContext* reflectContext);
        
        static AZ::Entity* CreateDataSlot(const AZ::EntityId& nodeId, const DataSlotConfiguration& dataSlotConfiguration);
        
        DataSlotComponent();
        DataSlotComponent(const DataSlotConfiguration& dataSlotConfiguration);
        ~DataSlotComponent();
        
        // Component
        void Init() override;
        void Activate() override;
        void Deactivate() override;
        ////

        // SlotRequestBus
        void DisplayProposedConnection(const AZ::EntityId& connectionId, const Endpoint& endpoint) override;
        void RemoveProposedConnection(const AZ::EntityId& connectionId, const Endpoint& endpoint) override;

        void AddConnectionId(const AZ::EntityId& connectionId, const Endpoint& endpoint) override;
        void RemoveConnectionId(const AZ::EntityId& connectionId, const Endpoint& endpoint) override;

        void SetNode(const AZ::EntityId& nodeId) override;

        SlotConfiguration* CloneSlotConfiguration() const override;
        ////

        // DataSlotRequestBus
        bool ConvertToReference(bool isNewSlot = false) override;
        bool CanConvertToReference(bool isNewSlot = false) const override;

        bool ConvertToValue() override;
        bool CanConvertToValue() const override;

        bool IsUserSlot() const override;

        DataSlotType GetDataSlotType() const override;
        DataValueType GetDataValueType() const override;

        AZ::Uuid GetDataTypeId() const override;
        void SetDataTypeId(AZ::Uuid typeId) override;

        const Styling::StyleHelper* GetDataColorPalette() const override;
        
        size_t GetContainedTypesCount() const override;
        AZ::Uuid GetContainedTypeId(size_t index) const override;
        const Styling::StyleHelper* GetContainedTypeColorPalette(size_t index) const override;

        void SetDataAndContainedTypeIds(AZ::Uuid typeId, const AZStd::vector<AZ::Uuid>& typeIds, DataValueType valueType) override;
        ////

    protected:
        void UpdateDisplay();
        void RestoreDisplay(bool updateDisplay = false);

        // SlotComponent
        void OnFinalizeDisplay() override;
        ////

        void UpdatePropertyDisplayState();

    private:
        DataSlotComponent(const DataSlotComponent&) = delete;
        DataSlotComponent& operator=(const DataSlotComponent&) = delete;
        AZ::Entity* ConstructConnectionEntity(const Endpoint& sourceEndpoint, const Endpoint& targetEndpoint, bool createModelConnection) override;

        bool            m_canConvertSlotTypes;
        DataSlotType    m_dataSlotType;
        DataValueType   m_valueType;

        bool            m_isUserSlot;

        AZ::Uuid                m_dataTypeId;
        AZStd::vector<AZ::Uuid> m_containedTypeIds;

        AZ::EntityId    m_displayedConnection;

        // Cached information for the display
        DataSlotType    m_previousDataSlotType;
    };
}