프로젝트

일반

사용자정보

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

markus / Markus.ImageComparer / ImageCompareBase.cs @ 0c575433

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

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

    
16
namespace Markus.Image
17
{
18
    public class ImageCompareBase : IDisposable
19
    {
20
        Mat OriginalImageData = null;
21
        Mat TargatImageData = null;
22

    
23
        double contoursLongCount = 0;
24
        double ComparisonCount = 0;
25
        public CompareProgress Progress = new CompareProgress();
26

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

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

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

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

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

    
53
                // 원본이미지의 크키와 Format24bppRgb로 타켓 이미지를 변경
54
                // 크기가 틀린 경우 비교시 바이트배열 오류 발생
55
                OriginalImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(Originalbitmap);
56
                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
                Mat outputData = new Mat(TargatImageData.Size(), MatType.CV_8UC1);
67

    
68
                Cv2.Absdiff(OriginalImageData, TargatImageData, outputData);
69

    
70
                // 틀린부분을 반환
71
                Cv2.BitwiseNot(outputData, result);
72
            }
73
            catch (Exception ex)
74
            {
75
                throw ex;
76
            }
77
            finally
78
            {
79
            }
80
            
81
            return result;
82
        }
83

    
84
        protected System.Drawing.Bitmap ChangeBitmapFormatAndSize(System.Drawing.Bitmap bitmap, Size newSize, PixelFormat pixelFormat)
85
        {
86
            Bitmap result = bitmap;
87

    
88
            if (pixelFormat != bitmap.PixelFormat)
89
            {
90
                Point originPoint = new Point(0, 0);
91
                Rectangle rect = new Rectangle(originPoint, bitmap.Size);
92
                result = bitmap.Clone(rect, pixelFormat);
93
            }
94

    
95
            if (bitmap.Size != newSize)
96
            {
97
                result = new Bitmap(newSize.Width, newSize.Height);
98

    
99
                using (Graphics g = Graphics.FromImage(result))
100
                {
101
                    g.DrawImage(bitmap, 0, 0, newSize.Width, newSize.Height);
102
                    g.Dispose();
103
                }
104
            }
105

    
106
            return result;
107
        }
108

    
109
        /// <summary>
110
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
111
        /// </summary>
112
        /// <param name="data"></param>
113
        /// <param name="block"></param>
114
        /// <returns></returns>
115
        protected List<System.Windows.Rect> GetMatchPixels(Mat data, Size block)
116
        {
117
            SetStatus("Detection", 0, CompareStatus.Detection);
118

    
119
            int width = data.Cols;
120
            int height = data.Rows;
121

    
122
            List<System.Windows.Rect> results = new List<System.Windows.Rect>();
123

    
124
            System.Diagnostics.Stopwatch stopwatch = new System.Diagnostics.Stopwatch();
125
            stopwatch.Start();
126

    
127
            Mat outputMat = new Mat(data.Size(), MatType.CV_8UC1);
128

    
129
            HierarchyIndex[] hierarchy;
130
            OpenCvSharp.Point[][] contours;
131

    
132
            OpenCvSharp.Point testpoint = new OpenCvSharp.Point();
133
           
134
            Cv2.Threshold(data, data, 0, 30,ThresholdTypes.BinaryInv);
135

    
136
            Cv2.FindContours(data, out contours, out hierarchy,RetrievalModes.List,ContourApproximationModes.ApproxNone, testpoint);
137

    
138
            SetStatus("Comparison", 0, CompareStatus.Comparison);
139
            contoursLongCount = contours.Sum(x => x.Count());
140

    
141
            //var rects = contours.AsParallel()
142
            //                    .Select(points => GetRectList(points, block))
143
            //                    .SelectMany(x => x);
144

    
145
            var rects = contours.AsParallel()
146
                    .SelectMany(points => GetRectList(points, block));
147

    
148
            results.AddRange(rects);
149
             
150
            return results;
151
        }
152
        private List<System.Windows.Rect> GetRectList(OpenCvSharp.Point[] points, Size block)
153
        {
154
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
155
            HashSet<System.Windows.Rect> rectSet = new HashSet<System.Windows.Rect>();
156

    
157
            Parallel.For(0, points.Length, i =>
158
            {
159
                var rect = new System.Windows.Rect
160
                {
161
                    X = points[i].X - block.Width / 2,
162
                    Y = points[i].Y - block.Height / 2,
163
                    Width = block.Width,
164
                    Height = block.Height
165
                };
166

    
167
                lock (rectSet)
168
                {
169
                    if (!rectSet.Any(r => r.IntersectsWith(rect)))
170
                    {
171
                        rectSet.Add(rect);
172
                        result.Add(rect);
173
                    }
174
                }
175

    
176
                int comparisonCount = 0;
177
                Interlocked.Increment(ref comparisonCount);
178
                
179
                ComparisonCount = comparisonCount;
180

    
181
                SetStatus("Comparison", ComparisonCount / (double)contoursLongCount * 100, CompareStatus.Comparison);
182
            });
183

    
184
            return result;
185
        }
186

    
187
        private List<System.Windows.Rect> GetRectList2(OpenCvSharp.Point[] points,Size block)
188
        {
189
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
190

    
191
            for (int i = 0; i < points.Count(); i++)
192
            {
193
                var rect = new System.Windows.Rect
194
                {
195
                    X = points[i].X - block.Width / 2,
196
                    Y = points[i].Y - block.Height / 2,
197
                    Width = block.Width,
198
                    Height = block.Height
199
                };
200

    
201
                if (result.Count(r => r.IntersectsWith(rect)) == 0)
202
                {
203
                    result.Add(rect);
204
                }
205

    
206
                ComparisonCount++;
207

    
208
                SetStatus("Comparison", ComparisonCount/contoursLongCount*100, CompareStatus.Comparison);
209
            }
210

    
211
            return result;
212
        }
213

    
214
        /// <summary>
215
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
216
        /// </summary>
217
        /// <param name="data"></param>
218
        /// <param name="block"></param>
219
        /// <returns></returns>
220
        protected async Task<List<System.Windows.Rect>> GetMatchPixelsAsnc(Mat data, Size block)
221
        {
222
            return await Task.Factory.StartNew<List<System.Windows.Rect>>(() =>
223
            {
224
                return GetMatchPixels(data, block);
225
            });
226
        }
227

    
228
        protected Bitmap LoadPicture(string url)
229
        {
230
            HttpWebRequest wreq;
231
            HttpWebResponse wresp;
232
            Stream mystream;
233
            Bitmap bmp;
234

    
235
            bmp = null;
236
            mystream = null;
237
            wresp = null;
238
            try
239
            {
240
                wreq = (HttpWebRequest)WebRequest.Create(url);
241
                wreq.AllowWriteStreamBuffering = true;
242

    
243
                wresp = (HttpWebResponse)wreq.GetResponse();
244

    
245
                if ((mystream = wresp.GetResponseStream()) != null)
246
                    bmp = new Bitmap(mystream);
247
            }
248
            finally
249
            {
250
                if (mystream != null)
251
                {
252
                    mystream.Close();
253
                    mystream.Dispose();
254
                }
255

    
256
                if (wresp != null)
257
                {
258
                    wresp.Close();
259
                    wresp.Dispose();
260
                }
261
            }
262
            return (bmp);
263
        }
264

    
265
        #region  IDisposable Support 
266

    
267
        private bool disposedValue = false;
268

    
269
        // 중복 호출을 검색하려면 
270
        protected virtual void Dispose(bool disposing)
271
        {
272
            if (!disposedValue)
273
            {
274
                if (disposing)
275
                {
276
                    // TODO: 관리되는 상태(관리되는 개체)를 삭제합니다. 
277
                }
278
                // TODO: 관리되지 않는 리소스(관리되지 않는 개체)를 해제하고 아래의 종료자를 재정의합니다. 
279
                // TODO: 큰 필드를 null로 설정합니다. 
280
                disposedValue = true;
281
            }
282
        }
283

    
284
        // TODO: 위의 Dispose(bool disposing)에 관리되지 않는 리소스를 해제하는 코드가 포함되어 있는 경우에만 종료자를 재정의합니다.
285
        // 
286
        //~DisposeClass()
287
       // {
288
            // // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
289
            // Dispose(false); // 
290
       // } 
291

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

    
298

    
299
            if (OriginalImageData != null)
300
            {
301
                OriginalImageData.Dispose();
302
            }
303

    
304
            if (TargatImageData != null)
305
            {
306
                TargatImageData.Dispose();
307
            }
308

    
309
            // TODO: 위의 종료자가 재정의된 경우 다음 코드 줄의 주석 처리를 제거합니다. 
310
            //  GC.SuppressFinalize(this);
311
            GC.Collect();
312
            GC.SuppressFinalize(this);
313
            GC.Collect();
314
        } 
315
        #endregion
316
    }
317
}
클립보드 이미지 추가 (최대 크기: 500 MB)