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 = 0;
 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
 226 static void
 227 merge_pixel_with_background (unsigned char *orig, unsigned char *bg,
 228            unsigned char *base)
 229 {
 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;
 235 }
 236
 237 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
 238     return px + (x + y * width) * 4;
 239 }
 240
 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 };
 251
 252 GList * flood_channels_dropdown_items_list() {
 253     GList *glist = NULL;
 254
 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"));
 263
 264     return glist;
 265 }
 266
 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;
 308 }
 309
 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;
 323 }
 324
 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);
 332
 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());
 339
 340     long totalNodeCount = 0L;
 341
 342     double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 0.0);
 343
 344     for (unsigned int i=0 ; i<results.size() ; i++) {
 345         Inkscape::Trace::TracingEngineResult result = results[i];
 346         totalNodeCount += result.getNodeCount();
 347
 348         Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");
 349         /* Set style */
 350         sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
 351
 352         NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
 353         Path *path = bpath_to_Path(bpath);
 354         g_free(bpath);
 355
 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);
 369
 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             }
 390
 391             delete expanded_path;
 392
 393         } else {
 394             gchar *str = path->svg_dump_path();
 395             delete path;
 396             pathRepr->setAttribute("d", str);
 397             g_free(str);
 398         }
 399
 400         layer_repr->addChild(pathRepr, NULL);
 401
 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             }
 419
 420             Inkscape::Selection *selection = sp_desktop_selection(desktop);
 421
 422             pathRepr->setPosition(-1);
 423
 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             }
 432
 433         }
 434
 435         Inkscape::GC::release(pathRepr);
 436
 437     }
 438 }
 439
 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 };
 454
 455 enum ScanlineCheckResult {
 456     SCANLINE_CHECK_OK,
 457     SCANLINE_CHECK_ABORTED,
 458     SCANLINE_CHECK_BOUNDARY
 459 };
 460
 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;
 508 }
 509
 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);
 513
 514     /* Create new arena */
 515     NRArena *arena = NRArena::create();
 516     unsigned dkey = sp_item_display_key_new(1);
 517
 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());
 522
 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;
 530
 531     NR::Rect screen = desktop->get_display_area();
 532
 533     int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
 534     int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
 535
 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);
 548
 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);
 559
 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     }
 584
 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     }
 608
 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     }
 625
 626     bool aborted = false;
 627     int y_limit = height - 1;
 628
 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);
 631
 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       }
 645
 646     bool reached_screen_boundary = false;
 647
 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;
 657
 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         fill_queue.push_front(color_point);
 671
 672         while (!fill_queue.empty() && !aborted) {
 673             NR::Point cp = fill_queue.front();
 674             fill_queue.pop_front();
 675
 676             unsigned char *s = get_pixel(px, (int)cp[NR::X], (int)cp[NR::Y], width);
 677
 678             // same color at this point
 679             if (compare_pixels(s, orig_color, dtc, threshold, method)) {
 680                 int x = (int)cp[NR::X];
 681                 int y = (int)cp[NR::Y];
 682
 683                 bool top_fill = true;
 684                 bool bottom_fill = true;
 685
 686                 if (y > 0) {
 687                     top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y - 1, width, threshold, method, top_fill);
 688                 } else {
 689                     if (bbox->min()[NR::Y] > screen.min()[NR::Y]) {
 690                         aborted = true; break;
 691                     } else {
 692                         reached_screen_boundary = true;
 693                     }
 694                 }
 695                 if (y < y_limit) {
 696                     bottom_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y + 1, width, threshold, method, bottom_fill);
 697                   } else {
 698                       if (bbox->max()[NR::Y] < screen.max()[NR::Y]) {
 699                           aborted = true; break;
 700                       } else {
 701                           reached_screen_boundary = true;
 702                       }
 703                 }
 704
 705                 bci.is_left = true;
 706                 bci.x = x;
 707                 bci.y = y;
 708                 bci.top_fill = top_fill;
 709                 bci.bottom_fill = bottom_fill;
 710
 711                 ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
 712
 713                 switch (result) {
 714                     case SCANLINE_CHECK_ABORTED:
 715                         aborted = true;
 716                         break;
 717                     case SCANLINE_CHECK_BOUNDARY:
 718                         reached_screen_boundary = true;
 719                         break;
 720                     default:
 721                         break;
 722                 }
 723
 724                 bci.is_left = false;
 725                 bci.x = x + 1;
 726                 bci.y = y;
 727                 bci.top_fill = top_fill;
 728                 bci.bottom_fill = bottom_fill;
 729
 730                 result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
 731
 732                 switch (result) {
 733                     case SCANLINE_CHECK_ABORTED:
 734                         aborted = true;
 735                         break;
 736                     case SCANLINE_CHECK_BOUNDARY:
 737                         reached_screen_boundary = true;
 738                         break;
 739                     default:
 740                         break;
 741                 }
 742             }
 743         }
 744     }
 745
 746     g_free(px);
 747
 748     if (aborted) {
 749         g_free(trace_px);
 750         desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));
 751         return;
 752     }
 753
 754     if (reached_screen_boundary) {
 755         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."));
 756     }
 757
 758     GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px,
 759                                       GDK_COLORSPACE_RGB,
 760                                       TRUE,
 761                                       8, width, height, width * 4,
 762                                       (GdkPixbufDestroyNotify)g_free,
 763                                       NULL);
 764
 765     NR::Matrix inverted_affine = NR::Matrix(affine).inverse();
 766
 767     do_trace(pixbuf, desktop, inverted_affine, union_with_selection);
 768
 769     g_free(trace_px);
 770
 771     sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));
 772 }
 773
 774 static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)
 775 {
 776     gint ret = FALSE;
 777
 778     SPDesktop *desktop = event_context->desktop;
 779
 780     switch (event->type) {
 781     case GDK_BUTTON_PRESS:
 782         if (event->button.state & GDK_CONTROL_MASK) {
 783             NR::Point const button_w(event->button.x,
 784                                     event->button.y);
 785
 786             SPItem *item = sp_event_context_find_item (desktop, button_w, TRUE, TRUE);
 787
 788             Inkscape::XML::Node *pathRepr = SP_OBJECT_REPR(item);
 789             /* Set style */
 790             sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
 791             sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET, _("Set style on object"));
 792             ret = TRUE;
 793         }
 794         break;
 795     default:
 796         break;
 797     }
 798
 799     if (((SPEventContextClass *) parent_class)->item_handler) {
 800         ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);
 801     }
 802
 803     return ret;
 804 }
 805
 806 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
 807 {
 808     static bool dragging;
 809
 810     gint ret = FALSE;
 811     SPDesktop *desktop = event_context->desktop;
 812
 813     switch (event->type) {
 814     case GDK_BUTTON_PRESS:
 815         if ( event->button.button == 1 ) {
 816             if (!(event->button.state & GDK_CONTROL_MASK)) {
 817                 NR::Point const button_w(event->button.x,
 818                                         event->button.y);
 819
 820                 // save drag origin
 821                 event_context->xp = (gint) button_w[NR::X];
 822                 event_context->yp = (gint) button_w[NR::Y];
 823                 event_context->within_tolerance = true;
 824
 825                 dragging = true;
 826
 827                 NR::Point const p(desktop->w2d(button_w));
 828                 Inkscape::Rubberband::get()->setMode(RUBBERBAND_MODE_TOUCHPATH);
 829                 Inkscape::Rubberband::get()->start(desktop, p);
 830             }
 831         }
 832     case GDK_MOTION_NOTIFY:
 833         if ( dragging
 834              && ( event->motion.state & GDK_BUTTON1_MASK ) )
 835         {
 836             if ( event_context->within_tolerance
 837                  && ( abs( (gint) event->motion.x - event_context->xp ) < event_context->tolerance )
 838                  && ( abs( (gint) event->motion.y - event_context->yp ) < event_context->tolerance ) ) {
 839                 break; // do not drag if we're within tolerance from origin
 840             }
 841
 842             event_context->within_tolerance = false;
 843
 844             NR::Point const motion_pt(event->motion.x, event->motion.y);
 845             NR::Point const p(desktop->w2d(motion_pt));
 846             if (Inkscape::Rubberband::get()->is_started()) {
 847                 Inkscape::Rubberband::get()->move(p);
 848                 event_context->defaultMessageContext()->set(Inkscape::NORMAL_MESSAGE, _("<b>Draw over</b> areas to add to fill, hold <b>Alt</b> for touch fill"));
 849                 gobble_motion_events(GDK_BUTTON1_MASK);
 850             }
 851         }
 852         break;
 853
 854     case GDK_BUTTON_RELEASE:
 855         if ( event->button.button == 1 ) {
 856             Inkscape::Rubberband::Rubberband *r = Inkscape::Rubberband::get();
 857             if (r->is_started()) {
 858                 // set "busy" cursor
 859                 desktop->setWaitingCursor();
 860
 861                 if (SP_IS_EVENT_CONTEXT(event_context)) {
 862                     // Since setWaitingCursor runs main loop iterations, we may have already left this tool!
 863                     // So check if the tool is valid before doing anything
 864                     dragging = false;
 865
 866                     bool is_point_fill = event_context->within_tolerance;
 867                     bool is_touch_fill = event->button.state & GDK_MOD1_MASK;
 868
 869                     sp_flood_do_flood_fill(event_context, event, event->button.state & GDK_SHIFT_MASK, is_point_fill, is_touch_fill);
 870
 871                     desktop->clearWaitingCursor();
 872                     // restore cursor when done; note that it may already be different if e.g. user
 873                     // switched to another tool during interruptible tracing or drawing, in which case do nothing
 874
 875                     ret = TRUE;
 876                 }
 877
 878                 r->stop();
 879                 event_context->defaultMessageContext()->clear();
 880             }
 881         }
 882         break;
 883     case GDK_KEY_PRESS:
 884         switch (get_group0_keyval (&event->key)) {
 885         case GDK_Up:
 886         case GDK_Down:
 887         case GDK_KP_Up:
 888         case GDK_KP_Down:
 889             // prevent the zoom field from activation
 890             if (!MOD__CTRL_ONLY)
 891                 ret = TRUE;
 892             break;
 893         default:
 894             break;
 895         }
 896         break;
 897     default:
 898         break;
 899     }
 900
 901     if (!ret) {
 902         if (((SPEventContextClass *) parent_class)->root_handler) {
 903             ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);
 904         }
 905     }
 906
 907     return ret;
 908 }
 909
 910
 911 static void sp_flood_finish(SPFloodContext *rc)
 912 {
 913     rc->_message_context->clear();
 914
 915     if ( rc->item != NULL ) {
 916         SPDesktop * desktop;
 917
 918         desktop = SP_EVENT_CONTEXT_DESKTOP(rc);
 919
 920         SP_OBJECT(rc->item)->updateRepr();
 921
 922         sp_canvas_end_forced_full_redraws(desktop->canvas);
 923
 924         sp_desktop_selection(desktop)->set(rc->item);
 925         sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,
 926                         _("Fill bounded area"));
 927
 928         rc->item = NULL;
 929     }
 930 }
 931
 932 void flood_channels_set_channels( gint channels )
 933 {
 934     prefs_set_int_attribute("tools.paintbucket", "channels", channels);
 935 }
 936
 937 /*
 938   Local Variables:
 939   mode:c++
 940   c-file-style:"stroustrup"
 941   c-file-offsets:((innamespace . 0)(inline-open . 0)(case-label . +))
 942   indent-tabs-mode:nil
 943   fill-column:99
 944   End:
 945 */
 946 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99 :