#define __SP_FLOOD_CONTEXT_C__

#include "config.h"

#include <gdk/gdkkeysyms.h>

#include <queue>

#include "macros.h"

#include "display/sp-canvas.h"

#include "document.h"

#include "sp-namedview.h"

#include "sp-object.h"

#include "sp-rect.h"

#include "selection.h"

#include "selection-chemistry.h"

#include "desktop-handles.h"

#include "snap.h"

#include "desktop.h"

#include "desktop-style.h"

#include "message-stack.h"

#include "message-context.h"

#include "pixmaps/cursor-paintbucket.xpm"

#include "flood-context.h"

#include "sp-metrics.h"

#include <glibmm/i18n.h>

#include "object-edit.h"

#include "xml/repr.h"

#include "xml/node-event-vector.h"

#include "prefs-utils.h"

#include "context-fns.h"

#include "display/nr-arena-item.h"

#include "display/nr-arena.h"

#include "display/nr-arena-image.h"

#include "display/canvas-arena.h"

#include "libnr/nr-pixops.h"

#include "libnr/nr-matrix-rotate-ops.h"

#include "libnr/nr-matrix-translate-ops.h"

#include "libnr/nr-rotate-fns.h"

#include "libnr/nr-scale-ops.h"

#include "libnr/nr-scale-translate-ops.h"

#include "libnr/nr-translate-matrix-ops.h"

#include "libnr/nr-translate-scale-ops.h"

#include "libnr/nr-matrix-ops.h"

#include "sp-item.h"

#include "sp-root.h"

#include "sp-defs.h"

#include "splivarot.h"

#include "livarot/Path.h"

#include "livarot/Shape.h"

#include "libnr/n-art-bpath.h"

#include "svg/svg.h"

#include "trace/trace.h"

#include "trace/potrace/inkscape-potrace.h"

static void sp_flood_context_class_init(SPFloodContextClass *klass);

static void sp_flood_context_init(SPFloodContext *flood_context);

static void sp_flood_context_dispose(GObject *object);

static void sp_flood_context_setup(SPEventContext *ec);

static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event);

static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event);

static void sp_flood_finish(SPFloodContext *rc);

static SPEventContextClass *parent_class;

GtkType sp_flood_context_get_type()

{

static GType type = 0;

if (!type) {

GTypeInfo info = {

sizeof(SPFloodContextClass),

NULL, NULL,

(GClassInitFunc) sp_flood_context_class_init,

NULL, NULL,

sizeof(SPFloodContext),

4,

(GInstanceInitFunc) sp_flood_context_init,

NULL, /* value_table */

};

type = g_type_register_static(SP_TYPE_EVENT_CONTEXT, "SPFloodContext", &info, (GTypeFlags) 0);

}

return type;

}

static void sp_flood_context_class_init(SPFloodContextClass *klass)

{

GObjectClass *object_class = (GObjectClass *) klass;

SPEventContextClass *event_context_class = (SPEventContextClass *) klass;

parent_class = (SPEventContextClass *) g_type_class_peek_parent(klass);

object_class->dispose = sp_flood_context_dispose;

event_context_class->setup = sp_flood_context_setup;

event_context_class->root_handler = sp_flood_context_root_handler;

event_context_class->item_handler = sp_flood_context_item_handler;

}

static void sp_flood_context_init(SPFloodContext *flood_context)

{

SPEventContext *event_context = SP_EVENT_CONTEXT(flood_context);

event_context->cursor_shape = cursor_paintbucket_xpm;

event_context->hot_x = 11;

event_context->hot_y = 30;

event_context->xp = 0;

event_context->yp = 0;

event_context->tolerance = 0;

event_context->within_tolerance = false;

event_context->item_to_select = NULL;

event_context->shape_repr = NULL;

event_context->shape_knot_holder = NULL;

flood_context->item = NULL;

new (&flood_context->sel_changed_connection) sigc::connection();

}

static void sp_flood_context_dispose(GObject *object)

{

SPFloodContext *rc = SP_FLOOD_CONTEXT(object);

SPEventContext *ec = SP_EVENT_CONTEXT(object);

rc->sel_changed_connection.disconnect();

rc->sel_changed_connection.~connection();

/* fixme: This is necessary because we do not grab */

if (rc->item) {

sp_flood_finish(rc);

}

if (ec->shape_repr) { // remove old listener

sp_repr_remove_listener_by_data(ec->shape_repr, ec);

Inkscape::GC::release(ec->shape_repr);

ec->shape_repr = 0;

}

if (rc->_message_context) {

delete rc->_message_context;

}

G_OBJECT_CLASS(parent_class)->dispose(object);

}

static Inkscape::XML::NodeEventVector ec_shape_repr_events = {

NULL, /* child_added */

NULL, /* child_removed */

ec_shape_event_attr_changed,

NULL, /* content_changed */

NULL /* order_changed */

};

void sp_flood_context_selection_changed(Inkscape::Selection *selection, gpointer data)

{

SPFloodContext *rc = SP_FLOOD_CONTEXT(data);

SPEventContext *ec = SP_EVENT_CONTEXT(rc);

if (ec->shape_repr) { // remove old listener

sp_repr_remove_listener_by_data(ec->shape_repr, ec);

Inkscape::GC::release(ec->shape_repr);

ec->shape_repr = 0;

}

SPItem *item = selection->singleItem();

if (item) {

Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);

if (shape_repr) {

ec->shape_repr = shape_repr;

Inkscape::GC::anchor(shape_repr);

sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);

}

}

}

static void sp_flood_context_setup(SPEventContext *ec)

{

SPFloodContext *rc = SP_FLOOD_CONTEXT(ec);

if (((SPEventContextClass *) parent_class)->setup) {

((SPEventContextClass *) parent_class)->setup(ec);

}

SPItem *item = sp_desktop_selection(ec->desktop)->singleItem();

if (item) {

Inkscape::XML::Node *shape_repr = SP_OBJECT_REPR(item);

if (shape_repr) {

ec->shape_repr = shape_repr;

Inkscape::GC::anchor(shape_repr);

sp_repr_add_listener(shape_repr, &ec_shape_repr_events, ec);

}

}

rc->sel_changed_connection.disconnect();

rc->sel_changed_connection = sp_desktop_selection(ec->desktop)->connectChanged(

sigc::bind(sigc::ptr_fun(&sp_flood_context_selection_changed), (gpointer)rc)

);

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;

}

enum PaintBucketChannels {

FLOOD_CHANNELS_RGB,

FLOOD_CHANNELS_ALPHA

};

GList * flood_channels_dropdown_items_list() {

GList *glist = NULL;

glist = g_list_append (glist, _("Visible Colors"));

glist = g_list_append (glist, _("Alpha"));

return glist;

}

static bool compare_pixels(unsigned char *check, unsigned char *orig, unsigned char *dtc, int tolerance, PaintBucketChannels method) {

int diff = 0;

switch (method) {

case FLOOD_CHANNELS_ALPHA:

return ((int)abs(check[3] - orig[3]) <= (tolerance / 4));

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) <= ((tolerance * 3) / 4));

}

return false;

}

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 tolerance, PaintBucketChannels method, bool fill_switch) {

unsigned char *t = get_pixel(px, x, y, width);

if (compare_pixels(t, orig, dtc, tolerance, method)) {

unsigned char *trace_t = get_pixel(trace_px, x, y, width);

if (trace_t[3] != 255) {

if (fill_switch) {

fill_queue->push(NR::Point(x, y));

}

}

return false;

}

return true;

}

static void do_trace(GdkPixbuf *px, SPDesktop *desktop, NR::Matrix transform) {

SPDocument *document = sp_desktop_document(desktop);

Inkscape::Trace::Potrace::PotraceTracingEngine pte;

pte.setTraceType(Inkscape::Trace::Potrace::TRACE_BRIGHTNESS);

pte.setInvert(false);

Glib::RefPtr<Gdk::Pixbuf> pixbuf = Glib::wrap(px, true);

std::vector<Inkscape::Trace::TracingEngineResult> results = pte.trace(pixbuf);

Inkscape::XML::Node *layer_repr = SP_GROUP(desktop->currentLayer())->repr;

Inkscape::XML::Document *xml_doc = sp_document_repr_doc(desktop->doc());

long totalNodeCount = 0L;

double offset = prefs_get_double_attribute_limited("tools.paintbucket", "offset", 1.5, 0.0, 2.0);

for (unsigned int i=0 ; i<results.size() ; i++) {

Inkscape::Trace::TracingEngineResult result = results[i];

totalNodeCount += result.getNodeCount();

Inkscape::XML::Node *pathRepr = xml_doc->createElement("svg:path");

/* Set style */

sp_desktop_apply_style_tool (desktop, pathRepr, "tools.paintbucket", false);

NArtBpath *bpath = sp_svg_read_path(result.getPathData().c_str());

Path *path = bpath_to_Path(bpath);

g_free(bpath);

Shape *path_shape = new Shape();

path->ConvertWithBackData(0.03);

path->Fill(path_shape, 0);

delete path;

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);

Path *expanded_path = new Path();

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;

gchar *str = expanded_path->svg_dump_path();

delete expanded_path;

pathRepr->setAttribute("d", str);

g_free(str);

layer_repr->addChild(pathRepr, NULL);

SPObject *reprobj = document->getObjectByRepr(pathRepr);

if (reprobj) {

sp_item_write_transform(SP_ITEM(reprobj), pathRepr, transform, NULL);

// premultiply the item transform by the accumulated parent transform in the paste layer

NR::Matrix local = sp_item_i2doc_affine(SP_GROUP(desktop->currentLayer()));

if (!local.test_identity()) {

gchar const *t_str = pathRepr->attribute("transform");

NR::Matrix item_t (NR::identity());

if (t_str)

sp_svg_transform_read(t_str, &item_t);

item_t *= local.inverse();

// (we're dealing with unattached repr, so we write to its attr instead of using sp_item_set_transform)

gchar *affinestr=sp_svg_transform_write(item_t);

pathRepr->setAttribute("transform", affinestr);

g_free(affinestr);

}

Inkscape::Selection *selection = sp_desktop_selection(desktop);

selection->set(reprobj);

pathRepr->setPosition(-1);

}

Inkscape::GC::release(pathRepr);

}

}

struct bitmap_coords_info {

bool is_left;

int x;

int y;

int y_limit;

int width;

int tolerance;

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 ScanlineCheckResult perform_bitmap_scanline_check(std::queue<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;

do {

ok = false;

if (bci.is_left) {

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.tolerance, 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.tolerance, 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.dtc, bci.x, bci.y + 1, bci.width, bci.tolerance, bci.method, bci.bottom_fill);

}

if (bci.is_left) {

bci.x--;

} else {

bci.x++;

}

ok = true;

}

} else {

if (bci.bbox.min()[NR::X] > bci.screen.min()[NR::X]) {

aborted = true; break;

} else {

reached_screen_boundary = true;

}

}

} while (ok);

if (aborted) { return SCANLINE_CHECK_ABORTED; }

if (reached_screen_boundary) { return SCANLINE_CHECK_BOUNDARY; }

return SCANLINE_CHECK_OK;

}

static void sp_flood_do_flood_fill(SPEventContext *event_context, GdkEvent *event) {

SPDesktop *desktop = event_context->desktop;

SPDocument *document = sp_desktop_document(desktop);

/* Create new arena */

NRArena *arena = NRArena::create();

unsigned dkey = sp_item_display_key_new(1);

sp_document_ensure_up_to_date (document);

SPItem *document_root = SP_ITEM(SP_DOCUMENT_ROOT(document));

NR::Maybe<NR::Rect> bbox = document_root->getBounds(NR::identity());

if (!bbox) {

desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));

return;

}

double zoom_scale = desktop->current_zoom();

double padding = 1.6;

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);

NR::Point origin(screen.min()[NR::X],

sp_document_height(document) - screen.extent(NR::Y) - screen.min()[NR::Y]);

origin[NR::X] = origin[NR::X] + (screen.extent(NR::X) * ((1 - padding) / 2));

origin[NR::Y] = origin[NR::Y] + (screen.extent(NR::Y) * ((1 - padding) / 2));

NR::scale scale(zoom_scale, zoom_scale);

NR::Matrix affine = scale * NR::translate(-origin * scale);

/* Create ArenaItems and set transform */

NRArenaItem *root = sp_item_invoke_show(SP_ITEM(sp_document_root(document)), arena, dkey, SP_ITEM_SHOW_DISPLAY);

nr_arena_item_set_transform(NR_ARENA_ITEM(root), affine);

NRGC gc(NULL);

nr_matrix_set_identity(&gc.transform);

NRRectL final_bbox;

final_bbox.x0 = 0;

final_bbox.y0 = 0;//row;

final_bbox.x1 = width;

final_bbox.y1 = height;//row + num_rows;

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);

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);

// Hide items

sp_item_invoke_hide(SP_ITEM(sp_document_root(document)), dkey);

nr_arena_item_unref(root);

nr_object_unref((NRObject *) arena);

NR::Point pw = NR::Point(event->button.x / zoom_scale, sp_document_height(document) + (event->button.y / zoom_scale)) * affine;

pw[NR::X] = (int)MIN(width - 1, MAX(0, pw[NR::X]));

pw[NR::Y] = (int)MIN(height - 1, MAX(0, pw[NR::Y]));

guchar *trace_px = g_new(guchar, 4 * width * height);

memset(trace_px, 0x00, 4 * width * height);

std::queue<NR::Point> fill_queue;

fill_queue.push(pw);

bool aborted = false;

int y_limit = height - 1;

unsigned char orig_color[4];

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]; }

unsigned char merged_orig[4];

merge_pixel_with_background(orig_color, dtc, merged_orig);

int tolerance = (255 * prefs_get_int_attribute_limited("tools.paintbucket", "tolerance", 1, 0, 100)) / 100;

PaintBucketChannels method = (PaintBucketChannels)prefs_get_int_attribute("tools.paintbucket", "channels", 0);

bool reached_screen_boundary = false;

bitmap_coords_info bci;

bci.y_limit = y_limit;

bci.width = width;

bci.tolerance = tolerance;

bci.method = method;

bci.bbox = *bbox;

bci.screen = screen;

bci.dtc = dtc;

while (!fill_queue.empty() && !aborted) {

NR::Point cp = fill_queue.front();

fill_queue.pop();

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, tolerance, method)) {

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, tolerance, method, top_fill);

} else {

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, tolerance, method, bottom_fill);

} else {

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;

break;

case SCANLINE_CHECK_BOUNDARY:

reached_screen_boundary = true;

break;

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;

}

}

}

g_free(px);

if (aborted) {

g_free(trace_px);

desktop->messageStack()->flash(Inkscape::WARNING_MESSAGE, _("<b>Area is not bounded</b>, cannot fill."));

return;

}

if (reached_screen_boundary) {

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."));

}

GdkPixbuf* pixbuf = gdk_pixbuf_new_from_data(trace_px,

GDK_COLORSPACE_RGB,

TRUE,

8, width, height, width * 4,

(GdkPixbufDestroyNotify)g_free,

NULL);

NR::Matrix inverted_affine = NR::Matrix(affine).inverse();

do_trace(pixbuf, desktop, inverted_affine);

g_free(trace_px);

sp_document_done(document, SP_VERB_CONTEXT_PAINTBUCKET, _("Fill bounded area"));

}

static gint sp_flood_context_item_handler(SPEventContext *event_context, SPItem *item, GdkEvent *event)

{

gint ret = FALSE;

switch (event->type) {

case GDK_BUTTON_PRESS:

break;

// motion and release are always on root (why?)

default:

break;

}

if (((SPEventContextClass *) parent_class)->item_handler) {

ret = ((SPEventContextClass *) parent_class)->item_handler(event_context, item, event);

}

return ret;

}

static gint sp_flood_context_root_handler(SPEventContext *event_context, GdkEvent *event)

{

gint ret = FALSE;

switch (event->type) {

case GDK_BUTTON_PRESS:

if ( event->button.button == 1 ) {

sp_flood_do_flood_fill(event_context, event);

ret = TRUE;

}

break;

case GDK_KEY_PRESS:

switch (get_group0_keyval (&event->key)) {

case GDK_Up:

case GDK_Down:

case GDK_KP_Up:

case GDK_KP_Down:

// prevent the zoom field from activation

if (!MOD__CTRL_ONLY)

ret = TRUE;

break;

default:

break;

}

break;

default:

break;

}

if (!ret) {

if (((SPEventContextClass *) parent_class)->root_handler) {

ret = ((SPEventContextClass *) parent_class)->root_handler(event_context, event);

}

}

return ret;

}

static void sp_flood_finish(SPFloodContext *rc)

{

rc->_message_context->clear();

if ( rc->item != NULL ) {

SPDesktop * desktop;

desktop = SP_EVENT_CONTEXT_DESKTOP(rc);

SP_OBJECT(rc->item)->updateRepr();

sp_canvas_end_forced_full_redraws(desktop->canvas);

sp_desktop_selection(desktop)->set(rc->item);

sp_document_done(sp_desktop_document(desktop), SP_VERB_CONTEXT_PAINTBUCKET,

_("Fill bounded area"));

rc->item = NULL;

}

}

void flood_channels_changed(GtkComboBox *cbox, GtkWidget *tbl)

{

prefs_set_int_attribute("tools.paintbucket", "channels", (gint)gtk_combo_box_get_active(cbox));

}