Files
UnrealEngineUWP/Engine/Source/Editor/BlueprintGraph/Private/BlueprintGraphPanelPinFactory.cpp
ryan durand 627baf970a Updating copyright for Engine Editor.
#rnx
#rb none


#ROBOMERGE-SOURCE: CL 10869241 via CL 10869527 via CL 10869904
#ROBOMERGE-BOT: (v613-10869866)

[CL 10870586 by ryan durand in Main branch]
2019-12-26 15:33:43 -05:00

70 lines
2.3 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "BlueprintGraphPanelPinFactory.h"
#include "Engine/CurveTable.h"
#include "Engine/DataTable.h"
#include "Widgets/DeclarativeSyntaxSupport.h"
#include "EdGraphSchema_K2.h"
#include "K2Node_CallFunction.h"
#include "K2Node_GetDataTableRow.h"
#include "SGraphPin.h"
#include "SGraphPinNameList.h"
#include "SGraphPinDataTableRowName.h"
TSharedPtr<class SGraphPin> FBlueprintGraphPanelPinFactory::CreatePin(class UEdGraphPin* InPin) const
{
if (InPin->PinType.PinCategory == UEdGraphSchema_K2::PC_Name)
{
UObject* Outer = InPin->GetOuter();
// Create drop down combo boxes for DataTable and CurveTable RowName pins
const UEdGraphPin* DataTablePin = nullptr;
if (Outer->IsA(UK2Node_CallFunction::StaticClass()))
{
const UK2Node_CallFunction* CallFunctionNode = CastChecked<UK2Node_CallFunction>(Outer);
if (CallFunctionNode)
{
const UFunction* FunctionToCall = CallFunctionNode->GetTargetFunction();
if (FunctionToCall)
{
const FString& DataTablePinName = FunctionToCall->GetMetaData(FBlueprintMetadata::MD_DataTablePin);
DataTablePin = CallFunctionNode->FindPin(DataTablePinName);
}
}
}
else if (Outer->IsA(UK2Node_GetDataTableRow::StaticClass()))
{
const UK2Node_GetDataTableRow* GetDataTableRowNode = CastChecked<UK2Node_GetDataTableRow>(Outer);
DataTablePin = GetDataTableRowNode->GetDataTablePin();
}
if (DataTablePin)
{
if (DataTablePin->DefaultObject != nullptr && DataTablePin->LinkedTo.Num() == 0)
{
if (auto DataTable = Cast<UDataTable>(DataTablePin->DefaultObject))
{
return SNew(SGraphPinDataTableRowName, InPin, DataTable);
}
if (DataTablePin->DefaultObject->IsA(UCurveTable::StaticClass()))
{
UCurveTable* CurveTable = (UCurveTable*)DataTablePin->DefaultObject;
if (CurveTable)
{
TArray<TSharedPtr<FName>> RowNames;
/** Extract all the row names from the RowMap */
for (TMap<FName, FRealCurve*>::TConstIterator Iterator(CurveTable->GetRowMap()); Iterator; ++Iterator)
{
/** Create a simple array of the row names */
TSharedPtr<FName> RowNameItem = MakeShareable(new FName(Iterator.Key()));
RowNames.Add(RowNameItem);
}
return SNew(SGraphPinNameList, InPin, RowNames);
}
}
}
}
}
return nullptr;
}