ladybird/Userland/Libraries/LibGfx/ImageFormats/PAMLoader.cpp
Nico Weber 83ab9f7c2d LibGfx/PNM: Remove unnecessary line
This is already done at the caller decode() in
PortableImageLoaderCommon.h, as pointed out by @LucasChollet at
https://github.com/SerenityOS/serenity/pull/22935#discussion_r1467158789

No behavior change.
2024-01-26 14:53:33 +01:00

63 lines
2.7 KiB
C++

/*
* Copyright (c) 2024, the SerenityOS developers.
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "PAMLoader.h"
#include "PortableImageLoaderCommon.h"
namespace Gfx {
ErrorOr<void> read_image_data(PAMLoadingContext& context)
{
VERIFY(context.type == PAMLoadingContext::Type::RAWBITS);
// FIXME: Technically it's more to spec to check that a known tupl type has a minimum depth and then skip additional channels.
bool is_gray = context.format_details.depth == 1 && context.format_details.tupl_type == "GRAYSCALE"sv;
bool is_gray_alpha = context.format_details.depth == 2 && context.format_details.tupl_type == "GRAYSCALE_ALPHA"sv;
bool is_rgb = context.format_details.depth == 3 && context.format_details.tupl_type == "RGB"sv;
bool is_rgba = context.format_details.depth == 4 && context.format_details.tupl_type == "RGB_ALPHA"sv;
bool is_cmyk = context.format_details.depth == 4 && context.format_details.tupl_type == "CMYK"sv;
if (!is_gray && !is_gray_alpha && !is_rgb && !is_rgba && !is_cmyk)
return Error::from_string_view("Unsupported PAM depth"sv);
auto& stream = *context.stream;
if (is_cmyk) {
context.format_details.cmyk_bitmap = TRY(CMYKBitmap::create_with_size({ context.width, context.height }));
CMYK* data = context.format_details.cmyk_bitmap.value()->begin();
for (u64 i = 0; i < context.width * context.height; ++i) {
Array<u8, 4> pixel;
TRY(stream.read_until_filled(pixel));
data[i] = { pixel[0], pixel[1], pixel[2], pixel[3] };
}
} else {
TRY(create_bitmap(context));
for (u64 i = 0; i < context.width * context.height; ++i) {
if (is_gray) {
Array<u8, 1> pixel;
TRY(stream.read_until_filled(pixel));
context.bitmap->set_pixel(i % context.width, i / context.width, { pixel[0], pixel[0], pixel[0] });
} else if (is_gray_alpha) {
Array<u8, 2> pixel;
TRY(stream.read_until_filled(pixel));
context.bitmap->set_pixel(i % context.width, i / context.width, { pixel[0], pixel[0], pixel[0], pixel[1] });
} else if (is_rgb) {
Array<u8, 3> pixel;
TRY(stream.read_until_filled(pixel));
context.bitmap->set_pixel(i % context.width, i / context.width, { pixel[0], pixel[1], pixel[2] });
} else if (is_rgba) {
Array<u8, 4> pixel;
TRY(stream.read_until_filled(pixel));
context.bitmap->set_pixel(i % context.width, i / context.width, { pixel[0], pixel[1], pixel[2], pixel[3] });
}
}
}
return {};
}
}