Health/Assets/OpenCVForUnity/Examples/MainModules/imgproc/HoughLinesPExample/HoughLinesPExample.cs

69 lines
2.2 KiB
C#

using UnityEngine;
using UnityEngine.SceneManagement;
using System.Collections;
using OpenCVForUnity.CoreModule;
using OpenCVForUnity.ImgprocModule;
using OpenCVForUnity.UnityUtils;
namespace OpenCVForUnityExample
{
/// <summary>
/// HoughLinesP Example
/// An example of straight line detection using the Imgproc.HoughLinesP function.
/// http://docs.opencv.org/3.1.0/d9/db0/tutorial_hough_lines.html
/// </summary>
public class HoughLinesPExample : MonoBehaviour
{
// Use this for initialization
void Start()
{
Texture2D imgTexture = Resources.Load("chessboard") as Texture2D;
Mat imgMat = new Mat(imgTexture.height, imgTexture.width, CvType.CV_8UC3);
Utils.texture2DToMat(imgTexture, imgMat);
Debug.Log("imgMat.ToString() " + imgMat.ToString());
Mat grayMat = new Mat();
Imgproc.cvtColor(imgMat, grayMat, Imgproc.COLOR_RGB2GRAY);
Imgproc.Canny(grayMat, grayMat, 50, 200);
Mat lines = new Mat();
Imgproc.HoughLinesP(grayMat, lines, 1, Mathf.PI / 180, 50, 50, 10);
//Debug.Log ("lines.toStirng() " + lines.ToString ());
//Debug.Log ("lines.dump()" + lines.dump ());
int[] linesArray = new int[lines.cols() * lines.rows() * lines.channels()];
lines.get(0, 0, linesArray);
for (int i = 0; i < linesArray.Length; i = i + 4)
{
Imgproc.line(imgMat, new Point(linesArray[i + 0], linesArray[i + 1]), new Point(linesArray[i + 2], linesArray[i + 3]), new Scalar(255, 0, 0), 2);
}
Texture2D texture = new Texture2D(imgMat.cols(), imgMat.rows(), TextureFormat.RGBA32, false);
Utils.matToTexture2D(imgMat, texture);
gameObject.GetComponent<Renderer>().material.mainTexture = texture;
}
// Update is called once per frame
void Update()
{
}
/// <summary>
/// Raises the back button click event.
/// </summary>
public void OnBackButtonClick()
{
SceneManager.LoadScene("OpenCVForUnityExample");
}
}
}