SetTrackerHitExtensions.h

#ifndef SetTrackerHitExtensions_h
#define SetTrackerHitExtensions_h 1

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

#include <EVENT/TrackerHit.h>
#include <EVENT/SimTrackerHit.h>
#include <IMPL/LCCollectionVec.h>

#include <UTIL/BitField64.h>
#include <UTIL/ILDConf.h>

namespace EVENT{
  class MCParticle ;
  class Track ;
}

namespace IMPL {
  class TrackImpl ;
}

namespace UTIL{
  class LCRelationNavigator ;
}


namespace MarlinTrk{
  class IMarlinTrkSystem ;
}

class SetTrackerHitExtensions : public marlin::Processor {
  
public:
  
  virtual marlin::Processor*  newProcessor() { return new SetTrackerHitExtensions ; }
  
  
  SetTrackerHitExtensions() ;
  
  virtual void init() ;
  
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  virtual void processEvent( LCEvent * evt ) ; 
  
  
  virtual void check( LCEvent * evt ) ; 
  
  
  virtual void end() ;
  
protected:
  
  
  const LCObjectVec* getSimHits( TrackerHit* trkhit, const FloatVec* weights = NULL);
  
  UTIL::BitField64* _encoder;
  int getDetectorID(TrackerHit* hit) { _encoder->setValue(hit->getCellID0()); return (*_encoder)[lcio::ILDCellID0::subdet]; }
  int getSideID(TrackerHit* hit)     { _encoder->setValue(hit->getCellID0()); return (*_encoder)[lcio::ILDCellID0::side]; };
  int getLayerID(TrackerHit* hit)    { _encoder->setValue(hit->getCellID0()); return (*_encoder)[lcio::ILDCellID0::layer]; };
  int getModuleID(TrackerHit* hit)   { _encoder->setValue(hit->getCellID0()); return (*_encoder)[lcio::ILDCellID0::module]; };
  int getSensorID(TrackerHit* hit)   { _encoder->setValue(hit->getCellID0()); return (*_encoder)[lcio::ILDCellID0::sensor]; };

  
  LCCollection* GetCollection(  LCEvent * evt, std::string colName ) ;
  
  LCRelationNavigator* GetRelations( LCEvent * evt, std::string RelName ) ;
  
  void SetupInputCollections( LCEvent * evt ) ;
  
  
  
  std::vector< std::string > _colNamesTrackerHits;
 
  std::vector< std::string > _colNamesTrackerHitRelations;
  
  
//   int _nEventPrintout ;
  int _n_run ;
  int _n_evt ;
  int _current_evt_number ;
  
  
  std::vector< LCCollection* > _colTrackerHits;
  std::vector< LCRelationNavigator* > _navTrackerHitRel;
    
  
} ;

#endif