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.8 KiB
C++
67 lines
1.8 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 "PBMLoader.h"
|
|
#include "AK/Endian.h"
|
|
#include "PortableImageLoaderCommon.h"
|
|
#include "Userland/Libraries/LibGfx/Streamer.h"
|
|
#include <string.h>
|
|
|
|
namespace Gfx {
|
|
|
|
bool read_image_data(PBMLoadingContext& context, Streamer& streamer)
|
|
{
|
|
Vector<Gfx::Color> color_data;
|
|
|
|
auto const context_size = context.width * context.height;
|
|
color_data.resize(context_size);
|
|
|
|
if (context.type == PBMLoadingContext::Type::ASCII) {
|
|
for (u64 i = 0; i < context_size; ++i) {
|
|
u8 byte;
|
|
if (!streamer.read(byte))
|
|
return false;
|
|
if (byte == '0')
|
|
color_data[i] = Color::White;
|
|
else if (byte == '1')
|
|
color_data[i] = Color::Black;
|
|
else
|
|
i--;
|
|
}
|
|
} else if (context.type == PBMLoadingContext::Type::RAWBITS) {
|
|
for (u64 color_index = 0; color_index < context_size;) {
|
|
u8 byte;
|
|
if (!streamer.read(byte))
|
|
return false;
|
|
for (int i = 0; i < 8; i++) {
|
|
int val = byte & 0x80;
|
|
|
|
if (val == 0)
|
|
color_data[color_index] = Color::White;
|
|
else
|
|
color_data[color_index] = Color::Black;
|
|
|
|
byte = byte << 1;
|
|
color_index++;
|
|
|
|
if (color_index % context.width == 0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!create_bitmap(context)) {
|
|
return false;
|
|
}
|
|
|
|
set_pixels(context, color_data);
|
|
|
|
context.state = PBMLoadingContext::State::Bitmap;
|
|
return true;
|
|
}
|
|
}
|