9f0b041f3a09cacf53294d3bdc726c9eb81b38ea
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;
103 }
105 static void sp_flood_context_class_init(SPFloodContextClass *klass)
106 {
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;
117 }
119 static void sp_flood_context_init(SPFloodContext *flood_context)
120 {
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();
138 }
140 static void sp_flood_context_dispose(GObject *object)
141 {
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);
164 }
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)
179 {
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 }
198 }
200 static void sp_flood_context_setup(SPEventContext *ec)
201 {
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());
225 if (prefs_get_int_attribute("tools.paintbucket", "selcue", 0) != 0) {
226 rc->enableSelectionCue();
227 }
228 }
230 inline static void
231 merge_pixel_with_background (unsigned char *orig, unsigned char *bg,
232 unsigned char *base)
233 {
234 int precalc_bg_alpha = (255 * (255 - bg[3])) / 255;
236 for (int i = 0; i < 3; i++) {
237 base[i] = precalc_bg_alpha + (bg[i] * bg[3]) / 255;
238 base[i] = (base[i] * (255 - orig[3])) / 255 + (orig[i] * orig[3]) / 255;
239 }
240 }
242 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
243 return px + (x + y * width) * 4;
244 }
246 GList * flood_channels_dropdown_items_list() {
247 GList *glist = NULL;
249 glist = g_list_append (glist, _("Visible Colors"));
250 glist = g_list_append (glist, _("Red"));
251 glist = g_list_append (glist, _("Green"));
252 glist = g_list_append (glist, _("Blue"));
253 glist = g_list_append (glist, _("Hue"));
254 glist = g_list_append (glist, _("Saturation"));
255 glist = g_list_append (glist, _("Lightness"));
256 glist = g_list_append (glist, _("Alpha"));
258 return glist;
259 }
261 GList * flood_autogap_dropdown_items_list() {
262 GList *glist = NULL;
264 glist = g_list_append (glist, _("None"));
265 glist = g_list_append (glist, _("Small"));
266 glist = g_list_append (glist, _("Medium"));
267 glist = g_list_append (glist, _("Large"));
269 return glist;
270 }
272 static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *merged_orig_pixel, unsigned char *dtc, int threshold, PaintBucketChannels method) {
273 int diff = 0;
274 float hsl_check[3], hsl_orig[3];
276 if ((method == FLOOD_CHANNELS_H) ||
277 (method == FLOOD_CHANNELS_S) ||
278 (method == FLOOD_CHANNELS_L)) {
279 sp_color_rgb_to_hsl_floatv(hsl_check, check[0] / 255.0, check[1] / 255.0, check[2] / 255.0);
280 sp_color_rgb_to_hsl_floatv(hsl_orig, orig[0] / 255.0, orig[1] / 255.0, orig[2] / 255.0);
281 }
283 switch (method) {
284 case FLOOD_CHANNELS_ALPHA:
285 return ((int)abs(check[3] - orig[3]) <= threshold);
286 case FLOOD_CHANNELS_R:
287 return ((int)abs(check[0] - orig[0]) <= threshold);
288 case FLOOD_CHANNELS_G:
289 return ((int)abs(check[1] - orig[1]) <= threshold);
290 case FLOOD_CHANNELS_B:
291 return ((int)abs(check[2] - orig[2]) <= threshold);
292 case FLOOD_CHANNELS_RGB:
293 unsigned char merged_check[3];
295 merge_pixel_with_background(check, dtc, merged_check);
297 for (int i = 0; i < 3; i++) {
298 diff += (int)abs(merged_check[i] - merged_orig_pixel[i]);
299 }
300 return ((diff / 3) <= ((threshold * 3) / 4));
302 case FLOOD_CHANNELS_H:
303 return ((int)(fabs(hsl_check[0] - hsl_orig[0]) * 100.0) <= threshold);
304 case FLOOD_CHANNELS_S:
305 return ((int)(fabs(hsl_check[1] - hsl_orig[1]) * 100.0) <= threshold);
306 case FLOOD_CHANNELS_L:
307 return ((int)(fabs(hsl_check[2] - hsl_orig[2]) * 100.0) <= threshold);
308 }
310 return false;
311 }
313 static inline bool is_pixel_checked(unsigned char *t) { return t[0] == 1; }
314 static inline bool is_pixel_queued(unsigned char *t) { return t[1] == 1; }
315 static inline bool is_pixel_paintability_checked(unsigned char *t) { return t[2] != 0; }
316 static inline bool is_pixel_paintable(unsigned char *t) { return t[2] == 1; }
317 static inline bool is_pixel_colored(unsigned char *t) { return t[3] == 255; }
319 static inline void mark_pixel_checked(unsigned char *t) { t[0] = 1; }
320 static inline void mark_pixel_unchecked(unsigned char *t) { t[0] = 0; }
321 static inline void mark_pixel_queued(unsigned char *t) { t[1] = 1; }
322 static inline void mark_pixel_paintable(unsigned char *t) { t[2] = 1; }
323 static inline void mark_pixel_not_paintable(unsigned char *t) { t[2] = 2; }
324 static inline void mark_pixel_colored(unsigned char *t) { t[3] = 255; }
326 static inline void clear_pixel_paintability(unsigned char *t) { t[2] = 0; }
328 struct bitmap_coords_info {
329 bool is_left;
330 unsigned int x;
331 unsigned int y;
332 int y_limit;
333 unsigned int width;
334 unsigned int height;
335 unsigned int threshold;
336 unsigned int radius;
337 PaintBucketChannels method;
338 unsigned char *dtc;
339 unsigned char *merged_orig_pixel;
340 NR::Rect bbox;
341 NR::Rect screen;
342 unsigned int max_queue_size;
343 unsigned int current_step;
344 };
346 inline static bool check_if_pixel_is_paintable(guchar *px, unsigned char *trace_t, int x, int y, unsigned char *orig_color, bitmap_coords_info bci) {
347 if (is_pixel_paintability_checked(trace_t)) {
348 return is_pixel_paintable(trace_t);
349 } else {
350 unsigned char *t = get_pixel(px, x, y, bci.width);
351 if (compare_pixels(t, orig_color, bci.merged_orig_pixel, bci.dtc, bci.threshold, bci.method)) {
352 mark_pixel_paintable(trace_t);
353 return true;
354 } else {
355 mark_pixel_not_paintable(trace_t);
356 return false;
357 }
358 }
359 }
361 static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform, bool union_with_selection) {
362 SPDocument *document = sp_desktop_document(desktop);
364 Inkscape::Trace::Potrace::PotraceTracingEngine pte;
366 pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS);
367 pte.setInvert(false);
369 Glib::RefPtr<Gdk::Pixbuf> pixbuf = Glib::wrap(px, true);
371 std::vector<Inkscape::Trace::TracingEngineResult> results = pte.trace(pixbuf);
373 Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr;
374 Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc());
376 long totalNodeCount = 0L;
378 double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 0.0);
380 for (unsigned int i=0 ; i<results.size() ; i++) {
381 Inkscape::Trace::TracingEngineResult result = results[i];
382 totalNodeCount += result.getNodeCount();
384 Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");
385 /* Set style */
386 sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
388 NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
389 Path *path = bpath_to_Path(bpath);
390 g_free(bpath);
392 if (offset != 0) {
394 Shape *path_shape = new Shape();
396 path->ConvertWithBackData(0.03);
397 path->Fill(path_shape, 0);
398 delete path;
400 Shape *expanded_path_shape = new Shape();
402 expanded_path_shape->ConvertToShape(path_shape, fill_nonZero);
403 path_shape->MakeOffset(expanded_path_shape, offset * desktop->current_zoom(), join_round, 4);
404 expanded_path_shape->ConvertToShape(path_shape, fill_positive);
406 Path *expanded_path = new Path();
408 expanded_path->Reset();
409 expanded_path_shape->ConvertToForme(expanded_path);
410 expanded_path->ConvertEvenLines(1.0);
411 expanded_path->Simplify(1.0);
413 delete path_shape;
414 delete expanded_path_shape;
416 gchar *str = expanded_path->svg_dump_path();
417 if (str && *str) {
418 pathRepr->setAttribute("d", str);
419 g_free(str);
420 } else {
421 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Too much inset</b>, the result is empty."));
422 Inkscape::GC::release(pathRepr);
423 g_free(str);
424 return;
425 }
427 delete expanded_path;
429 } else {
430 gchar *str = path->svg_dump_path();
431 delete path;
432 pathRepr->setAttribute("d", str);
433 g_free(str);
434 }
436 layer_repr->addChild(pathRepr, NULL);
438 SPObject *reprobj = document->getObjectByRepr(pathRepr);
439 if (reprobj) {
440 sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL);
442 // premultiply the item transform by the accumulated parent transform in the paste layer
443 NR::Matrix local = sp_item_i2doc_affine(SP_GROUP(desktop->currentLayer()));
444 if (!local.test_identity()) {
445 gchar const *t_str = pathRepr->attribute("transform");
446 NR::Matrix item_t (NR::identity());
447 if (t_str)
448 sp_svg_transform_read(t_str, &item_t);
449 item_t *= local.inverse();
450 // (we're dealing with unattached repr, so we write to its attr instead of using sp_item_set_transform)
451 gchar *affinestr=sp_svg_transform_write(item_t);
452 pathRepr->setAttribute("transform", affinestr);
453 g_free(affinestr);
454 }
456 Inkscape::Selection *selection = sp_desktop_selection(desktop);
458 pathRepr->setPosition(-1);
460 if (union_with_selection) {
461 desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, ngettext("Area filled, path with <b>%d</b> node created and unioned with selection.","Area filled, path with <b>%d</b> nodes created and unioned with selection.",sp_nodes_in_path(SP_PATH(reprobj))), sp_nodes_in_path(SP_PATH(reprobj)));
462 selection->add(reprobj);
463 sp_selected_path_union_skip_undo();
464 } else {
465 desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, ngettext("Area filled, path with <b>%d</b> node created.","Area filled, path with <b>%d</b> nodes created.",sp_nodes_in_path(SP_PATH(reprobj))), sp_nodes_in_path(SP_PATH(reprobj)));
466 selection->set(reprobj);
467 }
469 }
471 Inkscape::GC::release(pathRepr);
473 }
474 }
476 enum ScanlineCheckResult {
477 SCANLINE_CHECK_OK,
478 SCANLINE_CHECK_ABORTED,
479 SCANLINE_CHECK_BOUNDARY
480 };
482 inline static bool coords_in_range(unsigned int x, unsigned int y, bitmap_coords_info bci) {
483 return (x < bci.width) &&
484 (y < bci.height);
485 }
487 #define PAINT_DIRECTION_LEFT 1
488 #define PAINT_DIRECTION_RIGHT 2
489 #define PAINT_DIRECTION_UP 4
490 #define PAINT_DIRECTION_DOWN 8
491 #define PAINT_DIRECTION_ALL 15
493 inline static unsigned int paint_pixel(guchar *px, guchar *trace_px, unsigned char *orig_color, bitmap_coords_info bci, unsigned char *original_point_trace_t) {
494 if (bci.radius == 0) {
495 mark_pixel_colored(original_point_trace_t);
496 return PAINT_DIRECTION_ALL;
497 } else {
498 unsigned char *trace_t;
500 bool can_paint_up = true;
501 bool can_paint_down = true;
502 bool can_paint_left = true;
503 bool can_paint_right = true;
505 for (unsigned int ty = bci.y - bci.radius; ty <= bci.y + bci.radius; ty++) {
506 for (unsigned int tx = bci.x - bci.radius; tx <= bci.x + bci.radius; tx++) {
507 if (coords_in_range(tx, ty, bci)) {
508 trace_t = get_pixel(trace_px, tx, ty, bci.width);
509 if (!is_pixel_colored(trace_t)) {
510 if (check_if_pixel_is_paintable(px, trace_t, tx, ty, orig_color, bci)) {
511 mark_pixel_colored(trace_t);
512 } else {
513 if (tx < bci.x) { can_paint_left = false; }
514 if (tx > bci.x) { can_paint_right = false; }
515 if (ty < bci.y) { can_paint_up = false; }
516 if (ty > bci.y) { can_paint_down = false; }
517 }
518 }
519 }
520 }
521 }
523 unsigned int paint_directions = 0;
524 if (can_paint_left) { paint_directions += PAINT_DIRECTION_LEFT; }
525 if (can_paint_right) { paint_directions += PAINT_DIRECTION_RIGHT; }
526 if (can_paint_up) { paint_directions += PAINT_DIRECTION_UP; }
527 if (can_paint_down) { paint_directions += PAINT_DIRECTION_DOWN; }
529 return paint_directions;
530 }
531 }
533 static void push_point_onto_queue(std::deque<NR::Point> *fill_queue, unsigned int max_queue_size, unsigned char *trace_t, unsigned int x, unsigned int y) {
534 if (!is_pixel_queued(trace_t)) {
535 if ((fill_queue->size() < max_queue_size)) {
536 fill_queue->push_back(NR::Point(x, y));
537 mark_pixel_queued(trace_t);
538 }
539 }
540 }
542 static void shift_point_onto_queue(std::deque<NR::Point> *fill_queue, unsigned int max_queue_size, unsigned char *trace_t, unsigned int x, unsigned int y) {
543 if (!is_pixel_queued(trace_t)) {
544 if ((fill_queue->size() < max_queue_size)) {
545 fill_queue->push_front(NR::Point(x, y));
546 mark_pixel_queued(trace_t);
547 }
548 }
549 }
551 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) {
552 bool aborted = false;
553 bool reached_screen_boundary = false;
554 bool ok;
556 bool keep_tracing;
557 bool initial_paint = true;
559 unsigned char *current_trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width);
560 unsigned int paint_directions;
562 bool currently_painting_top = false;
563 bool currently_painting_bottom = false;
565 unsigned int pix_width = bci.width * 4;
567 unsigned int top_ty = bci.y - 1;
568 unsigned int bottom_ty = bci.y + 1;
570 bool can_paint_top = (top_ty > 0);
571 bool can_paint_bottom = (bottom_ty < bci.height);
573 NR::Point t = fill_queue->front();
575 do {
576 ok = false;
577 if (bci.is_left) {
578 keep_tracing = (bci.x != 0);
579 } else {
580 keep_tracing = (bci.x < bci.width);
581 }
583 if (keep_tracing) {
584 if (check_if_pixel_is_paintable(px, current_trace_t, bci.x, bci.y, orig_color, bci)) {
585 paint_directions = paint_pixel(px, trace_px, orig_color, bci, current_trace_t);
586 if (bci.radius == 0) {
587 mark_pixel_checked(current_trace_t);
588 if ((t[NR::X] == bci.x) && (t[NR::Y] == bci.y)) {
589 fill_queue->pop_front(); t = fill_queue->front();
590 }
591 }
593 if (can_paint_top) {
594 if (paint_directions & PAINT_DIRECTION_UP) {
595 unsigned char *trace_t = current_trace_t - pix_width;
596 if (!is_pixel_queued(trace_t)) {
597 bool ok_to_paint = check_if_pixel_is_paintable(px, trace_t, bci.x, top_ty, orig_color, bci);
599 if (initial_paint) { currently_painting_top = !ok_to_paint; }
601 if (ok_to_paint && (!currently_painting_top)) {
602 currently_painting_top = true;
603 push_point_onto_queue(fill_queue, bci.max_queue_size, trace_t, bci.x, top_ty);
604 }
605 if ((!ok_to_paint) && currently_painting_top) {
606 currently_painting_top = false;
607 }
608 }
609 }
610 }
612 if (can_paint_bottom) {
613 if (paint_directions & PAINT_DIRECTION_DOWN) {
614 unsigned char *trace_t = current_trace_t + pix_width;
615 if (!is_pixel_queued(trace_t)) {
616 bool ok_to_paint = check_if_pixel_is_paintable(px, trace_t, bci.x, bottom_ty, orig_color, bci);
618 if (initial_paint) { currently_painting_bottom = !ok_to_paint; }
620 if (ok_to_paint && (!currently_painting_bottom)) {
621 currently_painting_bottom = true;
622 push_point_onto_queue(fill_queue, bci.max_queue_size, trace_t, bci.x, bottom_ty);
623 }
624 if ((!ok_to_paint) && currently_painting_bottom) {
625 currently_painting_bottom = false;
626 }
627 }
628 }
629 }
631 if (bci.is_left) {
632 if (paint_directions & PAINT_DIRECTION_LEFT) {
633 bci.x -= 1; current_trace_t -= 4;
634 ok = true;
635 }
636 } else {
637 if (paint_directions & PAINT_DIRECTION_RIGHT) {
638 bci.x += 1; current_trace_t += 4;
639 ok = true;
640 }
641 }
643 initial_paint = false;
644 }
645 } else {
646 if (bci.bbox.min()[NR::X] > bci.screen.min()[NR::X]) {
647 aborted = true; break;
648 } else {
649 reached_screen_boundary = true;
650 }
651 }
652 } while (ok);
654 if (aborted) { return SCANLINE_CHECK_ABORTED; }
655 if (reached_screen_boundary) { return SCANLINE_CHECK_BOUNDARY; }
656 return SCANLINE_CHECK_OK;
657 }
659 static bool sort_fill_queue_vertical(NR::Point a, NR::Point b) {
660 return a[NR::Y] > b[NR::Y];
661 }
663 static bool sort_fill_queue_horizontal(NR::Point a, NR::Point b) {
664 return a[NR::X] > b[NR::X];
665 }
667 static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *event, bool union_with_selection, bool is_point_fill, bool is_touch_fill) {
668 SPDesktop *desktop = event_context->desktop;
669 SPDocument *document = sp_desktop_document(desktop);
671 /* Create new arena */
672 NRArena *arena = NRArena::create();
673 unsigned dkey = sp_item_display_key_new(1);
675 sp_document_ensure_up_to_date (document);
677 SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document));
678 NR::Maybe<NR::Rect> bbox = document_root->getBounds(NR::identity());
680 if (!bbox) {
681 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
682 return;
683 }
685 double zoom_scale = desktop->current_zoom();
686 double padding = 1.6;
688 NR::Rect screen = desktop->get_display_area();
690 unsigned int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
691 unsigned int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
693 NR::Point origin(screen.min()[NR::X],
694 sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);
696 origin[NR::X] = origin[NR::X] + (screen.extent(NR::X) * ((1 - padding) / 2));
697 origin[NR::Y] = origin[NR::Y] + (screen.extent(NR::Y) * ((1 - padding) / 2));
699 NR::scale scale(zoom_scale, zoom_scale);
700 NR::Matrix affine = scale * NR::translate(-origin * scale);
702 /* Create ArenaItems and set transform */
703 NRArenaItem *root = sp_item_invoke_show(SP_ITEM(sp_document_root(document)), arena, dkey, SP_ITEM_SHOW_DISPLAY);
704 nr_arena_item_set_transform(NR_ARENA_ITEM(root), affine);
706 NRGC gc(NULL);
707 nr_matrix_set_identity(&gc.transform);
709 NRRectL final_bbox;
710 final_bbox.x0 = 0;
711 final_bbox.y0 = 0;//row;
712 final_bbox.x1 = width;
713 final_bbox.y1 = height;//row + num_rows;
715 nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE);
717 guchar *px = g_new(guchar, 4 * width * height);
719 NRPixBlock B;
720 nr_pixblock_setup_extern( &B, NR_PIXBLOCK_MODE_R8G8B8A8N,
721 final_bbox.x0, final_bbox.y0, final_bbox.x1, final_bbox.y1,
722 px, 4 * width, FALSE, FALSE );
724 SPNamedView *nv = sp_desktop_namedview(desktop);
725 unsigned long bgcolor = nv->pagecolor;
727 unsigned char dtc[4];
728 dtc[0] = NR_RGBA32_R(bgcolor);
729 dtc[1] = NR_RGBA32_G(bgcolor);
730 dtc[2] = NR_RGBA32_B(bgcolor);
731 dtc[3] = NR_RGBA32_A(bgcolor);
733 for (unsigned int fy = 0; fy < height; fy++) {
734 guchar *p = NR_PIXBLOCK_PX(&B) + fy * B.rs;
735 for (unsigned int fx = 0; fx < width; fx++) {
736 for (int i = 0; i < 4; i++) {
737 *p++ = dtc[i];
738 }
739 }
740 }
742 nr_arena_item_invoke_render(NULL, root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE );
743 nr_pixblock_release(&B);
745 // Hide items
746 sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey);
748 nr_arena_item_unref(root);
749 nr_object_unref((NRObject *) arena);
751 guchar *trace_px = g_new(guchar, 4 * width * height);
752 memset(trace_px, 0x00, 4 * width * height);
754 std::deque<NR::Point> fill_queue;
755 std::queue<NR::Point> color_queue;
757 std::vector<NR::Point> fill_points;
759 bool aborted = false;
760 int y_limit = height - 1;
762 PaintBucketChannels method = (PaintBucketChannels)prefs_get_int_attribute("tools.paintbucket", "channels", 0);
763 int threshold = prefs_get_int_attribute_limited("tools.paintbucket", "threshold", 1, 0, 100);
765 switch(method) {
766 case FLOOD_CHANNELS_ALPHA:
767 case FLOOD_CHANNELS_RGB:
768 case FLOOD_CHANNELS_R:
769 case FLOOD_CHANNELS_G:
770 case FLOOD_CHANNELS_B:
771 threshold = (255 * threshold) / 100;
772 break;
773 case FLOOD_CHANNELS_H:
774 case FLOOD_CHANNELS_S:
775 case FLOOD_CHANNELS_L:
776 break;
777 }
779 bitmap_coords_info bci;
781 bci.y_limit = y_limit;
782 bci.width = width;
783 bci.height = height;
784 bci.threshold = threshold;
785 bci.method = method;
786 bci.bbox = *bbox;
787 bci.screen = screen;
788 bci.dtc = dtc;
789 bci.radius = prefs_get_int_attribute_limited("tools.paintbucket", "autogap", 0, 0, 3);
790 bci.max_queue_size = (width * height) / 4;
791 bci.current_step = 0;
793 if (is_point_fill) {
794 fill_points.push_back(NR::Point(event->button.x, event->button.y));
795 } else {
796 Inkscape::Rubberband::Rubberband *r = Inkscape::Rubberband::get();
797 fill_points = r->getPoints();
798 }
800 for (unsigned int i = 0; i < fill_points.size(); i++) {
801 NR::Point pw = NR::Point(fill_points[i][NR::X] / zoom_scale, sp_document_height(document) + (fill_points[i][NR::Y] / zoom_scale)) * affine;
803 pw[NR::X] = (int)MIN(width - 1, MAX(0, pw[NR::X]));
804 pw[NR::Y] = (int)MIN(height - 1, MAX(0, pw[NR::Y]));
806 if (is_touch_fill) {
807 if (i == 0) {
808 color_queue.push(pw);
809 } else {
810 unsigned char *trace_t = get_pixel(trace_px, (int)pw[NR::X], (int)pw[NR::Y], width);
811 push_point_onto_queue(&fill_queue, bci.max_queue_size, trace_t, (int)pw[NR::X], (int)pw[NR::Y]);
812 }
813 } else {
814 color_queue.push(pw);
815 }
816 }
818 bool reached_screen_boundary = false;
820 bool first_run = true;
822 unsigned long sort_size_threshold = 5;
824 while (!color_queue.empty() && !aborted) {
825 NR::Point color_point = color_queue.front();
826 color_queue.pop();
828 int cx = (int)color_point[NR::X];
829 int cy = (int)color_point[NR::Y];
831 unsigned char *orig_px = get_pixel(px, cx, cy, width);
832 unsigned char orig_color[4];
833 for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; }
835 unsigned char merged_orig[3];
837 merge_pixel_with_background(orig_color, dtc, merged_orig);
839 bci.merged_orig_pixel = merged_orig;
841 unsigned char *trace_t = get_pixel(trace_px, cx, cy, width);
842 if (!is_pixel_checked(trace_t) && !is_pixel_colored(trace_t)) {
843 if (check_if_pixel_is_paintable(px, trace_px, cx, cy, orig_color, bci)) {
844 shift_point_onto_queue(&fill_queue, bci.max_queue_size, trace_t, cx, cy);
846 if (!first_run) {
847 for (unsigned int y = 0; y < height; y++) {
848 trace_t = get_pixel(trace_px, 0, y, width);
849 for (unsigned int x = 0; x < width; x++) {
850 clear_pixel_paintability(trace_t);
851 trace_t += 4;
852 }
853 }
854 }
855 first_run = false;
856 }
857 }
859 unsigned long old_fill_queue_size = fill_queue.size();
861 while (!fill_queue.empty() && !aborted) {
862 NR::Point cp = fill_queue.front();
864 if (bci.radius == 0) {
865 unsigned long new_fill_queue_size = fill_queue.size();
867 if (new_fill_queue_size > sort_size_threshold) {
868 if (new_fill_queue_size > old_fill_queue_size) {
869 std::sort(fill_queue.begin(), fill_queue.end(), sort_fill_queue_vertical);
871 std::deque<NR::Point>::iterator start_sort = fill_queue.begin();
872 std::deque<NR::Point>::iterator end_sort = fill_queue.begin();
873 unsigned int sort_y = (unsigned int)cp[NR::Y];
874 unsigned int current_y = sort_y;
876 for (std::deque<NR::Point>::iterator i = fill_queue.begin(); i != fill_queue.end(); i++) {
877 NR::Point current = *i;
878 current_y = (unsigned int)current[NR::Y];
879 if (current_y != sort_y) {
880 if (start_sort != end_sort) {
881 std::sort(start_sort, end_sort, sort_fill_queue_horizontal);
882 }
883 sort_y = current_y;
884 start_sort = i;
885 }
886 end_sort = i;
887 }
888 if (start_sort != end_sort) {
889 std::sort(start_sort, end_sort, sort_fill_queue_horizontal);
890 }
892 cp = fill_queue.front();
893 }
894 }
896 old_fill_queue_size = new_fill_queue_size;
897 }
899 fill_queue.pop_front();
901 int x = (int)cp[NR::X];
902 int y = (int)cp[NR::Y];
904 unsigned char *trace_t = get_pixel(trace_px, x, y, width);
905 if (!is_pixel_checked(trace_t)) {
906 mark_pixel_checked(trace_t);
908 if (y == 0) {
909 if (bbox->min()[NR::Y] > screen.min()[NR::Y]) {
910 aborted = true; break;
911 } else {
912 reached_screen_boundary = true;
913 }
914 }
916 if (y == y_limit) {
917 if (bbox->max()[NR::Y] < screen.max()[NR::Y]) {
918 aborted = true; break;
919 } else {
920 reached_screen_boundary = true;
921 }
922 }
924 bci.is_left = true;
925 bci.x = x;
926 bci.y = y;
928 ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
930 switch (result) {
931 case SCANLINE_CHECK_ABORTED:
932 aborted = true;
933 break;
934 case SCANLINE_CHECK_BOUNDARY:
935 reached_screen_boundary = true;
936 break;
937 default:
938 break;
939 }
941 if (bci.x < width) {
942 trace_t += 4;
943 if (!is_pixel_checked(trace_t) && !is_pixel_queued(trace_t)) {
944 mark_pixel_checked(trace_t);
945 bci.is_left = false;
946 bci.x = x + 1;
948 result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
950 switch (result) {
951 case SCANLINE_CHECK_ABORTED:
952 aborted = true;
953 break;
954 case SCANLINE_CHECK_BOUNDARY:
955 reached_screen_boundary = true;
956 break;
957 default:
958 break;
959 }
960 }
961 }
962 }
964 bci.current_step++;
966 if (bci.current_step > bci.max_queue_size) {
967 aborted = true;
968 }
969 }
970 }
972 g_free(px);
974 if (aborted) {
975 g_free(trace_px);
976 desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
977 return;
978 }
980 if (reached_screen_boundary) {
981 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."));
982 }
984 GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px,
985 GDK_COLORSPACE_RGB,
986 TRUE,
987 8, width, height, width * 4,
988 (GdkPixbufDestroyNotify)g_free,
989 NULL);
991 NR::Matrix inverted_affine = NR::Matrix(affine).inverse();
993 do_trace(pixbuf, desktop, inverted_affine, union_with_selection);
995 g_free(trace_px);
997 sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));
998 }
1000 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)
1001 {
1002 gint ret = FALSE;
1004 SPDesktop *desktop = event_context->desktop;
1006 switch (event->type) {
1007 case GDK_BUTTON_PRESS:
1008 if ((event->button.state & GDK_CONTROL_MASK) && event->button.button == 1 && !event_context->space_panning) {
1009 NR::Point const button_w(event->button.x,
1010 event->button.y);
1012 SPItem *item = sp_event_context_find_item (desktop, button_w, TRUE, TRUE);
1014 Inkscape::XML::Node *pathRepr = SP_OBJECT_REPR(item);
1015 /* Set style */
1016 sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
1017 sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET, _("Set style on object"));
1018 ret = TRUE;
1019 }
1020 break;
1021 default:
1022 break;
1023 }
1025 if (((SPEventContextClass *) parent_class)->item_handler) {
1026 ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);
1027 }
1029 return ret;
1030 }
1032 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
1033 {
1034 static bool dragging;
1036 gint ret = FALSE;
1037 SPDesktop *desktop = event_context->desktop;
1039 switch (event->type) {
1040 case GDK_BUTTON_PRESS:
1041 if (event->button.button == 1 && !event_context->space_panning) {
1042 if (!(event->button.state & GDK_CONTROL_MASK)) {
1043 NR::Point const button_w(event->button.x,
1044 event->button.y);
1046 if (Inkscape::have_viable_layer(desktop, event_context->defaultMessageContext())) {
1047 // save drag origin
1048 event_context->xp = (gint) button_w[NR::X];
1049 event_context->yp = (gint) button_w[NR::Y];
1050 event_context->within_tolerance = true;
1052 dragging = true;
1054 NR::Point const p(desktop->w2d(button_w));
1055 Inkscape::Rubberband::get()->setMode(RUBBERBAND_MODE_TOUCHPATH);
1056 Inkscape::Rubberband::get()->start(desktop, p);
1057 }
1058 }
1059 }
1060 case GDK_MOTION_NOTIFY:
1061 if ( dragging
1062 && ( event->motion.state & GDK_BUTTON1_MASK ) && !event_context->space_panning)
1063 {
1064 if ( event_context->within_tolerance
1065 && ( abs( (gint) event->motion.x - event_context->xp ) < event_context->tolerance )
1066 && ( abs( (gint) event->motion.y - event_context->yp ) < event_context->tolerance ) ) {
1067 break; // do not drag if we're within tolerance from origin
1068 }
1070 event_context->within_tolerance = false;
1072 NR::Point const motion_pt(event->motion.x, event->motion.y);
1073 NR::Point const p(desktop->w2d(motion_pt));
1074 if (Inkscape::Rubberband::get()->is_started()) {
1075 Inkscape::Rubberband::get()->move(p);
1076 event_context->defaultMessageContext()->set(Inkscape::NORMAL_MESSAGE, _("<b>Draw over</b> areas to add to fill, hold <b>Alt</b> for touch fill"));
1077 gobble_motion_events(GDK_BUTTON1_MASK);
1078 }
1079 }
1080 break;
1082 case GDK_BUTTON_RELEASE:
1083 if (event->button.button == 1 && !event_context->space_panning) {
1084 Inkscape::Rubberband::Rubberband *r = Inkscape::Rubberband::get();
1085 if (r->is_started()) {
1086 // set "busy" cursor
1087 desktop->setWaitingCursor();
1089 if (SP_IS_EVENT_CONTEXT(event_context)) {
1090 // Since setWaitingCursor runs main loop iterations, we may have already left this tool!
1091 // So check if the tool is valid before doing anything
1092 dragging = false;
1094 bool is_point_fill = event_context->within_tolerance;
1095 bool is_touch_fill = event->button.state & GDK_MOD1_MASK;
1097 sp_flood_do_flood_fill(event_context, event, event->button.state & GDK_SHIFT_MASK, is_point_fill, is_touch_fill);
1099 desktop->clearWaitingCursor();
1100 // restore cursor when done; note that it may already be different if e.g. user
1101 // switched to another tool during interruptible tracing or drawing, in which case do nothing
1103 ret = TRUE;
1104 }
1106 r->stop();
1108 if (SP_IS_EVENT_CONTEXT(event_context)) {
1109 event_context->defaultMessageContext()->clear();
1110 }
1111 }
1112 }
1113 break;
1114 case GDK_KEY_PRESS:
1115 switch (get_group0_keyval (&event->key)) {
1116 case GDK_Up:
1117 case GDK_Down:
1118 case GDK_KP_Up:
1119 case GDK_KP_Down:
1120 // prevent the zoom field from activation
1121 if (!MOD__CTRL_ONLY)
1122 ret = TRUE;
1123 break;
1124 default:
1125 break;
1126 }
1127 break;
1128 default:
1129 break;
1130 }
1132 if (!ret) {
1133 if (((SPEventContextClass *) parent_class)->root_handler) {
1134 ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);
1135 }
1136 }
1138 return ret;
1139 }
1142 static void sp_flood_finish(SPFloodContext *rc)
1143 {
1144 rc->_message_context->clear();
1146 if ( rc->item != NULL ) {
1147 SPDesktop * desktop;
1149 desktop = SP_EVENT_CONTEXT_DESKTOP(rc);
1151 SP_OBJECT(rc->item)->updateRepr();
1153 sp_canvas_end_forced_full_redraws(desktop->canvas);
1155 sp_desktop_selection(desktop)->set(rc->item);
1156 sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,
1157 _("Fill bounded area"));
1159 rc->item = NULL;
1160 }
1161 }
1163 void flood_channels_set_channels( gint channels )
1164 {
1165 prefs_set_int_attribute("tools.paintbucket", "channels", channels);
1166 }
1168 /*
1169 Local Variables:
1170 mode:c++
1171 c-file-style:"stroustrup"
1172 c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +))
1173 indent-tabs-mode:nil
1174 fill-column:99
1175 End:
1176 */
1177 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99 :