DialogueBrowserTreeVariableProperties.h

Go to the documentation of this file.

// Copyright Csaba Molnar, Daniel Butum. All Rights Reserved.
#pragma once
 
#include "CoreMinimal.h"
 
#include "DlgManager.h"
#include "TreeViewHelpers/DlgTreeViewVariableProperties.h"
 
class UDialogueGraphNode;
class UDialogueGraphNode_Edge;
class UDlgDialogue;
class FDialogueBrowserTreeVariableProperties;
 
class FDialogueBrowserTreeVariableProperties : public FDlgTreeViewVariableProperties
{
    typedef FDialogueBrowserTreeVariableProperties Self;
    typedef FDlgTreeViewVariableProperties Super;
 
public:
    FDialogueBrowserTreeVariableProperties(const TSet<TWeakObjectPtr<const UDlgDialogue>>& InDialogues);
 
    // Dialogues:
    void AddDialogue(TWeakObjectPtr<const UDlgDialogue> Dialogue) override;
 
    // GraphNodes:
    bool HasGraphNodeSet(const FGuid& DialogueGUID) { return GraphNodes.Find(DialogueGUID) != nullptr; }
    TSet<TWeakObjectPtr<const UDialogueGraphNode>>* GetMutableGraphNodeSet(const FGuid& DialogueGUID)
    {
        return GraphNodes.Find(DialogueGUID);
    }
    const TSet<TWeakObjectPtr<const UDialogueGraphNode>>& GetGraphNodeSet(const FGuid& DialogueGUID) const
    {
        auto* SetPtr = GraphNodes.Find(DialogueGUID);
        check(SetPtr);
        return *SetPtr;
    }
 
    // EdgeNodes:
    bool HasEdgeNodeSet(const FGuid& DialogueGUID) { return EdgeNodes.Find(DialogueGUID) != nullptr; }
    TSet<TWeakObjectPtr<const UDialogueGraphNode_Edge>>* GetMutableEdgeNodeSet(const FGuid& DialogueGUID)
    {
        return EdgeNodes.Find(DialogueGUID);
    }
    const TSet<TWeakObjectPtr<const UDialogueGraphNode_Edge>>& GetEdgeNodeSet(const FGuid& DialogueGUID) const
    {
        auto* SetPtr = EdgeNodes.Find(DialogueGUID);
        check(SetPtr);
        return *SetPtr;
    }
 
protected:
    TMap<FGuid, TSet<TWeakObjectPtr<const UDialogueGraphNode>>> GraphNodes;
 
    TMap<FGuid, TSet<TWeakObjectPtr<const UDialogueGraphNode_Edge>>> EdgeNodes;
};