2019-06-07 18:56:55 +08:00
|
|
|
package org.opencv.test.objdetect;
|
|
|
|
|
2020-01-27 03:18:42 +08:00
|
|
|
import java.util.List;
|
2019-06-07 18:56:55 +08:00
|
|
|
import org.opencv.core.Mat;
|
|
|
|
import org.opencv.objdetect.QRCodeDetector;
|
|
|
|
import org.opencv.imgcodecs.Imgcodecs;
|
|
|
|
import org.opencv.test.OpenCVTestCase;
|
2021-05-16 19:52:36 +08:00
|
|
|
import java.util.Arrays;
|
2020-01-27 03:18:42 +08:00
|
|
|
import java.util.ArrayList;
|
2021-05-16 19:52:36 +08:00
|
|
|
import java.util.HashSet;
|
|
|
|
import java.util.List;
|
2019-06-07 18:56:55 +08:00
|
|
|
|
|
|
|
public class QRCodeDetectorTest extends OpenCVTestCase {
|
|
|
|
|
|
|
|
private final static String ENV_OPENCV_TEST_DATA_PATH = "OPENCV_TEST_DATA_PATH";
|
|
|
|
private String testDataPath;
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void setUp() throws Exception {
|
|
|
|
super.setUp();
|
|
|
|
|
2024-01-31 20:09:57 +08:00
|
|
|
// relys on https://developer.android.com/reference/java/lang/System
|
|
|
|
isTestCaseEnabled = System.getProperties().getProperty("java.vm.name") != "Dalvik";
|
|
|
|
|
|
|
|
if (isTestCaseEnabled) {
|
|
|
|
testDataPath = System.getenv(ENV_OPENCV_TEST_DATA_PATH);
|
|
|
|
if (testDataPath == null)
|
|
|
|
throw new Exception(ENV_OPENCV_TEST_DATA_PATH + " has to be defined!");
|
|
|
|
}
|
2019-06-07 18:56:55 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
public void testDetectAndDecode() {
|
|
|
|
Mat img = Imgcodecs.imread(testDataPath + "/cv/qrcode/link_ocv.jpg");
|
2020-01-27 03:18:42 +08:00
|
|
|
assertFalse(img.empty());
|
2019-06-07 18:56:55 +08:00
|
|
|
QRCodeDetector detector = new QRCodeDetector();
|
2020-01-27 03:18:42 +08:00
|
|
|
assertNotNull(detector);
|
2019-06-07 18:56:55 +08:00
|
|
|
String output = detector.detectAndDecode(img);
|
|
|
|
assertEquals(output, "https://opencv.org/");
|
|
|
|
}
|
|
|
|
|
2020-01-27 03:18:42 +08:00
|
|
|
public void testDetectAndDecodeMulti() {
|
|
|
|
Mat img = Imgcodecs.imread(testDataPath + "/cv/qrcode/multiple/6_qrcodes.png");
|
|
|
|
assertFalse(img.empty());
|
|
|
|
QRCodeDetector detector = new QRCodeDetector();
|
|
|
|
assertNotNull(detector);
|
|
|
|
List < String > output = new ArrayList< String >();
|
|
|
|
boolean result = detector.detectAndDecodeMulti(img, output);
|
|
|
|
assertTrue(result);
|
|
|
|
assertEquals(output.size(), 6);
|
2021-05-16 19:52:36 +08:00
|
|
|
List < String > expectedResults = Arrays.asList("SKIP", "EXTRA", "TWO STEPS FORWARD", "STEP BACK", "QUESTION", "STEP FORWARD");
|
|
|
|
assertEquals(new HashSet<String>(output), new HashSet<String>(expectedResults));
|
2020-01-27 03:18:42 +08:00
|
|
|
}
|
2019-06-07 18:56:55 +08:00
|
|
|
}
|