You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
db7a692cc7
changed the access tracking to a bulk operation api to allow for single calls instead of O(n) calls #preflight 63fe5fc7ac6c61e2b553e9a9 [CL 24449378 by joe pribele in ue5-main branch]
124 lines
3.1 KiB
C++
124 lines
3.1 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#if WITH_LOW_LEVEL_TESTS
|
|
|
|
#include "TestHarness.h"
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "UObject/ObjectHandle.h"
|
|
#include "Misc/ScopeLock.h"
|
|
|
|
#if UE_WITH_OBJECT_HANDLE_LATE_RESOLVE
|
|
inline FObjectHandle MakeUnresolvedHandle(const UObject* Obj)
|
|
{
|
|
UE::CoreUObject::Private::FPackedObjectRef PackedObjectRef = UE::CoreUObject::Private::MakePackedObjectRef(Obj);
|
|
return { PackedObjectRef.EncodedRef };
|
|
}
|
|
|
|
#endif
|
|
|
|
class FObjectRefTrackingTestBase
|
|
{
|
|
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.InstallCallbacks();
|
|
}
|
|
|
|
~FSnapshotObjectRefMetrics()
|
|
{
|
|
Test.RemoveCallbacks();
|
|
}
|
|
|
|
void TestNumResolves(const TCHAR* What, uint32 ExpectedDelta)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
TEST_EQUAL(What, OriginalNumResolves + ExpectedDelta, Test.GetNumResolves());
|
|
#endif
|
|
}
|
|
|
|
void TestNumFailedResolves(const TCHAR* What, uint32 ExpectedDelta)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
TEST_EQUAL(What, OriginalNumFailedResolves + ExpectedDelta, Test.GetNumFailedResolves());
|
|
#endif
|
|
}
|
|
|
|
void TestNumReads(const TCHAR* What, uint32 ExpectedDelta, bool bAllowAdditionalReads = false)
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
bool bValue = false;
|
|
if (bAllowAdditionalReads)
|
|
{
|
|
INFO(What);
|
|
CHECK(Test.GetNumReads() >= OriginalNumReads + ExpectedDelta);
|
|
}
|
|
else
|
|
{
|
|
bValue = OriginalNumReads + ExpectedDelta == Test.GetNumReads();
|
|
TEST_TRUE(What, bValue);
|
|
}
|
|
#endif
|
|
}
|
|
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 (!ObjectRef.IsNull() && !Obj)
|
|
{
|
|
NumFailedResolves++;
|
|
}
|
|
}
|
|
static void OnRefRead(TArrayView<const UObject* const> Objects)
|
|
{
|
|
NumReads++;
|
|
}
|
|
#endif
|
|
|
|
static void InstallCallbacks()
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
ResolvedCallbackHandle = UE::CoreUObject::AddObjectHandleReferenceResolvedCallback(OnRefResolved);
|
|
HandleReadCallbackHandle = UE::CoreUObject::AddObjectHandleReadCallback(OnRefRead);
|
|
#endif
|
|
}
|
|
|
|
static void RemoveCallbacks()
|
|
{
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
UE::CoreUObject::RemoveObjectHandleReferenceResolvedCallback(ResolvedCallbackHandle);
|
|
UE::CoreUObject::RemoveObjectHandleReadCallback(HandleReadCallbackHandle);
|
|
#endif
|
|
}
|
|
|
|
#if UE_WITH_OBJECT_HANDLE_TRACKING
|
|
static UE::CoreUObject::FObjectHandleTrackingCallbackId ResolvedCallbackHandle;
|
|
static UE::CoreUObject::FObjectHandleTrackingCallbackId HandleReadCallbackHandle;
|
|
#endif
|
|
static thread_local uint32 NumResolves;
|
|
static thread_local uint32 NumFailedResolves;
|
|
static thread_local uint32 NumReads;
|
|
};
|
|
|
|
#endif |