ShaderCompilerMemoryLimit.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
/*=============================================================================
    ShaderCompilerMemoryLimit.h: Wrapper for Windows specific Process Tree functionality.
=============================================================================*/
 
#pragma once
 
#include "CoreMinimal.h"
#include "HAL/PlatformProcess.h"
 
#if PLATFORM_WINDOWS
 
class FWindowsResourceProcessTreeMemory
{
public:
    FWindowsResourceProcessTreeMemory(const FWindowsResourceProcessTreeMemory&) = delete;
    FWindowsResourceProcessTreeMemory& operator = (const FWindowsResourceProcessTreeMemory&) = delete;
 
    FWindowsResourceProcessTreeMemory() = default;
    ~FWindowsResourceProcessTreeMemory() = default;
 
    void AddRootProcessId(uint32 ProcessId);
 
    void Reset();
 
    bool TryGetMemoryUsage(FPlatformProcessMemoryStats& OutStats);
 
private:
    struct FProcessInfo {
        uint32 ProcessId = 0;
        uint32 ParentProcessId = 0;
    };
 
    /* The Process IDs we will measure */
    TSet<uint32> RootProcessIds;
 
    /* All Processes in the system */
    TArray<FProcessInfo> AllProcesses;
 
    /* The root Process IDs and all child Process IDs */
    TSet<uint32> TreeProcessIds;
 
    /* Populate the TreeProcessIds with RootProcessIds and their children. */
    void CollectTreeProcessIds();
 
    /* Recursively add the RootProcessId and children to the TreeProcessIds */
    void CollectTreeProcessIdsRecurse(uint32 RootProcessId);
};
 
using FResourceProcessTreeMemory = FWindowsResourceProcessTreeMemory;
 
#else
 
class FGenericResourceProcessTreeMemory
{
public:
    FGenericResourceProcessTreeMemory(const FGenericResourceProcessTreeMemory&) = delete;
    FGenericResourceProcessTreeMemory& operator = (const FGenericResourceProcessTreeMemory&) = delete;
 
    FGenericResourceProcessTreeMemory() = default;
    ~FGenericResourceProcessTreeMemory() = default;
 
    void AddRootProcessId(uint32 ProcessId);
    void Reset();
    bool TryGetMemoryUsage(FPlatformProcessMemoryStats& OutStats);
};
 
using FResourceProcessTreeMemory = FGenericResourceProcessTreeMemory;
 
#endif // PLATFORM_WINDOWS