Fixed C++ port's handling of reversed barcodes:
[zxing.git] / cpp / core / src / zxing / oned / OneDReader.cpp
1 /*
2  *  OneDReader.cpp
3  *  ZXing
4  *
5  *  Created by Lukasz Warchol on 10-01-15.
6  *  Copyright 2010 ZXing authors All rights reserved.
7  *
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *      http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  */
20
21 #include "OneDReader.h"
22 #include <zxing/ReaderException.h>
23 #include <zxing/oned/OneDResultPoint.h>
24 #include <math.h>
25 #include <limits.h>
26
27 namespace zxing {
28         namespace oned {
29                 using namespace std;
30                 
31                 OneDReader::OneDReader() {
32                 }
33                 
34                 Ref<Result> OneDReader::decode(Ref<BinaryBitmap> image) {
35                         try {
36                                 return doDecode(image);
37                         }catch (ReaderException re) {
38                                 if (false /*tryHarder && image.isRotateSupported()*/) {
39                                         /*
40                                         BinaryBitmap rotatedImage = image.rotateCounterClockwise();
41                                         Result result = doDecode(rotatedImage, hints);
42                                         // Record that we found it rotated 90 degrees CCW / 270 degrees CW
43                                         Hashtable metadata = result.getResultMetadata();
44                                         int orientation = 270;
45                                         if (metadata != null && metadata.containsKey(ResultMetadataType.ORIENTATION)) {
46                                                 // But if we found it reversed in doDecode(), add in that result here:
47                                                 orientation = (orientation +
48                                                                            ((Integer) metadata.get(ResultMetadataType.ORIENTATION)).intValue()) % 360;
49                                         }
50                                         result.putMetadata(ResultMetadataType.ORIENTATION, new Integer(orientation));
51                                         // Update result points
52                                         ResultPoint[] points = result.getResultPoints();
53                                         int height = rotatedImage.getHeight();
54                                         for (int i = 0; i < points.length; i++) {
55                                                 points[i] = new ResultPoint(height - points[i].getY() - 1, points[i].getX());
56                                         }
57                                         return result;
58                                         */
59                                 } else {
60                                         throw re;
61                                 }
62                         }
63                 }
64                 
65                 Ref<Result> OneDReader::doDecode(Ref<BinaryBitmap> image){
66                         int width = image->getWidth();
67                         int height = image->getHeight();
68                         Ref<BitArray> row(new BitArray(width));
69 //                      BitArray row = new BitArray(width);
70                         
71                         int middle = height >> 1;
72                         bool tryHarder = true;//hints != null && hints.containsKey(DecodeHintType.TRY_HARDER);
73                         int rowStep = (int)fmax(1, height >> (tryHarder ? 7 : 4));
74                         int maxLines;
75                         if (tryHarder) {
76                                 maxLines = height; // Look at the whole image, not just the center
77                         } else {
78                                 maxLines = 9; // Nine rows spaced 1/16 apart is roughly the middle half of the image
79                         }
80                         
81                         for (int x = 0; x < maxLines; x++) {
82                                 
83                                 // Scanning from the middle out. Determine which row we're looking at next:
84                                 int rowStepsAboveOrBelow = (x + 1) >> 1;
85                                 bool isAbove = (x & 0x01) == 0; // i.e. is x even?
86                                 int rowNumber = middle + rowStep * (isAbove ? rowStepsAboveOrBelow : -rowStepsAboveOrBelow);
87                                 if (rowNumber < 0 || rowNumber >= height) {
88                                         // Oops, if we run off the top or bottom, stop
89                                         break;
90                                 }
91                                         
92                                 // Estimate black point for this row and load it:
93                                 try {
94                                         row = image->getBlackRow(rowNumber, row);
95                                 }catch (ReaderException re) {
96                                         continue;
97                                 }catch (IllegalArgumentException re) {
98                                         continue;
99                                 }
100                                 
101                                 // While we have the image data in a BitArray, it's fairly cheap to reverse it in place to
102                                 // handle decoding upside down barcodes.
103                                 for (int attempt = 0; attempt < 2; attempt++) {
104                                         if (attempt == 1) { // trying again?
105                                                 row->reverse(); // reverse the row and continue
106                                         }
107                                         try {
108                                                 // Look for a barcode
109                                                 Ref<Result> result = decodeRow(rowNumber, row);
110                                                 // We found our barcode
111                                                 if (attempt == 1) {
112                                                         //                                              // But it was upside down, so note that
113                                                         //                                              result.putMetadata(ResultMetadataType.ORIENTATION, new Integer(180));
114                                                         //                                              // And remember to flip the result points horizontally.
115                                                         std::vector<Ref<ResultPoint> > points(result->getResultPoints());
116                                                         if (points.size() == 2) {
117                                                                 Ref<ResultPoint> pointZero(new OneDResultPoint(width - points[0]->getX() - 1, points[0]->getY()));
118                                                                 points[0] = pointZero;
119
120                                                                 Ref<ResultPoint> pointOne(new OneDResultPoint(width - points[1]->getX() - 1, points[1]->getY()));
121                                                                 points[1] = pointOne;
122
123                 result.reset(new Result(result->getText(),result->getRawBytes(),points,result->getBarcodeFormat()));
124                                                         }
125                                                                 
126                                                 }
127                                                 return result;
128                                         } catch (ReaderException re) {
129                                                 // continue -- just couldn't decode this row
130                                         }
131                                 }
132                         }
133                         throw ReaderException("doDecode() failed");
134                 }
135                 
136                 unsigned int OneDReader::patternMatchVariance(int counters[], int countersSize, const int pattern[], int maxIndividualVariance) {
137                         int numCounters = countersSize;
138                         unsigned int total = 0;
139                         unsigned int patternLength = 0;
140                         for (int i = 0; i < numCounters; i++) {
141                                 total += counters[i];
142                                 patternLength += pattern[i];
143                         }
144                         if (total < patternLength) {
145                                 // If we don't even have one pixel per unit of bar width, assume this is too small
146                                 // to reliably match, so fail:
147                                 return INT_MAX;
148                         }
149                         // We're going to fake floating-point math in integers. We just need to use more bits.
150                         // Scale up patternLength so that intermediate values below like scaledCounter will have
151                         // more "significant digits"
152                         unsigned int unitBarWidth = (total << INTEGER_MATH_SHIFT) / patternLength;
153                         maxIndividualVariance = (maxIndividualVariance * unitBarWidth) >> INTEGER_MATH_SHIFT;
154                         
155                         unsigned int totalVariance = 0;
156                         for (int x = 0; x < numCounters; x++) {
157                                 int counter = counters[x] << INTEGER_MATH_SHIFT;
158                                 int scaledPattern = pattern[x] * unitBarWidth;
159                                 int variance = counter > scaledPattern ? counter - scaledPattern : scaledPattern - counter;
160                                 if (variance > maxIndividualVariance) {
161                                         return INT_MAX;
162                                 }
163                                 totalVariance += variance;
164                         }
165                         return totalVariance / total;
166                 }
167                 
168                 void OneDReader::recordPattern(Ref<BitArray> row, int start, int counters[], int countersCount){
169                         int numCounters = countersCount;//sizeof(counters) / sizeof(int);
170                         for (int i = 0; i < numCounters; i++) {
171                                 counters[i] = 0;
172                         }
173                         int end = row->getSize();
174                         if (start >= end) {
175                                 throw ReaderException("recordPattern: start >= end");
176                         }
177                         bool isWhite = !row->get(start);
178                         int counterPosition = 0;
179                         int i = start;
180                         while (i < end) {
181                                 bool pixel = row->get(i);
182                                 if (pixel ^ isWhite) { // that is, exactly one is true
183                                         counters[counterPosition]++;
184                                 } else {
185                                         counterPosition++;
186                                         if (counterPosition == numCounters) {
187                                                 break;
188                                         } else {
189                                                 counters[counterPosition] = 1;
190                                                 isWhite ^= true; // isWhite = !isWhite;
191                                         }
192                                 }
193                                 i++;
194                         }
195                         // If we read fully the last section of pixels and filled up our counters -- or filled
196                         // the last counter but ran off the side of the image, OK. Otherwise, a problem.
197                         if (!(counterPosition == numCounters || (counterPosition == numCounters - 1 && i == end))) {
198                                 throw ReaderException("recordPattern");
199                         }
200                 }
201                 
202                 OneDReader::~OneDReader() {
203                 }
204         }
205 }