SIOReader.h

Go to the documentation of this file.

#ifndef SIO_SIOREADER_H
#define SIO_SIOREADER_H 1

// -- std headers
#include <set>

// -- lcio headers
#include "IO/LCReader.h"
#include "IO/LCEventListener.h"
#include "IO/LCRunListener.h"
#include "MT/LCReaderListener.h"
#include "MT/LCReader.h"
// #include "IOIMPL/LCEventIOImpl.h"
// #include "IOIMPL/LCRunHeaderIOImpl.h"
// #include "SIO/SIOHandlerMgr.h"
// #include "SIO/LCIORandomAccessMgr.h"
// #include "EVENT/LCIO.h"
// #include "LCIOTypes.h"

// -- sio headers
// #include "sio/definitions.h"
// #include "sio/buffer.h"

namespace SIO {

  class SIOReader : public IO::LCReader, private MT::LCReaderListener {  // TODO restart from here !
  public:
    SIOReader( int lcReaderFlag=0 ) ;

    SIOReader(const SIOReader&) = delete ;
    SIOReader& operator=(const SIOReader&) = delete ;

    virtual ~SIOReader() ;

    void open(const std::vector<std::string>& filenames) override ;

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

    EVENT::LCRunHeader * readNextRunHeader() override ;

    EVENT::LCRunHeader * readNextRunHeader(int accessMode) override ;


    EVENT::LCEvent* readNextEvent() override ;


    EVENT::LCEvent* readNextEvent( int accessMode) override ;


    int getNumberOfEvents() override ;


    int getNumberOfRuns() override ;


    void getRuns(EVENT::IntVec & runs) override ;

    void getEvents(EVENT::IntVec & events) override ;


    void skipNEvents(int n) override ;



    void setReadCollectionNames(const std::vector<std::string>& colnames) override ;


    EVENT::LCRunHeader * readRunHeader(int runNumber ) override ;

    EVENT::LCRunHeader * readRunHeader(int runNumber, int accessMode ) override ;

    EVENT::LCEvent * readEvent(int runNumber, int evtNumber) override ;


    EVENT::LCEvent * readEvent(int runNumber, int evtNumber, int accessMode) override ;

    void close() override ;

    // interface for listeners

    void registerLCEventListener(IO::LCEventListener * ls) override ;

    void removeLCEventListener(IO::LCEventListener * ls) override ;

    void registerLCRunListener(IO::LCRunListener * ls) override ;

    void removeLCRunListener(IO::LCRunListener * ls) override ;

    void readStream() override ;

    void readStream(int maxRecord) override ;
    
  private:
    void processEvent( std::shared_ptr<EVENT::LCEvent> event ) override ;
    void processRunHeader( std::shared_ptr<EVENT::LCRunHeader> hdr ) override ;

  private:
    MT::LCReader                         _reader ;
    std::set<IO::LCRunListener*>         _runListeners {} ;
    std::set<IO::LCEventListener*>       _evtListeners {} ;
    EVENT::LCEvent* _currentEvent = nullptr ;
    EVENT::LCRunHeader* _currentRun = nullptr ;

  }; // class
} // namespace

#endif /* ifndef SIO_SIOREADER_H */