DialogueGraphSchema.h

Go to the documentation of this file.

// Copyright Csaba Molnar, Daniel Butum. All Rights Reserved.
#pragma once
 
#include "CoreTypes.h"
#include "EdGraph/EdGraphSchema.h"
 
#include "Nodes/DlgNode.h"
#include "DialogueEditorUtilities.h"
#include "DialogueGraphConnectionDrawingPolicy.h"
#include "SchemaActions/NewComment_DialogueGraphSchemaAction.h"
 
#include "DialogueGraphSchema.generated.h"
 
class UDialogueGraphNode;
class UGraphNodeContextMenuContext;
class UToolMenu;
class FMenuBuilder;
class FSlateWindowElementList;
class UEdGraph;
 
UCLASS()
class UDialogueGraphSchema : public UEdGraphSchema
{
    GENERATED_BODY()
 
public:
    void GetPaletteActions(FGraphActionMenuBuilder& ActionMenuBuilder) const;
 
    bool ConnectionCausesLoop(const UEdGraphPin* InputPin, const UEdGraphPin* OutputPin) const;
 
    //~ Begin EdGraphSchema Interface
    void GetGraphContextActions(FGraphContextMenuBuilder& ContextMenuBuilder) const override;
 
#if ENGINE_MINOR_VERSION >= 24
    void GetContextMenuActions(UToolMenu* Menu, UGraphNodeContextMenuContext* Context) const override;
#else
    void GetContextMenuActions(
        const UEdGraph* CurrentGraph,
        const UEdGraphNode* InGraphNode,
        const UEdGraphPin* InGraphPin,
        FMenuBuilder* MenuBuilder,
        bool bIsDebugging
    ) const override;
#endif
 
    void CreateDefaultNodesForGraph(UEdGraph& Graph) const override;
 
 
    FPinConnectionResponse MovePinLinks(
        UEdGraphPin& MoveFromPin,
        UEdGraphPin& MoveToPin,
        bool bIsIntermediateMove = false,
        bool bNotifyLinkedNodes = false
    ) const override;
 
    FPinConnectionResponse CopyPinLinks(UEdGraphPin& CopyFromPin, UEdGraphPin& CopyToPin, bool bIsIntermediateCopy = false) const override;
 
    const FPinConnectionResponse CanCreateConnection(const UEdGraphPin* PinA, const UEdGraphPin* PinB) const override;
 
    bool TryCreateConnection(UEdGraphPin* PinA, UEdGraphPin* PinB) const override;
 
    bool CreateAutomaticConversionNodeAndConnections(UEdGraphPin* PinA, UEdGraphPin* PinB) const override;
 
    bool ShouldHidePinDefaultValue(UEdGraphPin* Pin) const override;
 
    void BreakNodeLinks(UEdGraphNode& TargetNode) const override;
 
    void BreakPinLinks(UEdGraphPin& TargetPin, bool bSendsNodeNotifcation) const override;
 
    void BreakSinglePinLink(UEdGraphPin* SourcePin, UEdGraphPin* TargetPin) const override;
 
    void DroppedAssetsOnGraph(const TArray<FAssetData>& Assets, const FVector2D& GraphPosition, UEdGraph* Graph) const override;
 
    void DroppedAssetsOnNode(const TArray<FAssetData>& Assets, const FVector2D& GraphPosition, UEdGraphNode* Node) const override;
 
    int32 GetNodeSelectionCount(const UEdGraph* Graph) const override { return FDialogueEditorUtilities::GetSelectedNodes(Graph).Num(); }
 
    bool ShouldAlwaysPurgeOnModification() const override { return true; }
 
    TSharedPtr<FEdGraphSchemaAction> GetCreateCommentAction() const override
    {
        return TSharedPtr<FEdGraphSchemaAction>(static_cast<FEdGraphSchemaAction*>(new FNewComment_DialogueGraphSchemaAction));
    }
 
    /* Returns new FConnectionDrawingPolicy from this schema */
    FConnectionDrawingPolicy* CreateConnectionDrawingPolicy(
        int32 InBackLayerID,
        int32 InFrontLayerID,
        float InZoomFactor,
        const FSlateRect& InClippingRect,
        FSlateWindowElementList& InDrawElements,
        UEdGraph* InGraphObj
    ) const override
    {
        return new FDialogueGraphConnectionDrawingPolicy(InBackLayerID, InFrontLayerID, InZoomFactor, InClippingRect, InDrawElements, InGraphObj);
    }
    //~ End EdGraphSchema Interface
 
    // Begin own functions
    void BreakLinkTo(UEdGraphPin* TargetPin, UEdGraphPin* ToPin, bool bSendsNodeNotifcation) const;
 
private:
    //~ Begin own functions
    void GetCommentAction(FGraphActionMenuBuilder& ActionMenuBuilder, const UEdGraph* CurrentGraph = nullptr) const;
 
    void GetConvertActions(FGraphActionMenuBuilder& ActionMenuBuilder, const UEdGraph* CurrentGraph) const;
 
    void GetAllDialogueNodeActions(FGraphActionMenuBuilder& ActionMenuBuilder) const;
 
    static void InitDialogueNodeClasses();
 
public:
    // Allowed PinType.PinCategory values
    static const FName PIN_CATEGORY_Input;
    static const FName PIN_CATEGORY_Output;
 
    // Categories for actions
    static const FText NODE_CATEGORY_Dialogue;
    static const FText NODE_CATEGORY_Graph;
    static const FText NODE_CATEGORY_Convert;
 
private:
    static TArray<TSubclassOf<UDlgNode>> DialogueNodeClasses;
 
    static bool bDialogueNodeClassesInitialized;
};