82 lines
2.8 KiB
C++
82 lines
2.8 KiB
C++
#include <vcg/complex/complex.h>
|
|
|
|
#include <wrap/io_trimesh/import.h>
|
|
#include <wrap/io_trimesh/export_ply.h>
|
|
|
|
#include <vcg/complex/algorithms/geodesic_heat.h>
|
|
#include <vcg/complex/algorithms/update/color.h>
|
|
|
|
#include <iostream>
|
|
|
|
|
|
using namespace vcg;
|
|
using namespace std;
|
|
|
|
class MyEdge;
|
|
class MyFace;
|
|
class MyVertex;
|
|
struct MyUsedTypes : public UsedTypes< Use<MyVertex> ::AsVertexType,
|
|
Use<MyEdge> ::AsEdgeType,
|
|
Use<MyFace> ::AsFaceType>{};
|
|
|
|
class MyVertex : public Vertex<MyUsedTypes, vertex::Coord3f, vertex::VFAdj, vertex::Color4b, vertex::Qualityf, vertex::BitFlags>{};
|
|
class MyFace : public Face< MyUsedTypes, face::VFAdj, face::FFAdj, face::VertexRef, face::Normal3f, face::BitFlags, vertex::Qualityf> {};
|
|
class MyEdge : public Edge<MyUsedTypes>{};
|
|
class MyMesh : public tri::TriMesh< vector<MyVertex>, vector<MyFace> , vector<MyEdge> > {};
|
|
|
|
int main( int argc, char **argv )
|
|
{
|
|
if(argc<2)
|
|
{
|
|
printf("Usage trimesh_geodesic <meshfilename> <backEulerStepfloat>\n");
|
|
return -1;
|
|
}
|
|
float param_m = 1.0;
|
|
if (argc>2){
|
|
param_m = atof(argv[2]);
|
|
}
|
|
|
|
MyMesh m;
|
|
if(tri::io::Importer<MyMesh>::Open(m, argv[1])!=0)
|
|
{
|
|
printf("Error reading file %s\n",argv[1]);
|
|
exit(0);
|
|
}
|
|
|
|
Point3f c = m.bbox.Center();
|
|
MyVertex* closest = &*m.vert.begin();
|
|
float minDist = Distance(closest->P(),c);
|
|
for(MyMesh::VertexIterator vi=m.vert.begin();vi!=m.vert.end(); ++vi){
|
|
if(Distance(vi->P(),c)<minDist){
|
|
minDist = Distance(vi->P(),c);
|
|
closest = &*vi;
|
|
}
|
|
}
|
|
vector<MyVertex*> seedVec;
|
|
seedVec.push_back(closest);
|
|
tri::Clean<MyMesh>::RemoveUnreferencedVertex(m);
|
|
|
|
bool success;
|
|
success = tri::GeodesicHeat<MyMesh>::Compute(m, seedVec, param_m);
|
|
if (!success){
|
|
printf("computation of HeatGeodesic has failed! %d", success);
|
|
exit(0);
|
|
}
|
|
pair<float,float> minmax = tri::Stat<MyMesh>::ComputePerVertexQualityMinMax(m);
|
|
tri::UpdateColor<MyMesh>::PerVertexQualityRamp(m);
|
|
printf("min %f max %f\n",minmax.first, minmax.second);
|
|
tri::io::ExporterPLY<MyMesh>::Save(m,"base.ply",tri::io::Mask::IOM_VERTCOLOR | tri::io::Mask::IOM_VERTQUALITY);
|
|
|
|
int t0=clock();
|
|
tri::GeodesicHeat<MyMesh>::GeodesicHeatCache cache = tri::GeodesicHeat<MyMesh>::BuildCache(m, 1.0);
|
|
tri::GeodesicHeat<MyMesh>::ComputeFromCache(m, seedVec, cache);
|
|
int t1=clock();
|
|
tri::GeodesicHeat<MyMesh>::ComputeFromCache(m, seedVec, cache);
|
|
int t2=clock();
|
|
printf("Non-Cached Time: %6.3f\n",float(t1-t0)/CLOCKS_PER_SEC);
|
|
printf("Cached Time : %6.3f\n",float(t2-t1)/CLOCKS_PER_SEC);
|
|
tri::io::ExporterPLY<MyMesh>::Save(m,"base_m1.ply",tri::io::Mask::IOM_VERTCOLOR | tri::io::Mask::IOM_VERTQUALITY);
|
|
|
|
return 0;
|
|
}
|