/*
 * Copyright (c) Contributors to the Open 3D Engine Project
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include <AzCore/RTTI/BehaviorContext.h>
#include <ScriptCanvas/Core/Core.h>
#include <ScriptCanvas/Execution/RuntimeComponent.h>

#include "Interpreted/ExecutionStateInterpreted.h"
#include "Interpreted/ExecutionStateInterpretedPure.h"
#include "Interpreted/ExecutionStateInterpretedPerActivation.h"
#include "Interpreted/ExecutionStateInterpretedSingleton.h"

#include "ExecutionState.h"

namespace ScriptCanvas
{
    ExecutionStateConfig::ExecutionStateConfig(AZ::Data::Asset<RuntimeAsset> runtimeAsset, RuntimeComponent& component)
        : asset(runtimeAsset)
        , component(component)
        , runtimeData(runtimeAsset.Get()->GetData())
    {}

    ExecutionState::ExecutionState(const ExecutionStateConfig& config)
        : m_component(&config.component)
    {}

    ExecutionStatePtr ExecutionState::Create(const ExecutionStateConfig& config)
    {
        Grammar::ExecutionStateSelection selection = config.runtimeData.m_input.m_executionSelection;

        switch (selection)
        {
        case Grammar::ExecutionStateSelection::InterpretedPure:
            return AZStd::make_shared<ExecutionStateInterpretedPure>(config);

        case Grammar::ExecutionStateSelection::InterpretedPureOnGraphStart:
            return AZStd::make_shared<ExecutionStateInterpretedPureOnGraphStart>(config);

        case Grammar::ExecutionStateSelection::InterpretedObject:
            return AZStd::make_shared<ExecutionStateInterpretedPerActivation>(config);

        case Grammar::ExecutionStateSelection::InterpretedObjectOnGraphStart:
            return AZStd::make_shared<ExecutionStateInterpretedPerActivationOnGraphStart>(config);

        default:
            AZ_Assert(false, "Unsupported ScriptCanvas execution selection");
            return nullptr;
        }
    }

    AZ::Data::AssetId ExecutionState::GetAssetId() const
    {
        return m_component->GetRuntimeDataOverrides().m_runtimeAsset.GetId();
    }

    AZ::EntityId ExecutionState::GetEntityId() const
    {
        return m_component->GetEntityId();
    }

    AZ::ComponentId ExecutionState::GetComponentId() const
    {
        return m_component->GetId();
    }

    GraphIdentifier ExecutionState::GetGraphIdentifier() const
    {
        return  m_component->GetGraphIdentifier();
    }

    GraphIdentifier ExecutionState::GetGraphIdentifier(const AZ::Data::AssetId& id) const
    {
        return GraphIdentifier(AZ::Data::AssetId(id.m_guid, 0), GetComponentId());
    }

    AZ::EntityId ExecutionState::GetScriptCanvasId() const
    {
        return m_component->GetScriptCanvasId();
    }

    const RuntimeDataOverrides& ExecutionState::GetRuntimeDataOverrides() const
    {
        return m_component->GetRuntimeDataOverrides();
    }

    void ExecutionState::Reflect(AZ::ReflectContext* reflectContext)
    {
        if (auto behaviorContext = azrtti_cast<AZ::BehaviorContext*>(reflectContext))
        {
            behaviorContext->Class<ExecutionState>()
                ->Method("GetEntityId", &ExecutionState::GetEntityId)
                ->Method("GetScriptCanvasId", &ExecutionState::GetScriptCanvasId)
                ->Method("ToString", &ExecutionState::ToString)
                ->Attribute(AZ::Script::Attributes::Operator, AZ::Script::Attributes::OperatorType::ToString)
                ;
        }

        ExecutionStateInterpreted::Reflect(reflectContext);
        ExecutionStateInterpretedPerActivation::Reflect(reflectContext);
        ExecutionStateInterpretedPerActivationOnGraphStart::Reflect(reflectContext);
        ExecutionStateInterpretedPure::Reflect(reflectContext);
        ExecutionStateInterpretedPureOnGraphStart::Reflect(reflectContext);
        ExecutionStateInterpretedSingleton::Reflect(reflectContext);
    }

    ExecutionStatePtr ExecutionState::SharedFromThis()
    {
        return shared_from_this();
    }

    ExecutionStateConstPtr ExecutionState::SharedFromThisConst() const
    {
        return shared_from_this();
    }

    AZStd::string ExecutionState::ToString() const
    {
        return AZStd::string::format("ExecutionState[%p]", this);
    }

    ExecutionStateWeakPtr ExecutionState::WeakFromThis()
    {
        return ExecutionStateWeakPtr(this);
    }

    ExecutionStateWeakConstPtr ExecutionState::WeakFromThisConst() const
    {
        return ExecutionStateWeakConstPtr(this);
    }

}