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

namespace OpenCVForUnityExample
{
    /// <summary>
    /// MatchTemplate Example
    /// An example of template matching using the Imgproc.matchTemplate function.
    /// http://docs.opencv.org/3.2.0/de/da9/tutorial_template_matching.html
    /// </summary>
    public class MatchTemplateExample : MonoBehaviour
    {
        // Use this for initialization
        void Start ()
        {
            Texture2D imgTexture = Resources.Load ("lena") as Texture2D;
            Texture2D tempTexture = Resources.Load ("template") as Texture2D;
            Mat imgMat = new Mat (imgTexture.height, imgTexture.width, CvType.CV_8UC4);
            Mat tempMat = new Mat (tempTexture.height, tempTexture.width, CvType.CV_8UC4);
            Utils.texture2DToMat (imgTexture, imgMat);
            Utils.texture2DToMat (tempTexture, tempMat);

            
            //Create the result mat
            int result_cols = imgMat.cols () - tempMat.cols () + 1;
            int result_rows = imgMat.rows () - tempMat.rows () + 1;
            Mat result = new Mat (result_rows, result_cols, CvType.CV_32FC1);
            
            int match_method = Imgproc.TM_CCOEFF_NORMED;
            

            Imgproc.matchTemplate (imgMat, tempMat, result, match_method);
            
            Imgproc.threshold (result, result, 0.8, 1.0, Imgproc.THRESH_TOZERO);//threshold = 0.8
            
            for (int i = 0; i < result.rows (); i++) {
                for (int j = 0; j < result.cols (); j++) {
                    if (result.get (i, j) [0] > 0) {
                        
                        Imgproc.rectangle (imgMat, new Point (j, i), new Point (j + tempMat.cols (), i + tempMat.rows ()), new Scalar (255, 0, 0, 255), 2);
                        Debug.Log ("value" + result.get (i, j) [0]);
                    }
                }
            }

            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");
        }
    }
}