131 lines
4.5 KiB
C#
Executable File
131 lines
4.5 KiB
C#
Executable File
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;
|
|
}
|
|
}
|
|
}
|
|
}
|