133 lines
4.3 KiB
C#
133 lines
4.3 KiB
C#
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("face") 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;
|
|
}
|
|
}
|
|
}
|
|
}
|