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

// AZ
#include <AzCore/Component/ComponentApplicationBus.h>
#include <AzCore/Serialization/SerializeContext.h>
#include <AzToolsFramework/Entity/EditorEntityInfoBus.h>

// Qt
#include <QString>

// Graph Model
#include <GraphModel/Model/Slot.h>

// Landscape Canvas
#include "BaseNode.h"
#include <Editor/Core/Core.h>
#include <Editor/Core/GraphContext.h>

namespace LandscapeCanvas
{
    void BaseNode::Reflect(AZ::ReflectContext* context)
    {
        AZ::SerializeContext* serializeContext = azrtti_cast<AZ::SerializeContext*>(context);
        if (serializeContext)
        {
            serializeContext->Class<BaseNode, GraphModel::Node>()
                ->Version(0)
                ->Field("m_vegetationEntityId", &BaseNode::m_vegetationEntityId)
                ->Field("m_componentId", &BaseNode::m_componentId)
                ;
        }
    }

    BaseNode::BaseNode(GraphModel::GraphPtr graph)
        : Node(graph)
    {
    }

    void BaseNode::CreateEntityNameSlot()
    {
        // Property field to show the name of the corresponding Vegetation Entity
        GraphModel::DataTypePtr stringDataType = GraphContext::GetInstance()->GetDataType(LandscapeCanvasDataTypeEnum::String);
        RegisterSlot(GraphModel::SlotDefinition::CreateProperty(
            ENTITY_NAME_SLOT_ID,
            ENTITY_NAME_SLOT_LABEL.toUtf8().constData(),
            stringDataType,
            stringDataType->GetDefaultValue(),
            ENTITY_NAME_SLOT_DESCRIPTION.toUtf8().constData()));
    }

    void BaseNode::SetVegetationEntityId(const AZ::EntityId& entityId)
    {
        m_vegetationEntityId = entityId;

        // Update the Entity Name (if we have the property slot)
        GraphModel::SlotPtr slot = GetSlot(ENTITY_NAME_SLOT_ID);
        if (slot)
        {
            AZStd::string name;
            AzToolsFramework::EditorEntityInfoRequestBus::EventResult(name, entityId, &AzToolsFramework::EditorEntityInfoRequestBus::Events::GetName);
            slot->SetValue(AZStd::any(name));
        }
    }

    void BaseNode::SetComponentId(const AZ::ComponentId& componentId)
    {
        m_componentId = componentId;
    }

    AZ::Component* BaseNode::GetComponent() const
    {
        AZ::Entity* gradientEntity = nullptr;
        AZ::ComponentApplicationBus::BroadcastResult(gradientEntity, &AZ::ComponentApplicationRequests::FindEntity, GetVegetationEntityId());
        if (!gradientEntity)
        {
            return nullptr;
        }

        AZ::Component* component = gradientEntity->FindComponent(GetComponentId());
        if (component)
        {
            return component;
        }

        return nullptr;
    }

    bool BaseNode::IsAreaExtender() const
    {
        BaseNodeType baseNodeType = GetBaseNodeType();
        return baseNodeType == LandscapeCanvas::BaseNode::VegetationAreaModifier
            || baseNodeType == LandscapeCanvas::BaseNode::VegetationAreaFilter
            || baseNodeType == LandscapeCanvas::BaseNode::VegetationAreaSelector
            ;
    }
} // namespace LandscapeCanvas