#define __SP_FLOOD_CONTEXT_C__ /* * Flood fill drawing context * * Author: * Lauris Kaplinski * bulia byak * John Bintz * * Copyright (C) 2006 Johan Engelen * Copyright (C) 2000-2005 authors * Copyright (C) 2000-2001 Ximian, Inc. * * Released under GNU GPL, read the file 'COPYING' for more information */ #include "config.h" #include #include #include "macros.h" #include "display/sp-canvas.h" #include "document.h" #include "sp-namedview.h" #include "sp-object.h" #include "sp-rect.h" #include "selection.h" #include "selection-chemistry.h" #include "desktop-handles.h" #include "snap.h" #include "desktop.h" #include "desktop-style.h" #include "message-stack.h" #include "message-context.h" #include "pixmaps/cursor-paintbucket.xpm" #include "flood-context.h" #include "sp-metrics.h" #include #include "object-edit.h" #include "xml/repr.h" #include "xml/node-event-vector.h" #include "prefs-utils.h" #include "context-fns.h" #include "display/nr-arena-item.h" #include "display/nr-arena.h" #include "display/nr-arena-image.h" #include "display/canvas-arena.h" #include "libnr/nr-pixops.h" #include "libnr/nr-matrix-rotate-ops.h" #include "libnr/nr-matrix-translate-ops.h" #include "libnr/nr-rotate-fns.h" #include "libnr/nr-scale-ops.h" #include "libnr/nr-scale-translate-ops.h" #include "libnr/nr-translate-matrix-ops.h" #include "libnr/nr-translate-scale-ops.h" #include "libnr/nr-matrix-ops.h" #include "sp-item.h" #include "sp-root.h" #include "sp-defs.h" #include "splivarot.h" #include "livarot/Path.h" #include "livarot/Shape.h" #include "libnr/n-art-bpath.h" #include "svg/svg.h" #include "trace/trace.h" #include "trace/potrace/inkscape-potrace.h" static void sp_flood_context_class_init(SPFloodContextClass *klass); static void sp_flood_context_init(SPFloodContext *flood_context); static void sp_flood_context_dispose(GObject *object); static void sp_flood_context_setup(SPEventContext *ec); static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event); static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event); static void sp_flood_finish(SPFloodContext *rc); static SPEventContextClass *parent_class; GtkType sp_flood_context_get_type() { static GType type = 0; if (!type) { GTypeInfo info = { sizeof(SPFloodContextClass), NULL, NULL, (GClassInitFunc) sp_flood_context_class_init, NULL, NULL, sizeof(SPFloodContext), 4, (GInstanceInitFunc) sp_flood_context_init, NULL, /* value_table */ }; type = g_type_register_static(SP_TYPE_EVENT_CONTEXT, "SPFloodContext", &info, (GTypeFlags) 0); } return type; } static void sp_flood_context_class_init(SPFloodContextClass *klass) { GObjectClass *object_class = (GObjectClass *) klass; SPEventContextClass *event_context_class = (SPEventContextClass *) klass; parent_class = (SPEventContextClass *) g_type_class_peek_parent(klass); object_class->dispose = sp_flood_context_dispose; event_context_class->setup = sp_flood_context_setup; event_context_class->root_handler = sp_flood_context_root_handler; event_context_class->item_handler = sp_flood_context_item_handler; } static void sp_flood_context_init(SPFloodContext *flood_context) { SPEventContext *event_context = SP_EVENT_CONTEXT(flood_context); event_context->cursor_shape = cursor_paintbucket_xpm; event_context->hot_x = 11; event_context->hot_y = 30; event_context->xp = 0; event_context->yp = 0; event_context->tolerance = 0; event_context->within_tolerance = false; event_context->item_to_select = NULL; event_context->shape_repr = NULL; event_context->shape_knot_holder = NULL; flood_context->item = NULL; new (&flood_context->sel_changed_connection) sigc::connection(); } static void sp_flood_context_dispose(GObject *object) { SPFloodContext *rc = SP_FLOOD_CONTEXT(object); SPEventContext *ec = SP_EVENT_CONTEXT(object); rc->sel_changed_connection.disconnect(); rc->sel_changed_connection.~connection(); /* fixme: This is necessary because we do not grab */ if (rc->item) { sp_flood_finish(rc); } if (ec->shape_repr) { // remove old listener sp_repr_remove_listener_by_data(ec->shape_repr, ec); Inkscape::GC::release(ec->shape_repr); ec->shape_repr = 0; } if (rc->_message_context) { delete rc->_message_context; } G_OBJECT_CLASS(parent_class)->dispose(object); } static Inkscape::XML::NodeEventVector ec_shape_repr_events = { NULL, /* child_added */ NULL, /* child_removed */ ec_shape_event_attr_changed, NULL, /* content_changed */ NULL /* order_changed */ }; /** \brief Callback that processes the "changed" signal on the selection; destroys old and creates new knotholder */ void sp_flood_context_selection_changed(Inkscape::Selection *selection, gpointer data) { SPFloodContext *rc = SP_FLOOD_CONTEXT(data); SPEventContext *ec = SP_EVENT_CONTEXT(rc); if (ec->shape_repr) { // remove old listener sp_repr_remove_listener_by_data(ec->shape_repr, ec); Inkscape::GC::release(ec->shape_repr); ec->shape_repr = 0; } SPItem *item = selection->singleItem(); if (item) { Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item); if (shape_repr) { ec->shape_repr = shape_repr; Inkscape::GC::anchor(shape_repr); sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec); } } } static void sp_flood_context_setup(SPEventContext *ec) { SPFloodContext *rc = SP_FLOOD_CONTEXT(ec); if (((SPEventContextClass *) parent_class)->setup) { ((SPEventContextClass *) parent_class)->setup(ec); } SPItem *item = sp_desktop_selection(ec->desktop)->singleItem(); if (item) { Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item); if (shape_repr) { ec->shape_repr = shape_repr; Inkscape::GC::anchor(shape_repr); sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec); } } rc->sel_changed_connection.disconnect(); rc->sel_changed_connection = sp_desktop_selection(ec->desktop)->connectChanged( sigc::bind(sigc::ptr_fun(&sp_flood_context_selection_changed), (gpointer)rc) ); rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack()); } static void merge_pixel_with_background(unsigned char *orig, unsigned char *bg, unsigned char *base) { for (int i = 0; i < 3; i++) { base[i] = (255 * (255 - bg[3])) / 255 + (bg[i] * bg[3]) / 255; base[i] = (base[i] * (255 - orig[3])) / 255 + (orig[i] * orig[3]) / 255; } base[3] = 255; } inline unsigned char * get_pixel(guchar *px, int x, int y, int width) { return px + (x + y * width) * 4; } enum PaintBucketChannels { FLOOD_CHANNELS_RGB, FLOOD_CHANNELS_ALPHA }; GList * flood_channels_dropdown_items_list() { GList *glist = NULL; glist = g_list_append (glist, _("Visible Colors")); glist = g_list_append (glist, _("Alpha")); return glist; } static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *dtc, int tolerance, PaintBucketChannels method) { int diff = 0; switch (method) { case FLOOD_CHANNELS_ALPHA: return ((int)abs(check[3] - orig[3]) <= (tolerance / 4)); case FLOOD_CHANNELS_RGB: unsigned char merged_orig[4]; unsigned char merged_check[4]; merge_pixel_with_background(orig, dtc, merged_orig); merge_pixel_with_background(check, dtc, merged_check); for (int i = 0; i < 3; i++) { diff += (int)abs(merged_check[i] - merged_orig[i]); } return ((diff / 3) <= ((tolerance * 3) / 4)); } return false; } static bool try_add_to_queue(std::queue *fill_queue, guchar *px, guchar *trace_px, unsigned char *orig, unsigned char *dtc, int x, int y, int width, int tolerance, PaintBucketChannels method, bool fill_switch) { unsigned char *t = get_pixel(px, x, y, width); if (compare_pixels(t, orig, dtc, tolerance, method)) { unsigned char *trace_t = get_pixel(trace_px, x, y, width); if (trace_t[3] != 255) { if (fill_switch) { fill_queue->push(NR::Point(x, y)); } } return false; } return true; } static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) { SPDocument *document = sp_desktop_document(desktop); Inkscape::Trace::Potrace::PotraceTracingEngine pte; pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS); pte.setInvert(false); Glib::RefPtr pixbuf = Glib::wrap(px, true); std::vector results = pte.trace(pixbuf); Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr; Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc()); long totalNodeCount = 0L; double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 2.0); for (unsigned int i=0 ; icreateElement("svg:path"); /* Set style */ sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false); NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str()); Path *path = bpath_to_Path(bpath); g_free(bpath); Shape *path_shape = new Shape(); path->ConvertWithBackData(0.03); path->Fill(path_shape, 0); delete path; Shape *expanded_path_shape = new Shape(); expanded_path_shape->ConvertToShape(path_shape, fill_nonZero); path_shape->MakeOffset(expanded_path_shape, offset, join_round, 4); expanded_path_shape->ConvertToShape(path_shape, fill_positive); Path *expanded_path = new Path(); expanded_path->Reset(); expanded_path_shape->ConvertToForme(expanded_path); expanded_path->ConvertEvenLines(1.0); expanded_path->Simplify(1.0); delete path_shape; delete expanded_path_shape; gchar *str = expanded_path->svg_dump_path(); delete expanded_path; pathRepr->setAttribute("d", str); g_free(str); layer_repr->addChild(pathRepr, NULL); SPObject *reprobj = document->getObjectByRepr(pathRepr); if (reprobj) { sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL); // premultiply the item transform by the accumulated parent transform in the paste layer NR::Matrix local = sp_item_i2doc_affine(SP_GROUP(desktop->currentLayer())); if (!local.test_identity()) { gchar const *t_str = pathRepr->attribute("transform"); NR::Matrix item_t (NR::identity()); if (t_str) sp_svg_transform_read(t_str, &item_t); item_t *= local.inverse(); // (we're dealing with unattached repr, so we write to its attr instead of using sp_item_set_transform) gchar *affinestr=sp_svg_transform_write(item_t); pathRepr->setAttribute("transform", affinestr); g_free(affinestr); } Inkscape::Selection *selection = sp_desktop_selection(desktop); selection->set(reprobj); pathRepr->setPosition(-1); } Inkscape::GC::release(pathRepr); } } struct bitmap_coords_info { bool is_left; int x; int y; int y_limit; int width; int tolerance; PaintBucketChannels method; unsigned char *dtc; bool top_fill; bool bottom_fill; NR::Rect bbox; NR::Rect screen; }; enum ScanlineCheckResult { SCANLINE_CHECK_OK, SCANLINE_CHECK_ABORTED, SCANLINE_CHECK_BOUNDARY }; static ScanlineCheckResult perform_bitmap_scanline_check(std::queue *fill_queue, guchar *px, guchar *trace_px, unsigned char *orig_color, bitmap_coords_info bci) { bool aborted = false; bool reached_screen_boundary = false; bool ok; bool keep_tracing; unsigned char *t, *trace_t; do { ok = false; if (bci.is_left) { keep_tracing = (bci.x >= 0); } else { keep_tracing = (bci.x < bci.width); } if (keep_tracing) { t = get_pixel(px, bci.x, bci.y, bci.width); if (compare_pixels(t, orig_color, bci.dtc, bci.tolerance, bci.method)) { for (int i = 0; i < 4; i++) { t[i] = 255 - t[i]; } trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width); trace_t[3] = 255; if (bci.y > 0) { bci.top_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.dtc, bci.x, bci.y - 1, bci.width, bci.tolerance, bci.method, bci.top_fill); } if (bci.y < bci.y_limit) { bci.bottom_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.dtc, bci.x, bci.y + 1, bci.width, bci.tolerance, bci.method, bci.bottom_fill); } if (bci.is_left) { bci.x--; } else { bci.x++; } ok = true; } } else { if (bci.bbox.min()[NR::X] > bci.screen.min()[NR::X]) { aborted = true; break; } else { reached_screen_boundary = true; } } } while (ok); if (aborted) { return SCANLINE_CHECK_ABORTED; } if (reached_screen_boundary) { return SCANLINE_CHECK_BOUNDARY; } return SCANLINE_CHECK_OK; } static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *event) { SPDesktop *desktop = event_context->desktop; SPDocument *document = sp_desktop_document(desktop); /* Create new arena */ NRArena *arena = NRArena::create(); unsigned dkey = sp_item_display_key_new(1); sp_document_ensure_up_to_date (document); SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document)); NR::Maybe bbox = document_root->getBounds(NR::identity()); if (!bbox) { desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("Area is not bounded, cannot fill.")); return; } double zoom_scale = desktop->current_zoom(); double padding = 1.6; NR::Rect screen = desktop->get_display_area(); int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding); int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding); NR::Point origin(screen.min()[NR::X], sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]); origin[NR::X] = origin[NR::X] + (screen.extent(NR::X) * ((1 - padding) / 2)); origin[NR::Y] = origin[NR::Y] + (screen.extent(NR::Y) * ((1 - padding) / 2)); NR::scale scale(zoom_scale, zoom_scale); NR::Matrix affine = scale * NR::translate(-origin * scale); /* Create ArenaItems and set transform */ NRArenaItem *root = sp_item_invoke_show(SP_ITEM(sp_document_root(document)), arena, dkey, SP_ITEM_SHOW_DISPLAY); nr_arena_item_set_transform(NR_ARENA_ITEM(root), affine); NRGC gc(NULL); nr_matrix_set_identity(&gc.transform); NRRectL final_bbox; final_bbox.x0 = 0; final_bbox.y0 = 0;//row; final_bbox.x1 = width; final_bbox.y1 = height;//row + num_rows; nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE); guchar *px = g_new(guchar, 4 * width * height); //memset(px, 0x00, 4 * width * height); NRPixBlock B; nr_pixblock_setup_extern( &B, NR_PIXBLOCK_MODE_R8G8B8A8N, final_bbox.x0, final_bbox.y0, final_bbox.x1, final_bbox.y1, px, 4 * width, FALSE, FALSE ); SPNamedView *nv = sp_desktop_namedview(desktop); unsigned long bgcolor = nv->pagecolor; unsigned char dtc[4]; dtc[0] = NR_RGBA32_R(bgcolor); dtc[1] = NR_RGBA32_G(bgcolor); dtc[2] = NR_RGBA32_B(bgcolor); dtc[3] = NR_RGBA32_A(bgcolor); for (int fy = 0; fy < height; fy++) { guchar *p = NR_PIXBLOCK_PX(&B) + fy * B.rs; for (int fx = 0; fx < width; fx++) { for (int i = 0; i < 4; i++) { *p++ = dtc[i]; } } } nr_arena_item_invoke_render(NULL, root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE ); nr_pixblock_release(&B); // Hide items sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey); nr_arena_item_unref(root); nr_object_unref((NRObject *) arena); NR::Point pw = NR::Point(event->button.x / zoom_scale, sp_document_height(document) + (event->button.y / zoom_scale)) * affine; pw[NR::X] = (int)MIN(width - 1, MAX(0, pw[NR::X])); pw[NR::Y] = (int)MIN(height - 1, MAX(0, pw[NR::Y])); guchar *trace_px = g_new(guchar, 4 * width * height); memset(trace_px, 0x00, 4 * width * height); std::queue fill_queue; fill_queue.push(pw); bool aborted = false; int y_limit = height - 1; unsigned char orig_color[4]; unsigned char *orig_px = get_pixel(px, (int)pw[NR::X], (int)pw[NR::Y], width); for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; } unsigned char merged_orig[4]; merge_pixel_with_background(orig_color, dtc, merged_orig); int tolerance = (255 * prefs_get_int_attribute_limited("tools.paintbucket", "tolerance", 1, 0, 100)) / 100; PaintBucketChannels method = (PaintBucketChannels)prefs_get_int_attribute("tools.paintbucket", "channels", 0); bool reached_screen_boundary = false; bitmap_coords_info bci; bci.y_limit = y_limit; bci.width = width; bci.tolerance = tolerance; bci.method = method; bci.bbox = *bbox; bci.screen = screen; bci.dtc = dtc; while (!fill_queue.empty() && !aborted) { NR::Point cp = fill_queue.front(); fill_queue.pop(); unsigned char *s = get_pixel(px, (int)cp[NR::X], (int)cp[NR::Y], width); // same color at this point if (compare_pixels(s, orig_color, dtc, tolerance, method)) { int x = (int)cp[NR::X]; int y = (int)cp[NR::Y]; bool top_fill = true; bool bottom_fill = true; if (y > 0) { top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y - 1, width, tolerance, method, top_fill); } else { if (bbox->min()[NR::Y] > screen.min()[NR::Y]) { aborted = true; break; } else { reached_screen_boundary = true; } } if (y < y_limit) { bottom_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y + 1, width, tolerance, method, bottom_fill); } else { if (bbox->max()[NR::Y] < screen.max()[NR::Y]) { aborted = true; break; } else { reached_screen_boundary = true; } } bci.is_left = true; bci.x = x; bci.y = y; bci.top_fill = top_fill; bci.bottom_fill = bottom_fill; ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci); switch (result) { case SCANLINE_CHECK_ABORTED: aborted = true; break; case SCANLINE_CHECK_BOUNDARY: reached_screen_boundary = true; break; default: break; } bci.is_left = false; bci.x = x + 1; bci.y = y; bci.top_fill = top_fill; bci.bottom_fill = bottom_fill; result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci); switch (result) { case SCANLINE_CHECK_ABORTED: aborted = true; break; case SCANLINE_CHECK_BOUNDARY: reached_screen_boundary = true; break; default: break; } } } g_free(px); if (aborted) { g_free(trace_px); desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("Area is not bounded, cannot fill.")); return; } if (reached_screen_boundary) { desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("Only the visible part of the bounded area was filled. If you want to fill all of the area, undo, zoom out, and fill again.")); } GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px, GDK_COLORSPACE_RGB, TRUE, 8, width, height, width * 4, (GdkPixbufDestroyNotify)g_free, NULL); NR::Matrix inverted_affine = NR::Matrix(affine).inverse(); do_trace(pixbuf, desktop, inverted_affine); g_free(trace_px); sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area")); } static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event) { gint ret = FALSE; switch (event->type) { case GDK_BUTTON_PRESS: break; // motion and release are always on root (why?) default: break; } if (((SPEventContextClass *) parent_class)->item_handler) { ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event); } return ret; } static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event) { gint ret = FALSE; switch (event->type) { case GDK_BUTTON_PRESS: if ( event->button.button == 1 ) { sp_flood_do_flood_fill(event_context, event); ret = TRUE; } break; case GDK_KEY_PRESS: switch (get_group0_keyval (&event->key)) { case GDK_Up: case GDK_Down: case GDK_KP_Up: case GDK_KP_Down: // prevent the zoom field from activation if (!MOD__CTRL_ONLY) ret = TRUE; break; default: break; } break; default: break; } if (!ret) { if (((SPEventContextClass *) parent_class)->root_handler) { ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event); } } return ret; } static void sp_flood_finish(SPFloodContext *rc) { rc->_message_context->clear(); if ( rc->item != NULL ) { SPDesktop * desktop; desktop = SP_EVENT_CONTEXT_DESKTOP(rc); SP_OBJECT(rc->item)->updateRepr(); sp_canvas_end_forced_full_redraws(desktop->canvas); sp_desktop_selection(desktop)->set(rc->item); sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area")); rc->item = NULL; } } void flood_channels_changed(GtkComboBox *cbox, GtkWidget *tbl) { prefs_set_int_attribute("tools.paintbucket", "channels", (gint)gtk_combo_box_get_active(cbox)); } /* Local Variables: mode:c++ c-file-style:"stroustrup" c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +)) indent-tabs-mode:nil fill-column:99 End: */ // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99 :