DlgTreeViewParticipantProperties.h

Go to the documentation of this file.

// Copyright Csaba Molnar, Daniel Butum. All Rights Reserved.
#pragma once
 
#include <type_traits>
#include "CoreMinimal.h"
 
#include "DlgManager.h"
#include "TreeViewHelpers/DlgTreeViewVariableProperties.h"
#include "TreeViewHelpers/DlgTreeViewHelper.h"
 
template <class VariablePropertyType>
class FDlgTreeViewParticipantProperties
{
    // Compile time Typecheck if subclass
    typedef typename std::enable_if<std::is_base_of<FDlgTreeViewVariableProperties, VariablePropertyType>::value>::type check;
 
public:
    FDlgTreeViewParticipantProperties(const TSet<TWeakObjectPtr<const UDlgDialogue>>& InDialogues) : Dialogues(InDialogues) {}
    FDlgTreeViewParticipantProperties(const TSet<TWeakObjectPtr<const UDlgDialogue>>&& InDialogues) : Dialogues(InDialogues) {}
 
    void Sort()
    {
        Dialogues.Sort(FDlgTreeViewHelper::PredicateSortDialogueWeakPtrAlphabeticallyAscending);
        for (const auto& Pair : Events)
        {
            Pair.Value->Sort();
        }
 
        FDlgHelper::SortDefault(Events);
        FDlgHelper::SortDefault(Conditions);
        FDlgHelper::SortDefault(Integers);
        FDlgHelper::SortDefault(Floats);
        FDlgHelper::SortDefault(Bools);
        FDlgHelper::SortDefault(FNames);
        FDlgHelper::SortDefault(ClassIntegers);
        FDlgHelper::SortDefault(ClassFloats);
        FDlgHelper::SortDefault(ClassBools);
        FDlgHelper::SortDefault(ClassFNames);
        FDlgHelper::SortDefault(ClassFTexts);
    }
 
    // Setters
 
    // Add Dialogue that containt this participant.
    void AddDialogue(TWeakObjectPtr<const UDlgDialogue> Dialogue) { Dialogues.Add(Dialogue); }
 
    // Returns the EventName Property
    TSharedPtr<VariablePropertyType> AddDialogueToEvent(FName EventName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&Events, EventName, Dialogue);
    }
 
    // Returns the EventClass Property
    TSharedPtr<VariablePropertyType> AddDialogueToCustomEvent(UClass* EventClass, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable<UClass*>(&CustomEvents, EventClass, Dialogue);
    }
 
    // Returns the ConditionName Property
    TSharedPtr<VariablePropertyType> AddDialogueToCondition(FName ConditionName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&Conditions, ConditionName, Dialogue);
    }
 
    // Returns the IntName Property
    TSharedPtr<VariablePropertyType> AddDialogueToIntVariable(FName IntVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&Integers, IntVariableName, Dialogue);
    }
 
    // Returns the FloatName Property
    TSharedPtr<VariablePropertyType> AddDialogueToFloatVariable(FName FloatVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&Floats, FloatVariableName, Dialogue);
    }
 
    // Returns the BoolName Property
    TSharedPtr<VariablePropertyType> AddDialogueToBoolVariable(FName BoolVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&Bools, BoolVariableName, Dialogue);
    }
 
    // Returns the FName Property
    TSharedPtr<VariablePropertyType> AddDialogueToFNameVariable(FName FNameVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&FNames, FNameVariableName, Dialogue);
    }
 
    // Returns the IntName Property
    TSharedPtr<VariablePropertyType> AddDialogueToClassIntVariable(FName IntVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&ClassIntegers, IntVariableName, Dialogue);
    }
 
    // Returns the FloatName Property
    TSharedPtr<VariablePropertyType> AddDialogueToClassFloatVariable(FName FloatVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&ClassFloats, FloatVariableName, Dialogue);
    }
 
    // Returns the BoolName Property
    TSharedPtr<VariablePropertyType> AddDialogueToClassBoolVariable(FName BoolVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&ClassBools, BoolVariableName, Dialogue);
    }
 
    // Returns the FName Property
    TSharedPtr<VariablePropertyType> AddDialogueToClassFNameVariable(FName FNameVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&ClassFNames, FNameVariableName, Dialogue);
    }
 
    // Returns the FText Property *
    TSharedPtr<VariablePropertyType> AddDialogueToClassFTextVariable(FName FTextVariableName, TWeakObjectPtr<const UDlgDialogue> Dialogue)
    {
        return AddDialogueToVariable(&ClassFTexts, FTextVariableName, Dialogue);
    }
 
    // Getters
    const TSet<TWeakObjectPtr<const UDlgDialogue>>& GetDialogues() const { return Dialogues; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetEvents() const { return Events; }
    const TMap<UClass*, TSharedPtr<VariablePropertyType>>& GetCustomEvents() const { return CustomEvents; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetConditions() const { return Conditions; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetIntegers() const { return Integers; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetFloats() const { return Floats; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetBools() const { return Bools; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetFNames() const { return FNames; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetClassIntegers() const { return ClassIntegers; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetClassFloats() const { return ClassFloats; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetClassBools() const { return ClassBools; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetClassFNames() const { return ClassFNames; }
    const TMap<FName, TSharedPtr<VariablePropertyType>>& GetClassFTexts() const { return ClassFTexts; }
 
    bool HasDialogueValues() const
    {
        return Integers.Num() > 0 || Floats.Num() > 0 || Bools.Num() > 0 || FNames.Num() > 0;
    }
 
    bool HasClassVariables() const
    {
        return ClassIntegers.Num() > 0 || ClassFloats.Num() > 0 || ClassBools.Num() > 0 || ClassFNames.Num() > 0 || ClassFTexts.Num() > 0;
    }
 
    bool HasDialogues() const { return Dialogues.Num() > 0; }
    bool HasEvents() const { return Events.Num() > 0; }
    bool HasCustomEvents() const { return CustomEvents.Num() > 0; }
    bool HasConditions() const { return Conditions.Num() > 0; }
    bool HasIntegers() const { return Integers.Num() > 0; }
    bool HasFloats() const { return Floats.Num() > 0; }
    bool HasBools() const { return Bools.Num() > 0; }
    bool HasFNames() const { return FNames.Num() > 0; }
    bool HasClassIntegers() const { return ClassIntegers.Num() > 0; }
    bool HasClassFloats() const { return ClassFloats.Num() > 0; }
    bool HasClassBools() const { return ClassBools.Num() > 0; }
    bool HasClassFNames() const { return ClassFNames.Num() > 0; }
    bool HasClassFTexts() const { return ClassFTexts.Num() > 0; }
 
protected:
    template <typename KeyType>
    TSharedPtr<VariablePropertyType> AddDialogueToVariable(
        TMap<KeyType, TSharedPtr<VariablePropertyType>>* VariableMap,
        KeyType VariableKeyValue,
        TWeakObjectPtr<const UDlgDialogue> Dialogue
    )
    {
        TSharedPtr<VariablePropertyType>* VariablePropsPtr = VariableMap->Find(VariableKeyValue);
        TSharedPtr<VariablePropertyType> VariableProps;
        if (VariablePropsPtr == nullptr)
        {
            // Variable does not exist for participant, create it
            const TSet<TWeakObjectPtr<const UDlgDialogue>> SetArgument{Dialogue};
            VariableProps = MakeShared<VariablePropertyType>(SetArgument);
            VariableMap->Add(VariableKeyValue, VariableProps);
        }
        else
        {
            // exists
            VariableProps = *VariablePropsPtr;
            VariableProps->AddDialogue(Dialogue);
        }
 
        return VariableProps;
    }
 
protected:
    TSet<TWeakObjectPtr<const UDlgDialogue>> Dialogues;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> Events;
 
    TMap<UClass*, TSharedPtr<VariablePropertyType>> CustomEvents;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> Conditions;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> Integers;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> Floats;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> Bools;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> FNames;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> ClassIntegers;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> ClassFloats;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> ClassBools;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> ClassFNames;
 
    TMap<FName, TSharedPtr<VariablePropertyType>> ClassFTexts;
};