프로젝트

일반

사용자정보

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

markus / ImageComparer / Markus.ImageComparer - 복사본 / ImageCompareBase.cs @ 2007ecaa

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

1 2007ecaa taeseongkim
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.Tasks;
12
using Point = System.Drawing.Point;
13
using Size = System.Drawing.Size;
14
15
namespace Markus.Image
16
{
17
    public class ImageCompareBase : IDisposable
18
    {
19
        Mat OriginalImageData = null;
20
        Mat TargatImageData = null;
21
22
        double contoursLongCount = 0;
23
        double 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)
42
        {
43
            Mat result = new Mat();
44
45
            try
46
            {
47
                SetStatus("Image Load", 0, CompareStatus.Loading);
48
49
                Originalbitmap = ChangeBitmapFormatAndSize(Originalbitmap, Originalbitmap.Size, PixelFormat.Format24bppRgb);
50
                TargatBitmap = ChangeBitmapFormatAndSize(TargatBitmap, Originalbitmap.Size, PixelFormat.Format24bppRgb);
51
52
                // 원본이미지의 크키와 Format24bppRgb로 타켓 이미지를 변경
53
                // 크기가 틀린 경우 비교시 바이트배열 오류 발생
54
                OriginalImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(Originalbitmap);
55
                TargatImageData = OpenCvSharp.Extensions.BitmapConverter.ToMat(TargatBitmap);
56
57
                if (OriginalImageData.Size() != TargatImageData.Size())
58
                {
59
                    Cv2.Resize(TargatImageData, TargatImageData, OriginalImageData.Size());
60
                }
61
                
62
                Cv2.CvtColor(OriginalImageData, OriginalImageData, ColorConversionCodes.BGR2GRAY);
63
                Cv2.CvtColor(TargatImageData, TargatImageData, ColorConversionCodes.BGR2GRAY);
64
65
                Mat outputData = new Mat(TargatImageData.Size(), MatType.CV_8UC1);
66
67
                Cv2.Absdiff(OriginalImageData, TargatImageData, outputData);
68
69
                // 틀린부분을 반환
70
                Cv2.BitwiseNot(outputData, result);
71
            }
72
            catch (Exception ex)
73
            {
74
                throw ex;
75
            }
76
            finally
77
            {
78
            }
79
            
80
            return result;
81
        }
82
83
        protected System.Drawing.Bitmap ChangeBitmapFormatAndSize(System.Drawing.Bitmap bitmap, Size newSize, PixelFormat pixelFormat)
84
        {
85
            Bitmap result = bitmap;
86
87
            if (pixelFormat != bitmap.PixelFormat)
88
            {
89
                Point originPoint = new Point(0, 0);
90
                Rectangle rect = new Rectangle(originPoint, bitmap.Size);
91
                result = bitmap.Clone(rect, pixelFormat);
92
            }
93
94
            if (bitmap.Size != newSize)
95
            {
96
                result = new Bitmap(newSize.Width, newSize.Height);
97
98
                using (Graphics g = Graphics.FromImage(result))
99
                {
100
                    g.DrawImage(bitmap, 0, 0, newSize.Width, newSize.Height);
101
                    g.Dispose();
102
                }
103
            }
104
105
            return result;
106
        }
107
108
        /// <summary>
109
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
110
        /// </summary>
111
        /// <param name="data"></param>
112
        /// <param name="block"></param>
113
        /// <returns></returns>
114
        protected List<System.Windows.Rect> GetMatchPixels(Mat data, Size block)
115
        {
116
            SetStatus("Detection", 0, CompareStatus.Detection);
117
118
            int width = data.Cols;
119
            int height = data.Rows;
120
121
            List<System.Windows.Rect> results = new List<System.Windows.Rect>();
122
123
            System.Diagnostics.Stopwatch stopwatch = new System.Diagnostics.Stopwatch();
124
            stopwatch.Start();
125
126
            Mat outputMat = new Mat(data.Size(), MatType.CV_8UC1);
127
128
            HierarchyIndex[] hierarchy;
129
            OpenCvSharp.Point[][] contours;
130
131
            OpenCvSharp.Point testpoint = new OpenCvSharp.Point();
132
           
133
            Cv2.Threshold(data, data, 0, 30,ThresholdTypes.BinaryInv);
134
135
            Cv2.FindContours(data, out contours, out hierarchy,RetrievalModes.List,ContourApproximationModes.ApproxNone, testpoint);
136
137
            SetStatus("Comparison", 0, CompareStatus.Comparison);
138
            contoursLongCount = contours.Sum(x => x.Count());
139
140
            var rects = contours.AsParallel()
141
                                .Select(points => GetRectList(points, block))
142
                                .SelectMany(x => x);
143
144
            results.AddRange(rects);
145
             
146
            return results;
147
        }
148
149
        private List<System.Windows.Rect> GetRectList(OpenCvSharp.Point[] points,Size block)
150
        {
151
            List<System.Windows.Rect> result = new List<System.Windows.Rect>();
152
153
            for (int i = 0; i < points.Count(); i++)
154
            {
155
                var rect = new System.Windows.Rect
156
                {
157
                    X = points[i].X - block.Width / 2,
158
                    Y = points[i].Y - block.Height / 2,
159
                    Width = block.Width,
160
                    Height = block.Height
161
                };
162
163
                if (result.Count(r => r.IntersectsWith(rect)) == 0)
164
                {
165
                    result.Add(rect);
166
                }
167
168
                ComparisonCount++;
169
170
                SetStatus("Comparison", ComparisonCount/contoursLongCount*100, CompareStatus.Comparison);
171
            }
172
173
            return result;
174
        }
175
176
        /// <summary>
177
        /// Image<TColor, TDepth>의 틀린 부분을 Rect로 반환
178
        /// </summary>
179
        /// <param name="data"></param>
180
        /// <param name="block"></param>
181
        /// <returns></returns>
182
        protected async Task<List<System.Windows.Rect>> GetMatchPixelsAsnc(Mat data, Size block)
183
        {
184
            return await Task.Factory.StartNew<List<System.Windows.Rect>>(() =>
185
            {
186
                return GetMatchPixels(data, block);
187
            });
188
        }
189
190
        protected Bitmap LoadPicture(string url)
191
        {
192
            HttpWebRequest wreq;
193
            HttpWebResponse wresp;
194
            Stream mystream;
195
            Bitmap bmp;
196
197
            bmp = null;
198
            mystream = null;
199
            wresp = null;
200
            try
201
            {
202
                wreq = (HttpWebRequest)WebRequest.Create(url);
203
                wreq.AllowWriteStreamBuffering = true;
204
205
                wresp = (HttpWebResponse)wreq.GetResponse();
206
207
                if ((mystream = wresp.GetResponseStream()) != null)
208
                    bmp = new Bitmap(mystream);
209
            }
210
            finally
211
            {
212
                if (mystream != null)
213
                {
214
                    mystream.Close();
215
                    mystream.Dispose();
216
                }
217
218
                if (wresp != null)
219
                {
220
                    wresp.Close();
221
                    wresp.Dispose();
222
                }
223
            }
224
            return (bmp);
225
        }
226
227
        #region  IDisposable Support 
228
229
        private bool disposedValue = false;
230
231
        // 중복 호출을 검색하려면 
232
        protected virtual void Dispose(bool disposing)
233
        {
234
            if (!disposedValue)
235
            {
236
                if (disposing)
237
                {
238
                    // TODO: 관리되는 상태(관리되는 개체)를 삭제합니다. 
239
                }
240
                // TODO: 관리되지 않는 리소스(관리되지 않는 개체)를 해제하고 아래의 종료자를 재정의합니다. 
241
                // TODO: 큰 필드를 null로 설정합니다. 
242
                disposedValue = true;
243
            }
244
        }
245
246
        // TODO: 위의 Dispose(bool disposing)에 관리되지 않는 리소스를 해제하는 코드가 포함되어 있는 경우에만 종료자를 재정의합니다.
247
        // 
248
        //~DisposeClass()
249
       // {
250
            // // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
251
            // Dispose(false); // 
252
       // } 
253
254
        // 삭제 가능한 패턴을 올바르게 구현하기 위해 추가된 코드입니다. 
255
        public void Dispose()
256
        {
257
            // 이 코드를 변경하지 마세요. 위의 Dispose(bool disposing)에 정리 코드를 입력하세요. 
258
            Dispose(true);
259
260
261
            if (OriginalImageData != null)
262
            {
263
                OriginalImageData.Dispose();
264
            }
265
266
            if (TargatImageData != null)
267
            {
268
                TargatImageData.Dispose();
269
            }
270
271
            // TODO: 위의 종료자가 재정의된 경우 다음 코드 줄의 주석 처리를 제거합니다. 
272
            //  GC.SuppressFinalize(this);
273
            GC.Collect();
274
            GC.SuppressFinalize(this);
275
            GC.Collect();
276
        } 
277
        #endregion
278
    }
279
}
클립보드 이미지 추가 (최대 크기: 500 MB)