Unity3D截屏工具

1.主要涉及到的知识：Unity3D中Texture2D类操作，开启协程；System.IO文件流操作，Delegate信息回调。

用途：用于屏幕截图

简介：CaptureScreenTool是一个工具类，调用CaptureScreenTool.Instance.saveScreenPic(Delegate callback,ImgType type=ImgType.PNG));可以截屏操作

思路：读取屏幕帧缓存，创建Texture，然后文件写入帧缓存格式化后数据bytes！

文件操作最好写一个文件管理工具类，这里只写出写入byte数组的方法。一次把数据全写入（可以循环分次写入）

1. #region FileControl
2.     class FIleMag
3.     {
4.         public static bool saveBytes(string path, byte[] bytes)
5.         {
6.             bool res = false;
7.             if (path == null || bytes == null)
8.                 return false;
9. 
   
10. 
    
11.             if (!Directory.Exists(Path.GetDirectoryName(path)))//得到文件的所在目录的全路径，没有该目录就创建
12.                 Directory.CreateDirectory(path);
13.             FileStream fs = null;
14.             try
15.             {
16.                 fs = new FileStream(path, FileMode.Create);//创建新的文件
17.                 fs.Write(bytes, 0, bytes.Length);//一次写入所有byte[]
18.                 fs.Flush();
19.                 fs.Close();
20.                 res = true;
21.             }
22.             catch (Exception ex)
23.             {
24.                 Debug.Log(ex);
25.                 return false;
26.             }
27.             return res;
28.         }
29.     }
30.     #endregion

由于开启协程只能在主线程中开启，所以当必须创建一个GameObject再挂载脚本。使用单例的创建模式，在Instance是对GameObject进行创建，当截屏完成是，对GameObject进行删除（你也可以在3-5秒后再删除，免于频繁截屏创建销毁的开销）

1. public class CaptureScreenTool : MonoBehaviour {
2.     private static CaptureScreenTool instance = null;
3.     private bool isRunning = false;//是否正在保存，防止开启多个协程多次截屏
4.     private Delegate callBack = null;//设置外部回调
5.     private ImgType curType = ImgType.PNG;//设置保存格式
6.     private const string saveDirectoryName = "captureData";//文件夹
7.     public enum ImgType
8.     {
9.         PNG,
10.         JPG
11.     }
12. 
    
13. 
    
14.     public static CaptureScreenTool Instance
15.     {
16.         get
17.         {
18.             if (instance == null)
19.             {
20. //初始化创建GameObject物品，挂载脚步，这样才能开启协程
21.                 GameObject temp = new GameObject();
22.                 instance = temp.AddComponent<CaptureScreenTool>();
23.             }
24.             return instance;
25.         }
26.     }
27.     private IEnumerator captureScreenPic(Delegate callBack)
28.     {
29.         yield return new WaitForEndOfFrame();//等待帧缓存结束
30.         int width=Screen.width;
31.         int height=Screen.height;
32. 
    
33. //创建一个Texture2D纹理，可查看手册
34.         Texture2D pic = new Texture2D(width, height, TextureFormat.RGB24, false);
35.         pic.ReadPixels(new Rect(0, 0, width, height), 0, 0, true);
36.         byte[] imageBytes = null;
37.         string imageName = null;
38.         switch (curType)
39.         {
40.             case ImgType.PNG:
41.                 imageName = System.DateTime.Now.ToFileTime().ToString() + ".png";//根据时间设置文件名，防止重复，也可反推日期
42.                 imageBytes = pic.EncodeToPNG();
43.                 break;
44.             case ImgType.JPG:
45.                 imageName = System.DateTime.Now.ToFileTime().ToString() + ".jpg";
46.                 imageBytes = pic.EncodeToJPG();
47.                 break;
48.         }
49.         if (imageBytes == null || imageName == null)
50.             StopCoroutine(captureScreenPic(callBack));
51.         string path=Application.persistentDataPath+"/"+saveDirectoryName+"/"+imageName;
52. //保存文件操作
53.         if (FIleMag.saveBytes(path, imageBytes))
54.         {
55.             if (callBack != null)
56.                 callBack.DynamicInvoke(path);//回调外部方法，返回路径，可用于获取文件路径进行分享什么的操作
57.         }
58. 
    
59. //销毁操作
60.         callBack = null;
61.         Destroy(gameObject);
62.         instance = null;
63.         isRunning = false;
64. 
    
65. 
    
66.     }
67. 
    
68. //外部调用方法，如CaptureScreenTool.Instance.saveScreenPic((Action<string>)save);
69.     public void saveScreenPic(Delegate callback, ImgType type = ImgType.PNG)
70.     {
71.         if (isRunning)
72.             return;
73. 
    
74.         Debug.Log("saveScreenPic");
75.         curType = type;
76.         StartCoroutine(captureScreenPic(callback));
77. 
    
78.     }
79. }

此方法只对整个屏幕进行截屏，不能对单个摄像机层次剔除进行分别的截屏