MARKUS

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 ImageCompare : ImageCompareBase

    {

        /// <summary>

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

        /// </summary>

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

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

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

        /// <param name="ResultImageSize">반환되는 이미지 사이즈. 없으면 targat사이즈로 반환</param>

        /// <returns>TargetBitmap에 ResultRectSize의 크기로 변환된 부분이 표시되어 반환</returns>

        public Bitmap CompareDrawRects(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize, Size? ResultImageSize = null)

        {

            Size resultImageSize = TargatBitmap.Size;

            if (ResultImageSize.HasValue)

            {

                resultImageSize = ResultImageSize.Value;

            }

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

            if (rects.Count != 0)

            {

                Bitmap newBitmap = new Bitmap(resultImageSize.Width, resultImageSize.Height, System.Drawing.Imaging.PixelFormat.Format24bppRgb);

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

                {

                    g.DrawImage(TargatBitmap, 0, 0);

                }

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

                {

                    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.Red, 5f), rect.ToArray());

                    g.Save();

                    g.Dispose();

                }

                Originalbitmap = newBitmap;

            }

            return Originalbitmap;

        }

        public System.Windows.Media.Imaging.BitmapSource CompareDrawRects(System.Windows.Media.Imaging.BitmapSource Originalbitmap, System.Windows.Media.Imaging.BitmapSource TargatBitmap,  Size ResultRectSize, Size? ResultImageSize = null)

        {

            var _Originalbitmap = CreateBitmapFromSource(Originalbitmap);

            var _TargatBitmap = CreateBitmapFromSource(TargatBitmap);

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

            if (rects.Count != 0)

            {

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

                {

                    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(_TargatBitmap);

        }

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

        {

            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, ResultImageSize);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

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

        {

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

            Bitmap Originalbitmap = null;

            Bitmap TargatBitmap = null;

            try

            {

                Originalbitmap = LoadPicture(OriginalbitmapUri);

                TargatBitmap = LoadPicture(TargatBitmapUri);

                Size resultImageSize = Originalbitmap.Size;

                if (IsResultTargetSize)

                {

                    resultImageSize = TargatBitmap.Size;

                }

                result = CompareReturnRects(Originalbitmap, TargatBitmap, ResultRectSize, resultImageSize);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

        public List<System.Windows.Rect> CompareReturnRects(System.Windows.Media.Imaging.BitmapFrame OriginalbitmapFrame, System.Windows.Media.Imaging.BitmapFrame TargatBitmapFrame, Size ResultRectSize, Size? ResultImageSize = null)

        {

            return CompareReturnRects(Converter.ConvertBitmapFrameToBitmap(OriginalbitmapFrame), Converter.ConvertBitmapFrameToBitmap(TargatBitmapFrame), ResultRectSize, ResultImageSize);

        }

        /// <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 , Size? ResultImageSize = null)

        {

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

            try

            {

                Size resultImageSize = TargatBitmap.Size;

                if (ResultImageSize.HasValue)

                {

                    resultImageSize = ResultImageSize.Value;

                }

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

                var data = MathchesImageData(Originalbitmap, TargatBitmap, resultImageSize);

                result =  GetMatchPixels(data, ResultRectSize);

                //var data2 = MathchesImageData(TargatBitmap,Originalbitmap, Originalbitmap.Size);

                //result.AddRange(GetMatchPixels(data2, ResultRectSize));

                //result = result.Distinct().ToList();

                //if (rects.Count() > 0)

                //{

                //    result = Merge(rects, ResultRectSize.Height);

                //}

                //result = JoinRectList(rects);

            }

            catch (Exception ex)

            {

                throw ex;

            }

            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<Bitmap> CompareDrawRectsAsync(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap, Size ResultRectSize, Size? ResultImageSize = null)

        {

            System.Drawing.Bitmap cloneOriginal = Originalbitmap;

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

            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, Size? ResultImageSize = null)

        {

            var _Originalbitmap = CreateBitmapFromSource(Originalbitmap);

            var _TargatBitmap = CreateBitmapFromSource(TargatBitmap);

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

            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="nwdBitmap"></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;

                Bitmap newBitmap = new Bitmap(bitmap.Width, bitmap.Height, PixelFormat.Format32bppArgb);

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

                {

                    g.DrawImage(bitmap, 0, 0);

                }

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

                                                                   newBitmap.HorizontalResolution , newBitmap.VerticalResolution,

                                                                   newBitmap.PixelFormat.Convert(), null);

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

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

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

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

                newBitmap.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, Size? ResultImageSize = null)

        {

            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, ResultImageSize);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

        public async Task<System.Windows.Media.Imaging.BitmapSource> CompareDrawRectsAsync(string OriginalbitmapUri, string TargatBitmapUri, Size ResultRectSize, Size? ResultImageSize = null)

        {

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

            Bitmap Originalbitmap = null;

            Bitmap TargatBitmap = null;

            try

            {

                Originalbitmap = LoadPicture(OriginalbitmapUri);

                TargatBitmap = LoadPicture(TargatBitmapUri);

                var bitmap = await CompareDrawRectsAsync(Originalbitmap, TargatBitmap, ResultRectSize, ResultImageSize);

                result = CreateBitmapSourceFromBitmap(bitmap);

            }

            catch (Exception)

            {

                throw;

            }

            finally

            {

                Originalbitmap.Dispose();

                TargatBitmap.Dispose();

            }

            return result;

        }

        public async Task<List<System.Windows.Rect>> CompareReturnRectsAsync(System.Windows.Media.Imaging.BitmapSource OriginalbitmapFrame, System.Windows.Media.Imaging.BitmapSource TargatBitmapFrame, Size ResultRectSize, Size? ResultImageSize = null)

        {

            return await CompareReturnRectsAsync(Converter.ConvertBitmapFrameToBitmap(OriginalbitmapFrame), Converter.ConvertBitmapFrameToBitmap(TargatBitmapFrame), ResultRectSize, ResultImageSize);

        }

            /// <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,Size? ResultImageSize = null)

        {

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

            try

            {

                Size resultImageSize = Originalbitmap.Size;

                if(ResultImageSize.HasValue)

                {

                    resultImageSize = ResultImageSize.Value;

                }

                var data = MathchesImageData(Originalbitmap, TargatBitmap, resultImageSize);

                result = await GetMatchPixelsAsnc(data, ResultRectSize);

                data = null;

            }

            catch (Exception ex)

            {

                throw ex;

            }

            finally

            {

            }

            return result;

        }

        protected override void Dispose(bool disposing)

        {

            base.Dispose(disposing);

        }

    }

}