flockbay 0.10.20 → 0.10.21

package/tools/unreal-mcp/upstream/MCPGameProject/Plugins/UnrealMCP/Source/UnrealMCP/Private/Commands/UnrealMCPEditorCommands.cpp

@@ -27,6 +27,8 @@
 #include "Components/StaticMeshComponent.h"
 #include "Materials/MaterialInterface.h"
 #include "Sound/SoundWave.h"
+#include "IImageWrapper.h"
+#include "IImageWrapperModule.h"
 #include "EditorSubsystem.h"
 #include "Subsystems/EditorActorSubsystem.h"
 #include "Engine/Blueprint.h"
@@ -54,6 +56,31 @@
 #include "Components/SceneCaptureComponent2D.h"
 #include "Engine/TextureRenderTarget2D.h"
 
+static bool SavePngToFile(const FString& FilePath, int32 Width, int32 Height, const TArray<FColor>& Bitmap)
+{
+	IImageWrapperModule& ImageWrapperModule = FModuleManager::LoadModuleChecked<IImageWrapperModule>(FName("ImageWrapper"));
+	TSharedPtr<IImageWrapper> Wrapper = ImageWrapperModule.CreateImageWrapper(EImageFormat::PNG);
+	if (!Wrapper.IsValid())
+	{
+		return false;
+	}
+
+	if (!Wrapper->SetRaw(Bitmap.GetData(), Bitmap.Num() * sizeof(FColor), Width, Height, ERGBFormat::BGRA, 8))
+	{
+		return false;
+	}
+
+	const TArray64<uint8>& Compressed = Wrapper->GetCompressed(100);
+	if (Compressed.Num() <= 0 || Compressed.Num() > MAX_int32)
+	{
+		return false;
+	}
+
+	TArray<uint8> Out;
+	Out.Append(Compressed.GetData(), static_cast<int32>(Compressed.Num()));
+	return FFileHelper::SaveArrayToFile(Out, *FilePath);
+}
+
 FUnrealMCPEditorCommands::FUnrealMCPEditorCommands()
 {
 }
@@ -219,7 +246,7 @@ TSharedPtr<FJsonObject> FUnrealMCPEditorCommands::HandleSaveAll(const TSharedPtr
     }
 
     TArray<UPackage*> DirtyBefore;
-    FEditorFileUtils::GetDirtyPackages(DirtyBefore, /*bIncludeMapPackages=*/true, /*bIncludeContentPackages=*/true);
+    FEditorFileUtils::GetDirtyPackages(DirtyBefore);
 
     TSet<FString> DirtyBeforeNames;
     for (UPackage* Pkg : DirtyBefore)
@@ -239,7 +266,7 @@ TSharedPtr<FJsonObject> FUnrealMCPEditorCommands::HandleSaveAll(const TSharedPtr
     );
 
     TArray<UPackage*> DirtyAfter;
-    FEditorFileUtils::GetDirtyPackages(DirtyAfter, /*bIncludeMapPackages=*/true, /*bIncludeContentPackages=*/true);
+    FEditorFileUtils::GetDirtyPackages(DirtyAfter);
 
     TSet<FString> DirtyAfterNames;
     for (UPackage* Pkg : DirtyAfter)
@@ -293,8 +320,7 @@ TSharedPtr<FJsonObject> FUnrealMCPEditorCommands::HandleSaveAll(const TSharedPtr
         Details->SetArrayField(TEXT("stillDirtyPackages"), StillDirty);
 
         TSharedPtr<FJsonObject> ErrObj = FUnrealMCPCommonUtils::CreateErrorResponse(
-            TEXT("Some packages are still dirty after save_all. They may require manual Save As, source control checkout, or have save errors. Resolve in-editor and retry save_all."),
-        );
+            TEXT("Some packages are still dirty after save_all. They may require manual Save As, source control checkout, or have save errors. Resolve in-editor and retry save_all."));
         ErrObj->SetObjectField(TEXT("details"), Details);
         return ErrObj;
     }
@@ -2445,9 +2471,7 @@ TSharedPtr<FJsonObject> FUnrealMCPEditorCommands::HandleTakeScreenshot(const TSh
 
         CaptureActor->Destroy();
 
-        TArray<uint8> CompressedBitmap;
-        FImageUtils::CompressImageArray(Width, Height, Bitmap, CompressedBitmap);
-        if (!FFileHelper::SaveArrayToFile(CompressedBitmap, *FilePath))
+        if (!SavePngToFile(FilePath, Width, Height, Bitmap))
         {
             return FUnrealMCPCommonUtils::CreateErrorResponse(TEXT("Failed to save screenshot PNG to disk."));
         }
@@ -2472,10 +2496,7 @@ TSharedPtr<FJsonObject> FUnrealMCPEditorCommands::HandleTakeScreenshot(const TSh
 
         if (Viewport->ReadPixels(Bitmap, FReadSurfaceDataFlags(), ViewportRect))
         {
-            TArray<uint8> CompressedBitmap;
-            FImageUtils::CompressImageArray(ViewSize.X, ViewSize.Y, Bitmap, CompressedBitmap);
-
-            if (FFileHelper::SaveArrayToFile(CompressedBitmap, *FilePath))
+            if (SavePngToFile(FilePath, ViewSize.X, ViewSize.Y, Bitmap))
             {
                 TSharedPtr<FJsonObject> ResultObj = MakeShared<FJsonObject>();
                 ResultObj->SetStringField(TEXT("filepath"), FilePath);

package/tools/unreal-mcp/upstream/MCPGameProject/Plugins/UnrealMCP/Source/UnrealMCP/UnrealMCP.Build.cs

@@ -45,6 +45,7 @@ public class UnrealMCP : ModuleRules
 				"EditorScriptingUtilities",
 				"EditorSubsystem",
 				"Landscape",
+				"ImageWrapper",
 				"Slate",
 				"SlateCore",
 				"MessageLog",