io_streamcount.h

Go to the documentation of this file.

/*******************************************************************************
 * \copyright   This file is part of the GADGET4 N-body/SPH code developed
 * \copyright   by Volker Springel. Copyright (C) 2014-2020 by Volker Springel
 * \copyright   (vspringel@mpa-garching.mpg.de) and all contributing authors.
 *******************************************************************************/
 
#ifndef IO_STREAMCOUNT_H
#define IO_STREAMCOUNT_H
 
#include <errno.h>
 
class io_streamcount
{
 public:
  long long byte_count = 0;
 
  size_t my_fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream)
  {
    size_t nwritten;
 
    if(!stream)
      return 0;
 
    if(size * nmemb > 0)
      {
        if((nwritten = fwrite(ptr, size, nmemb, stream)) != nmemb)
          {
            Terminate("I/O error (fwrite) has occured: %s\n", strerror(errno));
          }
      }
    else
      nwritten = 0;
 
    byte_count += size * nmemb;
 
    return nwritten;
  }
 
  size_t my_fread(void *ptr, size_t size, size_t nmemb, FILE *stream)
  {
    size_t nread;
 
    if(!stream)
      return 0;
 
    if(size * nmemb > 0)
      {
        if((nread = fread(ptr, size, nmemb, stream)) != nmemb)
          {
            if(feof(stream))
              {
                Terminate("I/O error (fread) has occured: end of file\n");
              }
            else
              Terminate("I/O error (fread) has occured: %s\n", strerror(errno));
          }
      }
    else
      nread = 0;
 
    byte_count += size * nmemb;
 
    return nread;
  }
 
  void reset_io_byte_count(void) { byte_count = 0; }
 
  long long get_io_byte_count(void) { return byte_count; }
};
 
#endif