mirror of
https://github.com/RGBCube/serenity
synced 2025-05-17 00:55:06 +00:00
55feecee36
Previously, the position of the mask used to calculate the new selection did not match the position of the active layer. The program would crash when trying to set a mask pixel outside the bounds of the active layer.
136 lines
4.7 KiB
C++
136 lines
4.7 KiB
C++
/*
|
|
* Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
|
|
* Copyright (c) 2022, the SerenityOS developers.
|
|
* Copyright (c) 2022, Timothy Slater <tslater2006@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "WandSelectTool.h"
|
|
#include "../ImageEditor.h"
|
|
#include "../Layer.h"
|
|
#include <AK/Queue.h>
|
|
#include <LibGUI/BoxLayout.h>
|
|
#include <LibGUI/Button.h>
|
|
#include <LibGUI/ComboBox.h>
|
|
#include <LibGUI/ItemListModel.h>
|
|
#include <LibGUI/Label.h>
|
|
#include <LibGUI/Model.h>
|
|
#include <LibGUI/Painter.h>
|
|
#include <LibGUI/ValueSlider.h>
|
|
|
|
namespace PixelPaint {
|
|
|
|
static void set_flood_selection(Gfx::Bitmap& bitmap, Image& image, Gfx::IntPoint start_position, Gfx::IntPoint selection_offset, int threshold, Selection::MergeMode merge_mode)
|
|
{
|
|
VERIFY(bitmap.bpp() == 32);
|
|
|
|
auto selection_mask = Mask::empty({ selection_offset, bitmap.size() });
|
|
|
|
auto pixel_reached = [&](Gfx::IntPoint location) {
|
|
selection_mask.set(selection_offset.x() + location.x(), selection_offset.y() + location.y(), 0xFF);
|
|
};
|
|
|
|
bitmap.flood_visit_from_point(start_position, threshold, move(pixel_reached));
|
|
|
|
selection_mask.shrink_to_fit();
|
|
image.selection().merge(selection_mask, merge_mode);
|
|
}
|
|
|
|
bool WandSelectTool::on_keydown(GUI::KeyEvent& key_event)
|
|
{
|
|
if (key_event.key() == KeyCode::Key_Escape) {
|
|
m_editor->image().selection().clear();
|
|
return true;
|
|
}
|
|
return Tool::on_keydown(key_event);
|
|
}
|
|
|
|
void WandSelectTool::on_mousedown(Layer* layer, MouseEvent& event)
|
|
{
|
|
if (!layer)
|
|
return;
|
|
|
|
auto& layer_event = event.layer_event();
|
|
if (!layer->rect().contains(layer_event.position()))
|
|
return;
|
|
|
|
auto selection_offset = layer->relative_rect().top_left();
|
|
|
|
m_editor->image().selection().begin_interactive_selection();
|
|
set_flood_selection(layer->currently_edited_bitmap(), m_editor->image(), layer_event.position(), selection_offset, m_threshold, m_merge_mode);
|
|
m_editor->image().selection().end_interactive_selection();
|
|
m_editor->update();
|
|
m_editor->did_complete_action(tool_name());
|
|
}
|
|
|
|
ErrorOr<GUI::Widget*> WandSelectTool::get_properties_widget()
|
|
{
|
|
if (m_properties_widget) {
|
|
return m_properties_widget.ptr();
|
|
}
|
|
|
|
auto properties_widget = TRY(GUI::Widget::try_create());
|
|
(void)TRY(properties_widget->try_set_layout<GUI::VerticalBoxLayout>());
|
|
|
|
auto threshold_container = TRY(properties_widget->try_add<GUI::Widget>());
|
|
threshold_container->set_fixed_height(20);
|
|
(void)TRY(threshold_container->try_set_layout<GUI::HorizontalBoxLayout>());
|
|
|
|
auto threshold_label = TRY(threshold_container->try_add<GUI::Label>("Threshold:"));
|
|
threshold_label->set_text_alignment(Gfx::TextAlignment::CenterLeft);
|
|
threshold_label->set_fixed_size(80, 20);
|
|
|
|
auto threshold_slider = TRY(threshold_container->try_add<GUI::ValueSlider>(Orientation::Horizontal, "%"_short_string));
|
|
threshold_slider->set_range(0, 100);
|
|
threshold_slider->set_value(m_threshold);
|
|
|
|
threshold_slider->on_change = [this](int value) {
|
|
m_threshold = value;
|
|
};
|
|
set_primary_slider(threshold_slider);
|
|
|
|
auto mode_container = TRY(properties_widget->try_add<GUI::Widget>());
|
|
mode_container->set_fixed_height(20);
|
|
(void)TRY(mode_container->try_set_layout<GUI::HorizontalBoxLayout>());
|
|
|
|
auto mode_label = TRY(mode_container->try_add<GUI::Label>());
|
|
mode_label->set_text("Mode:");
|
|
mode_label->set_text_alignment(Gfx::TextAlignment::CenterLeft);
|
|
mode_label->set_fixed_size(80, 20);
|
|
|
|
for (int i = 0; i < (int)Selection::MergeMode::__Count; i++) {
|
|
switch ((Selection::MergeMode)i) {
|
|
case Selection::MergeMode::Set:
|
|
TRY(m_merge_mode_names.try_append("Set"));
|
|
break;
|
|
case Selection::MergeMode::Add:
|
|
TRY(m_merge_mode_names.try_append("Add"));
|
|
break;
|
|
case Selection::MergeMode::Subtract:
|
|
TRY(m_merge_mode_names.try_append("Subtract"));
|
|
break;
|
|
case Selection::MergeMode::Intersect:
|
|
TRY(m_merge_mode_names.try_append("Intersect"));
|
|
break;
|
|
default:
|
|
VERIFY_NOT_REACHED();
|
|
}
|
|
}
|
|
|
|
auto mode_combo = TRY(mode_container->try_add<GUI::ComboBox>());
|
|
mode_combo->set_only_allow_values_from_model(true);
|
|
mode_combo->set_model(*GUI::ItemListModel<DeprecatedString>::create(m_merge_mode_names));
|
|
mode_combo->set_selected_index((int)m_merge_mode);
|
|
mode_combo->on_change = [this](auto&&, GUI::ModelIndex const& index) {
|
|
VERIFY(index.row() >= 0);
|
|
VERIFY(index.row() < (int)Selection::MergeMode::__Count);
|
|
|
|
m_merge_mode = (Selection::MergeMode)index.row();
|
|
};
|
|
|
|
m_properties_widget = properties_widget;
|
|
return m_properties_widget.ptr();
|
|
}
|
|
|
|
}
|