EnumUtilities.h

Go to the documentation of this file.

// Copyright (c) 2023 FrostBit Software Lab at the Lapland University of Applied Sciences
//
// This work is licensed under the terms of the MIT license.
// For a copy, see <https://opensource.org/licenses/MIT>.
 
#pragma once
 
#include "CoreMinimal.h"
#include "UObject/Class.h"
#include "Kismet/BlueprintFunctionLibrary.h"
 
#include "Agrarsense/Walker/WalkerParameters.h"
#include "Agrarsense/Foliage/FoliageTypes.h"
#include "Agrarsense/Prop/PropTypes.h"
#include "Agrarsense/Sensor/SensorTypes.h"
#include "Agrarsense/Vehicle/VehicleTypes.h"
#include "Agrarsense/Settings/QualityLevel.h"
#include "Agrarsense/Settings/AntiAliasing.h"
 
#include "EnumUtilities.generated.h"
 
UCLASS()
class AGRARSENSE_API UEnumUtilities : public UBlueprintFunctionLibrary
{
 
    GENERATED_BODY()
 
public:
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EFoliageTypes ConvertStringToFoliageType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertFoliageTypeToString(EFoliageTypes FoliageType);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EPropTypes ConvertStringToPropType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertPropTypeToString(EPropTypes PropType);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static ESensorTypes ConvertStringToSensorType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertSensorTypeToString(ESensorTypes Sensortype);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EVehicleTypes ConvertStringToVehicleType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertVehicleTypeToString(EVehicleTypes VehicleType);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EWalkerType ConvertStringToWalkerType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertWalkerTypeToString(EWalkerType WalkerType);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EWalkerAction ConvertStringToWalkerActionType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertWalkerActionTypeToString(EWalkerAction WalkerAction);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EWalkerEndAction ConvertStringToWalkerEndActionType(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertWalkerEndActionTypeToString(EWalkerEndAction WalkerEndAction);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EQualityLevel ConvertStringToQualityLevel(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertQualityLevelToString(EQualityLevel QualityLevel);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static EAntiAliasing ConvertStringToAntiAliasingMethod(const FString& String);
 
    UFUNCTION(BlueprintCallable, Category = "Agrarsense Conversion Utilities")
    static FString ConvertAntiAliasingMethodToString(EAntiAliasing AntiAliasingMethod);
 
    template <typename TEnum>
    static TMap<TEnum, FString> CreateEnumStringMap(const FString& EnumTypeName)
    {
        TMap<TEnum, FString> EnumMap;
 
        // EnumTypeName must be in the following format: "/Script/ProjectName.EnumClassName"
        UEnum* Enum = FindObject<UEnum>(nullptr, *EnumTypeName, true);
        if (Enum)
        {
            for (int32 EnumIndex = 0; EnumIndex < Enum->NumEnums(); EnumIndex++)
            {
                TEnum EnumValue = static_cast<TEnum>(Enum->GetValueByIndex(EnumIndex));
                FString EnumName = Enum->GetNameStringByIndex(EnumIndex);
 
                EnumMap.Add(EnumValue, EnumName);
            }
        }
 
        return EnumMap;
    }
 
    template <typename TEnum>
    static TMap<FString, TEnum> CreateStringEnumMap(const FString& EnumTypeName)
    {
        TMap<FString, TEnum> EnumMap;
 
        // EnumTypeName must be in the following format: "/Script/ProjectName.EnumClassName"
        UEnum* Enum = FindObject<UEnum>(nullptr, *EnumTypeName, true);
        if (Enum)
        {
            for (int32 EnumIndex = 0; EnumIndex < Enum->NumEnums(); EnumIndex++)
            {
                TEnum EnumValue = static_cast<TEnum>(Enum->GetValueByIndex(EnumIndex));
                FString EnumName = Enum->GetNameStringByIndex(EnumIndex);
 
                EnumMap.Add(EnumName, EnumValue);
            }
        }
 
        return EnumMap;
    }
 
private:
    template <typename TEnum>
    static TEnum ConvertStringToEnum(const FString& EnumString, const FString& EnumTypeName)
    {
        UEnum* Enum = FindObject<UEnum>(nullptr, *EnumTypeName, true);
        if (Enum)
        {
            int32 EnumIndex = Enum->GetIndexByName(FName(*EnumString));
            if (EnumIndex != INDEX_NONE)
            {
                return static_cast<TEnum>(Enum->GetValueByIndex(EnumIndex));
            }
        }
 
        return TEnum(0); // Default value if not found
    }
};