random-mirrors/librw
1
0
Fork 0
mirror of https://github.com/aap/librw.git synced 2024-11-25 21:25:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
1275b70e9d
librw/tools/playground/splines.cpp
aap 01f021fdbf bla
2020-07-22 14:09:07 +02:00

521 lines
11 KiB
C++
Raw Blame History

#include <rw.h>
#include <skeleton.h>
#include <vector>
using namespace rw;
using namespace RWDEVICE;
static Im3DVertex im3dVerts[1024];
static int numImVerts;
static rw::PrimitiveType imPrim;
static Im3DVertex imVert;
void
BeginIm3D(rw::PrimitiveType prim)
{
numImVerts = 0;
imPrim = prim;
}
void
EndIm3D(void)
{
rw::im3d::Transform(im3dVerts, numImVerts, nil, rw::im3d::EVERYTHING);
rw::im3d::RenderPrimitive(imPrim);
rw::im3d::End();
}
void
AddVertex(const rw::V3d &vert)
{
if(numImVerts >= 1024){
EndIm3D();
switch(imPrim){
case PRIMTYPEPOLYLINE:
im3dVerts[0] = im3dVerts[numImVerts-1];
numImVerts = 1;
break;
case PRIMTYPETRISTRIP:
// TODO: winding?
im3dVerts[0] = im3dVerts[numImVerts-2];
im3dVerts[1] = im3dVerts[numImVerts-1];
numImVerts = 2;
break;
case PRIMTYPETRIFAN:
im3dVerts[1] = im3dVerts[numImVerts-1];
numImVerts = 2;
break;
default:
numImVerts = 0;
}
}
imVert.setX(vert.x);
imVert.setY(vert.y);
imVert.setZ(vert.z);
im3dVerts[numImVerts++] = imVert;
}
float epsilon = 0.000001;
class RBCurve
{
public:
int degree;
std::vector<V3d> verts;
std::vector<float> weights; // for rational
std::vector<float> knots;
V3d eval(float u);
void drawHull(void);
void drawSpline(void);
};
class RBSurf
{
public:
int degreeU, degreeV;
int numU, numV;
std::vector<V3d> verts;
std::vector<float> weights;
std::vector<float> knotsU, knotsV;
void update(void);
V3d eval(float u, float v);
void drawHull(void);
void drawIsoparms(void);
void drawShaded(void);
};
float div0(float a, float b) { return b == 0.0f ? a : a/b; }
// naive algorithm
float
evalBasis(int i, int d, float u, float knots[])
{
if(d == 0){
if(knots[i] <= u && u < knots[i+1])
return 1.0f;
return 0.0f;
}
float b0 = evalBasis(i, d-1, u, knots);
float b1 = evalBasis(i+1, d-1, u, knots);
return b0*div0(u-knots[i], knots[i+d] - knots[i]) + b1*div0(knots[i+d+1]-u, knots[i+d+1] - knots[i+1]);
}
float
evalBasisFast(int i, int d, float u, float knots[])
{
int r, j;
float tmp[10];
// degree 0 values
for(j = 0; j < d+1; j++)
tmp[j] = knots[i+j] <= u && u < knots[i+j+1] ? 1.0f : 0.0f;
// build up from degree zero
for(r = d, d = 1; r > 0; r--, d++){
for(j = 0; j < r; j++){
float t1 = div0(u-knots[i+j], knots[i+j + d] - knots[i+j]);
float t2 = div0(knots[i+j + d+1]-u, knots[i+j + d+1] - knots[i+j + 1]);
tmp[j] = tmp[j]*t1 + tmp[j+1]*t2;
}
}
return tmp[0];
}
V3d
RBCurve::eval(float u)
{
int i;
V3d vert = { 0.0f, 0.0f, 0.0f };
float w = 0.0f;
// Find knots we're interested in
for(i = 0; i < knots.size(); i++)
if(knots[i] <= u && u < knots[i+1])
break;
int startI = i-degree;
int endI = i+1;
for(i = startI; i < endI; i++){
float r = evalBasisFast(i, degree, u, &knots[0]);
w += weights[i]*r;
vert = add(vert, scale(verts[i], weights[i]*r));
}
return scale(vert, div0(1.0f,w));
}
void
RBCurve::drawHull(void)
{
int i;
rw::SetRenderState(rw::TEXTURERASTER, nil);
rw::SetRenderState(rw::FOGENABLE, 0);
BeginIm3D(rw::PRIMTYPEPOLYLINE);
imVert.setU(0.0f);
imVert.setV(0.0f);
imVert.setColor(138, 72, 51, 255);
// imVert.setColor(228, 172, 121, 255);
for(i = 0; i < verts.size(); i++)
AddVertex(verts[i]);
EndIm3D();
}
void
RBCurve::drawSpline(void)
{
int i;
float u, endu;
V3d vert;
rw::SetRenderState(rw::TEXTURERASTER, nil);
rw::SetRenderState(rw::FOGENABLE, 0);
BeginIm3D(rw::PRIMTYPEPOLYLINE);
imVert.setU(0.0f);
imVert.setV(0.0f);
imVert.setColor(0, 4, 96, 255);
u = knots[0];
endu = knots[knots.size()-1] - epsilon;
float uinc = (endu-knots[0])/40.0f;
for(;; u += uinc){
if(u > endu)
u = endu;
AddVertex(eval(u));
if(u >= endu)
break;
}
EndIm3D();
}
void
RBSurf::update(void)
{
numU = knotsU.size() - degreeU - 1;
numV = knotsV.size() - degreeV - 1;
}
V3d
RBSurf::eval(float u, float v)
{
int i, j;
V3d vert = { 0.0f, 0.0f, 0.0f };
float w = 0.0f;
float basisU[10], basisV[10];
// Find knots we're interested in
int k;
for(k = 0; k < knotsU.size(); k++)
if(knotsU[k] <= u && u < knotsU[k+1])
break;
int startI = k-degreeU;
int endI = k+1;
for(k = 0; k < knotsV.size(); k++)
if(knotsV[k] <= v && v < knotsV[k+1])
break;
int startJ = k-degreeV;
int endJ = k+1;
for(i = startI; i < endI; i++) basisU[i-startI] = evalBasisFast(i, degreeU, u, &knotsU[0]);
for(j = startJ; j < endJ; j++) basisV[j-startJ] = evalBasisFast(j, degreeV, v, &knotsV[0]);
for(j = startJ; j < endJ; j++)
for(i = startI; i < endI; i++){
float r = basisV[j-startJ]*basisU[i-startI];
w += weights[j*numU + i]*r;
vert = add(vert, scale(verts[j*numU + i], weights[j*numU + i]*r));
}
return scale(vert, div0(1.0f,w));
}
void
RBSurf::drawHull(void)
{
rw::SetRenderState(rw::TEXTURERASTER, nil);
rw::SetRenderState(rw::FOGENABLE, 0);
imVert.setU(0.0f);
imVert.setV(0.0f);
imVert.setColor(138, 72, 51, 255);
// imVert.setColor(228, 172, 121, 255);
int iu, iv;
for(iv = 0; iv < numV; iv++){
BeginIm3D(rw::PRIMTYPEPOLYLINE);
for(iu = 0; iu < numU; iu++)
AddVertex(verts[iu + iv*numU]);
EndIm3D();
}
for(iu = 0; iu < numU; iu++){
BeginIm3D(rw::PRIMTYPEPOLYLINE);
for(iv = 0; iv < numV; iv++)
AddVertex(verts[iu + iv*numU]);
EndIm3D();
}
}
void
RBSurf::drawIsoparms(void)
{
V3d vert;
int iu, iv;
float u, v;
rw::SetRenderState(rw::TEXTURERASTER, nil);
rw::SetRenderState(rw::FOGENABLE, 0);
imVert.setU(0.0f);
imVert.setV(0.0f);
imVert.setColor(0, 4, 96, 255);
float endu = knotsU[knotsU.size()-1] - epsilon;
float endv = knotsU[knotsV.size()-1] - epsilon;
float uinc = (endu-knotsU[0])/40.0f;
float vinc = (endv-knotsV[0])/40.0f;
v = -100000.0f;
for(iv = 0; iv < knotsV.size(); iv++){
if(knotsV[iv] <= v) continue;
v = knotsV[iv];
if(v > endv) v = endv;
BeginIm3D(rw::PRIMTYPEPOLYLINE);
for(u = knotsU[0];; u += uinc){
if(u > endu)
u = endu;
AddVertex(eval(u, v));
if(u >= endu)
break;
}
EndIm3D();
}
u = -100000.0f;
for(iu = 0; iu < knotsU.size(); iu++){
if(knotsU[iu] <= u) continue;
u = knotsU[iu];
if(u > endu) u = endu;
BeginIm3D(rw::PRIMTYPEPOLYLINE);
for(v = knotsV[0];; v += vinc){
if(v > endv)
v = endv;
AddVertex(eval(u, v));
if(v >= endv)
break;
}
EndIm3D();
}
}
void
RBSurf::drawShaded(void)
{
V3d vert;
int iu, iv;
float u, v;
rw::SetRenderState(rw::TEXTURERASTER, nil);
rw::SetRenderState(rw::FOGENABLE, 0);
imVert.setU(0.0f);
imVert.setV(0.0f);
imVert.setColor(0, 128, 240, 255);
float endu = knotsU[knotsU.size()-1] - epsilon;
float endv = knotsU[knotsV.size()-1] - epsilon;
float uinc = (endu-knotsU[0])/40.0f;
float vinc = (endv-knotsV[0])/40.0f;
float vnext;
for(v = knotsV[0];; v = vnext){
if(v > endv)
v = endv;
vnext = v + vinc;
BeginIm3D(rw::PRIMTYPETRISTRIP);
for(u = knotsU[0];; u += uinc){
if(u > endu)
u = endu;
AddVertex(eval(u, v));
AddVertex(eval(u, vnext));
if(u >= endu)
break;
}
EndIm3D();
if(vnext >= endv)
break;
}
}
RBCurve testspline1, testspline2;
RBSurf testsurf;
void
initsplines(void)
{
V3d vert;
testspline1.degree = 3;
testspline1.verts.clear();
testspline1.weights.clear();
vert.set(-30.63383, 22.65459, 0);
vert = scale(vert, 1.0f/20.0f);
testspline1.verts.push_back(vert);
testspline1.weights.push_back(1.0f);
vert.set(13.50783, 33.01786, 15.06403);
vert = scale(vert, 1.0f/20.0f);
testspline1.verts.push_back(vert);
testspline1.weights.push_back(1.0f);
vert.set(34.252, -10.36327, 15.06403);
vert = scale(vert, 1.0f/20.0f);
testspline1.verts.push_back(vert);
testspline1.weights.push_back(1.0f);
vert.set(-7.959972, -1.205032, 0);
vert = scale(vert, 1.0f/20.0f);
testspline1.verts.push_back(vert);
testspline1.weights.push_back(1.0f);
vert.set(6.995127, -41.32158, -18.19684);
vert = scale(vert, 1.0f/20.0f);
testspline1.verts.push_back(vert);
testspline1.weights.push_back(1.0f);
testspline1.knots.clear();
testspline1.knots.push_back(0);
testspline1.knots.push_back(0);
testspline1.knots.push_back(0);
testspline1.knots.push_back(0);
testspline1.knots.push_back(1);
testspline1.knots.push_back(2);
testspline1.knots.push_back(2);
testspline1.knots.push_back(2);
testspline1.knots.push_back(2);
testspline2.degree = 2;
testspline2.verts.clear();
#define V(x, y, z) \
vert.set(x, y, z); \
testspline2.verts.push_back(scale(vert, 1.0f/20.0f)); \
testspline2.weights.push_back(1.0f);
V(-61.9913, 9.158239, 0);
V(-32.32231, 27.23371, 0);
V(25.80961, -4.820126, 0);
testspline2.knots.clear();
testspline2.knots.push_back(0);
testspline2.knots.push_back(0);
testspline2.knots.push_back(0);
testspline2.knots.push_back(1);
testspline2.knots.push_back(1);
testspline2.knots.push_back(1);
/*
testspline2 = testspline1;
// testspline2.knots.clear();
// testspline2.knots.push_back(0);
// testspline2.knots.push_back(0);
// testspline2.knots.push_back(0);
// testspline2.knots.push_back(0);
// testspline2.knots.push_back(3);
// testspline2.knots.push_back(4);
// testspline2.knots.push_back(4);
// testspline2.knots.push_back(4);
// testspline2.knots.push_back(4);
testspline1.weights[2] = 5.0f;
*/
#define V(x, y, z) \
vert.set(x, y, z); \
testsurf.verts.push_back(scale(vert, 1.0f/20.0f)); \
testsurf.weights.push_back(1.0f);
testsurf.degreeU = 3;
testsurf.degreeV = 3;
testsurf.verts.clear();
testsurf.weights.clear();
V(-69.22764, -0, 12.77366);
V(-48.72468, 0, 29.16251);
V(-24.84476, 0, 39.52605);
V(22.43265, -0, 45.79238);
V(36.4229, 0, 28.9215);
V(61.02645, 0, 6.74835);
V(86.35364, -0, 20.96809);
V(-69.22764, 9.286676, 12.77366);
V(-48.72468, 9.286676, 29.16251);
V(-24.84476, 9.286676, 39.52605);
V(22.43265, 9.286676, 45.79238);
V(36.4229, 9.286676, 28.9215);
V(61.02645, 9.286676, 6.74835);
V(86.35364, 9.286676, 20.96809);
V(-68.13416, 23.51821, 6.114491);
V(-48.19925, 23.51821, 22.27994);
V(-26.91943, 23.51821, 33.20763);
V(18.69658, 23.51821, 39.0488);
V(27.88844, 23.51821, 30.80604);
V(57.49908, 23.51821, -0.6488895);
V(86.35364, 23.51821, 14.21974);
V(-67.00163, 52.46369, -0.7825046);
V(-47.65504, 52.46369, 15.15157);
V(-29.06818, 52.46369, 26.66356);
V(14.82709, 52.46369, 32.06438);
V(19.04919, 52.46369, 32.75788);
V(53.84574, 52.46369, -8.310311);
V(86.35364, 52.46369, 7.230374);
V(-67.86079, 70.07219, 4.449699);
V(-48.06789, 70.07219, 20.5593);
V(-27.43809, 70.07219, 31.62803);
V(17.76257, 70.07219, 37.36291);
V(25.75483, 70.07219, 31.27717);
V(56.61724, 70.07219, -2.498198);
V(86.35364, 70.07219, 12.53265);
testsurf.knotsU.clear();
testsurf.knotsU.push_back(0);
testsurf.knotsU.push_back(0);
testsurf.knotsU.push_back(0);
testsurf.knotsU.push_back(0);
testsurf.knotsU.push_back(0.25);
testsurf.knotsU.push_back(0.5);
testsurf.knotsU.push_back(0.75);
testsurf.knotsU.push_back(1);
testsurf.knotsU.push_back(1);
testsurf.knotsU.push_back(1);
testsurf.knotsU.push_back(1);
testsurf.knotsV.clear();
testsurf.knotsV.push_back(0);
testsurf.knotsV.push_back(0);
testsurf.knotsV.push_back(0);
testsurf.knotsV.push_back(0);
testsurf.knotsV.push_back(0.5);
testsurf.knotsV.push_back(1);
testsurf.knotsV.push_back(1);
testsurf.knotsV.push_back(1);
testsurf.knotsV.push_back(1);
testsurf.update();
}
void
rendersplines(void)
{
testspline1.drawHull();
testspline1.drawSpline();
// testspline2.drawHull();
// testspline2.drawSpline();
testsurf.drawHull();
testsurf.drawShaded();
testsurf.drawIsoparms();
}
Reference in New Issue View Git Blame Copy Permalink