system: Linux mars.sprixweb.com 3.10.0-1160.119.1.el7.x86_64 #1 SMP Tue Jun 4 14:43:51 UTC 2024 x86_64
// Copyright 2022 Google LLC
// SPDX-License-Identifier: BSD-2-Clause
// Compares two files and returns whether they are the same once decoded.
#include <iostream>
#include <string>
#include "aviftest_helpers.h"
#include "avifutil.h"
using libavif::testutil::AvifImagePtr;
int main(int argc, char** argv) {
if (argc != 4 && argc != 5) {
std::cerr << "Wrong argument: " << argv[0]
<< " file1 file2 ignore_alpha_flag [psnr_threshold]" << std::endl;
return 2;
}
AvifImagePtr decoded[2] = {
AvifImagePtr(avifImageCreateEmpty(), avifImageDestroy),
AvifImagePtr(avifImageCreateEmpty(), avifImageDestroy)};
if (!decoded[0] || !decoded[1]) {
std::cerr << "Cannot create AVIF images." << std::endl;
return 2;
}
uint32_t depth[2];
// Request the bit depth closest to the bit depth of the input file.
constexpr int kRequestedDepth = 0;
constexpr avifPixelFormat requestedFormat = AVIF_PIXEL_FORMAT_NONE;
for (int i : {0, 1}) {
// Make sure no color conversion happens.
decoded[i]->matrixCoefficients = AVIF_MATRIX_COEFFICIENTS_IDENTITY;
if (avifReadImage(argv[i + 1], requestedFormat, kRequestedDepth,
AVIF_CHROMA_DOWNSAMPLING_AUTOMATIC,
/*ignoreColorProfile==*/AVIF_FALSE,
/*ignoreExif=*/AVIF_FALSE,
/*ignoreXMP=*/AVIF_FALSE, /*allowChangingCicp=*/AVIF_TRUE,
decoded[i].get(), &depth[i], nullptr,
nullptr) == AVIF_APP_FILE_FORMAT_UNKNOWN) {
std::cerr << "Image " << argv[i + 1] << " cannot be read." << std::endl;
return 2;
}
}
if (depth[0] != depth[1]) {
std::cerr << "Images " << argv[1] << " and " << argv[2]
<< " have different depths." << std::endl;
return 1;
}
bool ignore_alpha = std::stoi(argv[3]) != 0;
if (argc == 4) {
if (!libavif::testutil::AreImagesEqual(*decoded[0], *decoded[1],
ignore_alpha)) {
std::cerr << "Images " << argv[1] << " and " << argv[2]
<< " are different." << std::endl;
return 1;
}
std::cout << "Images " << argv[1] << " and " << argv[2] << " are identical."
<< std::endl;
} else {
auto psnr =
libavif::testutil::GetPsnr(*decoded[0], *decoded[1], ignore_alpha);
if (psnr < std::stod(argv[4])) {
std::cerr << "PSNR: " << psnr << ", images " << argv[1] << " and "
<< argv[2] << " are not similar." << std::endl;
return 1;
}
std::cout << "PSNR: " << psnr << ", images " << argv[1] << " and "
<< argv[2] << " are similar." << std::endl;
}
return 0;
}