프로젝트

일반

사용자정보

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

markus / Markus.ImageComparer / ImageCompareBase.cs @ a860bafc

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

1 2007ecaa taeseongkim
using OpenCvSharp;
2
using System;
3 a860bafc taeseongkim
using System.Collections.Concurrent;
4 2007ecaa taeseongkim
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 0c575433 taeseongkim
using System.Threading;
13 2007ecaa taeseongkim
using System.Threading.Tasks;
14 a860bafc taeseongkim
using System.Windows.Media.Imaging;
15 2007ecaa taeseongkim
using Point = System.Drawing.Point;
16
using Size = System.Drawing.Size;
17
18
namespace Markus.Image
19
{
20
    public class ImageCompareBase : IDisposable
21
    {
22
        Mat OriginalImageData = null;
23
        Mat TargatImageData = null;
24
25
        double contoursLongCount = 0;
26 a860bafc taeseongkim
        int ComparisonCount = 0;
27 2007ecaa taeseongkim
        public CompareProgress Progress = new CompareProgress();
28
29
        private void SetStatus(string message,double percentage,CompareStatus status)
30
        {
31 a860bafc taeseongkim
            //System.Diagnostics.Debug.WriteLine(percentage);
32 2007ecaa taeseongkim
33
            Progress.Message = message;
34
            Progress.Percentage = percentage;
35
            Progress.Status = status;
36
        }
37
38
        /// <summary>
39
        /// Originalbitmap에서 TargatBitmap과 비교하여 틀린 부분의 데이터를 Emgu.CV.TDepth형식으로 반환한다.
40
        /// </summary>
41
        /// <param name="Originalbitmap">원본 이미지</param>
42
        /// <param name="TargatBitmap">비교대상 이미지</param>
43
        /// <returns>Emgu.CV.TDepth형식의 byte[,,]</returns>
44 0c575433 taeseongkim
        protected Mat MathchesImageData(System.Drawing.Bitmap Originalbitmap, System.Drawing.Bitmap TargatBitmap,Size resultSize)
45 2007ecaa taeseongkim
        {
46
            Mat result = new Mat();
47
48
            try
49
            {
50
                SetStatus("Image Load", 0, CompareStatus.Loading);
51
52 0c575433 taeseongkim
                Originalbitmap = ChangeBitmapFormatAndSize(Originalbitmap, resultSize, PixelFormat.Format24bppRgb);
53
                TargatBitmap = ChangeBitmapFormatAndSize(TargatBitmap, resultSize, PixelFormat.Format24bppRgb);
54 2007ecaa taeseongkim
55
                // 원본이미지의 크키와 Format24bppRgb로 타켓 이미지를 변경
56
                // 크기가 틀린 경우 비교시 바이트배열 오류 발생
57
                OriginalImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(Originalbitmap);
58
                TargatImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(TargatBitmap);
59
60
                if (OriginalImageData.Size() != TargatImageData.Size())
61
                {
62
                    Cv2.Resize(TargatImageData, TargatImageData, OriginalImageData.Size());
63
                }
64
                
65
                Cv2.CvtColor(OriginalImageData, OriginalImageData, ColorConversionCodes.BGR2GRAY);
66
                Cv2.CvtColor(TargatImageData, TargatImageData, ColorConversionCodes.BGR2GRAY);
67
68 a860bafc taeseongkim
                Mat outputData = new Mat(new OpenCvSharp.Size(resultSize.Width, resultSize.Height), MatType.CV_8UC1);
69 2007ecaa taeseongkim
70
                Cv2.Absdiff(OriginalImageData, TargatImageData, outputData);
71
72
                // 틀린부분을 반환
73
                Cv2.BitwiseNot(outputData, result);
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 a860bafc taeseongkim
            Cv2.Threshold(data, data, 50, 255,ThresholdTypes.BinaryInv);
113 2007ecaa taeseongkim
114 a860bafc taeseongkim
            Cv2.FindContours(data, out contours, out hierarchy,RetrievalModes.List,ContourApproximationModes.ApproxSimple, testpoint);
115 2007ecaa taeseongkim
116
            SetStatus("Comparison", 0, CompareStatus.Comparison);
117
            contoursLongCount = contours.Sum(x => x.Count());
118
119 0c575433 taeseongkim
            //var rects = contours.AsParallel()
120
            //                    .Select(points => GetRectList(points, block))
121
            //                    .SelectMany(x => x);
122
123 2007ecaa taeseongkim
            var rects = contours.AsParallel()
124 0c575433 taeseongkim
                    .SelectMany(points => GetRectList(points, block));
125 2007ecaa taeseongkim
126
            results.AddRange(rects);
127
             
128
            return results;
129
        }
130 a860bafc taeseongkim
131 0c575433 taeseongkim
        private List<System.Windows.Rect> GetRectList(OpenCvSharp.Point[] points, Size block)
132
        {
133
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
134 a860bafc taeseongkim
            ConcurrentBag<System.Windows.Rect> rectBag = new ConcurrentBag<System.Windows.Rect>();
135 0c575433 taeseongkim
136
            Parallel.For(0, points.Length, i =>
137
            {
138
                var rect = new System.Windows.Rect
139
                {
140
                    X = points[i].X - block.Width / 2,
141
                    Y = points[i].Y - block.Height / 2,
142
                    Width = block.Width,
143
                    Height = block.Height
144
                };
145
146 a860bafc taeseongkim
                if (!rectBag.Any(r => r.IntersectsWith(rect)))
147 0c575433 taeseongkim
                {
148 a860bafc taeseongkim
                    rectBag.Add(rect);
149
                    result.Add(rect);
150 0c575433 taeseongkim
                }
151
152 a860bafc taeseongkim
                Interlocked.Increment(ref ComparisonCount);
153 0c575433 taeseongkim
154 a860bafc taeseongkim
                SetStatus("Comparison", (double)ComparisonCount / (double)contoursLongCount * 100, CompareStatus.Comparison);
155 0c575433 taeseongkim
            });
156
157
            return result;
158
        }
159 2007ecaa taeseongkim
160 a860bafc taeseongkim
        protected System.Drawing.Bitmap ChangeBitmapFormatAndSize(System.Drawing.Bitmap bitmap, Size newSize, PixelFormat pixelFormat)
161
        {
162
            Bitmap result = bitmap;
163
164
            if (pixelFormat != bitmap.PixelFormat)
165
            {
166
                Point originPoint = new Point(0, 0);
167
                Rectangle rect = new Rectangle(originPoint, bitmap.Size);
168
                result = bitmap.Clone(rect, pixelFormat);
169
            }
170
171
            if (bitmap.Size != newSize)
172
            {
173
                result = new Bitmap(newSize.Width, newSize.Height);
174
175
                using (Graphics g = Graphics.FromImage(result))
176
                {
177
                    g.DrawImage(bitmap, 0, 0, newSize.Width, newSize.Height);
178
                    g.Dispose();
179
                }
180
            }
181
182
            return result;
183
        }
184
185
186 0c575433 taeseongkim
        private List<System.Windows.Rect> GetRectList2(OpenCvSharp.Point[] points,Size block)
187 2007ecaa taeseongkim
        {
188
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
189
190
            for (int i = 0; i < points.Count(); i++)
191
            {
192
                var rect = new System.Windows.Rect
193
                {
194
                    X = points[i].X - block.Width / 2,
195
                    Y = points[i].Y - block.Height / 2,
196
                    Width = block.Width,
197
                    Height = block.Height
198
                };
199
200
                if (result.Count(r => r.IntersectsWith(rect)) == 0)
201
                {
202
                    result.Add(rect);
203
                }
204
205
                ComparisonCount++;
206
207
                SetStatus("Comparison", ComparisonCount/contoursLongCount*100, CompareStatus.Comparison);
208
            }
209
210
            return result;
211
        }
212
213
        /// <summary>
214
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
215
        /// </summary>
216
        /// <param name="data"></param>
217
        /// <param name="block"></param>
218
        /// <returns></returns>
219
        protected async Task<List<System.Windows.Rect>> GetMatchPixelsAsnc(Mat data, Size block)
220
        {
221
            return await Task.Factory.StartNew<List<System.Windows.Rect>>(() =>
222
            {
223
                return GetMatchPixels(data, block);
224
            });
225
        }
226
227
        protected Bitmap LoadPicture(string url)
228
        {
229
            HttpWebRequest wreq;
230
            HttpWebResponse wresp;
231
            Stream mystream;
232
            Bitmap bmp;
233
234
            bmp = null;
235
            mystream = null;
236
            wresp = null;
237
            try
238
            {
239
                wreq = (HttpWebRequest)WebRequest.Create(url);
240
                wreq.AllowWriteStreamBuffering = true;
241
242
                wresp = (HttpWebResponse)wreq.GetResponse();
243
244
                if ((mystream = wresp.GetResponseStream()) != null)
245
                    bmp = new Bitmap(mystream);
246
            }
247
            finally
248
            {
249
                if (mystream != null)
250
                {
251
                    mystream.Close();
252
                    mystream.Dispose();
253
                }
254
255
                if (wresp != null)
256
                {
257
                    wresp.Close();
258
                    wresp.Dispose();
259
                }
260
            }
261
            return (bmp);
262
        }
263
264
        #region  IDisposable Support 
265
266
        private bool disposedValue = false;
267
268
        // 중복 호출을 검색하려면 
269
        protected virtual void Dispose(bool disposing)
270
        {
271
            if (!disposedValue)
272
            {
273
                if (disposing)
274
                {
275
                    // TODO: 관리되는 상태(관리되는 개체)를 삭제합니다. 
276
                }
277
                // TODO: 관리되지 않는 리소스(관리되지 않는 개체)를 해제하고 아래의 종료자를 재정의합니다. 
278
                // TODO: 큰 필드를 null로 설정합니다. 
279
                disposedValue = true;
280
            }
281
        }
282
283
        // TODO: 위의 Dispose(bool disposing)에 관리되지 않는 리소스를 해제하는 코드가 포함되어 있는 경우에만 종료자를 재정의합니다.
284
        // 
285
        //~DisposeClass()
286
       // {
287
            // // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
288
            // Dispose(false); // 
289
       // } 
290
291
        // 삭제 가능한 패턴을 올바르게 구현하기 위해 추가된 코드입니다. 
292
        public void Dispose()
293
        {
294
            // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
295
            Dispose(true);
296
297
298
            if (OriginalImageData != null)
299
            {
300
                OriginalImageData.Dispose();
301
            }
302
303
            if (TargatImageData != null)
304
            {
305
                TargatImageData.Dispose();
306
            }
307
308
            // TODO: 위의 종료자가 재정의된 경우 다음 코드 줄의 주석 처리를 제거합니다. 
309
            //  GC.SuppressFinalize(this);
310
            GC.Collect();
311
            GC.SuppressFinalize(this);
312
            GC.Collect();
313
        } 
314
        #endregion
315
    }
316
}
클립보드 이미지 추가 (최대 크기: 500 MB)