Code

Final fix for gradient fill algorithm
[inkscape.git] / src / flood-context.cpp
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>
22 #include <deque>
24 #include "macros.h"
25 #include "display/sp-canvas.h"
26 #include "document.h"
27 #include "sp-namedview.h"
28 #include "sp-object.h"
29 #include "sp-rect.h"
30 #include "selection.h"
31 #include "desktop-handles.h"
32 #include "desktop.h"
33 #include "desktop-style.h"
34 #include "message-stack.h"
35 #include "message-context.h"
36 #include "pixmaps/cursor-paintbucket.xpm"
37 #include "flood-context.h"
38 #include "sp-metrics.h"
39 #include <glibmm/i18n.h>
40 #include "object-edit.h"
41 #include "xml/repr.h"
42 #include "xml/node-event-vector.h"
43 #include "prefs-utils.h"
44 #include "context-fns.h"
45 #include "rubberband.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 "libnr/nr-pixops.h"
52 #include "libnr/nr-matrix-translate-ops.h"
53 #include "libnr/nr-scale-ops.h"
54 #include "libnr/nr-scale-translate-ops.h"
55 #include "libnr/nr-translate-matrix-ops.h"
56 #include "libnr/nr-translate-scale-ops.h"
57 #include "libnr/nr-matrix-ops.h"
58 #include "sp-item.h"
59 #include "sp-root.h"
60 #include "sp-defs.h"
61 #include "sp-path.h"
62 #include "splivarot.h"
63 #include "livarot/Path.h"
64 #include "livarot/Shape.h"
65 #include "libnr/n-art-bpath.h"
66 #include "svg/svg.h"
67 #include "color.h"
69 #include "trace/trace.h"
70 #include "trace/potrace/inkscape-potrace.h"
72 static void sp_flood_context_class_init(SPFloodContextClass *klass);
73 static void sp_flood_context_init(SPFloodContext *flood_context);
74 static void sp_flood_context_dispose(GObject *object);
76 static void sp_flood_context_setup(SPEventContext *ec);
78 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event);
79 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event);
81 static void sp_flood_finish(SPFloodContext *rc);
83 static SPEventContextClass *parent_class;
86 GtkType sp_flood_context_get_type()
87 {
88     static GType type = 0;
89     if (!type) {
90         GTypeInfo info = {
91             sizeof(SPFloodContextClass),
92             NULL, NULL,
93             (GClassInitFunc) sp_flood_context_class_init,
94             NULL, NULL,
95             sizeof(SPFloodContext),
96             4,
97             (GInstanceInitFunc) sp_flood_context_init,
98             NULL,    /* value_table */
99         };
100         type = g_type_register_static(SP_TYPE_EVENT_CONTEXT, "SPFloodContext", &info, (GTypeFlags) 0);
101     }
102     return type;
105 static void sp_flood_context_class_init(SPFloodContextClass *klass)
107     GObjectClass *object_class = (GObjectClass *) klass;
108     SPEventContextClass *event_context_class = (SPEventContextClass *) klass;
110     parent_class = (SPEventContextClass *) g_type_class_peek_parent(klass);
112     object_class->dispose = sp_flood_context_dispose;
114     event_context_class->setup = sp_flood_context_setup;
115     event_context_class->root_handler  = sp_flood_context_root_handler;
116     event_context_class->item_handler  = sp_flood_context_item_handler;
119 static void sp_flood_context_init(SPFloodContext *flood_context)
121     SPEventContext *event_context = SP_EVENT_CONTEXT(flood_context);
123     event_context->cursor_shape = cursor_paintbucket_xpm;
124     event_context->hot_x = 11;
125     event_context->hot_y = 30;
126     event_context->xp = 0;
127     event_context->yp = 0;
128     event_context->tolerance = 4;
129     event_context->within_tolerance = false;
130     event_context->item_to_select = NULL;
132     event_context->shape_repr = NULL;
133     event_context->shape_knot_holder = NULL;
135     flood_context->item = NULL;
137     new (&flood_context->sel_changed_connection) sigc::connection();
140 static void sp_flood_context_dispose(GObject *object)
142     SPFloodContext *rc = SP_FLOOD_CONTEXT(object);
143     SPEventContext *ec = SP_EVENT_CONTEXT(object);
145     rc->sel_changed_connection.disconnect();
146     rc->sel_changed_connection.~connection();
148     /* fixme: This is necessary because we do not grab */
149     if (rc->item) {
150         sp_flood_finish(rc);
151     }
153     if (ec->shape_repr) { // remove old listener
154         sp_repr_remove_listener_by_data(ec->shape_repr, ec);
155         Inkscape::GC::release(ec->shape_repr);
156         ec->shape_repr = 0;
157     }
159     if (rc->_message_context) {
160         delete rc->_message_context;
161     }
163     G_OBJECT_CLASS(parent_class)->dispose(object);
166 static Inkscape::XML::NodeEventVector ec_shape_repr_events = {
167     NULL, /* child_added */
168     NULL, /* child_removed */
169     ec_shape_event_attr_changed,
170     NULL, /* content_changed */
171     NULL  /* order_changed */
172 };
174 /**
175 \brief  Callback that processes the "changed" signal on the selection;
176 destroys old and creates new knotholder
177 */
178 void sp_flood_context_selection_changed(Inkscape::Selection *selection, gpointer data)
180     SPFloodContext *rc = SP_FLOOD_CONTEXT(data);
181     SPEventContext *ec = SP_EVENT_CONTEXT(rc);
183     if (ec->shape_repr) { // remove old listener
184         sp_repr_remove_listener_by_data(ec->shape_repr, ec);
185         Inkscape::GC::release(ec->shape_repr);
186         ec->shape_repr = 0;
187     }
189     SPItem *item = selection->singleItem();
190     if (item) {
191         Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);
192         if (shape_repr) {
193             ec->shape_repr = shape_repr;
194             Inkscape::GC::anchor(shape_repr);
195             sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);
196         }
197     }
200 static void sp_flood_context_setup(SPEventContext *ec)
202     SPFloodContext *rc = SP_FLOOD_CONTEXT(ec);
204     if (((SPEventContextClass *) parent_class)->setup) {
205         ((SPEventContextClass *) parent_class)->setup(ec);
206     }
208     SPItem *item = sp_desktop_selection(ec->desktop)->singleItem();
209     if (item) {
210         Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);
211         if (shape_repr) {
212             ec->shape_repr = shape_repr;
213             Inkscape::GC::anchor(shape_repr);
214             sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);
215         }
216     }
218     rc->sel_changed_connection.disconnect();
219     rc->sel_changed_connection = sp_desktop_selection(ec->desktop)->connectChanged(
220         sigc::bind(sigc::ptr_fun(&sp_flood_context_selection_changed), (gpointer)rc)
221     );
223     rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack());
226 static void
227 merge_pixel_with_background (unsigned char *orig, unsigned char *bg,
228            unsigned char *base)
230     for (int i = 0; i < 3; i++) {
231         base[i] = (255 * (255 - bg[3])) / 255 + (bg[i] * bg[3]) / 255;
232         base[i] = (base[i] * (255 - orig[3])) / 255 + (orig[i] * orig[3]) / 255;
233     }
234     base[3] = 255;
237 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
238     return px + (x + y * width) * 4;
241 enum PaintBucketChannels {
242     FLOOD_CHANNELS_RGB,
243     FLOOD_CHANNELS_R,
244     FLOOD_CHANNELS_G,
245     FLOOD_CHANNELS_B,
246     FLOOD_CHANNELS_H,
247     FLOOD_CHANNELS_S,
248     FLOOD_CHANNELS_L,
249     FLOOD_CHANNELS_ALPHA
250 };
252 GList * flood_channels_dropdown_items_list() {
253     GList *glist = NULL;
255     glist = g_list_append (glist, _("Visible Colors"));
256     glist = g_list_append (glist, _("Red"));
257     glist = g_list_append (glist, _("Green"));
258     glist = g_list_append (glist, _("Blue"));
259     glist = g_list_append (glist, _("Hue"));
260     glist = g_list_append (glist, _("Saturation"));
261     glist = g_list_append (glist, _("Lightness"));
262     glist = g_list_append (glist, _("Alpha"));
264     return glist;
267 static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *dtc, int threshold, PaintBucketChannels method) {
268     int diff = 0;
269     float hsl_check[3], hsl_orig[3];
270     
271     if ((method == FLOOD_CHANNELS_H) ||
272         (method == FLOOD_CHANNELS_S) ||
273         (method == FLOOD_CHANNELS_L)) {
274         sp_color_rgb_to_hsl_floatv(hsl_check, check[0] / 255.0, check[1] / 255.0, check[2] / 255.0);
275         sp_color_rgb_to_hsl_floatv(hsl_orig, orig[0] / 255.0, orig[1] / 255.0, orig[2] / 255.0);
276     }
277     
278     switch (method) {
279         case FLOOD_CHANNELS_ALPHA:
280             return ((int)abs(check[3] - orig[3]) <= threshold);
281         case FLOOD_CHANNELS_R:
282             return ((int)abs(check[0] - orig[0]) <= threshold);
283         case FLOOD_CHANNELS_G:
284             return ((int)abs(check[1] - orig[1]) <= threshold);
285         case FLOOD_CHANNELS_B:
286             return ((int)abs(check[2] - orig[2]) <= threshold);
287         case FLOOD_CHANNELS_RGB:
288             unsigned char merged_orig[4];
289             unsigned char merged_check[4];
290             
291             merge_pixel_with_background(orig, dtc, merged_orig);
292             merge_pixel_with_background(check, dtc, merged_check);
293             
294             for (int i = 0; i < 3; i++) {
295               diff += (int)abs(merged_check[i] - merged_orig[i]);
296             }
297             return ((diff / 3) <= ((threshold * 3) / 4));
298         
299         case FLOOD_CHANNELS_H:
300             return ((int)(fabs(hsl_check[0] - hsl_orig[0]) * 100.0) <= threshold);
301         case FLOOD_CHANNELS_S:
302             return ((int)(fabs(hsl_check[1] - hsl_orig[1]) * 100.0) <= threshold);
303         case FLOOD_CHANNELS_L:
304             return ((int)(fabs(hsl_check[2] - hsl_orig[2]) * 100.0) <= threshold);
305     }
306     
307     return false;
310 static bool try_add_to_queue(std::deque<NR::Point> *fill_queue, guchar *px, guchar *trace_px, unsigned char *orig, unsigned char *dtc, int x, int y, int width, int threshold, PaintBucketChannels method, bool fill_switch) {
311     unsigned char *t = get_pixel(px, x, y, width);
312     if (compare_pixels(t, orig, dtc, threshold, method)) {
313         unsigned char *trace_t = get_pixel(trace_px, x, y, width);
314         if (trace_t[3] != 255 && trace_t[0] != 255) {
315             if (fill_switch) {
316                 fill_queue->push_back(NR::Point(x, y));
317                 trace_t[0] = 255;
318             }
319         }
320         return false;
321     }
322     return true;
325 static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform, bool union_with_selection) {
326     SPDocument *document = sp_desktop_document(desktop);
327     
328     Inkscape::Trace::Potrace::PotraceTracingEngine pte;
329         
330     pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS);
331     pte.setInvert(false);
333     Glib::RefPtr<Gdk::Pixbuf> pixbuf = Glib::wrap(px, true);
334     
335     std::vector<Inkscape::Trace::TracingEngineResult> results = pte.trace(pixbuf);
336     
337     Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr;
338     Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc());
340     long totalNodeCount = 0L;
342     double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 0.0);
344     for (unsigned int i=0 ; i<results.size() ; i++) {
345         Inkscape::Trace::TracingEngineResult result = results[i];
346         totalNodeCount += result.getNodeCount();
348         Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");
349         /* Set style */
350         sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
352         NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
353         Path *path = bpath_to_Path(bpath);
354         g_free(bpath);
356         if (offset != 0) {
357         
358             Shape *path_shape = new Shape();
359         
360             path->ConvertWithBackData(0.03);
361             path->Fill(path_shape, 0);
362             delete path;
363         
364             Shape *expanded_path_shape = new Shape();
365         
366             expanded_path_shape->ConvertToShape(path_shape, fill_nonZero);
367             path_shape->MakeOffset(expanded_path_shape, offset * desktop->current_zoom(), join_round, 4);
368             expanded_path_shape->ConvertToShape(path_shape, fill_positive);
370             Path *expanded_path = new Path();
371         
372             expanded_path->Reset();
373             expanded_path_shape->ConvertToForme(expanded_path);
374             expanded_path->ConvertEvenLines(1.0);
375             expanded_path->Simplify(1.0);
376         
377             delete path_shape;
378             delete expanded_path_shape;
379         
380             gchar *str = expanded_path->svg_dump_path();
381             if (str && *str) {
382                 pathRepr->setAttribute("d", str);
383                 g_free(str);
384             } else {
385                 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Too much inset</b>, the result is empty."));
386                 Inkscape::GC::release(pathRepr);
387                 g_free(str);
388                 return;
389             }
391             delete expanded_path;
393         } else {
394             gchar *str = path->svg_dump_path();
395             delete path;
396             pathRepr->setAttribute("d", str);
397             g_free(str);
398         }
400         layer_repr->addChild(pathRepr, NULL);
402         SPObject *reprobj = document->getObjectByRepr(pathRepr);
403         if (reprobj) {
404             sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL);
405             
406             // premultiply the item transform by the accumulated parent transform in the paste layer
407             NR::Matrix local = sp_item_i2doc_affine(SP_GROUP(desktop->currentLayer()));
408             if (!local.test_identity()) {
409                 gchar const *t_str = pathRepr->attribute("transform");
410                 NR::Matrix item_t (NR::identity());
411                 if (t_str)
412                     sp_svg_transform_read(t_str, &item_t);
413                 item_t *= local.inverse();
414                 // (we're dealing with unattached repr, so we write to its attr instead of using sp_item_set_transform)
415                 gchar *affinestr=sp_svg_transform_write(item_t);
416                 pathRepr->setAttribute("transform", affinestr);
417                 g_free(affinestr);
418             }
420             Inkscape::Selection *selection = sp_desktop_selection(desktop);
422             pathRepr->setPosition(-1);
424             if (union_with_selection) {
425                 desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, _("Area filled, path with <b>%d</b> nodes created and unioned with selection."), sp_nodes_in_path(SP_PATH(reprobj)));
426                 selection->add(reprobj);
427                 sp_selected_path_union_skip_undo();
428             } else {
429                 desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, _("Area filled, path with <b>%d</b> nodes created."), sp_nodes_in_path(SP_PATH(reprobj)));
430                 selection->set(reprobj);
431             }
433         }
435         Inkscape::GC::release(pathRepr);
437     }
440 struct bitmap_coords_info {
441     bool is_left;
442     int x;
443     int y;
444     int y_limit;
445     int width;
446     int threshold;
447     PaintBucketChannels method;
448     unsigned char *dtc;
449     bool top_fill;
450     bool bottom_fill;
451     NR::Rect bbox;
452     NR::Rect screen;
453 };
455 enum ScanlineCheckResult {
456     SCANLINE_CHECK_OK,
457     SCANLINE_CHECK_ABORTED,
458     SCANLINE_CHECK_BOUNDARY
459 };
461 static ScanlineCheckResult perform_bitmap_scanline_check(std::deque<NR::Point> *fill_queue, guchar *px, guchar *trace_px, unsigned char *orig_color, bitmap_coords_info bci) {
462     bool aborted = false;
463     bool reached_screen_boundary = false;
464     bool ok;
465   
466     bool keep_tracing;
467     unsigned char *t, *trace_t;
468   
469     do {
470         ok = false;
471         if (bci.is_left) {
472             keep_tracing = (bci.x >= 0);
473         } else {
474             keep_tracing = (bci.x < bci.width);
475         }
476         
477         if (keep_tracing) {
478             t = get_pixel(px, bci.x, bci.y, bci.width);
479             if (compare_pixels(t, orig_color, bci.dtc, bci.threshold, bci.method)) {
480                 for (int i = 0; i < 4; i++) { t[i] = 255 - t[i]; }
481                 trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width);
482                 trace_t[3] = 255; 
483                 if (bci.y > 0) { 
484                     bci.top_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.dtc, bci.x, bci.y - 1, bci.width, bci.threshold, bci.method, bci.top_fill);
485                 }
486                 if (bci.y < bci.y_limit) { 
487                     bci.bottom_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.dtc, bci.x, bci.y + 1, bci.width, bci.threshold, bci.method, bci.bottom_fill);
488                 }
489                 if (bci.is_left) {
490                     bci.x--;
491                 } else {
492                     bci.x++;
493                 }
494                 ok = true;
495             }
496         } else {
497             if (bci.bbox.min()[NR::X] > bci.screen.min()[NR::X]) {
498                 aborted = true; break;
499             } else {
500                 reached_screen_boundary = true;
501             }
502         }
503     } while (ok);
504     
505     if (aborted) { return SCANLINE_CHECK_ABORTED; }
506     if (reached_screen_boundary) { return SCANLINE_CHECK_BOUNDARY; }
507     return SCANLINE_CHECK_OK;
510 static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *event, bool union_with_selection, bool is_point_fill, bool is_touch_fill) {
511     SPDesktop *desktop = event_context->desktop;
512     SPDocument *document = sp_desktop_document(desktop);
514     /* Create new arena */
515     NRArena *arena = NRArena::create();
516     unsigned dkey = sp_item_display_key_new(1);
518     sp_document_ensure_up_to_date (document);
519     
520     SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document));
521     NR::Maybe<NR::Rect> bbox = document_root->getBounds(NR::identity());
523     if (!bbox) {
524         desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
525         return;
526     }
527     
528     double zoom_scale = desktop->current_zoom();
529     double padding = 1.6;
531     NR::Rect screen = desktop->get_display_area();
533     int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
534     int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
536     NR::Point origin(screen.min()[NR::X],
537                      sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);
538                     
539     origin[NR::X] = origin[NR::X] + (screen.extent(NR::X) * ((1 - padding) / 2));
540     origin[NR::Y] = origin[NR::Y] + (screen.extent(NR::Y) * ((1 - padding) / 2));
541     
542     NR::scale scale(zoom_scale, zoom_scale);
543     NR::Matrix affine = scale * NR::translate(-origin * scale);
544     
545     /* Create ArenaItems and set transform */
546     NRArenaItem *root = sp_item_invoke_show(SP_ITEM(sp_document_root(document)), arena, dkey, SP_ITEM_SHOW_DISPLAY);
547     nr_arena_item_set_transform(NR_ARENA_ITEM(root), affine);
549     NRGC gc(NULL);
550     nr_matrix_set_identity(&gc.transform);
551     
552     NRRectL final_bbox;
553     final_bbox.x0 = 0;
554     final_bbox.y0 = 0;//row;
555     final_bbox.x1 = width;
556     final_bbox.y1 = height;//row + num_rows;
557     
558     nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE);
560     guchar *px = g_new(guchar, 4 * width * height);
561     
562     NRPixBlock B;
563     nr_pixblock_setup_extern( &B, NR_PIXBLOCK_MODE_R8G8B8A8N,
564                               final_bbox.x0, final_bbox.y0, final_bbox.x1, final_bbox.y1,
565                               px, 4 * width, FALSE, FALSE );
566     
567     SPNamedView *nv = sp_desktop_namedview(desktop);
568     unsigned long bgcolor = nv->pagecolor;
569     
570     unsigned char dtc[4];
571     dtc[0] = NR_RGBA32_R(bgcolor);
572     dtc[1] = NR_RGBA32_G(bgcolor);
573     dtc[2] = NR_RGBA32_B(bgcolor);
574     dtc[3] = NR_RGBA32_A(bgcolor);
575     
576     for (int fy = 0; fy < height; fy++) {
577         guchar *p = NR_PIXBLOCK_PX(&B) + fy * B.rs;
578         for (int fx = 0; fx < width; fx++) {
579             for (int i = 0; i < 4; i++) { 
580                 *p++ = dtc[i];
581             }
582         }
583     }
585     nr_arena_item_invoke_render(NULL, root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE );
586     nr_pixblock_release(&B);
587     
588     // Hide items
589     sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey);
590     
591     nr_arena_item_unref(root);
592     nr_object_unref((NRObject *) arena);
593     
594     guchar *trace_px = g_new(guchar, 4 * width * height);
595     memset(trace_px, 0x00, 4 * width * height);
596     
597     std::deque<NR::Point> fill_queue;
598     std::queue<NR::Point> color_queue;
599     
600     std::vector<NR::Point> fill_points;
601     
602     if (is_point_fill) {
603         fill_points.push_back(NR::Point(event->button.x, event->button.y));
604     } else {
605         Inkscape::Rubberband::Rubberband *r = Inkscape::Rubberband::get();
606         fill_points = r->getPoints();
607     }
609     for (unsigned int i = 0; i < fill_points.size(); i++) {
610         NR::Point pw = NR::Point(fill_points[i][NR::X] / zoom_scale, sp_document_height(document) + (fill_points[i][NR::Y] / zoom_scale)) * affine;
611         
612         pw[NR::X] = (int)MIN(width - 1, MAX(0, pw[NR::X]));
613         pw[NR::Y] = (int)MIN(height - 1, MAX(0, pw[NR::Y]));
614         
615         if (is_touch_fill) {
616             if (i == 0) {
617                 color_queue.push(pw);
618             } else {
619                 fill_queue.push_back(pw);
620             }
621         } else {
622             color_queue.push(pw);
623         }
624     }
626     bool aborted = false;
627     int y_limit = height - 1;
629     PaintBucketChannels method = (PaintBucketChannels)prefs_get_int_attribute("tools.paintbucket", "channels", 0);
630     int threshold = prefs_get_int_attribute_limited("tools.paintbucket", "threshold", 1, 0, 100);
632     switch(method) {
633         case FLOOD_CHANNELS_ALPHA:
634         case FLOOD_CHANNELS_RGB:
635         case FLOOD_CHANNELS_R:
636         case FLOOD_CHANNELS_G:
637         case FLOOD_CHANNELS_B:
638             threshold = (255 * threshold) / 100;
639             break;
640         case FLOOD_CHANNELS_H:
641         case FLOOD_CHANNELS_S:
642         case FLOOD_CHANNELS_L:
643             break;
644       }
646     bool reached_screen_boundary = false;
648     bitmap_coords_info bci;
649     
650     bci.y_limit = y_limit;
651     bci.width = width;
652     bci.threshold = threshold;
653     bci.method = method;
654     bci.bbox = *bbox;
655     bci.screen = screen;
656     bci.dtc = dtc;
658     while (!color_queue.empty() && !aborted) {
659         NR::Point color_point = color_queue.front();
660         color_queue.pop();
661         
662         unsigned char *orig_px = get_pixel(px, (int)color_point[NR::X], (int)color_point[NR::Y], width);
663         unsigned char orig_color[4];
664         for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; }
665         
666         unsigned char merged_orig[4];
667     
668         merge_pixel_with_background(orig_color, dtc, merged_orig);
669         
670         unsigned char *trace_t = get_pixel(trace_px, (int)color_point[NR::X], (int)color_point[NR::Y], width);
671         if ((trace_t[0] != 255) && (trace_t[3] != 255)) {
672           fill_queue.push_front(color_point);
673         }
674         
675         while (!fill_queue.empty() && !aborted) {
676             NR::Point cp = fill_queue.front();
677             fill_queue.pop_front();
678             
679             unsigned char *s = get_pixel(px, (int)cp[NR::X], (int)cp[NR::Y], width);
680             
681             // same color at this point
682             if (compare_pixels(s, orig_color, dtc, threshold, method)) {
683                 int x = (int)cp[NR::X];
684                 int y = (int)cp[NR::Y];
685                 
686                 bool top_fill = true;
687                 bool bottom_fill = true;
688                 
689                 if (y > 0) { 
690                     top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y - 1, width, threshold, method, top_fill);
691                 } else {
692                     if (bbox->min()[NR::Y] > screen.min()[NR::Y]) {
693                         aborted = true; break;
694                     } else {
695                         reached_screen_boundary = true;
696                     }
697                 }
698                 if (y < y_limit) { 
699                     bottom_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y + 1, width, threshold, method, bottom_fill);
700                   } else {
701                       if (bbox->max()[NR::Y] < screen.max()[NR::Y]) {
702                           aborted = true; break;
703                       } else {
704                           reached_screen_boundary = true;
705                       }
706                 }
707                 
708                 bci.is_left = true;
709                 bci.x = x;
710                 bci.y = y;
711                 bci.top_fill = top_fill;
712                 bci.bottom_fill = bottom_fill;
713                 
714                 ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
715                 
716                 switch (result) {
717                     case SCANLINE_CHECK_ABORTED:
718                         aborted = true;
719                         break;
720                     case SCANLINE_CHECK_BOUNDARY:
721                         reached_screen_boundary = true;
722                         break;
723                     default:
724                         break;
725                 }
726                 
727                 bci.is_left = false;
728                 bci.x = x + 1;
729                 bci.y = y;
730                 bci.top_fill = top_fill;
731                 bci.bottom_fill = bottom_fill;
732                 
733                 result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
734                 
735                 switch (result) {
736                     case SCANLINE_CHECK_ABORTED:
737                         aborted = true;
738                         break;
739                     case SCANLINE_CHECK_BOUNDARY:
740                         reached_screen_boundary = true;
741                         break;
742                     default:
743                         break;
744                 }
745             }
746         }
747     }
748     
749     g_free(px);
750     
751     if (aborted) {
752         g_free(trace_px);
753         desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
754         return;
755     }
756     
757     if (reached_screen_boundary) {
758         desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Only the visible part of the bounded area was filled.</b> If you want to fill all of the area, undo, zoom out, and fill again.")); 
759     }
760     
761     GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px,
762                                       GDK_COLORSPACE_RGB,
763                                       TRUE,
764                                       8, width, height, width * 4,
765                                       (GdkPixbufDestroyNotify)g_free,
766                                       NULL);
768     NR::Matrix inverted_affine = NR::Matrix(affine).inverse();
769     
770     do_trace(pixbuf, desktop, inverted_affine, union_with_selection);
772     g_free(trace_px);
773     
774     sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));
777 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)
779     gint ret = FALSE;
781     SPDesktop *desktop = event_context->desktop;
783     switch (event->type) {
784     case GDK_BUTTON_PRESS:
785         if (event->button.state & GDK_CONTROL_MASK) {
786             NR::Point const button_w(event->button.x,
787                                     event->button.y);
788             
789             SPItem *item = sp_event_context_find_item (desktop, button_w, TRUE, TRUE);
790             
791             Inkscape::XML::Node *pathRepr = SP_OBJECT_REPR(item);
792             /* Set style */
793             sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
794             sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET, _("Set style on object"));
795             ret = TRUE;
796         }
797         break;
798     default:
799         break;
800     }
802     if (((SPEventContextClass *) parent_class)->item_handler) {
803         ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);
804     }
806     return ret;
809 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
811     static bool dragging;
812     
813     gint ret = FALSE;
814     SPDesktop *desktop = event_context->desktop;
816     switch (event->type) {
817     case GDK_BUTTON_PRESS:
818         if ( event->button.button == 1 ) {
819             if (!(event->button.state & GDK_CONTROL_MASK)) {
820                 NR::Point const button_w(event->button.x,
821                                         event->button.y);
822     
823                 // save drag origin
824                 event_context->xp = (gint) button_w[NR::X];
825                 event_context->yp = (gint) button_w[NR::Y];
826                 event_context->within_tolerance = true;
827                 
828                 dragging = true;
829                 
830                 NR::Point const p(desktop->w2d(button_w));
831                 Inkscape::Rubberband::get()->setMode(RUBBERBAND_MODE_TOUCHPATH);
832                 Inkscape::Rubberband::get()->start(desktop, p);
833             }
834         }
835     case GDK_MOTION_NOTIFY:
836         if ( dragging
837              && ( event->motion.state & GDK_BUTTON1_MASK ) )
838         {
839             if ( event_context->within_tolerance
840                  && ( abs( (gint) event->motion.x - event_context->xp ) < event_context->tolerance )
841                  && ( abs( (gint) event->motion.y - event_context->yp ) < event_context->tolerance ) ) {
842                 break; // do not drag if we're within tolerance from origin
843             }
844             
845             event_context->within_tolerance = false;
846             
847             NR::Point const motion_pt(event->motion.x, event->motion.y);
848             NR::Point const p(desktop->w2d(motion_pt));
849             if (Inkscape::Rubberband::get()->is_started()) {
850                 Inkscape::Rubberband::get()->move(p);
851                 event_context->defaultMessageContext()->set(Inkscape::NORMAL_MESSAGE, _("<b>Draw over</b> areas to add to fill, hold <b>Alt</b> for touch fill"));
852                 gobble_motion_events(GDK_BUTTON1_MASK);
853             }
854         }
855         break;
857     case GDK_BUTTON_RELEASE:
858         if ( event->button.button == 1 ) {
859             Inkscape::Rubberband::Rubberband *r = Inkscape::Rubberband::get();
860             if (r->is_started()) {
861                 // set "busy" cursor
862                 desktop->setWaitingCursor();
864                 if (SP_IS_EVENT_CONTEXT(event_context)) { 
865                     // Since setWaitingCursor runs main loop iterations, we may have already left this tool!
866                     // So check if the tool is valid before doing anything
867                     dragging = false;
869                     bool is_point_fill = event_context->within_tolerance;
870                     bool is_touch_fill = event->button.state & GDK_MOD1_MASK;
871                     
872                     sp_flood_do_flood_fill(event_context, event, event->button.state & GDK_SHIFT_MASK, is_point_fill, is_touch_fill);
873                     
874                     desktop->clearWaitingCursor();
875                     // restore cursor when done; note that it may already be different if e.g. user 
876                     // switched to another tool during interruptible tracing or drawing, in which case do nothing
878                     ret = TRUE;
879                 }
881                 r->stop();
882                 event_context->defaultMessageContext()->clear();
883             }
884         }
885         break;
886     case GDK_KEY_PRESS:
887         switch (get_group0_keyval (&event->key)) {
888         case GDK_Up:
889         case GDK_Down:
890         case GDK_KP_Up:
891         case GDK_KP_Down:
892             // prevent the zoom field from activation
893             if (!MOD__CTRL_ONLY)
894                 ret = TRUE;
895             break;
896         default:
897             break;
898         }
899         break;
900     default:
901         break;
902     }
904     if (!ret) {
905         if (((SPEventContextClass *) parent_class)->root_handler) {
906             ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);
907         }
908     }
910     return ret;
914 static void sp_flood_finish(SPFloodContext *rc)
916     rc->_message_context->clear();
918     if ( rc->item != NULL ) {
919         SPDesktop * desktop;
921         desktop = SP_EVENT_CONTEXT_DESKTOP(rc);
923         SP_OBJECT(rc->item)->updateRepr();
925         sp_canvas_end_forced_full_redraws(desktop->canvas);
927         sp_desktop_selection(desktop)->set(rc->item);
928         sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,
929                         _("Fill bounded area"));
931         rc->item = NULL;
932     }
935 void flood_channels_set_channels( gint channels )
937     prefs_set_int_attribute("tools.paintbucket", "channels", channels);
940 /*
941   Local Variables:
942   mode:c++
943   c-file-style:"stroustrup"
944   c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +))
945   indent-tabs-mode:nil
946   fill-column:99
947   End:
948 */
949 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99 :