DialogueGraph.h

Go to the documentation of this file.

// Copyright Csaba Molnar, Daniel Butum. All Rights Reserved.
#pragma once
 
#include "CoreMinimal.h"
#include "EdGraph/EdGraph.h"
 
#include "DlgDialogue.h"
 
#include "DialogueGraph.generated.h"
 
class UDlgNode;
class UDialogueGraphNode_Base;
class UDialogueGraphNode;
class UDialogueGraphNode_Root;
class UDialogueGraphNode_Edge;
class UDialogueGraphSchema;
 
UCLASS()
class UDialogueGraph : public UEdGraph
{
    GENERATED_BODY()
 
public:
    // Begin UObject Interface.
    bool Modify(bool bAlwaysMarkDirty = true) override;
 
    // Begin UEdGraph
    bool RemoveGraphNode(UEdGraphNode* NodeToRemove);
 
    // Begin Own methods
    UDlgDialogue* GetDialogue() const
    {
        // Unreal engine magic, get the object that owns this graph, that is our Dialogue.
        return CastChecked<UDlgDialogue>(GetOuter());
    }
 
    UDialogueGraphNode_Root* GetRootGraphNode() const;
 
    const TArray<UEdGraphNode*>& GetAllGraphNodes() const { return Nodes;  }
 
    TArray<UDialogueGraphNode_Base*> GetAllBaseDialogueGraphNodes() const;
 
    TArray<UDialogueGraphNode*> GetAllDialogueGraphNodes() const;
 
    TArray<UDialogueGraphNode_Edge*> GetAllEdgeDialogueGraphNodes() const;
 
    void CreateGraphNodesFromDialogue();
 
    void LinkGraphNodesFromDialogue() const;
 
    void AutoPositionGraphNodes() const;
 
    void RemoveAllNodes();
 
    const UDialogueGraphSchema* GetDialogueGraphSchema() const;
 
private:
    UDialogueGraph(const FObjectInitializer& ObjectInitializer);
 
    // Link the specified node to all it's children
    void LinkGraphNodeToChildren(
        const TArray<UDlgNode*>& NodesDialogue,
        const UDlgNode& NodeDialogue,
        UDialogueGraphNode* NodeGraph
    ) const;
};