/* PIKA - Photo and Image Kooker Application * a rebranding of The GNU Image Manipulation Program (created with heckimp) * A derived work which may be trivial. However, any changes may be (C)2023 by Aldercone Studio * * Original copyright, applying to most contents (license remains unchanged): * Copyright (C) 1995 Spencer Kimball and Peter Mattis * * PikaForegroundSelectTool * Copyright (C) 2005 Sven Neumann * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include "config.h" #include #include #include #include #include "libpikamath/pikamath.h" #include "libpikabase/pikabase.h" #include "libpikacolor/pikacolor.h" #include "libpikawidgets/pikawidgets.h" #include "tools-types.h" #include "config/pikaguiconfig.h" #include "gegl/pika-gegl-loops.h" #include "gegl/pika-gegl-mask.h" #include "gegl/pika-gegl-utils.h" #include "core/pika.h" #include "core/pikachannel-select.h" #include "core/pikadrawable-foreground-extract.h" #include "core/pikaerror.h" #include "core/pikaimage.h" #include "core/pikalayer.h" #include "core/pikalayermask.h" #include "core/pikaprogress.h" #include "core/pikascanconvert.h" #include "widgets/pikahelp-ids.h" #include "widgets/pikawidgets-utils.h" #include "display/pikacanvasitem.h" #include "display/pikacanvasbufferpreview.h" #include "display/pikadisplay.h" #include "display/pikadisplayshell.h" #include "display/pikatoolgui.h" #include "pikaforegroundselecttool.h" #include "pikaforegroundselectoptions.h" #include "pikatoolcontrol.h" #include "pika-intl.h" #define FAR_OUTSIDE -10000 typedef struct _StrokeUndo StrokeUndo; struct _StrokeUndo { GeglBuffer *saved_trimap; gint trimap_x; gint trimap_y; PikaMattingDrawMode draw_mode; gint stroke_width; }; static void pika_foreground_select_tool_finalize (GObject *object); static gboolean pika_foreground_select_tool_initialize (PikaTool *tool, PikaDisplay *display, GError **error); static void pika_foreground_select_tool_control (PikaTool *tool, PikaToolAction action, PikaDisplay *display); static void pika_foreground_select_tool_button_press (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaButtonPressType press_type, PikaDisplay *display); static void pika_foreground_select_tool_button_release (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaButtonReleaseType release_type, PikaDisplay *display); static void pika_foreground_select_tool_motion (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaDisplay *display); static gboolean pika_foreground_select_tool_key_press (PikaTool *tool, GdkEventKey *kevent, PikaDisplay *display); static void pika_foreground_select_tool_modifier_key (PikaTool *tool, GdkModifierType key, gboolean press, GdkModifierType state, PikaDisplay *display); static void pika_foreground_select_tool_active_modifier_key (PikaTool *tool, GdkModifierType key, gboolean press, GdkModifierType state, PikaDisplay *display); static void pika_foreground_select_tool_oper_update (PikaTool *tool, const PikaCoords *coords, GdkModifierType state, gboolean proximity, PikaDisplay *display); static void pika_foreground_select_tool_cursor_update (PikaTool *tool, const PikaCoords *coords, GdkModifierType state, PikaDisplay *display); static const gchar * pika_foreground_select_tool_can_undo (PikaTool *tool, PikaDisplay *display); static const gchar * pika_foreground_select_tool_can_redo (PikaTool *tool, PikaDisplay *display); static gboolean pika_foreground_select_tool_undo (PikaTool *tool, PikaDisplay *display); static gboolean pika_foreground_select_tool_redo (PikaTool *tool, PikaDisplay *display); static void pika_foreground_select_tool_options_notify (PikaTool *tool, PikaToolOptions *options, const GParamSpec *pspec); static void pika_foreground_select_tool_draw (PikaDrawTool *draw_tool); static void pika_foreground_select_tool_confirm (PikaPolygonSelectTool *poly_sel, PikaDisplay *display); static void pika_foreground_select_tool_halt (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_commit (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_set_trimap (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_set_preview (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_preview (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_stroke_paint (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_cancel_paint (PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_response (PikaToolGui *gui, gint response_id, PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_preview_toggled(GtkToggleButton *button, PikaForegroundSelectTool *fg_select); static void pika_foreground_select_tool_update_gui (PikaForegroundSelectTool *fg_select); static StrokeUndo * pika_foreground_select_undo_new (GeglBuffer *trimap, GArray *stroke, PikaMattingDrawMode draw_mode, gint stroke_width); static void pika_foreground_select_undo_pop (StrokeUndo *undo, GeglBuffer *trimap); static void pika_foreground_select_undo_free (StrokeUndo *undo); G_DEFINE_TYPE (PikaForegroundSelectTool, pika_foreground_select_tool, PIKA_TYPE_POLYGON_SELECT_TOOL) #define parent_class pika_foreground_select_tool_parent_class void pika_foreground_select_tool_register (PikaToolRegisterCallback callback, gpointer data) { (* callback) (PIKA_TYPE_FOREGROUND_SELECT_TOOL, PIKA_TYPE_FOREGROUND_SELECT_OPTIONS, pika_foreground_select_options_gui, PIKA_CONTEXT_PROP_MASK_FOREGROUND | PIKA_CONTEXT_PROP_MASK_BACKGROUND, "pika-foreground-select-tool", _("Foreground Select"), _("Foreground Select Tool: Select a region containing foreground objects"), N_("F_oreground Select"), NULL, NULL, PIKA_HELP_TOOL_FOREGROUND_SELECT, PIKA_ICON_TOOL_FOREGROUND_SELECT, data); } static void pika_foreground_select_tool_class_init (PikaForegroundSelectToolClass *klass) { GObjectClass *object_class = G_OBJECT_CLASS (klass); PikaToolClass *tool_class = PIKA_TOOL_CLASS (klass); PikaDrawToolClass *draw_tool_class = PIKA_DRAW_TOOL_CLASS (klass); PikaPolygonSelectToolClass *polygon_select_tool_class; polygon_select_tool_class = PIKA_POLYGON_SELECT_TOOL_CLASS (klass); object_class->finalize = pika_foreground_select_tool_finalize; tool_class->initialize = pika_foreground_select_tool_initialize; tool_class->control = pika_foreground_select_tool_control; tool_class->button_press = pika_foreground_select_tool_button_press; tool_class->button_release = pika_foreground_select_tool_button_release; tool_class->motion = pika_foreground_select_tool_motion; tool_class->key_press = pika_foreground_select_tool_key_press; tool_class->modifier_key = pika_foreground_select_tool_modifier_key; tool_class->active_modifier_key = pika_foreground_select_tool_active_modifier_key; tool_class->oper_update = pika_foreground_select_tool_oper_update; tool_class->cursor_update = pika_foreground_select_tool_cursor_update; tool_class->can_undo = pika_foreground_select_tool_can_undo; tool_class->can_redo = pika_foreground_select_tool_can_redo; tool_class->undo = pika_foreground_select_tool_undo; tool_class->redo = pika_foreground_select_tool_redo; tool_class->options_notify = pika_foreground_select_tool_options_notify; draw_tool_class->draw = pika_foreground_select_tool_draw; polygon_select_tool_class->confirm = pika_foreground_select_tool_confirm; } static void pika_foreground_select_tool_init (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); pika_tool_control_set_motion_mode (tool->control, PIKA_MOTION_MODE_EXACT); pika_tool_control_set_scroll_lock (tool->control, FALSE); pika_tool_control_set_preserve (tool->control, FALSE); pika_tool_control_set_dirty_mask (tool->control, PIKA_DIRTY_IMAGE_SIZE | PIKA_DIRTY_ACTIVE_DRAWABLE); pika_tool_control_set_precision (tool->control, PIKA_CURSOR_PRECISION_SUBPIXEL); pika_tool_control_set_tool_cursor (tool->control, PIKA_TOOL_CURSOR_FREE_SELECT); pika_tool_control_set_action_size (tool->control, "tools-foreground-select-brush-size-set"); fg_select->state = MATTING_STATE_FREE_SELECT; fg_select->grayscale_preview = NULL; } static void pika_foreground_select_tool_finalize (GObject *object) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (object); g_clear_object (&fg_select->gui); fg_select->preview_toggle = NULL; if (fg_select->stroke) g_warning ("%s: stroke should be NULL at this point", G_STRLOC); if (fg_select->mask) g_warning ("%s: mask should be NULL at this point", G_STRLOC); if (fg_select->trimap) g_warning ("%s: mask should be NULL at this point", G_STRLOC); G_OBJECT_CLASS (parent_class)->finalize (object); } static gboolean pika_foreground_select_tool_initialize (PikaTool *tool, PikaDisplay *display, GError **error) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); PikaGuiConfig *config = PIKA_GUI_CONFIG (display->pika->config); PikaImage *image = pika_display_get_image (display); PikaDisplayShell *shell = pika_display_get_shell (display); GList *drawables = pika_image_get_selected_drawables (image); PikaDrawable *drawable; if (g_list_length (drawables) != 1) { if (g_list_length (drawables) > 1) g_set_error_literal (error, PIKA_ERROR, PIKA_FAILED, _("Cannot select from multiple layers.")); else g_set_error_literal (error, PIKA_ERROR, PIKA_FAILED, _("No selected drawables.")); g_list_free (drawables); return FALSE; } drawable = drawables->data; g_list_free (drawables); if (! pika_item_is_visible (PIKA_ITEM (drawable)) && ! config->edit_non_visible) { g_set_error_literal (error, PIKA_ERROR, PIKA_FAILED, _("The active layer is not visible.")); return FALSE; } tool->display = display; /* enable double click for the FreeSelectTool, because it may have been * disabled if the tool has switched to MATTING_STATE_PAINT_TRIMAP, * in pika_foreground_select_tool_set_trimap(). */ pika_tool_control_set_wants_double_click (tool->control, TRUE); fg_select->state = MATTING_STATE_FREE_SELECT; if (! fg_select->gui) { fg_select->gui = pika_tool_gui_new (tool->tool_info, NULL, _("Dialog for foreground select"), NULL, NULL, pika_widget_get_monitor (GTK_WIDGET (shell)), TRUE, _("_Cancel"), GTK_RESPONSE_CANCEL, _("_Select"), GTK_RESPONSE_APPLY, NULL); pika_tool_gui_set_auto_overlay (fg_select->gui, TRUE); g_signal_connect (fg_select->gui, "response", G_CALLBACK (pika_foreground_select_tool_response), fg_select); fg_select->preview_toggle = gtk_check_button_new_with_mnemonic (_("_Preview mask")); gtk_box_pack_start (GTK_BOX (pika_tool_gui_get_vbox (fg_select->gui)), fg_select->preview_toggle, FALSE, FALSE, 0); gtk_widget_show (fg_select->preview_toggle); g_signal_connect (fg_select->preview_toggle, "toggled", G_CALLBACK (pika_foreground_select_tool_preview_toggled), fg_select); } pika_tool_gui_set_description (fg_select->gui, _("Select foreground pixels")); pika_tool_gui_set_response_sensitive (fg_select->gui, GTK_RESPONSE_APPLY, FALSE); gtk_widget_set_sensitive (fg_select->preview_toggle, FALSE); gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (fg_select->preview_toggle), FALSE); pika_tool_gui_set_shell (fg_select->gui, shell); pika_tool_gui_set_viewable (fg_select->gui, PIKA_VIEWABLE (drawable)); pika_tool_gui_show (fg_select->gui); return TRUE; } static void pika_foreground_select_tool_control (PikaTool *tool, PikaToolAction action, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); switch (action) { case PIKA_TOOL_ACTION_PAUSE: case PIKA_TOOL_ACTION_RESUME: break; case PIKA_TOOL_ACTION_HALT: pika_foreground_select_tool_halt (fg_select); break; case PIKA_TOOL_ACTION_COMMIT: pika_foreground_select_tool_commit (fg_select); break; } PIKA_TOOL_CLASS (parent_class)->control (tool, action, display); } static void pika_foreground_select_tool_button_press (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaButtonPressType press_type, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); PikaDrawTool *draw_tool = PIKA_DRAW_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_TOOL_CLASS (parent_class)->button_press (tool, coords, time, state, press_type, display); } else { PikaVector2 point = pika_vector2_new (coords->x, coords->y); pika_draw_tool_pause (draw_tool); if (pika_draw_tool_is_active (draw_tool) && draw_tool->display != display) pika_draw_tool_stop (draw_tool); pika_tool_control_activate (tool->control); fg_select->last_coords = *coords; g_return_if_fail (fg_select->stroke == NULL); fg_select->stroke = g_array_new (FALSE, FALSE, sizeof (PikaVector2)); g_array_append_val (fg_select->stroke, point); if (! pika_draw_tool_is_active (draw_tool)) pika_draw_tool_start (draw_tool, display); pika_draw_tool_resume (draw_tool); } } static void pika_foreground_select_tool_button_release (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaButtonReleaseType release_type, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_TOOL_CLASS (parent_class)->button_release (tool, coords, time, state, release_type, display); } else { pika_draw_tool_pause (PIKA_DRAW_TOOL (tool)); pika_tool_control_halt (tool->control); if (release_type == PIKA_BUTTON_RELEASE_CANCEL) { pika_foreground_select_tool_cancel_paint (fg_select); } else { pika_foreground_select_tool_stroke_paint (fg_select); if (fg_select->state == MATTING_STATE_PREVIEW_MASK) pika_foreground_select_tool_preview (fg_select); else pika_foreground_select_tool_set_trimap (fg_select); } pika_draw_tool_resume (PIKA_DRAW_TOOL (tool)); } } static void pika_foreground_select_tool_motion (PikaTool *tool, const PikaCoords *coords, guint32 time, GdkModifierType state, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_TOOL_CLASS (parent_class)->motion (tool, coords, time, state, display); } else { PikaVector2 *last = &g_array_index (fg_select->stroke, PikaVector2, fg_select->stroke->len - 1); pika_draw_tool_pause (PIKA_DRAW_TOOL (tool)); fg_select->last_coords = *coords; if (last->x != (gint) coords->x || last->y != (gint) coords->y) { PikaVector2 point = pika_vector2_new (coords->x, coords->y); g_array_append_val (fg_select->stroke, point); } pika_draw_tool_resume (PIKA_DRAW_TOOL (tool)); } } static gboolean pika_foreground_select_tool_key_press (PikaTool *tool, GdkEventKey *kevent, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { return PIKA_TOOL_CLASS (parent_class)->key_press (tool, kevent, display); } else { if (display != tool->display) return FALSE; switch (kevent->keyval) { case GDK_KEY_Return: case GDK_KEY_KP_Enter: case GDK_KEY_ISO_Enter: if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (fg_select->preview_toggle), TRUE); else pika_foreground_select_tool_response (fg_select->gui, GTK_RESPONSE_APPLY, fg_select); return TRUE; case GDK_KEY_Escape: if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) pika_foreground_select_tool_response (fg_select->gui, GTK_RESPONSE_CANCEL, fg_select); else gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (fg_select->preview_toggle), FALSE); return TRUE; default: return FALSE; } } } static void pika_foreground_select_tool_modifier_key (PikaTool *tool, GdkModifierType key, gboolean press, GdkModifierType state, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_TOOL_CLASS (parent_class)->modifier_key (tool, key, press, state, display); } else { #if 0 if (key == pika_get_toggle_behavior_mask ()) { PikaForegroundSelectOptions *options; options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (tool); g_object_set (options, "background", ! options->background, NULL); } #endif } } static void pika_foreground_select_tool_active_modifier_key (PikaTool *tool, GdkModifierType key, gboolean press, GdkModifierType state, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_TOOL_CLASS (parent_class)->active_modifier_key (tool, key, press, state, display); } } static void pika_foreground_select_tool_oper_update (PikaTool *tool, const PikaCoords *coords, GdkModifierType state, gboolean proximity, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); PikaForegroundSelectOptions *options; const gchar *status_stage = NULL; const gchar *status_mode = NULL; options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (fg_select); PIKA_TOOL_CLASS (parent_class)->oper_update (tool, coords, state, proximity, display); if (fg_select->state == MATTING_STATE_FREE_SELECT) { if (PIKA_SELECTION_TOOL (tool)->function == SELECTION_SELECT) { gint n_points; pika_polygon_select_tool_get_points (PIKA_POLYGON_SELECT_TOOL (tool), NULL, &n_points); if (n_points > 2) { status_mode = _("Roughly outline the object to extract"); status_stage = _("press Enter to refine."); } else { status_stage = _("Roughly outline the object to extract"); } } } else { PikaDrawTool *draw_tool = PIKA_DRAW_TOOL (tool); pika_draw_tool_pause (draw_tool); if (proximity) { fg_select->last_coords = *coords; } else { fg_select->last_coords.x = FAR_OUTSIDE; fg_select->last_coords.y = FAR_OUTSIDE; } pika_draw_tool_resume (draw_tool); if (options->draw_mode == PIKA_MATTING_DRAW_MODE_FOREGROUND) status_mode = _("Selecting foreground"); else if (options->draw_mode == PIKA_MATTING_DRAW_MODE_BACKGROUND) status_mode = _("Selecting background"); else status_mode = _("Selecting unknown"); if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) status_stage = _("press Enter to preview."); else status_stage = _("press Escape to exit preview or Enter to apply."); } if (proximity && status_stage) { if (status_mode) pika_tool_replace_status (tool, display, "%s, %s", status_mode, status_stage); else pika_tool_replace_status (tool, display, "%s", status_stage); } } static void pika_foreground_select_tool_cursor_update (PikaTool *tool, const PikaCoords *coords, GdkModifierType state, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) { switch (PIKA_SELECTION_TOOL (tool)->function) { case SELECTION_MOVE_MASK: case SELECTION_MOVE: case SELECTION_MOVE_COPY: case SELECTION_ANCHOR: return; default: break; } } PIKA_TOOL_CLASS (parent_class)->cursor_update (tool, coords, state, display); } static const gchar * pika_foreground_select_tool_can_undo (PikaTool *tool, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->undo_stack) { StrokeUndo *undo = fg_select->undo_stack->data; const gchar *desc; if (pika_enum_get_value (PIKA_TYPE_MATTING_DRAW_MODE, undo->draw_mode, NULL, NULL, &desc, NULL)) { return desc; } } return NULL; } static const gchar * pika_foreground_select_tool_can_redo (PikaTool *tool, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); if (fg_select->redo_stack) { StrokeUndo *undo = fg_select->redo_stack->data; const gchar *desc; if (pika_enum_get_value (PIKA_TYPE_MATTING_DRAW_MODE, undo->draw_mode, NULL, NULL, &desc, NULL)) { return desc; } } return NULL; } static gboolean pika_foreground_select_tool_undo (PikaTool *tool, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); StrokeUndo *undo = fg_select->undo_stack->data; pika_foreground_select_undo_pop (undo, fg_select->trimap); fg_select->undo_stack = g_list_remove (fg_select->undo_stack, undo); fg_select->redo_stack = g_list_prepend (fg_select->redo_stack, undo); if (fg_select->state == MATTING_STATE_PREVIEW_MASK) pika_foreground_select_tool_preview (fg_select); else pika_foreground_select_tool_set_trimap (fg_select); return TRUE; } static gboolean pika_foreground_select_tool_redo (PikaTool *tool, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); StrokeUndo *undo = fg_select->redo_stack->data; pika_foreground_select_undo_pop (undo, fg_select->trimap); fg_select->redo_stack = g_list_remove (fg_select->redo_stack, undo); fg_select->undo_stack = g_list_prepend (fg_select->undo_stack, undo); if (fg_select->state == MATTING_STATE_PREVIEW_MASK) pika_foreground_select_tool_preview (fg_select); else pika_foreground_select_tool_set_trimap (fg_select); return TRUE; } static void pika_foreground_select_tool_options_notify (PikaTool *tool, PikaToolOptions *options, const GParamSpec *pspec) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); PikaForegroundSelectOptions *fg_options; fg_options = PIKA_FOREGROUND_SELECT_OPTIONS (options); if (! tool->display) return; if (! strcmp (pspec->name, "mask-color") || ! strcmp (pspec->name, "preview-mode")) { if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) { pika_foreground_select_tool_set_trimap (fg_select); } else if (fg_select->state == MATTING_STATE_PREVIEW_MASK) { pika_foreground_select_tool_set_preview (fg_select); } } else if (! strcmp (pspec->name, "engine")) { if (fg_select->state == MATTING_STATE_PREVIEW_MASK) { pika_foreground_select_tool_preview (fg_select); } } else if (! strcmp (pspec->name, "iterations")) { if (fg_options->engine == PIKA_MATTING_ENGINE_GLOBAL && fg_select->state == MATTING_STATE_PREVIEW_MASK) { pika_foreground_select_tool_preview (fg_select); } } else if (! strcmp (pspec->name, "levels") || ! strcmp (pspec->name, "active-levels")) { if (fg_options->engine == PIKA_MATTING_ENGINE_LEVIN && fg_select->state == MATTING_STATE_PREVIEW_MASK) { pika_foreground_select_tool_preview (fg_select); } } } static void pika_foreground_select_tool_get_area (GeglBuffer *mask, gint *x1, gint *y1, gint *x2, gint *y2) { gint width; gint height; pika_gegl_mask_bounds (mask, x1, y1, x2, y2); width = *x2 - *x1; height = *y2 - *y1; *x1 = MAX (*x1 - width / 2, 0); *y1 = MAX (*y1 - height / 2, 0); *x2 = MIN (*x2 + width / 2, pika_item_get_width (PIKA_ITEM (mask))); *y2 = MIN (*y2 + height / 2, pika_item_get_height (PIKA_ITEM (mask))); } static void pika_foreground_select_tool_draw (PikaDrawTool *draw_tool) { PikaTool *tool = PIKA_TOOL (draw_tool); PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (tool); PikaForegroundSelectOptions *options; options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (tool); if (fg_select->state == MATTING_STATE_FREE_SELECT) { PIKA_DRAW_TOOL_CLASS (parent_class)->draw (draw_tool); return; } else { gint x = fg_select->last_coords.x; gint y = fg_select->last_coords.y; gdouble radius = options->stroke_width / 2.0f; if (fg_select->stroke) { PikaDisplayShell *shell = pika_display_get_shell (draw_tool->display); pika_draw_tool_add_pen (draw_tool, (const PikaVector2 *) fg_select->stroke->data, fg_select->stroke->len, PIKA_CONTEXT (options), PIKA_ACTIVE_COLOR_FOREGROUND, options->stroke_width * shell->scale_y); } /* warn if the user is drawing outside of the working area */ if (FALSE) { gint x1, y1; gint x2, y2; pika_foreground_select_tool_get_area (fg_select->mask, &x1, &y1, &x2, &y2); if (x < x1 + radius || x > x2 - radius || y < y1 + radius || y > y2 - radius) { pika_draw_tool_add_rectangle (draw_tool, FALSE, x1, y1, x2 - x1, y2 - y1); } } if (x > FAR_OUTSIDE && y > FAR_OUTSIDE) pika_draw_tool_add_arc (draw_tool, FALSE, x - radius, y - radius, 2 * radius, 2 * radius, 0.0, 2.0 * G_PI); if (fg_select->grayscale_preview) pika_draw_tool_add_preview (draw_tool, fg_select->grayscale_preview); } } static void pika_foreground_select_tool_confirm (PikaPolygonSelectTool *poly_sel, PikaDisplay *display) { PikaForegroundSelectTool *fg_select = PIKA_FOREGROUND_SELECT_TOOL (poly_sel); PikaImage *image = pika_display_get_image (display); GList *drawables = pika_image_get_selected_drawables (image); PikaDrawable *drawable; PikaItem *item; g_return_if_fail (g_list_length (drawables) == 1); drawable = drawables->data; item = PIKA_ITEM (drawable); g_list_free (drawables); if (drawable && fg_select->state == MATTING_STATE_FREE_SELECT) { PikaScanConvert *scan_convert = pika_scan_convert_new (); const PikaVector2 *points; gint n_points; pika_polygon_select_tool_get_points (poly_sel, &points, &n_points); pika_scan_convert_add_polyline (scan_convert, n_points, points, TRUE); fg_select->trimap = gegl_buffer_new (GEGL_RECTANGLE (pika_item_get_offset_x (item), pika_item_get_offset_y (item), pika_item_get_width (item), pika_item_get_height (item)), pika_image_get_mask_format (image)); pika_scan_convert_render_value (scan_convert, fg_select->trimap, 0, 0, 0.5); pika_scan_convert_free (scan_convert); fg_select->grayscale_preview = pika_canvas_buffer_preview_new (pika_display_get_shell (display), fg_select->trimap); pika_foreground_select_tool_set_trimap (fg_select); } } static void pika_foreground_select_tool_halt (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); PikaDrawTool *draw_tool = PIKA_DRAW_TOOL (fg_select); if (draw_tool->preview) { pika_draw_tool_remove_preview (draw_tool, fg_select->grayscale_preview); } g_clear_object (&fg_select->grayscale_preview); g_clear_object (&fg_select->trimap); g_clear_object (&fg_select->mask); if (fg_select->undo_stack) { g_list_free_full (fg_select->undo_stack, (GDestroyNotify) pika_foreground_select_undo_free); fg_select->undo_stack = NULL; } if (fg_select->redo_stack) { g_list_free_full (fg_select->redo_stack, (GDestroyNotify) pika_foreground_select_undo_free); fg_select->redo_stack = NULL; } if (tool->display) pika_display_shell_set_mask (pika_display_get_shell (tool->display), NULL, 0, 0, NULL, FALSE); pika_tool_control_set_tool_cursor (tool->control, PIKA_TOOL_CURSOR_FREE_SELECT); pika_tool_control_set_toggle_tool_cursor (tool->control, PIKA_TOOL_CURSOR_FREE_SELECT); pika_tool_control_set_toggled (tool->control, FALSE); /* set precision to SUBPIXEL, because it may have been changed to * PIXEL_CENTER if the tool has switched to MATTING_STATE_PAINT_TRIMAP, * in pika_foreground_select_tool_set_trimap(). */ pika_tool_control_set_precision (tool->control, PIKA_CURSOR_PRECISION_SUBPIXEL); fg_select->state = MATTING_STATE_FREE_SELECT; /* update the undo actions / menu items */ if (tool->display) pika_image_flush (pika_display_get_image (tool->display)); tool->display = NULL; g_list_free (tool->drawables); tool->drawables = NULL; if (fg_select->gui) pika_tool_gui_hide (fg_select->gui); } static void pika_foreground_select_tool_commit (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); PikaSelectionOptions *options = PIKA_SELECTION_TOOL_GET_OPTIONS (fg_select); if (tool->display && fg_select->state != MATTING_STATE_FREE_SELECT) { PikaImage *image = pika_display_get_image (tool->display); if (fg_select->state != MATTING_STATE_PREVIEW_MASK) pika_foreground_select_tool_preview (fg_select); pika_channel_select_buffer (pika_image_get_mask (image), C_("command", "Foreground Select"), fg_select->mask, 0, /* x offset */ 0, /* y offset */ options->operation, options->feather, options->feather_radius, options->feather_radius); pika_image_flush (image); } } static void pika_foreground_select_tool_set_trimap (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); PikaForegroundSelectOptions *options; g_return_if_fail (fg_select->trimap != NULL); options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (tool); pika_polygon_select_tool_halt (PIKA_POLYGON_SELECT_TOOL (fg_select)); if (options->preview_mode == PIKA_MATTING_PREVIEW_MODE_ON_COLOR) { if (fg_select->grayscale_preview) pika_canvas_item_set_visible (fg_select->grayscale_preview, FALSE); pika_display_shell_set_mask (pika_display_get_shell (tool->display), fg_select->trimap, 0, 0, &options->mask_color, TRUE); } else { pika_display_shell_set_mask (pika_display_get_shell (tool->display), NULL, 0, 0, NULL, FALSE); if (fg_select->grayscale_preview) { g_object_set (fg_select->grayscale_preview, "buffer", fg_select->trimap, NULL); pika_canvas_item_set_visible (fg_select->grayscale_preview, TRUE); } } pika_tool_control_set_tool_cursor (tool->control, PIKA_TOOL_CURSOR_PAINTBRUSH); pika_tool_control_set_toggle_tool_cursor (tool->control, PIKA_TOOL_CURSOR_PAINTBRUSH); pika_tool_control_set_toggled (tool->control, FALSE); /* disable double click in paint trimap state */ pika_tool_control_set_wants_double_click (tool->control, FALSE); /* set precision to PIXEL_CENTER in paint trimap state */ pika_tool_control_set_precision (tool->control, PIKA_CURSOR_PRECISION_PIXEL_CENTER); fg_select->state = MATTING_STATE_PAINT_TRIMAP; pika_foreground_select_tool_update_gui (fg_select); } static void pika_foreground_select_tool_set_preview (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); PikaForegroundSelectOptions *options; g_return_if_fail (fg_select->mask != NULL); options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (tool); if (options->preview_mode == PIKA_MATTING_PREVIEW_MODE_ON_COLOR) { if (fg_select->grayscale_preview) pika_canvas_item_set_visible (fg_select->grayscale_preview, FALSE); pika_display_shell_set_mask (pika_display_get_shell (tool->display), fg_select->mask, 0, 0, &options->mask_color, TRUE); } else { pika_display_shell_set_mask (pika_display_get_shell (tool->display), NULL, 0, 0, NULL, FALSE); if (fg_select->grayscale_preview) { g_object_set (fg_select->grayscale_preview, "buffer", fg_select->mask, NULL); pika_canvas_item_set_visible (fg_select->grayscale_preview, TRUE); } } pika_tool_control_set_tool_cursor (tool->control, PIKA_TOOL_CURSOR_PAINTBRUSH); pika_tool_control_set_toggle_tool_cursor (tool->control, PIKA_TOOL_CURSOR_PAINTBRUSH); pika_tool_control_set_toggled (tool->control, FALSE); fg_select->state = MATTING_STATE_PREVIEW_MASK; pika_foreground_select_tool_update_gui (fg_select); } static void pika_foreground_select_tool_preview (PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); PikaForegroundSelectOptions *options; PikaImage *image = pika_display_get_image (tool->display); GList *drawables = pika_image_get_selected_drawables (image); PikaDrawable *drawable; g_return_if_fail (g_list_length (drawables) == 1); drawable = drawables->data; g_list_free (drawables); options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (tool); g_clear_object (&fg_select->mask); fg_select->mask = pika_drawable_foreground_extract (drawable, options->engine, options->iterations, options->levels, options->active_levels, fg_select->trimap, PIKA_PROGRESS (fg_select)); pika_foreground_select_tool_set_preview (fg_select); } static void pika_foreground_select_tool_stroke_paint (PikaForegroundSelectTool *fg_select) { PikaForegroundSelectOptions *options; PikaScanConvert *scan_convert; StrokeUndo *undo; gint width; gdouble opacity; options = PIKA_FOREGROUND_SELECT_TOOL_GET_OPTIONS (fg_select); g_return_if_fail (fg_select->stroke != NULL); width = ROUND ((gdouble) options->stroke_width); if (fg_select->redo_stack) { g_list_free_full (fg_select->redo_stack, (GDestroyNotify) pika_foreground_select_undo_free); fg_select->redo_stack = NULL; } undo = pika_foreground_select_undo_new (fg_select->trimap, fg_select->stroke, options->draw_mode, width); if (! undo) { g_array_free (fg_select->stroke, TRUE); fg_select->stroke = NULL; return; } fg_select->undo_stack = g_list_prepend (fg_select->undo_stack, undo); scan_convert = pika_scan_convert_new (); if (fg_select->stroke->len == 1) { PikaVector2 points[2]; points[0] = points[1] = ((PikaVector2 *) fg_select->stroke->data)[0]; points[1].x += 0.01; points[1].y += 0.01; pika_scan_convert_add_polyline (scan_convert, 2, points, FALSE); } else { pika_scan_convert_add_polyline (scan_convert, fg_select->stroke->len, (PikaVector2 *) fg_select->stroke->data, FALSE); } pika_scan_convert_stroke (scan_convert, width, PIKA_JOIN_ROUND, PIKA_CAP_ROUND, 10.0, 0.0, NULL); if (options->draw_mode == PIKA_MATTING_DRAW_MODE_FOREGROUND) opacity = 1.0; else if (options->draw_mode == PIKA_MATTING_DRAW_MODE_BACKGROUND) opacity = 0.0; else opacity = 0.5; pika_scan_convert_compose_value (scan_convert, fg_select->trimap, 0, 0, opacity); pika_scan_convert_free (scan_convert); g_array_free (fg_select->stroke, TRUE); fg_select->stroke = NULL; /* update the undo actions / menu items */ pika_image_flush (pika_display_get_image (PIKA_TOOL (fg_select)->display)); } static void pika_foreground_select_tool_cancel_paint (PikaForegroundSelectTool *fg_select) { g_return_if_fail (fg_select->stroke != NULL); g_array_free (fg_select->stroke, TRUE); fg_select->stroke = NULL; } static void pika_foreground_select_tool_response (PikaToolGui *gui, gint response_id, PikaForegroundSelectTool *fg_select) { PikaTool *tool = PIKA_TOOL (fg_select); switch (response_id) { case GTK_RESPONSE_APPLY: pika_tool_control (tool, PIKA_TOOL_ACTION_COMMIT, tool->display); break; default: pika_tool_control (tool, PIKA_TOOL_ACTION_HALT, tool->display); break; } } static void pika_foreground_select_tool_preview_toggled (GtkToggleButton *button, PikaForegroundSelectTool *fg_select) { if (fg_select->state != MATTING_STATE_FREE_SELECT) { if (gtk_toggle_button_get_active (button)) { if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) pika_foreground_select_tool_preview (fg_select); } else { if (fg_select->state == MATTING_STATE_PREVIEW_MASK) pika_foreground_select_tool_set_trimap (fg_select); } } } static void pika_foreground_select_tool_update_gui (PikaForegroundSelectTool *fg_select) { if (fg_select->state == MATTING_STATE_PAINT_TRIMAP) { pika_tool_gui_set_description (fg_select->gui, _("Paint mask")); } else if (fg_select->state == MATTING_STATE_PREVIEW_MASK) { pika_tool_gui_set_description (fg_select->gui, _("Preview")); } pika_tool_gui_set_response_sensitive (fg_select->gui, GTK_RESPONSE_APPLY, TRUE); gtk_widget_set_sensitive (fg_select->preview_toggle, TRUE); } static StrokeUndo * pika_foreground_select_undo_new (GeglBuffer *trimap, GArray *stroke, PikaMattingDrawMode draw_mode, gint stroke_width) { StrokeUndo *undo; const GeglRectangle *extent; gint x1, y1, x2, y2; gint width, height; gint i; extent = gegl_buffer_get_extent (trimap); x1 = G_MAXINT; y1 = G_MAXINT; x2 = G_MININT; y2 = G_MININT; for (i = 0; i < stroke->len; i++) { PikaVector2 *point = &g_array_index (stroke, PikaVector2, i); x1 = MIN (x1, floor (point->x)); y1 = MIN (y1, floor (point->y)); x2 = MAX (x2, ceil (point->x)); y2 = MAX (y2, ceil (point->y)); } x1 -= (stroke_width + 1) / 2; y1 -= (stroke_width + 1) / 2; x2 += (stroke_width + 1) / 2; y2 += (stroke_width + 1) / 2; x1 = MAX (x1, extent->x); y1 = MAX (y1, extent->y); x2 = MIN (x2, extent->x + extent->width); y2 = MIN (y2, extent->x + extent->height); width = x2 - x1; height = y2 - y1; if (width <= 0 || height <= 0) return NULL; undo = g_slice_new0 (StrokeUndo); undo->saved_trimap = gegl_buffer_new (GEGL_RECTANGLE (x1, y1, width, height), gegl_buffer_get_format (trimap)); pika_gegl_buffer_copy ( trimap, GEGL_RECTANGLE (x1, y1, width, height), GEGL_ABYSS_NONE, undo->saved_trimap, NULL); undo->trimap_x = x1; undo->trimap_y = y1; undo->draw_mode = draw_mode; undo->stroke_width = stroke_width; return undo; } static void pika_foreground_select_undo_pop (StrokeUndo *undo, GeglBuffer *trimap) { GeglBuffer *buffer; gint width, height; buffer = pika_gegl_buffer_dup (undo->saved_trimap); width = gegl_buffer_get_width (buffer); height = gegl_buffer_get_height (buffer); pika_gegl_buffer_copy (trimap, GEGL_RECTANGLE (undo->trimap_x, undo->trimap_y, width, height), GEGL_ABYSS_NONE, undo->saved_trimap, NULL); pika_gegl_buffer_copy (buffer, GEGL_RECTANGLE (undo->trimap_x, undo->trimap_y, width, height), GEGL_ABYSS_NONE, trimap, NULL); g_object_unref (buffer); } static void pika_foreground_select_undo_free (StrokeUndo *undo) { g_clear_object (&undo->saved_trimap); g_slice_free (StrokeUndo, undo); }