1// Copyright 2014 PDFium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6// Original code is licensed as follows:
7/*
8 * Copyright 2010 ZXing authors
9 *
10 * Licensed under the Apache License, Version 2.0 (the "License");
11 * you may not use this file except in compliance with the License.
12 * You may obtain a copy of the License at
13 *
14 *      http://www.apache.org/licenses/LICENSE-2.0
15 *
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 * See the License for the specific language governing permissions and
20 * limitations under the License.
21 */
22
23#include "xfa/src/fxbarcode/barcode.h"
24#include "xfa/src/fxbarcode/common/BC_WhiteRectangleDetector.h"
25#include "xfa/src/fxbarcode/common/BC_CommonBitMatrix.h"
26#include "xfa/src/fxbarcode/BC_ResultPoint.h"
27const int32_t CBC_WhiteRectangleDetector::INIT_SIZE = 30;
28const int32_t CBC_WhiteRectangleDetector::CORR = 1;
29CBC_WhiteRectangleDetector::CBC_WhiteRectangleDetector(
30    CBC_CommonBitMatrix* image) {
31  m_image = image;
32  m_height = image->GetHeight();
33  m_width = image->GetWidth();
34  m_leftInit = (m_width - INIT_SIZE) >> 1;
35  m_rightInit = (m_width + INIT_SIZE) >> 1;
36  m_upInit = (m_height - INIT_SIZE) >> 1;
37  m_downInit = (m_height + INIT_SIZE) >> 1;
38}
39void CBC_WhiteRectangleDetector::Init(int32_t& e) {
40  if (m_upInit < 0 || m_leftInit < 0 || m_downInit >= m_height ||
41      m_rightInit >= m_width) {
42    e = BCExceptionNotFound;
43    BC_EXCEPTION_CHECK_ReturnVoid(e);
44  }
45}
46CBC_WhiteRectangleDetector::CBC_WhiteRectangleDetector(
47    CBC_CommonBitMatrix* image,
48    int32_t initSize,
49    int32_t x,
50    int32_t y) {
51  m_image = image;
52  m_height = image->GetHeight();
53  m_width = image->GetWidth();
54  int32_t halfsize = initSize >> 1;
55  m_leftInit = x - halfsize;
56  m_rightInit = x + halfsize;
57  m_upInit = y - halfsize;
58  m_downInit = y + halfsize;
59}
60CBC_WhiteRectangleDetector::~CBC_WhiteRectangleDetector() {}
61CFX_PtrArray* CBC_WhiteRectangleDetector::Detect(int32_t& e) {
62  int32_t left = m_leftInit;
63  int32_t right = m_rightInit;
64  int32_t up = m_upInit;
65  int32_t down = m_downInit;
66  FX_BOOL sizeExceeded = FALSE;
67  FX_BOOL aBlackPointFoundOnBorder = TRUE;
68  FX_BOOL atLeastOneBlackPointFoundOnBorder = FALSE;
69  while (aBlackPointFoundOnBorder) {
70    aBlackPointFoundOnBorder = FALSE;
71    FX_BOOL rightBorderNotWhite = TRUE;
72    while (rightBorderNotWhite && right < m_width) {
73      rightBorderNotWhite = ContainsBlackPoint(up, down, right, FALSE);
74      if (rightBorderNotWhite) {
75        right++;
76        aBlackPointFoundOnBorder = TRUE;
77      }
78    }
79    if (right >= m_width) {
80      sizeExceeded = TRUE;
81      break;
82    }
83    FX_BOOL bottomBorderNotWhite = TRUE;
84    while (bottomBorderNotWhite && down < m_height) {
85      bottomBorderNotWhite = ContainsBlackPoint(left, right, down, TRUE);
86      if (bottomBorderNotWhite) {
87        down++;
88        aBlackPointFoundOnBorder = TRUE;
89      }
90    }
91    if (down >= m_height) {
92      sizeExceeded = TRUE;
93      break;
94    }
95    FX_BOOL leftBorderNotWhite = TRUE;
96    while (leftBorderNotWhite && left >= 0) {
97      leftBorderNotWhite = ContainsBlackPoint(up, down, left, FALSE);
98      if (leftBorderNotWhite) {
99        left--;
100        aBlackPointFoundOnBorder = TRUE;
101      }
102    }
103    if (left < 0) {
104      sizeExceeded = TRUE;
105      break;
106    }
107    FX_BOOL topBorderNotWhite = TRUE;
108    while (topBorderNotWhite && up >= 0) {
109      topBorderNotWhite = ContainsBlackPoint(left, right, up, TRUE);
110      if (topBorderNotWhite) {
111        up--;
112        aBlackPointFoundOnBorder = TRUE;
113      }
114    }
115    if (up < 0) {
116      sizeExceeded = TRUE;
117      break;
118    }
119    if (aBlackPointFoundOnBorder) {
120      atLeastOneBlackPointFoundOnBorder = TRUE;
121    }
122  }
123  if (!sizeExceeded && atLeastOneBlackPointFoundOnBorder) {
124    int32_t maxSize = right - left;
125    CBC_AutoPtr<CBC_ResultPoint> z(NULL);
126    for (int32_t i = 1; i < maxSize; i++) {
127      z = CBC_AutoPtr<CBC_ResultPoint>(
128          GetBlackPointOnSegment((FX_FLOAT)left, (FX_FLOAT)(down - i),
129                                 (FX_FLOAT)(left + i), (FX_FLOAT)(down)));
130      if (z.get() != NULL) {
131        break;
132      }
133    }
134    if (z.get() == NULL) {
135      e = BCExceptionNotFound;
136      BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
137    }
138    CBC_AutoPtr<CBC_ResultPoint> t(NULL);
139    for (int32_t j = 1; j < maxSize; j++) {
140      t = CBC_AutoPtr<CBC_ResultPoint>(
141          GetBlackPointOnSegment((FX_FLOAT)left, (FX_FLOAT)(up + j),
142                                 (FX_FLOAT)(left + j), (FX_FLOAT)up));
143      if (t.get() != NULL) {
144        break;
145      }
146    }
147    if (t.get() == NULL) {
148      e = BCExceptionNotFound;
149      BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
150    }
151    CBC_AutoPtr<CBC_ResultPoint> x(NULL);
152    for (int32_t k = 1; k < maxSize; k++) {
153      x = CBC_AutoPtr<CBC_ResultPoint>(
154          GetBlackPointOnSegment((FX_FLOAT)right, (FX_FLOAT)(up + k),
155                                 (FX_FLOAT)(right - k), (FX_FLOAT)up));
156      if (x.get() != NULL) {
157        break;
158      }
159    }
160    if (x.get() == NULL) {
161      e = BCExceptionNotFound;
162      BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
163    }
164    CBC_AutoPtr<CBC_ResultPoint> y(NULL);
165    for (int32_t m = 1; m < maxSize; m++) {
166      y = CBC_AutoPtr<CBC_ResultPoint>(
167          GetBlackPointOnSegment((FX_FLOAT)right, (FX_FLOAT)(down - m),
168                                 (FX_FLOAT)(right - m), (FX_FLOAT)down));
169      if (y.get() != NULL) {
170        break;
171      }
172    }
173    if (y.get() == NULL) {
174      e = BCExceptionNotFound;
175      BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
176    }
177    return CenterEdges(y.get(), z.get(), x.get(), t.get());
178  } else {
179    e = BCExceptionNotFound;
180    BC_EXCEPTION_CHECK_ReturnValue(e, NULL);
181  }
182  return NULL;
183}
184int32_t CBC_WhiteRectangleDetector::Round(FX_FLOAT d) {
185  return (int32_t)(d + 0.5f);
186}
187CBC_ResultPoint* CBC_WhiteRectangleDetector::GetBlackPointOnSegment(
188    FX_FLOAT aX,
189    FX_FLOAT aY,
190    FX_FLOAT bX,
191    FX_FLOAT bY) {
192  int32_t dist = DistanceL2(aX, aY, bX, bY);
193  float xStep = (bX - aX) / dist;
194  float yStep = (bY - aY) / dist;
195  for (int32_t i = 0; i < dist; i++) {
196    int32_t x = Round(aX + i * xStep);
197    int32_t y = Round(aY + i * yStep);
198    if (m_image->Get(x, y)) {
199      return new CBC_ResultPoint((FX_FLOAT)x, (FX_FLOAT)y);
200    }
201  }
202  return NULL;
203}
204int32_t CBC_WhiteRectangleDetector::DistanceL2(FX_FLOAT aX,
205                                               FX_FLOAT aY,
206                                               FX_FLOAT bX,
207                                               FX_FLOAT bY) {
208  float xDiff = aX - bX;
209  float yDiff = aY - bY;
210  return Round((float)sqrt(xDiff * xDiff + yDiff * yDiff));
211}
212CFX_PtrArray* CBC_WhiteRectangleDetector::CenterEdges(CBC_ResultPoint* y,
213                                                      CBC_ResultPoint* z,
214                                                      CBC_ResultPoint* x,
215                                                      CBC_ResultPoint* t) {
216  float yi = y->GetX();
217  float yj = y->GetY();
218  float zi = z->GetX();
219  float zj = z->GetY();
220  float xi = x->GetX();
221  float xj = x->GetY();
222  float ti = t->GetX();
223  float tj = t->GetY();
224  if (yi < m_width / 2) {
225    CFX_PtrArray* result = new CFX_PtrArray;
226    result->SetSize(4);
227    (*result)[0] = new CBC_ResultPoint(ti - CORR, tj + CORR);
228    (*result)[1] = new CBC_ResultPoint(zi + CORR, zj + CORR);
229    (*result)[2] = new CBC_ResultPoint(xi - CORR, xj - CORR);
230    (*result)[3] = new CBC_ResultPoint(yi + CORR, yj - CORR);
231    return result;
232  } else {
233    CFX_PtrArray* result = new CFX_PtrArray;
234    result->SetSize(4);
235    (*result)[0] = new CBC_ResultPoint(ti + CORR, tj + CORR);
236    (*result)[1] = new CBC_ResultPoint(zi + CORR, zj - CORR);
237    (*result)[2] = new CBC_ResultPoint(xi - CORR, xj + CORR);
238    (*result)[3] = new CBC_ResultPoint(yi - CORR, yj - CORR);
239    return result;
240  }
241}
242FX_BOOL CBC_WhiteRectangleDetector::ContainsBlackPoint(int32_t a,
243                                                       int32_t b,
244                                                       int32_t fixed,
245                                                       FX_BOOL horizontal) {
246  if (horizontal) {
247    for (int32_t x = a; x <= b; x++) {
248      if (m_image->Get(x, fixed)) {
249        return TRUE;
250      }
251    }
252  } else {
253    for (int32_t y = a; y <= b; y++) {
254      if (m_image->Get(fixed, y)) {
255        return TRUE;
256      }
257    }
258  }
259  return FALSE;
260}
261