RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-26 05:27:46 +00:00
Code Issues Activity Actions

LibGfx/PortableFormat: Add PortableFormatWriter

Browse source 
Currently, the encoder only output ppm files, but it can easily be
extended to support both PBM and PGM.
This commit is contained in:
Lucas CHOLLET 2023-03-12 22:18:16 -04:00 • committed by Linus Groh
parent e32267c902
commit 9c9bd271c8
3 changed files with 93 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

54
Userland/Libraries/LibGfx/PortableFormatWriter.cpp Normal file
View file

@ -0,0 +1,54 @@
/*
* Copyright (c) 2023, Lucas Chollet <lucas.chollet@serenityos.org >
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "PortableFormatWriter.h"
#include <AK/String.h>
namespace Gfx {
ErrorOr<ByteBuffer> PortableFormatWriter::encode(Bitmap const& bitmap, Options options)
{
ByteBuffer buffer;
// FIXME: Add support for PBM and PGM
TRY(add_header(buffer, options, bitmap.width(), bitmap.height(), 255));
TRY(add_pixels(buffer, options, bitmap));
return buffer;
}
ErrorOr<void> PortableFormatWriter::add_header(ByteBuffer& buffer, Options const& options, u32 width, u32 height, u32 maximal_value)
{
TRY(buffer.try_append(TRY(String::formatted("P{}\n", options.format == Options::Format::ASCII ? "3"sv : "6"sv)).bytes()));
TRY(buffer.try_append(TRY(String::formatted("# {}\n", options.comment)).bytes()));
TRY(buffer.try_append(TRY(String::formatted("{} {}\n", width, height)).bytes()));
TRY(buffer.try_append(TRY(String::formatted("{}\n", maximal_value)).bytes()));
return {};
}
ErrorOr<void> PortableFormatWriter::add_pixels(ByteBuffer& buffer, Options const& options, Bitmap const& bitmap)
{
for (int i = 0; i < bitmap.height(); ++i) {
for (int j = 0; j < bitmap.width(); ++j) {
auto color = bitmap.get_pixel(j, i);
if (options.format == Options::Format::ASCII) {
TRY(buffer.try_append(TRY(String::formatted("{} {} {}\t", color.red(), color.green(), color.blue())).bytes()));
} else {
TRY(buffer.try_append(color.red()));
TRY(buffer.try_append(color.green()));
TRY(buffer.try_append(color.blue()));
}
}
if (options.format == Options::Format::ASCII)
TRY(buffer.try_append('\n'));
}
return {};
}
}