vcglib/apps/test/segmentation3d/segmentator.h

1049 lines
24 KiB
C
Raw Normal View History

2005-02-01 19:07:14 +01:00
#ifndef __SEGMENTATOR
#define __SEGMENTATOR
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
/////need vertex-face topology in case of extended marching cubes
//#ifdef _EXTENDED_MARCH
// #include <vcg/simplex/vertex/with/afvn.h>
// #include <vcg/simplex/face/with/afavfnfmrt.h>
//#else
#include <vcg/simplex/vertex/with/vn.h>
#include <vcg/simplex/face/with/affnfmrt.h>
//#endif
2004-12-17 12:17:36 +01:00
#include <sim/particle/with/basic_physics.h>
#include <sim/methods/mass_spring/triangle.h>
#include <vcg/complex/trimesh/base.h>
#include <vcg/complex/trimesh/allocate.h>
#include <vcg/complex/trimesh/update/topology.h>
#include <vcg/complex/trimesh/update/normal.h>
2005-02-01 19:07:14 +01:00
#include <vcg/complex/trimesh/update/bounding.h>
#include <vcg/complex/trimesh/update/edges.h>
2004-12-17 12:17:36 +01:00
#include <vcg/complex/trimesh/refine.h>
2005-02-01 19:07:14 +01:00
#include <vcg/complex/trimesh/create/platonic.h>
2004-12-17 12:17:36 +01:00
#include <volume_dataset.h>
2005-02-01 19:07:14 +01:00
#include <vcg/complex/trimesh/create/resampler.h>
2004-12-17 12:17:36 +01:00
#include <vcg/space/point3.h>
#include <vcg/space/box3.h>
#include <sim/pde_integrator.h>
#include <partial_container.h>
#include <vector>
#include <time.h>
#include <math.h>
#include <collision_detection.h>
#include <vcg/complex/trimesh/smooth.h>
2005-04-11 11:26:22 +02:00
#include <vcg/simplex/face/topology.h>
#include <vcg/complex/trimesh/update/topology.h>
2005-02-01 19:07:14 +01:00
2004-12-17 12:17:36 +01:00
class Segmentator{
public:
struct DummyEdge;
struct DummyTetra;
struct MyFace;
2005-02-01 19:07:14 +01:00
#ifdef _EXTENDED_MARCH
struct MyVertex: public ParticleBasic<vcg::VertexAFVNf<DummyEdge,MyFace,DummyTetra> >
#else
struct MyVertex: public ParticleBasic<vcg::VertexVNf<DummyEdge,MyFace,DummyTetra> >
#endif
2004-12-17 12:17:36 +01:00
{
public:
bool blocked;//optimize after with vertex flags
bool stopped;
MyVertex()
{
blocked=false;
stopped=false;
Acc()=Point3f(0,0,0);
Vel()=Point3f(0,0,0);
2005-02-01 19:07:14 +01:00
ClearFlags();
2005-04-11 11:26:22 +02:00
//__super::
2004-12-17 12:17:36 +01:00
//neeed call of the super class
}
void UpdateAcceleration()
{
if ((!blocked)&&(!stopped))
{
Acc()=(IntForce()+ExtForce())/Mass();
}
else
{
Acc()=Point3f(0,0,0);
Vel()=Point3f(0,0,0);
}
}
void Reset()
{
IntForce()=Point3f(0.f,0.f,0.f);
}
2005-04-11 11:26:22 +02:00
void SetRestPos()
{
RPos()=P();
}
2004-12-17 12:17:36 +01:00
};
2005-04-11 11:26:22 +02:00
///this class implements the deformable triangle in a mass spring system
struct MyFace : public TriangleMassSpring< vcg::FaceAFFNFMRT<MyVertex,DummyEdge,MyFace> >
2004-12-17 12:17:36 +01:00
{
2005-04-11 11:26:22 +02:00
public:
bool intersected;
float kdihedral;
ScalarType AreaRep;
int _Mark;
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
MyFace()
{
intersected=false;
ClearFlags();
}
2005-04-11 11:26:22 +02:00
void Init ( double k, double mass,float k_dihedral )
{
__super::Init(k,mass);
kdihedral=k_dihedral;
AreaRep=((V(1)->RPos() - V(0)->RPos()) ^ (V(2)->RPos() - V(0)->RPos())).Norm();
2005-04-11 11:26:22 +02:00
SetS();
}
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
bool IsActive()
{
return((!(((V(0)->blocked)||(V(0)->stopped))&&
((V(1)->blocked)||(V(1)->stopped))&&
((V(2)->blocked)||(V(2)->stopped))))&&(!intersected));
}
2005-04-20 17:20:13 +02:00
bool HaveVertexActive()
{
return((V(0)->blocked==false)||(V(1)->blocked==false)||(V(2)->blocked==false));
}
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
bool IsBlocked()
2005-02-08 18:47:50 +01:00
{
2005-04-11 11:26:22 +02:00
return((V(0)->blocked)&&(V(1)->blocked)&&(V(2)->blocked));
}
2005-04-20 17:20:13 +02:00
2005-04-11 11:26:22 +02:00
double DiedralAngle(int edge)
2004-12-17 12:17:36 +01:00
{
2005-04-11 11:26:22 +02:00
MyFace *fopp=FFp(edge);
CoordType norm1=NormalizedNormal();
CoordType norm2=fopp->NormalizedNormal();
2005-04-20 17:20:13 +02:00
return (norm1*norm2);
2005-04-11 11:26:22 +02:00
}
2005-02-08 18:47:50 +01:00
int &Mark()
{return (_Mark);}
2005-04-11 11:26:22 +02:00
///return the bounding box of the simplex
vcg::Box3<float> BBox()
{
vcg::Box3<float> bb;
GetBBox(bb);
return (bb);
}
2004-12-17 12:17:36 +01:00
2005-04-20 17:20:13 +02:00
ScalarType ForceValue(ScalarType l0,ScalarType l1)
{
2005-04-22 11:23:09 +02:00
ScalarType diff=(l0-l1);
if (diff>0)//compression
2005-04-26 11:20:15 +02:00
return ((diff/(l1))*_k)+(diff *_k);
2005-04-22 11:23:09 +02:00
else
2005-04-26 11:20:15 +02:00
return (diff *_k);
2005-04-20 17:20:13 +02:00
}
2005-04-11 11:26:22 +02:00
///update of the internal forces using the dihedral angle
bool Update ( void )
{
2005-04-20 17:20:13 +02:00
if (!IsD()&&(!intersected))//if this face is not deleted
2005-04-11 11:26:22 +02:00
{
for (int i=0;i<3;i++)
2004-12-17 12:17:36 +01:00
{
2005-04-11 11:26:22 +02:00
if (!IsBorder(i))
{
MyFace *fopp=FFp(i);
MyFace *myAddr=this;
assert(!fopp->IsD());
assert(fopp!=this);
2005-04-20 17:20:13 +02:00
if ((fopp<myAddr)&&(!fopp->intersected))//test do not duplicate updates per edge
2005-04-11 11:26:22 +02:00
{
//normal and area based diadedral angle calcolus
CoordType DirEdge=(V(i)->P()-V((i+1)%3)->P()).Normalize();
fopp=FFp(i);
CoordType Ver=(NormalizedNormal()^fopp->NormalizedNormal()).Normalize();
ScalarType diaedral=DiedralAngle(i);
2005-04-20 17:20:13 +02:00
ScalarType Force=(((-diaedral)+1.f)*kdihedral);
CoordType VectF=NormalizedNormal()*(Force);
2005-04-11 11:26:22 +02:00
if ((Ver*DirEdge)<=0)///convex
2005-04-20 17:20:13 +02:00
{
V((i+2)%3)->IntForce()+=VectF;
fopp->V((FFi(i)+2)%3)->IntForce()+=VectF;///opposite vertex
V(i)->IntForce()-=VectF;
V((i+1)%3)->IntForce()-=VectF;
2005-04-11 11:26:22 +02:00
}
else ///non-convex
{
2005-04-20 17:20:13 +02:00
V((i+2)%3)->IntForce()-=VectF;
fopp->V((FFi(i)+2)%3)->IntForce()-=VectF;///opposite vertex
V(i)->IntForce()+=VectF;
V((i+1)%3)->IntForce()+=VectF;
2005-04-11 11:26:22 +02:00
}
2005-04-20 17:20:13 +02:00
2005-04-11 11:26:22 +02:00
}
}
2004-12-17 12:17:36 +01:00
}
2005-04-11 11:26:22 +02:00
2005-04-22 11:23:09 +02:00
////area changing constrain penalize area goes to zero
//CoordType m0=(V(0)->P()+V(1)->P())/2.f;
//CoordType m1=(V(1)->P()+V(2)->P())/2.f;
//CoordType m2=(V(2)->P()+V(0)->P())/2.f;
//m0=(V(2)->P()-m0).Normalize();//directions
//m1=(V(0)->P()-m1).Normalize();
//m2=(V(1)->P()-m2).Normalize();
//ScalarType FArea=((ScalarType)1/AreaRep-Area())*_k;
//V(0)->IntForce()+=m0*FArea/3.f;
//V(1)->IntForce()+=m1*FArea/3.f;
//V(2)->IntForce()+=m2*FArea/3.f;
////end area constrain
2005-04-11 11:26:22 +02:00
return(__super::Update());
///new
//double stretch;
//CoordType direction;
/////for each vertex
//for (int i=0;i<3;i++)
//{
// //spring on the face
// direction=(V(i)->RPos()-V(i)->P());
// stretch=direction.Norm();
// direction.Normalize();
// V(i)-> IntForce()+=direction*(stretch * _k)/2.f;
//}
2004-12-17 12:17:36 +01:00
}
2005-04-11 11:26:22 +02:00
return true;
///new
2004-12-17 12:17:36 +01:00
}
};
struct MyTriMesh: public vcg::tri::TriMesh<std::vector<MyVertex>,std::vector<MyFace> >{};
typedef Partial_Container<std::vector<MyVertex*>,MyVertex> Part_VertexContainer;
typedef Partial_Container<std::vector<MyFace*>,MyFace> Part_FaceContainer;
typedef PDEIntegrator<Part_FaceContainer,Part_VertexContainer,float> myIntegrator;
typedef Collision_Detector<std::vector<MyFace> > Collision;
2005-02-01 19:07:14 +01:00
////typedef Walker<MyTriMesh,MyTriMesh> MyWalk;
//
//#ifdef _EXTENDED_MARCH
// typedef vcg::tri::ExtendedMarchingCubes<MyTriMesh, MyWalk> MarchingCubes;
//#else
// typedef vcg::tri::MarchingCubes<MyTriMesh, MyWalk> MarchingCubes;
//#endif
2004-12-17 12:17:36 +01:00
public:
Point3f scale;
//VolumetricDataset<int> d;
2005-02-01 19:07:14 +01:00
/*MyTriMesh m;
MyTriMesh new_m;*/
MyTriMesh *m;
MyTriMesh *new_m;
2004-12-17 12:17:36 +01:00
Part_FaceContainer P_Faces;
Part_VertexContainer P_Vertex;
Part_VertexContainer V_Stopped;
myIntegrator *TrINT;
MyTriMesh::CoordType InitialBarycenter;
float mass;
float k_elanst;
float k_dihedral;
float edge_size;
int tolerance;
int gray_init;
float time_stamp;
float edge_precision;
2005-04-20 17:20:13 +02:00
float edge_init;
2004-12-17 12:17:36 +01:00
bool end_loop;
bool refined;
clock_t interval_reinit;
clock_t interval_selfcollision;
Collision *CollDet;
//Volume_Dataset_Optimized<short> V;
Volume_Dataset <short> V;
vcg::Box3<float> bbox;
char *inDir;
char *outDir;
//attention static members
/*int BlockFlag;
int StoppedFlag;*/
Segmentator()
{
2005-02-01 19:07:14 +01:00
m=new MyTriMesh();
CollDet=new Collision(m->face);
2004-12-17 12:17:36 +01:00
}
~Segmentator()
{
}
private:
/////return integer coordinete in volumetric dataset
//Point3i MapToDataset(MyTriMesh::CoordType p)
//{
// MyTriMesh::ScalarType x=((MyTriMesh::ScalarType)p.V(0));
// MyTriMesh::ScalarType y=((MyTriMesh::ScalarType)p.V(1));
// MyTriMesh::ScalarType z=((MyTriMesh::ScalarType)p.V(2));
// return Point3i((int)p.V(0),(int)p.V(1),(int)p.V(2));
//}
//
///map to space coordinate from dataset coordinates
MyTriMesh::CoordType MapToSpace(Point3i p)
{
MyTriMesh::ScalarType x=((MyTriMesh::ScalarType)p.V(0));
MyTriMesh::ScalarType y=((MyTriMesh::ScalarType)p.V(1));
MyTriMesh::ScalarType z=((MyTriMesh::ScalarType)p.V(2));
return (MyTriMesh::CoordType(x,y,z));
}
2004-12-21 18:38:42 +01:00
#ifdef _TORUS
///torus version
2005-04-20 17:20:13 +02:00
/*float getColor(MyTriMesh::CoordType p)
2004-12-21 18:38:42 +01:00
{
2005-04-20 17:20:13 +02:00
static float ray=25.f;
static float diameter=4.f;
float dist=p.Norm();
2004-12-21 18:38:42 +01:00
float difference=abs(ray-dist);
2005-04-20 17:20:13 +02:00
float z=fabs(p.Z());
if((difference<diameter)&&(z<diameter))
return (100.f);
else
return (0.f);
}*/
bool InTorus(MyTriMesh::CoordType p)
{
static float ray=25.f;
static float diameter=4.f;
double fatt=(p.X()*p.X()+p.Y()*p.Y());
double fatt1=(ray-math::Sqrt(fatt));
double fatt2=fatt1*fatt1+p.Z()*p.Z();
double conf=diameter*diameter;
///now swap
return (fatt2<=conf);
}
2005-04-22 11:23:09 +02:00
2005-04-20 17:20:13 +02:00
float getColor(MyTriMesh::CoordType p)
{
MyTriMesh::CoordType p1=MyTriMesh::CoordType(p.X(),p.Z(),p.Y());
2005-04-22 11:23:09 +02:00
MyTriMesh::CoordType p2=MyTriMesh::CoordType(p.Z(),p.Y(),p.X());
if (InTorus(p)||InTorus(p1)||InTorus(p2))
2005-04-20 17:20:13 +02:00
return (100.f);
2004-12-21 18:38:42 +01:00
else
2005-04-20 17:20:13 +02:00
return (0.f);
2004-12-21 18:38:42 +01:00
}
2005-04-20 17:20:13 +02:00
2005-04-22 11:23:09 +02:00
2004-12-21 18:38:42 +01:00
#else
2004-12-17 12:17:36 +01:00
///return integer coordinete in volumetric dataset
float getColor(MyTriMesh::CoordType p)
{
2004-12-20 18:56:01 +01:00
float lx=(p.V(0)-(int)p.V(0))*scale.V(0);//da rivedere bene per lo scale
float ly=(p.V(1)-(int)p.V(1))*scale.V(1);//da rivedere bene per lo scale
float lz=(p.V(2)-(int)p.V(2))*scale.V(2);//da rivedere bene per lo scale
p=Scale(p);
2004-12-17 12:17:36 +01:00
Point3i base=Point3i((int)p.V(0),(int)p.V(1),(int)p.V(2));
float v[8];
Point3i px;
for (int i=0;i<8;i++)
{
px=base+Point3i((i%2),(i/4),((i/2)%2));
v[i]=(float)V.getAt(px);
}
float color=lx*v[1]+v[0]+lz*v[0]*lx-v[0]*lx-ly*v[0]+ly*v[2]-lz*v[0]+ly*v[0]*lx-ly*lx*v[1]-ly*v[2]*lx
-lz*ly*v[0]*lx+lz*ly*lx*v[1]+lz*ly*v[2]*lx-lz*ly*lx*v[3]+lz*ly*lx*v[4]-lz*ly*lx*v[5]-lz*ly*
v[6]*lx+lz*ly*lx*v[7]+ly*lx*v[3]-lz*lx*v[1]+lz*ly*v[0]-lz*ly*v[2]-lz*lx*v[4]+lz*lx*v[5]-lz*ly*
v[4]+lz*ly*v[6]+lz*v[4];
return color;
}
2004-12-21 18:38:42 +01:00
#endif
2004-12-17 12:17:36 +01:00
///maximixe the gradient of the movement
MyTriMesh::CoordType Gradient(MyTriMesh::CoordType p,float h=0.01f)
{
float value=getColor(p);
MyTriMesh::CoordType h0=MyTriMesh::CoordType(h,0,0);
MyTriMesh::CoordType h1=MyTriMesh::CoordType(0,h,0);
MyTriMesh::CoordType h2=MyTriMesh::CoordType(0,0,h);
float dx=(getColor(p+h0)-value)/h;
/*dx=v[1]+lz*v[0]-v[0]*lx+ly*v[0]-ly*v[1]-ly*v[2]
-lz*ly*v[0]+lz*ly*v[1]+lz*ly*v[2]-lz*ly*v[3]+lz*ly*v[4]-lz*ly*v[5]-lz*ly*
v[6]+lz*ly*v[7]+ly*v[3]-lz*v[1]-lz*v[4]+lz*v[5]-lz*ly*v[4]+lz*ly*v[6]+lz*v[4];
dy=-v[0]+v[2]+v[0]*lx-lx*v[1]-v[2]*lx
-lz*v[0]*lx+lz*lx*v[1]+lz*v[2]*lx-lz*lx*v[3]+lz*lx*v[4]-lz*lx*v[5]-lz*v[6]*lx+lz*ly*lx*v[7]+ly*lx*v[3]-lz*lx*v[1]+lz*ly*v[0]-lz*ly*v[2]-lz*lx*v[4]+lz*lx*v[5]-lz*ly*
v[4]+lz*ly*v[6]+lz*v[4];*/
float dy=(getColor(p+h1)-value)/h;
float dz=(getColor(p+h2)-value)/h;
MyTriMesh::CoordType ret=MyTriMesh::CoordType(dx,dy,dz);
return (ret);
}
2004-12-20 18:56:01 +01:00
///scale the coordinates of a point
MyTriMesh::CoordType UnScale(MyTriMesh::CoordType p)
{
MyTriMesh::ScalarType x=(p.V(0))*scale.V(0);
MyTriMesh::ScalarType y=(p.V(1))*scale.V(1);
MyTriMesh::ScalarType z=(p.V(2))*scale.V(2);
return (MyTriMesh::CoordType(x,y,z));
}
2004-12-17 12:17:36 +01:00
///scale the coordinates of a point
MyTriMesh::CoordType Scale(MyTriMesh::CoordType p)
{
MyTriMesh::ScalarType x=(p.V(0))/scale.V(0);
MyTriMesh::ScalarType y=(p.V(1))/scale.V(1);
MyTriMesh::ScalarType z=(p.V(2))/scale.V(2);
return (MyTriMesh::CoordType(x,y,z));
}
///return true if a coordinate is out of limits
bool OutOfLimits(MyTriMesh::CoordType p)
{
2004-12-20 18:56:01 +01:00
Point3f test=p;
2005-02-08 18:47:50 +01:00
Point3f max=BBox().max-Point3f(1.f,1.f,1.f);//last change
Point3f min=BBox().min;//+Point3f(1.f,1.f,1.f);//last change
2004-12-17 12:17:36 +01:00
for (int i=0;i<3;i++)
{
if(((test.V(i)>=max.V(i))||(test.V(i)<=min.V(i))))
return true;
}
return false;
}
bool IsBlocked(MyVertex *v)
{
//return ((v->Flags()& BlockFlag!=0));
return (v->blocked);
}
void SetBlocked(MyVertex *v)
{
//v->Flags()|= BlockFlag;
v->blocked=true;
//v->SetS();//for refine
}
void SetBlockedFace(MyFace *f)
{
SetBlocked(f->V(0));
SetBlocked(f->V(1));
SetBlocked(f->V(2));
}
void SetIntersectedFace(MyFace *f)
{
f->intersected=true;
2005-04-11 11:26:22 +02:00
f->ClearS();
//if ((!f->intersected)&&(!f->IsD()))
//{
// f->intersected=true;
// /////detach from Face-Face Topology
// for (int i=0;i<3;i++)
// if (!f->IsBorder(i))
// vcg::face::FFDetach<MyFace>(*f,i);
// f->SetD();
// m->fn--;
//}
2004-12-17 12:17:36 +01:00
}
bool IsStopped(MyVertex *v)
{
//return ((v->Flags()& StoppedFlag!=0));
return (v->stopped);
}
void SetStopped(MyVertex *v)
{
//v->Flags()|= StoppedFlag;
v->stopped=true;
V_Stopped.push_back(v);
}
void ClearStopped(MyVertex *v)
{
//v->Flags()&= ~StoppedFlag;
v->stopped=false;
}
2005-02-08 18:47:50 +01:00
/////re-set physical pararmeters on the mesh
//void InitPhysParam(float k_elanst,float mass,float k_dihedral)
//{
// for (unsigned int i=0;i<m->face.size();i++)
// {
// if (!m->face[i].IsD())
// m->face[i].Init(k_elanst,mass,k_dihedral);
// }
//}
2004-12-17 12:17:36 +01:00
///set the initial mesh of deformable object
2005-02-01 19:07:14 +01:00
void InitMesh(MyTriMesh *m)
2004-12-17 12:17:36 +01:00
{
2005-02-01 19:07:14 +01:00
m->Clear();
2005-02-08 18:47:50 +01:00
2005-02-01 19:07:14 +01:00
vcg::tri::Icosahedron<MyTriMesh>(*m);
2004-12-17 12:17:36 +01:00
2005-02-01 19:07:14 +01:00
vcg::tri::UpdateTopology<MyTriMesh>::FaceFace(*m);
2004-12-17 12:17:36 +01:00
/* P_Vertex.clear();
P_Faces.clear();*/
2005-02-01 19:07:14 +01:00
for (unsigned int i=0;i<m->vert.size();i++)
2004-12-17 12:17:36 +01:00
{
2005-02-01 19:07:14 +01:00
m->vert[i].P()=UnScale(m->vert[i].P());///last change
m->vert[i].P()+=InitialBarycenter;
2005-04-11 11:26:22 +02:00
m->vert[i].SetRestPos();
2005-02-08 18:47:50 +01:00
// m.vert[i].P()=UnScale(m.vert[i].P());
2004-12-17 12:17:36 +01:00
// P_Vertex.push_back(&m.vert[i]);
}
2005-02-01 19:07:14 +01:00
vcg::tri::UpdateNormals<MyTriMesh>::PerVertexNormalized(*m);
2004-12-20 18:56:01 +01:00
2004-12-17 12:17:36 +01:00
}
///return true if the gray level of the vertex v differ from graylevel less than tolerance
bool InTolerance(MyTriMesh::VertexType *v)
{
2005-02-08 18:47:50 +01:00
return (fabs(getColor(v->P())-(float)gray_init)<(float)tolerance);
2004-12-17 12:17:36 +01:00
}
///add to the vertex v a containing force basing on diffence from tolerance
MyTriMesh::CoordType ContainingForce(MyTriMesh::VertexType *v)
{
//float dinstance=fabs((FindGrayMedia(v))-gray_init);
float dinstance=fabs((getColor(v->P()))-(float)gray_init);
assert(dinstance<=tolerance);
MyTriMesh::CoordType ret=(-v->N()*((dinstance)/(float)tolerance));
return (ret);
}
///find the gradient factor
MyTriMesh::CoordType GradientFactor(MyTriMesh::VertexType *v)
{
MyTriMesh::CoordType value=Gradient(v->P());
/*float d0=getColor(v->P()+value);
float d1=getColor(v->P()-value);
if ((fabs(d0-(float)gray_init))>(fabs(d1-(float)gray_init)))
return (-value);
else */
return (value*(gray_init-getColor(v->P())));
}
///add the external forces to the deformable mesh
2004-12-21 18:38:42 +01:00
2004-12-17 12:17:36 +01:00
void AddExtForces()
{
Part_VertexContainer::iterator vi;
2005-04-20 17:20:13 +02:00
//PartialUpdateNormals();
2004-12-17 12:17:36 +01:00
end_loop=true;
for (vi=P_Vertex.begin();vi<P_Vertex.end();++vi)
{
if (!(*vi).IsD())
{
if (OutOfLimits((*vi).P()))
2005-02-08 18:47:50 +01:00
{
2004-12-17 12:17:36 +01:00
SetBlocked(&*vi);
2005-02-08 18:47:50 +01:00
(*vi).ExtForce()=MyTriMesh::CoordType(0,0,0);
}
else
2004-12-17 12:17:36 +01:00
if ((!IsBlocked(&*vi))&&(!IsStopped(&*vi)))
{
end_loop=false;
if (!InTolerance(&*vi))
{
SetBlocked(&*vi);
(*vi).ExtForce()=MyTriMesh::CoordType(0,0,0);
}
else
{
MyTriMesh::CoordType Inflating=(*vi).N();
MyTriMesh::CoordType Containing0=ContainingForce(&*vi);
2004-12-20 18:56:01 +01:00
if (Containing0.Norm()>1)
Containing0.Normalize();
2005-04-20 17:20:13 +02:00
Containing0*=0.75;
2004-12-20 18:56:01 +01:00
(*vi).ExtForce()=Inflating+Containing0;/*+Containing1+Containing0*/;
2004-12-17 12:17:36 +01:00
}
}
else
(*vi).ExtForce()=MyTriMesh::CoordType(0,0,0);
}
}
}
///reinit the partial integration vectors that describe active vertices
void Reinit_PVectors()
{
V_Stopped.clear();
P_Vertex.clear();
MyTriMesh::VertexIterator vi;
2005-02-01 19:07:14 +01:00
for (vi=m->vert.begin();vi<m->vert.end();vi++)
2004-12-17 12:17:36 +01:00
{
if ((!vi->IsD())&&(!vi->blocked))
P_Vertex.push_back(&(*vi));
if ((!vi->IsD())&&((*vi).stopped)&&(!vi->blocked))
V_Stopped.push_back(&(*vi));
}
P_Faces.clear();
MyTriMesh::FaceIterator fi;
2005-02-01 19:07:14 +01:00
for (fi=m->face.begin();fi<m->face.end();fi++)
2004-12-17 12:17:36 +01:00
{
if ((!fi->IsD())&&(!fi->IsBlocked()))
P_Faces.push_back(&(*fi));
}
}
///erase the stopped entities from the partial containers
void Refresh_PVectors()
{
Part_FaceContainer P_FacesAux;
Part_VertexContainer P_VertexAux;
P_FacesAux.clear();
P_VertexAux.clear();
2005-02-01 19:07:14 +01:00
unsigned int i=0;
2004-12-17 12:17:36 +01:00
for (i=0;i<P_Vertex.size();i++)
{
2005-02-08 18:47:50 +01:00
if ((!P_Vertex[i]->IsD())&&(!P_Vertex[i]->blocked))
2004-12-17 12:17:36 +01:00
P_VertexAux.push_back(P_Vertex[i]);
}
for (i=0;i<P_Faces.size();i++)
{
2005-02-08 18:47:50 +01:00
if ((!P_Faces[i]->IsD())&&(!P_Faces[i]->IsBlocked()))
2004-12-17 12:17:36 +01:00
P_FacesAux.push_back(P_Faces[i]);
}
P_Faces.clear();
P_Vertex.clear();
P_Faces=P_FacesAux;
P_Vertex=P_VertexAux;
}
///add the new elements on partial vectors when allocate space for new vertices
void AddNewElements(MyTriMesh::VertexIterator vi,MyTriMesh::FaceIterator fi)
{
2005-02-01 19:07:14 +01:00
while (vi!=m->vert.end())
2004-12-17 12:17:36 +01:00
{
2005-02-08 18:47:50 +01:00
if ((!(*vi).IsD())&&(!(*vi).blocked)&&(!(*vi).stopped))
2004-12-17 12:17:36 +01:00
P_Vertex.push_back(&(*vi));
vi++;
}
2005-02-01 19:07:14 +01:00
while (fi!=m->face.end())
2004-12-17 12:17:36 +01:00
{
2005-02-08 18:47:50 +01:00
if ((!(*fi).IsD())&&((*fi).IsActive()))
2004-12-17 12:17:36 +01:00
P_Faces.push_back(&(*fi));
fi++;
}
}
///verify and eventually stop the vertices of the mesh
void VerifyForces()
{
float proj;
Part_VertexContainer::iterator vi;
for (vi=P_Vertex.begin();vi<P_Vertex.end();++vi)
{
if (!IsStopped(&*vi))
{
MyTriMesh::CoordType accn=(*vi).Acc();
proj=accn*(*vi).N();
if ((proj)<=0)
SetStopped(&*vi);
}
}
}
bool TimeReinit()
{
static clock_t time=0;
clock_t elapsedsecs=abs(time-clock());
if (elapsedsecs>interval_reinit)
{
time=clock();
return true;
}
return false;
}
bool TimeSelfIntersection()
{
static clock_t time=0;
clock_t elapsedsecs=abs(time-clock());
if (elapsedsecs>interval_selfcollision)
{
time=clock();
return true;
}
return false;
}
2005-02-08 18:47:50 +01:00
void PartialUpdateNormals()
{
2005-04-20 17:20:13 +02:00
//vcg::tri::UpdateNormals<MyTriMesh>::PerFaceNormalized(*m);
//vcg::tri::UpdateNormals<MyTriMesh>::PerVertexNormalized(*m);
2005-02-08 18:47:50 +01:00
Part_FaceContainer::iterator fi;
for (fi=P_Faces.begin();fi<P_Faces.end();++fi)
if (!(*fi).IsD())
(*fi).ComputeNormalizedNormal();
Part_VertexContainer::iterator vi;
for (vi=P_Vertex.begin();vi<P_Vertex.end();++vi)
if( !(*vi).IsD() && (*vi).IsRW() )
(*vi).N() = MyVertex::NormalType(0.f,0.f,0.f);
for(fi=P_Faces.begin();fi!=P_Faces.end();++fi)
if( !(*fi).IsD() && (*fi).IsR() )
{
MyFace::NormalType t = (*fi).Normal();
for(int j=0; j<3; ++j)
if( !(*fi).V(j)->IsD() && (*fi).V(j)->IsRW() )
(*fi).V(j)->N() += t;
}
for(vi=P_Vertex.begin();vi!=P_Vertex.end();++vi)
if( !(*vi).IsD() && (*vi).IsRW() )
(*vi).N().Normalize();
}
2005-04-11 11:26:22 +02:00
//bool SelectToRefine()
//{
// MyTriMesh::FaceIterator fi;
// for (fi=m->face.begin();pfi<m->face.end();++pfi)
// {
// if (((*fi).QualityFace()<0.1f))
// fi->ClearS();
// }
//}
template<class MESH_TYPE>
struct MyMidPoint:vcg::MidPoint<MESH_TYPE>
{
void operator()(typename MESH_TYPE::VertexType &nv, face::Pos<typename MESH_TYPE::FaceType> ep){
nv.P()= (ep.f->V(ep.z)->P()+ep.f->V1(ep.z)->P())/2.0;
nv.RPos()=(ep.f->V(ep.z)->RPos()+ep.f->V1(ep.z)->RPos())/2.0;
2005-04-20 17:20:13 +02:00
2005-04-11 11:26:22 +02:00
if( MESH_TYPE::HasPerVertexNormal())
nv.N()= (ep.f->V(ep.z)->N()+ep.f->V1(ep.z)->N()).Normalize();
if( MESH_TYPE::HasPerVertexColor())
nv.C().lerp(ep.f->V(ep.z)->C(),ep.f->V1(ep.z)->C(),.5f);
}
};
2004-12-17 12:17:36 +01:00
///refine the mesh and re-update eventually
void RefineStep(float _edge_size)
{
2005-02-01 19:07:14 +01:00
MyTriMesh::VertexIterator vinit=m->vert.begin();
MyTriMesh::FaceIterator finit=m->face.begin();
MyTriMesh::VertexIterator vend=m->vert.end();
MyTriMesh::FaceIterator fend=m->face.end();
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
// bool to_refine=SelectToRefine();
/*if (to_refine)*/
//refined=vcg::Refine(*m,MidPoint<MyTriMesh>(),_edge_size,true);
refined=vcg::Refine(*m,MyMidPoint<MyTriMesh>(),_edge_size,true);
/*else
refined=false;*/
2004-12-17 12:17:36 +01:00
if (refined)
{
2005-04-11 11:26:22 +02:00
2005-02-01 19:07:14 +01:00
MyTriMesh::VertexIterator vinit2=m->vert.begin();
MyTriMesh::FaceIterator finit2=m->face.begin();
2004-12-17 12:17:36 +01:00
if ((vinit2!=vinit)||(finit2!=finit))
2005-02-08 18:47:50 +01:00
{
2004-12-17 12:17:36 +01:00
Reinit_PVectors();
2005-02-08 18:47:50 +01:00
CollDet->RefreshElements();
}
2004-12-17 12:17:36 +01:00
else
2005-02-08 18:47:50 +01:00
{
2004-12-17 12:17:36 +01:00
AddNewElements(vend,fend);
2005-02-08 18:47:50 +01:00
CollDet->UpdateStep<Part_FaceContainer>(P_Faces);
}
2005-04-11 11:26:22 +02:00
/*vcg::tri::UpdateNormals<MyTriMesh>::PerFaceNormalized(*m);
vcg::tri::UpdateNormals<MyTriMesh>::PerVertexNormalized(*m);*/
2005-04-20 17:20:13 +02:00
2005-02-08 18:47:50 +01:00
PartialUpdateNormals();
2005-04-11 11:26:22 +02:00
//#ifdef _DEBUG
// vcg::tri::UpdateTopology<MyTriMesh>::TestFaceFace(*m);
//#endif
2005-02-08 18:47:50 +01:00
//CollDet->RefreshElements();
2004-12-17 12:17:36 +01:00
}
}
///reset vertex position and unblock them
void ReinitPhysicMesh()
{
Part_FaceContainer::iterator pfi;
2005-04-20 17:20:13 +02:00
/*for (pfi=P_Faces.begin();pfi<P_Faces.end();++pfi)
2005-02-08 18:47:50 +01:00
if((!(*pfi).IsD())&&((!(*pfi).intersected)))
2005-04-20 17:20:13 +02:00
(*pfi).Init(k_elanst,mass,k_dihedral);*/
2004-12-17 12:17:36 +01:00
2005-04-11 11:26:22 +02:00
Part_VertexContainer::iterator pvi;
for (pvi=P_Vertex.begin();pvi<P_Vertex.end();++pvi)
if(!(*pvi).IsD())
(*pvi).SetRestPos();
2005-04-20 17:20:13 +02:00
for (pfi=P_Faces.begin();pfi<P_Faces.end();++pfi)
if((!(*pfi).IsD())&&((!(*pfi).intersected)))
(*pfi).Init(k_elanst,mass,k_dihedral);
//PartialUpdateNormals();
2004-12-17 12:17:36 +01:00
/*for (MyTriMesh::VertexIterator vi=m.vert.begin();vi<m.vert.end();vi++)
ClearStopped(&*vi);*/
}
///clear the stopped vertex
void ClearStopped()
{
//for (MyTriMesh::VertexIterator vi=m.vert.begin();vi<m.vert.end();vi++)
// ClearStopped(&*vi);
Part_VertexContainer::iterator vi;
for (vi=V_Stopped.begin();vi<V_Stopped.end();++vi)
{
ClearStopped(&*vi);
}
V_Stopped.clear();
}
///do one step of controls for self collision detetction
void CollisionDetection()
{
2005-02-01 19:07:14 +01:00
CollDet->UpdateStep<Part_FaceContainer>(P_Faces);
2004-12-17 12:17:36 +01:00
std::vector<MyFace*> coll=CollDet->computeSelfIntersection();
for (std::vector<MyFace*>::iterator it=coll.begin();it<coll.end();it++)
{
2005-02-08 18:47:50 +01:00
if (!(*it)->IsD())
{
SetBlockedFace(*it);
SetIntersectedFace(*it);
}
2004-12-17 12:17:36 +01:00
}
}
public:
///set the initial barycenter where the triangle mesh start to expand
2004-12-20 18:56:01 +01:00
//void SetInitialBarycenter(MyTriMesh::CoordType b)
//{
// InitialBarycenter=b;
// gray_init=getColor(b);
// /*InitMesh(m);*/
//}
2004-12-17 12:17:36 +01:00
///set the input output directory of images
void LoadFromDir(char *in, char *out)
{
inDir=in;
outDir=out;
//caso optimized
/*V.Resample(inDir,outDir);
V.Init(1000,outDir);*/
V.Load(inDir);
2004-12-20 18:56:01 +01:00
bbox=vcg::Box3<float>(MapToSpace((V.Min())),(MapToSpace(V.Max())));
2004-12-17 12:17:36 +01:00
}
2004-12-20 18:56:01 +01:00
///set parameters for segmentation
2005-02-08 18:47:50 +01:00
void SetSegmentParameters(int tol,float Mass=0.5f,float K_elanst=0.2f,float Dihedral=0.2f,float Time_stamp=0.2f,
2004-12-20 18:56:01 +01:00
float Edge_precision=4.f,Point3f ScaleFactor=Point3f(1.f,1.f,1.f),
2005-04-20 17:20:13 +02:00
clock_t _interval=2000,clock_t _interval2=250)
2004-12-17 12:17:36 +01:00
{
mass=Mass;
k_elanst=K_elanst;
tolerance=tol;
interval_reinit=_interval;
interval_selfcollision=_interval2;
edge_size=16.f;
2005-04-20 17:20:13 +02:00
edge_init=edge_size;
2004-12-17 12:17:36 +01:00
edge_precision=Edge_precision;
time_stamp=Time_stamp;
k_dihedral=Dihedral;
scale=ScaleFactor;
2004-12-21 18:38:42 +01:00
bbox=vcg::Box3<float>(UnScale(MapToSpace(V.Min())),(UnScale(MapToSpace(V.Max()))));//last change!
2004-12-20 18:56:01 +01:00
}
2004-12-21 18:38:42 +01:00
2004-12-20 18:56:01 +01:00
///init the segmentation of the mesh
void InitSegmentation(MyTriMesh::CoordType b)
{
InitialBarycenter=b;
gray_init=getColor(b);
2004-12-17 12:17:36 +01:00
TrINT= new myIntegrator(P_Faces,P_Vertex);
TrINT->SetPDESolver(PDESolvers::EULER_METHOD);
2004-12-20 18:56:01 +01:00
InitMesh(m);
2005-02-08 18:47:50 +01:00
2004-12-17 12:17:36 +01:00
//init the mesh with new
Reinit_PVectors();
2005-02-08 18:47:50 +01:00
2004-12-17 12:17:36 +01:00
ReinitPhysicMesh();
CollDet->Init(bbox.min,bbox.max,5.f);
}
///return the bounding box of the mesh
2005-04-20 17:20:13 +02:00
vcg::Box3<float>& BBox()
2004-12-17 12:17:36 +01:00
{
return (bbox);
}
///one step of moving for the deformable object
void Step(float t,float _edge_size)
{
2005-02-01 19:07:14 +01:00
if (m->face.size()!=0)
2004-12-17 12:17:36 +01:00
{
AddExtForces();
TrINT->Step(t);
VerifyForces();
Refresh_PVectors();
if (end_loop)
{
RefineStep(_edge_size);
ReinitPhysicMesh();
ClearStopped();
}
if (TimeSelfIntersection())
CollisionDetection();
}
}
void Smooth()
{
2005-02-01 19:07:14 +01:00
ScaleLaplacianSmooth<MyTriMesh>(*m,1,0.5);
2004-12-17 12:17:36 +01:00
}
void AutoStep()
{
refined=false;
Step(time_stamp,edge_size);
//test on 80% of the vertex blocked
2005-02-01 19:07:14 +01:00
if ((((float)P_Vertex.size()/(float)m->vn)<0.2)&&(end_loop)&&(!refined)&&(edge_size>edge_precision))
2004-12-17 12:17:36 +01:00
{
edge_size/=2.f;
if (edge_size<edge_precision)
edge_size=edge_precision;
time_stamp/=2.f;
}
}
2005-04-20 17:20:13 +02:00
///set as deleted intersected vertices
void ClearIntersectedFaces()
{
MyTriMesh::FaceIterator fi;
for (fi=m->face.begin();fi<m->face.end();fi++)
#ifdef _TORUS
if (((*fi).intersected==true))///||((*fi).HaveVertexActive()))
{
m->fn--;
(*fi).SetD();
}
#else
if (((*fi).intersected==true))//||((*fi).HaveVertexActive()))
{
m->fn--;
(*fi).SetD();
}
#endif
}
2005-02-08 18:47:50 +01:00
2005-02-01 19:07:14 +01:00
///first version
void Resample()
{
2005-04-20 17:20:13 +02:00
ClearIntersectedFaces();
2005-02-01 19:07:14 +01:00
new_m=new MyTriMesh();
2005-02-08 18:47:50 +01:00
vcg::tri::UpdateBounding<MyTriMesh>::Box(*m);
2005-04-20 17:20:13 +02:00
#ifdef _TORUS
2005-02-01 19:07:14 +01:00
vcg::trimesh::Resampler<MyTriMesh,MyTriMesh>::Resample<vcg::trimesh::RES::MMarchingCubes>(*m,*new_m,
2005-04-20 17:20:13 +02:00
Point3i((int) edge_precision,(int) edge_precision,(int) edge_precision),edge_init);
#else
vcg::trimesh::Resampler<MyTriMesh,MyTriMesh>::Resample<vcg::trimesh::RES::MMarchingCubes>(*m,*new_m,
Point3i((int) edge_precision,(int) edge_precision,(int) edge_precision),edge_size*2.f);
#endif
// delete(new_m0);
2005-02-01 19:07:14 +01:00
delete(m);
m=new_m;
Reinit_PVectors();
ReinitPhysicMesh();
CollDet->Init(bbox.min,bbox.max,5.f);
}
2004-12-17 12:17:36 +01:00
};
#endif