summaryrefslogtreecommitdiff
path: root/vendor/exr/tests/across_compression.rs
diff options
context:
space:
mode:
authorValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
committerValentin Popov <valentin@popov.link>2024-01-08 00:21:28 +0300
commit1b6a04ca5504955c571d1c97504fb45ea0befee4 (patch)
tree7579f518b23313e8a9748a88ab6173d5e030b227 /vendor/exr/tests/across_compression.rs
parent5ecd8cf2cba827454317368b68571df0d13d7842 (diff)
downloadfparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.tar.xz
fparkan-1b6a04ca5504955c571d1c97504fb45ea0befee4.zip
Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
Diffstat (limited to 'vendor/exr/tests/across_compression.rs')
-rw-r--r--vendor/exr/tests/across_compression.rs263
1 files changed, 263 insertions, 0 deletions
diff --git a/vendor/exr/tests/across_compression.rs b/vendor/exr/tests/across_compression.rs
new file mode 100644
index 0000000..b1776e9
--- /dev/null
+++ b/vendor/exr/tests/across_compression.rs
@@ -0,0 +1,263 @@
+use std::path::Path;
+use exr::prelude::*;
+use exr::image::validate_results::ValidateResult;
+
+fn dir() -> &'static Path { Path::new("tests/images/valid/custom/compression_methods") }
+
+fn expect_eq_other(sub_dir: &str, image_name: &str, expected: &str) {
+ let path = dir().join(sub_dir).join(image_name);
+
+ match read_first_flat_layer_from_file(path) {
+ Err(Error::NotSupported(message)) => println!("skipping ({})", message),
+ Err(error) => panic!("unexpected error: {}", error),
+ Ok(mut decompressed) => {
+ let decompressed_path = dir().join(sub_dir).join(expected);
+ let mut expected_decompressed = read_first_flat_layer_from_file(decompressed_path)
+ .expect("uncompressed image could not be loaded");
+
+ // HACK: make metadata match artificially, to avoid failing the check due to meta data mismatch
+ // (the name of the compression methods should not be equal, as we test between compression methods)
+ expected_decompressed.layer_data.encoding.compression = Compression::Uncompressed;
+ decompressed.layer_data.encoding.compression = Compression::Uncompressed;
+
+ debug_assert_eq!(expected_decompressed.layer_data.attributes, decompressed.layer_data.attributes, "attributes should not be affected by compression");
+ debug_assert_eq!(expected_decompressed.layer_data.size, decompressed.layer_data.size, "size should not be affected by compression");
+
+ // Note: Unimplemented methods may still work, if each compressed tile would be larger than uncompressed.
+ expected_decompressed.assert_equals_result(&decompressed);
+ }
+ }
+}
+
+// comparing to a different format, png,
+// is the only real way to check that
+// little endian data is unpacked correctly on big endian systems
+// it does not attempt to compare NaN
+fn expect_eq_png(image_name: &str) {
+ type Rgb16Image = ::image::ImageBuffer<::image::Rgb<u16>, Vec<u16>>;
+
+ let exr_path = dir().join("u16").join(image_name);
+ let png_from_exr = read_first_rgba_layer_from_file(
+ exr_path,
+
+ |resolution, _channels: &RgbaChannels| -> Rgb16Image {
+ ::image::ImageBuffer::new(
+ resolution.width() as u32,
+ resolution.height() as u32
+ )
+ },
+
+ // set each pixel in the png buffer from the exr file
+ |png_pixels: &mut Rgb16Image, position: Vec2<usize>, (r,g,b,_): (f32,f32,f32,f32)| {
+ png_pixels.put_pixel(
+ position.x() as u32, position.y() as u32,
+ ::image::Rgb([to_u16(r), to_u16(g), to_u16(b)])
+ );
+ }
+ );
+
+ fn to_u16(num: f32) -> u16 { (num.powf(1.0/2.14).clamp(0.0, 1.0) * u16::MAX as f32).round() as u16 }
+
+ match png_from_exr {
+ Err(Error::NotSupported(message)) => println!("skipping ({})", message),
+ Err(error) => panic!("unexpected error: {}", error),
+ Ok(decompressed) => {
+ let truth_path = dir().join("u16").join("ground_truth.png");
+ let truth_dyn_img = image::open(truth_path).unwrap();
+
+ let ground_truth_png = truth_dyn_img.to_rgb16();
+ let exr_as_png_px = decompressed.layer_data.channel_data.pixels;
+ debug_assert_eq!(ground_truth_png.dimensions(), exr_as_png_px.dimensions(), "size should not be affected by compression");
+
+ let expected_px = ground_truth_png.pixels()
+ .flat_map(|px| px.0.iter().copied());
+
+ let actual_px = exr_as_png_px.pixels()
+ .flat_map(|px| px.0.iter().copied());
+
+ let max_diff = u16::MAX/10;
+ for (exp, val) in expected_px.zip(actual_px) {
+ assert!(
+ exp.abs_diff(val) < max_diff,
+ "values not similar enough: found {}, expected {}", val, exp
+ );
+ }
+ }
+ }
+}
+
+fn expect_eq_uncompressed(sub_dir: &str, image_name: &str) {
+ expect_eq_other(sub_dir, image_name, "uncompressed.exr")
+}
+
+
+#[test]
+fn compare_compression_contents_zip_f32() {
+ expect_eq_uncompressed("f32", "zip.exr");
+}
+
+#[test]
+fn compare_compression_contents_zip_f16() {
+ expect_eq_uncompressed("f16", "zip.exr");
+}
+
+#[test]
+fn compare_compression_contents_zips_f32() {
+ expect_eq_uncompressed("f32", "zips.exr");
+}
+
+#[test]
+fn compare_compression_contents_zips_f16() {
+ expect_eq_uncompressed("f16", "zips.exr");
+}
+
+#[test]
+fn compare_compression_contents_b44_f32() {
+ expect_eq_uncompressed("f32", "b44.exr"); // f32s are not compressed in b44 and can be compared exactly
+}
+
+#[test]
+fn compare_compression_contents_b44_f16() {
+ expect_eq_other("f16", "b44.exr", "decompressed_b44.exr");
+}
+
+#[test]
+fn compare_compression_contents_b44a_f32() {
+ expect_eq_uncompressed("f32", "b44a.exr"); // f32s are not compressed in b44 and can be compared exactly
+}
+
+#[test]
+fn compare_compression_contents_b44a_f16() {
+ expect_eq_other("f16", "b44a.exr", "decompressed_b44a.exr");
+}
+
+#[test]
+fn compare_compression_contents_dwaa_f32() {
+ expect_eq_other("f32", "dwaa.exr", "decompressed_dwaa.exr");
+}
+
+#[test]
+fn compare_compression_contents_dwaa_f16() {
+ expect_eq_other("f16", "dwaa.exr", "decompressed_dwaa.exr");
+}
+
+#[test]
+fn compare_compression_contents_dwab_f32() {
+ expect_eq_other("f32", "dwab.exr", "decompressed_dwab.exr");
+}
+
+#[test]
+fn compare_compression_contents_dwab_f16() {
+ expect_eq_other("f16", "dwab.exr", "decompressed_dwab.exr");
+}
+
+#[test]
+fn compare_compression_contents_piz_f32() {
+ expect_eq_uncompressed("f32", "piz.exr");
+}
+
+#[test]
+fn compare_compression_contents_piz_f16() {
+ expect_eq_uncompressed("f16", "piz.exr");
+}
+
+#[test]
+fn compare_compression_contents_rle_f32() {
+ expect_eq_uncompressed("f32", "rle.exr");
+}
+
+#[test]
+fn compare_compression_contents_rle_f16() {
+ expect_eq_uncompressed("f16", "rle.exr");
+}
+
+
+#[test]
+#[cfg(target_endian = "little")] // TODO make it work on big endian
+fn compare_compression_contents_pxr24_f16() {
+ expect_eq_other("f16", "pxr24.exr", "decompressed_pxr24.exr");
+}
+
+#[test]
+#[cfg(target_endian = "little")] // TODO make it work on big endian
+fn compare_compression_contents_pxr24_f32() {
+ expect_eq_other("f32", "pxr24.exr", "decompressed_pxr24.exr");
+}
+
+#[test]
+fn compare_png_to_uncompressed_f16() {
+ expect_eq_png("f16_uncompressed.exr");
+}
+
+#[test]
+fn compare_png_to_piz_f16() {
+ expect_eq_png("f16_piz.exr");
+}
+
+#[test]
+fn compare_png_to_rle_f16() {
+ expect_eq_png("f16_rle.exr");
+}
+
+#[test]
+fn compare_png_to_zip_f16() {
+ expect_eq_png("f16_zip.exr");
+}
+
+#[test]
+fn compare_png_to_zips_f16() {
+ expect_eq_png("f16_zips.exr");
+}
+
+#[test]
+fn compare_png_to_dwaa_f16() {
+ expect_eq_png("f16_dwaa.exr");
+}
+
+#[test]
+fn compare_png_to_b44_f16() {
+ expect_eq_png("f16_b44.exr");
+}
+
+#[test]
+fn compare_png_to_b44a_f16() {
+ expect_eq_png("f16_b44a.exr");
+}
+
+#[test]
+#[cfg(target_endian = "little")] // TODO make it work on big endian
+fn compare_png_to_pxr24_f16() {
+ expect_eq_png("f16_pxr24.exr");
+}
+
+
+#[test]
+fn compare_png_to_uncompressed_f32() {
+ expect_eq_png("f32_uncompressed.exr");
+}
+
+#[test]
+fn compare_png_to_piz_f32() {
+ expect_eq_png("f32_piz.exr");
+}
+
+#[test]
+fn compare_png_to_rle_f32() {
+ expect_eq_png("f32_rle.exr");
+}
+
+#[test]
+fn compare_png_to_zip_f32() {
+ expect_eq_png("f32_zip.exr");
+}
+
+#[test]
+fn compare_png_to_dwaa_f32() {
+ expect_eq_png("f32_dwaa.exr");
+}
+
+#[test]
+#[cfg(target_endian = "little")] // TODO make it work on big endian
+fn compare_png_to_pxr24_f32() {
+ expect_eq_png("f32_pxr24.exr");
+}