| // Copyright 2014 PDFium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| // Original code is licensed as follows: |
| /* |
| * Copyright 2010 ZXing authors |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "xfa/fxbarcode/common/BC_WhiteRectangleDetector.h" |
| |
| #include <memory> |
| |
| #include "xfa/fxbarcode/BC_ResultPoint.h" |
| #include "xfa/fxbarcode/common/BC_CommonBitMatrix.h" |
| #include "xfa/fxbarcode/utils.h" |
| |
| CBC_WhiteRectangleDetector::CBC_WhiteRectangleDetector( |
| CBC_CommonBitMatrix* image) { |
| m_image = image; |
| m_height = image->GetHeight(); |
| m_width = image->GetWidth(); |
| m_leftInit = (m_width - INIT_SIZE) >> 1; |
| m_rightInit = (m_width + INIT_SIZE) >> 1; |
| m_upInit = (m_height - INIT_SIZE) >> 1; |
| m_downInit = (m_height + INIT_SIZE) >> 1; |
| } |
| void CBC_WhiteRectangleDetector::Init(int32_t& e) { |
| if (m_upInit < 0 || m_leftInit < 0 || m_downInit >= m_height || |
| m_rightInit >= m_width) { |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnVoid(e); |
| } |
| } |
| CBC_WhiteRectangleDetector::CBC_WhiteRectangleDetector( |
| CBC_CommonBitMatrix* image, |
| int32_t initSize, |
| int32_t x, |
| int32_t y) { |
| m_image = image; |
| m_height = image->GetHeight(); |
| m_width = image->GetWidth(); |
| int32_t halfsize = initSize >> 1; |
| m_leftInit = x - halfsize; |
| m_rightInit = x + halfsize; |
| m_upInit = y - halfsize; |
| m_downInit = y + halfsize; |
| } |
| |
| CBC_WhiteRectangleDetector::~CBC_WhiteRectangleDetector() {} |
| |
| CFX_ArrayTemplate<CBC_ResultPoint*>* CBC_WhiteRectangleDetector::Detect( |
| int32_t& e) { |
| int32_t left = m_leftInit; |
| int32_t right = m_rightInit; |
| int32_t up = m_upInit; |
| int32_t down = m_downInit; |
| FX_BOOL sizeExceeded = FALSE; |
| FX_BOOL aBlackPointFoundOnBorder = TRUE; |
| FX_BOOL atLeastOneBlackPointFoundOnBorder = FALSE; |
| while (aBlackPointFoundOnBorder) { |
| aBlackPointFoundOnBorder = FALSE; |
| FX_BOOL rightBorderNotWhite = TRUE; |
| while (rightBorderNotWhite && right < m_width) { |
| rightBorderNotWhite = ContainsBlackPoint(up, down, right, FALSE); |
| if (rightBorderNotWhite) { |
| right++; |
| aBlackPointFoundOnBorder = TRUE; |
| } |
| } |
| if (right >= m_width) { |
| sizeExceeded = TRUE; |
| break; |
| } |
| FX_BOOL bottomBorderNotWhite = TRUE; |
| while (bottomBorderNotWhite && down < m_height) { |
| bottomBorderNotWhite = ContainsBlackPoint(left, right, down, TRUE); |
| if (bottomBorderNotWhite) { |
| down++; |
| aBlackPointFoundOnBorder = TRUE; |
| } |
| } |
| if (down >= m_height) { |
| sizeExceeded = TRUE; |
| break; |
| } |
| FX_BOOL leftBorderNotWhite = TRUE; |
| while (leftBorderNotWhite && left >= 0) { |
| leftBorderNotWhite = ContainsBlackPoint(up, down, left, FALSE); |
| if (leftBorderNotWhite) { |
| left--; |
| aBlackPointFoundOnBorder = TRUE; |
| } |
| } |
| if (left < 0) { |
| sizeExceeded = TRUE; |
| break; |
| } |
| FX_BOOL topBorderNotWhite = TRUE; |
| while (topBorderNotWhite && up >= 0) { |
| topBorderNotWhite = ContainsBlackPoint(left, right, up, TRUE); |
| if (topBorderNotWhite) { |
| up--; |
| aBlackPointFoundOnBorder = TRUE; |
| } |
| } |
| if (up < 0) { |
| sizeExceeded = TRUE; |
| break; |
| } |
| if (aBlackPointFoundOnBorder) { |
| atLeastOneBlackPointFoundOnBorder = TRUE; |
| } |
| } |
| if (!sizeExceeded && atLeastOneBlackPointFoundOnBorder) { |
| int32_t maxSize = right - left; |
| std::unique_ptr<CBC_ResultPoint> z; |
| for (int32_t i = 1; i < maxSize; i++) { |
| z.reset(GetBlackPointOnSegment((FX_FLOAT)left, (FX_FLOAT)(down - i), |
| (FX_FLOAT)(left + i), (FX_FLOAT)(down))); |
| if (z) |
| break; |
| } |
| if (!z.get()) { |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnValue(e, nullptr); |
| } |
| std::unique_ptr<CBC_ResultPoint> t; |
| for (int32_t j = 1; j < maxSize; j++) { |
| t.reset(GetBlackPointOnSegment((FX_FLOAT)left, (FX_FLOAT)(up + j), |
| (FX_FLOAT)(left + j), (FX_FLOAT)up)); |
| if (t) |
| break; |
| } |
| if (!t.get()) { |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnValue(e, nullptr); |
| } |
| std::unique_ptr<CBC_ResultPoint> x; |
| for (int32_t k = 1; k < maxSize; k++) { |
| x.reset(GetBlackPointOnSegment((FX_FLOAT)right, (FX_FLOAT)(up + k), |
| (FX_FLOAT)(right - k), (FX_FLOAT)up)); |
| if (x) |
| break; |
| } |
| if (!x.get()) { |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnValue(e, nullptr); |
| } |
| std::unique_ptr<CBC_ResultPoint> y; |
| for (int32_t m = 1; m < maxSize; m++) { |
| y.reset(GetBlackPointOnSegment((FX_FLOAT)right, (FX_FLOAT)(down - m), |
| (FX_FLOAT)(right - m), (FX_FLOAT)down)); |
| if (y) |
| break; |
| } |
| if (!y.get()) { |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnValue(e, nullptr); |
| } |
| return CenterEdges(y.get(), z.get(), x.get(), t.get()); |
| } |
| |
| e = BCExceptionNotFound; |
| BC_EXCEPTION_CHECK_ReturnValue(e, nullptr); |
| return nullptr; |
| } |
| |
| int32_t CBC_WhiteRectangleDetector::Round(FX_FLOAT d) { |
| return (int32_t)(d + 0.5f); |
| } |
| CBC_ResultPoint* CBC_WhiteRectangleDetector::GetBlackPointOnSegment( |
| FX_FLOAT aX, |
| FX_FLOAT aY, |
| FX_FLOAT bX, |
| FX_FLOAT bY) { |
| int32_t dist = DistanceL2(aX, aY, bX, bY); |
| float xStep = (bX - aX) / dist; |
| float yStep = (bY - aY) / dist; |
| for (int32_t i = 0; i < dist; i++) { |
| int32_t x = Round(aX + i * xStep); |
| int32_t y = Round(aY + i * yStep); |
| if (m_image->Get(x, y)) { |
| return new CBC_ResultPoint((FX_FLOAT)x, (FX_FLOAT)y); |
| } |
| } |
| return NULL; |
| } |
| int32_t CBC_WhiteRectangleDetector::DistanceL2(FX_FLOAT aX, |
| FX_FLOAT aY, |
| FX_FLOAT bX, |
| FX_FLOAT bY) { |
| float xDiff = aX - bX; |
| float yDiff = aY - bY; |
| return Round((float)sqrt(xDiff * xDiff + yDiff * yDiff)); |
| } |
| |
| CFX_ArrayTemplate<CBC_ResultPoint*>* CBC_WhiteRectangleDetector::CenterEdges( |
| CBC_ResultPoint* y, |
| CBC_ResultPoint* z, |
| CBC_ResultPoint* x, |
| CBC_ResultPoint* t) const { |
| CFX_ArrayTemplate<CBC_ResultPoint*>* result = |
| new CFX_ArrayTemplate<CBC_ResultPoint*>(); |
| result->SetSize(4); |
| |
| float yi = y->GetX(); |
| float yj = y->GetY(); |
| float zi = z->GetX(); |
| float zj = z->GetY(); |
| float xi = x->GetX(); |
| float xj = x->GetY(); |
| float ti = t->GetX(); |
| float tj = t->GetY(); |
| |
| if (yi < m_width / 2) { |
| (*result)[0] = new CBC_ResultPoint(ti - CORR, tj + CORR); |
| (*result)[1] = new CBC_ResultPoint(zi + CORR, zj + CORR); |
| (*result)[2] = new CBC_ResultPoint(xi - CORR, xj - CORR); |
| (*result)[3] = new CBC_ResultPoint(yi + CORR, yj - CORR); |
| return result; |
| } |
| |
| (*result)[0] = new CBC_ResultPoint(ti + CORR, tj + CORR); |
| (*result)[1] = new CBC_ResultPoint(zi + CORR, zj - CORR); |
| (*result)[2] = new CBC_ResultPoint(xi - CORR, xj + CORR); |
| (*result)[3] = new CBC_ResultPoint(yi - CORR, yj - CORR); |
| return result; |
| } |
| |
| FX_BOOL CBC_WhiteRectangleDetector::ContainsBlackPoint(int32_t a, |
| int32_t b, |
| int32_t fixed, |
| FX_BOOL horizontal) { |
| if (horizontal) { |
| for (int32_t x = a; x <= b; x++) { |
| if (m_image->Get(x, fixed)) { |
| return TRUE; |
| } |
| } |
| } else { |
| for (int32_t y = a; y <= b; y++) { |
| if (m_image->Get(fixed, y)) { |
| return TRUE; |
| } |
| } |
| } |
| return FALSE; |
| } |