analyze_data_io.h

Go to the documentation of this file.

//--------------------------------------------------
//---------------------------------------------------

#ifndef IMEDGINE_ANALYZE_DATA_IO_H
#define IMEDGINE_ANALYZE_DATA_IO_H

#include "../global/global_definitions.h"
#include "dataset_definitions.h"
#include "../exceptions/invalid_cast_exception.h"
#include "data_io.h"

#include <string>

#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>

namespace imedgine
{
  //--------------------------------------------------
  //---------------------------------------------------
  
  class AnalyzeDataIO : public DataIO
  {
  public:
    
    //--------------------------------------------------
    
    AnalyzeDataIO();
    
    //--------------------------------------------------
    
    virtual ~AnalyzeDataIO();
    
    //--------------------------------------------------
    
    virtual std::pair<dataset_pointer_type, dataset_type> 
        loadDataset(std::string filename, dataset_key_type dataset_key);
    
    //--------------------------------------------------
    
    virtual void writeDataset(dataset_pointer_type dataset, std::string filename)
        throw(InvalidCastException);
    
  protected:
      
    //--------------------------------------------------
      
    template <class PixelDatasetType>
    pixel_dataset_pointer_type read(std::string filename, dataset_key_type dataset_key);
    
    //--------------------------------------------------
    
    template <class PixelDatasetPointerType>
    void write(PixelDatasetPointerType dataset, std::string filename);
  };
  
  //--------------------------------------------------

  template <class PixelDatasetType>
      pixel_dataset_pointer_type AnalyzeDataIO::read(
      std::string filename, dataset_key_type dataset_key)
  {
    typedef itk::ImageFileReader<typename PixelDatasetType::type> reader_type;
    typename reader_type::Pointer file_reader = reader_type::New();
    file_reader->SetFileName(filename.c_str());
    file_reader->UpdateLargestPossibleRegion();
    
    typename PixelDatasetType::type::Pointer dataset(file_reader->GetOutput());
    dataset->DisconnectPipeline();
    
    return(pixel_dataset_pointer_type(new PixelDatasetType(dataset_key, dataset)));
  }
  
  //--------------------------------------------------
  
  template <class PixelDatasetPointerType>
  void AnalyzeDataIO::write(
      PixelDatasetPointerType dataset, std::string filename)
  {
    typedef typename itk::ImageFileWriter<
        typename PixelDatasetPointerType::element_type::type> writer_type;
    
    typename writer_type::Pointer file_writer = writer_type::New();
    file_writer->SetFileName(filename.c_str());
    file_writer->SetInput(dataset->getDataset());
    file_writer->Update();
  }
}

#endif //IMEDGINE_ANALYZE_DATA_IO_H

---