Updates to C++ port:

- updated Binarizer, BinaryBitmap, and LuminanceSource implementations to match Java
 - updated Magick client

git-svn-id: https://zxing.googlecode.com/svn/trunk@1480 59b500cc-1b3d-0410-9834-0bbf25fbcc57
This commit is contained in:
flyashi 2010-07-13 18:43:54 +00:00
parent 4ed7e4ac96
commit 5a36cea5af
15 changed files with 205 additions and 378 deletions

35
cpp/blackboxtest.sh Executable file
View file

@ -0,0 +1,35 @@
#!/bin/sh
blackboxpath="../core/test/data/blackbox"
formats="ean13 ean8 upce upca qrcode"
passed=0;
failed=0;
oldcat="";
for format in $formats; do
for pic in `ls ${blackboxpath}/${format}-*/*.{jpg,JPG} 2>/dev/null`; do
category=${pic%/*};
category=${category##*/};
if [ "$oldcat" != "$category" ]; then
echo "***** $oldcat finished - $passed of $((passed+failed)) passed **** ***** ******* ***** *********************"
oldcat=$category;
passed=0;
failed=0;
fi
echo -n "Processing: $pic ... "
tmp="${pic%JPG}";
txt="${tmp%jpg}txt";
expected=`cat "$txt"`;
actual=`build/zxing . $pic`;
if [ "$expected" == "$actual" ]; then
echo "passed."
passed=$((passed+1));
else
echo -e "FAILED\n Expected: $expected\n Detected: $actual"
failed=$((failed+1));
fi
done
done
echo "***** $oldcat finished - $passed of $((passed+failed)) passed **** ***** ******* ***** *********************"

View file

@ -23,27 +23,13 @@
namespace zxing {
Binarizer::Binarizer(Ref<LuminanceSource> source) : source_(source), array_(NULL), matrix_(NULL), cached_y_(-1) {
Binarizer::Binarizer(Ref<LuminanceSource> source) : source_(source) {
}
Binarizer::~Binarizer() {
}
Ref<BitArray> Binarizer::getBlackRow(int y, Ref<BitArray> row){
if (array_ == NULL && cached_y_ != y) {
array_ = estimateBlackRow(y, row);
cached_y_ = y;
}
return array_;
}
Ref<BitMatrix> Binarizer::getBlackMatrix() {
if (matrix_ == NULL)
matrix_ = estimateBlackMatrix();
return matrix_;
}
Ref<LuminanceSource> Binarizer::getSource() {
Ref<LuminanceSource> Binarizer::getLuminanceSource() const {
return source_;
}

View file

@ -32,20 +32,16 @@ namespace zxing {
class Binarizer : public Counted {
private:
Ref<LuminanceSource> source_;
Ref<BitArray> array_;
Ref<BitMatrix> matrix_;
int cached_y_;
public:
Binarizer(Ref<LuminanceSource> source);
virtual ~Binarizer();
virtual Ref<BitArray> estimateBlackRow(int y, Ref<BitArray> row)=0;
Ref<BitArray> getBlackRow(int y, Ref<BitArray> row);
virtual Ref<BitArray> getBlackRow(int y, Ref<BitArray> row) = 0;
virtual Ref<BitMatrix> getBlackMatrix() = 0;
virtual Ref<BitMatrix> estimateBlackMatrix() = 0;
Ref<BitMatrix> getBlackMatrix();
Ref<LuminanceSource> getSource();
Ref<LuminanceSource> getLuminanceSource() const ;
virtual Ref<Binarizer> createBinarizer(Ref<LuminanceSource> source) = 0;
};
}

View file

@ -23,7 +23,7 @@
namespace zxing {
BinaryBitmap::BinaryBitmap(Ref<Binarizer> binarizer) : bits_(NULL), array_bits_(NULL), binarizer_(binarizer), cached_y_(-1) {
BinaryBitmap::BinaryBitmap(Ref<Binarizer> binarizer) : binarizer_(binarizer) {
}
@ -31,28 +31,23 @@ namespace zxing {
}
Ref<BitArray> BinaryBitmap::getBlackRow(int y, Ref<BitArray> row) {
if (array_bits_ == NULL && cached_y_ != y) {
array_bits_ = binarizer_->getBlackRow(y, row);
cached_y_ = y;
}
return array_bits_;
return binarizer_->getBlackRow(y, row);
}
Ref<BitMatrix> BinaryBitmap::getBlackMatrix() {
if (bits_ == NULL) {
bits_ = binarizer_->getBlackMatrix();
}
return bits_;
}
int BinaryBitmap::getWidth() {
return getSource()->getWidth();
}
int BinaryBitmap::getHeight() {
return getSource()->getHeight();
return binarizer_->getBlackMatrix();
}
Ref<LuminanceSource> BinaryBitmap::getSource() {
return binarizer_->getSource();
int BinaryBitmap::getWidth() const {
return getLuminanceSource()->getWidth();
}
int BinaryBitmap::getHeight() const {
return getLuminanceSource()->getHeight();
}
Ref<LuminanceSource> BinaryBitmap::getLuminanceSource() const {
return binarizer_->getLuminanceSource();
}
}

View file

@ -30,8 +30,6 @@ namespace zxing {
class BinaryBitmap : public Counted {
private:
Ref<BitMatrix> bits_;
Ref<BitArray> array_bits_;
Ref<Binarizer> binarizer_;
int cached_y_;
@ -41,10 +39,18 @@ namespace zxing {
Ref<BitArray> getBlackRow(int y, Ref<BitArray> row);
Ref<BitMatrix> getBlackMatrix();
Ref<LuminanceSource> getSource();
int getWidth();
int getHeight();
Ref<LuminanceSource> getLuminanceSource() const;
int getWidth() const;
int getHeight() const;
bool isRotateSupported() const;
Ref<BinaryBitmap> rotateCounterClockwise();
bool isCropSupported() const;
Ref<BinaryBitmap> crop(int left, int top, int width, int height);
};
}

View file

@ -19,6 +19,7 @@
*/
#include <zxing/LuminanceSource.h>
#include <zxing/common/IllegalArgumentException.h>
namespace zxing {
@ -28,16 +29,32 @@ LuminanceSource::LuminanceSource() {
LuminanceSource::~LuminanceSource() {
}
unsigned char* LuminanceSource::copyMatrix() {
unsigned char* LuminanceSource::getMatrix() {
int width = getWidth();
int height = getHeight();
unsigned char* matrix = new unsigned char[width * height];
unsigned char* row = new unsigned char[width];
for (int y = 0; y < height; y++) {
for (int x = 0; x < width; x++) {
matrix[y*width+x] = getPixel(x, y);
}
getRow(y, row);
memcpy(&matrix[y * width], row, width);
}
return matrix;
}
bool LuminanceSource::isCropSupported() const {
return false;
}
Ref<LuminanceSource> LuminanceSource::crop(int left, int top, int width, int height) {
throw IllegalArgumentException("This luminance source does not support cropping.");
}
bool LuminanceSource::isRotateSupported() const {
return false;
}
Ref<LuminanceSource> LuminanceSource::rotateCounterClockwise() {
throw IllegalArgumentException("This luminance source does not support rotation.");
}
}

View file

@ -22,6 +22,7 @@
#define LUMINANCESOURCE_H_
#include <zxing/common/Counted.h>
#include <string.h>
namespace zxing {
@ -33,8 +34,15 @@ public:
virtual int getWidth() const = 0;
virtual int getHeight() const = 0;
virtual unsigned char getPixel(int x, int y) const = 0;
virtual unsigned char* copyMatrix();
virtual unsigned char* getRow(int y, unsigned char* row) = 0;
virtual unsigned char* getMatrix();
virtual bool isCropSupported() const;
virtual Ref<LuminanceSource> crop(int left, int top, int width, int height);
virtual bool isRotateSupported() const;
virtual Ref<LuminanceSource> rotateCounterClockwise();
};
}

View file

@ -31,7 +31,7 @@ const int LUMINANCE_SHIFT = 8 - LUMINANCE_BITS;
const int LUMINANCE_BUCKETS = 1 << LUMINANCE_BITS;
GlobalHistogramBinarizer::GlobalHistogramBinarizer(Ref<LuminanceSource> source) :
Binarizer(source) {
Binarizer(source), cached_matrix_(NULL), cached_row_(NULL), cached_row_num_(-1) {
}
@ -39,10 +39,14 @@ GlobalHistogramBinarizer::~GlobalHistogramBinarizer() {
}
Ref<BitArray> GlobalHistogramBinarizer::estimateBlackRow(int y,
Ref<BitArray> row){
Ref<BitArray> GlobalHistogramBinarizer::getBlackRow(int y, Ref<BitArray> row) {
if (row == cached_row_num_) {
return cached_row_;
}
vector<int> histogram(LUMINANCE_BUCKETS, 0);
LuminanceSource& source = *getSource();
LuminanceSource& source = *getLuminanceSource();
int width = source.getWidth();
if (row == NULL || static_cast<int>(row->getSize()) < width) {
row = new BitArray(width);
@ -50,9 +54,11 @@ Ref<BitArray> GlobalHistogramBinarizer::estimateBlackRow(int y,
row->clear();
}
//TODO(flyashi): cache this instead of allocating and deleting per row
unsigned char* row_pixels = new unsigned char[width];
getLuminanceSource()->getRow(y,row_pixels);
for (int x = 0; x < width; x++) {
unsigned char pixel = source.getPixel(x, y);
histogram[pixel >> LUMINANCE_SHIFT]++;
histogram[row_pixels[x] >> LUMINANCE_SHIFT]++;
}
int blackPoint = estimate(histogram) << LUMINANCE_SHIFT;
@ -60,10 +66,10 @@ Ref<BitArray> GlobalHistogramBinarizer::estimateBlackRow(int y,
Ref<BitArray> array_ref(new BitArray(width));
BitArray& array = *array_ref;
int left = source.getPixel(0, y);
int center = source.getPixel(1, y);
int left = row_pixels[0];
int center = row_pixels[1];
for (int x = 1; x < width - 1; x++) {
int right = source.getPixel(x+1, y);
int right = row_pixels[x + 1];
// A simple -1 4 -1 box filter with a weight of 2.
int luminance = ((center << 2) - left - right) >> 1;
if (luminance < blackPoint) {
@ -73,12 +79,21 @@ Ref<BitArray> GlobalHistogramBinarizer::estimateBlackRow(int y,
center = right;
}
cached_row_ = array_ref;
cached_row_num_ = y;
delete [] row_pixels;
return array_ref;
}
Ref<BitMatrix> GlobalHistogramBinarizer::estimateBlackMatrix() {
Ref<BitMatrix> GlobalHistogramBinarizer::getBlackMatrix() {
if (cached_matrix_ != NULL) {
return cached_matrix_;
}
// Faster than working with the reference
LuminanceSource& source = *getSource();
LuminanceSource& source = *getLuminanceSource();
int width = source.getWidth();
int height = source.getHeight();
vector<int> histogram(LUMINANCE_BUCKETS, 0);
@ -87,14 +102,15 @@ Ref<BitMatrix> GlobalHistogramBinarizer::estimateBlackMatrix() {
// Quickly calculates the histogram by sampling four rows from the image.
// This proved to be more robust on the blackbox tests than sampling a
// diagonal as we used to do.
unsigned char* row = new unsigned char[width];
for (int y = 1; y < 5; y++) {
int row = height * y / 5;
int rownum = height * y / 5;
int right = (width << 2) / 5;
int sdf;
getLuminanceSource()->getRow(rownum,row);
for (int x = width / 5; x < right; x++) {
unsigned char pixel = source.getPixel(x, row);
histogram[pixel >> LUMINANCE_SHIFT]++;
sdf = histogram[pixel >> LUMINANCE_SHIFT];
histogram[row[x] >> LUMINANCE_SHIFT]++;
sdf = histogram[row[x] >> LUMINANCE_SHIFT];
}
}
@ -103,11 +119,16 @@ Ref<BitMatrix> GlobalHistogramBinarizer::estimateBlackMatrix() {
Ref<BitMatrix> matrix_ref(new BitMatrix(width, height));
BitMatrix& matrix = *matrix_ref;
for (int y = 0; y < height; y++) {
getLuminanceSource()->getRow(y,row);
for (int x = 0; x < width; x++) {
if (source.getPixel(x, y) <= blackPoint)
if (row[x] <= blackPoint)
matrix.set(x, y);
}
}
cached_matrix_ = matrix_ref;
delete [] row;
return matrix_ref;
}
@ -179,5 +200,9 @@ int GlobalHistogramBinarizer::estimate(vector<int> &histogram) {
return bestValley;
}
Ref<Binarizer> GlobalHistogramBinarizer::createBinarizer(Ref<LuminanceSource> source) {
return Ref<Binarizer> (new GlobalHistogramBinarizer(source));
}
} // namespace zxing

View file

@ -30,13 +30,19 @@
namespace zxing {
class GlobalHistogramBinarizer : public Binarizer {
private:
Ref<BitMatrix> cached_matrix_;
Ref<BitArray> cached_row_;
int cached_row_num_;
public:
GlobalHistogramBinarizer(Ref<LuminanceSource> source);
virtual ~GlobalHistogramBinarizer();
virtual Ref<BitArray> estimateBlackRow(int y, Ref<BitArray> row);
virtual Ref<BitMatrix> estimateBlackMatrix();
virtual Ref<BitArray> getBlackRow(int y, Ref<BitArray> row);
virtual Ref<BitMatrix> getBlackMatrix();
static int estimate(std::vector<int> &histogram);
Ref<Binarizer> createBinarizer(Ref<LuminanceSource> source);
};
}

View file

@ -1,196 +0,0 @@
/*
* LocalBlockBinarizer.cpp
* zxing
*
* Created by Ralf Kistner on 17/10/2009.
* Copyright 2008 ZXing authors All rights reserved.
*
* 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 <zxing/common/LocalBlockBinarizer.h>
namespace zxing {
const int GLOBAL = 0;
const int THRESHOLD = 1;
LocalBlockBinarizer::LocalBlockBinarizer(Ref<LuminanceSource> source) :
Binarizer(source) {
}
LocalBlockBinarizer::~LocalBlockBinarizer() {
}
Ref<BitArray> LocalBlockBinarizer::estimateBlackRow(int y, Ref<BitArray> row) {
//TODO: implement
return Ref<BitArray>();
}
// Calculates the final BitMatrix once for all requests. This could be called once from the
// constructor instead, but there are some advantages to doing it lazily, such as making
// profiling easier, and not doing heavy lifting when callers don't expect it.
Ref<BitMatrix> LocalBlockBinarizer::estimateBlackMatrix() {
Ref<LuminanceSource> source = getSource();
unsigned char* luminances = source->copyMatrix();
int width = source->getWidth();
int height = source->getHeight();
// Sharpening does not really help for 2d barcodes
// sharpenRow(luminances, width, height);
int subWidth = width >> 3;
int subHeight = height >> 3;
unsigned char* averages = new unsigned char[subWidth * subHeight];
unsigned char* types = new unsigned char[subWidth * subHeight];
calculateBlackPoints(luminances, averages, types, subWidth, subHeight, width);
Ref<BitMatrix> matrix(new BitMatrix(width, height));
calculateThresholdForBlock(luminances, subWidth, subHeight, width, averages, types, *matrix);
delete[] averages;
delete[] types;
delete[] luminances;
return matrix;
}
// For each 8x8 block in the image, calculate the average black point using a 5x5 grid
// of the blocks around it. Also handles the corner cases, but will ignore up to 7 pixels
// on the right edge and 7 pixels at the bottom of the image if the overall dimensions are not
// multiples of eight. In practice, leaving those pixels white does not seem to be a problem.
void LocalBlockBinarizer::calculateThresholdForBlock(const unsigned char* luminances, int subWidth, int subHeight,
int stride, const unsigned char* averages, const unsigned char* types, BitMatrix& matrix) {
// Calculate global average
int global = 0;
for (int y = 0; y < subHeight; y++) {
for (int x = 0; x < subWidth; x++) {
global += averages[y * subWidth + x];
}
}
global /= subWidth * subHeight;
for (int y = 0; y < subHeight; y++) {
for (int x = 0; x < subWidth; x++) {
int left = (x > 0) ? x : 1;
left = (left < subWidth - 1) ? left : subWidth - 2;
int top = (y > 0) ? y : 1;
top = (top < subHeight - 1) ? top : subHeight - 2;
int sum = 0;
int contrast = 0;
for (int z = -1; z <= 1; z++) {
// sum += averages[(top + z) * subWidth + left - 2];
sum += averages[(top + z) * subWidth + left - 1];
sum += averages[(top + z) * subWidth + left];
sum += averages[(top + z) * subWidth + left + 1];
// sum += averages[(top + z) * subWidth + left + 2];
// type += types[(top + z) * subWidth + left - 2];
contrast += types[(top + z) * subWidth + left - 1];
contrast += types[(top + z) * subWidth + left];
contrast += types[(top + z) * subWidth + left + 1];
// type += types[(top + z) * subWidth + left + 2];
}
int average = sum / 9;
if (contrast > 2)
threshold8x8Block(luminances, x << 3, y << 3, average, stride, matrix);
// else if(average < global) // Black
// matrix.setRegion(x << 3, y << 3, 8, 8);
// If white, we don't need to do anything - the block is already cleared.
}
}
}
// Applies a single threshold to an 8x8 block of pixels.
void LocalBlockBinarizer::threshold8x8Block(const unsigned char* luminances, int xoffset, int yoffset, int threshold,
int stride, BitMatrix& matrix) {
for (int y = 0; y < 8; y++) {
int offset = (yoffset + y) * stride + xoffset;
for (int x = 0; x < 8; x++) {
int pixel = luminances[offset + x];
if (pixel < threshold) {
matrix.set(xoffset + x, yoffset + y);
}
}
}
}
// Calculates a single black point for each 8x8 block of pixels and saves it away.
void LocalBlockBinarizer::calculateBlackPoints(const unsigned char* luminances, unsigned char* averages,
unsigned char* types, int subWidth, int subHeight, int stride) {
for (int y = 0; y < subHeight; y++) {
for (int x = 0; x < subWidth; x++) {
int sum = 0;
int min = 255;
int max = 0;
for (int yy = 0; yy < 8; yy++) {
int offset = ((y << 3) + yy) * stride + (x << 3);
const unsigned char* lumo = luminances + offset;
for (int xx = 0; xx < 8; xx++) {
int pixel = lumo[xx];
sum += pixel;
if (pixel < min) {
min = pixel;
}
if (pixel > max) {
max = pixel;
}
}
}
// If the contrast is inadequate, we treat the block as white.
// An arbitrary value is chosen here. Higher values mean less noise, but may also reduce
// the ability to recognise some barcodes.
int average = sum >> 6;
int type;
if (max - min > 30)
type = THRESHOLD;
else
type = GLOBAL;
// int average = (max - min > 24) ? (sum >> 6) : (min-1);
averages[y * subWidth + x] = average;
types[y * subWidth + x] = type;
}
}
}
// Applies a simple -1 4 -1 box filter with a weight of 2 to each row.
void LocalBlockBinarizer::sharpenRow(unsigned char* luminances, int width, int height) {
for (int y = 0; y < height; y++) {
int offset = y * width;
int left = luminances[offset];
int center = luminances[offset + 1];
for (int x = 1; x < width - 1; x++) {
unsigned char right = luminances[offset + x + 1];
int pixel = ((center << 2) - left - right) >> 1;
// Must clamp values to 0..255 so they will fit in a byte.
if (pixel > 255) {
pixel = 255;
} else if (pixel < 0) {
pixel = 0;
}
luminances[offset + x] = (unsigned char)pixel;
left = center;
center = right;
}
}
}
}

View file

@ -1,47 +0,0 @@
/*
* LocalBlockBinarizer.h
* zxing
*
* Created by Ralf Kistner on 17/10/2009.
* Copyright 2008 ZXing authors All rights reserved.
*
* 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.
*/
#ifndef LOCALBLOCKBINARIZER_H_
#define LOCALBLOCKBINARIZER_H_
#include <zxing/Binarizer.h>
#include <zxing/common/BitMatrix.h>
namespace zxing {
class LocalBlockBinarizer : public Binarizer {
public:
LocalBlockBinarizer(Ref<LuminanceSource> source);
virtual ~LocalBlockBinarizer();
virtual Ref<BitMatrix> estimateBlackMatrix();
Ref<BitArray> estimateBlackRow(int y, Ref<BitArray> row);
private:
void calculateThresholdForBlock(const unsigned char* luminances, int subWidth, int subHeight,
int stride, const unsigned char* averages, const unsigned char* types, BitMatrix& matrix);
void sharpenRow(unsigned char* luminances, int width, int height);
void calculateBlackPoints(const unsigned char* luminances, unsigned char* averages, unsigned char* types, int subWidth, int subHeight, int stride);
void threshold8x8Block(const unsigned char* luminances, int xoffset, int yoffset, int threshold,
int stride, BitMatrix& matrix);
};
}
#endif /* LOCALBLOCKBINARIZER_H_ */

View file

@ -24,6 +24,8 @@
using namespace Magick;
namespace zxing {
MagickBitmapSource::MagickBitmapSource(Image& image) : image_(image) {
width = image.columns();
height = image.rows();
@ -35,25 +37,30 @@ MagickBitmapSource::~MagickBitmapSource() {
}
int MagickBitmapSource::getWidth() {
int MagickBitmapSource::getWidth() const {
return width;
}
int MagickBitmapSource::getHeight() {
int MagickBitmapSource::getHeight() const {
return height;
}
unsigned char MagickBitmapSource::getPixel(int x, int y) {
unsigned char* MagickBitmapSource::getRow(int y, unsigned char* row) {
int width = getWidth();
if (row == NULL) {
row = new unsigned char[width];
}
for (int x = 0; x < width; x++) {
const PixelPacket* p = pixel_cache + y * width + x;
// We assume 16 bit values here
row[x] = (unsigned char)((306 * ((int)p->red >> 8) + 601 * ((int)p->green >> 8) + 117 * ((int)p->blue >> 8)) >> 10);
}
return row;
//return (unsigned char)((((int)p->red + (int)p->green + (int)p->blue) >> 8) / 3);
return (unsigned char)((306 * ((int)p->red >> 8) + 601 * ((int)p->green >> 8) + 117 * ((int)p->blue >> 8)) >> 10);
}
/** This is a more efficient implementation. */
unsigned char* MagickBitmapSource::copyMatrix() {
unsigned char* MagickBitmapSource::getMatrix() {
int width = getWidth();
int height = getHeight();
unsigned char* matrix = new unsigned char[width*height];
@ -68,4 +75,5 @@ unsigned char* MagickBitmapSource::copyMatrix() {
}
return matrix;
}
}

View file

@ -24,7 +24,9 @@
#include <Magick++.h>
#include <zxing/LuminanceSource.h>
class MagickBitmapSource : public zxing::LuminanceSource {
namespace zxing {
class MagickBitmapSource : public LuminanceSource {
private:
Magick::Image& image_;
int width;
@ -36,10 +38,12 @@ public:
~MagickBitmapSource();
int getWidth();
int getHeight();
unsigned char getPixel(int x, int y);
unsigned char* copyMatrix();
int getWidth() const;
int getHeight() const;
unsigned char* getRow(int y, unsigned char* row);
unsigned char* getMatrix();
};
}
#endif /* MAGICKMONOCHROMEBITMAPSOURCE_H_ */

View file

@ -75,7 +75,7 @@ int main(int argc, char** argv) {
bool local = true; // Use local thresholding
test_image(image, local);
decode_image(image, local);
}
return 0;
}

View file

@ -4,6 +4,7 @@
*
* Created by Ralf Kistner on 16/10/2009.
* Copyright 2008 ZXing authors All rights reserved.
* Modified by Yakov Okshtein (flyashi@gmail.com) to add 1D barcode support.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,10 +26,12 @@
#include "MagickBitmapSource.h"
#include <zxing/common/Counted.h>
#include <zxing/qrcode/QRCodeReader.h>
#include <zxing/Binarizer.h>
#include <zxing/oned/MultiFormatUPCEANReader.h>
#include <zxing/Result.h>
#include <zxing/ReaderException.h>
#include <zxing/common/GlobalHistogramBinarizer.h>
#include <zxing/common/LocalBlockBinarizer.h>
//#include <zxing/common/LocalBlockBinarizer.h>
#include <exception>
#include <zxing/Exception.h>
#include <zxing/common/IllegalArgumentException.h>
@ -106,28 +109,14 @@ void save_grid(Ref<BitMatrix> matrix, string filename, Ref<PerspectiveTransform>
image.write(filename);
}
Ref<Result> decode(string out_prefix, Ref<BinaryBitmap> image, string& cell_grid, string& cell_transformed) {
Ref<Result> decode2D(string out_prefix, Ref<BinaryBitmap> image, string& cell_grid, string& cell_transformed) {
Decoder decoder;
QREdgeDetector detector = QREdgeDetector(image->getBlackMatrix());
Ref<DetectorResult> detectorResult(detector.detect());
if (out_prefix.size()) {
// Grid image
string gridfile = out_prefix + ".grid.gif";
Ref<PerspectiveTransform> transform = detectorResult->getTransform();
int dimension = detectorResult->getBits()->getDimension();
save_grid(image->getBlackMatrix(), gridfile, transform, dimension);
cell_grid = "<img src=\"" + gridfile + "\" />";
// Transformed image
string tfile = out_prefix + ".transformed.png";
save_matrix(detectorResult->getBits(), tfile, 5);
cell_transformed = "<img src=\"" + tfile + "\" />";
}
vector<Ref<ResultPoint> > points(detectorResult->getPoints());
Ref<DecoderResult> decoderResult(decoder.decode(detectorResult->getBits()));
@ -136,11 +125,26 @@ Ref<Result> decode(string out_prefix, Ref<BinaryBitmap> image, string& cell_grid
decoderResult->getRawBytes(),
points,
BarcodeFormat_QR_CODE));
return result;
}
Ref<Result> decode1D(string out_prefix, Ref<BinaryBitmap> image, string& cell_grid, string& cell_transformed) {
Ref<Reader> reader(new oned::MultiFormatUPCEANReader);
Ref<Result> result(new Result(*reader->decode(image)));
return result;
}
//TODO(flyashi): Call MultiFormatReader directly
Ref<Result> decode(string out_prefix, Ref<BinaryBitmap> image, string& cell_grid, string& cell_transformed) {
try {
return decode1D(out_prefix,image,cell_grid,cell_transformed);
} catch (ReaderException re) {
return decode2D(out_prefix,image,cell_grid,cell_transformed);
}
}
int test_image(Image& image, string out_prefix, bool localized) {
@ -159,7 +163,7 @@ int test_image(Image& image, string out_prefix, bool localized) {
Ref<MagickBitmapSource> source(new MagickBitmapSource(image));
if (localized) {
binarizer = new LocalBlockBinarizer(source);
//binarizer = new LocalBlockBinarizer(source);
} else {
binarizer = new GlobalHistogramBinarizer(source);
}
@ -167,7 +171,7 @@ int test_image(Image& image, string out_prefix, bool localized) {
if (out_prefix.size()) {
string monofile = out_prefix + ".mono.png";
matrix = binarizer->getBlackMatrix();
save_matrix(matrix, monofile);
//save_matrix(matrix, monofile);
cell_mono = "<img src=\"" + monofile + "\" />";
}
@ -190,10 +194,8 @@ int test_image(Image& image, string out_prefix, bool localized) {
res = -5;
}
cout << "<td>" << cell_mono << "</td>" << endl;
cout << "<td>" << cell_grid << "</td>" << endl;
cout << "<td>" << cell_transformed << "</td>" << endl;
cout << "<td bgcolor=\"" << result_color << "\">" << cell_result << "</td>" << endl;
cout << cell_result;
return res;
}
@ -213,16 +215,15 @@ int main(int argc, char** argv) {
}
string outfolder = argv[1];
int total = argc - 2;
// int total = argc - 2;
int gonly = 0;
int lonly = 0;
int both = 0;
int neither = 0;
cout << "<html><body><table border=\"1\">" << endl;
for (int i = 2; i < argc; i++) {
string infilename = argv[i];
cerr << "Processing: " << infilename << endl;
// cerr << "Processing: " << infilename << endl;
Image image;
try {
image.read(infilename);
@ -230,10 +231,6 @@ int main(int argc, char** argv) {
cerr << "Unable to open image, ignoring" << endl;
continue;
}
cout << "<tr><td colspan=\"5\">" << infilename << "</td></tr>" << endl;
cout << "<tr>" << endl;
cout << "<td><img src=\"" << infilename << "\" /></td>" << endl;
int gresult = 1;
@ -241,36 +238,23 @@ int main(int argc, char** argv) {
if (outfolder == string("-")) {
gresult = test_image_global(image, "");
lresult = test_image_local(image, "");
// lresult = test_image_local(image, "");
} else {
replace(infilename.begin(), infilename.end(), '/', '_');
string prefix = string(outfolder) + string("/") + infilename;
gresult = test_image_global(image, prefix + ".g");
lresult = test_image_local(image, prefix + ".l");
// lresult = test_image_local(image, prefix + ".l");
}
gresult = gresult == 0;
lresult = lresult == 0;
// lresult = lresult == 0;
gonly += gresult && !lresult;
lonly += lresult && !gresult;
both += gresult && lresult;
neither += !gresult && !lresult;
cout << "</tr>" << endl;
}
cout << "</table>" << endl;
cout << "<table>" << endl;
cout << "<tr><td>Total</td><td>" << total << "</td></tr>" << endl;
cout << "<tr><td>Both correct</td><td>" << both << "</td></tr>" << endl;
cout << "<tr><td>Neither correct</td><td>" << neither << "</td></tr>" << endl;
cout << "<tr><td>Global only</td><td>" << gonly << "</td></tr>" << endl;
cout << "<tr><td>Local only</td><td>" << lonly << "</td></tr>" << endl;
cout << "</table>" << endl;
cout << "</body></html>" << endl;
return 0;
}