serenity/Userland/Applications/PixelPaint/Tools/WandSelectTool.cpp

/*
 * 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::IntRect layer_rect, int threshold, Selection::MergeMode merge_mode)
{
    VERIFY(bitmap.bpp() == 32);

    auto image_rect = image.rect();
    auto mask_rect = layer_rect.intersected(image_rect);
    auto selection_mask = Mask::empty(mask_rect);

    auto pixel_reached = [&](Gfx::IntPoint location) {
        auto point_to_set = layer_rect.top_left() + location;
        if (selection_mask.bounding_rect().contains(point_to_set))
            selection_mask.set(point_to_set, 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;

    m_editor->image().selection().begin_interactive_selection();
    set_flood_selection(layer->currently_edited_bitmap(), m_editor->image(), layer_event.position(), layer->relative_rect(), 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:"_string));
    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, "%"_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:"_string);
    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();
}

}