OverlayTiming.h

#ifndef OverlayTiming_h
#define OverlayTiming_h 1

#include "marlin/Processor.h"
#include "marlin/EventModifier.h"

#include "lcio.h"

#include <cmath>

namespace overlay {

  class OverlayTiming : public marlin::Processor, public marlin::EventModifier
  {
  public:
    virtual marlin::Processor*  newProcessor();

    OverlayTiming();

    virtual void init();

    virtual const std::string &name() const;

    virtual void processRunHeader(EVENT::LCRunHeader *run) ;

    virtual void modifyEvent(EVENT::LCEvent *evt);

    virtual void check(EVENT::LCEvent *evt);

    virtual void end();

  protected:
    float time_of_flight(float x, float y, float z) const;

    void define_time_windows(const std::string &Collection_name);

    void crop_collection(EVENT::LCCollection *collection);

    void merge_collections(EVENT::LCCollection *source_collection, EVENT::LCCollection *dest_collection, float time_offset);

    long long cellID2long(int id0, int id1) const;

    float _T_diff;
    int _nBunchTrain;

    unsigned int _nRun ;
    unsigned int _nEvt ;
    StringVec _inputFileNames ; 

    int _BX_phys;
    float _NOverlay;
    unsigned int overlay_file_list;
    float _BeamCal_int,_ETD_int, _EcalBarrel_int, _EcalBarrelPreShower_int, _EcalEndcap_int, _EcalEndcapPreShower_int, _EcalEndcapRing_int, _EcalEndcapRingPreShower_int;
    float _FTD_int, _HcalBarrelReg_int, _HcalEndCapRings_int, _HcalEndCaps_int, _LHcal_int, _LumiCal_int, _MuonBarrel_int, _MuonEndCap_int, _SET_int, _SIT_int, _VXD_int;
    float _EcalPlug_int, _VXDB_int, _VXDE_int, _ITB_int, _ITE_int, _OTB_int, _OTE_int;
    float _TPC_int, _TPCSpacePoint_int;

    IO::LCReader* overlay_Eventfile_reader;

    float this_start;
    float this_stop;
    int _ranSeed;

    std::string _mcParticleCollectionName;
    std::string _mcPhysicsParticleCollectionName;
    std::string currentDest;
    bool TPC_hits;

    float _tpcVdrift_mm_ns ;
    bool _randomBX, _Poisson;

    typedef std::map<long long, EVENT::SimCalorimeterHit*> DestMap;
    typedef std::map<std::string, DestMap> CollDestMap;
    CollDestMap collDestMap;
  };

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

  marlin::Processor *OverlayTiming::newProcessor()
  {
    return new OverlayTiming;
  }

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

  const std::string &OverlayTiming::name() const
  {
    return marlin::Processor::name();
  }

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

  inline float OverlayTiming::time_of_flight(float x, float y, float z) const
  {
    //returns the time of flight to the radius in ns
    // mm/m/s = 10^{-3}s = 10^6 ns d.h. 299 mm/ns
    return std::sqrt((x * x) + (y * y) + (z * z))/299.792458;
  }

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

  inline long long OverlayTiming::cellID2long(int id0, int id1) const
  {
    return ((long long) id0 << 32) | id1;
  }

} // namespace

#endif