95 lines
3.2 KiB
C++
95 lines
3.2 KiB
C++
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "WingServer.h"
|
|
#include "WingHandler.h"
|
|
#include "WingFetcher.h"
|
|
#include "WingFunctionArgs.h"
|
|
#include "WingUtils.h"
|
|
#include "Engine/Blueprint.h"
|
|
#include "EdGraphSchema_K2.h"
|
|
#include "Kismet2/BlueprintEditorUtils.h"
|
|
#include "EventDispatcher_Create.generated.h"
|
|
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
// ---------------------------------------------------------------------------
|
|
|
|
UCLASS()
|
|
class UWing_EventDispatcher_Create : public UObject, public IWingHandler
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
public:
|
|
UPROPERTY(meta=(Description="Blueprint name or package path"))
|
|
FString Blueprint;
|
|
|
|
UPROPERTY(meta=(Description="Name of the new event dispatcher"))
|
|
FString Dispatcher;
|
|
|
|
UPROPERTY(meta=(Description="Arguments expressed as: int x,float y"))
|
|
FString Arguments;
|
|
|
|
virtual FString GetDescription() const override
|
|
{
|
|
return TEXT("Add a new event dispatcher to a Blueprint.");
|
|
}
|
|
|
|
virtual void Handle() override
|
|
{
|
|
WingFetcher F;
|
|
UBlueprint* BP = F.Walk(Blueprint).Cast<UBlueprint>();
|
|
if (!BP) return;
|
|
|
|
// Check for valid proposed name
|
|
if (!WingUtils::FindExactlyNoneNamed(Dispatcher, BP->NewVariables, TEXT("Variable"))) return;
|
|
if (!WingUtils::FindExactlyNoneNamed(Dispatcher, WingUtils::AllGraphs(BP), TEXT("Graph"))) return;
|
|
FString InternalName = WingUtils::CheckProposedName(Dispatcher);
|
|
if (InternalName.IsEmpty()) return;
|
|
FName VarFName(InternalName);
|
|
|
|
// Make sure the argument types are valid.
|
|
if (!WingFunctionArgs::CheckArgs(Arguments)) return;
|
|
|
|
// Add the delegate variable
|
|
FEdGraphPinType DelegateType;
|
|
DelegateType.PinCategory = UEdGraphSchema_K2::PC_MCDelegate;
|
|
if (!FBlueprintEditorUtils::AddMemberVariable(BP, VarFName, DelegateType))
|
|
{
|
|
UWingServer::Printf(TEXT("ERROR: Failed to add event dispatcher '%s' to %s\n"), *Dispatcher, *WingUtils::FormatName(BP));
|
|
return;
|
|
}
|
|
|
|
// Create the signature graph
|
|
const UEdGraphSchema_K2* K2Schema = GetDefault<UEdGraphSchema_K2>();
|
|
UEdGraph* SigGraph = FBlueprintEditorUtils::CreateNewGraph(BP, VarFName,
|
|
UEdGraph::StaticClass(), UEdGraphSchema_K2::StaticClass());
|
|
if (!SigGraph)
|
|
{
|
|
UWingServer::Printf(TEXT("ERROR: Failed to create signature graph for '%s'\n"), *Dispatcher);
|
|
return;
|
|
}
|
|
|
|
SigGraph->bEditable = false;
|
|
K2Schema->CreateDefaultNodesForGraph(*SigGraph);
|
|
K2Schema->CreateFunctionGraphTerminators(*SigGraph, static_cast<UClass*>(nullptr));
|
|
K2Schema->AddExtraFunctionFlags(SigGraph, FUNC_BlueprintCallable | FUNC_BlueprintEvent | FUNC_Public);
|
|
K2Schema->MarkFunctionEntryAsEditable(SigGraph, true);
|
|
BP->DelegateSignatureGraphs.Add(SigGraph);
|
|
|
|
// Store the function arguments
|
|
TWeakObjectPtr<UK2Node_EditablePinBase> EntryNode;
|
|
TWeakObjectPtr<UK2Node_EditablePinBase> ResultNode;
|
|
FBlueprintEditorUtils::GetEntryAndResultNodes(SigGraph, EntryNode, ResultNode);
|
|
if (!EntryNode.IsValid())
|
|
{
|
|
UWingServer::Print(TEXT("ERROR: Entry node not found in delegate signature graph\n"));
|
|
return;
|
|
}
|
|
if (!WingFunctionArgs::SetArgs(EntryNode.Get(), Arguments)) return;
|
|
|
|
UWingServer::Printf(TEXT("Created event dispatcher %s in %s\n"), *Dispatcher, *WingUtils::FormatName(BP));
|
|
}
|
|
};
|