DialogueGraphNode_Base.h

Go to the documentation of this file.

// Copyright Csaba Molnar, Daniel Butum. All Rights Reserved.
#pragma once
 
#include "CoreTypes.h"
#include "UObject/ObjectMacros.h"
#include "EdGraph/EdGraphNode.h"
 
#include "DialogueEditor/Graph/DialogueGraphSchema.h"
#include "DialogueEditor/Graph/DialogueGraph.h"
 
#include "DialogueGraphNode_Base.generated.h"
 
class UEdGraphPin;
class UEdGraphSchema;
class UDialogueGraphSchema;
 
UCLASS(Abstract)
class UDialogueGraphNode_Base : public UEdGraphNode
{
    GENERATED_BODY()
 
public:
    //~ Begin UObject Interface.
    void PostLoad() override;
 
    void PostDuplicate(bool bDuplicateForPIE) override;
 
    void PostEditImport() override;
 
    // UEdGraphNode interface.
    void PostPlacedNewNode() override { RegisterListeners(); }
 
    void AllocateDefaultPins() override;
 
    void ReconstructNode() override;
 
    bool CanDuplicateNode() const override { return true; }
 
    bool CanUserDeleteNode() const override { return true; }
 
    void PrepareForCopying() override { Super::PrepareForCopying(); }
 
    void RemovePinAt(int32 PinIndex, EEdGraphPinDirection PinDirection) override {}
 
    bool CanSplitPin(const UEdGraphPin* Pin) const override { return false; }
 
    bool CanCreateUnderSpecifiedSchema(const UEdGraphSchema* Schema) const override
    {
        return Schema->IsA(UDialogueGraphSchema::StaticClass());
    }
 
    FString GetDocumentationLink() const override { return TEXT("Shared/DialogueGraphNode"); }
 
    bool ShowPaletteIconOnNode() const override { return true; }
 
    FLinearColor GetNodeTitleColor() const override { return GetNodeBackgroundColor(); }
 
    FSlateIcon GetIconAndTint(FLinearColor& OutColor) const override
    {
        static const FSlateIcon Icon = FSlateIcon(FEditorStyle::GetStyleSetName(), "Graph.StateNode.Icon");
        OutColor = GetNodeBackgroundColor();
        return Icon;
    }
 
    // Begin own functions
    UDialogueGraphNode_Base(const FObjectInitializer& ObjectInitializer);
 
    virtual void PostCopyNode() {}
 
    virtual bool HasOutputConnectionToNode(const UEdGraphNode* TargetNode) const;
 
    virtual bool CanHaveInputConnections() const { return true; }
 
    virtual bool CanHaveOutputConnections() const { return true; }
 
    virtual FLinearColor GetNodeBackgroundColor() const { return FLinearColor::Black; }
 
    virtual void CheckAll() const
    {
#if DO_CHECK
        GetInputPin();
        GetOutputPin();
#endif
    }
 
    virtual FIntPoint GetPosition() const { return FIntPoint(NodePosX, NodePosY); }
 
    virtual void SetPosition(int32 X, int32 Y)
    {
        NodePosX = X;
        NodePosY = Y;
    }
 
    // Compiler methods
    void ClearCompilerMessage();
 
    void SetCompilerWarningMessage(FString Message);
 
    bool HasInputPin() const
    {
        return Pins.IsValidIndex(INDEX_PIN_Input) && Pins[INDEX_PIN_Input] != nullptr &&
               Pins[INDEX_PIN_Input]->Direction == EGPD_Input;
    }
 
    bool HasOutputPin() const
    {
        return Pins.IsValidIndex(INDEX_PIN_Output) && Pins[INDEX_PIN_Output] != nullptr &&
               Pins[INDEX_PIN_Output]->Direction == EGPD_Output;
    }
 
    // @return the input pin for this dialogue Node
    UEdGraphPin* GetInputPin() const
    {
        check(HasInputPin());
        return Pins[INDEX_PIN_Input];
    }
 
    // @return the output pin for this dialogue Node
    UEdGraphPin* GetOutputPin() const
    {
        check(HasOutputPin());
        return Pins[INDEX_PIN_Output];
    }
 
    // Helper method to get directly the Dialogue Graph (which is our parent)
    UDialogueGraph* GetDialogueGraph() const { return CastChecked<UDialogueGraph>(GetGraph()); }
 
    // Helper method to get directly the Dialogue
    UDlgDialogue* GetDialogue() const { return GetDialogueGraph()->GetDialogue(); }
 
    // TODO fix UEdGraphSchema::BreakSinglePinLink, make it to const
    const UDialogueGraphSchema* GetDialogueGraphSchema() const { return GetDialogueGraph()->GetDialogueGraphSchema(); }
 
    TSharedPtr<SGraphNode> GetNodeWidget() const { return DEPRECATED_NodeWidget.Pin(); }
 
protected:
    // Begin own functions
    virtual void CreateInputPin() { unimplemented(); }
 
    virtual void CreateOutputPin() { unimplemented(); }
 
    virtual void OnDialoguePropertyChanged(const FPropertyChangedEvent& PropertyChangedEvent) {}
 
    virtual void RegisterListeners();
 
protected:
    // Constants for the location of the input/output pins in the Pins array
    static constexpr int32 INDEX_PIN_Input = 0;
    static constexpr int32 INDEX_PIN_Output = 1;
};