public static void checkEquals(
BufferedImage imgA, MultiSpectral imgB, boolean boofcvBandOrder, float tol) {
if (imgA.getRaster() instanceof ByteInterleavedRaster
&& imgA.getType() != BufferedImage.TYPE_BYTE_INDEXED) {
ByteInterleavedRaster raster = (ByteInterleavedRaster) imgA.getRaster();
if (raster.getNumBands() == 1) {
GImageSingleBand band = FactoryGImageSingleBand.wrap(imgB.getBand(0));
int strideA = raster.getScanlineStride();
int offsetA = raster.getDataOffset(0) - raster.getPixelStride() + 1;
// handle a special case where the RGB conversion is screwed
for (int i = 0; i < imgA.getHeight(); i++) {
for (int j = 0; j < imgA.getWidth(); j++) {
double valB = band.get(j, i).doubleValue();
int valA = raster.getDataStorage()[offsetA + i * strideA + j];
valA &= 0xFF;
if (Math.abs(valA - valB) > tol)
throw new RuntimeException("Images are not equal: A = " + valA + " B = " + valB);
}
}
return;
}
}
int bandOrder[];
if (boofcvBandOrder) {
if (imgB.getNumBands() == 4) {
bandOrder = new int[] {1, 2, 3, 0};
} else {
bandOrder = new int[] {0, 1, 2};
}
} else {
if (imgA.getType() == BufferedImage.TYPE_INT_RGB) {
bandOrder = new int[] {0, 1, 2};
} else if (imgA.getType() == BufferedImage.TYPE_INT_BGR
|| imgA.getType() == BufferedImage.TYPE_3BYTE_BGR) {
bandOrder = new int[] {2, 1, 0};
} else if (imgA.getType() == BufferedImage.TYPE_4BYTE_ABGR) {
bandOrder = new int[] {0, 3, 2, 1};
} else if (imgA.getType() == BufferedImage.TYPE_INT_ARGB) {
bandOrder = new int[] {0, 1, 2, 3};
} else {
bandOrder = new int[] {0, 1, 2};
}
}
int expected[] = new int[4];
for (int y = 0; y < imgA.getHeight(); y++) {
for (int x = 0; x < imgA.getWidth(); x++) {
// getRGB() automatically converts the band order to ARGB
int rgb = imgA.getRGB(x, y);
expected[0] = ((rgb >>> 24) & 0xFF); // alpha
expected[1] = ((rgb >>> 16) & 0xFF); // red
expected[2] = ((rgb >>> 8) & 0xFF); // green
expected[3] = (rgb & 0xFF); // blue
if (imgB.getNumBands() == 4) {
for (int i = 0; i < 4; i++) {
double found = GeneralizedImageOps.get(imgB.getBand(bandOrder[i]), x, y);
if (Math.abs(Math.exp(expected[i] - found)) > tol) {
for (int j = 0; j < 4; j++) {
System.out.println(
expected[j] + " " + GeneralizedImageOps.get(imgB.getBand(bandOrder[j]), x, y));
}
throw new RuntimeException(
"Images are not equal: band - " + i + " type " + imgA.getType());
}
}
} else if (imgB.getNumBands() == 3) {
for (int i = 0; i < 3; i++) {
double found = GeneralizedImageOps.get(imgB.getBand(bandOrder[i]), x, y);
if (Math.abs(expected[i + 1] - found) > tol) {
for (int j = 0; j < 3; j++) {
System.out.println(
expected[j + 1]
+ " "
+ GeneralizedImageOps.get(imgB.getBand(bandOrder[j]), x, y));
}
throw new RuntimeException(
"Images are not equal: band - " + i + " type " + imgA.getType());
}
}
} else {
throw new RuntimeException("Unexpectd number of bands");
}
}
}
}
