LCWriter.h

Go to the documentation of this file.

#ifndef MT_LCWRITER_H
#define MT_LCWRITER_H 1

// -- std headers
#include <string>
#include <vector>
#include <fstream>
#include <memory>
#include <mutex>
#include <atomic>
#include <set>

// -- lcio headers
#include "MT/Types.h"
#include "EVENT/LCIO.h"

// need to make forward declaration to
// avoid lcio dependencies to include directly sio headers

namespace sio {
  class buffer ;
}

namespace SIO {
  class LCIORandomAccessMgr ;
}

namespace MT {

  class LCWriter {
  public:
    LCWriter() = default ;
    LCWriter(const LCWriter&) = delete ;
    LCWriter& operator=(const LCWriter&) = delete ;
    ~LCWriter() = default ;

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

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

    void setCompressionLevel( int level ) ;

    void writeRunHeader( EVENT::LCRunHeader *hdr ) ;

    void writeEvent( EVENT::LCEvent *evt ) ;

    void writeEvent( EVENT::LCEvent *evt, const std::set<std::string> &colsOnly ) ;

    void close() ;

  private:
    static std::string getSIOFileName( const std::string& filename ) ;

  private:
    std::ofstream                               _stream {} ;
    std::atomic<unsigned int>                   _maxBufferSize {2*1204*1024} ;
    std::atomic<int>                            _compressionLevel {-1} ;
    std::shared_ptr<SIO::LCIORandomAccessMgr>   _raMgr {nullptr} ;
    std::mutex                                  _mutex {} ;
  };

}

#endif