itkVTKImageIO2.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkVTKImageIO2.h,v $
  Language:  C++
  Date:      $Date: 2010-06-23 16:11:17 $
  Version:   $Revision: 1.3 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

  Portions of this code are covered under the VTK copyright.
  See VTKCopyright.txt or http://www.kitware.com/VTKCopyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkVTKImageIO2_h
#define __itkVTKImageIO2_h

#ifdef _MSC_VER
#pragma warning ( disable : 4786 )
#endif

#include <fstream>
#include "itkStreamingImageIOBase.h"


namespace itk
{

class ITK_EXPORT VTKImageIO2 
  : public StreamingImageIOBase
{
public:
  typedef VTKImageIO2                 Self;
  typedef StreamingImageIOBase        Superclass;
  typedef SmartPointer<Self>          Pointer;
  typedef SmartPointer<const Self>    ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(VTKImageIO2, StreamingImageIOBase);

  /*-------- This part of the interface deals with reading data. ------ */

  virtual bool CanReadFile(const char*);

  virtual void ReadImageInformation();

  virtual void Read(void* buffer);

  /*-------- This part of the interfaces deals with writing data. ----- */

  virtual bool CanWriteFile(const char*);

  virtual void WriteImageInformation() {};

  virtual void Write(const void* buffer);

  
  virtual SizeType GetHeaderSize() const { return this->m_HeaderSize; }

protected:
  VTKImageIO2();
  ~VTKImageIO2();
  void PrintSelf(std::ostream& os, Indent indent) const;

  void InternalReadImageInformation(std::ifstream& file);
  void WriteImageInformation(const void* buffer);
  
  void ReadHeaderSize( std::ifstream& file );
  
private:
  VTKImageIO2(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  void SetPixelTypeFromString( const std::string & pixelType);

  SizeType m_HeaderSize;
};

} // end namespace itk

#endif // __itkVTKImageIO2_h