You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
bc9b168688
#rb Chris.Constantinescu #preflight 62458f77292f228e09013681 [CL 19571202 by mark lintott in ue5-main branch]
123 lines
2.8 KiB
C++
123 lines
2.8 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "TestFixtures/CoreTestFixture.h"
|
|
#include "UObject/ObjectHandle.h"
|
|
|
|
class FObjectRefTrackingTestBase : public FCoreTestFixture
|
|
{
|
|
public:
|
|
|
|
uint32 GetNumResolves() const { return NumResolves; }
|
|
uint32 GetNumFailedResolves() const { return NumFailedResolves; }
|
|
uint32 GetNumReads() const { return NumReads; }
|
|
|
|
struct FSnapshotObjectRefMetrics
|
|
{
|
|
public:
|
|
FSnapshotObjectRefMetrics(FObjectRefTrackingTestBase& InTest)
|
|
: Test(InTest)
|
|
, OriginalNumResolves(Test.GetNumResolves())
|
|
, OriginalNumFailedResolves(Test.GetNumFailedResolves())
|
|
, OriginalNumReads(Test.GetNumReads())
|
|
{
|
|
Test.ConditionalInstallCallbacks();
|
|
}
|
|
|
|
bool TestNumResolves(const TCHAR* What, uint32 ExpectedDelta)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
TEST_EQUAL(What, OriginalNumResolves + ExpectedDelta, Test.GetNumResolves());
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
bool TestNumFailedResolves(const TCHAR* What, uint32 ExpectedDelta)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
TEST_EQUAL(What, OriginalNumFailedResolves + ExpectedDelta, Test.GetNumFailedResolves());
|
|
#endif
|
|
return true;
|
|
|
|
}
|
|
|
|
bool TestNumReads(const TCHAR* What, uint32 ExpectedDelta, bool bAllowAdditionalReads = false)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
if (bAllowAdditionalReads)
|
|
{
|
|
TEST_TRUE(What, Test.GetNumReads() >= OriginalNumReads + ExpectedDelta);
|
|
}
|
|
else
|
|
{
|
|
TEST_EQUAL(What, OriginalNumReads + ExpectedDelta, Test.GetNumReads());
|
|
}
|
|
#endif
|
|
return true;
|
|
|
|
}
|
|
private:
|
|
FObjectRefTrackingTestBase& Test;
|
|
uint32 OriginalNumResolves;
|
|
uint32 OriginalNumFailedResolves;
|
|
uint32 OriginalNumReads;
|
|
};
|
|
|
|
private:
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
static void OnRefResolved(const FObjectRef& ObjectRef, UPackage* Pkg, UObject* Obj)
|
|
{
|
|
NumResolves++;
|
|
if (!IsObjectRefNull(ObjectRef) && !Obj)
|
|
{
|
|
NumFailedResolves++;
|
|
}
|
|
|
|
if (PrevResolvedFunc)
|
|
{
|
|
PrevResolvedFunc(ObjectRef, Pkg, Obj);
|
|
}
|
|
}
|
|
static void OnRefRead(UObject* Obj)
|
|
{
|
|
NumReads++;
|
|
if (PrevReadFunc)
|
|
{
|
|
PrevReadFunc(Obj);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static void ConditionalInstallCallbacks()
|
|
{
|
|
static bool bCallbacksInstalled = false;
|
|
static FCriticalSection CallbackInstallationLock;
|
|
|
|
if (bCallbacksInstalled)
|
|
{
|
|
return;
|
|
}
|
|
|
|
FScopeLock ScopeLock(&CallbackInstallationLock);
|
|
if (!bCallbacksInstalled)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
PrevResolvedFunc = SetObjectHandleReferenceResolvedCallback(OnRefResolved);
|
|
PrevReadFunc = SetObjectHandleReadCallback(OnRefRead);
|
|
#endif
|
|
bCallbacksInstalled = true;
|
|
}
|
|
}
|
|
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
static ObjectHandleReferenceResolvedFunction* PrevResolvedFunc;
|
|
static ObjectHandleReadFunction* PrevReadFunc;
|
|
#endif
|
|
static thread_local uint32 NumResolves;
|
|
static thread_local uint32 NumFailedResolves;
|
|
static thread_local uint32 NumReads;
|
|
};
|
|
|