polygon intersector: improved test routines
[swftools.git] / lib / gfxpoly / poly.c
index 48e92f9..72e72aa 100644 (file)
@@ -3,10 +3,12 @@
 #include <memory.h>
 #include <math.h>
 #include "../mem.h"
+#include "../types.h"
 #include "../q.h"
 #include "poly.h"
 #include "active.h"
 #include "xrow.h"
+#include "wind.h"
 
 //#define DEBUG
 //#undef assert
@@ -47,9 +49,12 @@ type_t point_type = {
 
 typedef struct _status {
     int y;
+    int num_polygons;
     actlist_t*actlist;
     heap_t*queue;
     edge_t*output;
+    xrow_t*xrow;
+    windrule_t*windrule;
 #ifdef DEBUG
     dict_t*seen_crossings; //list of crossing we saw so far
     dict_t*intersecting_segs; //list of segments intersecting in this scanline
@@ -80,25 +85,61 @@ int compare_events(const void*_a,const void*_b)
         return 0;
 }
 
-gfxpoly_t* gfxpoly_new()
+gfxpoly_t* gfxpoly_new(double gridsize)
 {
-    return 0;
+    gfxpoly_t*p = (gfxpoly_t*)rfx_calloc(sizeof(gfxpoly_t));
+    p->gridsize = gridsize;
+    return p;
 }
 void gfxpoly_destroy(gfxpoly_t*poly)
 {
-    edge_t* s = poly;
+    edge_t* s = poly->edges;
     while(s) {
         edge_t*next  = s->next;
         free(s);
         s = next;
     }
+    free(poly);
+}
+char gfxpoly_check(gfxpoly_t*poly)
+{
+    edge_t* s = poly->edges;
+    dict_t*d = dict_new2(&point_type);
+    while(s) {
+        if(!dict_contains(d, &s->a)) {
+            dict_put(d, &s->a, (void*)(ptroff_t)1);
+        } else {
+            int count = (ptroff_t)dict_lookup(d, &s->a);
+            dict_del(d, &s->a);
+            count++;
+            dict_put(d, &s->a, (void*)(ptroff_t)count);
+        }
+        if(!dict_contains(d, &s->b)) {
+            dict_put(d, &s->b, (void*)(ptroff_t)1);
+        } else {
+            int count = (ptroff_t)dict_lookup(d, &s->b);
+            dict_del(d, &s->b);
+            count++;
+            dict_put(d, &s->b, (void*)(ptroff_t)count);
+        }
+        s = s->next;
+    }
+    DICT_ITERATE_ITEMS(d, point_t*, p, void*, c) {
+        int count = (ptroff_t)c;
+        if(count&1) {
+            fprintf(stderr, "Point (%f,%f) occurs %d times\n", p->x*poly->gridsize, p->y*poly->gridsize, count);
+            return 0;
+        }
+    }
+    return 1;
 }
 
 void gfxpoly_dump(gfxpoly_t*poly)
 {
-    edge_t* s = (edge_t*)poly;
+    edge_t* s = poly->edges;
+    double g = poly->gridsize;
     while(s) {
-        fprintf(stderr, "(%d,%d) -> (%d,%d)\n", s->a.x, s->a.y, s->b.x, s->b.y);
+        fprintf(stderr, "(%f,%f) -> (%f,%f)\n", s->a.x*g, s->a.y*g, s->b.x*g, s->b.y*g);
         s = s->next;
     }
 }
@@ -112,7 +153,9 @@ inline static event_t event_new()
 
 void event_dump(event_t*e)
 {
-    if(e->type == EVENT_START) {
+    if(e->type == EVENT_HORIZONTAL) {
+        fprintf(stderr, "Horizontal [%d] (%d,%d) -> (%d,%d)\n", e->s1->nr, e->s1->a.x, e->s1->a.y, e->s1->b.x, e->s1->b.y);
+    } else if(e->type == EVENT_START) {
         fprintf(stderr, "event: segment [%d] starts at (%d,%d)\n", e->s1->nr, e->p.x, e->p.y);
     } else if(e->type == EVENT_END) {
         fprintf(stderr, "event: segment [%d] ends at (%d,%d)\n", e->s1->nr, e->p.x, e->p.y);
@@ -126,15 +169,24 @@ void event_dump(event_t*e)
 static inline max32(int32_t v1, int32_t v2) {return v1>v2?v1:v2;}
 static inline min32(int32_t v1, int32_t v2) {return v1<v2?v1:v2;}
 
-void segment_init(segment_t*s, int x1, int y1, int x2, int y2)
+void segment_init(segment_t*s, int x1, int y1, int x2, int y2, windstate_t windstate, int polygon_nr)
 {
-    assert(y1!=y2);
     if(y1<y2) {
         s->dir = DIR_DOWN;
-    } else {
+    } else if(y1>y2) {
         int x = x1;x1=x2;x2=x;
         int y = y1;y1=y2;y2=y;
         s->dir = DIR_UP;
+    } else {
+        /* up/down for horizontal segments is handled by "rotating"
+           them 90° anticlockwise in screen coordinates (tilt your head to 
+           the right) */
+        s->dir = DIR_UP;
+        if(x1>x2) {
+            s->dir = DIR_DOWN;
+            int x = x1;x1=x2;x2=x;
+            int y = y1;y1=y2;y2=y;
+        }
     }
     s->a.x = x1;
     s->a.y = y1;
@@ -145,9 +197,10 @@ void segment_init(segment_t*s, int x1, int y1, int x2, int y2)
     s->delta.x = x2-x1;
     s->delta.y = y2-y1;
     s->pos = s->a;
-    s->tmp = -1;
-#ifdef DEBUG
-    static int segment_count=0; //for debugging
+    s->polygon_nr = polygon_nr;
+#define XDEBUG
+#ifdef XDEBUG
+    static int segment_count=0;
     s->nr = segment_count++;
 #endif
 
@@ -170,10 +223,10 @@ void segment_init(segment_t*s, int x1, int y1, int x2, int y2)
     dict_init2(&s->scheduled_crossings, &ptr_type, 0);
 }
 
-segment_t* segment_new(int32_t x1, int32_t y1, int32_t x2, int32_t y2)
+segment_t* segment_new(int32_t x1, int32_t y1, int32_t x2, int32_t y2, windstate_t initial, int polygon_nr)
 {
     segment_t*s = (segment_t*)rfx_calloc(sizeof(segment_t));
-    segment_init(s, x1, y1, x2, y2);
+    segment_init(s, x1, y1, x2, y2, initial, polygon_nr);
     return s;
 }
 void segment_destroy(segment_t*s)
@@ -182,23 +235,27 @@ void segment_destroy(segment_t*s)
     free(s);
 }
 
-void gfxpoly_enqueue(edge_t*list, heap_t*queue)
+void gfxpoly_enqueue(edge_t*list, heap_t*queue, windstate_t initial, int polygon_nr)
 {
-    edge_t*l = list;
-    while(l) {
-        segment_t*s = segment_new(l->a.x, l->a.y, l->b.x, l->b.y);
+    edge_t*l;
+    for(l=list;l;l=l->next) {
+        if(l->a.x == l->b.x && 
+           l->a.y == l->b.y) {
+            fprintf(stderr, "Warning: intersector input contains zero-length segments\n");
+            continue;
+        }
+        segment_t*s = segment_new(l->a.x, l->a.y, l->b.x, l->b.y, initial, polygon_nr);
 #ifdef DEBUG
         fprintf(stderr, "[%d] (%d,%d) -> (%d,%d) %s\n",
                 s->nr, s->a.x, s->a.y, s->b.x, s->b.y,
                 s->dir==DIR_UP?"up":"down");
 #endif
         event_t e = event_new();
-        e.type = EVENT_START;
+        e.type = s->delta.y ? EVENT_START : EVENT_HORIZONTAL;
         e.p = s->a;
         e.s1 = s;
         e.s2 = 0;
         heap_put(queue, &e);
-        l = l->next;
     }
 }
 
@@ -242,6 +299,9 @@ void schedule_crossing(status_t*status, segment_t*s1, segment_t*s2)
     }
     
     if(dict_contains(&s1->scheduled_crossings, s2)) {
+        /* FIXME: this whole segment hashing thing is really slow */
+        //fprintf(stderr, "Encountered crossing between [%d] and [%d] twice\n", s1->nr, s2->nr);
+
         // we already know about this one
         return;
     }
@@ -369,27 +429,6 @@ void exchange_two(status_t*status, event_t*e)
         schedule_crossing(status, s1, right);
 }
 
-void insert_point_into_segment(status_t*status, segment_t*s, point_t p)
-{
-#ifdef DEBUG
-    if(s->pos.x == p.x && s->pos.y == p.y) {
-        fprintf(stderr, "Error: tried to add (%d,%d) to segment [%d] twice\n", p.x, p.y, s->nr);
-    }
-#endif
-    assert(s->pos.x != p.x || s->pos.y != p.y);
-#ifdef DEBUG
-    fprintf(stderr, "[%d] gets extra point (%d,%d)\n", s->nr, p.x, p.y);
-#endif
-
-    edge_t*e = malloc(sizeof(edge_t));
-    e->a = s->pos;
-    e->b = p;
-    e->next = status->output;
-    status->output = e;
-    
-    s->pos = p;
-}
-
 typedef struct _box {
     point_t left1, left2, right1, right2;
 } box_t;
@@ -402,7 +441,39 @@ static inline box_t box_new(int x, int y)
     box.left2.y = box.right2.y = y;
     return box;
 }
-        
+
+
+static void insert_point_into_segment(status_t*status, segment_t*s, point_t p)
+{
+    assert(s->pos.x != p.x || s->pos.y != p.y);
+
+#ifdef DEBUG
+    if(!dict_contains(status->segs_with_point, s))
+        dict_put(status->segs_with_point, s, 0);
+#endif
+
+    assert(s->fs_out_ok);
+    if(s->fs_out) {
+#ifdef DEBUG
+        fprintf(stderr, "[%d] receives next point (%d,%d) (drawing)\n", s->nr, p.x, p.y);
+#endif
+        // omit horizontal lines
+        if(s->pos.y != p.y) {
+            edge_t*e = malloc(sizeof(edge_t));
+            e->a = s->pos;
+            e->b = p;
+            assert(e->a.y != e->b.y);
+            e->next = status->output;
+            status->output = e;
+        }
+    } else {
+#ifdef DEBUG
+        fprintf(stderr, "[%d] receives next point (%d,%d) (omitting)\n", s->nr, p.x, p.y);
+#endif
+    }
+    s->pos = p;
+}
+
 /* possible optimizations:
    1.) keep two different active lists around, one for negative and one for
        positive slopes
@@ -411,23 +482,20 @@ static inline box_t box_new(int x, int y)
 */
 /*
    SLOPE_POSITIVE:
-       +     \ +        
+      \+     \ +        
 ------ I      \I       
       -I\----  I      
        I \   --I\---
        I  \    I \  -------
        +   \   +  \
 */
-void add_points_to_positively_sloped_segments(status_t*status, xrow_t*xrow, int32_t y)
+static void add_points_to_positively_sloped_segments(status_t*status, int32_t y)
 {
     int t;
-    for(t=0;t<xrow->num;t++) {
-        box_t box = box_new(xrow->x[t], y);
+    for(t=0;t<status->xrow->num;t++) {
+        box_t box = box_new(status->xrow->x[t], y);
         segment_t*seg = actlist_find(status->actlist, box.left2, box.left2);
-        if(seg) 
-            seg = seg->right;
-        else    
-            seg = actlist_leftmost(status->actlist);
+        seg = actlist_right(status->actlist, seg);
         while(seg) {
             if(seg->a.y == y) {
                 // this segment just started, ignore it
@@ -437,15 +505,12 @@ void add_points_to_positively_sloped_segments(status_t*status, xrow_t*xrow, int3
                 double d1 = LINE_EQ(box.right1, seg);
                 double d2 = LINE_EQ(box.right2, seg);
                 if(d1>=0 || d2>=0) {
-#ifdef DEBUG
-                    dict_put(status->segs_with_point, seg, 0);
-#endif
                     insert_point_into_segment(status, seg, box.right2);
                 } else {
                     break;
                 }
             }
-            seg = seg->right;
+            seg = actlist_right(status->actlist, seg);
         }
     }
 }
@@ -457,11 +522,11 @@ void add_points_to_positively_sloped_segments(status_t*status, xrow_t*xrow, int3
    |   I    | /I   /
    |  /+    |/ +  /
 */
-void add_points_to_negatively_sloped_segments(status_t*status, xrow_t*xrow, int32_t y)
+static void add_points_to_negatively_sloped_segments(status_t*status, int32_t y)
 {
     int t;
-    for(t=xrow->num-1;t>=0;t--) {
-        box_t box = box_new(xrow->x[t], y);
+    for(t=status->xrow->num-1;t>=0;t--) {
+        box_t box = box_new(status->xrow->x[t], y);
         segment_t*seg = actlist_find(status->actlist, box.right2, box.right2);
         while(seg) {
             if(seg->a.y == y) {
@@ -472,22 +537,84 @@ void add_points_to_negatively_sloped_segments(status_t*status, xrow_t*xrow, int3
                 double d1 = LINE_EQ(box.left1, seg);
                 double d2 = LINE_EQ(box.left2, seg);
                 if(d1<0 || d2<0) {
-#ifdef DEBUG
-                    dict_put(status->segs_with_point, seg, 0);
-#endif
                     insert_point_into_segment(status, seg, box.right2);
                 } else {
                     break;
                 }
             }
-            seg = seg->left;
+            seg = actlist_left(status->actlist, seg);
         }
     }
 }
 
+static void recalculate_windings(status_t*status)
+{
+    /* TODO: we could use some clever second linked list structure so that we
+             only need to process points we know we marked */
+    
+    segment_t*s = actlist_leftmost(status->actlist);
+    segment_t*last = 0;
+    while(s) {
+        windstate_t wind = last?last->wind:status->windrule->start(status->num_polygons);
+        s->wind = status->windrule->add(wind, s->fs, s->dir, s->polygon_nr);
+        s->fs_out = status->windrule->diff(&wind, &s->wind);
+        s->fs_out_ok = 1;
+#ifdef DEBUG
+        fprintf(stderr, "[%d] %s/%d/%s/%s ", s->nr, s->dir==DIR_UP?"up":"down", s->wind.wind_nr, s->wind.is_filled?"fill":"nofill", s->fs_out?"draw":"omit");
+#endif
+        last = s;
+        s = actlist_right(status->actlist, s);
+    }
+#ifdef DEBUG
+    fprintf(stderr, "\n");
+#endif
+
+}
+
+void intersect_with_horizontal(status_t*status, segment_t*h)
+{
+    segment_t* left = actlist_find(status->actlist, h->a, h->a);
+    segment_t* right = actlist_find(status->actlist, h->b, h->b);
+
+    segment_t* s = right;
+
+    while(s!=left) {
+        assert(s);
+        /*
+           x1 + ((x2-x1)*(y-y1)) / dy = 
+           (x1*(y2-y) + x2*(y-y1)) / dy
+        */
+        point_t p;
+        p.y = status->y;
+        p.x = XPOS(s, p.y);
+#ifdef DEBUG
+        fprintf(stderr, "...into [%d] (%d,%d) -> (%d,%d) at (%d,%d)\n", s->nr, 
+                s->a.x, s->a.y,
+                s->b.x, s->b.y,
+                p.x, p.y
+                );
+#endif
+        assert(p.x >= h->a.x);
+        assert(p.x <= h->b.x);
+        assert(s->delta.x > 0 && p.x >= s->a.x || s->delta.x <= 0 && p.x <= s->a.x);
+        assert(s->delta.x > 0 && p.x <= s->b.x || s->delta.x <= 0 && p.x >= s->b.x);
+        xrow_add(status->xrow, p.x);
+
+        s = actlist_left(status->actlist, s);
+    }
+    xrow_add(status->xrow, h->a.x);
+}
+
 void event_apply(status_t*status, event_t*e)
 {
     switch(e->type) {
+        case EVENT_HORIZONTAL: {
+#ifdef DEBUG
+            event_dump(e);
+#endif
+            intersect_with_horizontal(status, e->s1);
+            break;
+        }
         case EVENT_END: {
             //delete segment from list
             segment_t*s = e->s1;
@@ -520,12 +647,14 @@ void event_apply(status_t*status, event_t*e)
                 schedule_crossing(status, left, s);
             if(right)
                 schedule_crossing(status, s, right);
-
             schedule_endpoint(status, e->s1);
             break;
         }
         case EVENT_CROSS: {
-            // exchange two (or more) segments
+            // exchange two segments
+#ifdef DEBUG
+            event_dump(e);
+#endif
             if(actlist_right(status->actlist, e->s1) == e->s2 &&
                actlist_left(status->actlist, e->s2) == e->s1) {
                 exchange_two(status, e);
@@ -564,20 +693,23 @@ void check_status(status_t*status)
 }
 #endif
 
-edge_t* gfxpoly_process(edge_t*poly)
+gfxpoly_t* gfxpoly_process(gfxpoly_t*poly, windrule_t*windrule)
 {
     heap_t* queue = heap_new(sizeof(event_t), compare_events);
-    gfxpoly_enqueue(poly, queue);
+    
+    gfxpoly_enqueue(poly->edges, queue, windrule->start(1), /*polygon nr*/0);
+
     status_t status;
     memset(&status, 0, sizeof(status_t));
+    status.num_polygons = 1;
     status.queue = queue;
+    status.windrule = windrule;
     status.actlist = actlist_new();
 #ifdef DEBUG
     status.seen_crossings = dict_new2(&point_type);
-    gfxpoly_dump(poly);
 #endif
     
-    xrow_t*xrow = xrow_new();
+    status.xrow = xrow_new();
 
     event_t*e = heap_chopmax(queue);
     while(e) {
@@ -588,17 +720,20 @@ edge_t* gfxpoly_process(edge_t*poly)
         fprintf(stderr, "----------------------------------- %d\n", status.y);
         actlist_verify_and_dump(status.actlist, status.y-1);
 #endif
-        xrow_reset(xrow);
+        xrow_reset(status.xrow);
         do {
-            xrow_add(xrow, e->p.x);
+            if(e->type != EVENT_HORIZONTAL) {
+                xrow_add(status.xrow, e->p.x);
+            }
             event_apply(&status, e);
             free(e);
             e = heap_chopmax(queue);
         } while(e && status.y == e->p.y);
 
-        xrow_sort(xrow);
-        add_points_to_positively_sloped_segments(&status, xrow, status.y);
-        add_points_to_negatively_sloped_segments(&status, xrow, status.y);
+        xrow_sort(status.xrow);
+        add_points_to_positively_sloped_segments(&status, status.y);
+        add_points_to_negatively_sloped_segments(&status, status.y);
+        recalculate_windings(&status);
 #ifdef DEBUG
         check_status(&status);
         dict_destroy(status.intersecting_segs);
@@ -610,7 +745,9 @@ edge_t* gfxpoly_process(edge_t*poly)
 #endif
     actlist_destroy(status.actlist);
     heap_destroy(queue);
-    xrow_destroy(xrow);
+    xrow_destroy(status.xrow);
 
-    return status.output;
+    gfxpoly_t*p = gfxpoly_new(poly->gridsize);
+    p->edges = status.output;
+    return p;
 }