From 216385084bb04d796af216ee1f1b44386a03f3c9 Mon Sep 17 00:00:00 2001 From: BenJilks Date: Sun, 18 Oct 2020 14:39:22 +0000 Subject: [PATCH] LibGfx: Add BMPWriter This utility allows you to create a BMP file from a bitmap --- Libraries/LibGfx/BMPWriter.cpp | 135 ++++++++++++++++++++++++++++++++ Libraries/LibGfx/BMPWriter.h | 51 ++++++++++++ Libraries/LibGfx/CMakeLists.txt | 1 + 3 files changed, 187 insertions(+) create mode 100644 Libraries/LibGfx/BMPWriter.cpp create mode 100644 Libraries/LibGfx/BMPWriter.h diff --git a/Libraries/LibGfx/BMPWriter.cpp b/Libraries/LibGfx/BMPWriter.cpp new file mode 100644 index 0000000000..8ae2b00a8b --- /dev/null +++ b/Libraries/LibGfx/BMPWriter.cpp @@ -0,0 +1,135 @@ +/* + * Copyright (c) 2020, Ben Jilks + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include +#include +#include +#include + +namespace Gfx { + +constexpr int bytes_per_pixel = 3; + +#define FILE_HEADER_SIZE 14 +#define IMAGE_INFORMATION_SIZE 40 +#define PIXEL_DATA_OFFSET FILE_HEADER_SIZE + IMAGE_INFORMATION_SIZE + +class Streamer { +public: + Streamer(u8* data) + : m_data(data) + { + } + + void write_u8(u8 i) + { + *(m_data++) = i; + } + + void write_u16(u16 i) + { + *(m_data++) = i & 0xFF; + *(m_data++) = (i >> 8) & 0xFF; + } + + void write_u32(u32 i) + { + write_u16(i & 0xFFFF); + write_u16((i >> 16) & 0xFFFF); + } + + void write_i32(i32 i) + { + write_u32(static_cast(i)); + } + +private: + u8* m_data; +}; + +static ByteBuffer write_pixel_data(const RefPtr bitmap, int pixel_row_data_size) +{ + int image_size = pixel_row_data_size * bitmap->height(); + auto buffer = ByteBuffer::create_uninitialized(image_size); + + int current_row = 0; + for (int y = bitmap->height() - 1; y >= 0; --y) { + auto* row = buffer.data() + (pixel_row_data_size * current_row++); + for (int x = 0; x < bitmap->width(); x++) { + auto pixel = bitmap->get_pixel(x, y); + row[x * bytes_per_pixel + 0] = pixel.blue(); + row[x * bytes_per_pixel + 1] = pixel.green(); + row[x * bytes_per_pixel + 2] = pixel.red(); + } + } + + return buffer; +} + +static ByteBuffer compress_pixel_data(const ByteBuffer& pixel_data, BMPWriter::Compression compression) +{ + switch (compression) { + case BMPWriter::Compression::RGB: + return pixel_data; + } + + ASSERT_NOT_REACHED(); +} + +ByteBuffer BMPWriter::dump(const RefPtr bitmap) +{ + int pixel_row_data_size = (bytes_per_pixel * 8 * bitmap->width() + 31) / 32 * 4; + int image_size = pixel_row_data_size * bitmap->height(); + auto buffer = ByteBuffer::create_uninitialized(PIXEL_DATA_OFFSET); + + auto pixel_data = write_pixel_data(bitmap, pixel_row_data_size); + pixel_data = compress_pixel_data(pixel_data, m_compression); + + int file_size = PIXEL_DATA_OFFSET + pixel_data.size(); + Streamer streamer(buffer.data()); + streamer.write_u8('B'); + streamer.write_u8('M'); + streamer.write_u32(file_size); + streamer.write_u32(0); + streamer.write_u32(PIXEL_DATA_OFFSET); + + streamer.write_u32(IMAGE_INFORMATION_SIZE); // Header size + streamer.write_i32(bitmap->width()); // ImageWidth + streamer.write_i32(bitmap->height()); // ImageHeight + streamer.write_u16(1); // Planes + streamer.write_u16(bytes_per_pixel * 8); // BitsPerPixel + streamer.write_u32((u32)m_compression); // Compression + streamer.write_u32(image_size); // ImageSize + streamer.write_i32(0); // XpixelsPerMeter + streamer.write_i32(0); // YpixelsPerMeter + streamer.write_u32(0); // TotalColors + streamer.write_u32(0); // ImportantColors + + buffer.append(pixel_data.data(), pixel_data.size()); + return buffer; +} + +} diff --git a/Libraries/LibGfx/BMPWriter.h b/Libraries/LibGfx/BMPWriter.h new file mode 100644 index 0000000000..d881ab5f9d --- /dev/null +++ b/Libraries/LibGfx/BMPWriter.h @@ -0,0 +1,51 @@ +/* + * Copyright (c) 2020, Ben Jilks + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#include + +namespace Gfx { + +class Bitmap; + +class BMPWriter { +public: + BMPWriter() = default; + + ByteBuffer dump(const RefPtr); + + enum class Compression : u32 { + RGB = 0, + }; + + inline void set_compression(Compression compression) { m_compression = compression; } + +private: + Compression m_compression { Compression::RGB }; +}; + +} diff --git a/Libraries/LibGfx/CMakeLists.txt b/Libraries/LibGfx/CMakeLists.txt index 81545c35ff..b919b3246a 100644 --- a/Libraries/LibGfx/CMakeLists.txt +++ b/Libraries/LibGfx/CMakeLists.txt @@ -2,6 +2,7 @@ set(SOURCES AffineTransform.cpp Bitmap.cpp BMPLoader.cpp + BMPWriter.cpp CharacterBitmap.cpp ClassicStylePainter.cpp ClassicWindowTheme.cpp