e737b90f92adf825299dc55a41a3a7aaad984590
1 #define __SP_FLOOD_CONTEXT_C__
3 /*
4 * Flood fill drawing context
5 *
6 * Author:
7 * Lauris Kaplinski <lauris@kaplinski.com>
8 * bulia byak <buliabyak@users.sf.net>
9 * John Bintz <jcoswell@coswellproductions.org>
10 *
11 * Copyright (C) 2006 Johan Engelen <johan@shouraizou.nl>
12 * Copyright (C) 2000-2005 authors
13 * Copyright (C) 2000-2001 Ximian, Inc.
14 *
15 * Released under GNU GPL, read the file 'COPYING' for more information
16 */
18 #include "config.h"
20 #include <gdk/gdkkeysyms.h>
21 #include <queue>
23 #include "macros.h"
24 #include "display/sp-canvas.h"
25 #include "document.h"
26 #include "sp-namedview.h"
27 #include "sp-object.h"
28 #include "sp-rect.h"
29 #include "selection.h"
30 #include "selection-chemistry.h"
31 #include "desktop-handles.h"
32 #include "snap.h"
33 #include "desktop.h"
34 #include "desktop-style.h"
35 #include "message-stack.h"
36 #include "message-context.h"
37 #include "pixmaps/cursor-paintbucket.xpm"
38 #include "flood-context.h"
39 #include "sp-metrics.h"
40 #include <glibmm/i18n.h>
41 #include "object-edit.h"
42 #include "xml/repr.h"
43 #include "xml/node-event-vector.h"
44 #include "prefs-utils.h"
45 #include "context-fns.h"
47 #include "display/nr-arena-item.h"
48 #include "display/nr-arena.h"
49 #include "display/nr-arena-image.h"
50 #include "display/canvas-arena.h"
51 #include "helper/png-write.h"
52 #include "libnr/nr-pixops.h"
53 #include "libnr/nr-matrix-rotate-ops.h"
54 #include "libnr/nr-matrix-translate-ops.h"
55 #include "libnr/nr-rotate-fns.h"
56 #include "libnr/nr-scale-ops.h"
57 #include "libnr/nr-scale-translate-ops.h"
58 #include "libnr/nr-translate-matrix-ops.h"
59 #include "libnr/nr-translate-scale-ops.h"
60 #include "libnr/nr-matrix-ops.h"
61 #include "sp-item.h"
62 #include "sp-root.h"
63 #include "sp-defs.h"
64 #include "splivarot.h"
65 #include "livarot/Path.h"
66 #include "livarot/Shape.h"
67 #include "libnr/n-art-bpath.h"
68 #include "svg/svg.h"
70 #include "trace/trace.h"
71 #include "trace/potrace/inkscape-potrace.h"
73 static void sp_flood_context_class_init(SPFloodContextClass *klass);
74 static void sp_flood_context_init(SPFloodContext *flood_context);
75 static void sp_flood_context_dispose(GObject *object);
77 static void sp_flood_context_setup(SPEventContext *ec);
79 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event);
80 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event);
82 static void sp_flood_finish(SPFloodContext *rc);
84 static SPEventContextClass *parent_class;
87 struct SPEBP {
88 int width, height, sheight;
89 guchar r, g, b, a;
90 NRArenaItem *root; // the root arena item to show; it is assumed that all unneeded items are hidden
91 guchar *px;
92 unsigned (*status)(float, void *);
93 void *data;
94 };
96 GtkType sp_flood_context_get_type()
97 {
98 static GType type = 0;
99 if (!type) {
100 GTypeInfo info = {
101 sizeof(SPFloodContextClass),
102 NULL, NULL,
103 (GClassInitFunc) sp_flood_context_class_init,
104 NULL, NULL,
105 sizeof(SPFloodContext),
106 4,
107 (GInstanceInitFunc) sp_flood_context_init,
108 NULL, /* value_table */
109 };
110 type = g_type_register_static(SP_TYPE_EVENT_CONTEXT, "SPFloodContext", &info, (GTypeFlags) 0);
111 }
112 return type;
113 }
115 static void sp_flood_context_class_init(SPFloodContextClass *klass)
116 {
117 GObjectClass *object_class = (GObjectClass *) klass;
118 SPEventContextClass *event_context_class = (SPEventContextClass *) klass;
120 parent_class = (SPEventContextClass *) g_type_class_peek_parent(klass);
122 object_class->dispose = sp_flood_context_dispose;
124 event_context_class->setup = sp_flood_context_setup;
125 event_context_class->root_handler = sp_flood_context_root_handler;
126 event_context_class->item_handler = sp_flood_context_item_handler;
127 }
129 static void sp_flood_context_init(SPFloodContext *flood_context)
130 {
131 SPEventContext *event_context = SP_EVENT_CONTEXT(flood_context);
133 event_context->cursor_shape = cursor_paintbucket_xpm;
134 event_context->hot_x = 11;
135 event_context->hot_y = 30;
136 event_context->xp = 0;
137 event_context->yp = 0;
138 event_context->tolerance = 0;
139 event_context->within_tolerance = false;
140 event_context->item_to_select = NULL;
142 event_context->shape_repr = NULL;
143 event_context->shape_knot_holder = NULL;
145 flood_context->item = NULL;
147 new (&flood_context->sel_changed_connection) sigc::connection();
148 }
150 static void sp_flood_context_dispose(GObject *object)
151 {
152 SPFloodContext *rc = SP_FLOOD_CONTEXT(object);
153 SPEventContext *ec = SP_EVENT_CONTEXT(object);
155 rc->sel_changed_connection.disconnect();
156 rc->sel_changed_connection.~connection();
158 /* fixme: This is necessary because we do not grab */
159 if (rc->item) {
160 sp_flood_finish(rc);
161 }
163 if (ec->shape_repr) { // remove old listener
164 sp_repr_remove_listener_by_data(ec->shape_repr, ec);
165 Inkscape::GC::release(ec->shape_repr);
166 ec->shape_repr = 0;
167 }
169 if (rc->_message_context) {
170 delete rc->_message_context;
171 }
173 G_OBJECT_CLASS(parent_class)->dispose(object);
174 }
176 static Inkscape::XML::NodeEventVector ec_shape_repr_events = {
177 NULL, /* child_added */
178 NULL, /* child_removed */
179 ec_shape_event_attr_changed,
180 NULL, /* content_changed */
181 NULL /* order_changed */
182 };
184 /**
185 \brief Callback that processes the "changed" signal on the selection;
186 destroys old and creates new knotholder
187 */
188 void sp_flood_context_selection_changed(Inkscape::Selection *selection, gpointer data)
189 {
190 SPFloodContext *rc = SP_FLOOD_CONTEXT(data);
191 SPEventContext *ec = SP_EVENT_CONTEXT(rc);
193 if (ec->shape_repr) { // remove old listener
194 sp_repr_remove_listener_by_data(ec->shape_repr, ec);
195 Inkscape::GC::release(ec->shape_repr);
196 ec->shape_repr = 0;
197 }
199 SPItem *item = selection->singleItem();
200 if (item) {
201 Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);
202 if (shape_repr) {
203 ec->shape_repr = shape_repr;
204 Inkscape::GC::anchor(shape_repr);
205 sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);
206 }
207 }
208 }
210 static void sp_flood_context_setup(SPEventContext *ec)
211 {
212 SPFloodContext *rc = SP_FLOOD_CONTEXT(ec);
214 if (((SPEventContextClass *) parent_class)->setup) {
215 ((SPEventContextClass *) parent_class)->setup(ec);
216 }
218 SPItem *item = sp_desktop_selection(ec->desktop)->singleItem();
219 if (item) {
220 Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);
221 if (shape_repr) {
222 ec->shape_repr = shape_repr;
223 Inkscape::GC::anchor(shape_repr);
224 sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);
225 }
226 }
228 rc->sel_changed_connection.disconnect();
229 rc->sel_changed_connection = sp_desktop_selection(ec->desktop)->connectChanged(
230 sigc::bind(sigc::ptr_fun(&sp_flood_context_selection_changed), (gpointer)rc)
231 );
233 rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack());
234 }
236 /**
237 Hide all items which are not listed in list, recursively, skipping groups and defs
238 */
239 static void
240 hide_other_items_recursively(SPObject *o, GSList *list, unsigned dkey)
241 {
242 if (SP_IS_ITEM(o)
243 && !SP_IS_DEFS(o)
244 && !SP_IS_ROOT(o)
245 && !SP_IS_GROUP(o)
246 && !g_slist_find(list, o))
247 {
248 sp_item_invoke_hide(SP_ITEM(o), dkey);
249 }
251 // recurse
252 if (!g_slist_find(list, o)) {
253 for (SPObject *child = sp_object_first_child(o) ; child != NULL; child = SP_OBJECT_NEXT(child) ) {
254 hide_other_items_recursively(child, list, dkey);
255 }
256 }
257 }
259 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
260 return px + (x + y * width) * 4;
261 }
263 static bool compare_pixels(unsigned char *a, unsigned char *b, int tolerance) {
264 for (int i = 0; i < 4; i++) {
265 if (abs(a[i] - b[i]) > tolerance) {
266 return false;
267 }
268 }
269 return true;
270 }
272 static void try_add_to_queue(std::queue<NR::Point> *fill_queue, guchar *px, unsigned char *orig, int x, int y, int width, int tolerance) {
273 unsigned char *t = get_pixel(px, x, y, width);
274 if (compare_pixels(t, orig, tolerance)) {
275 fill_queue->push(NR::Point(x, y));
276 }
277 }
279 static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) {
280 SPDocument *document = sp_desktop_document(desktop);
282 Inkscape::Trace::Potrace::PotraceTracingEngine pte;
284 pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS);
285 pte.setInvert(false);
287 Glib::RefPtr<Gdk::Pixbuf> pixbuf = Glib::wrap(px, true);
289 std::vector<Inkscape::Trace::TracingEngineResult> results = pte.trace(pixbuf);
291 Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr;
292 Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc());
294 long totalNodeCount = 0L;
296 for (unsigned int i=0 ; i<results.size() ; i++) {
297 Inkscape::Trace::TracingEngineResult result = results[i];
298 totalNodeCount += result.getNodeCount();
300 Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");
301 /* Set style */
302 sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
304 NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
305 Path *path = bpath_to_Path(bpath);
306 g_free(bpath);
308 Shape *path_shape = new Shape();
310 path->ConvertWithBackData(0.03);
311 path->Fill(path_shape, 0);
312 delete path;
314 Shape *expanded_path_shape = new Shape();
316 expanded_path_shape->ConvertToShape(path_shape, fill_nonZero);
317 path_shape->MakeOffset(expanded_path_shape, 1.5, join_round, 4);
318 expanded_path_shape->ConvertToShape(path_shape, fill_positive);
320 Path *expanded_path = new Path();
322 expanded_path->Reset();
323 expanded_path_shape->ConvertToForme(expanded_path);
324 expanded_path->ConvertEvenLines(1.0);
325 expanded_path->Simplify(1.0);
327 delete path_shape;
328 delete expanded_path_shape;
330 gchar *str = expanded_path->svg_dump_path();
331 delete expanded_path;
332 pathRepr->setAttribute("d", str);
333 g_free(str);
335 layer_repr->addChild(pathRepr, NULL);
337 SPObject *reprobj = document->getObjectByRepr(pathRepr);
338 if (reprobj) {
339 sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL);
340 Inkscape::Selection *selection = sp_desktop_selection(desktop);
341 selection->set(reprobj);
342 pathRepr->setPosition(-1);
343 }
345 Inkscape::GC::release(pathRepr);
346 }
347 }
349 static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *event) {
350 SPDesktop *desktop = event_context->desktop;
351 SPDocument *document = sp_desktop_document(desktop);
353 /* Create new arena */
354 NRArena *arena = NRArena::create();
355 unsigned dkey = sp_item_display_key_new(1);
357 sp_document_ensure_up_to_date (document);
359 SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document));
360 NR::Rect bbox = document_root->invokeBbox(NR::identity());
362 if (bbox.isEmpty()) {
363 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
364 return;
365 }
367 double zoom_scale = desktop->current_zoom();
368 double padding = 1.6;
370 NR::Rect screen = desktop->get_display_area();
372 int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
373 int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
375 NR::Point origin(screen.min()[NR::X],
376 sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);
378 origin[NR::X] = origin[NR::X] + (screen.extent(NR::X) * ((1 - padding) / 2));
379 origin[NR::Y] = origin[NR::Y] + (screen.extent(NR::Y) * ((1 - padding) / 2));
381 NR::scale scale(zoom_scale, zoom_scale);
382 NR::Matrix affine = scale * NR::translate(-origin * scale);
384 /* Create ArenaItems and set transform */
385 NRArenaItem *root = sp_item_invoke_show(SP_ITEM(sp_document_root(document)), arena, dkey, SP_ITEM_SHOW_DISPLAY);
386 nr_arena_item_set_transform(NR_ARENA_ITEM(root), affine);
388 NRGC gc(NULL);
389 nr_matrix_set_identity(&gc.transform);
391 NRRectL final_bbox;
392 final_bbox.x0 = 0;
393 final_bbox.y0 = 0;//row;
394 final_bbox.x1 = width;
395 final_bbox.y1 = height;//row + num_rows;
397 nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE);
399 guchar *px = g_new(guchar, 4 * width * height);
400 memset(px, 0x00, 4 * width * height);
402 NRPixBlock B;
403 nr_pixblock_setup_extern( &B, NR_PIXBLOCK_MODE_R8G8B8A8N,
404 final_bbox.x0, final_bbox.y0, final_bbox.x1, final_bbox.y1,
405 px, 4 * width, FALSE, FALSE );
407 nr_arena_item_invoke_render( root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE );
408 nr_pixblock_release(&B);
410 // Hide items
411 sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey);
413 nr_arena_item_unref(root);
414 nr_object_unref((NRObject *) arena);
416 NR::Point pw = NR::Point(event->button.x / zoom_scale, sp_document_height(document) + (event->button.y / zoom_scale)) * affine;
418 pw[NR::X] = (int)MIN(width - 1, MAX(0, pw[NR::X]));
419 pw[NR::Y] = (int)MIN(height - 1, MAX(0, pw[NR::Y]));
421 guchar *trace_px = g_new(guchar, 4 * width * height);
422 memset(trace_px, 0x00, 4 * width * height);
424 std::queue<NR::Point> fill_queue;
425 fill_queue.push(pw);
427 bool aborted = false;
428 int y_limit = height - 1;
430 unsigned char orig_color[4];
431 unsigned char *orig_px = get_pixel(px, (int)pw[NR::X], (int)pw[NR::Y], width);
432 for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; }
434 int tolerance = prefs_get_int_attribute_limited("tools.paintbucket", "tolerance", 1, 0, 255);
436 while (!fill_queue.empty() && !aborted) {
437 NR::Point cp = fill_queue.front();
438 fill_queue.pop();
439 unsigned char *s = get_pixel(px, (int)cp[NR::X], (int)cp[NR::Y], width);
441 // same color at this point
442 if (compare_pixels(s, orig_color, tolerance)) {
443 int left = (int)cp[NR::X];
444 int right = (int)cp[NR::X] + 1;
445 int x = (int)cp[NR::X];
446 int y = (int)cp[NR::Y];
448 if (y > 0) {
449 try_add_to_queue(&fill_queue, px, orig_color, x, y - 1, width, tolerance);
450 } else {
451 aborted = true; break;
452 }
453 if (y < y_limit) {
454 try_add_to_queue(&fill_queue, px, orig_color, x, y + 1, width, tolerance);
455 } else {
456 aborted = true; break;
457 }
459 unsigned char *t, *trace_t;
460 bool ok = false;
462 do {
463 ok = false;
464 // go left
465 if (left >= 0) {
466 t = get_pixel(px, left, y, width);
467 if (compare_pixels(t, orig_color, tolerance)) {
468 for (int i = 0; i < 4; i++) { t[i] = 255 - t[i]; }
469 trace_t = get_pixel(trace_px, left, y, width);
470 trace_t[3] = 255;
471 if (y > 0) { try_add_to_queue(&fill_queue, px, orig_color, left, y - 1, width, tolerance); }
472 if (y < y_limit) { try_add_to_queue(&fill_queue, px, orig_color, left, y + 1, width, tolerance); }
473 left--; ok = true;
474 }
475 } else {
476 aborted = true; break;
477 }
478 } while (ok);
480 do {
481 ok = false;
482 // go right
483 if (right < width) {
484 t = get_pixel(px, right, y, width);
485 if (compare_pixels(t, orig_color, tolerance)) {
486 for (int i = 0; i < 4; i++) { t[i] = 255 - t[i]; }
487 trace_t = get_pixel(trace_px, right, y, width);
488 trace_t[3] = 255;
489 if (y > 0) { try_add_to_queue(&fill_queue, px, orig_color, right, y - 1, width, tolerance); }
490 if (y < y_limit) { try_add_to_queue(&fill_queue, px, orig_color, right, y + 1, width, tolerance); }
491 right++; ok = true;
492 }
493 } else {
494 aborted = true; break;
495 }
496 } while (ok);
497 }
498 }
500 g_free(px);
502 if (aborted) {
503 g_free(trace_px);
504 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
505 return;
506 }
508 GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px,
509 GDK_COLORSPACE_RGB,
510 TRUE,
511 8, width, height, width * 4,
512 (GdkPixbufDestroyNotify)g_free,
513 NULL);
515 NR::Matrix inverted_affine = NR::Matrix(affine).inverse();
517 do_trace(pixbuf, desktop, inverted_affine);
519 g_free(trace_px);
521 sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));
522 }
524 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)
525 {
526 gint ret = FALSE;
528 switch (event->type) {
529 case GDK_BUTTON_PRESS:
530 break;
531 // motion and release are always on root (why?)
532 default:
533 break;
534 }
536 if (((SPEventContextClass *) parent_class)->item_handler) {
537 ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);
538 }
540 return ret;
541 }
543 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
544 {
545 SPDesktop *desktop = event_context->desktop;
547 gint ret = FALSE;
548 switch (event->type) {
549 case GDK_BUTTON_PRESS:
550 if ( event->button.button == 1 ) {
551 sp_flood_do_flood_fill(event_context, event);
553 ret = TRUE;
554 }
555 break;
556 case GDK_KEY_PRESS:
557 switch (get_group0_keyval (&event->key)) {
558 case GDK_Up:
559 case GDK_Down:
560 case GDK_KP_Up:
561 case GDK_KP_Down:
562 // prevent the zoom field from activation
563 if (!MOD__CTRL_ONLY)
564 ret = TRUE;
565 break;
566 case GDK_Escape:
567 sp_desktop_selection(desktop)->clear();
568 default:
569 break;
570 }
571 break;
572 default:
573 break;
574 }
576 if (!ret) {
577 if (((SPEventContextClass *) parent_class)->root_handler) {
578 ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);
579 }
580 }
582 return ret;
583 }
586 static void sp_flood_finish(SPFloodContext *rc)
587 {
588 rc->_message_context->clear();
590 if ( rc->item != NULL ) {
591 SPDesktop * desktop;
593 desktop = SP_EVENT_CONTEXT_DESKTOP(rc);
595 SP_OBJECT(rc->item)->updateRepr();
597 sp_canvas_end_forced_full_redraws(desktop->canvas);
599 sp_desktop_selection(desktop)->set(rc->item);
600 sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,
601 _("Fill bounded area"));
603 rc->item = NULL;
604 }
605 }
607 /*
608 Local Variables:
609 mode:c++
610 c-file-style:"stroustrup"
611 c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +))
612 indent-tabs-mode:nil
613 fill-column:99
614 End:
615 */
616 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99 :