// Copyright Epic Games, Inc. All Rights Reserved.

#include "RigVMModel/Nodes/RigVMEnumNode.h"

const FString URigVMEnumNode::EnumName = TEXT("Enum");
const FString URigVMEnumNode::EnumValueName = TEXT("EnumValue");
const FString URigVMEnumNode::EnumIndexName = TEXT("EnumIndex");

URigVMEnumNode::URigVMEnumNode()
{
}

FString URigVMEnumNode::GetNodeTitle() const
{
	if(UEnum* Enum = GetEnum())
	{
		return FString::Printf(TEXT("Enum %s"), *Enum->GetName());
	}
	return TEXT("Enum");
}

UEnum* URigVMEnumNode::GetEnum() const
{
	return Cast<UEnum>(GetCPPTypeObject());
}

FString URigVMEnumNode::GetCPPType() const
{
	URigVMPin* EnumValuePin = FindPin(EnumValueName);
	if (EnumValuePin == nullptr)
	{
		return FString();
	}
	return EnumValuePin->GetCPPType();
}

UObject* URigVMEnumNode::GetCPPTypeObject() const
{
	URigVMPin* EnumValuePin = FindPin(EnumValueName);
	if (EnumValuePin == nullptr)
	{
		return nullptr;
	}
	return EnumValuePin->GetCPPTypeObject();
}

FString URigVMEnumNode::GetDefaultValue(const URigVMPin::FPinOverride& InOverride) const
{
	URigVMPin* EnumValuePin = FindPin(EnumValueName);
	if (EnumValuePin == nullptr)
	{
		return FString();
	}
	return EnumValuePin->GetDefaultValue(InOverride);
}