프로젝트

일반

사용자정보

통계
| 브랜치(Branch): | 개정판:

markus / Markus.ImageComparer / ImageCompareBase.cs @ 16231f58

이력 | 보기 | 이력해설 | 다운로드 (12 KB)

1
using OpenCvSharp;
2
using System;
3
using System.Collections.Concurrent;
4
using System.Collections.Generic;
5
using System.Drawing;
6
using System.Drawing.Imaging;
7
using System.IO;
8
using System.Linq;
9
using System.Net;
10
using System.Runtime.InteropServices;
11
using System.Text;
12
using System.Threading;
13
using System.Threading.Tasks;
14
using System.Windows.Media.Imaging;
15
using Point = System.Drawing.Point;
16
using Size = System.Drawing.Size;
17

    
18
namespace Markus.Image
19
{
20
    public class ImageCompareBase : IDisposable
21
    {
22
        double contoursLongCount = 0;
23
        int ComparisonCount = 0;
24
        public CompareProgress Progress = new CompareProgress();
25

    
26
        private void SetStatus(string message,double percentage,CompareStatus status)
27
        {
28
            //System.Diagnostics.Debug.WriteLine(percentage);
29

    
30
            Progress.Message = message;
31
            Progress.Percentage = percentage;
32
            Progress.Status = status;
33
        }
34

    
35
        /// <summary>
36
        /// Originalbitmap에서 TargatBitmap과 비교하여 틀린 부분의 데이터를 Emgu.CV.TDepth형식으로 반환한다.
37
        /// </summary>
38
        /// <param name="Originalbitmap">원본 이미지</param>
39
        /// <param name="TargatBitmap">비교대상 이미지</param>
40
        /// <returns>Emgu.CV.TDepth형식의 byte[,,]</returns>
41
        protected Mat MathchesImageData(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap,Size resultSize)
42
        {
43
            Mat result = new Mat();
44

    
45
            try
46
            {
47
                SetStatus("Image Load", 0, CompareStatus.Loading);
48

    
49
                Originalbitmap = ChangeBitmapFormatAndSize(Originalbitmap, resultSize, PixelFormat.Format24bppRgb);
50
                TargatBitmap = ChangeBitmapFormatAndSize(TargatBitmap, resultSize, PixelFormat.Format24bppRgb);
51

    
52
                // 원본이미지의 크키와 Format24bppRgb로 타켓 이미지를 변경
53
                // 크기가 틀린 경우 비교시 바이트배열 오류 발생
54
                using (Mat OriginalImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(Originalbitmap))
55
                {
56
                    using (Mat TargatImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(TargatBitmap))
57
                    {
58
                        if (OriginalImageData.Size() != TargatImageData.Size())
59
                        {
60
                            Cv2.Resize(TargatImageData, TargatImageData, OriginalImageData.Size());
61
                        }
62

    
63
                        Cv2.CvtColor(OriginalImageData, OriginalImageData, ColorConversionCodes.BGR2GRAY);
64
                        Cv2.CvtColor(TargatImageData, TargatImageData, ColorConversionCodes.BGR2GRAY);
65

    
66
                        using (Mat outputData = new Mat(new OpenCvSharp.Size(resultSize.Width, resultSize.Height), MatType.CV_8UC1))
67
                        {
68
                            Cv2.Absdiff(OriginalImageData, TargatImageData, outputData);
69
                            // 틀린부분을 반환
70
                            Cv2.BitwiseNot(outputData, result);
71
                        }
72
                    }
73
                }
74
            }
75
            catch (Exception ex)
76
            {
77
                throw ex;
78
            }
79
            finally
80
            {
81
            }
82
            
83
            return result;
84
        }
85

    
86

    
87
        /// <summary>
88
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
89
        /// </summary>
90
        /// <param name="data"></param>
91
        /// <param name="block"></param>
92
        /// <returns></returns>
93
        protected List<System.Windows.Rect> GetMatchPixels(Mat data, Size block)
94
        {
95
            SetStatus("Detection", 0, CompareStatus.Detection);
96

    
97
            int width = data.Cols;
98
            int height = data.Rows;
99

    
100
            List<System.Windows.Rect> results = new List<System.Windows.Rect>();
101

    
102
            System.Diagnostics.Stopwatch stopwatch = new System.Diagnostics.Stopwatch();
103
            stopwatch.Start();
104

    
105
            //Mat outputMat = new Mat(data.Size(), MatType.CV_8UC1);
106

    
107
            HierarchyIndex[] hierarchy;
108
            OpenCvSharp.Point[][] contours;
109

    
110
            OpenCvSharp.Point testpoint = new OpenCvSharp.Point();
111
           
112
            Cv2.Threshold(data, data, 50, 255,ThresholdTypes.BinaryInv);
113

    
114
            Cv2.FindContours(data, out contours, out hierarchy,RetrievalModes.List,ContourApproximationModes.ApproxSimple, testpoint);
115

    
116
            SetStatus("Comparison", 0, CompareStatus.Comparison);
117
            contoursLongCount = contours.Sum(x => x.Count());
118

    
119
            //var rects = contours.AsParallel()
120
            //                    .Select(points => GetRectList(points, block))
121
            //                    .SelectMany(x => x);
122

    
123
            var rects = contours.SelectMany(points => GetRectList(points, block));
124

    
125
            results.AddRange(rects);
126
             
127
            return results;
128
        }
129
        protected System.Drawing.Bitmap GetContoursImage(Mat data, Size resultSize, Color RectColor)
130
        {
131
            System.Drawing.Bitmap result = null;
132

    
133
            try
134
            {
135
                Mat transparentImage = new Mat(new OpenCvSharp.Size(resultSize.Width, resultSize.Height), MatType.CV_8UC4, new Scalar(0, 0, 0, 0));
136

    
137
                // 컨투어를 찾기 위한 변수 선언
138
                HierarchyIndex[] hierarchy;
139
                OpenCvSharp.Point[][] contours;
140
                OpenCvSharp.Point testpoint = new OpenCvSharp.Point();
141

    
142
                Cv2.Threshold(data, data, 100, 255, ThresholdTypes.BinaryInv);
143
                // 이진화된 이미지에서 컨투어 찾기
144
                Cv2.FindContours(data, out contours, out hierarchy, RetrievalModes.List, ContourApproximationModes.ApproxNone, testpoint);
145

    
146

    
147
                // Scalar(B,G,R,A)
148
                // 컨투어 그리기
149
                Cv2.DrawContours(transparentImage, contours, -1, new Scalar(RectColor.B, RectColor.G, RectColor.R, RectColor.A), 10);
150

    
151
                // Mat을 Bitmap으로 변환
152
                result = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(transparentImage);
153
            }
154
            catch (Exception ex)
155
            {
156
                throw new Exception("GetContoursImage Error.", ex);
157
            }
158
            finally
159
            {
160

    
161
            }
162

    
163
            return result;
164
        }
165
        private List<System.Windows.Rect> GetRectListAsParallel(OpenCvSharp.Point[] points, Size block)
166
        {
167
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
168
            ConcurrentBag<System.Windows.Rect> rectBag = new ConcurrentBag<System.Windows.Rect>();
169

    
170
            Parallel.For(0, points.Length, i =>
171
            {
172
                if (i >= points.Length) return; // 배열 길이 확인
173

    
174
                var rect = new System.Windows.Rect
175
                {
176
                    X = points[i].X - block.Width / 2,
177
                    Y = points[i].Y - block.Height / 2,
178
                    Width = block.Width,
179
                    Height = block.Height
180
                };
181

    
182
                if (!rectBag.Any(r => r.IntersectsWith(rect)))
183
                {
184
                    rectBag.Add(rect);
185
                    result.Add(rect);
186
                }
187

    
188
                Interlocked.Increment(ref ComparisonCount);
189

    
190
                SetStatus("Comparison", (double)ComparisonCount / (double)contoursLongCount * 100, CompareStatus.Comparison);
191
            });
192

    
193
            return result;
194
        }
195

    
196
        protected System.Drawing.Bitmap ChangeBitmapFormatAndSize(System.Drawing.Bitmap bitmap, Size newSize, PixelFormat pixelFormat)
197
        {
198
            Bitmap result = bitmap;
199

    
200
            if (pixelFormat != bitmap.PixelFormat)
201
            {
202
                Point originPoint = new Point(0, 0);
203
                Rectangle rect = new Rectangle(originPoint, bitmap.Size);
204
                result = bitmap.Clone(rect, pixelFormat);
205
            }
206

    
207
            if (bitmap.Size != newSize)
208
            {
209
                result = new Bitmap(newSize.Width, newSize.Height);
210

    
211
                using (Graphics g = Graphics.FromImage(result))
212
                {
213
                    g.DrawImage(bitmap, 0, 0, newSize.Width, newSize.Height);
214
                    g.Dispose();
215
                }
216
            }
217

    
218
            return result;
219
        }
220

    
221

    
222
        private List<System.Windows.Rect> GetRectList(OpenCvSharp.Point[] points,Size block)
223
        {
224
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
225

    
226
            for (int i = 0; i < points.Count(); i++)
227
            {
228
                var rect = new System.Windows.Rect
229
                {
230
                    X = points[i].X - block.Width / 2,
231
                    Y = points[i].Y - block.Height / 2,
232
                    Width = block.Width,
233
                    Height = block.Height
234
                };
235

    
236
                if (result.Count(r => r.IntersectsWith(rect)) == 0)
237
                {
238
                    result.Add(rect);
239
                }
240

    
241
                ComparisonCount++;
242

    
243
                SetStatus("Comparison", ComparisonCount/contoursLongCount*100, CompareStatus.Comparison);
244
            }
245

    
246
            return result;
247
        }
248

    
249
        /// <summary>
250
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
251
        /// </summary>
252
        /// <param name="data"></param>
253
        /// <param name="block"></param>
254
        /// <returns></returns>
255
        protected async Task<List<System.Windows.Rect>> GetMatchPixelsAsnc(Mat data, Size block)
256
        {
257
            return await Task.Factory.StartNew<List<System.Windows.Rect>>(() =>
258
            {
259
                return GetMatchPixels(data, block);
260
            });
261
        }
262

    
263
        protected Bitmap LoadPicture(string url)
264
        {
265
            HttpWebRequest wreq;
266
            HttpWebResponse wresp;
267
            Stream mystream;
268
            Bitmap bmp;
269

    
270
            bmp = null;
271
            mystream = null;
272
            wresp = null;
273
            try
274
            {
275
                wreq = (HttpWebRequest)WebRequest.Create(url);
276
                wreq.AllowWriteStreamBuffering = true;
277

    
278
                wresp = (HttpWebResponse)wreq.GetResponse();
279

    
280
                if ((mystream = wresp.GetResponseStream()) != null)
281
                    bmp = new Bitmap(mystream);
282
            }
283
            finally
284
            {
285
                if (mystream != null)
286
                {
287
                    mystream.Close();
288
                    mystream.Dispose();
289
                }
290

    
291
                if (wresp != null)
292
                {
293
                    wresp.Close();
294
                    wresp.Dispose();
295
                }
296
            }
297
            return (bmp);
298
        }
299

    
300
        #region  IDisposable Support 
301

    
302
        private bool disposedValue = false;
303

    
304
        // 중복 호출을 검색하려면 
305
        protected virtual void Dispose(bool disposing)
306
        {
307
            if (!disposedValue)
308
            {
309
                if (disposing)
310
                {
311
                    // TODO: 관리되는 상태(관리되는 개체)를 삭제합니다. 
312
                }
313
                // TODO: 관리되지 않는 리소스(관리되지 않는 개체)를 해제하고 아래의 종료자를 재정의합니다. 
314
                // TODO: 큰 필드를 null로 설정합니다. 
315
                disposedValue = true;
316
            }
317
        }
318

    
319
        // TODO: 위의 Dispose(bool disposing)에 관리되지 않는 리소스를 해제하는 코드가 포함되어 있는 경우에만 종료자를 재정의합니다.
320
        // 
321
        //~DisposeClass()
322
       // {
323
            // // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
324
            // Dispose(false); // 
325
       // } 
326

    
327
        // 삭제 가능한 패턴을 올바르게 구현하기 위해 추가된 코드입니다. 
328
        public void Dispose()
329
        {
330
            // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
331
            Dispose(true);
332

    
333
            // TODO: 위의 종료자가 재정의된 경우 다음 코드 줄의 주석 처리를 제거합니다. 
334
            //  GC.SuppressFinalize(this);
335
            GC.Collect();
336
            GC.SuppressFinalize(this);
337
            GC.Collect();
338
        } 
339
        #endregion
340
    }
341
}
클립보드 이미지 추가 (최대 크기: 500 MB)