SIOReader.h

#ifndef SIO_SIOREADER_H
#define SIO_SIOREADER_H 1

#include <string>
#include <set>
#include <map>
#include "IO/LCReader.h"
#include "IO/LCEventListener.h"
#include "IO/LCRunListener.h"

#include "IOIMPL/LCEventIOImpl.h"
#include "IOIMPL/LCRunHeaderIOImpl.h"

#include "LCIORandomAccessMgr.h"
#include "LCIOTypes.h"
#include "EVENT/LCIO.h"


class SIO_record ;
class SIO_stream ;    


namespace SIO {

class SIORunHeaderHandler ;
class SIOEventHandler ;
  
  class SIOReader : public IO::LCReader {
    
    //    typedef std::map< EVENT::long64 , EVENT::long64 > EventMap ;
    //    typedef RunEventMap EventMap ;
  public:
    
    SIOReader( int lcReaderFlag=0 ) ;
    
    // Destructor

    virtual ~SIOReader() ;
    

    virtual void open(const std::vector<std::string>& filenames) 
      throw (IO::IOException, std::exception) ;


    virtual void open(const std::string & filename) throw (IO::IOException, std::exception) ;
    
    virtual EVENT::LCRunHeader * readNextRunHeader() throw (IO::IOException, std::exception) ;

    virtual EVENT::LCRunHeader * readNextRunHeader(int accessMode) throw (IO::IOException, std::exception) ;


    virtual EVENT::LCEvent* readNextEvent() throw (IO::IOException, std::exception) ;
    

    virtual EVENT::LCEvent* readNextEvent( int accessMode) throw (IO::IOException, std::exception) ;
    

   virtual int getNumberOfEvents() throw (IO::IOException, std::exception ) ;


    virtual int getNumberOfRuns() throw (IO::IOException, std::exception ) ;


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

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


    virtual void skipNEvents(int n)   throw (IO::IOException, std::exception )  ;



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


    virtual EVENT::LCRunHeader * readRunHeader(int runNumber ) 
      throw (IO::IOException , std::exception) ;

    virtual EVENT::LCRunHeader * readRunHeader(int runNumber, int accessMode ) 
      throw (IO::IOException , std::exception) ;

    virtual EVENT::LCEvent * readEvent(int runNumber, int evtNumber) 
      throw (IO::IOException, std::exception ) ;


    virtual EVENT::LCEvent * readEvent(int runNumber, int evtNumber, int accessMode) 
      throw (IO::IOException, std::exception ) ;
    
    virtual void close() throw (IO::IOException, std::exception) ;
    
    // interface for listeners
 
    virtual void registerLCEventListener(IO::LCEventListener * ls) ;

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

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

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

    virtual void readStream() throw (IO::IOException, std::exception) ;

    virtual void readStream(int maxRecord) throw (IO::IOException, std::exception) ;




  protected:

    void setUpHandlers() ;
    void readRecord() throw (IO::IOException , IO::EndOfDataException , std::exception) ;


    void postProcessEvent() ;

    void getEventMap() ;

    void recreateEventMap() ;

  protected:
    
    // we need an SIO record for every type
//     SIO_record *_evtRecord ;
//     SIO_record *_hdrRecord ;
//     SIO_record *_runRecord ;
    SIO_record *_dummyRecord ;  // used for reading arbitrary records
    SIO_stream *_stream ;

    SIORunHeaderHandler* _runHandler ;
    SIOEventHandler* _evtHandler ;

  private:
    
    IOIMPL::LCEventIOImpl *_defaultEvt ; // used to add collections when reading 
    IOIMPL::LCEventIOImpl *_evt ;
    IOIMPL::LCRunHeaderIOImpl *_run ;

    std::set<IO::LCRunListener*> _runListeners ;
    std::set<IO::LCEventListener*> _evtListeners ;
    
    std::vector<std::string> _myFilenames ;
    unsigned int _currentFileIndex ;

    //    EventMap _evtMap ;
    bool _readEventMap ;
    
    //    RunEventMap _reMap ;
    LCIORandomAccessMgr _raMgr ;

  }; // class
} // namespace

#endif /* ifndef SIO_SIOREADER_H */