Code

Some additional Paint Bucket optimizations and cleanup
[inkscape.git] / src / flood-context.cpp
index 423f60980302a7012625f9511cd11478e8d5dcaf..2a18b3e3392f5d6874189f5d390447e00835d5b4 100644 (file)
@@ -125,7 +125,7 @@ static void sp_flood_context_init(SPFloodContext *flood_context)
     event_context->hot_y = 30;
     event_context->xp = 0;
     event_context->yp = 0;
-    event_context->tolerance = 0;
+    event_context->tolerance = 4;
     event_context->within_tolerance = false;
     event_context->item_to_select = NULL;
 
@@ -223,32 +223,22 @@ static void sp_flood_context_setup(SPEventContext *ec)
     rc->_message_context = new Inkscape::MessageContext((ec->desktop)->messageStack());
 }
 
-static void
+inline static void
 merge_pixel_with_background (unsigned char *orig, unsigned char *bg,
            unsigned char *base)
 {
+    int precalc_bg_alpha = (255 * (255 - bg[3])) / 255;
+    
     for (int i = 0; i < 3; i++) {
-        base[i] = (255 * (255 - bg[3])) / 255 + (bg[i] * bg[3]) / 255;
+        base[i] = precalc_bg_alpha + (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;
 }
 
-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;
 
@@ -264,7 +254,18 @@ GList * flood_channels_dropdown_items_list() {
     return glist;
 }
 
-static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *dtc, int threshold, PaintBucketChannels method) {
+GList * flood_autogap_dropdown_items_list() {
+    GList *glist = NULL;
+
+    glist = g_list_append (glist, _("None"));
+    glist = g_list_append (glist, _("Small"));
+    glist = g_list_append (glist, _("Medium"));
+    glist = g_list_append (glist, _("Large"));
+
+    return glist;
+}
+
+static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *merged_orig_pixel, unsigned char *dtc, int threshold, PaintBucketChannels method) {
     int diff = 0;
     float hsl_check[3], hsl_orig[3];
     
@@ -285,14 +286,12 @@ static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned c
         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];
+            unsigned char merged_check[3];
             
-            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]);
+              diff += (int)abs(merged_check[i] - merged_orig_pixel[i]);
             }
             return ((diff / 3) <= ((threshold * 3) / 4));
         
@@ -307,19 +306,51 @@ static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned c
     return false;
 }
 
-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) {
-    unsigned char *t = get_pixel(px, x, y, width);
-    if (compare_pixels(t, orig, dtc, threshold, method)) {
-        unsigned char *trace_t = get_pixel(trace_px, x, y, width);
-        if (trace_t[3] != 255 && trace_t[0] != 255) {
-            if (fill_switch) {
-                fill_queue->push_back(NR::Point(x, y));
-                trace_t[0] = 255;
-            }
+static inline bool is_pixel_checked(unsigned char *t) { return t[0] == 1; }
+static inline bool is_pixel_queued(unsigned char *t) { return t[1] == 1; }
+static inline bool is_pixel_paintability_checked(unsigned char *t) { return t[2] != 0; }
+static inline bool is_pixel_paintable(unsigned char *t) { return t[2] == 1; }
+static inline bool is_pixel_colored(unsigned char *t) { return t[3] == 255; }
+
+static inline void mark_pixel_checked(unsigned char *t) { t[0] = 1; }
+static inline void mark_pixel_unchecked(unsigned char *t) { t[0] = 0; }
+static inline void mark_pixel_queued(unsigned char *t) { t[1] = 1; }
+static inline void mark_pixel_paintable(unsigned char *t) { t[2] = 1; }
+static inline void mark_pixel_not_paintable(unsigned char *t) { t[2] = 2; }
+static inline void mark_pixel_colored(unsigned char *t) { t[3] = 255; }
+
+static inline void clear_pixel_paintability(unsigned char *t) { t[2] = 0; }
+
+struct bitmap_coords_info {
+    bool is_left;
+    unsigned int x;
+    unsigned int y;
+    int y_limit;
+    unsigned int width;
+    unsigned int height;
+    unsigned int threshold;
+    unsigned int radius;
+    PaintBucketChannels method;
+    unsigned char *dtc;
+    unsigned char *merged_orig_pixel;
+    NR::Rect bbox;
+    NR::Rect screen;
+    unsigned int max_queue_size;
+};
+
+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) {
+    if (is_pixel_paintability_checked(trace_t)) {
+        return is_pixel_paintable(trace_t);
+    } else {
+        unsigned char *t = get_pixel(px, x, y, bci.width);
+        if (compare_pixels(t, orig_color, bci.merged_orig_pixel, bci.dtc, bci.threshold, bci.method)) {
+            mark_pixel_paintable(trace_t);
+            return true;
+        } else {
+            mark_pixel_not_paintable(trace_t);
+            return false;
         }
-        return false;
     }
-    return true;
 }
 
 static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform, bool union_with_selection) {
@@ -437,61 +468,156 @@ static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform, bo
     }
 }
 
-struct bitmap_coords_info {
-    bool is_left;
-    int x;
-    int y;
-    int y_limit;
-    int width;
-    int threshold;
-    PaintBucketChannels method;
-    unsigned char *dtc;
-    bool top_fill;
-    bool bottom_fill;
-    NR::Rect bbox;
-    NR::Rect screen;
-};
-
 enum ScanlineCheckResult {
     SCANLINE_CHECK_OK,
     SCANLINE_CHECK_ABORTED,
     SCANLINE_CHECK_BOUNDARY
 };
 
+static bool coords_in_range(unsigned int x, unsigned int y, bitmap_coords_info bci) {
+    return (x < bci.width) &&
+           (y < bci.height);
+}
+
+#define PAINT_DIRECTION_LEFT 1
+#define PAINT_DIRECTION_RIGHT 2
+#define PAINT_DIRECTION_UP 4
+#define PAINT_DIRECTION_DOWN 8
+#define PAINT_DIRECTION_ALL 15
+
+static unsigned int paint_pixel(guchar *px, guchar *trace_px, unsigned char *orig_color, bitmap_coords_info bci, unsigned char *original_point_trace_t) {
+    if (bci.radius == 0) {
+        mark_pixel_colored(original_point_trace_t); 
+        return PAINT_DIRECTION_ALL;
+    } else {
+        unsigned char *trace_t;
+  
+        bool can_paint_up = true;
+        bool can_paint_down = true;
+        bool can_paint_left = true;
+        bool can_paint_right = true;
+      
+        for (int y = -bci.radius; y <= (int)bci.radius; y++) {
+            int ty = bci.y + y;
+            for (int x = -bci.radius; x <= (int)bci.radius; x++) {
+                int tx = bci.x + x;
+                
+                if (coords_in_range(tx, ty, bci)) {
+                    trace_t = get_pixel(trace_px, tx, ty, bci.width);
+                    if (!is_pixel_colored(trace_t)) {
+                        if (check_if_pixel_is_paintable(px, trace_t, tx, ty, orig_color, bci)) {
+                            mark_pixel_colored(trace_t); 
+                        } else {
+                            if (x < 0) { can_paint_left = false; }
+                            if (x > 0) { can_paint_right = false; }
+                            if (y < 0) { can_paint_up = false; }
+                            if (y > 0) { can_paint_down = false; }
+                        }
+                    }
+                }
+            }
+        }
+    
+        unsigned int paint_directions = 0;
+        if (can_paint_left) { paint_directions += PAINT_DIRECTION_LEFT; }
+        if (can_paint_right) { paint_directions += PAINT_DIRECTION_RIGHT; }
+        if (can_paint_up) { paint_directions += PAINT_DIRECTION_UP; }
+        if (can_paint_down) { paint_directions += PAINT_DIRECTION_DOWN; }
+        
+        return paint_directions;
+    }
+}
+
 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) {
     bool aborted = false;
     bool reached_screen_boundary = false;
     bool ok;
   
     bool keep_tracing;
-    unsigned char *t, *trace_t;
-  
+    bool initial_paint = true;
+    
+    unsigned char *current_trace_t = get_pixel(trace_px, bci.x, bci.y, bci.width);
+    unsigned int paint_directions;
+    
+    bool currently_painting_top = false;
+    bool currently_painting_bottom = false;
+    
+    unsigned int pix_width = bci.width * 4;
+    
     do {
         ok = false;
         if (bci.is_left) {
-            keep_tracing = (bci.x >= 0);
+            keep_tracing = (bci.x != 0);
         } else {
             keep_tracing = (bci.x < bci.width);
         }
         
         if (keep_tracing) {
-            t = get_pixel(px, bci.x, bci.y, bci.width);
-            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.dtc, bci.x, bci.y - 1, bci.width, bci.threshold, bci.method, bci.top_fill);
+            if (check_if_pixel_is_paintable(px, current_trace_t, bci.x, bci.y, orig_color, bci)) {
+                paint_directions = paint_pixel(px, trace_px, orig_color, bci, current_trace_t);
+                if (bci.radius == 0) {
+                    mark_pixel_checked(current_trace_t);
                 }
-                if (bci.y < bci.y_limit) { 
-                    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 (paint_directions & PAINT_DIRECTION_UP) { 
+                    unsigned int ty = bci.y - 1;
+                    if (ty > 0) {
+                        unsigned char *trace_t = current_trace_t - pix_width;
+                        if (!is_pixel_queued(trace_t)) {
+                            bool ok_to_paint = check_if_pixel_is_paintable(px, trace_t, bci.x, ty, orig_color, bci);
+                            
+                            if (initial_paint) { currently_painting_top = !ok_to_paint; }
+                            
+                            if (ok_to_paint && (!currently_painting_top)) {
+                                currently_painting_top = true;
+                                if ((fill_queue->size() < bci.max_queue_size)) {
+                                    fill_queue->push_back(NR::Point(bci.x, ty));
+                                    mark_pixel_queued(trace_t);
+                                }
+                            }
+                            if ((!ok_to_paint) && currently_painting_top) {
+                                currently_painting_top = false;
+                            }
+                        }
+                    }
                 }
+                
+                if (paint_directions & PAINT_DIRECTION_DOWN) { 
+                    unsigned int ty = bci.y + 1;
+                    if (ty < bci.height) {
+                        unsigned char *trace_t = current_trace_t + pix_width;
+                        if (!is_pixel_queued(trace_t)) {
+                            bool ok_to_paint = check_if_pixel_is_paintable(px, trace_t, bci.x, ty, orig_color, bci);
+                            
+                            if (initial_paint) { currently_painting_bottom = !ok_to_paint; }
+                            
+                            if (ok_to_paint && (!currently_painting_bottom)) {
+                                currently_painting_bottom = true;
+                                if ((fill_queue->size() < bci.max_queue_size)) {
+                                    fill_queue->push_back(NR::Point(bci.x, ty));
+                                    mark_pixel_queued(trace_t);
+                                }
+                            }
+                            if ((!ok_to_paint) && currently_painting_bottom) {
+                                currently_painting_bottom = false;
+                            }
+                        }
+                    }
+                }
+                
                 if (bci.is_left) {
-                    bci.x--;
+                    if (paint_directions & PAINT_DIRECTION_LEFT) {
+                        bci.x -= 1; current_trace_t -= 4;
+                        ok = true;
+                    }
                 } else {
-                    bci.x++;
+                    if (paint_directions & PAINT_DIRECTION_RIGHT) {
+                        bci.x += 1; current_trace_t += 4;
+                        ok = true;
+                    }
                 }
-                ok = true;
+                
+                initial_paint = false;
             }
         } else {
             if (bci.bbox.min()[NR::X] > bci.screen.min()[NR::X]) {
@@ -530,8 +656,8 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
 
     NR::Rect screen = desktop->get_display_area();
 
-    int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
-    int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
+    unsigned int width = (int)ceil(screen.extent(NR::X) * zoom_scale * padding);
+    unsigned int height = (int)ceil(screen.extent(NR::Y) * zoom_scale * padding);
 
     NR::Point origin(screen.min()[NR::X],
                      sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);
@@ -573,9 +699,9 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
     dtc[2] = NR_RGBA32_B(bgcolor);
     dtc[3] = NR_RGBA32_A(bgcolor);
     
-    for (int fy = 0; fy < height; fy++) {
+    for (unsigned int fy = 0; fy < height; fy++) {
         guchar *p = NR_PIXBLOCK_PX(&B) + fy * B.rs;
-        for (int fx = 0; fx < width; fx++) {
+        for (unsigned int fx = 0; fx < width; fx++) {
             for (int i = 0; i < 4; i++) { 
                 *p++ = dtc[i];
             }
@@ -649,11 +775,16 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
     
     bci.y_limit = y_limit;
     bci.width = width;
+    bci.height = height;
     bci.threshold = threshold;
     bci.method = method;
     bci.bbox = *bbox;
     bci.screen = screen;
     bci.dtc = dtc;
+    bci.radius = prefs_get_int_attribute_limited("tools.paintbucket", "autogap", 0, 0, 3);
+    bci.max_queue_size = width * height;
+
+    bool first_run = true;
 
     while (!color_queue.empty() && !aborted) {
         NR::Point color_point = color_queue.front();
@@ -663,53 +794,62 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
         unsigned char orig_color[4];
         for (int i = 0; i < 4; i++) { orig_color[i] = orig_px[i]; }
         
-        unsigned char merged_orig[4];
+        unsigned char merged_orig[3];
     
         merge_pixel_with_background(orig_color, dtc, merged_orig);
         
-        fill_queue.push_front(color_point);
+        bci.merged_orig_pixel = merged_orig;
+        
+        unsigned char *trace_t = get_pixel(trace_px, (int)color_point[NR::X], (int)color_point[NR::Y], width);
+        if (!is_pixel_checked(trace_t) && !is_pixel_colored(trace_t)) {
+            if (check_if_pixel_is_paintable(px, trace_px, (int)color_point[NR::X], (int)color_point[NR::Y], orig_color, bci)) {
+                fill_queue.push_front(color_point);
+                
+                if (!first_run) {
+                    for (unsigned int y = 0; y < height; y++) {
+                        trace_t = get_pixel(trace_px, 0, y, width);
+                        for (unsigned int x = 0; x < width; x++) {
+                            clear_pixel_paintability(trace_t);
+                            trace_t += 4;
+                        }
+                    }
+                }
+                first_run = false;
+            }
+        }
         
         while (!fill_queue.empty() && !aborted) {
             NR::Point cp = fill_queue.front();
             fill_queue.pop_front();
             
-            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, dtc, threshold, method)) {
+            unsigned char *trace_t = get_pixel(trace_px, (int)cp[NR::X], (int)cp[NR::Y], width);
+            if (!is_pixel_checked(trace_t)) {
+                mark_pixel_checked(trace_t);
                 int x = (int)cp[NR::X];
                 int y = (int)cp[NR::Y];
                 
-                bool top_fill = true;
-                bool bottom_fill = true;
-                
-                if (y > 0) { 
-                    top_fill = try_add_to_queue(&fill_queue, px, trace_px, orig_color, dtc, x, y - 1, width, threshold, method, top_fill);
-                } else {
+                if (y == 0) {
                     if (bbox->min()[NR::Y] > screen.min()[NR::Y]) {
                         aborted = true; break;
                     } else {
                         reached_screen_boundary = true;
                     }
                 }
-                if (y < y_limit) { 
-                    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;
-                      } else {
-                          reached_screen_boundary = true;
-                      }
+
+                if (y == y_limit) {
+                    if (bbox->max()[NR::Y] < screen.max()[NR::Y]) {
+                        aborted = true; break;
+                    } else {
+                        reached_screen_boundary = true;
+                    }
                 }
-                
+
                 bci.is_left = true;
                 bci.x = x;
                 bci.y = y;
-                bci.top_fill = top_fill;
-                bci.bottom_fill = bottom_fill;
-                
+
                 ScanlineCheckResult result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
-                
+
                 switch (result) {
                     case SCANLINE_CHECK_ABORTED:
                         aborted = true;
@@ -720,24 +860,27 @@ static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *even
                     default:
                         break;
                 }
-                
-                bci.is_left = false;
-                bci.x = x + 1;
-                bci.y = y;
-                bci.top_fill = top_fill;
-                bci.bottom_fill = bottom_fill;
-                
-                result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
-                
-                switch (result) {
-                    case SCANLINE_CHECK_ABORTED:
-                        aborted = true;
-                        break;
-                    case SCANLINE_CHECK_BOUNDARY:
-                        reached_screen_boundary = true;
-                        break;
-                    default:
-                        break;
+
+                if (bci.x < width) {
+                    trace_t += 4;
+                    if (!is_pixel_checked(trace_t) && !is_pixel_queued(trace_t)) {
+                        mark_pixel_checked(trace_t);
+                        bci.is_left = false;
+                        bci.x = x + 1;
+        
+                        result = perform_bitmap_scanline_check(&fill_queue, px, trace_px, orig_color, bci);
+        
+                        switch (result) {
+                            case SCANLINE_CHECK_ABORTED:
+                                aborted = true;
+                                break;
+                            case SCANLINE_CHECK_BOUNDARY:
+                                reached_screen_boundary = true;
+                                break;
+                            default:
+                                break;
+                        }
+                    }
                 }
             }
         }
@@ -817,16 +960,18 @@ static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEven
                 NR::Point const button_w(event->button.x,
                                         event->button.y);
     
-                // save drag origin
-                event_context->xp = (gint) button_w[NR::X];
-                event_context->yp = (gint) button_w[NR::Y];
-                event_context->within_tolerance = true;
-                
-                dragging = true;
-                
-                NR::Point const p(desktop->w2d(button_w));
-                Inkscape::Rubberband::get()->setMode(RUBBERBAND_MODE_TOUCHPATH);
-                Inkscape::Rubberband::get()->start(desktop, p);
+                if (Inkscape::have_viable_layer(desktop, event_context->defaultMessageContext())) {
+                    // save drag origin
+                    event_context->xp = (gint) button_w[NR::X];
+                    event_context->yp = (gint) button_w[NR::Y];
+                    event_context->within_tolerance = true;
+                    
+                    dragging = true;
+                    
+                    NR::Point const p(desktop->w2d(button_w));
+                    Inkscape::Rubberband::get()->setMode(RUBBERBAND_MODE_TOUCHPATH);
+                    Inkscape::Rubberband::get()->start(desktop, p);
+                }
             }
         }
     case GDK_MOTION_NOTIFY: