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

Mandelbrot: Add BMP and QOI image formats as options for export

Browse source 
Inspired by #13558 I added the option to export as QOI to the
Mandelbrot demo. I added BMP as well for completeness.
This commit is contained in:
EWouters 2022-04-13 19:04:22 +02:00 • committed by Tim Flynn
parent 65d4fb7649
commit d3bc3da873
1 changed files with 49 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

55
Userland/Demos/Mandelbrot/Mandelbrot.cpp
View file

@ -16,8 +16,10 @@
#include <LibGUI/Painter.h> #include <LibGUI/Painter.h>
#include <LibGUI/Widget.h> #include <LibGUI/Widget.h>
#include <LibGUI/Window.h> #include <LibGUI/Window.h>
#include <LibGfx/BMPWriter.h>
#include <LibGfx/Bitmap.h> #include <LibGfx/Bitmap.h>
#include <LibGfx/PNGWriter.h> #include <LibGfx/PNGWriter.h>
#include <LibGfx/QOIWriter.h>
#include <LibMain/Main.h> #include <LibMain/Main.h>
#include <unistd.h> #include <unistd.h>
@ -206,10 +208,16 @@ private:
} }
}; };
enum class ImageType {
BMP,
PNG,
QOI
};
class Mandelbrot : public GUI::Frame { class Mandelbrot : public GUI::Frame {
C_OBJECT(Mandelbrot) C_OBJECT(Mandelbrot)
void export_image(String const& export_path); void export_image(String const& export_path, ImageType image_type);
enum class Zoom { enum class Zoom {
In, In,
@ -358,17 +366,32 @@ void Mandelbrot::resize_event(GUI::ResizeEvent& event)
m_set.resize(event.size()); m_set.resize(event.size());
} }
void Mandelbrot::export_image(String const& export_path) void Mandelbrot::export_image(String const& export_path, ImageType image_type)
{ {
m_set.resize(Gfx::IntSize { 1920, 1080 }); m_set.resize(Gfx::IntSize { 1920, 1080 });
auto png = Gfx::PNGWriter::encode(m_set.bitmap()); ByteBuffer encoded_data;
switch (image_type) {
case ImageType::BMP: {
Gfx::BMPWriter dumper;
encoded_data = dumper.dump(m_set.bitmap());
break;
}
case ImageType::PNG:
encoded_data = Gfx::PNGWriter::encode(m_set.bitmap());
break;
case ImageType::QOI:
encoded_data = Gfx::QOIWriter::encode(m_set.bitmap());
break;
default:
VERIFY_NOT_REACHED();
}
m_set.resize(size()); m_set.resize(size());
auto file = fopen(export_path.characters(), "wb"); auto file = fopen(export_path.characters(), "wb");
if (!file) { if (!file) {
GUI::MessageBox::show(window(), String::formatted("Could not open '{}' for writing.", export_path), "Mandelbrot", GUI::MessageBox::Type::Error); GUI::MessageBox::show(window(), String::formatted("Could not open '{}' for writing.", export_path), "Mandelbrot", GUI::MessageBox::Type::Error);
return; return;
} }
fwrite(png.data(), 1, png.size(), file); fwrite(encoded_data.data(), 1, encoded_data.size(), file);
fclose(file); fclose(file);
GUI::MessageBox::show(window(), "Image was successfully exported.", "Mandelbrot", GUI::MessageBox::Type::Information); GUI::MessageBox::show(window(), "Image was successfully exported.", "Mandelbrot", GUI::MessageBox::Type::Information);
} }
@ -392,13 +415,33 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
auto mandelbrot = TRY(window->try_set_main_widget<Mandelbrot>()); auto mandelbrot = TRY(window->try_set_main_widget<Mandelbrot>());
auto file_menu = TRY(window->try_add_menu("&File")); auto file_menu = TRY(window->try_add_menu("&File"));
TRY(file_menu->try_add_action(GUI::Action::create("&Export...", { Mod_Ctrl | Mod_Shift, Key_S }, Gfx::Bitmap::try_load_from_file("/res/icons/16x16/save.png").release_value_but_fixme_should_propagate_errors(),
auto& export_submenu = file_menu->add_submenu("&Export");
TRY(export_submenu.try_add_action(GUI::Action::create("As &BMP",
[&](GUI::Action&) {
Optional<String> export_path = GUI::FilePicker::get_save_filepath(window, "untitled", "bmp");
if (!export_path.has_value())
return;
mandelbrot->export_image(export_path.value(), ImageType::BMP);
})));
TRY(export_submenu.try_add_action(GUI::Action::create("As &PNG", { Mod_Ctrl | Mod_Shift, Key_S },
[&](GUI::Action&) { [&](GUI::Action&) {
Optional<String> export_path = GUI::FilePicker::get_save_filepath(window, "untitled", "png"); Optional<String> export_path = GUI::FilePicker::get_save_filepath(window, "untitled", "png");
if (!export_path.has_value()) if (!export_path.has_value())
return; return;
mandelbrot->export_image(export_path.value()); mandelbrot->export_image(export_path.value(), ImageType::PNG);
}))); })));
TRY(export_submenu.try_add_action(GUI::Action::create("As &QOI",
[&](GUI::Action&) {
Optional<String> export_path = GUI::FilePicker::get_save_filepath(window, "untitled", "qoi");
if (!export_path.has_value())
return;
mandelbrot->export_image(export_path.value(), ImageType::QOI);
})));
export_submenu.set_icon(TRY(Gfx::Bitmap::try_load_from_file("/res/icons/16x16/save.png")));
TRY(file_menu->try_add_separator()); TRY(file_menu->try_add_separator());
TRY(file_menu->try_add_action(GUI::CommonActions::make_quit_action([&](auto&) { app->quit(); }))); TRY(file_menu->try_add_action(GUI::CommonActions::make_quit_action([&](auto&) { app->quit(); })));