MARKUS

using Emgu.CV;

using Emgu.CV.Structure;

using System;

using System.Collections.Generic;

using System.Drawing;

using System.Drawing.Imaging;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

namespace Markus.Image

{

    public class ImageComparer : ImageComparerBase

    {

        /// <summary>

        /// 이미지를 비교 후 원본 이미지에 Rect를 그린다.

        /// </summary>

        /// <param name="Originalbitmap">원본 이미지</param>

        /// <param name="TargatBitmap">비교대상 이미지</param>

        /// <param name="ResultRectSize">반환되는 Rect의 사이즈</param>

        /// <returns></returns>

        public Bitmap CompareDrawRects(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize)

        {

            var rects =  CompareReturnRects(Originalbitmap, TargatBitmap, ResultRectSize);

            if (rects.Count != 0)

            {

                using (Graphics g = Graphics.FromImage(Originalbitmap))

                {

                    var rect = rects.Select(x => new System.Drawing.Rectangle((int)x.X, (int)x.Y, (int)x.Width, (int)x.Height)).ToList();

                    g.DrawRectangles(new Pen(Brushes.Blue, 3f), rect.ToArray());

                    g.Save();

                    g.Dispose();

                }

            }

            return Originalbitmap;

        }

        // JoinRects: will return a rectangle composed of r1 and r2.

        private System.Windows.Rect JoinRects(System.Windows.Rect r1, System.Windows.Rect r2)

        {

            return new System.Windows.Rect(Math.Min(r1.X, r2.X),

                            Math.Min(r1.Y, r2.Y),

                            Math.Max(r1.Y + r1.Width, r2.Y + r2.Width),

                            Math.Max(r1.X + r1.Height, r2.X + r2.Height));

        }

        private System.Windows.Rect RectSizeUp(System.Windows.Rect rect,int UpSize)

        {

            return new System.Windows.Rect(rect.X - UpSize, rect.Y - UpSize, rect.Width + UpSize, rect.Height + UpSize);

        }

        public System.Windows.Media.Imaging.BitmapSource CompareDrawRects(System.Windows.Media.Imaging.BitmapSource Originalbitmap, System.Windows.Media.Imaging.BitmapSource TargatBitmap, Size ResultRectSize)

        {

            var _Originalbitmap = CreateBitmapFromSource(Originalbitmap);

            var _TargatBitmap = CreateBitmapFromSource(TargatBitmap);

            var rects = CompareReturnRects(_Originalbitmap, _TargatBitmap, ResultRectSize);

            if (rects.Count != 0)

            {

                using (Graphics g = Graphics.FromImage(_Originalbitmap))

                {

                    var rect = rects.Select(x => new System.Drawing.Rectangle((int)x.X, (int)x.Y, (int)x.Width, (int)x.Height));

                    g.DrawRectangles(new Pen(Brushes.Blue, 3f), rect.ToArray());

                    g.Save();

                    g.Dispose();

                }

            }

            return CreateBitmapSourceFromBitmap(_Originalbitmap);

        }

        public List<System.Windows.Rect> CompareReturnRects(string OriginalbitmapUri, string TargatBitmapUri, Size ResultRectSize)

        {

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            Bitmap Originalbitmap = null;

            Bitmap TargatBitmap = null;

            try

            {

                Originalbitmap = LoadPicture(OriginalbitmapUri);

                TargatBitmap = LoadPicture(TargatBitmapUri);

                result = CompareReturnRects(Originalbitmap, TargatBitmap, ResultRectSize);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

        /// <summary>

        /// 이미지를 비교 후 해당 영역을 Rect로 반환한다.

        /// </summary>

        /// <param name="Originalbitmap">원본 이미지</param>

        /// <param name="TargatBitmap">비교대상 이미지</param>

        /// <param name="ResultRectSize">반환되는 Rect의 사이즈</param>

        /// <returns></returns>

        public List<System.Windows.Rect> CompareReturnRects(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize)

        {

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            try

            {

                List<System.Windows.Rect> rects = new List<System.Windows.Rect>();

                byte[,,] data = MathchesImageData(Originalbitmap, TargatBitmap);

                rects =  GetMatchPixels(data, ResultRectSize);

                if (rects.Count() > 0)

                {

                    result = Merge(rects, ResultRectSize.Height);

                }

                //result = JoinRectList(rects);

            }

            catch (Exception ex)

            {

                throw ex;

            }

            finally

            {

            }

            return result;

        }

        private List<System.Windows.Rect> JoinRectList(List<System.Windows.Rect> rects)

        {

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            if (rects.Count() > 0)

            {

                System.Windows.Rect rect = rects[0];

                System.Windows.Rect joinRect = rects[0];

                while (rects.Count() > 0)

                {

                    var joinItems = rects.Where(x => RectSizeUp(joinRect, 1).Contains(x)).ToList();

                    if (joinItems.Count() == 0)

                    {

                        result.Add(joinRect);

                    }

                    else

                    {

                        for (int i = 0; i < joinItems.Count(); i++)

                        {

                            rect = JoinRects(rect, joinItems[i]);

                            rects.Remove(joinItems[i]);

                        }

                        if (rects.Count() > 0)

                        {

                            rects.RemoveAt(0);

                        }

                        result.Add(joinRect);

                    }

                }

            }

            return result;

        }

        private List<System.Windows.Rect> Merge(List<System.Windows.Rect> r,int rectHeight)

        {

            // Computing the bound

            System.Windows.Rect bound = new System.Windows.Rect(r[0].X, r[0].Y, r[0].Width, r[0].Height);

            for (int i = 1; i < r.Count(); ++i)

            {

                bound.X = Math.Min(bound.X, r[i].X);

                bound.Y = Math.Min(bound.Y, r[i].Y);

                bound.Width = Math.Max(bound.Right, r[i].Right) - bound.X;

                bound.Height = Math.Max(bound.Bottom, r[i].Bottom) - bound.Y;

            }

            // Create number of rectangle will be created by vertical strip expansion

            System.Windows.Rect[] m = new System.Windows.Rect[(int)bound.Height / (int)rectHeight];

            for (int i = 0; i < m.Length; ++i)

            {

                m[i] = new System.Windows.Rect(Int32.MaxValue,bound.Y + i * rectHeight, 0, rectHeight);

            }

            for (int i = 0; i < r.Count(); ++i)

            {

                int index = ((int)r[i].Y - (int)bound.Y) / rectHeight;

                if (m[index].Width <= 0)

                {

                    m[index].Width = r[i].Width;

                    m[index].X = r[i].X;

                }

                else

                {

                    int right = (int)m[index].Right;

                    m[index].X = Math.Min(m[index].X, r[i].X);

                    m[index].Width = Math.Max(right, r[i].Right) - m[index].X;

                }

            }

            // Merge horozontally

            for (int i = m.Length - 1; i > 0; --i)

            {

                // can only merge when two rect has the same X and Width

                if ((m[i].X == m[i - 1].X) && (m[i].Width == m[i - 1].Width))

                {

                    m[i - 1].Height += m[i].Height; // expanse the rectangle

                    m[i].Width = 0;                // remove one rectangle

                }

            }

            // Remove all the empty rectangle

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            for (int i = m.Length - 1; i >= 0; --i)

            {

                if (m[i].Width > 0)

                    result.Add(m[i]);

            }

            return result;

        }

        /// <summary>

        /// 이미지를 비교 후 원본 이미지에 Rect를 그린다.

        /// 메모리 문제 발생

        /// </summary>

        /// <param name="Originalbitmap">원본 이미지</param>

        /// <param name="TargatBitmap">비교대상 이미지</param>

        /// <param name="ResultRectSize">반환되는 Rect의 사이즈</param>

        /// <returns></returns>

        public async Task<Bitmap> CompareDrawRectsAsync(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize)

        {

            System.Drawing.Bitmap cloneOriginal = Originalbitmap;

            var rects = await CompareReturnRectsAsync(cloneOriginal, TargatBitmap, ResultRectSize);

            if (rects.Count != 0)

            {

                using (Graphics g = Graphics.FromImage(cloneOriginal))

                {

                    var rect = rects.Select(x => new System.Drawing.Rectangle((int)x.X, (int)x.Y, (int)x.Width, (int)x.Height));

                    g.DrawRectangles(new Pen(Brushes.Blue, 3f), rect.ToArray());

                    g.Save();

                    g.Dispose();

                }

            }

            return cloneOriginal;

        }

        /// <summary>

        /// 이미지를 비교 후 원본 이미지에 Rect를 그린다.

        /// 메모리 문제 발생

        /// </summary>

        /// <param name="Originalbitmap">원본 이미지</param>

        /// <param name="TargatBitmap">비교대상 이미지</param>

        /// <param name="ResultRectSize">반환되는 Rect의 사이즈</param>

        /// <returns></returns>

        public async Task<System.Windows.Media.Imaging.BitmapSource> CompareDrawRectsAsync(System.Windows.Media.Imaging.BitmapSource Originalbitmap, System.Windows.Media.Imaging.BitmapSource TargatBitmap, Size ResultRectSize)

        {

            var _Originalbitmap = CreateBitmapFromSource(Originalbitmap);

            var _TargatBitmap = CreateBitmapFromSource(TargatBitmap);

            var rects = await CompareReturnRectsAsync(_Originalbitmap, _TargatBitmap, ResultRectSize);

            if (rects.Count != 0)

            {

                using (Graphics g = Graphics.FromImage(_Originalbitmap))

                {

                    var rect = rects.Select(x => new System.Drawing.Rectangle((int)x.X, (int)x.Y, (int)x.Width, (int)x.Height));

                    g.DrawRectangles(new Pen(Brushes.Blue, 3f), rect.ToArray());

                    g.Save();

                    g.Dispose();

                }

            }

            return CreateBitmapSourceFromBitmap(_Originalbitmap);

        }

        public Bitmap CreateBitmapFromSource(System.Windows.Media.Imaging.BitmapSource bitmapsource)

        {

            //convert image format

            var src = new System.Windows.Media.Imaging.FormatConvertedBitmap();

            src.BeginInit();

            src.Source = bitmapsource;

            src.DestinationFormat = System.Windows.Media.PixelFormats.Bgr24;

            src.EndInit();

            //copy to bitmap

            Bitmap bitmap = new Bitmap(src.PixelWidth, src.PixelHeight, System.Drawing.Imaging.PixelFormat.Format32bppArgb);

            var data = bitmap.LockBits(new Rectangle(Point.Empty, bitmap.Size), System.Drawing.Imaging.ImageLockMode.WriteOnly, System.Drawing.Imaging.PixelFormat.Format32bppArgb);

            src.CopyPixels(System.Windows.Int32Rect.Empty, data.Scan0, data.Height * data.Stride, data.Stride);

            bitmap.UnlockBits(data);

            return bitmap;

        }

        /// <summary>

        /// 메모리 문제 발생

        /// </summary>

        /// <param name="bitmap"></param>

        /// <returns></returns>

        public System.Windows.Media.Imaging.WriteableBitmap CreateWriteableBitmapFromBitmap(Bitmap bitmap)

        {

            System.Windows.Media.Imaging.WriteableBitmap result = null;

            if (bitmap == null)

                throw new ArgumentNullException("bitmap");

            try

            {

                int bytesPerPixel = 4;

                result = new System.Windows.Media.Imaging.WriteableBitmap(bitmap.Width, bitmap.Height,

                                                                   bitmap.HorizontalResolution ,bitmap.VerticalResolution,

                                                                   bitmap.PixelFormat.Convert(), null);

                Rectangle colorBitmapRectangle = new Rectangle(0, 0, bitmap.Width, bitmap.Height);

                System.Windows.Int32Rect colorBitmapInt32Rect = new System.Windows.Int32Rect(0, 0, bitmap.Width, bitmap.Height);

                BitmapData data = bitmap.LockBits(colorBitmapRectangle, ImageLockMode.WriteOnly, bitmap.PixelFormat);

                result.WritePixels(colorBitmapInt32Rect, data.Scan0, data.Width * data.Height * bytesPerPixel, data.Stride);

                bitmap.UnlockBits(data);

            }

            catch (Exception ex)

            {

                //throw ex;

            }

            finally

            {

                bitmap.Dispose();

                bitmap = null;

                //GC.Collect(2);

            }

            return result;

        }

        /// <summary>

        /// 메모리 문제 발생

        /// </summary>

        /// <param name="bitmap"></param>

        /// <returns></returns>

        public System.Windows.Media.Imaging.BitmapSource CreateBitmapSourceFromBitmap(Bitmap bitmap)

        {

            System.Windows.Media.Imaging.BitmapSource result = null;

            if (bitmap == null)

                throw new ArgumentNullException("bitmap");

            try

            {

                using (var hbitmap = new SafeHBitmapHandle(bitmap.GetHbitmap(), true))

                {

                    result = System.Windows.Interop.Imaging.CreateBitmapSourceFromHBitmap(

                        hbitmap.DangerousGetHandle(),

                        IntPtr.Zero,

                        System.Windows.Int32Rect.Empty,

                        System.Windows.Media.Imaging.BitmapSizeOptions.FromEmptyOptions());

                }

            }

            catch (Exception ex)

            {

                //throw ex;

            }

            finally

            {

                bitmap.Dispose();

                bitmap = null;

                //GC.Collect(2);

            }

            return result;

        }

        public async Task<List<System.Windows.Rect>> CompareReturnRectsAsync(string OriginalbitmapUri, string TargatBitmapUri, Size ResultRectSize)

        {

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            Bitmap Originalbitmap = null;

            Bitmap TargatBitmap = null;

            try

            {

                Originalbitmap = LoadPicture(OriginalbitmapUri);

                TargatBitmap = LoadPicture(TargatBitmapUri);

                result = await CompareReturnRectsAsync(Originalbitmap, TargatBitmap, ResultRectSize);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

        /// <summary>

        /// 이미지를 비교 후 해당 영역을 Rect로 반환한다.

        /// </summary>

        /// <param name="Originalbitmap">원본 이미지</param>

        /// <param name="TargatBitmap">비교대상 이미지</param>

        /// <param name="ResultRectSize">반환되는 Rect의 사이즈</param>

        /// <returns></returns>

        public async Task<List<System.Windows.Rect>> CompareReturnRectsAsync(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize)

        {

            List<System.Windows.Rect> result = new List<System.Windows.Rect>();

            try

            {

                byte[,,] data = MathchesImageData(Originalbitmap, TargatBitmap);

                result = await GetMatchPixelsAsnc(data, ResultRectSize);

                //result = JoinRectList(result);

                if (result.Count() > 0)

                {

                    result = Merge(result, ResultRectSize.Height);

                }

                data = null;

            }

            catch (Exception ex)

            {

                throw ex;

            }

            finally

            {

            }

            return result;

        }

        protected override void Dispose(bool disposing)

        {

            base.Dispose(disposing);

        }

    }

}