mirror of
https://github.com/RGBCube/serenity
synced 2025-05-25 22:05:07 +00:00
d4d3c3f262
As we directly write to the stream, we don't need to store a copy of the entire image in memory. However, writing to a stream is heavier on the CPU than to a ByteBuffer. This commit unfortunately makes `add_pixels` two times slower.
52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) 2023, Lucas Chollet <lucas.chollet@serenityos.org >
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "PortableFormatWriter.h"
|
|
#include <AK/Stream.h>
|
|
|
|
namespace Gfx {
|
|
|
|
ErrorOr<void> PortableFormatWriter::encode(Stream& output, Bitmap const& bitmap, Options options)
|
|
{
|
|
// FIXME: Add support for PBM and PGM
|
|
|
|
TRY(add_header(output, options, bitmap.width(), bitmap.height(), 255));
|
|
TRY(add_pixels(output, options, bitmap));
|
|
|
|
return {};
|
|
}
|
|
|
|
ErrorOr<void> PortableFormatWriter::add_header(Stream& output, Options const& options, u32 width, u32 height, u32 maximal_value)
|
|
{
|
|
TRY(output.write_formatted("P{}\n", options.format == Options::Format::ASCII ? "3"sv : "6"sv));
|
|
TRY(output.write_formatted("# {}\n", options.comment));
|
|
TRY(output.write_formatted("{} {}\n", width, height));
|
|
TRY(output.write_formatted("{}\n", maximal_value));
|
|
|
|
return {};
|
|
}
|
|
|
|
ErrorOr<void> PortableFormatWriter::add_pixels(Stream& output, 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(output.write_formatted("{} {} {}\t", color.red(), color.green(), color.blue()));
|
|
} else {
|
|
TRY(output.write_value(color.red()));
|
|
TRY(output.write_value(color.green()));
|
|
TRY(output.write_value(color.blue()));
|
|
}
|
|
}
|
|
if (options.format == Options::Format::ASCII)
|
|
TRY(output.write_value('\n'));
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
}
|