1 /* === S Y N F I G ========================================================= */
2 /*! \file curvewarp.cpp
3 ** \brief Implementation of the "Curve Warp" layer
8 ** Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
9 ** Copyright (c) 2007-2008 Chris Moore
11 ** This package is free software; you can redistribute it and/or
12 ** modify it under the terms of the GNU General Public License as
13 ** published by the Free Software Foundation; either version 2 of
14 ** the License, or (at your option) any later version.
16 ** This package is distributed in the hope that it will be useful,
17 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
18 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19 ** General Public License for more details.
22 ** === N O T E S ===========================================================
24 ** ========================================================================= */
26 /* === H E A D E R S ======================================================= */
35 #include "curvewarp.h"
37 #include <synfig/context.h>
38 #include <synfig/paramdesc.h>
39 #include <synfig/surface.h>
40 #include <synfig/valuenode.h>
41 #include <ETL/calculus>
45 /* === M A C R O S ========================================================= */
47 #define FAKE_TANGENT_STEP 0.000001
50 /* === G L O B A L S ======================================================= */
52 SYNFIG_LAYER_INIT(CurveWarp);
53 SYNFIG_LAYER_SET_NAME(CurveWarp,"curve_warp");
54 SYNFIG_LAYER_SET_LOCAL_NAME(CurveWarp,N_("Curve Warp"));
55 SYNFIG_LAYER_SET_CATEGORY(CurveWarp,N_("Distortions"));
56 SYNFIG_LAYER_SET_VERSION(CurveWarp,"0.0");
57 SYNFIG_LAYER_SET_CVS_ID(CurveWarp,"$Id$");
59 /* === P R O C E D U R E S ================================================= */
61 inline float calculate_distance(const std::vector<synfig::BLinePoint>& bline)
63 std::vector<synfig::BLinePoint>::const_iterator iter,next,ret;
64 std::vector<synfig::BLinePoint>::const_iterator end(bline.end());
68 if (bline.empty()) return dist;
73 for(;next!=end;iter=next++)
76 etl::hermite<Vector> curve(iter->get_vertex(), next->get_vertex(), iter->get_tangent2(), next->get_tangent1());
83 std::vector<synfig::BLinePoint>::const_iterator
84 find_closest_to_bline(bool fast, const std::vector<synfig::BLinePoint>& bline,const Point& p,float& t, float& len, bool& extreme)
86 std::vector<synfig::BLinePoint>::const_iterator iter,next,ret;
87 std::vector<synfig::BLinePoint>::const_iterator end(bline.end());
90 float dist(100000000000.0);
92 float best_pos(0), best_len(0);
93 etl::hermite<Vector> best_curve;
97 bool first = true, last = false;
100 for(;next!=end;iter=next++)
103 etl::hermite<Vector> curve(iter->get_vertex(), next->get_vertex(), iter->get_tangent2(), next->get_tangent1());
109 #define POINT_CHECK(x) bp=curve(x); thisdist=(bp-p).mag_squared(); if(thisdist<dist) { extreme = (first&&x<0.01); ret=iter; best_len = total_len; dist=thisdist; best_curve=curve; last=true; best_pos=x;}
110 POINT_CHECK(0.0001); POINT_CHECK((1.0/6)); POINT_CHECK((2.0/6)); POINT_CHECK((3.0/6));
111 POINT_CHECK((4.0/6)); POINT_CHECK((5.0/6)); POINT_CHECK(0.9999);
115 float pos = curve.find_closest(fast, p);
116 thisdist=(curve(pos)-p).mag_squared();
119 extreme = (first && pos == 0);
124 best_len = total_len;
128 total_len += curve.length();
135 len = best_len + best_curve.find_distance(0,best_curve.find_closest(fast, p));
136 if (last && t > .99) extreme = true;
140 len = best_len + best_curve.find_distance(0,best_pos);
141 if (last && t == 1) extreme = true;
146 /* === M E T H O D S ======================================================= */
151 curve_length_=calculate_distance(bline);
152 perp_ = (end_point - start_point).perp().norm();
155 CurveWarp::CurveWarp():
158 start_point(-2.5,-0.5),
162 bline.push_back(BLinePoint());
163 bline.push_back(BLinePoint());
164 bline[0].set_vertex(Point(-2.5,0));
165 bline[1].set_vertex(Point( 2.5,0));
166 bline[0].set_tangent(Point(1, 0.1));
167 bline[1].set_tangent(Point(1, -0.1));
168 bline[0].set_width(1.0f);
169 bline[1].set_width(1.0f);
172 Layer::Vocab voc(get_param_vocab());
173 Layer::fill_static(voc);
177 CurveWarp::transform(const Point &point_, Real *dist, Real *along, int quality)const
183 bool edge_case = false;
190 else if(bline.size()==1)
192 tangent=bline.front().get_tangent1();
193 p1=bline.front().get_vertex();
194 thickness=bline.front().get_width();
200 Point point(point_-origin);
202 std::vector<synfig::BLinePoint>::const_iterator iter,next;
204 // Figure out the BLinePoint we will be using,
205 next=find_closest_to_bline(fast,bline,point,t,len,extreme);
208 if(next==bline.end()) next=bline.begin();
211 etl::hermite<Vector> curve(iter->get_vertex(), next->get_vertex(), iter->get_tangent2(), next->get_tangent1());
213 // Setup the derivative function
214 etl::derivative<etl::hermite<Vector> > deriv(curve);
216 int search_iterations(7);
218 if(quality<=6)search_iterations=7;
219 else if(quality<=7)search_iterations=6;
220 else if(quality<=8)search_iterations=5;
221 else search_iterations=4;
223 // Figure out the closest point on the curve
224 if (fast) t = curve.find_closest(fast, point,search_iterations);
226 // Calculate our values
227 p1=curve(t); // the closest point on the curve
228 tangent=deriv(t); // the tangent at that point
230 // if the point we're nearest to is at either end of the
231 // bline, our distance from the curve is the distance from the
232 // point on the curve. we need to know which side of the
233 // curve we're on, so find the average of the two tangents at
235 if (t<0.00001 || t>0.99999)
237 bool zero_tangent = (tangent[0] == 0 && tangent[1] == 0);
241 if (iter->get_split_tangent_flag() || zero_tangent)
243 // fake the current tangent if we need to
244 if (zero_tangent) tangent = curve(FAKE_TANGENT_STEP) - curve(0);
246 // calculate the other tangent
247 Vector other_tangent(iter->get_tangent1());
248 if (other_tangent[0] == 0 && other_tangent[1] == 0)
250 // find the previous blinepoint
251 std::vector<synfig::BLinePoint>::const_iterator prev;
252 if (iter != bline.begin()) (prev = iter)--;
255 etl::hermite<Vector> other_curve(prev->get_vertex(), iter->get_vertex(), prev->get_tangent2(), iter->get_tangent1());
256 other_tangent = other_curve(1) - other_curve(1-FAKE_TANGENT_STEP);
259 // normalise and sum the two tangents
260 tangent=(other_tangent.norm()+tangent.norm());
266 if (next->get_split_tangent_flag() || zero_tangent)
268 // fake the current tangent if we need to
269 if (zero_tangent) tangent = curve(1) - curve(1-FAKE_TANGENT_STEP);
271 // calculate the other tangent
272 Vector other_tangent(next->get_tangent2());
273 if (other_tangent[0] == 0 && other_tangent[1] == 0)
275 // find the next blinepoint
276 std::vector<synfig::BLinePoint>::const_iterator next2(next);
277 if (++next2 == bline.end())
280 etl::hermite<Vector> other_curve(next->get_vertex(), next2->get_vertex(), next->get_tangent2(), next2->get_tangent1());
281 other_tangent = other_curve(FAKE_TANGENT_STEP) - other_curve(0);
284 // normalise and sum the two tangents
285 tangent=(other_tangent.norm()+tangent.norm());
290 tangent = tangent.norm();
292 // the width of the bline at the closest point on the curve
293 thickness=(next->get_width()-iter->get_width())*t+iter->get_width();
296 if (thickness < TOO_THIN && thickness > -TOO_THIN)
298 if (thickness > 0) thickness = TOO_THIN;
299 else thickness = -TOO_THIN;
308 std::vector<synfig::BLinePoint>::const_iterator iter(bline.begin());
309 tangent = iter->get_tangent1().norm();
314 std::vector<synfig::BLinePoint>::const_iterator iter(--bline.end());
315 tangent = iter->get_tangent2().norm();
318 len += (point_-origin - p1)*tangent;
319 diff = tangent.perp();
323 diff=(p1-(point_-origin));
324 if(diff*tangent.perp()<0) diff=-diff;
330 // diff is a unit vector perpendicular to the bline
331 const Real unscaled_distance((point_-origin - p1)*diff);
332 if (dist) *dist = unscaled_distance;
333 if (along) *along = len;
334 return ((start_point + (end_point - start_point) * len / curve_length_) +
335 perp_ * unscaled_distance/(thickness*perp_width));
338 synfig::Layer::Handle
339 CurveWarp::hit_check(synfig::Context context, const synfig::Point &point)const
341 return context.hit_check(transform(point));
345 CurveWarp::set_param(const String & param, const ValueBase &value)
353 if(param=="bline" && value.get_type()==ValueBase::TYPE_LIST)
361 IMPORT_AS(origin,"offset");
367 CurveWarp::get_param(const String & param)const
383 CurveWarp::get_param_vocab()const
387 ret.push_back(ParamDesc("origin")
388 .set_local_name(_("Origin")));
390 ret.push_back(ParamDesc("perp_width")
391 .set_local_name(_("Width"))
392 .set_origin("start_point"));
394 ret.push_back(ParamDesc("start_point")
395 .set_local_name(_("Start Point"))
396 .set_connect("end_point"));
398 ret.push_back(ParamDesc("end_point")
399 .set_local_name(_("End Point")));
401 ret.push_back(ParamDesc("bline")
402 .set_local_name(_("Vertices"))
403 .set_origin("origin")
405 .set_description(_("A list of BLine Points")));
407 ret.push_back(ParamDesc("fast")
408 .set_local_name(_("Fast")));
414 CurveWarp::get_color(Context context, const Point &point)const
416 return context.get_color(transform(point));
420 CurveWarp::accelerated_render(Context context,Surface *surface,int quality, const RendDesc &renddesc, ProgressCallback *cb)const
422 SuperCallback stageone(cb,0,9000,10000);
423 SuperCallback stagetwo(cb,9000,10000,10000);
427 const Real pw(renddesc.get_pw()),ph(renddesc.get_ph());
428 Point tl(renddesc.get_tl());
429 Point br(renddesc.get_br());
430 const int w(renddesc.get_w());
431 const int h(renddesc.get_h());
433 // find a bounding rectangle for the context we need to render
434 // todo: find a better way of doing this - this way doesn't work
435 Rect src_rect(transform(tl));
438 Real min_dist(999999), max_dist(-999999), min_along(999999), max_along(-999999);
440 #define UPDATE_DIST \
441 if (dist < min_dist) min_dist = dist; \
442 if (dist > max_dist) max_dist = dist; \
443 if (along < min_along) min_along = along; \
444 if (along > max_along) max_along = along
446 // look along the top and bottom edges
447 pos1[0] = pos2[0] = tl[0]; pos1[1] = tl[1]; pos2[1] = br[1];
448 for (x = 0; x < w; x++, pos1[0] += pw, pos2[0] += pw)
450 src_rect.expand(transform(pos1, &dist, &along)); UPDATE_DIST;
451 src_rect.expand(transform(pos2, &dist, &along)); UPDATE_DIST;
454 // look along the left and right edges
455 pos1[0] = tl[0]; pos2[0] = br[0]; pos1[1] = pos2[1] = tl[1];
456 for (y = 0; y < h; y++, pos1[1] += ph, pos2[1] += ph)
458 src_rect.expand(transform(pos1, &dist, &along)); UPDATE_DIST;
459 src_rect.expand(transform(pos2, &dist, &along)); UPDATE_DIST;
462 // look along the diagonals
463 const int max_wh(std::max(w,h));
464 const Real inc_x((br[0]-tl[0])/max_wh),inc_y((br[1]-tl[1])/max_wh);
465 pos1[0] = pos2[0] = tl[0]; pos1[1] = tl[1]; pos2[1] = br[1];
466 for (x = 0; x < max_wh; x++, pos1[0] += inc_x, pos2[0] = pos1[0], pos1[1]+=inc_y, pos2[1]-=inc_y)
468 src_rect.expand(transform(pos1, &dist, &along)); UPDATE_DIST;
469 src_rect.expand(transform(pos2, &dist, &along)); UPDATE_DIST;
473 // look at each blinepoint
474 std::vector<synfig::BLinePoint>::const_iterator iter;
475 for (iter=bline.begin(); iter!=bline.end(); iter++)
476 src_rect.expand(transform(iter->get_vertex()+origin, &dist, &along)); UPDATE_DIST;
479 Point src_tl(src_rect.get_min());
480 Point src_br(src_rect.get_max());
482 Vector ab((end_point - start_point).norm());
483 Angle::tan ab_angle(ab[1], ab[0]);
485 Real used_length = max_along - min_along;
486 Real render_width = max_dist - min_dist;
488 int src_w = (abs(used_length*Angle::cos(ab_angle).get()) +
489 abs(render_width*Angle::sin(ab_angle).get())) / abs(pw);
490 int src_h = (abs(used_length*Angle::sin(ab_angle).get()) +
491 abs(render_width*Angle::cos(ab_angle).get())) / abs(ph);
493 Real src_pw((src_br[0] - src_tl[0]) / src_w);
494 Real src_ph((src_br[1] - src_tl[1]) / src_h);
496 if (src_pw > abs(pw))
498 src_w = int((src_br[0] - src_tl[0]) / abs(pw));
499 src_pw = (src_br[0] - src_tl[0]) / src_w;
502 if (src_ph > abs(ph))
504 src_h = int((src_br[1] - src_tl[1]) / abs(ph));
505 src_ph = (src_br[1] - src_tl[1]) / src_h;
509 if (src_w > MAXPIX) src_w = MAXPIX;
510 if (src_h > MAXPIX) src_h = MAXPIX;
512 // this is an attempt to remove artifacts around tile edges - the
513 // cubic interpolation uses at most 2 pixels either side of the
514 // target pixel, so add an extra 2 pixels around the tile on all
516 src_tl -= (Point(src_pw,src_ph)*2);
517 src_br += (Point(src_pw,src_ph)*2);
520 src_pw = (src_br[0] - src_tl[0]) / src_w;
521 src_ph = (src_br[1] - src_tl[1]) / src_h;
523 // set up a renddesc for the context to render
524 RendDesc src_desc(renddesc);
525 src_desc.clear_flags();
526 src_desc.set_tl(src_tl);
527 src_desc.set_br(src_br);
528 src_desc.set_wh(src_w, src_h);
530 // render the context onto a new surface
532 source.set_wh(src_w,src_h);
533 if(!context.accelerated_render(&source,quality,src_desc,&stageone))
539 surface->set_wh(w,h);
542 if(quality<=4) // CUBIC
543 for(y=0,pos[1]=tl[1];y<h;y++,pos[1]+=ph)
545 for(x=0,pos[0]=tl[0];x<w;x++,pos[0]+=pw)
548 u=(tmp[0]-src_tl[0])/src_pw;
549 v=(tmp[1]-src_tl[1])/src_ph;
550 if(u<0 || v<0 || u>=src_w || v>=src_h || isnan(u) || isnan(v))
551 (*surface)[y][x]=context.get_color(tmp);
553 (*surface)[y][x]=source.cubic_sample(u,v);
555 if((y&31)==0 && cb && !stagetwo.amount_complete(y,h)) return false;
557 else if (quality<=6) // INTERPOLATION_LINEAR
558 for(y=0,pos[1]=tl[1];y<h;y++,pos[1]+=ph)
560 for(x=0,pos[0]=tl[0];x<w;x++,pos[0]+=pw)
563 u=(tmp[0]-src_tl[0])/src_pw;
564 v=(tmp[1]-src_tl[1])/src_ph;
565 if(u<0 || v<0 || u>=src_w || v>=src_h || isnan(u) || isnan(v))
566 (*surface)[y][x]=context.get_color(tmp);
568 (*surface)[y][x]=source.linear_sample(u,v);
570 if((y&31)==0 && cb && !stagetwo.amount_complete(y,h)) return false;
572 else // NEAREST_NEIGHBOR
573 for(y=0,pos[1]=tl[1];y<h;y++,pos[1]+=ph)
575 for(x=0,pos[0]=tl[0];x<w;x++,pos[0]+=pw)
578 u=(tmp[0]-src_tl[0])/src_pw;
579 v=(tmp[1]-src_tl[1])/src_ph;
580 if(u<0 || v<0 || u>=src_w || v>=src_h || isnan(u) || isnan(v))
581 (*surface)[y][x]=context.get_color(tmp);
583 (*surface)[y][x]=source[floor_to_int(v)][floor_to_int(u)];
585 if((y&31)==0 && cb && !stagetwo.amount_complete(y,h)) return false;
588 // Mark our progress as finished
589 if(cb && !cb->amount_complete(10000,10000))