using System;
using System.IO;
using System.Linq;
using UnityEditor;
using UnityEngine;

public class ScreenCaptureTools : EditorWindow
{
string m_path;
Vector2 m_size;

int m_superSize = 1;

[MenuItem("Tools/ScrrenCaptureTools")]
static void GetWindow()
{
    GetWindow<ScreenCaptureTools>();
}

private void OnEnable()
{
    m_size = new Vector2(1920, 1080);
    m_path = Application.dataPath + "/ScreenShot/";
    if (!Directory.Exists(m_path))
        Directory.CreateDirectory(m_path);
}

private void OnGUI()
{
    m_path = EditorGUILayout.TextField("SavePath", m_path);
    m_size = EditorGUILayout.Vector2Field("Size:", m_size);

    var cameras = GameObject.FindObjectsOfType<Camera>().ToList();

    foreach (var c in cameras)
        c.enabled = GUILayout.Toggle(c.enabled, c.gameObject.name);

    if (GUILayout.Button("Custom Capture"))
    {
        var path = string.Format("{0}{1}.png", m_path, DateTime.Now.ToString("yyyy_MM_dd_HH24_mm_ss"));

        var rt = new RenderTexture((int)m_size.x, (int)m_size.y, 24, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Default);

        cameras.Sort((a, b) => a.depth.CompareTo(b.depth));

        foreach (var c in cameras)
        {
            if (c.enabled)
            {
                c.targetTexture = rt;
                c.Render();
            }
        }

        RenderTexture.active = rt;

        var tc = new Texture2D((int)m_size.x, (int)m_size.y, TextureFormat.ARGB32, false);

        tc.ReadPixels(new Rect(Vector2.zero, m_size), 0, 0);
        tc.Apply();

        foreach (var c in cameras)
            c.targetTexture = null;

        RenderTexture.active = null;

        GameObject.Destroy(rt);

        byte[] bytes = tc.EncodeToPNG();

        File.WriteAllBytes(path, bytes);

        GameObject.Destroy(tc);

        Debug.LogFormat("Save file '{0}'", path);
    }

    GUILayout.Space(50);

    m_superSize = EditorGUILayout.IntSlider("superSize:", m_superSize, 1, 10);

    if (GUILayout.Button("Capture"))
    {
        var path = string.Format("{0}{1}.png", m_path, DateTime.Now.ToString("yyyy_MM_dd_HH24_mm_ss"));

        ScreenCapture.CaptureScreenshot(path, m_superSize);
    }
}

}