Added template methods

[creaRigidRegistration.git] / lib / Surface.h

diff --git a/lib/Surface.h b/lib/Surface.h
index 49b67249f3095d2b231fcce8a9b461ded67fff87..20e0f8bd80082b41b6272f8aa1417598701983c3 100644 (file)
--- a/lib/Surface.h
+++ b/lib/Surface.h
@@ -1,6 +1,8 @@
 #include "vtkImageData.h"
 #include "vtkProp3D.h"
 #include "vtkRenderer.h"
+#include "vtkPoints.h"
+#include "vtkCellArray.h"
 
 #include <string>
 
@@ -34,9 +36,14 @@ class Surface
                vtkActor* getSurface();
                
                /*
-               constructing image substract
+               constructing image surface
                */
                void surface(vtkImageData* imageData);
+
+               /*
+               Template for constructing the surface by image type
+               */
+               template <class T> void surfaceByType(T* dataImagePointer, vtkImageData* imageData, T max);
                
 
 
@@ -63,4 +70,15 @@ class Surface
                The type of image being viewed
                */
                int imageType;
-};
\ No newline at end of file
+
+               /*
+               The points of the surface
+               */
+               vtkPoints* surfacePoints;
+
+               /*
+               Cells to be constructed from the surface
+               */
+               vtkCellArray* surfaceCells;
+};
+