LCSplitWriter.h

Go to the documentation of this file.

#ifndef UTIL_LCSplitWriter_H
#define UTIL_LCSplitWriter_H 1

#include "IO/LCWriter.h"
#include "LCIOTypes.h"
#include <string>

namespace UTIL{
  
  class LCSplitWriter : public IO::LCWriter {
    
  public:

    LCSplitWriter( IO::LCWriter* wrt,  EVENT::long64 maxBytes ) :
      _wrt( wrt ), 
      _maxBytes(maxBytes),
      _baseFilename(""),
      _filename(""),
      _extension(""),
      _count(0) ,    
      _lastCount(4294967295UL) {    
    }    
   
    LCSplitWriter(const LCSplitWriter&) = delete ;
    LCSplitWriter& operator=(const LCSplitWriter&) = delete ;

    virtual ~LCSplitWriter() {;}
    
    
    //--------------------  the LCWriter interface: -------------------------------------


    virtual void open(const std::string & filename)  ;

    virtual void open(const std::string & filename, int writeMode)  ;

    virtual void writeRunHeader(const EVENT::LCRunHeader * hdr)  ;

    virtual void setCompressionLevel(int level) { 
      _wrt->setCompressionLevel(level) ; 
    }  


    virtual void writeEvent(const EVENT::LCEvent * evt)   ;

    virtual void close()   ;


    virtual void flush()   ;

    
    //---------------------------------------------------------------------------------------------

    EVENT::long64 fileSize() ;



  protected:
    LCSplitWriter( );

    EVENT::long64 file_size( const char *fname) ;

    const std::string& getFilename() ;
    
    void setBaseFilename( const std::string& filename )  ;
    
    std::string getCountingString(unsigned count) ;

    IO::LCWriter*  _wrt ;
    EVENT::long64 _maxBytes ;
    std::string _baseFilename ;
    std::string _filename ;
    std::string _extension ;
    unsigned _count ;
    unsigned _lastCount ;
    
  }; // class
  
} // namespace UTIL

#endif /* ifndef UTIL_LCSplitWriter_H */