sketcher tweaks + strengthen, yuv alpha draw_pixel fix
[goodguy/cinelerra.git] / cinelerra-5.1 / plugins / sketcher / sketcher.C
1 /*
2  * CINELERRA
3  * Copyright (C) 1997-2015 Adam Williams <broadcast at earthling dot net>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18  *
19  */
20
21 #include<stdio.h>
22 #include<stdint.h>
23 #include<math.h>
24 #include<string.h>
25
26 #include "arraylist.h"
27 #include "bccmodels.h"
28 #include "bccolors.h"
29 #include "clip.h"
30 #include "edlsession.h"
31 #include "filexml.h"
32 #include "overlayframe.h"
33 #include "pluginserver.h"
34 #include "preferences.h"
35 #include "sketcher.h"
36 #include "sketcherwindow.h"
37 #include "language.h"
38 #include "vframe.h"
39
40 void SketcherPoint::init(int id, int arc, coord x, coord y)
41 {
42         this->id = id;  this->arc = arc;
43         this->x = x;    this->y = y;
44 }
45 SketcherPoint::SketcherPoint(int id)
46 {
47         init(id, ARC_LINE, 0, 0);
48 }
49 SketcherPoint::SketcherPoint(int id, int arc, coord x, coord y)
50 {
51         init(id, arc, x, y);
52 }
53 SketcherPoint::~SketcherPoint()
54 {
55 }
56 SketcherPoint::SketcherPoint(SketcherPoint &pt)
57 {
58         copy_from(pt);
59 }
60 int SketcherPoint::equivalent(SketcherPoint &that)
61 {
62         return this->id == that.id &&
63                 this->arc == that.arc &&
64                 EQUIV(this->x, that.x) &&
65                 EQUIV(this->y, that.y) ? 1 : 0;
66 }
67 void SketcherPoint::copy_from(SketcherPoint &that)
68 {
69         this->id = that.id;  this->arc = that.arc;
70         this->x = that.x;    this->y = that.y;
71 }
72 void SketcherPoint::save_data(FileXML &output)
73 {
74         char point[BCSTRLEN];
75         sprintf(point,"/POINT_%d",id);
76         output.tag.set_title(point+1);
77         output.tag.set_property("TYPE", arc);
78         output.tag.set_property("X", x);
79         output.tag.set_property("Y", y);
80         output.append_tag();
81         output.tag.set_title(point+0);
82         output.append_tag();
83         output.append_newline();
84 }
85 void SketcherPoint::read_data(FileXML &input)
86 {
87         id = atoi(input.tag.get_title() + 6);
88         arc = input.tag.get_property("TYPE", ARC_OFF);
89         x = input.tag.get_property("X", (coord)0);
90         y = input.tag.get_property("Y", (coord)0);
91         bclamp(arc, 0, ARC_SZ-1);
92 }
93
94 void SketcherCurve::init(int id, int pen, int width, int color)
95 {
96         this->id = id;
97         this->width = width;
98         this->pen = pen;
99         this->color = color;
100 }
101 SketcherCurve::SketcherCurve(int id)
102 {
103         init(id, 1, ARC_LINE, CV_COLOR);
104 }
105 SketcherCurve::SketcherCurve(int id, int pen, int width, int color)
106 {
107         init(id, pen, width, color);
108 }
109 SketcherCurve::~SketcherCurve()
110 {
111 }
112 SketcherCurve::SketcherCurve(SketcherCurve &cv)
113 {
114         copy_from(cv);
115 }
116 int SketcherCurve::equivalent(SketcherCurve &that)
117 {
118         if( this->id != that.id ) return 0;
119         if( this->pen != that.pen ) return 0;
120         if( this->width != that.width ) return 0;
121         if( this->color != that.color ) return 0;
122         int n = this->points.size();
123         if( n != that.points.size() ) return 0;
124         for( int i=0; i<n; ++i ) {
125                 if( !points[i]->equivalent(*that.points[i]) ) return 0;
126         }
127         return 1;
128 }
129 void SketcherCurve::copy_from(SketcherCurve &that)
130 {
131         this->id = that.id;
132         this->pen = that.pen;
133         this->width = that.width;
134         this->color = that.color;
135         int m = points.size(), n = that.points.size();
136         while( m > n ) points.remove_object_number(--m);
137         while( m < n ) { points.append(new SketcherPoint());  ++m; }
138         for( int i=0; i<n; ++i ) points[i]->copy_from(*that.points[i]);
139 }
140 void SketcherCurve::save_data(FileXML &output)
141 {
142         char curve[BCSTRLEN];
143         sprintf(curve,"/CURVE_%d",id);
144         output.tag.set_title(curve+1);
145         output.tag.set_property("PEN", pen);
146         output.tag.set_property("RADIUS", width);
147         output.tag.set_property("COLOR", color);
148         output.append_tag();
149         output.append_newline();
150         for( int i=0,n=points.size(); i<n; ++i )
151                 points[i]->save_data(output);
152         output.tag.set_title(curve+0);
153         output.append_tag();
154         output.append_newline();
155 }
156 void SketcherCurve::read_data(FileXML &input)
157 {
158         id = atoi(input.tag.get_title() + 6);
159         pen = input.tag.get_property("PEN", PEN_OFF);
160         width = input.tag.get_property("RADIUS", 1.);
161         color = input.tag.get_property("COLOR", CV_COLOR);
162         bclamp(pen, 0, PEN_SZ-1);
163 }
164
165 int Sketcher::new_curve(int pen, int width, int color)
166 {
167         SketcherCurves &curves = config.curves;
168         int k = curves.size(), id = 1;
169         for( int i=k; --i>=0; ) {
170                 int n = config.curves[i]->id;
171                 if( n >= id ) id = n + 1;
172         }
173         SketcherCurve *cv = new SketcherCurve(id, pen, width, color);
174         curves.append(cv);
175         config.cv_selected = k;
176         return k;
177 }
178
179 int Sketcher::new_curve()
180 {
181         return new_curve(PEN_XLANT, 1, CV_COLOR);
182 }
183
184 int Sketcher::new_point(SketcherCurve *cv, int arc, coord x, coord y, int idx)
185 {
186         int id = 1;
187         for( int i=cv->points.size(); --i>=0; ) {
188                 int n = cv->points[i]->id;
189                 if( n >= id ) id = n + 1;
190         }
191         SketcherPoint *pt = new SketcherPoint(id, arc, x, y);
192         int n = cv->points.size();
193         if( idx < 0 || idx > n ) idx = n;
194         cv->points.insert(pt, idx);
195         return idx;
196 }
197
198 int Sketcher::new_point(int idx, int arc)
199 {
200         int ci = config.cv_selected;
201         if( ci < 0 || ci >= config.curves.size() )
202                 return -1;
203         SketcherCurve *cv = config.curves[ci];
204         EDLSession *session = get_edlsession();
205         coord x = !session ? 0.f : session->output_w / 2.f;
206         coord y = !session ? 0.f : session->output_h / 2.f;
207         return new_point(cv, arc, x, y, idx);
208 }
209
210 double SketcherCurve::nearest_point(int &pi, coord x, coord y)
211 {
212         pi = -1;
213         double dist = DBL_MAX;
214         for( int i=0; i<points.size(); ++i ) {
215                 SketcherPoint *p = points[i];
216                 double d = DISTANCE(x,y, p->x,p->y);
217                 if( d < dist ) { dist = d;  pi = i;  }
218         }
219         return pi >= 0 ? dist : -1.;
220 }
221
222 double SketcherConfig::nearest_point(int &ci, int &pi, coord x, coord y)
223 {
224         double dist = DBL_MAX;
225         ci = -1;  pi = -1;
226         for( int i=0; i<curves.size(); ++i ) {
227                 SketcherCurve *crv = curves[i];
228                 SketcherPoints &points = crv->points;
229                 for( int k=0; k<points.size(); ++k ) {
230                         SketcherPoint *p = points[k];
231                         double d = DISTANCE(x,y, p->x,p->y);
232                         if( d < dist ) {  dist = d; ci = i;  pi = k; }
233                 }
234         }
235         return pi >= 0 ? dist : -1.;
236 }
237
238
239 REGISTER_PLUGIN(Sketcher)
240
241 SketcherConfig::SketcherConfig()
242 {
243         drag = 1;
244         cv_selected = 0;
245         pt_selected = 0;
246 }
247 SketcherConfig::~SketcherConfig()
248 {
249 }
250
251 int SketcherConfig::equivalent(SketcherConfig &that)
252 {
253         if( this->drag != that.drag ) return 0;
254         if( this->cv_selected != that.cv_selected ) return 0;
255         if( this->pt_selected != that.pt_selected ) return 0;
256         if( this->curves.size() != that.curves.size() ) return 0;
257         for( int i=0, n=curves.size(); i<n; ++i ) {
258                 if( !curves[i]->equivalent(*that.curves[i]) ) return 0;
259         }
260         return 1;
261 }
262
263 void SketcherConfig::copy_from(SketcherConfig &that)
264 {
265         this->drag = that.drag;
266         this->cv_selected = that.cv_selected;
267         this->pt_selected = that.pt_selected;
268         int m = curves.size(), n = that.curves.size();
269         while( m > n ) curves.remove_object_number(--m);
270         while( m < n ) { curves.append(new SketcherCurve());  ++m; }
271         for( int i=0; i<n; ++i ) curves[i]->copy_from(*that.curves[i]);
272 }
273
274 void SketcherConfig::interpolate(SketcherConfig &prev, SketcherConfig &next,
275                 long prev_frame, long next_frame, long current_frame)
276 {
277         this->cv_selected = prev.cv_selected;
278         this->pt_selected = prev.pt_selected;
279         this->drag = prev.drag;
280
281         double next_scale = (double)(current_frame - prev_frame) / (next_frame - prev_frame);
282         double prev_scale = (double)(next_frame - current_frame) / (next_frame - prev_frame);
283
284         curves.remove_all_objects();
285         int prev_cv_sz = prev.curves.size();
286         int next_cv_sz = next.curves.size();
287         for( int i=0; i<prev_cv_sz; ++i ) {
288                 SketcherCurve *pcv = prev.curves[i], *ncv = 0;
289                 SketcherCurve *cv = curves.append(new SketcherCurve());
290                 int k = next_cv_sz;  // associated by id in next
291                 while( --k >= 0 && pcv->id != (ncv=next.curves[k])->id );
292                 if( k >= 0 ) {
293                         cv->id = pcv->id;
294                         cv->pen = pcv->pen;
295                         cv->width = pcv->width == ncv->width ? pcv->width :
296                                 pcv->width*prev_scale + ncv->width*next_scale + 0.5;
297                         int pr =  (pcv->color>>16)&0xff, nr =  (ncv->color>>16)&0xff;
298                         int pg =  (pcv->color>> 8)&0xff, ng =  (ncv->color>> 8)&0xff;
299                         int pb =  (pcv->color>> 0)&0xff, nb =  (ncv->color>> 0)&0xff;
300                         int pa = (~pcv->color>>24)&0xff, na = (~ncv->color>>24)&0xff;
301                         int r = pr == nr ? pr : pr*prev_scale + nr*next_scale + 0.5;
302                         int g = pg == ng ? pg : pg*prev_scale + ng*next_scale + 0.5;
303                         int b = pb == nb ? pb : pb*prev_scale + nb*next_scale + 0.5;
304                         int a = pa == na ? pa : pa*prev_scale + na*next_scale + 0.5;
305                         bclamp(r,0,255); bclamp(g,0,255); bclamp(b,0,255); bclamp(a,0,255);
306                         cv->color = (~a<<24) | (r<<16) | (g<<8) | (b<<0);
307                         int prev_pt_sz = pcv->points.size(), next_pt_sz = ncv->points.size();
308                         for( int j=0; j<prev_pt_sz; ++j ) {
309                                 SketcherPoint &pt = *pcv->points[j], *nt = 0;
310                                 k = next_pt_sz;  // associated by id in next
311                                 while( --k >= 0 && pt.id != (nt=ncv->points[k])->id );
312                                 coord x = pt.x, y = pt.y;
313                                 if( k >= 0 ) {
314                                         if( x != nt->x )
315                                                 x = x * prev_scale + nt->x * next_scale;
316                                         if( y != nt->y )
317                                                 y = y * prev_scale + nt->y * next_scale;
318                                 }
319                                 cv->points.append(new SketcherPoint(pt.id, pt.arc, x, y));
320                         }
321                 }
322                 else
323                         cv->copy_from(*pcv);
324         }
325 }
326
327 void SketcherConfig::limits()
328 {
329 }
330
331
332 Sketcher::Sketcher(PluginServer *server)
333  : PluginVClient(server)
334 {
335         img = 0;
336         out = 0;
337         overlay_frame = 0;
338 }
339
340 Sketcher::~Sketcher()
341 {
342         delete img;
343         delete out;
344         delete overlay_frame;
345 }
346
347 const char* Sketcher::plugin_title() { return N_("Sketcher"); }
348 int Sketcher::is_realtime() { return 1; }
349
350 NEW_WINDOW_MACRO(Sketcher, SketcherWindow);
351 LOAD_CONFIGURATION_MACRO(Sketcher, SketcherConfig)
352
353 void Sketcher::save_data(KeyFrame *keyframe)
354 {
355         FileXML output;
356 // cause data to be stored directly in text
357         output.set_shared_output(keyframe->xbuf);
358
359         output.tag.set_title("SKETCHER");
360         output.tag.set_property("DRAG", config.drag);
361         output.tag.set_property("CV_SELECTED", config.cv_selected);
362         output.tag.set_property("PT_SELECTED", config.pt_selected);
363         output.append_tag();
364         output.append_newline();
365         for( int i=0,n=config.curves.size(); i<n; ++i ) {
366                 config.curves[i]->save_data(output);
367         }
368         output.tag.set_title("/SKETCHER");
369         output.append_tag();
370         output.append_newline();
371         output.terminate_string();
372 }
373
374 void Sketcher::read_data(KeyFrame *keyframe)
375 {
376         FileXML input;
377         input.set_shared_input(keyframe->xbuf);
378         config.curves.remove_all_objects();
379         int result = 0;
380         SketcherCurve *cv = 0;
381
382         while( !(result=input.read_tag()) ) {
383                 if( input.tag.title_is("SKETCHER") ) {
384                         config.drag = input.tag.get_property("DRAG", config.drag);
385                         config.cv_selected = input.tag.get_property("CV_SELECTED", 0);
386                         config.pt_selected = input.tag.get_property("PT_SELECTED", 0);
387                 }
388                 else if( !strncmp(input.tag.get_title(),"CURVE_",6) ) {
389                         cv = new SketcherCurve();
390                         cv->read_data(input);
391                         config.curves.append(cv);
392                 }
393                 else if( !strncmp(input.tag.get_title(),"/CURVE_",7) )
394                         cv = 0;
395                 else if( !strncmp(input.tag.get_title(),"POINT_",6) ) {
396                         if( cv ) {
397                                 SketcherPoint *pt = new SketcherPoint();
398                                 pt->read_data(input);
399                                 cv->points.append(pt);
400                         }
401                         else
402                                 printf("Sketcher::read_data: no curve for point\n");
403                 }
404         }
405
406         if( !config.curves.size() )
407                 new_curve();
408         config.limits();
409 }
410
411 void Sketcher::update_gui()
412 {
413         if( !thread ) return;
414         thread->window->lock_window("Sketcher::update_gui");
415         if( load_configuration() ) {
416                 SketcherWindow *window = (SketcherWindow*)thread->window;
417                 window->update_gui();
418                 window->flush();
419         }
420         thread->window->unlock_window();
421 }
422
423 void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color, int d)
424 {
425         int r = d/2+1, x = pt->x, y = pt->y;
426         vfrm->set_pixel_color(color);
427         vfrm->draw_smooth(x-r,y+0, x-r, y-r, x+0,y-r);
428         vfrm->draw_smooth(x+0,y-r, x+r, y-r, x+r,y+0);
429         vfrm->draw_smooth(x+r,y+0, x+r, y+r, x+0,y+r);
430         vfrm->draw_smooth(x+0,y+r, x-r, y+r, x-r,y+0);
431         vfrm->draw_x(pt->x, pt->y, d);
432 }
433 void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color)
434 {
435         draw_point(vfrm, pt, color, bmax(w,h)/200 + 2);
436 }
437
438
439 int SketcherVPen::draw_pixel(int x, int y)
440 {
441         if( x >= 0 && x < vfrm->get_w() &&
442             y >= 0 && y < vfrm->get_h() )
443                 msk[vfrm->get_w()*y + x] = 0xff;
444         return 0;
445 }
446
447 int SketcherPenSquare::draw_pixel(int x, int y)
448 {
449         vfrm->draw_line(x-n, y, x+n, y);
450         for( int i=-n; i<n; ++i )
451                 vfrm->draw_line(x-n, y+i, x+n, y+i);
452         return SketcherVPen::draw_pixel(x, y);
453 }
454 int SketcherPenPlus::draw_pixel(int x, int y)
455 {
456         if( n > 1 ) {
457                 vfrm->draw_line(x-n, y, x+n, y);
458                 vfrm->draw_line(x, y-n, x, y+n);
459         }
460         else
461                 vfrm->draw_pixel(x, y);
462         return SketcherVPen::draw_pixel(x, y);
463 }
464 int SketcherPenSlant::draw_pixel(int x, int y)
465 {
466         vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
467         vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
468         vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
469         return SketcherVPen::draw_pixel(x, y);
470 }
471 int SketcherPenXlant::draw_pixel(int x, int y)
472 {
473         vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
474         vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
475         vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
476         vfrm->draw_line(x-n,   y-n,   x+n,   y+n);
477         vfrm->draw_line(x-n+1, y-n,   x+n+1, y+n);
478         vfrm->draw_line(x-n,   y-n+1, x+n,   y-n+1);
479         return SketcherVPen::draw_pixel(x, y);
480 }
481
482
483 SketcherVPen *SketcherCurve::new_vpen(VFrame *out)
484 {
485         switch( pen ) {
486         case PEN_SQUARE: return new SketcherPenSquare(out, width);
487         case PEN_PLUS:   return new SketcherPenPlus(out, width);
488         case PEN_SLANT:  return new SketcherPenSlant(out, width);
489         case PEN_XLANT:  return new SketcherPenXlant(out, width);
490         }
491         return 0;
492 }
493
494 static int intersects_at(float &x, float &y,
495                 float ax,float ay, float bx, float by, float cx,float cy,  // line slope ab thru c
496                 float dx,float dy, float ex, float ey, float fx,float fy, // line slope de thru f
497                 float mx=0)
498 {
499         float badx = bx - ax, bady = by - ay;
500         float eddx = ex - dx, eddy = ey - dy;
501         float d = badx*eddy - bady*eddx;
502         int ret = 0;
503         if( fabsf(d) < 1 ) { ret = 1;  d = signbit(d) ? -1 : 1; }
504         x = (badx*cy*eddx - badx*eddx*fy + badx*eddy*fx - bady*cx*eddx) / d;
505         y = (badx*cy*eddy - bady*cx*eddy - bady*eddx*fy + bady*eddy*fx) / d;
506         if( mx > 0 ) { bclamp(x, -mx,mx);  bclamp(y, -mx,mx); }
507         return ret;
508 }
509
510 static void smooth_axy(float &ax, float &ay,
511         float bx, float by, float cx, float cy, float dx, float dy)
512 {
513 //middle of bd reflected around ctr
514 // point ctr = (b+d)/2, dv=c-ctr, a=ctr-dv;
515         float xc = (bx+dx)*.5f, yc = (by+dy)*.5f;
516         float xd = cx - xc, yd = cy - yc;
517         ax = xc - xd;  ay = yc - yd;
518 }
519 static void smooth_dxy(float &dx, float &dy,
520         float ax, float ay, float bx, float by, float cx, float cy)
521 {
522 //middle of ac reflected around ctr
523 // point ctr = (a+c)/2, dv=c-ctr, d=ctr-dv;
524         float xc = (ax+cx)*.5f, yc = (ay+cy)*.5f;
525         float xd = bx - xc, yd = by - yc;
526         dx = xc - xd;  dy = yc - yd;
527 }
528
529 #if 0
530 static int convex(float ax,float ay, float bx,float by,
531                   float cx,float cy, float dx,float dy)
532 {
533         float abdx = bx-ax, abdy = by-ay;
534         float acdx = cx-ax, acdy = cy-ay;
535         float bcdx = cx-bx, bcdy = cy-by;
536         float bddx = dx-bx, bddy = dy-by;
537         float abc = abdx*acdy - abdy*acdx;
538         float bcd = bcdx*bddy - bcdy*bddx;
539         float v = abc * bcd;
540         return !v ? 0 : v>0 ? 1 : -1;
541 }
542 #endif
543
544
545 class FillRegion
546 {
547         class segment { public: int y, lt, rt; };
548         ArrayList<segment> stack;
549
550         void push(int y, int lt, int rt) {
551                 segment &seg = stack.append();
552                 seg.y = y;  seg.lt = lt;  seg.rt = rt;
553         }
554         void pop(int &y, int &lt, int &rt) {
555                 segment &seg = stack.last();
556                 y = seg.y;  lt = seg.lt;  rt = seg.rt;
557                 stack.remove();
558         }
559  
560         VFrame *img;
561         uint8_t *msk;
562         int w, h, nxt;
563         SketcherPoints &points;
564 public:
565         SketcherPoint *next();
566         bool exists() { return stack.size() > 0; }
567         void start_at(int x, int y);
568         void run();
569         FillRegion(SketcherPoints &pts, SketcherVPen *vpen);
570         ~FillRegion();
571 };
572
573 FillRegion::FillRegion(SketcherPoints &pts, SketcherVPen *vpen)
574  : points(pts)
575 {
576         this->img = vpen->vfrm;
577         this->msk = vpen->msk;
578         this->w = img->get_w();
579         this->h = img->get_h();
580         nxt = 0;
581 }
582 FillRegion::~FillRegion()
583 {
584 }
585
586 void FillRegion::start_at(int x, int y)
587 {
588         bclamp(x, 0, w-1);
589         bclamp(y, 0, h-1);
590         push(y, x, x);
591 }
592
593 void FillRegion::run()
594 {
595         while( stack.size() > 0 ) {
596                 int y, ilt, irt;
597                 pop(y, ilt, irt);
598                 int ofs = y*w + ilt;
599                 for( int x=ilt; x<=irt; ++x,++ofs ) {
600                         if( msk[ofs] ) continue;
601                         msk[ofs] = 0xff;
602                         img->draw_pixel(x, y);
603                         int lt = x, rt = x;
604                         int lofs = ofs;
605                         for( int i=lt; --i>=0; ) {
606                                 if( msk[--lofs] ) break;
607                                 img->draw_pixel(i, y);
608                                 msk[lofs] = 0xff;  lt = i;
609                         }
610                         int rofs = ofs;
611                         for( int i=rt; ++i< w; ) {
612                                 if( msk[++rofs] ) break;
613                                 img->draw_pixel(i, y);
614                                 msk[rofs] = 0xff;  rt = i;
615                         }
616                         if( y+1 <  h ) push(y+1, lt, rt);
617                         if( y-1 >= 0 ) push(y-1, lt, rt);
618                 }
619         }
620 }
621
622 SketcherPoint *FillRegion::next()
623 {
624         while( nxt < points.size() ) {
625                 SketcherPoint *pt = points[nxt];
626                 if( pt->arc == ARC_OFF ) continue;
627                 if( pt->arc != ARC_FILL ) break;
628                 start_at(pt->x, pt->y);
629                 ++nxt;
630         }
631         return nxt < points.size() ? points[nxt++] : 0;
632 }
633
634
635 void SketcherCurve::draw(VFrame *img)
636 {
637         if( !points.size() ) return;
638         const float fmx = 16383;
639         SketcherVPen *vpen = new_vpen(img);
640         FillRegion fill(points, vpen);
641         SketcherPoint *pnt0 = fill.next();
642         SketcherPoint *pnt1 = pnt0 ? fill.next() : 0;
643         SketcherPoint *pnt2 = pnt1 ? fill.next() : 0;
644         if( pnt0 && pnt1 && pnt2 ) {
645                 SketcherPoint *pt0 = pnt0, *pt1 = pnt1, *pt2 = pnt2;
646                 float ax,ay, bx,by, cx,cy, dx,dy, sx,sy;
647                 bx = pt0->x;  by = pt0->y;
648                 cx = pt1->x;  cy = pt1->y;
649                 dx = pt2->x;  dy = pt2->y;
650                 smooth_axy(ax,ay, bx,by, cx,cy, dx,dy);
651                 while( pt2 ) {
652                         dx = pt2->x;  dy = pt2->y;
653                         switch( pt0->arc ) {
654                         case ARC_LINE:
655                                 vpen->draw_line(bx, by, cx, cy);
656                                 break;
657                         case ARC_CURVE: {
658                                 // s = ac thru b x bd thru c
659                                 intersects_at(sx,sy, ax,ay,cx,cy,bx,by, bx,by,dx,dy,cx,cy,fmx);
660                                 vpen->draw_smooth(bx,by, sx,sy, cx,cy);
661                                 break; }
662                         }
663                         ax = bx;  ay = by;  pt0 = pt1;
664                         bx = cx;  by = cy;  pt1 = pt2;
665                         cx = dx;  cy = dy;  pt2 = fill.next();
666                 }
667                 switch( pt1->arc ) {
668                 case ARC_LINE:
669                         vpen->draw_line(bx, by, cx, cy);
670                         if( fill.exists() ) {
671                                 dx = pnt0->x;  dy = pnt0->y;
672                                 vpen->draw_line(cx,cy, dx,dy);
673                         }
674                         break;
675                 case ARC_CURVE: {
676                         if( fill.exists() ) {
677                                 dx = pnt0->x;  dy = pnt0->y;
678                                 intersects_at(sx,sy, ax,ay,cx,cy,bx,by, bx,by,dx,dy,cx,cy,fmx);
679                                 vpen->draw_smooth(bx,by, sx,sy, cx,cy);
680                                 ax = bx;  ay = by;
681                                 bx = cx;  by = cy;
682                                 cx = dx;  cy = dy;
683                                 dx = pnt1->x;  dy = pnt1->y;
684                         }
685                         else
686                                 smooth_dxy(dx,dy, ax,ay, bx,by, cx,cy);
687                         intersects_at(sx,sy, ax,ay,cx,cy,bx,by, bx,by,dx,dy,cx,cy,fmx);
688                         vpen->draw_smooth(bx,by, sx,sy, cx,cy);
689                         break; }
690                 }
691                 fill.run();
692         }
693         else if( pnt0 && pnt1 ) {
694                 vpen->draw_line(pnt0->x, pnt0->y, pnt1->x, pnt1->y);
695         }
696         else if( pnt0 ) {
697                 vpen->draw_pixel(pnt0->x, pnt0->y);
698         }
699         delete vpen;
700 }
701
702 int Sketcher::process_realtime(VFrame *input, VFrame *output)
703 {
704         this->input = input;  this->output = output;
705         w = output->get_w();  h = output->get_h();
706
707         load_configuration();
708
709         int out_color_model = output->get_color_model();
710         int color_model = out_color_model;
711         switch( color_model ) { // add alpha if needed
712         case BC_RGB888:         color_model = BC_RGBA8888;      break;
713         case BC_YUV888:         color_model = BC_YUVA8888;      break;
714         case BC_RGB161616:      color_model = BC_RGBA16161616;  break;
715         case BC_YUV161616:      color_model = BC_YUVA16161616;  break;
716         case BC_RGB_FLOAT:      color_model = BC_RGBA_FLOAT;    break;
717         case BC_RGB_FLOATP:     color_model = BC_RGBA_FLOATP;   break;
718         }
719         if( color_model == out_color_model ) {
720                 delete out;  out = output;
721                 if( output != input )
722                         output->transfer_from(input);
723         }
724         else {
725                 VFrame::get_temp(out, w, h, color_model);
726                 out->transfer_from(input);
727         }
728         VFrame::get_temp(img, w, h, color_model);
729         
730         if( !overlay_frame ) {
731                 int cpus = server->preferences->project_smp;
732                 int max = (w*h)/0x80000 + 2;
733                 if( cpus > max ) cpus = max;
734                 overlay_frame = new OverlayFrame(cpus);
735         }
736
737         for( int ci=0, n=config.curves.size(); ci<n; ++ci ) {
738                 SketcherCurve *cv = config.curves[ci];
739                 if( cv->pen == PEN_OFF ) continue;
740                 int m = cv->points.size();
741                 if( !m ) continue;
742                 img->clear_frame();
743                 img->set_pixel_color(cv->color, (~cv->color>>24)&0xff);
744                 cv->draw(img);
745                 overlay_frame->overlay(out, img, 0,0,w,h, 0,0,w,h,
746                                 1.f, TRANSFER_NORMAL, NEAREST_NEIGHBOR);
747         }
748
749         if( config.drag ) {
750                 for( int ci=0, n=config.curves.size(); ci<n; ++ci ) {
751                         SketcherCurve *cv = config.curves[ci];
752                         for( int pi=0,m=cv->points.size(); pi<m; ++pi ) {
753                                 int color = pi==config.pt_selected && ci==config.cv_selected ?
754                                         RED : cv->color ; 
755                                 draw_point(out, cv->points[pi], color);
756                         }
757                 }
758         }
759
760         if( output != out )
761                 output->transfer_from(out);
762         else
763                 out = 0;
764
765         return 0;
766 }
767
768 void SketcherPoints::dump()
769 {
770         for( int i=0; i<size(); ++i ) {
771                 SketcherPoint *pt = get(i);
772                 printf("  Pt %d, id=%d, arc=%s, x=%0.1f, y=%0.1f\n",
773                         i, pt->id, pt_type[pt->arc], pt->x, pt->y);
774         }
775 }
776 void SketcherCurves::dump()
777 {
778         for( int i=0; i<size(); ++i ) {
779                 SketcherCurve *cv = get(i);
780                 printf("Curve %d, id=%d, pen=%s, r=%d, color=%02x%02x%02x%02x, %d points\n",
781                         i, cv->id, cv_pen[cv->pen], cv->width, (~cv->color>>24)&0xff,
782                         (cv->color>>16)&0xff, (cv->color>>8)&0xff, (cv->color>>0)&0xff,
783                         cv->points.size());
784                 cv->points.dump();
785         }
786 }
787 void SketcherConfig::dump()
788 {
789         printf("Config drag=%d, cv_selected=%d, pt_selected=%d %d curves\n",
790                         drag, cv_selected, pt_selected, curves.size());
791         curves.dump();
792 }
793