85 lines
2.9 KiB
C++
85 lines
2.9 KiB
C++
|
// Fill out your copyright notice in the Description page of Project Settings.
|
|||
|
|
|
|||
|
|
#include "DialogueGraphSchema.h"
|
|||
|
|
#include "Node/DialogueGraphNode_Base.h"
|
|||
|
|
#include "Node/DialogueGraphNode_NPC.h"
|
|||
|
|
#include "Node/DialogueGraphNode_Player.h"
|
|||
|
|
#include "Node/DialogueGraphNode_Root.h"
|
|||
|
|
|
|||
|
|
#define LOCTEXT_NAMESPACE "DialogueGraphSchema"
|
|||
|
|
|
|||
|
|
UEdGraphNode* FDialogueGraphSchemaAction_NewNode::PerformAction(UEdGraph* ParentGraph, UEdGraphPin* FromPin, const FVector2D Location, bool bSelectNewNode)
|
|||
|
|
{
|
|||
|
|
UDialogueGraphNode_Base* NewNode = NewObject<UDialogueGraphNode_Base>(ParentGraph, ClassTemplate);
|
|||
|
|
NewNode->CreateNewGuid();
|
|||
|
|
NewNode->SetPostion(Location);
|
|||
|
|
NewNode->InitializeNodeData();
|
|||
|
|
NewNode->AllocateDefaultPins();
|
|||
|
|
|
|||
|
|
ParentGraph->Modify();
|
|||
|
|
ParentGraph->AddNode(NewNode, true, bSelectNewNode);
|
|||
|
|
|
|||
|
|
// 如果从某个Pin拖拽创建，自动连接
|
|||
|
|
if (FromPin && NewNode->Pins.Num() > 0)
|
|||
|
|
{
|
|||
|
|
for (UEdGraphPin* Pin : NewNode->Pins)
|
|||
|
|
{
|
|||
|
|
if (Pin->Direction != FromPin->Direction)
|
|||
|
|
{
|
|||
|
|
GetDefault<UDialogueGraphSchema>()->TryCreateConnection(FromPin, Pin);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return NewNode;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void UDialogueGraphSchema::CreateDefaultNodesForGraph(UEdGraph& Graph) const
|
|||
|
|
{
|
|||
|
|
FGraphNodeCreator<UDialogueGraphNode_Root> NodeCreator(Graph);
|
|||
|
|
UDialogueGraphNode_Root* MyNode = NodeCreator.CreateNode(true, UDialogueGraphNode_Root::StaticClass());
|
|||
|
|
NodeCreator.Finalize();
|
|||
|
|
SetNodeMetaData(MyNode, FNodeMetadata::DefaultGraphNode);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void UDialogueGraphSchema::GetGraphContextActions(FGraphContextMenuBuilder& ContextMenuBuilder) const
|
|||
|
|
{
|
|||
|
|
// 添加npc对话节点
|
|||
|
|
TSharedPtr<FDialogueGraphSchemaAction_NewNode> NPCNodeAction = MakeShareable(new FDialogueGraphSchemaAction_NewNode(
|
|||
|
|
UDialogueGraphNode_NPC::StaticClass(),
|
|||
|
|
LOCTEXT("DialogueNode_NPC_Category", "NPC Dialogue"),
|
|||
|
|
LOCTEXT("DialogueNode_NPC_Desc", "NPC Dialogue Node"),
|
|||
|
|
LOCTEXT("DialogueNode_NPC_Tooltip", "Create a npc dialogue node")
|
|||
|
|
));
|
|||
|
|
ContextMenuBuilder.AddAction(NPCNodeAction);
|
|||
|
|
// 添加npc对话节点
|
|||
|
|
TSharedPtr<FDialogueGraphSchemaAction_NewNode> PlayerNodeAction = MakeShareable(new FDialogueGraphSchemaAction_NewNode(
|
|||
|
|
UDialogueGraphNode_Player::StaticClass(),
|
|||
|
|
LOCTEXT("DialogueNode_Player_Category", "Player Dialogue"),
|
|||
|
|
LOCTEXT("DialogueNode_Player_Desc", "Player Dialogue Node"),
|
|||
|
|
LOCTEXT("DialogueNode_Player_Tooltip", "Create a Player dialogue node")
|
|||
|
|
));
|
|||
|
|
ContextMenuBuilder.AddAction(PlayerNodeAction);
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const FPinConnectionResponse UDialogueGraphSchema::CanCreateConnection(const UEdGraphPin* A, const UEdGraphPin* B) const
|
|||
|
|
{
|
|||
|
|
// 不能连接自己
|
|||
|
|
if (A->GetOwningNode() == B->GetOwningNode())
|
|||
|
|
{
|
|||
|
|
return FPinConnectionResponse(CONNECT_RESPONSE_DISALLOW, LOCTEXT("ConnectionSameNode", "Cannot connect to self"));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 必须一个输入一个输出
|
|||
|
|
if (A->Direction == B->Direction)
|
|||
|
|
{
|
|||
|
|
return FPinConnectionResponse(CONNECT_RESPONSE_DISALLOW, LOCTEXT("ConnectionSameDirection", "Must connect input to output"));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return FPinConnectionResponse(CONNECT_RESPONSE_MAKE, TEXT(""));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#undef LOCTEXT_NAMESPACE
|