Files
ReshadePluginsCore/libs/UEngine/UETools.hpp

105 lines
3.1 KiB
C++

#pragma once
#include "UMG_classes.hpp"
#include <spdlog/spdlog.h>
#include <unordered_map>
#include <atomic>
inline std::atomic_bool g_Console_Enabled { false };
namespace SDK {
class UEngine;
class UObject;
class UWorld;
class UWidget;
class APawn;
class UGameplayStatics;
class UConsole;
class UInputSettings;
class UKismetStringLibrary;
class UGameInstance;
class ULocalPlayer;
class APlayerController;
}
struct WidgetTrackInfo {
std::string ClassName;
std::string RootWidgetClass;
std::string OuterClass;
bool WasInViewport = false;
int ConstructCount = 0;
int DestructCount = 0;
std::chrono::steady_clock::time_point FirstSeen;
};
static std::unordered_map<std::string, WidgetTrackInfo> g_WidgetTracker;
// Unreal Engine functions
/**
* @brief Gets the current player Pawn from the world.
* @param world Optional UWorld pointer.
* @return Player Pawn or nullptr.
*/
SDK::APawn* GetPawnFromWorld(SDK::UWorld* world = nullptr);
/**
* @brief Reactivate the development console.
* @param logger spdlog.
*/
void ReactivateDevConsole(std::shared_ptr<spdlog::logger> logger);
void ApplyOffsetsSmart(SDK::UWidget* Widget, float Left, float Right, int MaxDepth);
/**
* @brief Apply offsets recursively to UVCanvasPanelSlot.
* @param widget UUserWidget* pointer.
* @param left offset.
* @param right offset.
* @param ExcludeObjects excluded objects in depth check
* @param ExcludeClass excluded class widgets in depth check
*/
void ApplyOffsetsRecursive(SDK::UWidget* widget, float left, float right = 0,
const std::vector<SDK::UObject*>& ExcludeObjects = {},
const std::vector<std::string>& ExcludeClass = {}, int MaxDepth = INT_MAX);
/**
* @brief Apply offsets recursively to Overlay.
* @param widget UUserWidget* pointer.
* @param Offset (left and right offsets).
* @param alignment widget (left, right, center, fill ...)
* @param ExcludeNames excluded class widgets in depth check
* @param MaxDepth Max depth to go through root component
*/
void ApplyOverlayOffsetRecursive(SDK::UWidget* Widget, float left, float right, SDK::EHorizontalAlignment alignment,
const std::vector<std::string>& ExcludeNames = {}, int MaxDepth = INT_MAX);
void FindAndApplyCanvasRecursive(SDK::UWidget* widget, float offset, int MaxDepth = INT_MAX, int currentDepth = 0);
/**
* @brief Tracks potential widgets candidates for HUD & UI at construction
* @param widget UUserWidget* pointer.
*/
void TrackWidgetConstruct(SDK::UUserWidget* widget);
/**
* @brief Tracks potential widgets candidates for HUD & UI at destruction
* @param widget UUserWidget* pointer.
*/
void TrackWidgetDestruct(SDK::UUserWidget* widget);
/**
* @brief Dump and log previously searched UI & HUD wisgets
* @param logger the log object.
*/
void DumpUIAnalysis(std::shared_ptr<spdlog::logger> logger);
/**
* @brief Clear previously tracked widgets
*/
void ClearWidgetTracking();
/**
* @brief Dump all widgets
* @param Widget a UWidget instance to browse
* @param Depth the depth to start from
* @param MaxDepth the maximal depth to reach
*/
void DumpWidgetRecursive(SDK::UWidget* Widget, int Depth = 0, int MaxDepth = 2);