OLD | NEW |
1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 // Original code is licensed as follows: | 6 // Original code is licensed as follows: |
7 /* | 7 /* |
8 * Copyright 2007 ZXing authors | 8 * Copyright 2007 ZXing authors |
9 * | 9 * |
10 * Licensed under the Apache License, Version 2.0 (the "License"); | 10 * Licensed under the Apache License, Version 2.0 (the "License"); |
(...skipping 24 matching lines...) Expand all Loading... |
35 const int32_t CBC_QRFinderPatternFinder::MIN_SKIP = 3; | 35 const int32_t CBC_QRFinderPatternFinder::MIN_SKIP = 3; |
36 const int32_t CBC_QRFinderPatternFinder::MAX_MODULES = 57; | 36 const int32_t CBC_QRFinderPatternFinder::MAX_MODULES = 57; |
37 const int32_t CBC_QRFinderPatternFinder::INTEGER_MATH_SHIFT = 8; | 37 const int32_t CBC_QRFinderPatternFinder::INTEGER_MATH_SHIFT = 8; |
38 | 38 |
39 CBC_QRFinderPatternFinder::CBC_QRFinderPatternFinder( | 39 CBC_QRFinderPatternFinder::CBC_QRFinderPatternFinder( |
40 CBC_CommonBitMatrix* image) { | 40 CBC_CommonBitMatrix* image) { |
41 m_image = image; | 41 m_image = image; |
42 m_crossCheckStateCount.SetSize(5); | 42 m_crossCheckStateCount.SetSize(5); |
43 m_hasSkipped = FALSE; | 43 m_hasSkipped = FALSE; |
44 } | 44 } |
| 45 |
45 CBC_QRFinderPatternFinder::~CBC_QRFinderPatternFinder() { | 46 CBC_QRFinderPatternFinder::~CBC_QRFinderPatternFinder() { |
46 for (int32_t i = 0; i < m_possibleCenters.GetSize(); i++) { | 47 for (int32_t i = 0; i < m_possibleCenters.GetSize(); i++) |
47 delete (CBC_QRFinderPattern*)m_possibleCenters[i]; | 48 delete m_possibleCenters[i]; |
48 } | |
49 m_possibleCenters.RemoveAll(); | |
50 } | 49 } |
| 50 |
51 class ClosestToAverageComparator { | 51 class ClosestToAverageComparator { |
52 private: | 52 private: |
53 FX_FLOAT m_averageModuleSize; | 53 FX_FLOAT m_averageModuleSize; |
54 | 54 |
55 public: | 55 public: |
56 ClosestToAverageComparator(FX_FLOAT averageModuleSize) | 56 ClosestToAverageComparator(FX_FLOAT averageModuleSize) |
57 : m_averageModuleSize(averageModuleSize) {} | 57 : m_averageModuleSize(averageModuleSize) {} |
58 int32_t operator()(FinderPattern* a, FinderPattern* b) { | 58 int32_t operator()(FinderPattern* a, FinderPattern* b) { |
59 FX_FLOAT dA = | 59 FX_FLOAT dA = |
60 (FX_FLOAT)fabs(a->GetEstimatedModuleSize() - m_averageModuleSize); | 60 (FX_FLOAT)fabs(a->GetEstimatedModuleSize() - m_averageModuleSize); |
(...skipping 12 matching lines...) Expand all Loading... |
73 return m_image; | 73 return m_image; |
74 } | 74 } |
75 CFX_Int32Array& CBC_QRFinderPatternFinder::GetCrossCheckStateCount() { | 75 CFX_Int32Array& CBC_QRFinderPatternFinder::GetCrossCheckStateCount() { |
76 m_crossCheckStateCount[0] = 0; | 76 m_crossCheckStateCount[0] = 0; |
77 m_crossCheckStateCount[1] = 0; | 77 m_crossCheckStateCount[1] = 0; |
78 m_crossCheckStateCount[2] = 0; | 78 m_crossCheckStateCount[2] = 0; |
79 m_crossCheckStateCount[3] = 0; | 79 m_crossCheckStateCount[3] = 0; |
80 m_crossCheckStateCount[4] = 0; | 80 m_crossCheckStateCount[4] = 0; |
81 return m_crossCheckStateCount; | 81 return m_crossCheckStateCount; |
82 } | 82 } |
83 CFX_PtrArray* CBC_QRFinderPatternFinder::GetPossibleCenters() { | 83 |
| 84 CFX_ArrayTemplate<CBC_QRFinderPattern*>* |
| 85 CBC_QRFinderPatternFinder::GetPossibleCenters() { |
84 return &m_possibleCenters; | 86 return &m_possibleCenters; |
85 } | 87 } |
| 88 |
86 CBC_QRFinderPatternInfo* CBC_QRFinderPatternFinder::Find(int32_t hint, | 89 CBC_QRFinderPatternInfo* CBC_QRFinderPatternFinder::Find(int32_t hint, |
87 int32_t& e) { | 90 int32_t& e) { |
88 int32_t maxI = m_image->GetHeight(); | 91 int32_t maxI = m_image->GetHeight(); |
89 int32_t maxJ = m_image->GetWidth(); | 92 int32_t maxJ = m_image->GetWidth(); |
90 int32_t iSkip = (3 * maxI) / (4 * MAX_MODULES); | 93 int32_t iSkip = (3 * maxI) / (4 * MAX_MODULES); |
91 if (iSkip < MIN_SKIP || 0) { | 94 if (iSkip < MIN_SKIP || 0) { |
92 iSkip = MIN_SKIP; | 95 iSkip = MIN_SKIP; |
93 } | 96 } |
94 FX_BOOL done = FALSE; | 97 FX_BOOL done = FALSE; |
95 CFX_Int32Array stateCount; | 98 CFX_Int32Array stateCount; |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
154 if (FoundPatternCross(stateCount)) { | 157 if (FoundPatternCross(stateCount)) { |
155 FX_BOOL confirmed = HandlePossibleCenter(stateCount, i, maxJ); | 158 FX_BOOL confirmed = HandlePossibleCenter(stateCount, i, maxJ); |
156 if (confirmed) { | 159 if (confirmed) { |
157 iSkip = stateCount[0]; | 160 iSkip = stateCount[0]; |
158 if (m_hasSkipped) { | 161 if (m_hasSkipped) { |
159 done = HaveMultiplyConfirmedCenters(); | 162 done = HaveMultiplyConfirmedCenters(); |
160 } | 163 } |
161 } | 164 } |
162 } | 165 } |
163 } | 166 } |
164 std::unique_ptr<CFX_PtrArray> patternInfo(SelectBestpatterns(e)); | 167 std::unique_ptr<CFX_ArrayTemplate<CBC_QRFinderPattern*>> patternInfo( |
| 168 SelectBestpatterns(e)); |
165 BC_EXCEPTION_CHECK_ReturnValue(e, NULL); | 169 BC_EXCEPTION_CHECK_ReturnValue(e, NULL); |
166 OrderBestPatterns(patternInfo.get()); | 170 OrderBestPatterns(patternInfo.get()); |
167 return new CBC_QRFinderPatternInfo(patternInfo.get()); | 171 return new CBC_QRFinderPatternInfo(patternInfo.get()); |
168 } | 172 } |
169 void CBC_QRFinderPatternFinder::OrderBestPatterns(CFX_PtrArray* patterns) { | 173 void CBC_QRFinderPatternFinder::OrderBestPatterns( |
170 FX_FLOAT abDistance = Distance((CBC_ResultPoint*)(*patterns)[0], | 174 CFX_ArrayTemplate<CBC_QRFinderPattern*>* patterns) { |
171 (CBC_ResultPoint*)(*patterns)[1]); | 175 FX_FLOAT abDistance = Distance((*patterns)[0], (*patterns)[1]); |
172 FX_FLOAT bcDistance = Distance((CBC_ResultPoint*)(*patterns)[1], | 176 FX_FLOAT bcDistance = Distance((*patterns)[1], (*patterns)[2]); |
173 (CBC_ResultPoint*)(*patterns)[2]); | 177 FX_FLOAT acDistance = Distance((*patterns)[0], (*patterns)[2]); |
174 FX_FLOAT acDistance = Distance((CBC_ResultPoint*)(*patterns)[0], | 178 CBC_QRFinderPattern* topLeft; |
175 (CBC_ResultPoint*)(*patterns)[2]); | 179 CBC_QRFinderPattern* topRight; |
176 CBC_QRFinderPattern *topLeft, *topRight, *bottomLeft; | 180 CBC_QRFinderPattern* bottomLeft; |
177 if (bcDistance >= abDistance && bcDistance >= acDistance) { | 181 if (bcDistance >= abDistance && bcDistance >= acDistance) { |
178 topLeft = (CBC_QRFinderPattern*)(*patterns)[0]; | 182 topLeft = (*patterns)[0]; |
179 topRight = (CBC_QRFinderPattern*)(*patterns)[1]; | 183 topRight = (*patterns)[1]; |
180 bottomLeft = (CBC_QRFinderPattern*)(*patterns)[2]; | 184 bottomLeft = (*patterns)[2]; |
181 } else if (acDistance >= bcDistance && acDistance >= abDistance) { | 185 } else if (acDistance >= bcDistance && acDistance >= abDistance) { |
182 topLeft = (CBC_QRFinderPattern*)(*patterns)[1]; | 186 topLeft = (*patterns)[1]; |
183 topRight = (CBC_QRFinderPattern*)(*patterns)[0]; | 187 topRight = (*patterns)[0]; |
184 bottomLeft = (CBC_QRFinderPattern*)(*patterns)[2]; | 188 bottomLeft = (*patterns)[2]; |
185 } else { | 189 } else { |
186 topLeft = (CBC_QRFinderPattern*)(*patterns)[2]; | 190 topLeft = (*patterns)[2]; |
187 topRight = (CBC_QRFinderPattern*)(*patterns)[0]; | 191 topRight = (*patterns)[0]; |
188 bottomLeft = (CBC_QRFinderPattern*)(*patterns)[1]; | 192 bottomLeft = (*patterns)[1]; |
189 } | 193 } |
190 if ((bottomLeft->GetY() - topLeft->GetY()) * | 194 if ((bottomLeft->GetY() - topLeft->GetY()) * |
191 (topRight->GetX() - topLeft->GetX()) < | 195 (topRight->GetX() - topLeft->GetX()) < |
192 (bottomLeft->GetX() - topLeft->GetX()) * | 196 (bottomLeft->GetX() - topLeft->GetX()) * |
193 (topRight->GetY() - topLeft->GetY())) { | 197 (topRight->GetY() - topLeft->GetY())) { |
194 CBC_QRFinderPattern* temp = topRight; | 198 CBC_QRFinderPattern* temp = topRight; |
195 topRight = bottomLeft; | 199 topRight = bottomLeft; |
196 bottomLeft = temp; | 200 bottomLeft = temp; |
197 } | 201 } |
198 (*patterns)[0] = bottomLeft; | 202 (*patterns)[0] = bottomLeft; |
(...skipping 169 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
368 FX_FLOAT centerI = | 372 FX_FLOAT centerI = |
369 CrossCheckVertical(i, (int32_t)centerJ, stateCount[2], stateCountTotal); | 373 CrossCheckVertical(i, (int32_t)centerJ, stateCount[2], stateCountTotal); |
370 if (!FXSYS_isnan(centerI)) { | 374 if (!FXSYS_isnan(centerI)) { |
371 centerJ = CrossCheckHorizontal((int32_t)centerJ, (int32_t)centerI, | 375 centerJ = CrossCheckHorizontal((int32_t)centerJ, (int32_t)centerI, |
372 stateCount[2], stateCountTotal); | 376 stateCount[2], stateCountTotal); |
373 if (!FXSYS_isnan(centerJ)) { | 377 if (!FXSYS_isnan(centerJ)) { |
374 FX_FLOAT estimatedModuleSize = (FX_FLOAT)stateCountTotal / 7.0f; | 378 FX_FLOAT estimatedModuleSize = (FX_FLOAT)stateCountTotal / 7.0f; |
375 FX_BOOL found = FALSE; | 379 FX_BOOL found = FALSE; |
376 int32_t max = m_possibleCenters.GetSize(); | 380 int32_t max = m_possibleCenters.GetSize(); |
377 for (int32_t index = 0; index < max; index++) { | 381 for (int32_t index = 0; index < max; index++) { |
378 CBC_QRFinderPattern* center = | 382 CBC_QRFinderPattern* center = m_possibleCenters[index]; |
379 (CBC_QRFinderPattern*)(m_possibleCenters[index]); | |
380 if (center->AboutEquals(estimatedModuleSize, centerI, centerJ)) { | 383 if (center->AboutEquals(estimatedModuleSize, centerI, centerJ)) { |
381 center->IncrementCount(); | 384 center->IncrementCount(); |
382 found = TRUE; | 385 found = TRUE; |
383 break; | 386 break; |
384 } | 387 } |
385 } | 388 } |
386 if (!found) { | 389 if (!found) { |
387 m_possibleCenters.Add( | 390 m_possibleCenters.Add( |
388 new CBC_QRFinderPattern(centerJ, centerI, estimatedModuleSize)); | 391 new CBC_QRFinderPattern(centerJ, centerI, estimatedModuleSize)); |
389 } | 392 } |
390 return TRUE; | 393 return TRUE; |
391 } | 394 } |
392 } | 395 } |
393 return FALSE; | 396 return FALSE; |
394 } | 397 } |
395 int32_t CBC_QRFinderPatternFinder::FindRowSkip() { | 398 int32_t CBC_QRFinderPatternFinder::FindRowSkip() { |
396 int32_t max = m_possibleCenters.GetSize(); | 399 int32_t max = m_possibleCenters.GetSize(); |
397 if (max <= 1) { | 400 if (max <= 1) { |
398 return 0; | 401 return 0; |
399 } | 402 } |
400 FinderPattern* firstConfirmedCenter = NULL; | 403 FinderPattern* firstConfirmedCenter = NULL; |
401 for (int32_t i = 0; i < max; i++) { | 404 for (int32_t i = 0; i < max; i++) { |
402 CBC_QRFinderPattern* center = (CBC_QRFinderPattern*)m_possibleCenters[i]; | 405 CBC_QRFinderPattern* center = m_possibleCenters[i]; |
403 if (center->GetCount() >= CENTER_QUORUM) { | 406 if (center->GetCount() >= CENTER_QUORUM) { |
404 if (firstConfirmedCenter == NULL) { | 407 if (firstConfirmedCenter == NULL) { |
405 firstConfirmedCenter = center; | 408 firstConfirmedCenter = center; |
406 } else { | 409 } else { |
407 m_hasSkipped = TRUE; | 410 m_hasSkipped = TRUE; |
408 return (int32_t)((fabs(firstConfirmedCenter->GetX() - center->GetX()) - | 411 return (int32_t)((fabs(firstConfirmedCenter->GetX() - center->GetX()) - |
409 fabs(firstConfirmedCenter->GetY() - center->GetY())) / | 412 fabs(firstConfirmedCenter->GetY() - center->GetY())) / |
410 2); | 413 2); |
411 } | 414 } |
412 } | 415 } |
413 } | 416 } |
414 return 0; | 417 return 0; |
415 } | 418 } |
416 FX_BOOL CBC_QRFinderPatternFinder::HaveMultiplyConfirmedCenters() { | 419 FX_BOOL CBC_QRFinderPatternFinder::HaveMultiplyConfirmedCenters() { |
417 int32_t confirmedCount = 0; | 420 int32_t confirmedCount = 0; |
418 FX_FLOAT totalModuleSize = 0.0f; | 421 FX_FLOAT totalModuleSize = 0.0f; |
419 int32_t max = m_possibleCenters.GetSize(); | 422 int32_t max = m_possibleCenters.GetSize(); |
420 int32_t i; | 423 int32_t i; |
421 for (i = 0; i < max; i++) { | 424 for (i = 0; i < max; i++) { |
422 CBC_QRFinderPattern* pattern = (CBC_QRFinderPattern*)m_possibleCenters[i]; | 425 CBC_QRFinderPattern* pattern = m_possibleCenters[i]; |
423 if (pattern->GetCount() >= CENTER_QUORUM) { | 426 if (pattern->GetCount() >= CENTER_QUORUM) { |
424 confirmedCount++; | 427 confirmedCount++; |
425 totalModuleSize += pattern->GetEstimatedModuleSize(); | 428 totalModuleSize += pattern->GetEstimatedModuleSize(); |
426 } | 429 } |
427 } | 430 } |
428 if (confirmedCount < 3) { | 431 if (confirmedCount < 3) { |
429 return FALSE; | 432 return FALSE; |
430 } | 433 } |
431 FX_FLOAT average = totalModuleSize / (FX_FLOAT)max; | 434 FX_FLOAT average = totalModuleSize / (FX_FLOAT)max; |
432 FX_FLOAT totalDeviation = 0.0f; | 435 FX_FLOAT totalDeviation = 0.0f; |
433 for (i = 0; i < max; i++) { | 436 for (i = 0; i < max; i++) { |
434 CBC_QRFinderPattern* pattern = (CBC_QRFinderPattern*)m_possibleCenters[i]; | 437 CBC_QRFinderPattern* pattern = m_possibleCenters[i]; |
435 totalDeviation += fabs(pattern->GetEstimatedModuleSize() - average); | 438 totalDeviation += fabs(pattern->GetEstimatedModuleSize() - average); |
436 } | 439 } |
437 return totalDeviation <= 0.05f * totalModuleSize; | 440 return totalDeviation <= 0.05f * totalModuleSize; |
438 } | 441 } |
439 inline FX_BOOL centerComparator(void* a, void* b) { | 442 |
440 return ((CBC_QRFinderPattern*)b)->GetCount() < | 443 CFX_ArrayTemplate<CBC_QRFinderPattern*>* |
441 ((CBC_QRFinderPattern*)a)->GetCount(); | 444 CBC_QRFinderPatternFinder::SelectBestpatterns(int32_t& e) { |
442 } | |
443 CFX_PtrArray* CBC_QRFinderPatternFinder::SelectBestpatterns(int32_t& e) { | |
444 int32_t startSize = m_possibleCenters.GetSize(); | 445 int32_t startSize = m_possibleCenters.GetSize(); |
445 if (m_possibleCenters.GetSize() < 3) { | 446 if (m_possibleCenters.GetSize() < 3) { |
446 e = BCExceptionRead; | 447 e = BCExceptionRead; |
447 BC_EXCEPTION_CHECK_ReturnValue(e, NULL); | 448 BC_EXCEPTION_CHECK_ReturnValue(e, NULL); |
448 } | 449 } |
449 FX_FLOAT average = 0.0f; | 450 FX_FLOAT average = 0.0f; |
450 if (startSize > 3) { | 451 if (startSize > 3) { |
451 FX_FLOAT totalModuleSize = 0.0f; | 452 FX_FLOAT totalModuleSize = 0.0f; |
452 for (int32_t i = 0; i < startSize; i++) { | 453 for (int32_t i = 0; i < startSize; i++) |
453 totalModuleSize += ((CBC_QRFinderPattern*)m_possibleCenters[i]) | 454 totalModuleSize += m_possibleCenters[i]->GetEstimatedModuleSize(); |
454 ->GetEstimatedModuleSize(); | 455 |
455 } | |
456 average = totalModuleSize / (FX_FLOAT)startSize; | 456 average = totalModuleSize / (FX_FLOAT)startSize; |
457 for (int32_t j = 0; | 457 for (int32_t j = 0; |
458 j < m_possibleCenters.GetSize() && m_possibleCenters.GetSize() > 3; | 458 j < m_possibleCenters.GetSize() && m_possibleCenters.GetSize() > 3; |
459 j++) { | 459 j++) { |
460 CBC_QRFinderPattern* pattern = (CBC_QRFinderPattern*)m_possibleCenters[j]; | 460 CBC_QRFinderPattern* pattern = m_possibleCenters[j]; |
461 if (fabs(pattern->GetEstimatedModuleSize() - average) > 0.2f * average) { | 461 if (fabs(pattern->GetEstimatedModuleSize() - average) > 0.2f * average) { |
462 delete pattern; | 462 delete pattern; |
463 m_possibleCenters.RemoveAt(j); | 463 m_possibleCenters.RemoveAt(j); |
464 j--; | 464 j--; |
465 } | 465 } |
466 } | 466 } |
467 } | 467 } |
468 if (m_possibleCenters.GetSize() > 3) { | 468 if (m_possibleCenters.GetSize() > 3) { |
469 BC_FX_PtrArray_Sort(m_possibleCenters, centerComparator); | 469 std::sort(m_possibleCenters.GetData(), |
| 470 m_possibleCenters.GetData() + m_possibleCenters.GetSize(), |
| 471 [](const CBC_QRFinderPattern* a, CBC_QRFinderPattern* b) { |
| 472 return a->GetCount() > b->GetCount(); // e.g. Descending. |
| 473 }); |
470 } | 474 } |
471 CFX_PtrArray* vec = new CFX_PtrArray(); | 475 CFX_ArrayTemplate<CBC_QRFinderPattern*>* vec = |
| 476 new CFX_ArrayTemplate<CBC_QRFinderPattern*>(); |
472 vec->SetSize(3); | 477 vec->SetSize(3); |
473 (*vec)[0] = ((CBC_QRFinderPattern*)m_possibleCenters[0])->Clone(); | 478 (*vec)[0] = m_possibleCenters[0]->Clone(); |
474 (*vec)[1] = ((CBC_QRFinderPattern*)m_possibleCenters[1])->Clone(); | 479 (*vec)[1] = m_possibleCenters[1]->Clone(); |
475 (*vec)[2] = ((CBC_QRFinderPattern*)m_possibleCenters[2])->Clone(); | 480 (*vec)[2] = m_possibleCenters[2]->Clone(); |
476 return vec; | 481 return vec; |
477 } | 482 } |
OLD | NEW |