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