X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=common%2FclitkDicomRT_ROI_ConvertToImageFilter.h;h=33d74b63df6acde0dd1027bdcc819b262deb27e3;hb=ff7b62d284bd4f5eadce3aff4b217ae722b369a7;hp=d5d27b31bbcd03bdf31f84f1d0ce9ba33307b18f;hpb=163fa80ef3873595f3e3cf75fb03e53536d1a6ee;p=clitk.git

diff --git a/common/clitkDicomRT_ROI_ConvertToImageFilter.h b/common/clitkDicomRT_ROI_ConvertToImageFilter.h
index d5d27b3..33d74b6 100644
--- a/common/clitkDicomRT_ROI_ConvertToImageFilter.h
+++ b/common/clitkDicomRT_ROI_ConvertToImageFilter.h
@@ -23,6 +23,8 @@
 #include "clitkDicomRT_ROI.h"
 #include "clitkImageCommon.h"
 #include <vtkImageData.h>
+#include <itkImage.h>
+#include <itkVTKImageToImageFilter.h>
 
 namespace clitk {
 
@@ -34,25 +36,46 @@ namespace clitk {
     ~DicomRT_ROI_ConvertToImageFilter();
 
     void SetROI(clitk::DicomRT_ROI * roi);
+    ///This is used to create a mask with the same characteristics as an input 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 Update();    
     vtkImageData * GetOutput();
+    template <int Dimension> typename itk::Image<unsigned char,Dimension>::ConstPointer GetITKOutput();
     void SetCropMaskEnabled(bool b);
 
   protected:
-    bool mImageInfoIsSet;
+    bool ImageInfoIsSet() const;
     bool mWriteOutput;
     bool mCropMask;
     std::string mOutputFilename;
     std::vector<double> mSpacing;
     std::vector<double> mOrigin;
-    std::vector<int> mSize;
+    std::vector<unsigned long> mSize;
     clitk::DicomRT_ROI * mROI;
     vtkImageData * mBinaryImage;
   };
   //--------------------------------------------------------------------
 
 } // end namespace clitk
+
+
+//--------------------------------------------------------------------
+
+template <int Dimension> 
+typename itk::Image<unsigned char,Dimension>::ConstPointer clitk::DicomRT_ROI_ConvertToImageFilter::GetITKOutput()
+{
+  assert(mBinaryImage);
+  typedef itk::Image<unsigned char,Dimension> ConnectorImageType;
+  typedef itk::VTKImageToImageFilter <ConnectorImageType> ConnectorType;
+  typename ConnectorType::Pointer connector = ConnectorType::New();
+  connector->SetInput(mBinaryImage);
+  connector->Update();
+  return connector->GetOutput();
+}
+//--------------------------------------------------------------------
 #endif // CLITKDICOMRT_ROI_CONVERTTOIMAGEFILTER_H