CalcTrackerHitResiduals.h

#ifndef CalcTrackerHitResiduals_h
#define CalcTrackerHitResiduals_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 TFile;
class TH1F;

#define MAXBUFFERSIZE 100

class CalcTrackerHitResiduals : public marlin::Processor {
  
public:
  
  virtual marlin::Processor*  newProcessor() { return new CalcTrackerHitResiduals ; }
  
  
  CalcTrackerHitResiduals() ;
  
  virtual void init() ;
  
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  virtual void processEvent( LCEvent * evt ) ; 
  
  
  virtual void check( LCEvent * evt ) ; 
  
  
  virtual void end() ;
  
protected:
  
//  void bookHistograms();
  
  void createHistogramBuffers();
  
  void fill_histo(const std::string& name, float value);
  
  void write_buffer_to_histo( std::map<std::string, std::list<float>*>::iterator it_buffer );
  
  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;
    
  
  TFile* _root_file;
  
  std::map<std::string, TH1F*> _histo_map;
  std::map<std::string, TH1F*>::iterator _histo_map_it;
  
  std::map<std::string, std::list<float>*> _histo_buffer_map;
  
  
} ;

#endif