Merge pull request #4599 from talha-opteran/patch-1

Jonathan · web-flow · commit efe2d23543da · 2022-07-17T10:26:24.000-03:00
Fix load level for UE editor
diff --git a/Unreal/Plugins/AirSim/Source/AirBlueprintLib.cpp b/Unreal/Plugins/AirSim/Source/AirBlueprintLib.cpp
@@ -76,11 +76,19 @@ void UAirBlueprintLib::setSimulatePhysics(AActor* actor, bool simulate_physics)
 
 bool UAirBlueprintLib::loadLevel(UObject* context, const FString& level_name)
 {
-    FString LongPackageName;
-    const bool success = FPackageName::SearchForPackageOnDisk(level_name, &LongPackageName);
-    if (success) {
-        context->GetWorld()->SetNewWorldOrigin(FIntVector(0, 0, 0));
-        UGameplayStatics::OpenLevel(context->GetWorld(), FName(*LongPackageName), true);
+
+    bool success{ false };
+    // Get name of current level
+    auto currMap = context->GetWorld()->GetMapName();
+    currMap.RemoveFromStart(context->GetWorld()->StreamingLevelsPrefix);
+    // Only load new level if different from current level
+    if (!currMap.Equals(level_name)) {
+        FString LongPackageName;
+        success = FPackageName::SearchForPackageOnDisk(level_name, &LongPackageName);
+        if (success) {
+            context->GetWorld()->SetNewWorldOrigin(FIntVector(0, 0, 0));
+            UGameplayStatics::OpenLevel(context->GetWorld(), FName(*LongPackageName), true);
+        }
     }
     return success;
 }
diff --git a/Unreal/Plugins/AirSim/Source/SimHUD/SimHUD.cpp b/Unreal/Plugins/AirSim/Source/SimHUD/SimHUD.cpp
@@ -22,9 +22,12 @@ void ASimHUD::BeginPlay()
 
     try {
         UAirBlueprintLib::OnBeginPlay();
+        initializeSettings();
         loadLevel();
 
-        initializeSettings();
+        // Prevent a MavLink connection being established if changing levels
+        if (map_changed_) return;
+
         setUnrealEngineSettings();
         createSimMode();
         createMainWidget();
@@ -253,7 +256,7 @@ std::string ASimHUD::getSimModeFromUser()
 
 void ASimHUD::loadLevel()
 {
-    UAirBlueprintLib::RunCommandOnGameThread([&]() { UAirBlueprintLib::loadLevel(this->GetWorld(), FString(AirSimSettings::singleton().level_name.c_str())); }, true);
+    UAirBlueprintLib::RunCommandOnGameThread([&]() { this->map_changed_ = UAirBlueprintLib::loadLevel(this->GetWorld(), FString(AirSimSettings::singleton().level_name.c_str())); }, true);
 }
 
 void ASimHUD::createSimMode()
diff --git a/Unreal/Plugins/AirSim/Source/SimHUD/SimHUD.h b/Unreal/Plugins/AirSim/Source/SimHUD/SimHUD.h
@@ -73,4 +73,5 @@ class AIRSIM_API ASimHUD : public AHUD
     ASimModeBase* simmode_;
 
     APIPCamera* subwindow_cameras_[AirSimSettings::kSubwindowCount];
+    bool map_changed_;
 };
