/vol/vipdata/irtk/common++/include/irtkCofstream.h

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

  Library   : Image Registration Toolkit (IRTK)
  Module    : $Id: irtkCofstream.h 2 2008-12-23 12:40:14Z dr $
  Copyright : Imperial College, Department of Computing
              Visual Information Processing (VIP), 2008 onwards
  Date      : $Date: 2008-12-23 12:40:14 +0000 (Tue, 23 Dec 2008) $
  Version   : $Revision: 2 $
  Changes   : $Author: dr $

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

#ifndef _IRTKCOFSTREAM_H

#define _IRTKCOFSTREAM_H

class irtkCofstream : public irtkObject
{

  FILE *_uncompressedFile;

#ifdef HAS_ZLIB
  gzFile _compressedFile;
#endif

protected:

  int _compressed;

  int _swapped;

public:

  irtkCofstream();

  ~irtkCofstream();

  void Write(char *data, int offset, int length);

  void WriteAsChar  (char data, int offset = -1);
  void WriteAsChar  (char *data, int length, int offset = -1);

  void WriteAsUChar (unsigned char data, int offset = -1);
  void WriteAsUChar (unsigned char *data, int length, int offset = -1);

  void WriteAsShort (short data, int offset = -1);
  void WriteAsShort (short *data, int length, int offset = -1);

  void WriteAsUShort(unsigned short data, int offset = -1);
  void WriteAsUShort(unsigned short *data, int length, int offset = -1);

  void WriteAsInt   (int data, int offset = -1);
  void WriteAsInt   (int *data, int length, int offset = -1);

  void WriteAsUInt  (unsigned int data, int offset = -1);
  void WriteAsUInt  (unsigned int *data, int length, int offset = -1);

  void WriteAsFloat (float data, int offset = -1);
  void WriteAsFloat (float *data, int length, int offset = -1);

  void WriteAsDouble(double data, int offset = -1);
  void WriteAsDouble(double *data, int length, int offset = -1);

  void WriteAsString(char *data, int offset = -1);

  void Open(const char *);

  void Close();

  int  IsCompressed();

  void IsCompressed(int);

  int  IsSwapped();

  void IsSwapped(int);

};

inline void irtkCofstream::Open(const char *filename)
{
  if (strstr(basename2(filename), ".gz") == NULL) {
    _compressed = False;
    _uncompressedFile = fopen(filename, "wb");

    // Check whether file was opened successful
    if (_uncompressedFile == NULL) {
      cerr << "cofstream::Open: Can't open file " << filename << endl;
      exit(1);
    }
  } else {
#ifdef HAS_ZLIB
    _compressed = True;
    _compressedFile = gzopen(filename, "wb");

    // Check whether file was opened successful
    if (_compressedFile == NULL) {
      cerr << "cofstream::Open: Can't open file " << filename << endl;
      exit(1);
    }
#else
    cerr << "cofstream::Open: Can't write compressed file without zlib" << endl;
    exit(1);
#endif
  }
}

inline void irtkCofstream::Close()
{
#ifdef HAS_ZLIB
  if (_compressedFile != NULL) {
    gzclose(_compressedFile);
    _compressedFile = NULL;
  }
#endif
  if (_uncompressedFile != NULL) {
    fclose(_uncompressedFile);
    _uncompressedFile = NULL;
  }
}

inline int irtkCofstream::IsCompressed()
{
  return _compressed;
}

inline void irtkCofstream::IsCompressed(int compressed)
{
  _compressed = compressed;
}

inline int irtkCofstream::IsSwapped()
{
  return _swapped;
}

inline void irtkCofstream::IsSwapped(int swapped)
{
  _swapped = swapped;
}

#endif

Generated on Mon Mar 23 16:55:52 2009 for IRTK by Doxygen 1.5.4