GHZMRSpaceXR/Assets/OpenCVForUnity/Examples/MainModules/imgcodecs/ImwriteScreenCaptureExample/ImwriteScreenCaptureExample.cs

using System.Collections;
using UnityEngine;
using UnityEngine.UI;
using UnityEngine.SceneManagement;
using OpenCVForUnity.CoreModule;
using OpenCVForUnity.ImgprocModule;
using OpenCVForUnity.ImgcodecsModule;
using OpenCVForUnity.UnityUtils;

namespace OpenCVForUnityExample
{
    [RequireComponent (typeof(Camera))]
    public class ImwriteScreenCaptureExample : MonoBehaviour
    {
        /// <summary>
        /// The cube.
        /// </summary>
        public GameObject cube;

        /// <summary>
        /// The save path input field.
        /// </summary>
        public InputField savePathInputField;

        /// <summary>
        /// The capture flag.
        /// </summary>
        bool captureFlag = false;

        /// <summary>
        /// The save path.
        /// </summary>
        string savePath;

        // Use this for initialization
        void Start ()
        {
            savePath = Application.persistentDataPath + "/ImwriteScreenCaptureExample_output.jpg";

            //if true, The error log of the Native side OpenCV will be displayed on the Unity Editor Console.
            Utils.setDebugMode (true);


            Texture2D imgTexture = Resources.Load ("lena") as Texture2D;

            Mat imgMat = new Mat (imgTexture.height, imgTexture.width, CvType.CV_8UC4);

            Utils.texture2DToMat (imgTexture, imgMat);
            Debug.Log ("imgMat.ToString() " + imgMat.ToString ());

            Texture2D texture = new Texture2D (imgMat.cols (), imgMat.rows (), TextureFormat.RGBA32, false);

            Utils.matToTexture2D (imgMat, texture);

            cube.GetComponent<Renderer> ().material.mainTexture = texture;


            Utils.setDebugMode (false);
        }

        /// <summary>
        /// Raises the render image event.
        /// </summary>
        /// <param name="source">Source.</param>
        /// <param name="destination">Destination.</param>
        void OnRenderImage (RenderTexture source, RenderTexture destination)
        {
            if (captureFlag) {
                //            Debug.Log ("source.width " + source.width + "source.height " + source.height);

                Mat cameraMat = new Mat (source.height, source.width, CvType.CV_8UC4);
                Texture2D texture = new Texture2D (cameraMat.width (), cameraMat.height (), TextureFormat.ARGB32, false);

                Utils.textureToTexture2D (source, texture);
                Utils.texture2DToMat (texture, cameraMat);

                Imgproc.cvtColor (cameraMat, cameraMat, Imgproc.COLOR_RGBA2BGRA);

                Imgproc.rectangle (cameraMat, new Point (0, 0), new Point (cameraMat.width (), cameraMat.height ()), new Scalar (0, 0, 255, 255), 3);
                Imgproc.putText (cameraMat, "SavePath:", new Point (5, cameraMat.rows () - 30), Imgproc.FONT_HERSHEY_SIMPLEX, 0.8, new Scalar (0, 0, 255), 2, Imgproc.LINE_AA, false);
                Imgproc.putText (cameraMat, savePath, new Point (5, cameraMat.rows () - 8), Imgproc.FONT_HERSHEY_SIMPLEX, 0.5, new Scalar (255, 255, 255), 0, Imgproc.LINE_AA, false);

                Imgcodecs.imwrite (savePath, cameraMat);

                savePathInputField.text = savePath;
                Debug.Log ("savePath: " + savePath);

                captureFlag = false;
            }

            Graphics.Blit (source, destination);
        }

        /// <summary>
        /// Raises the back button click event.
        /// </summary>
        public void OnBackButtonClick ()
        {
            SceneManager.LoadScene ("OpenCVForUnityExample");
        }

        /// <summary>
        /// Raises the capture screen button click event.
        /// </summary>
        public void OnCaptureScreenButtonClick ()
        {
            captureFlag = true;
        }

        /// <summary>
        /// Raises the load screen button click event.
        /// </summary>
        public void OnLoadScreenButtonClick ()
        {
            Mat loadMat = Imgcodecs.imread (savePath);
            Debug.Log ("loadMat.ToString() " + loadMat.ToString ());

            if (loadMat.width () != 0 && loadMat.height () != 0) {
                
                Texture2D texture = new Texture2D (loadMat.width (), loadMat.height (), TextureFormat.RGBA32, false);

                Imgproc.cvtColor (loadMat, loadMat, Imgproc.COLOR_BGR2RGB);

                Utils.matToTexture2D (loadMat, texture);

                cube.GetComponent<Renderer> ().material.mainTexture = texture;
            }
        }
    }
}