UnrealEngineUWP/Engine/Source/Developer/RigVMDeveloper/Private/RigVMModel/Nodes/RigVMVariableNode.cpp

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

#include "RigVMModel/Nodes/RigVMVariableNode.h"

#include "RigVMModel/RigVMGraph.h"

const FString URigVMVariableNode::VariableName = TEXT("Variable");
const FString URigVMVariableNode::ValueName = TEXT("Value");

URigVMVariableNode::URigVMVariableNode()
{
}

FString URigVMVariableNode::GetNodeTitle() const
{
	if (IsGetter())
	{
		return FString::Printf(TEXT("Get %s"), *GetVariableName().ToString());
	}
	return FString::Printf(TEXT("Set %s"), *GetVariableName().ToString());
}

FName URigVMVariableNode::GetVariableName() const
{
	URigVMPin* VariablePin = FindPin(VariableName);
	if (VariablePin == nullptr)
	{
		return NAME_None;
	}
	return *VariablePin->GetDefaultValue();
}

bool URigVMVariableNode::IsGetter() const
{
	URigVMPin* ValuePin = FindPin(ValueName);
	if (ValuePin == nullptr)
	{
		return false;
	}
	return ValuePin->GetDirection() == ERigVMPinDirection::Output;
}

bool URigVMVariableNode::IsExternalVariable() const
{
	return !IsLocalVariable() && !IsInputArgument();
}

bool URigVMVariableNode::IsLocalVariable() const
{
	const FName CurrentVariableName = GetVariableName();
	
	if(URigVMGraph* Graph = GetGraph())
	{
		const TArray<FRigVMGraphVariableDescription>& LocalVariables = Graph->GetLocalVariables();
		for (const FRigVMGraphVariableDescription& LocalVariable : LocalVariables)
		{
			if(LocalVariable.Name == CurrentVariableName)
			{
				return true;
			}
		}
	}
	return false;
}

bool URigVMVariableNode::IsInputArgument() const
{
	const FName CurrentVariableName = GetVariableName();
	
	if(URigVMGraph* Graph = GetGraph())
	{
		if (URigVMFunctionEntryNode* EntryNode = Graph->GetEntryNode())
		{
			for (const URigVMPin* Pin : EntryNode->GetPins())
			{
				if(Pin->GetFName() == CurrentVariableName)
				{
					return true;
				}
			}
		}
	}
	return false;
}

FString URigVMVariableNode::GetCPPType() const
{
	URigVMPin* ValuePin = FindPin(ValueName);
	if (ValuePin == nullptr)
	{
		return FString();
	}
	return ValuePin->GetCPPType();
}

UObject* URigVMVariableNode::GetCPPTypeObject() const
{
	URigVMPin* ValuePin = FindPin(ValueName);
	if (ValuePin == nullptr)
	{
		return nullptr;
	}
	return ValuePin->GetCPPTypeObject();
}

FString URigVMVariableNode::GetDefaultValue() const
{

	URigVMPin* ValuePin = FindPin(ValueName);
	if (ValuePin == nullptr)
	{
		return FString();
	}
	return ValuePin->GetDefaultValue();
}

FRigVMGraphVariableDescription URigVMVariableNode::GetVariableDescription() const
{
	FRigVMGraphVariableDescription Variable;
	Variable.Name = GetVariableName();
	Variable.CPPType = GetCPPType();
	Variable.CPPTypeObject = GetCPPTypeObject();
	Variable.DefaultValue = GetDefaultValue();
	return Variable;
}

URigVMPin* URigVMVariableNode::GetVariableNamePin() const
{
	return FindPin(VariableName);
}

URigVMPin* URigVMVariableNode::GetValuePin() const
{
	return FindPin(ValueName);
}