OverlayBX.h

#ifndef OverlayBX_h
#define OverlayBX_h 1

#include "marlin/Processor.h"
#include "marlin/EventModifier.h"
#include "lcio.h"
#include <string>
#include <vector>


#ifdef MARLIN_USE_AIDA
#include <vector>
#include <AIDA/AIDA.h>
typedef std::vector< AIDA::IHistogram1D* > Hist1DVec ;
#endif


#include "CLHEP/Vector/TwoVector.h"

namespace overlay{

  struct VXDLadder{
    double phi ;   // phi off ladder 
    CLHEP::Hep2Vector p0 ;  // 'left' end of ladder in r-phi
    CLHEP::Hep2Vector p1 ;  // 'right' end of ladder in r-phi
    CLHEP::Hep2Vector u  ;  // unit vector along ladder in r-phi
  };
  typedef std::vector< std::vector< VXDLadder > > VXDLadders ;

  struct VXDLayer{
    int nBX ;
    double width ;
    double ladderArea ;
    int nLadders ;
    VXDLayer(): nBX(0), width(0), ladderArea(0), nLadders(0) {}
  };
  typedef std::vector< VXDLayer >  VXDLayers ;


  // /** Helper struct for TPC parameters */
  // struct TPCParameters{ };


  class OverlayBX : public marlin::Processor, public marlin::EventModifier {
    
  public:
  
    virtual marlin::Processor*  newProcessor() { return new OverlayBX ; }
  
  
    OverlayBX() ;
  
    virtual const std::string & name() const { return Processor::name() ; }
  
    virtual void modifyEvent( LCEvent * evt ) ; 


    virtual void init() ;
  
    virtual void processRunHeader( LCRunHeader* run ) ;
  
    //  virtual void processEvent( LCEvent * evt ) ; 
  
  
    virtual void check( LCEvent * evt ) ; 
  
  
    virtual void end() ;
  
  
  protected:
  
    //  LCEvent*  readNextEvent() ;

    LCEvent*  readNextEvent(int bxNum) ;

    void init_geometry() ;
    int mergeVXDColsFromBX( LCCollection* vxdCol , LCCollection* vxdBGCol , int bxNum )  ;
    int mergeTPCColsFromBX( LCCollection* tpcCol , LCCollection* tpcBGCol , float zPosShift ) ;

    // ---- variables for processor parameters ----- 
    StringVec   _inputFileNames ;
    int         _eventsPerBX;
    float       _bxTime_s ;
    float       _tpcVdrift_mm_s ;
    int         _maxBXsTPC ;

    bool        _keepPairsTruthInfo ;
    bool        _phiRotateTPCHits ;

    FloatVec    _vxdLayerReadOutTimes ;

    StringVec   _tpcCollections ;
    //  std::string _tpcCollection ;

    std::string _mcpCollection ;
  
    std::string _vxdCollection ;
    StringVec   _mergeCollections ;
    int         _ranSeed  ;

    //---- class member variables ------
    typedef std::map<std::string, std::string> StrMap ;
    StrMap _tpcMap;
    StrMap _colMap;
    //  std::map<std::string, std::string> _colMap;

    std::vector< LCReader* > _lcReaders ;
    int _maxBXs ;
    int _nRun ;
    int _nEvt ;
    //  VXDLadders _vxdLadders ;
    VXDLayers  _vxdLayers ;

    int _lastBXNum  ;
    int _lastEvent  ; 
    int _currentRdr ;
  
#ifdef MARLIN_USE_AIDA
    Hist1DVec _hist1DVec ;
#endif

  } ;

} // namespace 


#endif