package com.google.zxing.common;
+import com.google.zxing.BarcodeFormat;
+import com.google.zxing.DecodeHintType;
import com.google.zxing.MonochromeBitmapSource;
import com.google.zxing.Reader;
import com.google.zxing.ReaderException;
import junit.framework.TestCase;
import javax.imageio.ImageIO;
+import java.awt.geom.AffineTransform;
+import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URL;
+import java.util.Hashtable;
+import java.util.Vector;
/**
* @author srowen@google.com (Sean Owen)
+ * @author dswitkin@google.com (Daniel Switkin)
*/
public abstract class AbstractBlackBoxTestCase extends TestCase {
+ private static final Hashtable TRY_HARDER_HINT;
+ static {
+ TRY_HARDER_HINT = new Hashtable();
+ TRY_HARDER_HINT.put(DecodeHintType.TRY_HARDER, Boolean.TRUE);
+ }
+
private static final FilenameFilter IMAGE_NAME_FILTER = new FilenameFilter() {
public boolean accept(File dir, String name) {
String lowerCase = name.toLowerCase();
- return
- lowerCase.endsWith(".jpg") || lowerCase.endsWith(".jpeg") ||
- lowerCase.endsWith(".gif") || lowerCase.endsWith(".png") ||
- lowerCase.endsWith(".url");
+ return lowerCase.endsWith(".jpg") || lowerCase.endsWith(".jpeg") ||
+ lowerCase.endsWith(".gif") || lowerCase.endsWith(".png") ||
+ lowerCase.endsWith(".url");
}
};
+ private class TestResult {
+ public int mustPassCount;
+ public float rotation;
+
+ TestResult(int mustPassCount, float rotation) {
+ this.mustPassCount = mustPassCount;
+ this.rotation = rotation;
+ }
+ }
+
private final File testBase;
private final Reader barcodeReader;
- private final double passPercent;
+ private final BarcodeFormat expectedFormat;
+ private Vector<TestResult> testResults;
- protected AbstractBlackBoxTestCase(File testBase, Reader barcodeReader, double passPercent) {
+ protected AbstractBlackBoxTestCase(File testBase,
+ Reader barcodeReader,
+ BarcodeFormat expectedFormat) {
this.testBase = testBase;
this.barcodeReader = barcodeReader;
- this.passPercent = passPercent;
+ this.expectedFormat = expectedFormat;
+ testResults = new Vector<TestResult>();
}
- public void testBlackBox() throws IOException {
+ /**
+ * Adds a new test for the current directory of images.
+ *
+ * @param mustPassCount The number of images which must decode for the test to pass.
+ * @param rotation The rotation in degrees clockwise to use for this test.
+ */
+ protected void addTest(int mustPassCount, float rotation) {
+ testResults.add(new TestResult(mustPassCount, rotation));
+ }
+ public void testBlackBox() throws IOException {
+ assertTrue(testResults.size() > 0);
assertTrue("Please run from the 'core' directory", testBase.exists());
File[] imageFiles = testBase.listFiles(IMAGE_NAME_FILTER);
- int passedCount = 0;
+ int[] passedCounts = new int[testResults.size()];
for (File testImage : imageFiles) {
System.out.println("Starting " + testImage.getAbsolutePath());
- BufferedImage image;
- if (testImage.getName().endsWith(".url")) {
- String urlString = readFileAsString(testImage);
- image = ImageIO.read(new URL(urlString));
- } else {
- image = ImageIO.read(testImage);
- }
- MonochromeBitmapSource source = new BufferedImageMonochromeBitmapSource(image);
- Result result;
- try {
- result = barcodeReader.decode(source);
- } catch (ReaderException re) {
- System.out.println(re);
- continue;
+ BufferedImage image;
+ if (testImage.getName().endsWith(".url")) {
+ String urlString = readFileAsString(testImage);
+ image = ImageIO.read(new URL(urlString));
+ } else {
+ image = ImageIO.read(testImage);
}
String testImageFileName = testImage.getName();
- File expectedTextFile =
- new File(testBase, testImageFileName.substring(0, testImageFileName.indexOf('.')) + ".txt");
+ File expectedTextFile = new File(testBase,
+ testImageFileName.substring(0, testImageFileName.indexOf('.')) + ".txt");
String expectedText = readFileAsString(expectedTextFile);
- String resultText = result.getText();
- if (expectedText.equals(resultText)) {
- passedCount++;
- } else {
- System.out.println("Mismatch: expected '" + expectedText + "' but got '" + resultText + '\'');
+ for (int x = 0; x < testResults.size(); x++) {
+ if (testOneImage(image, testResults.get(x).rotation, expectedText)) {
+ passedCounts[x]++;
+ }
}
+ }
+
+ for (int x = 0; x < testResults.size(); x++) {
+ System.out.println("Rotation " + testResults.get(x).rotation + " degrees: " + passedCounts[x] +
+ " of " + imageFiles.length + " images passed (" + testResults.get(x).mustPassCount +
+ " required)");
+ assertTrue("Rotation " + testResults.get(x).rotation + " degrees: Too many images failed",
+ passedCounts[x] >= testResults.get(x).mustPassCount);
+ }
+ }
+
+ private boolean testOneImage(BufferedImage image, float rotationInDegrees, String expectedText) {
+ BufferedImage rotatedImage = rotateImage(image, rotationInDegrees);
+ MonochromeBitmapSource source = new BufferedImageMonochromeBitmapSource(rotatedImage);
+ Result result;
+ try {
+ result = barcodeReader.decode(source);
+ } catch (ReaderException re) {
+ System.out.println(re);
+ return false;
+ }
+
+ if (expectedFormat != result.getBarcodeFormat()) {
+ System.out.println("Format mismatch: expected '" + expectedFormat + "' but got '" +
+ result.getBarcodeFormat() + "' (rotation: " + rotationInDegrees + ")");
+ return false;
+ }
+ String resultText = result.getText();
+ if (!expectedText.equals(resultText)) {
+ System.out.println("Mismatch: expected '" + expectedText + "' but got '" + resultText +
+ "' (rotation: " + rotationInDegrees + ")");
+ return false;
}
- System.out.println(passedCount + " of " + imageFiles.length + " images passed");
- assertTrue("Too many images failed", passedCount >= (int) (imageFiles.length * passPercent));
+ // Try "try harder" mode
+ try {
+ result = barcodeReader.decode(source, TRY_HARDER_HINT);
+ } catch (ReaderException re) {
+ fail("Normal mode succeeded but \"try harder\" failed");
+ return false;
+ }
+ if (expectedFormat != result.getBarcodeFormat()) {
+ System.out.println("Try Harder Format mismatch: expected '" + expectedFormat + "' but got '" +
+ result.getBarcodeFormat() + "' (rotation: " + rotationInDegrees + ")");
+ } else if (!expectedText.equals(resultText)) {
+ System.out.println("Try Harder Mismatch: expected '" + expectedText + "' but got '" +
+ resultText + "' (rotation: " + rotationInDegrees + ")");
+ }
+ return true;
}
private static String readFileAsString(File file) throws IOException {
return result.toString();
}
+ private static BufferedImage rotateImage(BufferedImage original, float degrees) {
+ if (degrees != 0.0f) {
+ AffineTransform at = new AffineTransform();
+ at.rotate(Math.toRadians(degrees), original.getWidth() / 2, original.getHeight() / 2);
+ AffineTransformOp op = new AffineTransformOp(at, AffineTransformOp.TYPE_BICUBIC);
+ return op.filter(original, null);
+ } else {
+ return original;
+ }
+ }
+
}
\ No newline at end of file