初始提交: UE5.3项目基础框架

Plugins/CesiumForUnreal/Source/CesiumRuntime/Private/Tests/CesiumTestHelpers.cpp

@@ -0,0 +1,111 @@
+// Copyright 2020-2024 CesiumGS, Inc. and Contributors
+
+#include "CesiumTestHelpers.h"
+#include "CesiumGeoreference.h"
+#include "Engine/Engine.h"
+
+#if WITH_EDITOR
+#include "Editor/EditorPerformanceSettings.h"
+#include "Interfaces/IPluginManager.h"
+#endif
+
+namespace CesiumTestHelpers {
+
+UWorld* getGlobalWorldContext() {
+  const TIndirectArray<FWorldContext>& worldContexts =
+      GEngine->GetWorldContexts();
+  FWorldContext firstWorldContext = worldContexts[0];
+  return firstWorldContext.World();
+}
+
+void TestRotatorsAreEquivalent(
+    FAutomationTestBase* pSpec,
+    ACesiumGeoreference* pGeoreferenceExpected,
+    const FRotator& rotatorExpected,
+    ACesiumGeoreference* pGeoreferenceActual,
+    const FRotator& rotatorActual) {
+  FVector xEcefExpected =
+      pGeoreferenceExpected->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorExpected.RotateVector(FVector::XAxisVector));
+  FVector yEcefExpected =
+      pGeoreferenceExpected->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorExpected.RotateVector(FVector::YAxisVector));
+  FVector zEcefExpected =
+      pGeoreferenceExpected->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorExpected.RotateVector(FVector::ZAxisVector));
+
+  FVector xEcefActual =
+      pGeoreferenceActual->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorActual.RotateVector(FVector::XAxisVector));
+  FVector yEcefActual =
+      pGeoreferenceActual->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorActual.RotateVector(FVector::YAxisVector));
+  FVector zEcefActual =
+      pGeoreferenceActual->TransformUnrealDirectionToEarthCenteredEarthFixed(
+          rotatorActual.RotateVector(FVector::ZAxisVector));
+
+  pSpec->TestEqual("xEcefActual", xEcefActual, xEcefExpected);
+  pSpec->TestEqual("yEcefActual", yEcefActual, yEcefExpected);
+  pSpec->TestEqual("zEcefActual", zEcefActual, zEcefExpected);
+}
+
+void waitForNextFrame(
+    const FDoneDelegate& done,
+    UWorld* pWorld,
+    float timeoutSeconds) {
+  uint64 start = GFrameCounter;
+  waitFor(done, pWorld, timeoutSeconds, [start]() {
+    uint64 current = GFrameCounter;
+    return current > start;
+  });
+}
+
+FName getUniqueTag(AActor* pActor) {
+  return FName(FString::Printf(TEXT("%lld"), pActor));
+}
+
+FName getUniqueTag(UActorComponent* pComponent) {
+  return FName(FString::Printf(TEXT("%lld"), pComponent));
+}
+
+#if WITH_EDITOR
+namespace {
+size_t timesAllowingEditorTick = 0;
+bool originalEditorTickState = true;
+} // namespace
+#endif
+
+void pushAllowTickInEditor() {
+#if WITH_EDITOR
+  if (timesAllowingEditorTick == 0) {
+    UEditorPerformanceSettings* pSettings =
+        GetMutableDefault<UEditorPerformanceSettings>();
+    originalEditorTickState = pSettings->bThrottleCPUWhenNotForeground;
+    pSettings->bThrottleCPUWhenNotForeground = false;
+  }
+
+  ++timesAllowingEditorTick;
+#endif
+}
+
+void popAllowTickInEditor() {
+#if WITH_EDITOR
+  --timesAllowingEditorTick;
+  if (timesAllowingEditorTick == 0) {
+    UEditorPerformanceSettings* pSettings =
+        GetMutableDefault<UEditorPerformanceSettings>();
+    pSettings->bThrottleCPUWhenNotForeground = originalEditorTickState;
+  }
+#endif
+}
+
+void trackForPlay(AActor* pEditorActor) {
+  pEditorActor->Tags.Add(getUniqueTag(pEditorActor));
+}
+
+void trackForPlay(UActorComponent* pEditorComponent) {
+  trackForPlay(pEditorComponent->GetOwner());
+  pEditorComponent->ComponentTags.Add(getUniqueTag(pEditorComponent));
+}
+
+} // namespace CesiumTestHelpers