itkImageIOBase.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkImageIOBase.h,v $
  Language:  C++
  Date:      $Date: 2007/02/22 00:21:07 $
  Version:   $Revision: 1.40 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.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 __itkImageIOBase_h
#define __itkImageIOBase_h

#include "itkLightProcessObject.h"
#include "itkObjectFactory.h"
#include "itkIndent.h"
#include "itkImageIORegion.h"
#include "vnl/vnl_vector.h"

#include <string>

namespace itk
{

class ITK_EXPORT ImageIOBase : public LightProcessObject
{
public:
  typedef ImageIOBase            Self;
  typedef LightProcessObject  Superclass;
  typedef SmartPointer<Self>  Pointer;

  itkTypeMacro(ImageIOBase, Superclass);

  itkSetStringMacro(FileName);
  itkGetStringMacro(FileName);

  class UnknownType {};

  typedef  enum {UNKNOWNPIXELTYPE,SCALAR,RGB,RGBA,OFFSET,VECTOR,
                 POINT,COVARIANTVECTOR,SYMMETRICSECONDRANKTENSOR,
                 DIFFUSIONTENSOR3D,COMPLEX,FIXEDARRAY}  IOPixelType;

  typedef  enum {UNKNOWNCOMPONENTTYPE,UCHAR,CHAR,USHORT,SHORT,UINT,INT,
                 ULONG,LONG, FLOAT,DOUBLE} IOComponentType;

  void SetNumberOfDimensions(unsigned int);
  itkGetMacro(NumberOfDimensions, unsigned int);

  virtual void SetDimensions(unsigned int i, unsigned int dim);
  virtual unsigned int GetDimensions(unsigned int i) const
  { return m_Dimensions[i]; }

  virtual void SetOrigin(unsigned int i, double origin);
  virtual double GetOrigin(unsigned int i) const
  { return m_Origin[i]; }

  virtual void SetSpacing(unsigned int i, double spacing);
  virtual double GetSpacing(unsigned int i) const
  { return m_Spacing[i]; }

  virtual void SetDirection(unsigned int i, std::vector<double> &direction);
  virtual void SetDirection(unsigned int i, vnl_vector<double> &direction);
  virtual std::vector<double> GetDirection(unsigned int i) const
  { return m_Direction[i]; }

  itkSetMacro(IORegion, ImageIORegion);
  itkGetMacro(IORegion, ImageIORegion);

  itkSetMacro(PixelType, IOPixelType);
  itkGetConstReferenceMacro(PixelType, IOPixelType);

  virtual bool SetPixelTypeInfo(const std::type_info& ptype);

  itkSetMacro(ComponentType,IOComponentType);
  itkGetConstReferenceMacro(ComponentType,IOComponentType);
  virtual const std::type_info& GetComponentTypeInfo() const;

  itkSetMacro(NumberOfComponents,unsigned int);
  itkGetConstReferenceMacro(NumberOfComponents,unsigned int);

  itkSetMacro(UseCompression,bool);
  itkGetConstReferenceMacro(UseCompression,bool);

  std::string GetComponentTypeAsString(IOComponentType) const;

  std::string GetPixelTypeAsString(IOPixelType) const;

  typedef  enum {ASCII,Binary,TypeNotApplicable} FileType;

  typedef  enum {BigEndian,LittleEndian,OrderNotApplicable} ByteOrder;

  itkSetMacro(FileType,FileType);
  itkGetConstReferenceMacro(FileType,FileType);
  void SetFileTypeToASCII()
  { this->SetFileType(ASCII); }
  void SetFileTypeToBinary()
  { this->SetFileType(Binary); }

  itkSetMacro(ByteOrder,ByteOrder);
  itkGetConstReferenceMacro(ByteOrder,ByteOrder);
  void SetByteOrderToBigEndian()
  { this->SetByteOrder(BigEndian); }
  void SetByteOrderToLittleEndian()
  { this->SetByteOrder(LittleEndian); }

  std::string GetFileTypeAsString(FileType) const;

  std::string GetByteOrderAsString(ByteOrder) const;

  virtual unsigned int GetPixelStride () const;

  unsigned int GetImageSizeInPixels() const;

  unsigned int GetImageSizeInBytes() const;

  unsigned int GetImageSizeInComponents() const;

  /*-------- This part of the interfaces deals with reading data ----- */

  virtual bool CanReadFile(const char*) = 0;

  virtual bool CanStreamRead() { return false; };

  virtual void ReadImageInformation() = 0;

  virtual void Read(void* buffer) = 0;


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

  virtual bool CanWriteFile(const char*)  = 0;

  virtual bool CanStreamWrite() { return false; };

  virtual void WriteImageInformation() = 0;

  virtual void Write( const void* buffer) = 0;

  /* --- Support reading and writing data as a series of files. --- */

  virtual bool SupportsDimension(unsigned long dim)
  {return (dim == 2);}

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

  IOPixelType m_PixelType;

  IOComponentType m_ComponentType;

  ByteOrder      m_ByteOrder;
  FileType       m_FileType;

  bool m_Initialized;

  std::string m_FileName;

  unsigned int m_NumberOfComponents;

  unsigned int m_NumberOfDimensions;

  bool m_UseCompression;

  ImageIORegion m_IORegion;

  std::vector<unsigned int> m_Dimensions;

  std::vector<double> m_Spacing;

  std::vector<double> m_Origin;

  std::vector<std::vector<double> > m_Direction;

  std::vector<unsigned int> m_Strides;

  virtual void Reset(const bool freeDynamic = true);

  void Resize(const unsigned int numDimensions, 
              const unsigned int* dimensions);

  void ComputeStrides();

  virtual unsigned int GetPixelSize() const;

  virtual unsigned int GetComponentSize() const;

  unsigned int GetComponentStride() const;

  unsigned int GetRowStride () const;

  unsigned int GetSliceStride () const;

  void WriteBufferAsASCII(std::ostream& os, const void *buffer,
                          IOComponentType ctype,
                          unsigned int numComp);

  void ReadBufferAsASCII(std::istream& os, void *buffer,
                         IOComponentType ctype,
                         unsigned int numComp);

  bool ReadBufferAsBinary(std::istream& os, void *buffer,unsigned int numComp);

  
private:
  ImageIOBase(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented


};

} // end namespace itk

#endif // __itkImageIOBase_h

---