PathTree.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/Map.h"
#include "Containers/Set.h"
#include "HAL/Platform.h"
#include "UObject/NameTypes.h"
 
template <typename FuncType> class TFunctionRef;
 
class FPathTree
{
public:
    ASSETREGISTRY_API FPathTree();
    
    ASSETREGISTRY_API void EnsureAdditionalCapacity(int32 NumNewPaths);
    
    ASSETREGISTRY_API bool CachePath(FName Path, TFunctionRef<void(FName)> OnPathAdded);
 
    ASSETREGISTRY_API bool RemovePath(FName Path, TFunctionRef<void(FName)> OnPathRemoved);
 
    ASSETREGISTRY_API bool PathExists(FName Path) const;
 
    ASSETREGISTRY_API bool GetAllPaths(TSet<FName>& OutPaths) const;
 
    ASSETREGISTRY_API void EnumerateAllPaths(TFunctionRef<bool(FName)> Callback) const;
 
    ASSETREGISTRY_API bool GetSubPaths(FName BasePath, TSet<FName>& OutPaths, bool bRecurse = true) const;
 
    ASSETREGISTRY_API bool EnumerateSubPaths(FName BasePath, TFunctionRef<bool(FName)> Callback, bool bRecurse = true) const;
 
    ASSETREGISTRY_API void Shrink();
 
    int32 NumPaths() const
    {
        return ParentPathToChildPaths.Num();
    }
 
    SIZE_T GetAllocatedSize(void) const
    {
        SIZE_T AllocatedSize = ParentPathToChildPaths.GetAllocatedSize() + ChildPathToParentPath.GetAllocatedSize();
 
        for (const TPair<FName, TSet<FName>>& Pair : ParentPathToChildPaths)
        {
            AllocatedSize += Pair.Value.GetAllocatedSize();
        }
 
        return AllocatedSize;
    }
 
    static ASSETREGISTRY_API FName NormalizePackagePath(FName In);
 
private:
    TMap<FName, TSet<FName>> ParentPathToChildPaths;
 
    TMap<FName, FName> ChildPathToParentPath;
};