src/flood-context.cpp

   1 #define __SP_FLOOD_CONTEXT_C__
   2
   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 */
  17
  18 #include "config.h"
  19
  20 #include <gdk/gdkkeysyms.h>
  21 #include <queue>
  22 #include <deque>
  23
  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"
  46
  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"
  68
  69 #include "trace/trace.h"
  70 #include "trace/potrace/inkscape-potrace.h"
  71
  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);
  75
  76 static void sp_flood_context_setup(SPEventContext *ec);
  77
  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);
  80
  81 static void sp_flood_finish(SPFloodContext *rc);
  82
  83 static SPEventContextClass *parent_class;
  84
  85
  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 }
 104
 105 static void sp_flood_context_class_init(SPFloodContextClass *klass)
 106 {
 107     GObjectClass *object_class = (GObjectClass *) klass;
 108     SPEventContextClass *event_context_class = (SPEventContextClass *) klass;
 109
 110     parent_class = (SPEventContextClass *) g_type_class_peek_parent(klass);
 111
 112     object_class->dispose = sp_flood_context_dispose;
 113
 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 }
 118
 119 static void sp_flood_context_init(SPFloodContext *flood_context)
 120 {
 121     SPEventContext *event_context = SP_EVENT_CONTEXT(flood_context);
 122
 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;
 131
 132     event_context->shape_repr = NULL;
 133     event_context->shape_knot_holder = NULL;
 134
 135     flood_context->item = NULL;
 136
 137     new (&flood_context->sel_changed_connection) sigc::connection();
 138 }
 139
 140 static void sp_flood_context_dispose(GObject *object)
 141 {
 142     SPFloodContext *rc = SP_FLOOD_CONTEXT(object);
 143     SPEventContext *ec = SP_EVENT_CONTEXT(object);
 144
 145     rc->sel_changed_connection.disconnect();
 146     rc->sel_changed_connection.~connection();
 147
 148     /* fixme: This is necessary because we do not grab */
 149     if (rc->item) {
 150         sp_flood_finish(rc);
 151     }
 152
 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     }
 158
 159     if (rc->_message_context) {
 160         delete rc->_message_context;
 161     }
 162
 163     G_OBJECT_CLASS(parent_class)->dispose(object);
 164 }
 165
 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 };
 173
 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);
 182
 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     }
 188
 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 }
 199
 200 static void sp_flood_context_setup(SPEventContext *ec)
 201 {
 202     SPFloodContext *rc = SP_FLOOD_CONTEXT(ec);
 203
 204     if (((SPEventContextClass *) parent_class)->setup) {
 205         ((SPEventContextClass *) parent_class)->setup(ec);
 206     }
 207
 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     }
 217
 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     );
 222
 223     rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack());
 224
 225     if (prefs_get_int_attribute("tools.paintbucket", "selcue", 0) != 0) {
 226         rc->enableSelectionCue();
 227     }
 228 }
 229
 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;
 235
 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 }
 241
 242 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
 243     return px + (x + y * width) * 4;
 244 }
 245
 246 GList * flood_channels_dropdown_items_list() {
 247     GList *glist = NULL;
 248
 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"));
 257
 258     return glist;
 259 }
 260
 261 GList * flood_autogap_dropdown_items_list() {
 262     GList *glist = NULL;
 263
 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"));
 268
 269     return glist;
 270 }
 271
 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];
 275
 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     }
 282
 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];
 294
 295             merge_pixel_with_background(check, dtc, merged_check);
 296
 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));
 301
 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     }
 309
 310     return false;
 311 }
 312
 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; }
 318
 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; }
 325
 326 static inline void clear_pixel_paintability(unsigned char *t) { t[2] = 0; }
 327
 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 };
 345
 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 }
 360
 361 static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform, bool union_with_selection) {
 362     SPDocument *document = sp_desktop_document(desktop);
 363
 364     Inkscape::Trace::Potrace::PotraceTracingEngine pte;
 365
 366     pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS);
 367     pte.setInvert(false);
 368
 369     Glib::RefPtr<Gdk::Pixbuf> pixbuf = Glib::wrap(px, true);
 370
 371     std::vector<Inkscape::Trace::TracingEngineResult> results = pte.trace(pixbuf);
 372
 373     Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr;
 374     Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc());
 375
 376     long totalNodeCount = 0L;
 377
 378     double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 0.0);
 379
 380     for (unsigned int i=0 ; i<results.size() ; i++) {
 381         Inkscape::Trace::TracingEngineResult result = results[i];
 382         totalNodeCount += result.getNodeCount();
 383
 384         Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");
 385         /* Set style */
 386         sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
 387
 388         NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
 389         Path *path = bpath_to_Path(bpath);
 390         g_free(bpath);
 391
 392         if (offset != 0) {
 393
 394             Shape *path_shape = new Shape();
 395
 396             path->ConvertWithBackData(0.03);
 397             path->Fill(path_shape, 0);
 398             delete path;
 399
 400             Shape *expanded_path_shape = new Shape();
 401
 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);
 405
 406             Path *expanded_path = new Path();
 407
 408             expanded_path->Reset();
 409             expanded_path_shape->ConvertToForme(expanded_path);
 410             expanded_path->ConvertEvenLines(1.0);
 411             expanded_path->Simplify(1.0);
 412
 413             delete path_shape;
 414             delete expanded_path_shape;
 415
 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             }
 426
 427             delete expanded_path;
 428
 429         } else {
 430             gchar *str = path->svg_dump_path();
 431             delete path;
 432             pathRepr->setAttribute("d", str);
 433             g_free(str);
 434         }
 435
 436         layer_repr->addChild(pathRepr, NULL);
 437
 438         SPObject *reprobj = document->getObjectByRepr(pathRepr);
 439         if (reprobj) {
 440             sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL);
 441
 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             }
 455
 456             Inkscape::Selection *selection = sp_desktop_selection(desktop);
 457
 458             pathRepr->setPosition(-1);
 459
 460             if (union_with_selection) {
 461                 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)));
 462                 selection->add(reprobj);
 463                 sp_selected_path_union_skip_undo();
 464             } else {
 465                 desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, _("Area filled, path with <b>%d</b> nodes created."), sp_nodes_in_path(SP_PATH(reprobj)));
 466                 selection->set(reprobj);
 467             }
 468
 469         }
 470
 471         Inkscape::GC::release(pathRepr);
 472
 473     }
 474 }
 475
 476 enum ScanlineCheckResult {
 477     SCANLINE_CHECK_OK,
 478     SCANLINE_CHECK_ABORTED,
 479     SCANLINE_CHECK_BOUNDARY
 480 };
 481
 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 }
 486
 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
 492
 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;
 499
 500         bool can_paint_up = true;
 501         bool can_paint_down = true;
 502         bool can_paint_left = true;
 503         bool can_paint_right = true;
 504
 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         }
 522
 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; }
 528
 529         return paint_directions;
 530     }
 531 }
 532
 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 }
 541
 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 }
 550
 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;
 555
 556     bool keep_tracing;
 557     bool initial_paint = true;
 558
 559     unsigned char *current_trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width);
 560     unsigned int paint_directions;
 561
 562     bool currently_painting_top = false;
 563     bool currently_painting_bottom = false;
 564
 565     unsigned int pix_width = bci.width * 4;
 566
 567     unsigned int top_ty = bci.y - 1;
 568     unsigned int bottom_ty = bci.y + 1;
 569
 570     bool can_paint_top = (top_ty > 0);
 571     bool can_paint_bottom = (bottom_ty < bci.height);
 572
 573     NR::Point t = fill_queue->front();
 574
 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         }
 582
 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                 }
 592
 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);
 598
 599                             if (initial_paint) { currently_painting_top = !ok_to_paint; }
 600
 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                 }
 611
 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);
 617
 618                             if (initial_paint) { currently_painting_bottom = !ok_to_paint; }
 619
 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                 }
 630
 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                 }
 642
 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);
 653
 654     if (aborted) { return SCANLINE_CHECK_ABORTED; }
 655     if (reached_screen_boundary) { return SCANLINE_CHECK_BOUNDARY; }
 656     return SCANLINE_CHECK_OK;
 657 }
 658
 659 static bool sort_fill_queue_vertical(NR::Point a, NR::Point b) {
 660     return a[NR::Y] > b[NR::Y];
 661 }
 662
 663 static bool sort_fill_queue_horizontal(NR::Point a, NR::Point b) {
 664     return a[NR::X] > b[NR::X];
 665 }
 666
 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);
 670
 671     /* Create new arena */
 672     NRArena *arena = NRArena::create();
 673     unsigned dkey = sp_item_display_key_new(1);
 674
 675     sp_document_ensure_up_to_date (document);
 676
 677     SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document));
 678     NR::Maybe<NR::Rect> bbox = document_root->getBounds(NR::identity());
 679
 680     if (!bbox) {
 681         desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
 682         return;
 683     }
 684
 685     double zoom_scale = desktop->current_zoom();
 686     double padding = 1.6;
 687
 688     NR::Rect screen = desktop->get_display_area();
 689
 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);
 692
 693     NR::Point origin(screen.min()[NR::X],
 694                      sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);
 695
 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));
 698
 699     NR::scale scale(zoom_scale, zoom_scale);
 700     NR::Matrix affine = scale * NR::translate(-origin * scale);
 701
 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);
 705
 706     NRGC gc(NULL);
 707     nr_matrix_set_identity(&gc.transform);
 708
 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;
 714
 715     nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE);
 716
 717     guchar *px = g_new(guchar, 4 * width * height);
 718
 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 );
 723
 724     SPNamedView *nv = sp_desktop_namedview(desktop);
 725     unsigned long bgcolor = nv->pagecolor;
 726
 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);
 732
 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     }
 741
 742     nr_arena_item_invoke_render(NULL, root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE );
 743     nr_pixblock_release(&B);
 744
 745     // Hide items
 746     sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey);
 747
 748     nr_arena_item_unref(root);
 749     nr_object_unref((NRObject *) arena);
 750
 751     guchar *trace_px = g_new(guchar, 4 * width * height);
 752     memset(trace_px, 0x00, 4 * width * height);
 753
 754     std::deque<NR::Point> fill_queue;
 755     std::queue<NR::Point> color_queue;
 756
 757     std::vector<NR::Point> fill_points;
 758
 759     bool aborted = false;
 760     int y_limit = height - 1;
 761
 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);
 764
 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     }
 778
 779     bitmap_coords_info bci;
 780
 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;
 792
 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     }
 799
 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;
 802
 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]));
 805
 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     }
 817
 818     bool reached_screen_boundary = false;
 819
 820     bool first_run = true;
 821
 822     unsigned long sort_size_threshold = 5;
 823
 824     while (!color_queue.empty() && !aborted) {
 825         NR::Point color_point = color_queue.front();
 826         color_queue.pop();
 827
 828         int cx = (int)color_point[NR::X];
 829         int cy = (int)color_point[NR::Y];
 830
 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]; }
 834
 835         unsigned char merged_orig[3];
 836
 837         merge_pixel_with_background(orig_color, dtc, merged_orig);
 838
 839         bci.merged_orig_pixel = merged_orig;
 840
 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);
 845
 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         }
 858
 859         unsigned long old_fill_queue_size = fill_queue.size();
 860
 861         while (!fill_queue.empty() && !aborted) {
 862             NR::Point cp = fill_queue.front();
 863
 864             if (bci.radius == 0) {
 865                 unsigned long new_fill_queue_size = fill_queue.size();
 866
 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);
 870
 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;
 875
 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                         }
 891
 892                         cp = fill_queue.front();
 893                     }
 894                 }
 895
 896                 old_fill_queue_size = new_fill_queue_size;
 897             }
 898
 899             fill_queue.pop_front();
 900
 901             int x = (int)cp[NR::X];
 902             int y = (int)cp[NR::Y];
 903
 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);
 907
 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                 }
 915
 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                 }
 923
 924                 bci.is_left = true;
 925                 bci.x = x;
 926                 bci.y = y;
 927
 928                 ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
 929
 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                 }
 940
 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;
 947
 948                         result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
 949
 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             }
 963
 964             bci.current_step++;
 965
 966             if (bci.current_step > bci.max_queue_size) {
 967                 aborted = true;
 968             }
 969         }
 970     }
 971
 972     g_free(px);
 973
 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     }
 979
 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     }
 983
 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);
 990
 991     NR::Matrix inverted_affine = NR::Matrix(affine).inverse();
 992
 993     do_trace(pixbuf, desktop, inverted_affine, union_with_selection);
 994
 995     g_free(trace_px);
 996
 997     sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));
 998 }
 999
1000 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)
1001 {
1002     gint ret = FALSE;
1003
1004     SPDesktop *desktop = event_context->desktop;
1005
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);
1011
1012             SPItem *item = sp_event_context_find_item (desktop, button_w, TRUE, TRUE);
1013
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     }
1024
1025     if (((SPEventContextClass *) parent_class)->item_handler) {
1026         ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);
1027     }
1028
1029     return ret;
1030 }
1031
1032 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
1033 {
1034     static bool dragging;
1035
1036     gint ret = FALSE;
1037     SPDesktop *desktop = event_context->desktop;
1038
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);
1045
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;
1051
1052                     dragging = true;
1053
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             }
1069
1070             event_context->within_tolerance = false;
1071
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;
1081
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();
1088
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;
1093
1094                     bool is_point_fill = event_context->within_tolerance;
1095                     bool is_touch_fill = event->button.state & GDK_MOD1_MASK;
1096
1097                     sp_flood_do_flood_fill(event_context, event, event->button.state & GDK_SHIFT_MASK, is_point_fill, is_touch_fill);
1098
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
1102
1103                     ret = TRUE;
1104                 }
1105
1106                 r->stop();
1107
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     }
1131
1132     if (!ret) {
1133         if (((SPEventContextClass *) parent_class)->root_handler) {
1134             ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);
1135         }
1136     }
1137
1138     return ret;
1139 }
1140
1141
1142 static void sp_flood_finish(SPFloodContext *rc)
1143 {
1144     rc->_message_context->clear();
1145
1146     if ( rc->item != NULL ) {
1147         SPDesktop * desktop;
1148
1149         desktop = SP_EVENT_CONTEXT_DESKTOP(rc);
1150
1151         SP_OBJECT(rc->item)->updateRepr();
1152
1153         sp_canvas_end_forced_full_redraws(desktop->canvas);
1154
1155         sp_desktop_selection(desktop)->set(rc->item);
1156         sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,
1157                         _("Fill bounded area"));
1158
1159         rc->item = NULL;
1160     }
1161 }
1162
1163 void flood_channels_set_channels( gint channels )
1164 {
1165     prefs_set_int_attribute("tools.paintbucket", "channels", channels);
1166 }
1167
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 :