vv/vvImageWriter.h

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ======================================================================-====*/
  18 #ifndef vvImageWriter_H
  19 #define vvImageWriter_H
  20 #include <string>
  21
  22 // ITK includes
  23 #include "itkImage.h"
  24 #include "itkCommand.h"
  25 #include "vvImage.h"
  26
  27 class vvImageWriter {
  28
  29 public:
  30     // constructor
  31     vvImageWriter();
  32     ~vvImageWriter();
  33
  34     void SetOutputFileName(std::string filename);
  35     void SetObserver(itk::Command::Pointer o) {
  36         mUseAnObserver = true;
  37         mObserver = o;
  38     }
  39     void DisableObserver() {
  40         mUseAnObserver = false;
  41     }
  42
  43     void SetInput(vvImage::Pointer image) {
  44         mImage = image;
  45     }
  46
  47     std::string GetLastError() {
  48         return mLastError;
  49     }
  50
  51     //====================================================================
  52     // Main function
  53     void Update() {
  54         Update(mImage->GetNumberOfDimensions(),mImage->GetScalarTypeAsString());
  55     }
  56     void Update(int dim, std::string OutputPixelType);
  57
  58 protected:
  59     //====================================================================
  60     std::string mOutputFilename;
  61     itk::Command::Pointer mObserver;
  62
  63     std::string mLastError;
  64     bool mUseAnObserver;
  65
  66     //====================================================================
  67     template<unsigned int VImageDimension>
  68     void UpdateWithDim(std::string OutputPixelType);
  69
  70     //====================================================================
  71     template<class OutputPixelType, unsigned int VImageDimension>
  72     void UpdateWithDimAndOutputPixelType();
  73
  74 private:
  75     vvImage::Pointer mImage;
  76
  77 }; // end class vvImageWriter
  78
  79 #endif /* end #define vvImageWriter_H */