---

CTBzfile.cxx

Go to the documentation of this file.

/*----------------------------------------------------------------------------*/
/* C Tool Box: Designed and implemented by:                                   */
/*    Walter F.J. Mueller   Gesellschaft fuer Schwerionenforschung (GSI)      */
/*                          Planckstrasse 1, D-64291 Darmstadt, Germany       */
/*                  Email:  W.F.J.Mueller@gsi.de                              */
/*                  WWW:    http://www-kp3.gsi.de/www/kp3/people/mueller.html */
/*------------------------------------------+---------------------------------*/

#include <stdio.h>
#include <string.h>
#include <zlib.h>

#include "CTB.hxx"
#include "CTB_Trace.hxx"
#include "CTBosFill.hxx"
#include "CTBzfile.hxx"

//------------------------------------------+-----------------------------------

CTBzfile::CTBzfile()
  : mi_type(closed),
    m_file(0),
    mb_bad(false),
    mb_eof(false)
{
  CTB_Trace("CTBzfile()");
}

//------------------------------------------+-----------------------------------

CTBzfile::~CTBzfile()
{
  CTB_Trace("~CTBzfile()");

  Close();
}

//------------------------------------------+-----------------------------------

bool CTBzfile::Open(const char *c_name, const char *c_mode)
{
  CTB_Trace("CTBzfile::Open(const char*, const char*)");
  CTBint i_lenname = c_name ? strlen(c_name) : 0;

  Close();
  mb_bad = false;
  mb_eof = false;

  if (i_lenname >= 4 && strcmp(c_name+i_lenname-3,".gz") == 0) {
    mi_type = gzfile;
    m_file  = gzopen(c_name,c_mode);
  } else {
    mi_type = cfile;
    m_file = fopen(c_name,c_mode);
  }
  
  if (m_file == 0) mi_type = closed;

  return m_file;
}

//------------------------------------------+-----------------------------------

void CTBzfile::Close()
{
  CTB_Trace("CTBzfile::Close()");

  switch(mi_type) {

  case cfile:
    fclose((FILE*) m_file);
    mi_type = closed;
    break;

  case gzfile:
    gzclose(m_file);
    mi_type = closed;
    break;
  }
  
  return;
}

//------------------------------------------+-----------------------------------

CTBint CTBzfile::Read(void* p_buf, CTBint i_size)
{
  CTB_Trace("CTBzfile::Read(void*, CTBint)");
  CTBint i_nbyte = 0;

  switch(mi_type) {

  case cfile:
    i_nbyte = fread(p_buf, 1, i_size, (FILE*) m_file);
    break;

  case gzfile:
    i_nbyte = gzread(m_file, p_buf, i_size);
    if (i_size > 0) {
      if (i_nbyte == 0) mb_eof = true;
      if (i_nbyte  < 0) mb_bad = true;
    }
    
    break;
  }
  
  return i_nbyte;
}

//------------------------------------------+-----------------------------------

CTBint CTBzfile::Write(const void* p_buf, CTBint i_size)
{
  CTB_Trace("CTBzfile::Write(void*, CTBint)");

  CTBint i_nbyte = 0;

  switch(mi_type) {

  case cfile:
    i_nbyte = fwrite(p_buf, 1, i_size, (FILE*) m_file);
    break;

  case gzfile:
    i_nbyte = gzwrite(m_file, (void*) p_buf, i_size);
    if (i_size > 0) {
      if (i_nbyte < 0) mb_bad = true;
    }
    break;
  }
  
  
  return i_nbyte;
}

//------------------------------------------+-----------------------------------

CTBint CTBzfile::ReadObject(void* p_buf, CTBint i_size)
{
  CTB_Trace("CTBzfile::ReadObject(void*, CTBint)");
  CTBint i_done = 0;
  CTBint i_nbyte;

  switch(mi_type) {

  case cfile:
    i_done = fread(p_buf, i_size, 1, (FILE*) m_file);
    break;

  case gzfile:
    i_nbyte = gzread(m_file, p_buf, i_size);
    if (i_size > 0) {
      if (i_nbyte == 0)           mb_eof = true;
      else if (i_nbyte != i_size) mb_bad = true;
      else i_done = 1;
    }
    
    break;
  }
  
  return i_done;
}

//------------------------------------------+-----------------------------------

CTBint CTBzfile::WriteObject(const void* p_buf, CTBint i_size)
{
  CTB_Trace("CTBzfile::WriteObject(void*, CTBint)");

  CTBint i_done = 0;
  CTBint i_nbyte;

  switch(mi_type) {

  case cfile:
    i_done = fwrite(p_buf, i_size, 1, (FILE*) m_file);
    break;

  case gzfile:
    i_nbyte = gzwrite(m_file, (void*) p_buf, i_size);
    if (i_size > 0) {
      if (i_nbyte != i_size) mb_bad = true;
      else i_done = 1;
    }
    break;
  }
  
  
  return i_done;
}

//------------------------------------------+-----------------------------------

bool CTBzfile::Good()
{
  CTB_Trace("CTBzfile::Good()");
  
  return !Fail();
}

//------------------------------------------+-----------------------------------

bool CTBzfile::Bad()
{
  CTB_Trace("CTBzfile::Bad()");
  
  switch(mi_type) {
  case cfile:  return ferror((FILE*) m_file);
  case gzfile: return mb_bad;
  }
  
  return false;
}

//------------------------------------------+-----------------------------------

bool CTBzfile::Fail()
{
  CTB_Trace("CTBzfile::Fail()");
  
  return !m_file || Bad() || Eof();
}

//------------------------------------------+-----------------------------------

bool CTBzfile::Eof()
{
  CTB_Trace("CTBzfile::Eof()");
  
  switch(mi_type) {
  case cfile:  return feof((FILE*) m_file);
  case gzfile: return mb_eof;
  }
  
  return true;
}

//------------------------------------------+-----------------------------------

void CTBzfile::Dump(int i_indent, ostream& os, const char* p_text) const
{
  CTBosFill bl(i_indent);

  os << bl << "-- CTBzfile ";
  if (p_text) os << p_text;
  os << " @ " << this << endl;

  os << bl << "    mi_type:    " << mi_type << "\n"; 
  os << bl << "    m_file:     " << m_file << "\n"; 
  os << bl << "    mb_bad:     " << mb_bad << "\n"; 
  os << bl << "    mb_eof:     " << mb_eof << "\n"; 

  return;
}

---