我正在处理 Java 应用程序的一部分，该应用程序将图像作为字节数组处理，将其读入 java.awt.image.BufferedImage 实例中并将其传递给第三方库进行处理。
为了进行单元测试，我想要获取一张图片（从磁盘上的文件），并断言它与代码处理后的相同的图片是相等的。
BufferedImage 从磁盘上的 PNG 文件使用 ImageIO.read(URL) 读取。BufferedImage 中，并将其作为 PNG 写入字节数组，以提供给待测试系统。当待测试系统将字节数组写入新的 BufferedImage 时，我希望断言这两个图像在有意义的方式上是相等的。使用 equals() 方法（从 Object 继承而来）不能很好地工作。比较 BufferedImage.toString() 值也不行，因为输出的字符串包括对象引用信息。
有人知道任何快捷方法吗？我不想在一个大型应用程序的小部分中为单个单元测试带来第三方库。
这是最佳的方法。不需要保留一个变量来告诉图像是否仍然相等。当条件为假时，立即返回false即可。短路评估有助于在比较失败后节省时间循环遍历像素，就像在trumpetlick的答案中所述。
/**
* Compares two images pixel by pixel.
*
* @param imgA the first image.
* @param imgB the second image.
* @return whether the images are both the same or not.
*/
public static boolean compareImages(BufferedImage imgA, BufferedImage imgB) {
// The images must be the same size.
if (imgA.getWidth() != imgB.getWidth() || imgA.getHeight() != imgB.getHeight()) {
return false;
}
int width = imgA.getWidth();
int height = imgA.getHeight();
// Loop over every pixel.
for (int y = 0; y < height; y++) {
for (int x = 0; x < width; x++) {
// Compare the pixels for equality.
if (imgA.getRGB(x, y) != imgB.getRGB(x, y)) {
return false;
}
}
}
return true;
}
BufferedImages的位深度、排列等相同（这似乎在这里必须成立），则可以这样做：DataBuffer dbActual = myBufferedImage.getRaster().getDataBuffer();
DataBuffer dbExpected = bufferImageReadFromAFile.getRaster().getDataBuffer();
DataBufferInt。DataBufferInt actualDBAsDBInt = (DataBufferInt) dbActual ;
DataBufferInt expectedDBAsDBInt = (DataBufferInt) dbExpected ;
首先对DataBuffers的大小和bank进行一些"合理性检查"，然后进入循环。
for (int bank = 0; bank < actualDBAsDBInt.getNumBanks(); bank++) {
int[] actual = actualDBAsDBInt.getData(bank);
int[] expected = expectedDBAsDBInt.getData(bank);
// this line may vary depending on your test framework
assertTrue(Arrays.equals(actual, expected));
}
这是尽可能快的方式，因为您一次获取一块数据而不是逐个获取。
int width;
int height;
boolean imagesEqual = true;
if( image1.getWidth() == ( width = image2.getWidth() ) &&
image1.getHeight() == ( height = image2.getHeight() ) ){
for(int x = 0;imagesEqual == true && x < width; x++){
for(int y = 0;imagesEqual == true && y < height; y++){
if( image1.getRGB(x, y) != image2.getRGB(x, y) ){
imagesEqual = false;
}
}
}
}else{
imagesEqual = false;
}
if块内将布尔值设置为true。 - Thilobreak 语句不会有太大作用，因为你有嵌套的 for 循环。 - JeffreyimagesEqual == true，只需要 imagesEqual 就可以了。那个新宝宝肯定会让你很辛苦的。 - Jeffrey我在Groovy中更改了按像素相等的函数，可能会有帮助：
boolean imagesAreEqual(BufferedImage image1, BufferedImage image2) {
if (image1.width != image2.width || image1.height != image2.height) {
return false
}
for (int x = 1; x < image2.width; x++) {
for (int y = 1; y < image2.height; y++) {
if (image1.getRGB(x, y) != image2.getRGB(x, y)) {
return false
}
}
}
return true
}
import org.mockito.ArgumentMatcher;
public class BufferedImageMatcher extends ArgumentMatcher<BufferedImage> {
private final BufferedImage expected;
public BufferedImageMatcher(BufferedImage expected) {
this.expected = expected;
}
@Override
public boolean matches(Object argument) {
BufferedImage actual = (BufferedImage) argument;
assertEquals(expected.getWidth(), actual.getWidth());
assertEquals(expected.getHeight(), actual.getHeight());
for (int x = 0; x < actual.getWidth(); x++) {
for (int y = 0; y < actual.getHeight(); y++) {
assertEquals(expected.getRGB(x, y), actual.getRGB(x, y));
}
}
return true;
}
}
assertThat(actual, new BufferedImageMatcher(expected));
功能正常但效率不高
public static boolean compareImage(File fileA, File fileB) {
try {
// take buffer data from botm image files //
BufferedImage biA = ImageIO.read(fileA);
DataBuffer dbA = biA.getData().getDataBuffer();
int sizeA = dbA.getSize();
BufferedImage biB = ImageIO.read(fileB);
DataBuffer dbB = biB.getData().getDataBuffer();
int sizeB = dbB.getSize();
// compare data-buffer objects //
if(sizeA == sizeB) {
for(int i=0; i<sizeA; i++) {
if(dbA.getElem(i) != dbB.getElem(i)) {
return false;
}
}
return true;
}
else {
return false;
}
}
catch (Exception e) {
e.printStackTrace();
return false;
}
}
我除了使用暴力循环之外，想不到其他的方法：
BufferedImage bi1, bi2, ...
...
Raster r1 = bi1.getData();
DataBuffer db1 = r1.getDataBuffer();
if (db1.getSize() != db2.getSize ())
...
for (int i = 0; i < db1.getSize(); i++) {
int px = db1.getElem(i);
}
getRGB()更快，但只有测试才能确定。 - Andrew Thompson你可以通过OutputStream将图像写入byte[]，并使用imageio进行操作。
在我的代码中，大致如下：
byte[] encodeJpegLossless(BufferedImage img){...using imageio...}
...
Assert.assertTrue(Arrays.equals(encodeJpegLossless(img1)
,encodeJpegLossless(img2)
)
);
.equals()不能起作用吗？ - Alexis KingBufferedImage没有重写Object#equals()方法。http://www.docjar.com/html/api/java/awt/image/BufferedImage.java.html - Matt Ball