using System.Collections;
using System.Collections.Generic;
using System.IO;
using UnityEngine;
using UnityEditor;

[CustomEditor(typeof(RenderTextureSaveToPNG))]
public class RenderTextureSaveToPNGEditor : Editor
{
    RenderTextureSaveToPNG inst;

    private void OnEnable()
    {
        inst = (RenderTextureSaveToPNG)target;
    }

    public override void OnInspectorGUI()
    {
        base.OnInspectorGUI();

        //
        if(GUILayout.Button("RenderTextureSaveToPNG", GUILayout.Height(50)))
        {
            SaveRenderToPng();
        }
    }

    Texture2D SaveRenderToPng()
    {
        int width = inst.inputTex.width;
        int height = inst.inputTex.height;
        Texture2D tex2d = new Texture2D(width, height, TextureFormat.ARGB32, false);
        RenderTexture.active = inst.inputTex;
        tex2d.ReadPixels(new Rect(0, 0, width, height), 0, 0);

        for(int y = 0; y < tex2d.height; ++y)
        {
            for (int x = 0; x < tex2d.width; ++x)
            {
                if(tex2d.GetPixel(x, y).grayscale > inst.maskSrength)
                {
                    tex2d.SetPixel(x, y, Color.white);
                }
            }
        }

        tex2d.Apply();

        File.WriteAllBytes(Application.dataPath + "/Materials/" + inst.inputTex.name + ".png", tex2d.EncodeToPNG());
        AssetDatabase.Refresh();

        return tex2d;
    }
}