raw_ofstream.cxx

Go to the documentation of this file.

#include "RawFile/raw_ofstream.h"
#include "RawFile/RawFileTools.h"
#include <cstdlib>
#include <unistd.h>
 
#define MAX_RAWFILE_SiZE 2000000000 // an approximate value, not exactly
 
// static data members
int             raw_ofstream::_nHandler     = 0;
raw_ofstream*   raw_ofstream::_instance     = 0;
pthread_mutex_t raw_ofstream::_pthread_lock = PTHREAD_MUTEX_INITIALIZER;
 
raw_ofstream* raw_ofstream::instance( const std::string& fname, int run ) {
  lock();
 
  if ( _instance == 0 ) { _instance = new raw_ofstream( fname, run ); }
 
  ++_nHandler;
 
  unlock();
 
  return _instance;
}
 
void raw_ofstream::release() {
  lock();
 
  if ( _nHandler > 0 && --_nHandler == 0 )
  {
    delete _instance;
    _instance = 0;
  }
 
  unlock();
}
 
raw_ofstream::raw_ofstream( const std::string& fname, int run )
    : m_nevt( 0 ), m_nfile( 0 ), m_fname( fname ) {
  m_runParametersRecord.setRunNumber( run );
  init_fstream();
}
 
raw_ofstream::~raw_ofstream() { this->close(); }
 
int raw_ofstream::write_event( const char* pbuf, int size ) {
  uint32_t fsize = tellp();
  if ( fsize >= MAX_RAWFILE_SiZE )
  {
    this->close();
    init_fstream();
  }
 
  m_dataSeparatorRecord.setDataBlockNumber( ++m_nevt );
  m_dataSeparatorRecord.setDataBlockSize( size );
 
  ( *this ) << m_dataSeparatorRecord;
  std::ofstream::write( pbuf, size );
 
  return m_nfile;
}
 
void raw_ofstream::close() {
  if ( is_open() )
  {
    m_fileEndRecord.setEventsInFile( m_nevt );
    m_nevt = 0;
 
    ( *this ) << m_fileEndRecord;
    std::ofstream::close();
 
    std::cout << "[RawFile] Finished writing file: " << real_fname() << std::endl;
  }
}
 
void raw_ofstream::init_fstream() {
  ++m_nfile;
 
  std::string fname = real_fname();
 
  if ( access( fname.c_str(), F_OK ) == 0 )
  {
    std::cerr << "[RawFile] Attempt to create an exist file: " << fname << std::endl;
    exit( 1 );
  }
 
  std::cout << "[RawFile] Creating a new file: " << real_fname() << std::endl;
  open( fname.c_str(), std::ios::binary );
 
  ( *this ) << m_fileStartRecord << m_fileNameStrings << m_runParametersRecord;
}
 
std::string raw_ofstream::real_fname() {
  std::string fname = m_fname;
 
  if ( m_nfile > 1 ) { fname += ".part" + RawFileTools::itoa( m_nfile ); }
 
  return fname;
}