Path: blob/master/thirdparty/embree/kernels/subdiv/catmullrom_curve.h
9913 views
// Copyright 2009-2021 Intel Corporation1// SPDX-License-Identifier: Apache-2.023#pragma once45#include "../common/default.h"6#include "bezier_curve.h"7#include "../common/scene_curves.h"89/*1011Implements Catmul Rom curves with control points p0, p1, p2, p3. At12t=0 the curve goes through p1, with tangent (p2-p0)/2, and for t=113the curve goes through p2 with tangent (p3-p2)/2.1415*/1617namespace embree18{19class CatmullRomBasis20{21public:2223template<typename T>24static __forceinline Vec4<T> eval(const T& u)25{26const T t = u;27const T s = T(1.0f) - u;28const T n0 = - t * s * s;29const T n1 = 2.0f + t * t * (3.0f * t - 5.0f);30const T n2 = 2.0f + s * s * (3.0f * s - 5.0f);31const T n3 = - s * t * t;32return T(0.5f) * Vec4<T>(n0, n1, n2, n3);33}3435template<typename T>36static __forceinline Vec4<T> derivative(const T& u)37{38const T t = u;39const T s = 1.0f - u;40const T n0 = - s * s + 2.0f * s * t;41const T n1 = 2.0f * t * (3.0f * t - 5.0f) + 3.0f * t * t;42const T n2 = 2.0f * s * (3.0f * t + 2.0f) - 3.0f * s * s;43const T n3 = -2.0f * s * t + t * t;44return T(0.5f) * Vec4<T>(n0, n1, n2, n3);45}4647template<typename T>48static __forceinline Vec4<T> derivative2(const T& u)49{50const T t = u;51const T n0 = -3.0f * t + 2.0f;52const T n1 = 9.0f * t - 5.0f;53const T n2 = -9.0f * t + 4.0f;54const T n3 = 3.0f * t - 1.0f;55return Vec4<T>(n0, n1, n2, n3);56}57};5859struct PrecomputedCatmullRomBasis60{61enum { N = 16 };62public:63PrecomputedCatmullRomBasis() {}64PrecomputedCatmullRomBasis(int shift);6566/* basis for bspline evaluation */67public:68float c0[N+1][N+1];69float c1[N+1][N+1];70float c2[N+1][N+1];71float c3[N+1][N+1];7273/* basis for bspline derivative evaluation */74public:75float d0[N+1][N+1];76float d1[N+1][N+1];77float d2[N+1][N+1];78float d3[N+1][N+1];79};80extern PrecomputedCatmullRomBasis catmullrom_basis0;81extern PrecomputedCatmullRomBasis catmullrom_basis1;8283template<typename Vertex>84struct CatmullRomCurveT85{86Vertex v0,v1,v2,v3;8788__forceinline CatmullRomCurveT() {}8990__forceinline CatmullRomCurveT(const Vertex& v0, const Vertex& v1, const Vertex& v2, const Vertex& v3)91: v0(v0), v1(v1), v2(v2), v3(v3) {}9293__forceinline Vertex begin() const {94return v1;95}9697__forceinline Vertex end() const {98return v2;99}100101__forceinline Vertex center() const {102return 0.5f*(v0+v1);103}104105__forceinline BBox<Vertex> bounds() const {106return merge(BBox<Vertex>(v0),BBox<Vertex>(v1),BBox<Vertex>(v2),BBox<Vertex>(v3));107}108109__forceinline friend CatmullRomCurveT operator -( const CatmullRomCurveT& a, const Vertex& b ) {110return CatmullRomCurveT(a.v0-b,a.v1-b,a.v2-b,a.v3-b);111}112113__forceinline CatmullRomCurveT<Vec3ff> xfm_pr(const LinearSpace3fa& space, const Vec3fa& p) const114{115const Vec3ff q0(xfmVector(space,v0-p), v0.w);116const Vec3ff q1(xfmVector(space,v1-p), v1.w);117const Vec3ff q2(xfmVector(space,v2-p), v2.w);118const Vec3ff q3(xfmVector(space,v3-p), v3.w);119return CatmullRomCurveT<Vec3ff>(q0,q1,q2,q3);120}121122__forceinline Vertex eval(const float t) const123{124const Vec4<float> b = CatmullRomBasis::eval(t);125return madd(b.x,v0,madd(b.y,v1,madd(b.z,v2,b.w*v3)));126}127128__forceinline Vertex eval_du(const float t) const129{130const Vec4<float> b = CatmullRomBasis::derivative(t);131return madd(b.x,v0,madd(b.y,v1,madd(b.z,v2,b.w*v3)));132}133134__forceinline Vertex eval_dudu(const float t) const135{136const Vec4<float> b = CatmullRomBasis::derivative2(t);137return madd(b.x,v0,madd(b.y,v1,madd(b.z,v2,b.w*v3)));138}139140__forceinline void eval(const float t, Vertex& p, Vertex& dp) const141{142p = eval(t);143dp = eval_du(t);144}145146__forceinline void eval(const float t, Vertex& p, Vertex& dp, Vertex& ddp) const147{148p = eval(t);149dp = eval_du(t);150ddp = eval_dudu(t);151}152153template<int M>154__forceinline Vec4vf<M> veval(const vfloat<M>& t) const155{156const Vec4vf<M> b = CatmullRomBasis::eval(t);157return madd(b.x, Vec4vf<M>(v0), madd(b.y, Vec4vf<M>(v1), madd(b.z, Vec4vf<M>(v2), b.w * Vec4vf<M>(v3))));158}159160template<int M>161__forceinline Vec4vf<M> veval_du(const vfloat<M>& t) const162{163const Vec4vf<M> b = CatmullRomBasis::derivative(t);164return madd(b.x, Vec4vf<M>(v0), madd(b.y, Vec4vf<M>(v1), madd(b.z, Vec4vf<M>(v2), b.w * Vec4vf<M>(v3))));165}166167template<int M>168__forceinline Vec4vf<M> veval_dudu(const vfloat<M>& t) const169{170const Vec4vf<M> b = CatmullRomBasis::derivative2(t);171return madd(b.x, Vec4vf<M>(v0), madd(b.y, Vec4vf<M>(v1), madd(b.z, Vec4vf<M>(v2), b.w * Vec4vf<M>(v3))));172}173174template<int M>175__forceinline void veval(const vfloat<M>& t, Vec4vf<M>& p, Vec4vf<M>& dp) const176{177p = veval<M>(t);178dp = veval_du<M>(t);179}180181template<int M>182__forceinline Vec4vf<M> eval0(const int ofs, const int size) const183{184assert(size <= PrecomputedCatmullRomBasis::N);185assert(ofs <= size);186return madd(vfloat<M>::loadu(&catmullrom_basis0.c0[size][ofs]), Vec4vf<M>(v0),187madd(vfloat<M>::loadu(&catmullrom_basis0.c1[size][ofs]), Vec4vf<M>(v1),188madd(vfloat<M>::loadu(&catmullrom_basis0.c2[size][ofs]), Vec4vf<M>(v2),189vfloat<M>::loadu(&catmullrom_basis0.c3[size][ofs]) * Vec4vf<M>(v3))));190}191192template<int M>193__forceinline Vec4vf<M> eval1(const int ofs, const int size) const194{195assert(size <= PrecomputedCatmullRomBasis::N);196assert(ofs <= size);197return madd(vfloat<M>::loadu(&catmullrom_basis1.c0[size][ofs]), Vec4vf<M>(v0),198madd(vfloat<M>::loadu(&catmullrom_basis1.c1[size][ofs]), Vec4vf<M>(v1),199madd(vfloat<M>::loadu(&catmullrom_basis1.c2[size][ofs]), Vec4vf<M>(v2),200vfloat<M>::loadu(&catmullrom_basis1.c3[size][ofs]) * Vec4vf<M>(v3))));201}202203template<int M>204__forceinline Vec4vf<M> derivative0(const int ofs, const int size) const205{206assert(size <= PrecomputedCatmullRomBasis::N);207assert(ofs <= size);208return madd(vfloat<M>::loadu(&catmullrom_basis0.d0[size][ofs]), Vec4vf<M>(v0),209madd(vfloat<M>::loadu(&catmullrom_basis0.d1[size][ofs]), Vec4vf<M>(v1),210madd(vfloat<M>::loadu(&catmullrom_basis0.d2[size][ofs]), Vec4vf<M>(v2),211vfloat<M>::loadu(&catmullrom_basis0.d3[size][ofs]) * Vec4vf<M>(v3))));212}213214template<int M>215__forceinline Vec4vf<M> derivative1(const int ofs, const int size) const216{217assert(size <= PrecomputedCatmullRomBasis::N);218assert(ofs <= size);219return madd(vfloat<M>::loadu(&catmullrom_basis1.d0[size][ofs]), Vec4vf<M>(v0),220madd(vfloat<M>::loadu(&catmullrom_basis1.d1[size][ofs]), Vec4vf<M>(v1),221madd(vfloat<M>::loadu(&catmullrom_basis1.d2[size][ofs]), Vec4vf<M>(v2),222vfloat<M>::loadu(&catmullrom_basis1.d3[size][ofs]) * Vec4vf<M>(v3))));223}224225/* calculates bounds of catmull-rom curve geometry */226__forceinline BBox3fa accurateRoundBounds() const227{228const int N = 7;229const float scale = 1.0f/(3.0f*(N-1));230Vec4vfx pl(pos_inf), pu(neg_inf);231for (int i=0; i<=N; i+=VSIZEX)232{233vintx vi = vintx(i)+vintx(step);234vboolx valid = vi <= vintx(N);235const Vec4vfx p = eval0<VSIZEX>(i,N);236const Vec4vfx dp = derivative0<VSIZEX>(i,N);237const Vec4vfx pm = p-Vec4vfx(scale)*select(vi!=vintx(0),dp,Vec4vfx(zero));238const Vec4vfx pp = p+Vec4vfx(scale)*select(vi!=vintx(N),dp,Vec4vfx(zero));239pl = select(valid,min(pl,p,pm,pp),pl); // FIXME: use masked min240pu = select(valid,max(pu,p,pm,pp),pu); // FIXME: use masked min241}242const Vec3fa lower(reduce_min(pl.x),reduce_min(pl.y),reduce_min(pl.z));243const Vec3fa upper(reduce_max(pu.x),reduce_max(pu.y),reduce_max(pu.z));244const float r_min = reduce_min(pl.w);245const float r_max = reduce_max(pu.w);246const Vec3fa upper_r = Vec3fa(max(abs(r_min),abs(r_max)));247return enlarge(BBox3fa(lower,upper),upper_r);248}249250/* calculates bounds when tessellated into N line segments */251__forceinline BBox3fa accurateFlatBounds(int N) const252{253if (likely(N == 4))254{255const Vec4vf4 pi = eval0<4>(0,4);256const Vec3fa lower(reduce_min(pi.x),reduce_min(pi.y),reduce_min(pi.z));257const Vec3fa upper(reduce_max(pi.x),reduce_max(pi.y),reduce_max(pi.z));258const Vec3fa upper_r = Vec3fa(reduce_max(abs(pi.w)));259const Vec3ff pe = end();260return enlarge(BBox3fa(min(lower,pe),max(upper,pe)),max(upper_r,Vec3fa(abs(pe.w))));261}262else263{264Vec3vfx pl(pos_inf), pu(neg_inf); vfloatx ru(0.0f);265for (int i=0; i<=N; i+=VSIZEX)266{267vboolx valid = vintx(i)+vintx(step) <= vintx(N);268const Vec4vfx pi = eval0<VSIZEX>(i,N);269270pl.x = select(valid,min(pl.x,pi.x),pl.x); // FIXME: use masked min271pl.y = select(valid,min(pl.y,pi.y),pl.y);272pl.z = select(valid,min(pl.z,pi.z),pl.z);273274pu.x = select(valid,max(pu.x,pi.x),pu.x); // FIXME: use masked min275pu.y = select(valid,max(pu.y,pi.y),pu.y);276pu.z = select(valid,max(pu.z,pi.z),pu.z);277278ru = select(valid,max(ru,abs(pi.w)),ru);279}280const Vec3fa lower(reduce_min(pl.x),reduce_min(pl.y),reduce_min(pl.z));281const Vec3fa upper(reduce_max(pu.x),reduce_max(pu.y),reduce_max(pu.z));282const Vec3fa upper_r(reduce_max(ru));283return enlarge(BBox3fa(lower,upper),upper_r);284}285}286287friend __forceinline embree_ostream operator<<(embree_ostream cout, const CatmullRomCurveT& curve) {288return cout << "CatmullRomCurve { v0 = " << curve.v0 << ", v1 = " << curve.v1 << ", v2 = " << curve.v2 << ", v3 = " << curve.v3 << " }";289}290};291292template<typename Vertex>293__forceinline void convert(const CatmullRomCurveT<Vertex>& icurve, BezierCurveT<Vertex>& ocurve)294{295const Vertex v0 = icurve.v1;296const Vertex v1 = icurve.v1+(icurve.v2-icurve.v0)*(1.0f/6.0f);297const Vertex v2 = icurve.v2+(icurve.v1-icurve.v3)*(1.0f/6.0f);298const Vertex v3 = icurve.v2;299ocurve = BezierCurveT<Vertex>(v0,v1,v2,v3);300}301302template<typename CurveGeometry>303__forceinline CatmullRomCurveT<Vec3ff> enlargeRadiusToMinWidth(const RayQueryContext* context, const CurveGeometry* geom, const Vec3fa& ray_org, const CatmullRomCurveT<Vec3ff>& curve)304{305return CatmullRomCurveT<Vec3ff>(enlargeRadiusToMinWidth(context,geom,ray_org,curve.v0),306enlargeRadiusToMinWidth(context,geom,ray_org,curve.v1),307enlargeRadiusToMinWidth(context,geom,ray_org,curve.v2),308enlargeRadiusToMinWidth(context,geom,ray_org,curve.v3));309}310311typedef CatmullRomCurveT<Vec3fa> CatmullRomCurve3fa;312}313314315316