X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FclitkDicomRTStruct2ImageFilter.h;h=f93cbb2969ca833f70682d1307b7f4993eed4cb0;hb=543b72e23ad001ac2a7743b9beacf48e2d0054ac;hp=1add60e469c5ccab31a9221cb732d87f00348389;hpb=6c46b5b1d02d65fdb3fdccc8174346b49a05a32f;p=clitk.git

diff --git a/common/clitkDicomRTStruct2ImageFilter.h b/common/clitkDicomRTStruct2ImageFilter.h
index 1add60e..f93cbb2 100644
--- a/common/clitkDicomRTStruct2ImageFilter.h
+++ b/common/clitkDicomRTStruct2ImageFilter.h
@@ -18,11 +18,12 @@
   =========================================================================*/
 
 #ifndef CLITKDICOMRTSTRUCT2IMAGEFILTER_H
-#define CLITKDICOMRT_TRUCT2IMAGEFILTER_H
+#define CLITKDICOMRTSTRUCT2IMAGEFILTER_H
 
 #include "clitkDicomRT_ROI.h"
 #include "clitkImageCommon.h"
 #include <vtkImageData.h>
+#include <vtkMatrix4x4.h>
 #include <itkImage.h>
 #include <itkVTKImageToImageFilter.h>
 
@@ -37,24 +38,30 @@ namespace clitk {
 
     void SetROI(clitk::DicomRT_ROI * roi);
     ///This is used to create a mask with the same characteristics as an input image
+    void SetImage(vvImage::Pointer image);
     void SetImageFilename(std::string s);
     void SetOutputOrigin(const double* origin);
     void SetOutputSpacing(const double* spacing);
     void SetOutputSize(const unsigned long* size);
     void SetOutputImageFilename(std::string s);
+    void SetWriteMesh(bool b);
     void Update();    
     vtkImageData * GetOutput();
     template <int Dimension> typename itk::Image<unsigned char,Dimension>::ConstPointer GetITKOutput();
     void SetCropMaskEnabled(bool b);
+    void SetWriteOutputFlag(bool b);
 
   protected:
     bool ImageInfoIsSet() const;
     bool mWriteOutput;
+    bool mWriteMesh;
     bool mCropMask;
     std::string mOutputFilename;
     std::vector<double> mSpacing;
     std::vector<double> mOrigin;
     std::vector<unsigned long> mSize;
+    std::vector< std::vector< double> > mDirection;
+    vtkSmartPointer<vtkMatrix4x4> mTransformMatrix;
     clitk::DicomRT_ROI * mROI;
     vtkSmartPointer<vtkImageData> mBinaryImage;
   };