fix for closing unclosed shapes automatically
[swftools.git] / lib / modules / swfdraw.c
1 // swfdraw.c
2
3 typedef struct _SWFSHAPEDRAWER
4 {
5     SHAPE*shape;
6     TAG*tag;
7     int tagfree;
8     SCOORD firstx;
9     SCOORD firsty;
10     SCOORD lastx;
11     SCOORD lasty;
12     SRECT bbox;
13     char isfinished;
14 } SWFSHAPEDRAWER;
15
16 static void swf_ShapeDrawerSetLineStyle(drawer_t*draw, void*style);
17 static void swf_ShapeDrawerSetFillStyle(drawer_t*draw, void*style);
18 static void swf_ShapeDrawerMoveTo(drawer_t*draw, FPOINT * to);
19 static void swf_ShapeDrawerLineTo(drawer_t*draw, FPOINT * to);
20 static void swf_ShapeDrawerSplineTo(drawer_t*draw, FPOINT * c1, FPOINT*  to);
21 static void swf_ShapeDrawerFinish(drawer_t*draw);
22 static void swf_ShapeDrawerClear(drawer_t*draw);
23
24 static void swf_ShapeDrawerInit(drawer_t*draw, TAG*tag, int fillstylebits, int linestylebits)
25 {
26     SWFSHAPEDRAWER*sdraw = malloc(sizeof(SWFSHAPEDRAWER));
27     memset(sdraw, 0, sizeof(SWFSHAPEDRAWER));
28     draw->internal = sdraw;
29
30     draw->setLineStyle = swf_ShapeDrawerSetLineStyle;
31     draw->setFillStyle = swf_ShapeDrawerSetFillStyle;
32     draw->moveTo = swf_ShapeDrawerMoveTo;
33     draw->lineTo = swf_ShapeDrawerLineTo;
34     draw->splineTo = swf_ShapeDrawerSplineTo;
35     draw->finish = swf_ShapeDrawerFinish;
36     draw->dealloc = swf_ShapeDrawerClear;
37     
38     sdraw->tagfree = 0;
39     if(tag == 0) {
40         tag = swf_InsertTag(0, ST_DEFINESHAPE);
41         sdraw->tagfree = 1;
42     }
43     sdraw->tag = tag;
44     swf_ShapeNew(&sdraw->shape);
45     draw->pos.x = 0;
46     draw->pos.y = 0;
47
48     swf_SetU8(sdraw->tag,0);
49     sdraw->shape->bits.fill = fillstylebits;
50     sdraw->shape->bits.line = linestylebits;
51     
52     sdraw->bbox.xmin = sdraw->bbox.ymin = SCOORD_MAX;
53     sdraw->bbox.xmax = sdraw->bbox.ymax = SCOORD_MIN;
54
55     sdraw->isfinished = 0;
56
57     swf_ShapeSetStyle(sdraw->tag,sdraw->shape,linestylebits?1:0,fillstylebits?1:0,0/*?*/);
58 }
59
60 void swf_Shape01DrawerInit(drawer_t*draw, TAG*tag)
61 {
62     swf_ShapeDrawerInit(draw, tag, 1, 0);
63 }
64
65 void swf_Shape11DrawerInit(drawer_t*draw, TAG*tag)
66 {
67     swf_ShapeDrawerInit(draw, tag, 1, 1);
68 }
69
70 static void swf_ShapeDrawerSetLineStyle(drawer_t*draw, void*style)
71 {
72     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
73 }
74 static void swf_ShapeDrawerSetFillStyle(drawer_t*draw, void*style)
75 {
76     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
77 }
78 static void fixEndPoint(drawer_t*draw)
79 {
80     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
81     if(   sdraw->firstx != sdraw->lastx 
82        || sdraw->firsty != sdraw->lasty) {
83         /* fix non-closing shapes */
84         /* TODO: do this only if the shape is filled */
85         FPOINT to;
86         to.x = sdraw->firstx/20.0;
87         to.y = sdraw->firsty/20.0;
88         draw->lineTo(draw, &to);
89     }
90 }
91 static void swf_ShapeDrawerMoveTo(drawer_t*draw, FPOINT * to)
92 {
93     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
94     int x = to->x*20;
95     int y = to->y*20;
96     if(sdraw->lastx != x || sdraw->lasty != y) {
97         fixEndPoint(draw);
98         swf_ShapeSetMove(sdraw->tag,sdraw->shape,x,y);
99         sdraw->firstx = sdraw->lastx = x;
100         sdraw->firsty = sdraw->lasty = y;
101         draw->pos = *to;
102     }
103 }
104 static void swf_ShapeDrawerLineTo(drawer_t*draw, FPOINT * to)
105 {
106     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
107     int x = to->x*20;
108     int y = to->y*20;
109     if(sdraw->lastx < sdraw->bbox.xmin) sdraw->bbox.xmin = sdraw->lastx;
110     if(sdraw->lasty < sdraw->bbox.ymin) sdraw->bbox.ymin = sdraw->lasty;
111     if(sdraw->lastx > sdraw->bbox.xmax) sdraw->bbox.xmax = sdraw->lastx;
112     if(sdraw->lasty > sdraw->bbox.ymax) sdraw->bbox.ymax = sdraw->lasty;
113     if(x < sdraw->bbox.xmin) sdraw->bbox.xmin = x;
114     if(y < sdraw->bbox.ymin) sdraw->bbox.ymin = y;
115     if(x > sdraw->bbox.xmax) sdraw->bbox.xmax = x;
116     if(y > sdraw->bbox.ymax) sdraw->bbox.ymax = y;
117     swf_ShapeSetLine(sdraw->tag,sdraw->shape,x-sdraw->lastx,y-sdraw->lasty);
118     sdraw->lastx = x;
119     sdraw->lasty = y;
120     draw->pos = *to;
121 }
122 static void swf_ShapeDrawerSplineTo(drawer_t*draw, FPOINT * c1, FPOINT*  to)
123 {
124     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
125     int tx = c1->x*20;
126     int ty = c1->y*20;
127     int x = to->x*20;
128     int y = to->y*20;
129     if(sdraw->lastx < sdraw->bbox.xmin) sdraw->bbox.xmin = sdraw->lastx;
130     if(sdraw->lasty < sdraw->bbox.ymin) sdraw->bbox.ymin = sdraw->lasty;
131     if(sdraw->lastx > sdraw->bbox.xmax) sdraw->bbox.xmax = sdraw->lastx;
132     if(sdraw->lasty > sdraw->bbox.ymax) sdraw->bbox.ymax = sdraw->lasty;
133     if(x < sdraw->bbox.xmin) sdraw->bbox.xmin = x;
134     if(y < sdraw->bbox.ymin) sdraw->bbox.ymin = y;
135     if(x > sdraw->bbox.xmax) sdraw->bbox.xmax = x;
136     if(y > sdraw->bbox.ymax) sdraw->bbox.ymax = y;
137     if(tx < sdraw->bbox.xmin) sdraw->bbox.xmin = tx;
138     if(ty < sdraw->bbox.ymin) sdraw->bbox.ymin = ty;
139     if(tx > sdraw->bbox.xmax) sdraw->bbox.xmax = tx;
140     if(ty > sdraw->bbox.ymax) sdraw->bbox.ymax = ty;
141     swf_ShapeSetCurve(sdraw->tag,sdraw->shape, tx-sdraw->lastx,ty-sdraw->lasty, x-tx,y-ty);
142     sdraw->lastx = x;
143     sdraw->lasty = y;
144     draw->pos = *to;
145 }
146 static void swf_ShapeDrawerFinish(drawer_t*draw)
147 {
148     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
149         
150     fixEndPoint(draw);
151
152     if(sdraw->bbox.xmin == SCOORD_MAX) {
153         /* no points at all -> empty bounding box */
154         sdraw->bbox.xmin = sdraw->bbox.ymin = 
155         sdraw->bbox.xmax = sdraw->bbox.ymax = 0;
156     }
157     sdraw->isfinished = 1;
158     swf_ShapeSetEnd(sdraw->tag);
159 }
160
161 static void swf_ShapeDrawerClear(drawer_t*draw)
162 {
163     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
164     if(sdraw->tagfree) {
165         swf_DeleteTag(sdraw->tag);
166         sdraw->tag = 0;
167     }
168     swf_ShapeFree(sdraw->shape);
169     sdraw->shape = 0;
170
171     free(draw->internal);
172     draw->internal = 0;
173 }
174
175 SRECT swf_ShapeDrawerGetBBox(drawer_t*draw)
176 {
177     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
178     return sdraw->bbox;
179 }
180
181 SHAPE* swf_ShapeDrawerToShape(drawer_t*draw)
182 {
183     SWFSHAPEDRAWER*sdraw = (SWFSHAPEDRAWER*)draw->internal;
184     SHAPE* shape = malloc(sizeof(SHAPE));
185     if(!sdraw->isfinished) {
186         fprintf(stderr, "Warning: you should Finish() your drawer before calling DrawerToShape");
187         swf_ShapeDrawerFinish(draw);
188     }
189     memcpy(shape, sdraw->shape, sizeof(SHAPE));
190     shape->bitlen = (sdraw->tag->len-1)*8;
191     shape->data = (U8*)malloc(sdraw->tag->len-1);
192     memcpy(shape->data, &sdraw->tag->data[1], sdraw->tag->len-1);
193     return shape;
194 }