ladybird/Userland/Utilities/test-jpeg-roundtrip.cpp
2024-01-14 11:35:40 +00:00

105 lines
3.1 KiB
C++

/*
* Copyright (c) 2024, the SerenityOS developers.
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibGfx/DeltaE.h>
#include <LibGfx/ICC/Profile.h>
#include <LibGfx/ICC/WellKnownProfiles.h>
#include <LibGfx/ImageFormats/JPEGLoader.h>
#include <LibGfx/ImageFormats/JPEGWriter.h>
#include <LibMain/Main.h>
struct Fixpoint {
Gfx::Color fixpoint;
int number_of_iterations {};
};
static ErrorOr<Fixpoint> compute_fixpoint(Gfx::Color start_color)
{
auto bitmap = TRY(Gfx::Bitmap::create(Gfx::BitmapFormat::BGRx8888, { 8, 8 }));
bitmap->fill(start_color);
int number_of_iterations = 1;
Color last_color = start_color;
while (true) {
AllocatingMemoryStream stream;
TRY(Gfx::JPEGWriter::encode(stream, *bitmap));
auto data = TRY(stream.read_until_eof());
auto plugin_decoder = TRY(Gfx::JPEGImageDecoderPlugin::create(data));
auto frame = TRY(plugin_decoder->frame(0));
Color current_color = frame.image->get_pixel(4, 4);
if (current_color == last_color)
break;
++number_of_iterations;
last_color = current_color;
bitmap = *frame.image;
}
return Fixpoint { last_color, number_of_iterations };
}
static ErrorOr<float> perceived_distance_in_sRGB(Gfx::Color a, Gfx::Color b)
{
auto sRGB = TRY(Gfx::ICC::sRGB());
Array<u8, 3> array_a { a.red(), a.green(), a.blue() };
Array<u8, 3> array_b { b.red(), b.green(), b.blue() };
return DeltaE(TRY(sRGB->to_lab(array_a)), TRY(sRGB->to_lab(array_b)));
}
struct Stats {
float max_delta {};
int max_number_of_iterations {};
};
static ErrorOr<void> test(Gfx::Color color, Stats& stats)
{
auto fixpoint = TRY(compute_fixpoint(color));
float perceived_distance = TRY(perceived_distance_in_sRGB(color, fixpoint.fixpoint));
outln("color {} converges to {} after saving {} times, delta {}", color, fixpoint.fixpoint, fixpoint.number_of_iterations, perceived_distance);
stats.max_delta = max(stats.max_delta, perceived_distance);
stats.max_number_of_iterations = max(stats.max_number_of_iterations, fixpoint.number_of_iterations);
return {};
}
ErrorOr<int> serenity_main(Main::Arguments)
{
Stats stats;
TRY(test(Gfx::Color::Red, stats));
TRY(test(Gfx::Color::Green, stats));
TRY(test(Gfx::Color::Blue, stats));
TRY(test(Gfx::Color::LightBlue, stats));
TRY(test(Gfx::Color::MidRed, stats));
TRY(test(Gfx::Color::MidGreen, stats));
TRY(test(Gfx::Color::MidBlue, stats));
TRY(test(Gfx::Color::DarkRed, stats));
TRY(test(Gfx::Color::DarkGreen, stats));
TRY(test(Gfx::Color::DarkBlue, stats));
TRY(test(Gfx::Color::Cyan, stats));
TRY(test(Gfx::Color::Magenta, stats));
TRY(test(Gfx::Color::Yellow, stats));
TRY(test(Gfx::Color::Black, stats));
TRY(test(Gfx::Color::DarkGray, stats));
TRY(test(Gfx::Color::MidGray, stats));
TRY(test(Gfx::Color::LightGray, stats));
TRY(test(Gfx::Color::White, stats));
outln();
outln("max delta {}, max number of iterations {}", stats.max_delta, stats.max_number_of_iterations);
return 0;
}