Code

sodipodi:docbase finally goes packing, along with a lauris comment
[inkscape.git] / src / flood-context.cpp
index 35e053fe06944537fef8001ecf528b8ad25ee6b1..aa6eab7b36cb7390b76ac8cd697a73a5f333378f 100644 (file)
 #include "sp-item.h"
 #include "sp-root.h"
 #include "sp-defs.h"
+#include "sp-path.h"
 #include "splivarot.h"
 #include "livarot/Path.h"
 #include "livarot/Shape.h"
 #include "libnr/n-art-bpath.h"
 #include "svg/svg.h"
+#include "color.h"
 
 #include "trace/trace.h"
 #include "trace/potrace/inkscape-potrace.h"
@@ -223,21 +225,90 @@ static void sp_flood_context_setup(SPEventContext *ec)
     rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack());
 }
 
+static void merge_pixel_with_background(unsigned char *orig, unsigned char *bg, unsigned char *base) {
+  for (int i = 0; i < 3; i++) { 
+    base[i] = (255 * (255 - bg[3])) / 255 + (bg[i] * bg[3]) / 255;
+    base[i] = (base[i] * (255 - orig[3])) / 255 + (orig[i] * orig[3]) / 255;
+  }
+  base[3] = 255;
+}
+
 inline unsigned char * get_pixel(guchar *px, int x, int y, int width) {
   return px + (x + y * width) * 4;
 }
 
-static bool compare_pixels(unsigned char *a, unsigned char *b, int tolerance) {
+enum PaintBucketChannels {
+    FLOOD_CHANNELS_RGB,
+    FLOOD_CHANNELS_R,
+    FLOOD_CHANNELS_G,
+    FLOOD_CHANNELS_B,
+    FLOOD_CHANNELS_H,
+    FLOOD_CHANNELS_S,
+    FLOOD_CHANNELS_L,
+    FLOOD_CHANNELS_ALPHA
+};
+
+GList * flood_channels_dropdown_items_list() {
+    GList *glist = NULL;
+
+    glist = g_list_append (glist, _("Visible Colors"));
+    glist = g_list_append (glist, _("Red"));
+    glist = g_list_append (glist, _("Green"));
+    glist = g_list_append (glist, _("Blue"));
+    glist = g_list_append (glist, _("Hue"));
+    glist = g_list_append (glist, _("Saturation"));
+    glist = g_list_append (glist, _("Lightness"));
+    glist = g_list_append (glist, _("Alpha"));
+
+    return glist;
+}
+
+static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *dtc, int threshold, PaintBucketChannels method) {
   int diff = 0;
-  for (int i = 0; i < 4; i++) {
-    diff += (int)abs(a[i] - b[i]);
+  float hsl_check[3], hsl_orig[3];
+  
+  if ((method == FLOOD_CHANNELS_H) ||
+      (method == FLOOD_CHANNELS_S) ||
+      (method == FLOOD_CHANNELS_L)) {
+    sp_color_rgb_to_hsl_floatv(hsl_check, check[0] / 255.0, check[1] / 255.0, check[2] / 255.0);
+    sp_color_rgb_to_hsl_floatv(hsl_orig, orig[0] / 255.0, orig[1] / 255.0, orig[2] / 255.0);
   }
-  return ((diff / 4) <= tolerance);
+  
+  switch (method) {
+    case FLOOD_CHANNELS_ALPHA:
+      return ((int)abs(check[3] - orig[3]) <= threshold);
+    case FLOOD_CHANNELS_R:
+      return ((int)abs(check[0] - orig[0]) <= threshold);
+    case FLOOD_CHANNELS_G:
+      return ((int)abs(check[1] - orig[1]) <= threshold);
+    case FLOOD_CHANNELS_B:
+      return ((int)abs(check[2] - orig[2]) <= threshold);
+    case FLOOD_CHANNELS_RGB:
+      unsigned char merged_orig[4];
+      unsigned char merged_check[4];
+      
+      merge_pixel_with_background(orig, dtc, merged_orig);
+      merge_pixel_with_background(check, dtc, merged_check);
+      
+      for (int i = 0; i < 3; i++) {
+        diff += (int)abs(merged_check[i] - merged_orig[i]);
+      }
+      return ((diff / 3) <= ((threshold * 3) / 4));
+    
+    case FLOOD_CHANNELS_H:
+      return ((int)(fabs(hsl_check[0] - hsl_orig[0]) * 100.0) <= threshold);
+    case FLOOD_CHANNELS_S:
+      return ((int)(fabs(hsl_check[1] - hsl_orig[1]) * 100.0) <= threshold);
+    case FLOOD_CHANNELS_L:
+      return ((int)(fabs(hsl_check[2] - hsl_orig[2]) * 100.0) <= threshold);
+  }
+  
+  return false;
 }
 
-static bool try_add_to_queue(std::queue<NR::Point> *fill_queue, guchar *px, guchar *trace_px, unsigned char *orig, int x, int y, int width, int tolerance, bool fill_switch) {
+static bool try_add_to_queue(std::queue<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) {
   unsigned char *t = get_pixel(px, x, y, width);
-  if (compare_pixels(t, orig, tolerance)) {
+  if (compare_pixels(t, orig, dtc, threshold, method)) {
     unsigned char *trace_t = get_pixel(trace_px, x, y, width);
     if (trace_t[3] != 255) {
       if (fill_switch) {
@@ -266,7 +337,7 @@ static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) {
 
     long totalNodeCount = 0L;
 
-    double offset = prefs_get_double_attribute_limited("tools.paintbucket", "offset", 1.5, 0.0, 2.0);
+    double offset = prefs_get_double_attribute("tools.paintbucket", "offset", 0.0);
 
     for (unsigned int i=0 ; i<results.size() ; i++) {
         Inkscape::Trace::TracingEngineResult result = results[i];
@@ -279,33 +350,50 @@ static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) {
         NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());
         Path *path = bpath_to_Path(bpath);
         g_free(bpath);
+
+        if (offset != 0) {
         
-        Shape *path_shape = new Shape();
+            Shape *path_shape = new Shape();
         
-        path->ConvertWithBackData(0.03);
-        path->Fill(path_shape, 0);
-        delete path;
+            path->ConvertWithBackData(0.03);
+            path->Fill(path_shape, 0);
+            delete path;
         
-        Shape *expanded_path_shape = new Shape();
+            Shape *expanded_path_shape = new Shape();
         
-        expanded_path_shape->ConvertToShape(path_shape, fill_nonZero);
-        path_shape->MakeOffset(expanded_path_shape, offset, join_round, 4);
-        expanded_path_shape->ConvertToShape(path_shape, fill_positive);
+            expanded_path_shape->ConvertToShape(path_shape, fill_nonZero);
+            path_shape->MakeOffset(expanded_path_shape, offset * desktop->current_zoom(), join_round, 4);
+            expanded_path_shape->ConvertToShape(path_shape, fill_positive);
 
-        Path *expanded_path = new Path();
+            Path *expanded_path = new Path();
         
-        expanded_path->Reset();
-        expanded_path_shape->ConvertToForme(expanded_path);
-        expanded_path->ConvertEvenLines(1.0);
-        expanded_path->Simplify(1.0);
+            expanded_path->Reset();
+            expanded_path_shape->ConvertToForme(expanded_path);
+            expanded_path->ConvertEvenLines(1.0);
+            expanded_path->Simplify(1.0);
         
-        delete path_shape;
-        delete expanded_path_shape;
+            delete path_shape;
+            delete expanded_path_shape;
         
-        gchar *str = expanded_path->svg_dump_path();
-        delete expanded_path;
-        pathRepr->setAttribute("d", str);
-        g_free(str);
+            gchar *str = expanded_path->svg_dump_path();
+            if (str && *str) {
+                pathRepr->setAttribute("d", str);
+                g_free(str);
+            } else {
+                desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Too much inset</b>, the result is empty."));
+                Inkscape::GC::release(pathRepr);
+                g_free(str);
+                return;
+            }
+
+            delete expanded_path;
+
+        } else {
+            gchar *str = path->svg_dump_path();
+            delete path;
+            pathRepr->setAttribute("d", str);
+            g_free(str);
+        }
 
         layer_repr->addChild(pathRepr, NULL);
 
@@ -330,9 +418,12 @@ static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) {
             Inkscape::Selection *selection = sp_desktop_selection(desktop);
             selection->set(reprobj);
             pathRepr->setPosition(-1);
+
+            desktop->messageStack()->flashF(Inkscape::WARNING_MESSAGE, _("Area filled, path with <b>%d</b> nodes created."), sp_nodes_in_path(SP_PATH(reprobj)));
         }
         
         Inkscape::GC::release(pathRepr);
+
     }
 }
 
@@ -342,7 +433,9 @@ struct bitmap_coords_info {
   int y;
   int y_limit;
   int width;
-  int tolerance;
+  int threshold;
+  PaintBucketChannels method;
+  unsigned char *dtc;
   bool top_fill;
   bool bottom_fill;
   NR::Rect bbox;
@@ -373,15 +466,15 @@ static ScanlineCheckResult perform_bitmap_scanline_check(std::queue<NR::Point> *
         
         if (keep_tracing) {
             t = get_pixel(px, bci.x, bci.y, bci.width);
-            if (compare_pixels(t, orig_color, bci.tolerance)) {
+            if (compare_pixels(t, orig_color, bci.dtc, bci.threshold, bci.method)) {
                 for (int i = 0; i < 4; i++) { t[i] = 255 - t[i]; }
                 trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width);
                 trace_t[3] = 255; 
                 if (bci.y > 0) { 
-                    bci.top_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.x, bci.y - 1, bci.width, bci.tolerance, bci.top_fill);
+                    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);
                 }
                 if (bci.y < bci.y_limit) { 
-                    bci.bottom_fill = try_add_to_queue(fill_queue, px, trace_px, orig_color, bci.x, bci.y + 1, bci.width, bci.tolerance, bci.bottom_fill);
+                    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);
                 }
                 if (bci.is_left) {
                     bci.x--;
@@ -455,13 +548,31 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
     nr_arena_item_invoke_update(root, &final_bbox, &gc, NR_ARENA_ITEM_STATE_ALL, NR_ARENA_ITEM_STATE_NONE);
 
     guchar *px = g_new(guchar, 4 * width * height);
-    memset(px, 0x00, 4 * width * height);
-
+    //memset(px, 0x00, 4 * width * height);
+    
     NRPixBlock B;
     nr_pixblock_setup_extern( &B, NR_PIXBLOCK_MODE_R8G8B8A8N,
                               final_bbox.x0, final_bbox.y0, final_bbox.x1, final_bbox.y1,
                               px, 4 * width, FALSE, FALSE );
     
+    SPNamedView *nv = sp_desktop_namedview(desktop);
+    unsigned long bgcolor = nv->pagecolor;
+    
+    unsigned char dtc[4];
+    dtc[0] = NR_RGBA32_R(bgcolor);
+    dtc[1] = NR_RGBA32_G(bgcolor);
+    dtc[2] = NR_RGBA32_B(bgcolor);
+    dtc[3] = NR_RGBA32_A(bgcolor);
+    
+    for (int fy = 0; fy < height; fy++) {
+      guchar *p = NR_PIXBLOCK_PX(&B) + fy * B.rs;
+      for (int fx = 0; fx < width; fx++) {
+        for (int i = 0; i < 4; i++) { 
+          *p++ = dtc[i];
+        }
+      }
+    }
+
     nr_arena_item_invoke_render(NULL, root, &final_bbox, &B, NR_ARENA_ITEM_RENDER_NO_CACHE );
     nr_pixblock_release(&B);
     
@@ -489,7 +600,26 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
     unsigned char *orig_px = get_pixel(px, (int)pw[NR::X], (int)pw[NR::Y], width);
     for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; }
 
-    int tolerance = (255 * prefs_get_int_attribute_limited("tools.paintbucket", "tolerance", 1, 0, 100)) / 100;
+    unsigned char merged_orig[4];
+
+    merge_pixel_with_background(orig_color, dtc, merged_orig);
+    
+    PaintBucketChannels method = (PaintBucketChannels)prefs_get_int_attribute("tools.paintbucket", "channels", 0);
+    int threshold = prefs_get_int_attribute_limited("tools.paintbucket", "threshold", 1, 0, 100);
+
+    switch(method) {
+      case FLOOD_CHANNELS_ALPHA:
+      case FLOOD_CHANNELS_RGB:
+      case FLOOD_CHANNELS_R:
+      case FLOOD_CHANNELS_G:
+      case FLOOD_CHANNELS_B:
+        threshold = (255 * threshold) / 100;
+        break;
+      case FLOOD_CHANNELS_H:
+      case FLOOD_CHANNELS_S:
+      case FLOOD_CHANNELS_L:
+        break;
+    }
 
     bool reached_screen_boundary = false;
 
@@ -497,9 +627,11 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
     
     bci.y_limit = y_limit;
     bci.width = width;
-    bci.tolerance = tolerance;
+    bci.threshold = threshold;
+    bci.method = method;
     bci.bbox = *bbox;
     bci.screen = screen;
+    bci.dtc = dtc;
 
     while (!fill_queue.empty() && !aborted) {
       NR::Point cp = fill_queue.front();
@@ -507,7 +639,7 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
       unsigned char *s = get_pixel(px, (int)cp[NR::X], (int)cp[NR::Y], width);
       
       // same color at this point
-      if (compare_pixels(s, orig_color, tolerance)) {
+      if (compare_pixels(s, orig_color, dtc, threshold, method)) {
         int x = (int)cp[NR::X];
         int y = (int)cp[NR::Y];
         
@@ -515,7 +647,7 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
         bool bottom_fill = true;
         
         if (y > 0) { 
-          top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, x, y - 1, width, tolerance, top_fill);
+          top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y - 1, width, threshold, method, top_fill);
         } else {
           if (bbox->min()[NR::Y] > screen.min()[NR::Y]) {
             aborted = true; break;
@@ -524,7 +656,7 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
           }
         }
         if (y < y_limit) { 
-          bottom_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, x, y + 1, width, tolerance, bottom_fill);
+          bottom_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y + 1, width, threshold, method, bottom_fill);
         } else {
           if (bbox->max()[NR::Y] < screen.max()[NR::Y]) {
             aborted = true; break;
@@ -605,10 +737,22 @@ static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem
 {
     gint ret = FALSE;
 
+    SPDesktop *desktop = event_context->desktop;
+
     switch (event->type) {
     case GDK_BUTTON_PRESS:
+        if (event->button.state & GDK_CONTROL_MASK) {
+            NR::Point const button_w(event->button.x,
+                                     event->button.y);
+            
+            SPItem *item = sp_event_context_find_item (desktop, button_w, TRUE, TRUE);
+            
+            Inkscape::XML::Node *pathRepr = SP_OBJECT_REPR(item);
+            /* Set style */
+            sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);
+            ret = TRUE;
+        }
         break;
-        // motion and release are always on root (why?)
     default:
         break;
     }
@@ -623,12 +767,24 @@ static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem
 static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)
 {
     gint ret = FALSE;
+    SPDesktop *desktop = event_context->desktop;
+
     switch (event->type) {
     case GDK_BUTTON_PRESS:
         if ( event->button.button == 1 ) {
-            sp_flood_do_flood_fill(event_context, event);
-
-            ret = TRUE;
+            if (!(event->button.state & GDK_CONTROL_MASK)) {
+                // set "busy" cursor
+                GdkCursor *waiting = gdk_cursor_new(GDK_WATCH);
+                gdk_window_set_cursor(GTK_WIDGET(sp_desktop_canvas(desktop))->window, waiting);
+    
+                sp_flood_do_flood_fill(event_context, event);
+    
+                // restore cursor when done
+                gdk_window_set_cursor(GTK_WIDGET(sp_desktop_canvas(desktop))->window, event_context->cursor);
+                gdk_cursor_unref(waiting);
+    
+                ret = TRUE;
+            }
         }
         break;
     case GDK_KEY_PRESS:
@@ -680,6 +836,11 @@ static void sp_flood_finish(SPFloodContext *rc)
     }
 }
 
+void flood_channels_changed(GtkComboBox *cbox, GtkWidget *tbl)
+{
+    prefs_set_int_attribute("tools.paintbucket", "channels", (gint)gtk_combo_box_get_active(cbox));
+}
+
 /*
   Local Variables:
   mode:c++