Only report "fusing" different colors once per pair

This commit is contained in:
ISSOtm
2022-04-24 19:10:31 +02:00
committed by Eldred Habert
parent 82012f698e
commit 568fb5e4c8

View File

@@ -11,12 +11,14 @@
#include <algorithm> #include <algorithm>
#include <assert.h> #include <assert.h>
#include <cinttypes> #include <cinttypes>
#include <climits>
#include <errno.h> #include <errno.h>
#include <fstream> #include <fstream>
#include <memory> #include <memory>
#include <optional> #include <optional>
#include <png.h> #include <png.h>
#include <setjmp.h> #include <setjmp.h>
#include <stdint.h>
#include <string.h> #include <string.h>
#include <tuple> #include <tuple>
#include <unordered_set> #include <unordered_set>
@@ -38,7 +40,12 @@ class ImagePalette {
public: public:
ImagePalette() = default; ImagePalette() = default;
void registerColor(Rgba const &rgba) { /**
* Registers a color in the palette.
* If the newly inserted color "conflicts" with another one (different color, but same CGB
* color), then the other color is returned. Otherwise, `nullptr` is returned.
*/
[[nodiscard]] Rgba const *registerColor(Rgba const &rgba) {
decltype(_colors)::value_type &slot = _colors[rgba.cgbColor()]; decltype(_colors)::value_type &slot = _colors[rgba.cgbColor()];
if (rgba.cgbColor() == Rgba::transparent) { if (rgba.cgbColor() == Rgba::transparent) {
@@ -48,9 +55,10 @@ public:
if (!slot.has_value()) { if (!slot.has_value()) {
slot.emplace(rgba); slot.emplace(rgba);
} else if (*slot != rgba) { } else if (*slot != rgba) {
warning("Different colors melded together (#%08x into #%08x as %04x)", rgba.toCSS(), assert(slot->cgbColor() != UINT16_MAX);
slot->toCSS(), rgba.cgbColor()); // TODO: indicate position return &*slot;
} }
return nullptr;
} }
size_t size() const { size_t size() const {
@@ -310,15 +318,36 @@ public:
size_t nbRowBytes = png_get_rowbytes(png, info); size_t nbRowBytes = png_get_rowbytes(png, info);
assert(nbRowBytes != 0); assert(nbRowBytes != 0);
DefaultInitVec<png_byte> row(nbRowBytes); DefaultInitVec<png_byte> row(nbRowBytes);
// Holds known-conflicting color pairs to avoid warning about them twice.
// We don't need to worry about transitivity, as ImagePalette slots are immutable once
// assigned, and conflicts always occur between that and another color.
// For the same reason, we don't need to worry about order, either.
std::vector<std::tuple<uint32_t, uint32_t>> conflicts;
// Assign a color to the given position, and register it in the image palette as well
auto assignColor = [this, &conflicts](png_uint_32 x, png_uint_32 y, Rgba &&color) {
if (Rgba const *other = colors.registerColor(color); other) {
std::tuple conflicting{color.toCSS(), other->toCSS()};
// Do not report combinations twice
if (std::find(conflicts.begin(), conflicts.end(), conflicting) == conflicts.end()) {
warning("Fusing colors #%08x and #%08x into Game Boy color $%04x",
std::get<0>(conflicting), std::get<1>(conflicting),
color.cgbColor()); // TODO: indicate position
// Do not report this combination again
conflicts.emplace_back(conflicting);
}
}
pixel(x, y) = color;
};
if (interlaceType == PNG_INTERLACE_NONE) { if (interlaceType == PNG_INTERLACE_NONE) {
for (png_uint_32 y = 0; y < height; ++y) { for (png_uint_32 y = 0; y < height; ++y) {
png_read_row(png, row.data(), nullptr); png_read_row(png, row.data(), nullptr);
for (png_uint_32 x = 0; x < width; ++x) { for (png_uint_32 x = 0; x < width; ++x) {
Rgba rgba(row[x * 4], row[x * 4 + 1], row[x * 4 + 2], row[x * 4 + 3]); assignColor(x, y,
colors.registerColor(rgba); Rgba(row[x * 4], row[x * 4 + 1], row[x * 4 + 2], row[x * 4 + 3]));
pixel(x, y) = rgba;
} }
} }
} else { } else {
@@ -339,9 +368,7 @@ public:
png_read_row(png, ptr, nullptr); png_read_row(png, ptr, nullptr);
for (png_uint_32 x = PNG_PASS_START_COL(pass); x < width; x += xStep) { for (png_uint_32 x = PNG_PASS_START_COL(pass); x < width; x += xStep) {
Rgba rgba(ptr[0], ptr[1], ptr[2], ptr[3]); assignColor(x, y, Rgba(ptr[0], ptr[1], ptr[2], ptr[3]));
colors.registerColor(rgba);
pixel(x, y) = rgba;
ptr += 4; ptr += 4;
} }
} }