DGtal  0.9.4beta
volMarchingCubes.cpp
1 
32 #include <iostream>
35 #include <queue>
36 #include "DGtal/base/Common.h"
37 #include "DGtal/kernel/CanonicEmbedder.h"
38 #include "DGtal/io/readers/VolReader.h"
39 #include "DGtal/images/ImageSelector.h"
40 #include "DGtal/images/imagesSetsUtils/SetFromImage.h"
41 #include "DGtal/images/ImageLinearCellEmbedder.h"
42 #include "DGtal/shapes/Shapes.h"
43 #include "DGtal/helpers/StdDefs.h"
44 #include "DGtal/topology/helpers/Surfaces.h"
45 #include "DGtal/topology/DigitalSurface.h"
46 #include "DGtal/topology/SetOfSurfels.h"
48 
50 
51 using namespace std;
52 using namespace DGtal;
53 using namespace Z3i;
54 
56 
57 void usage( int, char** argv )
58 {
59  std::cerr << "Usage: " << argv[ 0 ] << " <fileName.vol> <minT> <maxT> <Adj>" << std::endl;
60  std::cerr << "\t - displays the boundary of the shape stored in vol file <fileName.vol>." << std::endl;
61  std::cerr << "\t - voxel v belongs to the shape iff its value I(v) follows minT <= I(v) <= maxT." << std::endl;
62  std::cerr << "\t - 0: interior adjacency, 1: exterior adjacency." << std::endl;
63 
64 }
65 
66 int main( int argc, char** argv )
67 {
68  if ( argc < 5 )
69  {
70  usage( argc, argv );
71  return 1;
72  }
73  std::string inputFilename = argv[ 1 ];
74  unsigned int minThreshold = atoi( argv[ 2 ] );
75  unsigned int maxThreshold = atoi( argv[ 3 ] );
76  bool intAdjacency = atoi( argv[ 4 ] ) == 0;
77 
79  trace.beginBlock( "Reading vol file into an image." );
81  Image image = VolReader<Image>::importVol(inputFilename);
82  DigitalSet set3d (image.domain());
83  SetFromImage<DigitalSet>::append<Image>(set3d, image,
84  minThreshold, maxThreshold);
85  trace.endBlock();
87 
88 
90  trace.beginBlock( "Construct the Khalimsky space from the image domain." );
91  KSpace ks;
92  bool space_ok = ks.init( image.domain().lowerBound(),
93  image.domain().upperBound(), true );
94  if (!space_ok)
95  {
96  trace.error() << "Error in the Khamisky space construction."<<std::endl;
97  return 2;
98  }
99  trace.endBlock();
101 
103  typedef SurfelAdjacency<KSpace::dimension> MySurfelAdjacency;
104  MySurfelAdjacency surfAdj( intAdjacency ); // interior in all directions.
106 
108  trace.beginBlock( "Extracting boundary by scanning the space. " );
109  typedef KSpace::SurfelSet SurfelSet;
110  typedef SetOfSurfels< KSpace, SurfelSet > MySetOfSurfels;
112  MySetOfSurfels theSetOfSurfels( ks, surfAdj );
113  Surfaces<KSpace>::sMakeBoundary( theSetOfSurfels.surfelSet(),
114  ks, set3d,
115  image.domain().lowerBound(),
116  image.domain().upperBound() );
117  MyDigitalSurface digSurf( theSetOfSurfels );
118  trace.info() << "Digital surface has " << digSurf.size() << " surfels."
119  << std::endl;
120  trace.endBlock();
122 
124  trace.beginBlock( "Making OFF surface <marching-cube.off>. " );
125  typedef CanonicEmbedder< Space > MyEmbedder;
126  typedef
128  CellEmbedder cellEmbedder;
129  MyEmbedder trivialEmbedder;
130  // The +0.5 is to avoid isosurface going exactly through a voxel
131  // center, especially for binary volumes.
132  cellEmbedder.init( ks, image, trivialEmbedder,
133  ( (double) minThreshold ) + 0.5 );
134  ofstream out( "marching-cube.off" );
135  if ( out.good() )
136  digSurf.exportEmbeddedSurfaceAs3DOFF( out, cellEmbedder );
137  out.close();
138  trace.endBlock();
140  return 0;
141 }
142 
void beginBlock(const std::string &keyword="")
DigitalSetSelector< Domain, BIG_DS+HIGH_BEL_DS >::Type DigitalSet
Definition: StdDefs.h:100
Aim: implements association bewteen points lying in a digital domain and values.
Definition: Image.h:69
Trace trace
Definition: Common.h:137
Aim: A utility class for constructing surfaces (i.e. set of (n-1)-cells).
Definition: Surfaces.h:78
KhalimskySpaceND< 2, Integer > KSpace
Definition: StdDefs.h:77
STL namespace.
double endBlock()
Aim: Represents a set of n-1-cells in a nD space, together with adjacency relation between these cell...
Aim: implements methods to read a "Vol" file format.
Definition: VolReader.h:89
DGtal is the top-level namespace which contains all DGtal functions and types.
Aim: Define utilities to convert a digital set into an image.
Definition: SetFromImage.h:63
Aim: A trivial embedder for digital points, which corresponds to the canonic injection of Zn into Rn...
std::ostream & info()
Aim: A model of CDigitalSurfaceContainer which defines the digital surface as connected surfels...
Definition: SetOfSurfels.h:73
const Domain & domain() const
Definition: Image.h:192
void init(ConstAlias< KSpace > K, ConstAlias< Image > f, ConstAlias< Embedder > e, double iso_value)
std::ostream & error()
Aim: a cellular embedder for images. (default constructible, copy constructible, assignable). Model of CCellEmbedder.