356 lines
12 KiB
C#
356 lines
12 KiB
C#
using UnityEngine;
|
||
using System.IO;
|
||
using System.Collections;
|
||
using System;
|
||
using Zenith;
|
||
|
||
public class TransparentCaptureSystem<T> : MonoBehaviour
|
||
{
|
||
[Header("相机设置")] public Camera targetCamera;
|
||
public KeyCode captureKey = KeyCode.C;
|
||
|
||
[Header("原始截图设置")] public int captureWidth = 1080;
|
||
public int captureHeight = 1920;
|
||
|
||
[Header("裁剪设置")] public int cropWidth = 1080;
|
||
public int cropHeight = 1920;
|
||
public Vector2 cropOffset = Vector2.zero; // 裁剪偏移量 (0-1)
|
||
|
||
[Header("透明通道")] public bool preserveAlpha = true;
|
||
public Color backgroundColor = new Color(0, 0, 0, 0); // 透明背景
|
||
|
||
[Header("输出设置")] public string saveFolder = "Screenshots";
|
||
public string fileNamePrefix = "Screenshot";
|
||
public bool saveAsPNG = true;
|
||
public string onlyFileName = "";
|
||
|
||
private RenderTexture _renderTexture;
|
||
private Texture2D _originalTexture;
|
||
private Texture2D _croppedTexture;
|
||
|
||
public Transform uiRoot;
|
||
public static T Instance;
|
||
|
||
private void Awake()
|
||
{
|
||
var allText = File.ReadAllText(Application.streamingAssetsPath + "/picSavePath.txt");
|
||
if (string.IsNullOrEmpty(allText))
|
||
{
|
||
#if !UNITY_EDITOR
|
||
saveFolder = Path.Combine(Application.dataPath, "Screenshots");
|
||
#endif
|
||
}
|
||
else
|
||
{
|
||
saveFolder = allText;
|
||
}
|
||
|
||
Init();
|
||
}
|
||
|
||
void Start()
|
||
{
|
||
if (targetCamera == null) targetCamera = Camera.main;
|
||
}
|
||
|
||
protected virtual void Init()
|
||
{
|
||
}
|
||
|
||
public IEnumerator CaptureAndCrop(Action<byte[]> callback = null)
|
||
{
|
||
// 确保尺寸有效
|
||
ValidateSizes();
|
||
|
||
// 创建原始截图
|
||
yield return StartCoroutine(CaptureOriginalScreenshot());
|
||
|
||
// 裁剪图片（保留透明通道）
|
||
CropWithTransparency();
|
||
|
||
// 保存结果
|
||
SaveCroppedImage();
|
||
|
||
// 清理资源
|
||
CleanupResources();
|
||
callback?.Invoke(bytes);
|
||
}
|
||
// 【核心修改】优化截图渲染流程，避免RT显示在屏幕上
|
||
private IEnumerator CaptureOriginalScreenshot()
|
||
{
|
||
// 1. 保存相机原始状态（增加关键状态保存）
|
||
RenderTexture originalRT = targetCamera.targetTexture;
|
||
CameraClearFlags originalClearFlags = targetCamera.clearFlags;
|
||
Color originalBackgroundColor = targetCamera.backgroundColor;
|
||
bool originalEnabled = targetCamera.enabled; // 保存相机启用状态
|
||
// bool originalRenderInEditMode = targetCamera.renderInEditMode; // 编辑模式渲染状态
|
||
|
||
// 2. 创建临时RenderTexture
|
||
_renderTexture = RenderTexture.GetTemporary(
|
||
captureWidth,
|
||
captureHeight,
|
||
32,
|
||
RenderTextureFormat.ARGB32,
|
||
RenderTextureReadWrite.Linear
|
||
);
|
||
|
||
try
|
||
{
|
||
// 3. 配置相机仅渲染到RT，不显示在屏幕
|
||
targetCamera.enabled = false; // 禁用相机的自动渲染
|
||
// targetCamera.renderInEditMode = true; // 允许编辑模式渲染
|
||
targetCamera.targetTexture = _renderTexture;
|
||
targetCamera.clearFlags = CameraClearFlags.SolidColor;
|
||
targetCamera.backgroundColor = preserveAlpha ? backgroundColor : Color.black;
|
||
|
||
// 4. 手动渲染相机到RT（此操作不会显示在屏幕）
|
||
targetCamera.Render();
|
||
|
||
// 5. 读取RT像素到Texture2D
|
||
RenderTexture.active = _renderTexture;
|
||
_originalTexture = new Texture2D(captureWidth, captureHeight,
|
||
preserveAlpha ? TextureFormat.ARGB32 : TextureFormat.RGB24, false);
|
||
_originalTexture.ReadPixels(new Rect(0, 0, captureWidth, captureHeight), 0, 0);
|
||
_originalTexture.Apply();
|
||
}
|
||
finally
|
||
{
|
||
// 6. 无论是否出错，都强制恢复相机设置
|
||
targetCamera.targetTexture = originalRT;
|
||
targetCamera.clearFlags = originalClearFlags;
|
||
targetCamera.backgroundColor = originalBackgroundColor;
|
||
targetCamera.enabled = originalEnabled;
|
||
// targetCamera.renderInEditMode = originalRenderInEditMode;
|
||
RenderTexture.active = null; // 强制取消RT激活状态
|
||
}
|
||
|
||
yield return null;
|
||
}
|
||
private IEnumerator CaptureOriginalScreenshot1()
|
||
{
|
||
// 创建RenderTexture（带alpha通道）
|
||
_renderTexture = new RenderTexture(captureWidth, captureHeight, 32, RenderTextureFormat.ARGB32);
|
||
_renderTexture.Create();
|
||
|
||
// 保存原始相机设置
|
||
RenderTexture originalRT = targetCamera.targetTexture;
|
||
CameraClearFlags originalClearFlags = targetCamera.clearFlags;
|
||
Color originalBackgroundColor = targetCamera.backgroundColor;
|
||
|
||
// 配置相机透明背景
|
||
targetCamera.targetTexture = _renderTexture;
|
||
targetCamera.clearFlags = CameraClearFlags.SolidColor;
|
||
targetCamera.backgroundColor = preserveAlpha ? backgroundColor : Color.black;
|
||
|
||
// 渲染相机
|
||
targetCamera.Render();
|
||
|
||
// 创建Texture2D（带alpha通道）
|
||
_originalTexture = new Texture2D(captureWidth, captureHeight,
|
||
preserveAlpha ? TextureFormat.ARGB32 : TextureFormat.RGB24,
|
||
false);
|
||
|
||
// 读取像素（包括alpha）
|
||
RenderTexture.active = _renderTexture;
|
||
_originalTexture.ReadPixels(new Rect(0, 0, captureWidth, captureHeight), 0, 0);
|
||
_originalTexture.Apply();
|
||
RenderTexture.active = null;
|
||
|
||
// 恢复相机设置
|
||
targetCamera.targetTexture = originalRT;
|
||
targetCamera.clearFlags = originalClearFlags;
|
||
targetCamera.backgroundColor = originalBackgroundColor;
|
||
|
||
yield return null;
|
||
}
|
||
|
||
string fullPath;
|
||
public string FullPath => fullPath;
|
||
|
||
private void CropWithTransparency()
|
||
{
|
||
if (_originalTexture == null)
|
||
{
|
||
Debug.LogError("原始截图未创建!");
|
||
return;
|
||
}
|
||
|
||
// 计算裁剪起始位置
|
||
int startX = Mathf.FloorToInt((_originalTexture.width - cropWidth) * cropOffset.x);
|
||
int startY = Mathf.FloorToInt((_originalTexture.height - cropHeight) * cropOffset.y);
|
||
|
||
// 确保在有效范围内
|
||
startX = Mathf.Clamp(startX, 0, _originalTexture.width - cropWidth);
|
||
startY = Mathf.Clamp(startY, 0, _originalTexture.height - cropHeight);
|
||
|
||
// 创建裁剪后的纹理（带alpha通道）
|
||
_croppedTexture = new Texture2D(cropWidth, cropHeight,
|
||
preserveAlpha ? TextureFormat.ARGB32 : TextureFormat.RGB24,
|
||
false);
|
||
|
||
// 获取原始纹理的像素数据（包括alpha）
|
||
Color[] pixels = _originalTexture.GetPixels(startX, startY, cropWidth, cropHeight);
|
||
|
||
// 设置到裁剪纹理
|
||
_croppedTexture.SetPixels(pixels);
|
||
_croppedTexture.Apply();
|
||
}
|
||
|
||
// public static void SaveImage(Texture2D tex, string name)
|
||
// {
|
||
// byte[] bytes = tex.EncodeToPNG();
|
||
// var dirPath = "SaveImages/";
|
||
// if (!Directory.Exists(dirPath))
|
||
// {
|
||
// Directory.CreateDirectory(dirPath);
|
||
// }
|
||
// File.WriteAllBytes(dirPath + name + ".png", B83.Image.PNGTools.ChangePPI(bytes, 300, 300));
|
||
// }
|
||
private void SaveCroppedImage()
|
||
{
|
||
if (_croppedTexture == null)
|
||
{
|
||
Debug.LogError("裁剪后的纹理未创建!");
|
||
return;
|
||
}
|
||
|
||
// 创建保存目录
|
||
string folderPath = saveFolder;
|
||
if (!Directory.Exists(folderPath)) Directory.CreateDirectory(folderPath);
|
||
|
||
// 生成文件名
|
||
string timestamp = DateTime.Now.ToString("yyyyMMdd_HHmmss");
|
||
string extension = saveAsPNG ? ".png" : ".jpg";
|
||
string fileName;
|
||
string dpiName;
|
||
if (string.IsNullOrEmpty(onlyFileName))
|
||
fileName = $"{fileNamePrefix}_{cropWidth}x{cropHeight}_{timestamp}{extension}";
|
||
else
|
||
fileName = $"{onlyFileName}{extension}";
|
||
dpiName = $"{fileNamePrefix}_{cropWidth}x{cropHeight}_{timestamp}_DPI_{extension}";
|
||
|
||
fullPath = Path.Combine(folderPath, fileName);
|
||
var dpiFullPath = Path.Combine(folderPath, dpiName);
|
||
|
||
// 编码保存
|
||
byte[] imageBytes;
|
||
|
||
if (saveAsPNG)
|
||
{
|
||
imageBytes = _croppedTexture.EncodeToPNG();
|
||
}
|
||
else
|
||
{
|
||
// JPG不支持透明通道，强制去除alpha
|
||
Texture2D jpgTexture =
|
||
new Texture2D(_croppedTexture.width, _croppedTexture.height, TextureFormat.RGB24, false);
|
||
jpgTexture.SetPixels(_croppedTexture.GetPixels());
|
||
jpgTexture.Apply();
|
||
imageBytes = jpgTexture.EncodeToJPG(90);
|
||
Destroy(jpgTexture);
|
||
}
|
||
|
||
// File.WriteAllBytes(fullPath, imageBytes);
|
||
|
||
ImageDpiProcessor.SaveTextureWithDpi(_croppedTexture, fullPath, 300);
|
||
Debug.Log($"save as: {fullPath}");
|
||
|
||
// 验证透明度
|
||
if (saveAsPNG && preserveAlpha)
|
||
{
|
||
// 检查左上角像素是否透明
|
||
Color topLeft = _croppedTexture.GetPixel(0, _croppedTexture.height - 1);
|
||
Debug.Log($"保存的图片左上角透明度: {topLeft.a}");
|
||
|
||
// 检查中心像素是否透明
|
||
Color center = _croppedTexture.GetPixel(_croppedTexture.width / 2, _croppedTexture.height / 2);
|
||
Debug.Log($"保存的图片中心透明度: {center.a}");
|
||
}
|
||
|
||
Debug.Log($"截图已保存至: {fullPath}");
|
||
bytes = File.ReadAllBytes(fullPath);
|
||
// NetworkUpload.instance.miniCooperData.items[0].preview1 = System.Convert.ToBase64String(bytes);
|
||
|
||
#if UNITY_EDITOR
|
||
UnityEditor.EditorUtility.RevealInFinder(fullPath);
|
||
#endif
|
||
}
|
||
|
||
byte[] bytes;
|
||
|
||
private void ValidateSizes()
|
||
{
|
||
// 确保原始尺寸有效
|
||
captureWidth = Mathf.Clamp(captureWidth, 32, SystemInfo.maxTextureSize);
|
||
captureHeight = Mathf.Clamp(captureHeight, 32, SystemInfo.maxTextureSize);
|
||
|
||
// 确保裁剪尺寸有效
|
||
cropWidth = Mathf.Clamp(cropWidth, 32, captureWidth);
|
||
cropHeight = Mathf.Clamp(cropHeight, 32, captureHeight);
|
||
|
||
// 确保偏移量在0-1范围
|
||
cropOffset.x = Mathf.Clamp01(cropOffset.x);
|
||
cropOffset.y = Mathf.Clamp01(cropOffset.y);
|
||
}
|
||
|
||
private void CleanupResources()
|
||
{
|
||
if (_renderTexture != null)
|
||
{
|
||
Destroy(_renderTexture);
|
||
_renderTexture = null;
|
||
}
|
||
|
||
if (_originalTexture != null)
|
||
{
|
||
Destroy(_originalTexture);
|
||
_originalTexture = null;
|
||
}
|
||
|
||
// 保留裁剪纹理用于预览或后续使用
|
||
// 如果不再需要，可以取消注释下面一行
|
||
Destroy(_croppedTexture);
|
||
}
|
||
|
||
[ContextMenu("立即截图并裁剪")]
|
||
public virtual void CaptureAndCropNow(Action<byte[]> callback = null)
|
||
{
|
||
StartCoroutine(CaptureAndCrop(callback));
|
||
}
|
||
|
||
void OnGUI()
|
||
{
|
||
// 显示原始截图预览
|
||
if (_originalTexture != null)
|
||
{
|
||
GUI.DrawTexture(new Rect(10, 10, 200, 200 * (float)captureHeight / captureWidth),
|
||
_originalTexture, ScaleMode.ScaleToFit);
|
||
GUI.Label(new Rect(10, 220, 200, 30), $"原始尺寸: {captureWidth}x{captureHeight}");
|
||
}
|
||
|
||
// 显示裁剪后截图预览
|
||
if (_croppedTexture != null)
|
||
{
|
||
GUI.DrawTexture(new Rect(Screen.width - 210, 10, 200, 200 * (float)cropHeight / cropWidth),
|
||
_croppedTexture, ScaleMode.ScaleToFit);
|
||
GUI.Label(new Rect(Screen.width - 210, 220, 200, 30), $"裁剪尺寸: {cropWidth}x{cropHeight}");
|
||
|
||
// 显示透明度信息
|
||
if (preserveAlpha)
|
||
{
|
||
GUI.Label(new Rect(Screen.width - 210, 250, 200, 30), "包含透明通道");
|
||
}
|
||
}
|
||
}
|
||
|
||
void OnDestroy()
|
||
{
|
||
CleanupResources();
|
||
|
||
if (_croppedTexture != null)
|
||
{
|
||
Destroy(_croppedTexture);
|
||
_croppedTexture = null;
|
||
}
|
||
}
|
||
} |