b9574c180e
The `read_image_data` function of each one of[PBM, PGM, PPM]Loader use the same structure to read an image. This patch harmonizes the three functions and use finite loops instead of reading until EOF. It allows to quit early on bloated file, but it's mainly done for refactoring purpose.
67 lines
1.9 KiB
C++
67 lines
1.9 KiB
C++
/*
|
|
* Copyright (c) 2020, Hüseyin ASLITÜRK <asliturk@hotmail.com>
|
|
* Copyright (c) 2022, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/Endian.h>
|
|
#include <LibGfx/ImageFormats/PGMLoader.h>
|
|
#include <LibGfx/ImageFormats/PortableImageLoaderCommon.h>
|
|
#include <LibGfx/Streamer.h>
|
|
#include <string.h>
|
|
|
|
namespace Gfx {
|
|
|
|
static void set_adjusted_pixels(PGMLoadingContext& context, Vector<Gfx::Color> const& color_data)
|
|
{
|
|
size_t index = 0;
|
|
for (size_t y = 0; y < context.height; ++y) {
|
|
for (size_t x = 0; x < context.width; ++x) {
|
|
Color color = color_data.at(index);
|
|
if (context.format_details.max_val < 255) {
|
|
color = adjust_color(context.format_details.max_val, color);
|
|
}
|
|
context.bitmap->set_pixel(x, y, color);
|
|
++index;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool read_image_data(PGMLoadingContext& context, Streamer& streamer)
|
|
{
|
|
Vector<Gfx::Color> color_data;
|
|
auto const context_size = context.width * context.height;
|
|
|
|
color_data.resize(context_size);
|
|
|
|
if (context.type == PGMLoadingContext::Type::ASCII) {
|
|
for (u64 i = 0; i < context_size; ++i) {
|
|
auto number_or_error = read_number(streamer);
|
|
if (number_or_error.is_error())
|
|
return false;
|
|
auto value = number_or_error.value();
|
|
|
|
if (read_whitespace(context, streamer).is_error())
|
|
return false;
|
|
|
|
color_data[i] = { (u8)value, (u8)value, (u8)value };
|
|
}
|
|
} else if (context.type == PGMLoadingContext::Type::RAWBITS) {
|
|
for (u64 i = 0; i < context_size; ++i) {
|
|
u8 pixel;
|
|
if (!streamer.read(pixel))
|
|
return false;
|
|
color_data[i] = { pixel, pixel, pixel };
|
|
}
|
|
}
|
|
|
|
if (!create_bitmap(context))
|
|
return false;
|
|
|
|
set_adjusted_pixels(context, color_data);
|
|
|
|
context.state = PGMLoadingContext::State::Bitmap;
|
|
return true;
|
|
}
|
|
}
|