TrackDetMatchInfo.h

Go to the documentation of this file.

#ifndef TrackAssociator_TrackDetMatchInfo_h
#define TrackAssociator_TrackDetMatchInfo_h

#include "DataFormats/CaloTowers/interface/CaloTower.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/HcalRecHit/interface/HBHERecHit.h"
#include "DataFormats/HcalRecHit/interface/HORecHit.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "TrackingTools/TrackAssociator/interface/TAMuonChamberMatch.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
class TrackDetMatchInfo {
 public:
   enum EnergyType { EcalRecHits, HcalRecHits, HORecHits, TowerTotal, TowerEcal, TowerHcal, TowerHO };
   
   TrackDetMatchInfo();
   
   double crossedEnergy( EnergyType );
   
   double coneEnergy( double dR, EnergyType );
   
   double nXnEnergy(const DetId&, EnergyType, int gridSize = 1);
   
   double nXnEnergy(EnergyType, int gridSize = 1);

   DetId findMaxDeposition( EnergyType );
   DetId findMaxDeposition( EnergyType, int gridSize );
   DetId findMaxDeposition( const DetId&, EnergyType, int gridSize );
   
   math::XYZPoint trkGlobPosAtEcal;
   math::XYZPoint trkGlobPosAtHcal;
   math::XYZPoint trkGlobPosAtHO;
   
  GlobalVector trkMomAtEcal;
  GlobalVector trkMomAtHcal;
  GlobalVector trkMomAtHO;
   
   bool isGoodEcal;
   bool isGoodHcal;
   bool isGoodCalo;
   bool isGoodHO;
   bool isGoodMuon;
   
   std::vector<const EcalRecHit*> ecalRecHits;
   std::vector<const HBHERecHit*> hcalRecHits;
   std::vector<const HORecHit*>   hoRecHits;
   std::vector<const CaloTower*>  towers;

   std::vector<const EcalRecHit*> crossedEcalRecHits;
   std::vector<const HBHERecHit*> crossedHcalRecHits;
   std::vector<const HORecHit*>   crossedHORecHits;
   std::vector<const CaloTower*>  crossedTowers;

   std::vector<DetId>      crossedEcalIds;
   std::vector<DetId>      crossedHcalIds;
   std::vector<DetId>      crossedHOIds;
   std::vector<DetId>      crossedTowerIds;
   std::vector<DetId>      crossedPreshowerIds;
   
   std::vector<TAMuonChamberMatch> chambers;

   FreeTrajectoryState stateAtIP;
   
   const SimTrack* simTrack;
   double ecalTrueEnergy;
   double hcalTrueEnergy;
   double hcalTrueEnergyCorrected;
   
   reco::TrackRef trackRef_;
   SimTrackRef simTrackRef_;
   
   double ecalCrossedEnergy();
   double ecalConeEnergy();
   double hcalCrossedEnergy();
   double hcalConeEnergy();
   double hoCrossedEnergy();
   double hoConeEnergy();

   double ecalTowerEnergy() { return crossedEnergy(TowerEcal); }
   double ecalTowerConeEnergy() { return coneEnergy(999,TowerEcal); }
   double hcalTowerEnergy() { return crossedEnergy(TowerHcal); }
   double hcalTowerConeEnergy() { return coneEnergy(999, TowerHcal); }
   double hoTowerEnergy() { return crossedEnergy(TowerHO); }
   double hoTowerConeEnergy() { return coneEnergy(999, TowerHO); }

   double ecalEnergy() { return ecalCrossedEnergy(); }
   double hcalEnergy() { return hcalCrossedEnergy(); }
   double hoEnergy() { return hoCrossedEnergy(); }
   
   int numberOfSegments() const;
   int numberOfSegmentsInStation(int station) const;
   int numberOfSegmentsInStation(int station, int detector) const;
   int numberOfSegmentsInDetector(int detector) const;
   
   void setCaloGeometry( edm::ESHandle<CaloGeometry> geometry ) { caloGeometry = geometry; }
   GlobalPoint getPosition( const DetId& );
   std::string dumpGeometry( const DetId& );
 private:
   bool insideCone(const DetId&, const double);
   edm::ESHandle<CaloGeometry> caloGeometry;
};
#endif