wehub-resource-sync
783 lines
34 KiB
C#
783 lines
34 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using UnityEngine;
|
|
|
|
namespace MCPForUnity.Runtime.Helpers
|
|
//The reason for having another Runtime Utilities in additional to Editor Utilities is to avoid Editor-only dependencies in this runtime code.
|
|
{
|
|
public readonly struct ScreenshotCaptureResult
|
|
{
|
|
public ScreenshotCaptureResult(string fullPath, string projectRelativePath, int superSize)
|
|
: this(fullPath, projectRelativePath, superSize, isAsync: false, imageBase64: null, imageWidth: 0, imageHeight: 0)
|
|
{
|
|
}
|
|
|
|
public ScreenshotCaptureResult(string fullPath, string projectRelativePath, int superSize, bool isAsync)
|
|
: this(fullPath, projectRelativePath, superSize, isAsync, imageBase64: null, imageWidth: 0, imageHeight: 0)
|
|
{
|
|
}
|
|
|
|
public ScreenshotCaptureResult(string fullPath, string projectRelativePath, int superSize, bool isAsync,
|
|
string imageBase64, int imageWidth, int imageHeight)
|
|
{
|
|
FullPath = fullPath;
|
|
ProjectRelativePath = projectRelativePath;
|
|
SuperSize = superSize;
|
|
IsAsync = isAsync;
|
|
ImageBase64 = imageBase64;
|
|
ImageWidth = imageWidth;
|
|
ImageHeight = imageHeight;
|
|
}
|
|
|
|
public string FullPath { get; }
|
|
/// <summary>Path relative to the Unity project root (e.g. "Captures/foo.png"). Suitable for ScreenCapture.CaptureScreenshot.</summary>
|
|
public string ProjectRelativePath { get; }
|
|
public int SuperSize { get; }
|
|
public bool IsAsync { get; }
|
|
/// <summary>Base64-encoded PNG image data. Only populated when include_image is true.</summary>
|
|
public string ImageBase64 { get; }
|
|
public int ImageWidth { get; }
|
|
public int ImageHeight { get; }
|
|
}
|
|
|
|
public static class ScreenshotUtility
|
|
{
|
|
/// <summary>
|
|
/// Built-in default folder for screenshots, project-relative.
|
|
/// Callers can override per-call via the <c>folderOverride</c> parameter on capture methods,
|
|
/// or globally via <c>ScreenshotPreferences</c> in the Editor assembly.
|
|
/// </summary>
|
|
public const string DefaultFolder = "Assets/Screenshots";
|
|
|
|
private static Camera FindAvailableCamera()
|
|
{
|
|
var main = Camera.main;
|
|
if (main != null)
|
|
{
|
|
return main;
|
|
}
|
|
|
|
try
|
|
{
|
|
var cams = UnityFindObjectsCompat.FindAll<Camera>();
|
|
return cams.FirstOrDefault();
|
|
}
|
|
catch
|
|
{
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static ScreenshotCaptureResult CaptureToProjectFolder(
|
|
string fileName = null,
|
|
int superSize = 1,
|
|
bool ensureUniqueFileName = true,
|
|
string folderOverride = null)
|
|
{
|
|
ScreenshotCaptureResult result = PrepareCaptureResult(fileName, superSize, ensureUniqueFileName, folderOverride, isAsync: true);
|
|
// ScreenCapture.CaptureScreenshot accepts paths relative to the project root.
|
|
ScreenCapture.CaptureScreenshot(result.ProjectRelativePath, result.SuperSize);
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Captures a screenshot from a specific camera by rendering into a temporary RenderTexture (works in Edit Mode).
|
|
/// When <paramref name="includeImage"/> is true, the result includes a base64-encoded PNG (optionally
|
|
/// downscaled so the longest edge is at most <paramref name="maxResolution"/>).
|
|
/// </summary>
|
|
public static ScreenshotCaptureResult CaptureFromCameraToProjectFolder(
|
|
Camera camera,
|
|
string fileName = null,
|
|
int superSize = 1,
|
|
bool ensureUniqueFileName = true,
|
|
bool includeImage = false,
|
|
int maxResolution = 0,
|
|
string folderOverride = null)
|
|
{
|
|
if (camera == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(camera));
|
|
}
|
|
|
|
ScreenshotCaptureResult result = PrepareCaptureResult(fileName, superSize, ensureUniqueFileName, folderOverride, isAsync: false);
|
|
int size = result.SuperSize;
|
|
|
|
int width = Mathf.Max(1, camera.pixelWidth > 0 ? camera.pixelWidth : Screen.width);
|
|
int height = Mathf.Max(1, camera.pixelHeight > 0 ? camera.pixelHeight : Screen.height);
|
|
width *= size;
|
|
height *= size;
|
|
|
|
RenderTexture prevRT = camera.targetTexture;
|
|
RenderTexture prevActive = RenderTexture.active;
|
|
var rt = RenderTexture.GetTemporary(width, height, 24, RenderTextureFormat.ARGB32);
|
|
Texture2D tex = null;
|
|
Texture2D downscaled = null;
|
|
string imageBase64 = null;
|
|
int imgW = 0, imgH = 0;
|
|
try
|
|
{
|
|
camera.targetTexture = rt;
|
|
camera.Render();
|
|
|
|
RenderTexture.active = rt;
|
|
tex = new Texture2D(width, height, TextureFormat.RGBA32, false);
|
|
tex.ReadPixels(new Rect(0, 0, width, height), 0, 0);
|
|
tex.Apply();
|
|
|
|
byte[] png = tex.EncodeToPNG();
|
|
File.WriteAllBytes(result.FullPath, png);
|
|
|
|
if (includeImage)
|
|
{
|
|
int targetMax = maxResolution > 0 ? maxResolution : 640;
|
|
if (width > targetMax || height > targetMax)
|
|
{
|
|
downscaled = DownscaleTexture(tex, targetMax);
|
|
byte[] smallPng = downscaled.EncodeToPNG();
|
|
imageBase64 = System.Convert.ToBase64String(smallPng);
|
|
imgW = downscaled.width;
|
|
imgH = downscaled.height;
|
|
}
|
|
else
|
|
{
|
|
imageBase64 = System.Convert.ToBase64String(png);
|
|
imgW = width;
|
|
imgH = height;
|
|
}
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
camera.targetTexture = prevRT;
|
|
RenderTexture.active = prevActive;
|
|
RenderTexture.ReleaseTemporary(rt);
|
|
DestroyTexture(tex);
|
|
DestroyTexture(downscaled);
|
|
}
|
|
|
|
if (includeImage && imageBase64 != null)
|
|
{
|
|
return new ScreenshotCaptureResult(
|
|
result.FullPath, result.ProjectRelativePath, result.SuperSize, false,
|
|
imageBase64, imgW, imgH);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
#if UNITY_EDITOR
|
|
// Synchronously drive a WaitForEndOfFrame ScreenshotCapturer by pumping the editor's
|
|
// player loop. Play-mode only; EditorApplication.Step is a no-op in edit mode.
|
|
private static Texture2D CaptureCompositedAfterFrame(int superSize, int timeoutSteps = 5)
|
|
{
|
|
Texture2D result = null;
|
|
bool done = false;
|
|
bool callerReturned = false;
|
|
ScreenshotCapturer.Begin(superSize, tex =>
|
|
{
|
|
// Late completion after the spin loop timed out: caller will never consume
|
|
// the texture, so destroy it here to avoid leaking a Unity object.
|
|
if (callerReturned)
|
|
{
|
|
if (tex != null) DestroyTexture(tex);
|
|
return;
|
|
}
|
|
result = tex;
|
|
done = true;
|
|
});
|
|
// Step() pauses play mode as a side effect; restore the prior state so a screenshot
|
|
// doesn't leave a running game paused (an already-paused game stays paused).
|
|
bool wasPaused = UnityEditor.EditorApplication.isPaused;
|
|
try
|
|
{
|
|
for (int i = 0; i < timeoutSteps && !done; i++)
|
|
{
|
|
UnityEditor.EditorApplication.Step();
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
if (!wasPaused)
|
|
UnityEditor.EditorApplication.isPaused = false;
|
|
}
|
|
callerReturned = true;
|
|
return result;
|
|
}
|
|
#endif
|
|
|
|
/// <summary>
|
|
/// Captures a screenshot using ScreenCapture.CaptureScreenshotAsTexture, which captures the
|
|
/// final composited frame including UI Toolkit overlays, post-processing, etc.
|
|
/// Falls back to camera-based capture if ScreenCapture returns null at runtime.
|
|
/// </summary>
|
|
public static ScreenshotCaptureResult CaptureComposited(
|
|
string fileName = null,
|
|
int superSize = 1,
|
|
bool ensureUniqueFileName = true,
|
|
bool includeImage = false,
|
|
int maxResolution = 0,
|
|
string folderOverride = null)
|
|
{
|
|
ScreenshotCaptureResult result = PrepareCaptureResult(fileName, superSize, ensureUniqueFileName, folderOverride: folderOverride, isAsync: false);
|
|
Texture2D tex = null;
|
|
Texture2D downscaled = null;
|
|
string imageBase64 = null;
|
|
int imgW = 0, imgH = 0;
|
|
try
|
|
{
|
|
#if UNITY_EDITOR
|
|
// In play mode, inline ScreenCapture reads a backbuffer before UITK has
|
|
// composited; route through WaitForEndOfFrame instead.
|
|
tex = Application.isPlaying
|
|
? CaptureCompositedAfterFrame(result.SuperSize)
|
|
: ScreenCapture.CaptureScreenshotAsTexture(result.SuperSize);
|
|
#else
|
|
tex = ScreenCapture.CaptureScreenshotAsTexture(result.SuperSize);
|
|
#endif
|
|
if (tex == null)
|
|
{
|
|
// Fallback to camera-based if ScreenCapture fails
|
|
var cam = FindAvailableCamera();
|
|
if (cam != null)
|
|
return CaptureFromCameraToProjectFolder(cam, fileName, superSize, ensureUniqueFileName,
|
|
includeImage, maxResolution, folderOverride: folderOverride);
|
|
throw new InvalidOperationException("ScreenCapture.CaptureScreenshotAsTexture returned null and no fallback camera available.");
|
|
}
|
|
|
|
int width = tex.width;
|
|
int height = tex.height;
|
|
|
|
byte[] png = tex.EncodeToPNG();
|
|
File.WriteAllBytes(result.FullPath, png);
|
|
|
|
if (includeImage)
|
|
{
|
|
int targetMax = maxResolution > 0 ? maxResolution : 640;
|
|
if (width > targetMax || height > targetMax)
|
|
{
|
|
downscaled = DownscaleTexture(tex, targetMax);
|
|
byte[] smallPng = downscaled.EncodeToPNG();
|
|
imageBase64 = System.Convert.ToBase64String(smallPng);
|
|
imgW = downscaled.width;
|
|
imgH = downscaled.height;
|
|
}
|
|
else
|
|
{
|
|
imageBase64 = System.Convert.ToBase64String(png);
|
|
imgW = width;
|
|
imgH = height;
|
|
}
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
DestroyTexture(tex);
|
|
DestroyTexture(downscaled);
|
|
}
|
|
|
|
if (includeImage && imageBase64 != null)
|
|
{
|
|
return new ScreenshotCaptureResult(
|
|
result.FullPath, result.ProjectRelativePath, result.SuperSize, false,
|
|
imageBase64, imgW, imgH);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Renders a camera to a Texture2D without saving to disk. Used for multi-angle captures.
|
|
/// Returns the base64-encoded PNG, downscaled to fit within <paramref name="maxResolution"/>.
|
|
/// </summary>
|
|
public static (string base64, int width, int height) RenderCameraToBase64(Camera camera, int maxResolution = 640)
|
|
{
|
|
if (camera == null) throw new ArgumentNullException(nameof(camera));
|
|
|
|
int width = Mathf.Max(1, camera.pixelWidth > 0 ? camera.pixelWidth : Screen.width);
|
|
int height = Mathf.Max(1, camera.pixelHeight > 0 ? camera.pixelHeight : Screen.height);
|
|
|
|
RenderTexture prevRT = camera.targetTexture;
|
|
RenderTexture prevActive = RenderTexture.active;
|
|
var rt = RenderTexture.GetTemporary(width, height, 24, RenderTextureFormat.ARGB32);
|
|
Texture2D tex = null;
|
|
Texture2D downscaled = null;
|
|
try
|
|
{
|
|
camera.targetTexture = rt;
|
|
camera.Render();
|
|
|
|
RenderTexture.active = rt;
|
|
tex = new Texture2D(width, height, TextureFormat.RGBA32, false);
|
|
tex.ReadPixels(new Rect(0, 0, width, height), 0, 0);
|
|
tex.Apply();
|
|
|
|
int targetMax = maxResolution > 0 ? maxResolution : 640;
|
|
if (width > targetMax || height > targetMax)
|
|
{
|
|
downscaled = DownscaleTexture(tex, targetMax);
|
|
string b64 = System.Convert.ToBase64String(downscaled.EncodeToPNG());
|
|
return (b64, downscaled.width, downscaled.height);
|
|
}
|
|
else
|
|
{
|
|
string b64 = System.Convert.ToBase64String(tex.EncodeToPNG());
|
|
return (b64, width, height);
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
camera.targetTexture = prevRT;
|
|
RenderTexture.active = prevActive;
|
|
RenderTexture.ReleaseTemporary(rt);
|
|
DestroyTexture(tex);
|
|
DestroyTexture(downscaled);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Renders a camera to a Texture2D without saving to disk.
|
|
/// Caller owns the returned texture and must destroy it.
|
|
/// </summary>
|
|
public static Texture2D RenderCameraToTexture(Camera camera, int maxResolution = 640)
|
|
{
|
|
if (camera == null) throw new ArgumentNullException(nameof(camera));
|
|
|
|
int width = Mathf.Max(1, camera.pixelWidth > 0 ? camera.pixelWidth : Screen.width);
|
|
int height = Mathf.Max(1, camera.pixelHeight > 0 ? camera.pixelHeight : Screen.height);
|
|
|
|
RenderTexture prevRT = camera.targetTexture;
|
|
RenderTexture prevActive = RenderTexture.active;
|
|
var rt = RenderTexture.GetTemporary(width, height, 24, RenderTextureFormat.ARGB32);
|
|
Texture2D tex = null;
|
|
try
|
|
{
|
|
camera.targetTexture = rt;
|
|
camera.Render();
|
|
|
|
RenderTexture.active = rt;
|
|
tex = new Texture2D(width, height, TextureFormat.RGBA32, false);
|
|
tex.ReadPixels(new Rect(0, 0, width, height), 0, 0);
|
|
tex.Apply();
|
|
|
|
int targetMax = maxResolution > 0 ? maxResolution : 640;
|
|
if (width > targetMax || height > targetMax)
|
|
{
|
|
var downscaled = DownscaleTexture(tex, targetMax);
|
|
DestroyTexture(tex);
|
|
tex = null;
|
|
return downscaled;
|
|
}
|
|
var result = tex;
|
|
tex = null; // transfer ownership to caller
|
|
return result;
|
|
}
|
|
finally
|
|
{
|
|
camera.targetTexture = prevRT;
|
|
RenderTexture.active = prevActive;
|
|
RenderTexture.ReleaseTemporary(rt);
|
|
DestroyTexture(tex);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Composites a list of tile textures into a single contact-sheet grid image.
|
|
/// Labels are drawn as white text on a dark banner at the bottom of each tile.
|
|
/// Returns base64 PNG plus dimensions. Destroys all input tile textures.
|
|
/// </summary>
|
|
public static (string base64, int width, int height) ComposeContactSheet(
|
|
List<Texture2D> tiles, List<string> labels, int padding = 4)
|
|
{
|
|
if (tiles == null || tiles.Count == 0)
|
|
throw new ArgumentException("No tiles to compose.", nameof(tiles));
|
|
|
|
int tileW = tiles[0].width;
|
|
int tileH = tiles[0].height;
|
|
int count = tiles.Count;
|
|
|
|
// Calculate grid: prefer wider than tall (cols >= rows)
|
|
int cols = Mathf.CeilToInt(Mathf.Sqrt(count));
|
|
int rows = Mathf.CeilToInt((float)count / cols);
|
|
|
|
int labelHeight = Mathf.Max(14, tileH / 12);
|
|
int cellW = tileW + padding;
|
|
int cellH = tileH + labelHeight + padding;
|
|
|
|
int sheetW = cols * cellW + padding;
|
|
int sheetH = rows * cellH + padding;
|
|
|
|
Texture2D sheet = null;
|
|
try
|
|
{
|
|
sheet = new Texture2D(sheetW, sheetH, TextureFormat.RGBA32, false);
|
|
|
|
// Build the full sheet in a Color32[] buffer, then upload once
|
|
var bgColor = new Color32(30, 30, 30, 255);
|
|
Color32[] sheetPixels = new Color32[sheetW * sheetH];
|
|
for (int i = 0; i < sheetPixels.Length; i++) sheetPixels[i] = bgColor;
|
|
|
|
// Track label draw requests so we can apply them after the bulk upload
|
|
var labelDraws = new List<(string text, int x, int y, int h)>();
|
|
|
|
for (int idx = 0; idx < count; idx++)
|
|
{
|
|
int col = idx % cols;
|
|
int row = idx / cols;
|
|
// Place tiles top-left to bottom-right (Unity Texture2D y=0 is bottom)
|
|
int x = padding + col * cellW;
|
|
int y = sheetH - padding - (row + 1) * cellH + padding;
|
|
|
|
// Copy tile pixels row-by-row using bulk operations
|
|
Color32[] tilePixels = tiles[idx].GetPixels32();
|
|
for (int ty = 0; ty < tileH; ty++)
|
|
{
|
|
int srcOffset = ty * tileW;
|
|
int dstOffset = (y + labelHeight + ty) * sheetW + x;
|
|
System.Array.Copy(tilePixels, srcOffset, sheetPixels, dstOffset, tileW);
|
|
}
|
|
|
|
// Draw label banner (dark background strip below tile)
|
|
var bannerColor = new Color32(20, 20, 20, 220);
|
|
for (int ly = 0; ly < labelHeight; ly++)
|
|
{
|
|
int dstOffset = (y + ly) * sheetW + x;
|
|
for (int lx = 0; lx < tileW; lx++)
|
|
{
|
|
sheetPixels[dstOffset + lx] = bannerColor;
|
|
}
|
|
}
|
|
|
|
// Queue label text drawing (applied after bulk pixel upload)
|
|
if (labels != null && idx < labels.Count && !string.IsNullOrEmpty(labels[idx]))
|
|
{
|
|
labelDraws.Add((labels[idx], x + 3, y + 2, labelHeight - 4));
|
|
}
|
|
}
|
|
|
|
// Upload all tile + banner pixels in one SetPixels32 call
|
|
sheet.SetPixels32(sheetPixels);
|
|
|
|
// Draw label text on top (small glyph-based writes, negligible cost)
|
|
foreach (var (text, lx, ly, lh) in labelDraws)
|
|
{
|
|
DrawText(sheet, text, lx, ly, lh, Color.white);
|
|
}
|
|
|
|
sheet.Apply();
|
|
|
|
byte[] png = sheet.EncodeToPNG();
|
|
string b64 = System.Convert.ToBase64String(png);
|
|
return (b64, sheetW, sheetH);
|
|
}
|
|
finally
|
|
{
|
|
foreach (var tile in tiles) DestroyTexture(tile);
|
|
DestroyTexture(sheet);
|
|
}
|
|
}
|
|
|
|
private static void DrawText(Texture2D tex, string text, int startX, int startY, int charHeight, Color color)
|
|
{
|
|
// Simple 5x7 bitmap font for basic ASCII characters
|
|
int charWidth = Mathf.Max(4, charHeight * 5 / 7);
|
|
int spacing = Mathf.Max(1, charWidth / 5);
|
|
int x = startX;
|
|
|
|
foreach (char c in text)
|
|
{
|
|
if (x + charWidth > tex.width) break;
|
|
ulong glyph = GetGlyph(c);
|
|
if (glyph != 0)
|
|
{
|
|
for (int row = 0; row < 7; row++)
|
|
{
|
|
for (int col = 0; col < 5; col++)
|
|
{
|
|
bool on = ((glyph >> ((6 - row) * 5 + (4 - col))) & 1) == 1;
|
|
if (!on) continue;
|
|
// Scale the 5x7 glyph to charWidth x charHeight
|
|
int px0 = x + col * charWidth / 5;
|
|
int px1 = x + (col + 1) * charWidth / 5;
|
|
int py0 = startY + (6 - row) * charHeight / 7;
|
|
int py1 = startY + (7 - row) * charHeight / 7;
|
|
for (int py = py0; py < py1 && py < tex.height; py++)
|
|
for (int px = px0; px < px1 && px < tex.width; px++)
|
|
tex.SetPixel(px, py, color);
|
|
}
|
|
}
|
|
}
|
|
x += charWidth + spacing;
|
|
}
|
|
}
|
|
|
|
private static ulong GetGlyph(char c)
|
|
{
|
|
// 5x7 pixel font stored as 35-bit values (row0=bits34-30 ... row6=bits4-0)
|
|
// Each row is 5 wide, MSB=left. Row 0 is top.
|
|
switch (char.ToUpperInvariant(c))
|
|
{
|
|
case 'A': return 0b01110_10001_10001_11111_10001_10001_10001UL;
|
|
case 'B': return 0b11110_10001_10001_11110_10001_10001_11110UL;
|
|
case 'C': return 0b01110_10001_10000_10000_10000_10001_01110UL;
|
|
case 'D': return 0b11100_10010_10001_10001_10001_10010_11100UL;
|
|
case 'E': return 0b11111_10000_10000_11110_10000_10000_11111UL;
|
|
case 'F': return 0b11111_10000_10000_11110_10000_10000_10000UL;
|
|
case 'G': return 0b01110_10001_10000_10111_10001_10001_01110UL;
|
|
case 'H': return 0b10001_10001_10001_11111_10001_10001_10001UL;
|
|
case 'I': return 0b01110_00100_00100_00100_00100_00100_01110UL;
|
|
case 'K': return 0b10001_10010_10100_11000_10100_10010_10001UL;
|
|
case 'L': return 0b10000_10000_10000_10000_10000_10000_11111UL;
|
|
case 'M': return 0b10001_11011_10101_10101_10001_10001_10001UL;
|
|
case 'N': return 0b10001_11001_10101_10011_10001_10001_10001UL;
|
|
case 'O': return 0b01110_10001_10001_10001_10001_10001_01110UL;
|
|
case 'R': return 0b11110_10001_10001_11110_10100_10010_10001UL;
|
|
case 'S': return 0b01110_10001_10000_01110_00001_10001_01110UL;
|
|
case 'T': return 0b11111_00100_00100_00100_00100_00100_00100UL;
|
|
case 'U': return 0b10001_10001_10001_10001_10001_10001_01110UL;
|
|
case 'V': return 0b10001_10001_10001_10001_01010_01010_00100UL;
|
|
case 'W': return 0b10001_10001_10001_10101_10101_11011_10001UL;
|
|
case 'Y': return 0b10001_10001_01010_00100_00100_00100_00100UL;
|
|
case '0': return 0b01110_10011_10101_10101_10101_11001_01110UL;
|
|
case '1': return 0b00100_01100_00100_00100_00100_00100_01110UL;
|
|
case '2': return 0b01110_10001_00001_00010_00100_01000_11111UL;
|
|
case '3': return 0b01110_10001_00001_00110_00001_10001_01110UL;
|
|
case '4': return 0b00010_00110_01010_10010_11111_00010_00010UL;
|
|
case '5': return 0b11111_10000_11110_00001_00001_10001_01110UL;
|
|
case '6': return 0b01110_10001_10000_11110_10001_10001_01110UL;
|
|
case '7': return 0b11111_00001_00010_00100_01000_01000_01000UL;
|
|
case '8': return 0b01110_10001_10001_01110_10001_10001_01110UL;
|
|
case '9': return 0b01110_10001_10001_01111_00001_10001_01110UL;
|
|
case 'J': return 0b00111_00010_00010_00010_00010_10010_01100UL;
|
|
case 'P': return 0b11110_10001_10001_11110_10000_10000_10000UL;
|
|
case 'Q': return 0b01110_10001_10001_10001_10101_10010_01101UL;
|
|
case 'X': return 0b10001_01010_00100_00100_00100_01010_10001UL;
|
|
case 'Z': return 0b11111_00001_00010_00100_01000_10000_11111UL;
|
|
case '-': return 0b00000_00000_00000_11111_00000_00000_00000UL;
|
|
case '_': return 0b00000_00000_00000_00000_00000_00000_11111UL;
|
|
case ' ': return 0UL;
|
|
case '+': return 0b00000_00100_00100_11111_00100_00100_00000UL;
|
|
default: return 0UL;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Downscales a Texture2D so that its longest edge is at most <paramref name="maxEdge"/> pixels.
|
|
/// Uses bilinear filtering via a temporary RenderTexture blit.
|
|
/// Caller must destroy the returned Texture2D.
|
|
/// </summary>
|
|
public static Texture2D DownscaleTexture(Texture2D source, int maxEdge)
|
|
{
|
|
if (source == null)
|
|
throw new System.ArgumentNullException(nameof(source));
|
|
if (maxEdge <= 0)
|
|
throw new System.ArgumentOutOfRangeException(nameof(maxEdge), maxEdge, "maxEdge must be > 0.");
|
|
|
|
int srcW = source.width;
|
|
int srcH = source.height;
|
|
float scale = Mathf.Min((float)maxEdge / srcW, (float)maxEdge / srcH);
|
|
scale = Mathf.Min(scale, 1f); // never upscale
|
|
int dstW = Mathf.Max(1, Mathf.RoundToInt(srcW * scale));
|
|
int dstH = Mathf.Max(1, Mathf.RoundToInt(srcH * scale));
|
|
|
|
RenderTexture prevActive = RenderTexture.active;
|
|
var rt = RenderTexture.GetTemporary(dstW, dstH, 0, RenderTextureFormat.ARGB32);
|
|
rt.filterMode = FilterMode.Bilinear;
|
|
try
|
|
{
|
|
Graphics.Blit(source, rt);
|
|
RenderTexture.active = rt;
|
|
var dst = new Texture2D(dstW, dstH, TextureFormat.RGBA32, false);
|
|
dst.ReadPixels(new Rect(0, 0, dstW, dstH), 0, 0);
|
|
dst.Apply();
|
|
return dst;
|
|
}
|
|
finally
|
|
{
|
|
RenderTexture.active = prevActive;
|
|
RenderTexture.ReleaseTemporary(rt);
|
|
}
|
|
}
|
|
|
|
private static void DestroyTexture(Texture2D tex)
|
|
{
|
|
if (tex == null) return;
|
|
if (Application.isPlaying)
|
|
UnityEngine.Object.Destroy(tex);
|
|
else
|
|
UnityEngine.Object.DestroyImmediate(tex);
|
|
}
|
|
|
|
private static ScreenshotCaptureResult PrepareCaptureResult(string fileName, int superSize, bool ensureUniqueFileName, string folderOverride, bool isAsync)
|
|
{
|
|
int size = Mathf.Max(1, superSize);
|
|
string resolvedName = BuildFileName(fileName);
|
|
string folderAbsolute = ResolveFolderAbsolute(folderOverride);
|
|
Directory.CreateDirectory(folderAbsolute);
|
|
|
|
string fullPath = Path.Combine(folderAbsolute, resolvedName);
|
|
if (ensureUniqueFileName)
|
|
{
|
|
fullPath = EnsureUnique(fullPath);
|
|
}
|
|
|
|
string normalizedFullPath = fullPath.Replace('\\', '/');
|
|
string projectRelativePath = ToProjectRelativePath(normalizedFullPath);
|
|
|
|
return new ScreenshotCaptureResult(normalizedFullPath, projectRelativePath, size, isAsync);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Resolves a folder spec (which may be project-relative like "Assets/Screenshots",
|
|
/// absolute, or null/empty) to an absolute filesystem path inside the project root.
|
|
/// Throws on traversal escape or absolute paths outside the project.
|
|
/// </summary>
|
|
public static string ResolveFolderAbsolute(string folderOverride)
|
|
{
|
|
string projectRoot = GetProjectRootPath().TrimEnd('/');
|
|
string requested = string.IsNullOrWhiteSpace(folderOverride) ? DefaultFolder : folderOverride.Trim();
|
|
requested = requested.Replace('\\', '/').TrimEnd('/');
|
|
|
|
string combined = Path.IsPathRooted(requested)
|
|
? requested
|
|
: Path.Combine(projectRoot, requested);
|
|
|
|
string fullFolder = Path.GetFullPath(combined).Replace('\\', '/').TrimEnd('/');
|
|
string normalizedRoot = projectRoot;
|
|
|
|
// Reject paths that escape the project root (case-insensitive on Windows, exact elsewhere).
|
|
var rootComparison = Application.platform == RuntimePlatform.WindowsEditor
|
|
? StringComparison.OrdinalIgnoreCase
|
|
: StringComparison.Ordinal;
|
|
if (!fullFolder.Equals(normalizedRoot, rootComparison) &&
|
|
!fullFolder.StartsWith(normalizedRoot + "/", rootComparison))
|
|
{
|
|
throw new InvalidOperationException(
|
|
$"Screenshot folder '{folderOverride}' resolves outside the Unity project root ('{fullFolder}'). " +
|
|
$"Use a project-relative path (e.g. 'Assets/Screenshots' or 'Captures').");
|
|
}
|
|
|
|
return fullFolder;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Converts an absolute filesystem path inside the project to a project-relative path
|
|
/// (forward slashes, no leading separator). Returns the input unchanged when it does
|
|
/// not live under the project root.
|
|
/// </summary>
|
|
public static string ToProjectRelativePath(string normalizedFullPath)
|
|
{
|
|
if (string.IsNullOrEmpty(normalizedFullPath)) return normalizedFullPath;
|
|
string projectRoot = GetProjectRootPath();
|
|
string normalized = normalizedFullPath.Replace('\\', '/');
|
|
if (normalized.StartsWith(projectRoot, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return normalized.Substring(projectRoot.Length).TrimStart('/');
|
|
}
|
|
return normalized;
|
|
}
|
|
|
|
/// <summary>
|
|
/// True when <paramref name="projectRelativePath"/> lives under the Unity Assets/ folder
|
|
/// (and therefore should be imported via AssetDatabase).
|
|
/// </summary>
|
|
public static bool IsUnderAssets(string projectRelativePath)
|
|
{
|
|
if (string.IsNullOrEmpty(projectRelativePath)) return false;
|
|
string norm = projectRelativePath.Replace('\\', '/').TrimStart('/');
|
|
return norm.Equals("Assets", StringComparison.OrdinalIgnoreCase)
|
|
|| norm.StartsWith("Assets/", StringComparison.OrdinalIgnoreCase);
|
|
}
|
|
|
|
private static string BuildFileName(string fileName)
|
|
{
|
|
string name = string.IsNullOrWhiteSpace(fileName)
|
|
? $"screenshot-{DateTime.Now:yyyyMMdd-HHmmss}"
|
|
: fileName.Trim();
|
|
|
|
name = SanitizeFileName(name);
|
|
|
|
if (!name.EndsWith(".png", StringComparison.OrdinalIgnoreCase) &&
|
|
!name.EndsWith(".jpg", StringComparison.OrdinalIgnoreCase) &&
|
|
!name.EndsWith(".jpeg", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
name += ".png";
|
|
}
|
|
|
|
return name;
|
|
}
|
|
|
|
private static string SanitizeFileName(string fileName)
|
|
{
|
|
// GetInvalidFileNameChars() doesn't include '\' or '/' on Unix, so a caller-supplied
|
|
// name like "foo\bar" would survive and later get spliced into the directory portion.
|
|
var invalidChars = Path.GetInvalidFileNameChars();
|
|
string cleaned = new string(
|
|
fileName.Select(ch => invalidChars.Contains(ch) || ch == '/' || ch == '\\' ? '_' : ch).ToArray());
|
|
|
|
return string.IsNullOrWhiteSpace(cleaned) ? "screenshot" : cleaned;
|
|
}
|
|
|
|
private static string EnsureUnique(string path)
|
|
{
|
|
if (!File.Exists(path))
|
|
{
|
|
return path;
|
|
}
|
|
|
|
string directory = Path.GetDirectoryName(path) ?? string.Empty;
|
|
string baseName = Path.GetFileNameWithoutExtension(path);
|
|
string extension = Path.GetExtension(path);
|
|
int counter = 1;
|
|
|
|
string candidate;
|
|
do
|
|
{
|
|
candidate = Path.Combine(directory, $"{baseName}-{counter}{extension}");
|
|
counter++;
|
|
} while (File.Exists(candidate));
|
|
|
|
return candidate;
|
|
}
|
|
|
|
private static string GetProjectRootPath()
|
|
{
|
|
string root = Path.GetFullPath(Path.Combine(Application.dataPath, ".."));
|
|
root = root.Replace('\\', '/');
|
|
if (!root.EndsWith("/", StringComparison.Ordinal))
|
|
{
|
|
root += "/";
|
|
}
|
|
return root;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Transient MonoBehaviour that yields WaitForEndOfFrame, calls
|
|
/// ScreenCapture.CaptureScreenshotAsTexture, invokes the callback, and self-destructs.
|
|
/// </summary>
|
|
public sealed class ScreenshotCapturer : MonoBehaviour
|
|
{
|
|
private int _superSize = 1;
|
|
private Action<Texture2D> _onComplete;
|
|
|
|
/// <summary>Spawns a hidden GameObject, attaches a capturer, returns immediately.</summary>
|
|
public static void Begin(int superSize, Action<Texture2D> onComplete)
|
|
{
|
|
var go = new GameObject("__MCP_ScreenshotCapturer__") { hideFlags = HideFlags.HideAndDontSave };
|
|
var c = go.AddComponent<ScreenshotCapturer>();
|
|
c._superSize = Mathf.Max(1, superSize);
|
|
c._onComplete = onComplete;
|
|
}
|
|
|
|
private System.Collections.IEnumerator Start()
|
|
{
|
|
yield return new WaitForEndOfFrame();
|
|
Texture2D tex = null;
|
|
try { tex = ScreenCapture.CaptureScreenshotAsTexture(_superSize); }
|
|
catch (Exception ex) { Debug.LogError($"[MCP for Unity] CaptureScreenshotAsTexture failed: {ex.Message}"); }
|
|
_onComplete?.Invoke(tex);
|
|
Destroy(gameObject);
|
|
}
|
|
}
|
|
}
|