Files
UnrealEngineUWP/Engine/Source/Editor/BlueprintGraph/Private/K2Node_ClassDynamicCast.cpp
Matthew Griffin bb70b349ce Merging CL 2804086 from //UE4/Release-4.11 to Dev-Main (//UE4/Dev-Main) to isolate copyright update
#lockdown Nick.Penwarden

[CL 2819020 by Matthew Griffin in Main branch]
2016-01-07 08:17:16 -05:00

101 lines
3.2 KiB
C++

// Copyright 1998-2016 Epic Games, Inc. All Rights Reserved.
#include "BlueprintGraphPrivatePCH.h"
#include "K2Node_ClassDynamicCast.h"
#include "DynamicCastHandler.h"
#include "GraphEditorSettings.h"
#define LOCTEXT_NAMESPACE "K2Node_ClassDynamicCast"
struct FClassDynamicCastHelper
{
static const FString CastSuccessPinName;
static const FString& GetClassToCastName()
{
static FString ClassToCastName(TEXT("Class"));
return ClassToCastName;
}
};
const FString FClassDynamicCastHelper::CastSuccessPinName = TEXT("bSuccess");
UK2Node_ClassDynamicCast::UK2Node_ClassDynamicCast(const FObjectInitializer& ObjectInitializer)
: Super(ObjectInitializer)
{
}
void UK2Node_ClassDynamicCast::AllocateDefaultPins()
{
// Check to track down possible BP comms corruption
//@TODO: Move this somewhere more sensible
ensure((TargetType == NULL) || (!TargetType->HasAnyClassFlags(CLASS_NewerVersionExists)));
const UEdGraphSchema_K2* K2Schema = Cast<UEdGraphSchema_K2>(GetSchema());
check(K2Schema != nullptr);
if (!K2Schema->DoesGraphSupportImpureFunctions(GetGraph()))
{
bIsPureCast = true;
}
if (!bIsPureCast)
{
// Input - Execution Pin
CreatePin(EGPD_Input, K2Schema->PC_Exec, TEXT(""), NULL, false, false, K2Schema->PN_Execute);
// Output - Execution Pins
CreatePin(EGPD_Output, K2Schema->PC_Exec, TEXT(""), NULL, false, false, K2Schema->PN_CastSucceeded);
CreatePin(EGPD_Output, K2Schema->PC_Exec, TEXT(""), NULL, false, false, K2Schema->PN_CastFailed);
}
// Input - Source type Pin
CreatePin(EGPD_Input, K2Schema->PC_Class, TEXT(""), UObject::StaticClass(), false, false, FClassDynamicCastHelper::GetClassToCastName());
// Output - Data Pin
if (TargetType != NULL)
{
const FString CastResultPinName = K2Schema->PN_CastedValuePrefix + TargetType->GetDisplayNameText().ToString();
CreatePin(EGPD_Output, K2Schema->PC_Class, TEXT(""), *TargetType, false, false, CastResultPinName);
}
UEdGraphPin* BoolSuccessPin = CreatePin(EGPD_Output, K2Schema->PC_Boolean, TEXT(""), nullptr, /*bIsArray =*/false, /*bIsReference =*/false, FClassDynamicCastHelper::CastSuccessPinName);
BoolSuccessPin->bHidden = !bIsPureCast;
UK2Node::AllocateDefaultPins();
}
FLinearColor UK2Node_ClassDynamicCast::GetNodeTitleColor() const
{
return GetDefault<UGraphEditorSettings>()->ClassPinTypeColor;
}
FText UK2Node_ClassDynamicCast::GetNodeTitle(ENodeTitleType::Type TitleType) const
{
if (CachedNodeTitle.IsOutOfDate(this))
{
CachedNodeTitle.SetCachedText(FText::Format(LOCTEXT("NodeTitle", "{0} Class"), Super::GetNodeTitle(TitleType)), this);
}
return CachedNodeTitle;
}
UEdGraphPin* UK2Node_ClassDynamicCast::GetCastSourcePin() const
{
UEdGraphPin* Pin = FindPinChecked(FClassDynamicCastHelper::GetClassToCastName());
check(Pin->Direction == EGPD_Input);
return Pin;
}
UEdGraphPin* UK2Node_ClassDynamicCast::GetBoolSuccessPin() const
{
UEdGraphPin* Pin = FindPin(FClassDynamicCastHelper::CastSuccessPinName);
check((Pin == nullptr) || (Pin->Direction == EGPD_Output));
return Pin;
}
FNodeHandlingFunctor* UK2Node_ClassDynamicCast::CreateNodeHandler(FKismetCompilerContext& CompilerContext) const
{
return new FKCHandler_DynamicCast(CompilerContext, KCST_MetaCast);
}
#undef LOCTEXT_NAMESPACE