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 pty, coord x, coord y)
  41 {
  42         this->id = id;  this->pty = pty;
  43         this->x = x;    this->y = y;
  44 }
  45 SketcherPoint::SketcherPoint(int id)
  46 {
  47         init(id, PTY_LINE, 0, 0);
  48 }
  49 SketcherPoint::SketcherPoint(int id, int pty, coord x, coord y)
  50 {
  51         init(id, pty, 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->pty == that.pty &&
  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->pty = that.pty;
  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", pty);
  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         pty = input.tag.get_property("TYPE", PTY_OFF);
  89         x = input.tag.get_property("X", (coord)0);
  90         y = input.tag.get_property("Y", (coord)0);
  91         bclamp(pty, 0, PTY_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, PTY_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         this->pen = pen;  this->color = color;
 143         char curve[BCSTRLEN];
 144         sprintf(curve,"/CURVE_%d",id);
 145         output.tag.set_title(curve+1);
 146         output.tag.set_property("PEN", pen);
 147         output.tag.set_property("RADIUS", width);
 148         output.tag.set_property("COLOR", color);
 149         output.append_tag();
 150         output.append_newline();
 151         for( int i=0,n=points.size(); i<n; ++i )
 152                 points[i]->save_data(output);
 153         output.tag.set_title(curve+0);
 154         output.append_tag();
 155         output.append_newline();
 156 }
 157 void SketcherCurve::read_data(FileXML &input)
 158 {
 159         id = atoi(input.tag.get_title() + 6);
 160         pen = input.tag.get_property("PEN", PTY_OFF);
 161         width = input.tag.get_property("RADIUS", 1.);
 162         color = input.tag.get_property("COLOR", CV_COLOR);
 163         bclamp(pen, 0, PEN_SZ-1);
 164 }
 165
 166 int Sketcher::new_curve(int pen, int width, int color)
 167 {
 168         SketcherCurves &curves = config.curves;
 169         int k = curves.size(), id = 1;
 170         for( int i=k; --i>=0; ) {
 171                 int n = config.curves[i]->id;
 172                 if( n >= id ) id = n + 1;
 173         }
 174         SketcherCurve *cv = new SketcherCurve(id, pen, width, color);
 175         curves.append(cv);
 176         config.cv_selected = k;
 177         return k;
 178 }
 179
 180 int Sketcher::new_curve()
 181 {
 182         return new_curve(PEN_XLANT, 1, CV_COLOR);
 183 }
 184
 185 int Sketcher::new_point(SketcherCurve *cv, int pty, coord x, coord y, int idx)
 186 {
 187         int id = 1;
 188         for( int i=cv->points.size(); --i>=0; ) {
 189                 int n = cv->points[i]->id;
 190                 if( n >= id ) id = n + 1;
 191         }
 192         SketcherPoint *pt = new SketcherPoint(id, pty, x, y);
 193         int n = cv->points.size();
 194         if( idx < 0 || idx > n ) idx = n;
 195         cv->points.insert(pt, idx);
 196         return idx;
 197 }
 198
 199 int Sketcher::new_point(int idx)
 200 {
 201         int ci = config.cv_selected;
 202         if( ci < 0 || ci >= config.curves.size() )
 203                 return -1;
 204         SketcherCurve *cv = config.curves[ci];
 205         EDLSession *session = get_edlsession();
 206         coord x = !session ? 0.f : session->output_w / 2.f;
 207         coord y = !session ? 0.f : session->output_h / 2.f;
 208         return new_point(cv, PTY_LINE, x, y, idx);
 209 }
 210
 211 double SketcherCurve::nearest_point(int &pi, coord x, coord y)
 212 {
 213         pi = -1;
 214         double dist = DBL_MAX;
 215         for( int i=0; i<points.size(); ++i ) {
 216                 SketcherPoint *p = points[i];
 217                 double d = DISTANCE(x,y, p->x,p->y);
 218                 if( d < dist ) { dist = d;  pi = i;  }
 219         }
 220         return pi >= 0 ? dist : -1.;
 221 }
 222
 223 double SketcherConfig::nearest_point(int &ci, int &pi, coord x, coord y)
 224 {
 225         double dist = DBL_MAX;
 226         ci = -1;  pi = -1;
 227         for( int i=0; i<curves.size(); ++i ) {
 228                 SketcherCurve *crv = curves[i];
 229                 SketcherPoints &points = crv->points;
 230                 for( int k=0; k<points.size(); ++k ) {
 231                         SketcherPoint *p = points[k];
 232                         double d = DISTANCE(x,y, p->x,p->y);
 233                         if( d < dist ) {  dist = d; ci = i;  pi = k; }
 234                 }
 235         }
 236         return pi >= 0 ? dist : -1.;
 237 }
 238
 239
 240 REGISTER_PLUGIN(Sketcher)
 241
 242 SketcherConfig::SketcherConfig()
 243 {
 244         drag = 1;
 245         cv_selected = 0;
 246         pt_selected = 0;
 247 }
 248 SketcherConfig::~SketcherConfig()
 249 {
 250 }
 251
 252 int SketcherConfig::equivalent(SketcherConfig &that)
 253 {
 254         if( this->drag != that.drag ) return 0;
 255         if( this->cv_selected != that.cv_selected ) return 0;
 256         if( this->pt_selected != that.pt_selected ) return 0;
 257         if( this->curves.size() != that.curves.size() ) return 0;
 258         for( int i=0, n=curves.size(); i<n; ++i ) {
 259                 if( !curves[i]->equivalent(*that.curves[i]) ) return 0;
 260         }
 261         return 1;
 262 }
 263
 264 void SketcherConfig::copy_from(SketcherConfig &that)
 265 {
 266         this->drag = that.drag;
 267         this->cv_selected = that.cv_selected;
 268         this->pt_selected = that.pt_selected;
 269         int m = curves.size(), n = that.curves.size();
 270         while( m > n ) curves.remove_object_number(--m);
 271         while( m < n ) { curves.append(new SketcherCurve());  ++m; }
 272         for( int i=0; i<n; ++i ) curves[i]->copy_from(*that.curves[i]);
 273 }
 274
 275 void SketcherConfig::interpolate(SketcherConfig &prev, SketcherConfig &next,
 276                 long prev_frame, long next_frame, long current_frame)
 277 {
 278         this->cv_selected = prev.cv_selected;
 279         this->pt_selected = prev.pt_selected;
 280         this->drag = prev.drag;
 281
 282         double next_scale = (double)(current_frame - prev_frame) / (next_frame - prev_frame);
 283         double prev_scale = (double)(next_frame - current_frame) / (next_frame - prev_frame);
 284
 285         curves.remove_all_objects();
 286         int prev_cv_sz = prev.curves.size();
 287         int next_cv_sz = next.curves.size();
 288         for( int i=0; i<prev_cv_sz; ++i ) {
 289                 SketcherCurve *pcv = prev.curves[i], *ncv = 0;
 290                 SketcherCurve *cv = curves.append(new SketcherCurve());
 291                 int k = next_cv_sz;  // associated by id in next
 292                 while( --k >= 0 && pcv->id != (ncv=next.curves[k])->id );
 293                 if( k >= 0 ) {
 294                         cv->id = pcv->id;
 295                         cv->pen = pcv->pen;
 296                         cv->width = pcv->width == ncv->width ? pcv->width :
 297                                 pcv->width*prev_scale + ncv->width*next_scale + 0.5;
 298                         int pr =  (pcv->color>>16)&0xff, nr =  (ncv->color>>16)&0xff;
 299                         int pg =  (pcv->color>> 8)&0xff, ng =  (ncv->color>> 8)&0xff;
 300                         int pb =  (pcv->color>> 0)&0xff, nb =  (ncv->color>> 0)&0xff;
 301                         int pa = (~pcv->color>>24)&0xff, na = (~ncv->color>>24)&0xff;
 302                         int r = pr == nr ? pr : pr*prev_scale + nr*next_scale + 0.5;
 303                         int g = pg == ng ? pg : pg*prev_scale + ng*next_scale + 0.5;
 304                         int b = pb == nb ? pb : pb*prev_scale + nb*next_scale + 0.5;
 305                         int a = pa == na ? pa : pa*prev_scale + na*next_scale + 0.5;
 306                         bclamp(r,0,255); bclamp(g,0,255); bclamp(b,0,255); bclamp(a,0,255);
 307                         cv->color = (~a<<24) | (r<<16) | (g<<8) | (b<<0);
 308                         int prev_pt_sz = pcv->points.size(), next_pt_sz = ncv->points.size();
 309                         for( int j=0; j<prev_pt_sz; ++j ) {
 310                                 SketcherPoint &pt = *pcv->points[j], *nt = 0;
 311                                 k = next_pt_sz;  // associated by id in next
 312                                 while( --k >= 0 && pt.id != (nt=ncv->points[k])->id );
 313                                 coord x = pt.x, y = pt.y;
 314                                 if( k >= 0 ) {
 315                                         if( x != nt->x )
 316                                                 x = x * prev_scale + nt->x * next_scale;
 317                                         if( y != nt->y )
 318                                                 y = y * prev_scale + nt->y * next_scale;
 319                                 }
 320                                 cv->points.append(new SketcherPoint(pt.id, pt.pty, x, y));
 321                         }
 322                 }
 323                 else
 324                         cv->copy_from(*pcv);
 325         }
 326 }
 327
 328 void SketcherConfig::limits()
 329 {
 330 }
 331
 332
 333 Sketcher::Sketcher(PluginServer *server)
 334  : PluginVClient(server)
 335 {
 336         img = 0;
 337         out = 0;
 338         overlay_frame = 0;
 339 }
 340
 341 Sketcher::~Sketcher()
 342 {
 343         delete img;
 344         delete out;
 345         delete overlay_frame;
 346 }
 347
 348 const char* Sketcher::plugin_title() { return N_("Sketcher"); }
 349 int Sketcher::is_realtime() { return 1; }
 350
 351 NEW_WINDOW_MACRO(Sketcher, SketcherWindow);
 352 LOAD_CONFIGURATION_MACRO(Sketcher, SketcherConfig)
 353
 354 void Sketcher::save_data(KeyFrame *keyframe)
 355 {
 356         FileXML output;
 357 // cause data to be stored directly in text
 358         output.set_shared_output(keyframe->xbuf);
 359
 360         output.tag.set_title("SKETCHER");
 361         output.tag.set_property("DRAG", config.drag);
 362         output.tag.set_property("CV_SELECTED", config.cv_selected);
 363         output.tag.set_property("PT_SELECTED", config.pt_selected);
 364         output.append_tag();
 365         output.append_newline();
 366         for( int i=0,n=config.curves.size(); i<n; ++i ) {
 367                 config.curves[i]->save_data(output);
 368         }
 369         output.tag.set_title("/SKETCHER");
 370         output.append_tag();
 371         output.append_newline();
 372         output.terminate_string();
 373 }
 374
 375 void Sketcher::read_data(KeyFrame *keyframe)
 376 {
 377         FileXML input;
 378         input.set_shared_input(keyframe->xbuf);
 379         config.curves.remove_all_objects();
 380         int result = 0;
 381         SketcherCurve *cv = 0;
 382
 383         while( !(result=input.read_tag()) ) {
 384                 if( input.tag.title_is("SKETCHER") ) {
 385                         config.drag = input.tag.get_property("DRAG", config.drag);
 386                         config.cv_selected = input.tag.get_property("CV_SELECTED", 0);
 387                         config.pt_selected = input.tag.get_property("PT_SELECTED", 0);
 388                 }
 389                 else if( !strncmp(input.tag.get_title(),"CURVE_",6) ) {
 390                         cv = new SketcherCurve();
 391                         cv->read_data(input);
 392                         config.curves.append(cv);
 393                 }
 394                 else if( !strncmp(input.tag.get_title(),"/CURVE_",7) )
 395                         cv = 0;
 396                 else if( !strncmp(input.tag.get_title(),"POINT_",6) ) {
 397                         if( cv ) {
 398                                 SketcherPoint *pt = new SketcherPoint();
 399                                 pt->read_data(input);
 400                                 cv->points.append(pt);
 401                         }
 402                         else
 403                                 printf("Sketcher::read_data: no curve for point\n");
 404                 }
 405         }
 406
 407         if( !config.curves.size() )
 408                 new_curve();
 409         config.limits();
 410 }
 411
 412 void Sketcher::update_gui()
 413 {
 414         if( !thread ) return;
 415         thread->window->lock_window("Sketcher::update_gui");
 416         if( load_configuration() ) {
 417                 SketcherWindow *window = (SketcherWindow*)thread->window;
 418                 window->update_gui();
 419                 window->flush();
 420         }
 421         thread->window->unlock_window();
 422 }
 423
 424 void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color, int d)
 425 {
 426         int r = d/2+1, x = pt->x, y = pt->y;
 427         vfrm->set_pixel_color(color);
 428         vfrm->draw_smooth(x-r,y+0, x-r, y-r, x+0,y-r);
 429         vfrm->draw_smooth(x+0,y-r, x+r, y-r, x+r,y+0);
 430         vfrm->draw_smooth(x+r,y+0, x+r, y+r, x+0,y+r);
 431         vfrm->draw_smooth(x+0,y+r, x-r, y+r, x-r,y+0);
 432         vfrm->draw_x(pt->x, pt->y, d);
 433 }
 434 void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color)
 435 {
 436         draw_point(vfrm, pt, color, bmax(w,h)/200 + 2);
 437 }
 438
 439
 440 int SketcherVPen::draw_pixel(int x, int y)
 441 {
 442         if( x >= 0 && x < vfrm->get_w() &&
 443             y >= 0 && y < vfrm->get_h() )
 444                 msk[vfrm->get_w()*y + x] = 0xff;
 445         return 0;
 446 }
 447
 448 int SketcherPenSquare::draw_pixel(int x, int y)
 449 {
 450         vfrm->draw_line(x-n, y, x+n, y);
 451         for( int i=-n; i<n; ++i )
 452                 vfrm->draw_line(x-n, y+i, x+n, y+i);
 453         return SketcherVPen::draw_pixel(x, y);
 454 }
 455 int SketcherPenPlus::draw_pixel(int x, int y)
 456 {
 457         if( n > 1 ) {
 458                 vfrm->draw_line(x-n, y, x+n, y);
 459                 vfrm->draw_line(x, y-n, x, y+n);
 460         }
 461         else
 462                 vfrm->draw_pixel(x, y);
 463         return SketcherVPen::draw_pixel(x, y);
 464 }
 465 int SketcherPenSlant::draw_pixel(int x, int y)
 466 {
 467         vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
 468         vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
 469         vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
 470         return SketcherVPen::draw_pixel(x, y);
 471 }
 472 int SketcherPenXlant::draw_pixel(int x, int y)
 473 {
 474         vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
 475         vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
 476         vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
 477         vfrm->draw_line(x-n,   y-n,   x+n,   y+n);
 478         vfrm->draw_line(x-n+1, y-n,   x+n+1, y+n);
 479         vfrm->draw_line(x-n,   y-n+1, x+n,   y-n+1);
 480         return SketcherVPen::draw_pixel(x, y);
 481 }
 482
 483
 484 SketcherVPen *SketcherCurve::new_vpen(VFrame *out)
 485 {
 486         switch( pen ) {
 487         case PEN_SQUARE: return new SketcherPenSquare(out, width);
 488         case PEN_PLUS:   return new SketcherPenPlus(out, width);
 489         case PEN_SLANT:  return new SketcherPenSlant(out, width);
 490         case PEN_XLANT:  return new SketcherPenXlant(out, width);
 491         }
 492         return 0;
 493 }
 494
 495 static int intersects_at(float &x, float &y,
 496                 float ax,float ay, float bx, float by, float cx,float cy,  // line slope ab thru c
 497                 float dx,float dy, float ex, float ey, float fx,float fy, // line slope de thru f
 498                 float mx=0)
 499 {
 500         float badx = bx - ax, bady = by - ay;
 501         float eddx = ex - dx, eddy = ey - dy;
 502         float d = badx*eddy - bady*eddx;
 503         int ret = 0;
 504         if( fabsf(d) < 1 ) { ret = 1;  d = signbit(d) ? -1 : 1; }
 505         x = (badx*cy*eddx - badx*eddx*fy + badx*eddy*fx - bady*cx*eddx) / d;
 506         y = (badx*cy*eddy - bady*cx*eddy - bady*eddx*fy + bady*eddy*fx) / d;
 507         if( mx > 0 ) { bclamp(x, -mx,mx);  bclamp(y, -mx,mx); }
 508         return ret;
 509 }
 510
 511 static void smooth_axy(float &ax, float &ay,
 512         float bx, float by, float cx, float cy, float dx, float dy)
 513 {
 514 //middle of bd reflected around ctr
 515 // point ctr = (b+d)/2, dv=c-ctr, a=ctr-dv;
 516         float xc = (bx+dx)*.5f, yc = (by+dy)*.5f;
 517         float xd = cx - xc, yd = cy - yc;
 518         ax = xc - xd;  ay = yc - yd;
 519 }
 520 static void smooth_dxy(float &dx, float &dy,
 521         float ax, float ay, float bx, float by, float cx, float cy)
 522 {
 523 //middle of ac reflected around ctr
 524 // point ctr = (a+c)/2, dv=c-ctr, d=ctr-dv;
 525         float xc = (ax+cx)*.5f, yc = (ay+cy)*.5f;
 526         float xd = bx - xc, yd = by - yc;
 527         dx = xc - xd;  dy = yc - yd;
 528 }
 529
 530 #if 0
 531 static int convex(float ax,float ay, float bx,float by,
 532                   float cx,float cy, float dx,float dy)
 533 {
 534         float abdx = bx-ax, abdy = by-ay;
 535         float acdx = cx-ax, acdy = cy-ay;
 536         float bcdx = cx-bx, bcdy = cy-by;
 537         float bddx = dx-bx, bddy = dy-by;
 538         float abc = abdx*acdy - abdy*acdx;
 539         float bcd = bcdx*bddy - bcdy*bddx;
 540         float v = abc * bcd;
 541         return !v ? 0 : v>0 ? 1 : -1;
 542 }
 543 #endif
 544
 545
 546 class FillRegion
 547 {
 548         class segment { public: int y, lt, rt; };
 549         ArrayList<segment> stack;
 550
 551         void push(int y, int lt, int rt) {
 552                 segment &seg = stack.append();
 553                 seg.y = y;  seg.lt = lt;  seg.rt = rt;
 554         }
 555         void pop(int &y, int &lt, int &rt) {
 556                 segment &seg = stack.last();
 557                 y = seg.y;  lt = seg.lt;  rt = seg.rt;
 558                 stack.remove();
 559         }
 560
 561         VFrame *img;
 562         uint8_t *msk;
 563         int w, h, nxt;
 564         SketcherPoints &points;
 565 public:
 566         SketcherPoint *next();
 567         bool exists() { return stack.size() > 0; }
 568         void start_at(int x, int y);
 569         void run();
 570         FillRegion(SketcherPoints &pts, SketcherVPen *vpen);
 571         ~FillRegion();
 572 };
 573
 574 FillRegion::FillRegion(SketcherPoints &pts, SketcherVPen *vpen)
 575  : points(pts)
 576 {
 577         this->img = vpen->vfrm;
 578         this->msk = vpen->msk;
 579         this->w = img->get_w();
 580         this->h = img->get_h();
 581         nxt = 0;
 582 }
 583 FillRegion::~FillRegion()
 584 {
 585 }
 586
 587 void FillRegion::start_at(int x, int y)
 588 {
 589         bclamp(x, 0, w-1);
 590         bclamp(y, 0, h-1);
 591         push(y, x, x);
 592 }
 593
 594 void FillRegion::run()
 595 {
 596         while( stack.size() > 0 ) {
 597                 int y, ilt, irt;
 598                 pop(y, ilt, irt);
 599                 int ofs = y*w + ilt;
 600                 for( int x=ilt; x<=irt; ++x,++ofs ) {
 601                         if( msk[ofs] ) continue;
 602                         msk[ofs] = 0xff;
 603                         img->draw_pixel(x, y);
 604                         int lt = x, rt = x;
 605                         int lofs = ofs;
 606                         for( int i=lt; --i>=0; ) {
 607                                 if( msk[--lofs] ) break;
 608                                 img->draw_pixel(i, y);
 609                                 msk[lofs] = 0xff;  lt = i;
 610                         }
 611                         int rofs = ofs;
 612                         for( int i=rt; ++i< w; ) {
 613                                 if( msk[++rofs] ) break;
 614                                 img->draw_pixel(i, y);
 615                                 msk[rofs] = 0xff;  rt = i;
 616                         }
 617                         if( y+1 <  h ) push(y+1, lt, rt);
 618                         if( y-1 >= 0 ) push(y-1, lt, rt);
 619                 }
 620         }
 621 }
 622
 623 SketcherPoint *FillRegion::next()
 624 {
 625         while( nxt < points.size() ) {
 626                 SketcherPoint *pt = points[nxt];
 627                 if( pt->pty != PTY_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->pty ) {
 654                         case PTY_LINE:
 655                                 vpen->draw_line(bx, by, cx, cy);
 656                                 break;
 657                         case PTY_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->pty ) {
 668                 case PTY_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 PTY_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                 int m = cv->points.size();
 740                 if( !m || cv->pen == PTY_OFF ) continue;
 741                 img->clear_frame();
 742                 img->set_pixel_color(cv->color, (~cv->color>>24)&0xff);
 743                 cv->draw(img);
 744                 overlay_frame->overlay(out, img, 0,0,w,h, 0,0,w,h,
 745                                 1.f, TRANSFER_NORMAL, NEAREST_NEIGHBOR);
 746         }
 747
 748         if( config.drag ) {
 749                 for( int ci=0, n=config.curves.size(); ci<n; ++ci ) {
 750                         SketcherCurve *cv = config.curves[ci];
 751                         for( int pi=0,m=cv->points.size(); pi<m; ++pi ) {
 752                                 int color = pi==config.pt_selected && ci==config.cv_selected ?
 753                                         RED : cv->color ;
 754                                 draw_point(out, cv->points[pi], color);
 755                         }
 756                 }
 757         }
 758
 759         if( output != out )
 760                 output->transfer_from(out);
 761         else
 762                 out = 0;
 763
 764         return 0;
 765 }
 766
 767 void SketcherPoints::dump()
 768 {
 769         for( int i=0; i<size(); ++i ) {
 770                 SketcherPoint *pt = get(i);
 771                 printf("  Pt %d, id=%d, pty=%s, x=%0.1f, y=%0.1f\n",
 772                         i, pt->id, pt_type[pt->pty], pt->x, pt->y);
 773         }
 774 }
 775 void SketcherCurves::dump()
 776 {
 777         for( int i=0; i<size(); ++i ) {
 778                 SketcherCurve *cv = get(i);
 779                 printf("Curve %d, id=%d, pen=%s, r=%d, color=%02x%02x%02x%02x, %d points\n",
 780                         i, cv->id, cv_pen[cv->pen], cv->width, (~cv->color>>24)&0xff,
 781                         (cv->color>>16)&0xff, (cv->color>>8)&0xff, (cv->color>>0)&0xff,
 782                         cv->points.size());
 783                 cv->points.dump();
 784         }
 785 }
 786 void SketcherConfig::dump()
 787 {
 788         printf("Config drag=%d, cv_selected=%d, pt_selected=%d %d curves\n",
 789                         drag, cv_selected, pt_selected, curves.size());
 790         curves.dump();
 791 }
 792