TrackAssociator.h

Go to the documentation of this file.

#ifndef HTrackAssociator_HTrackAssociator_h
#define HTrackAssociator_HTrackAssociator_h 1

// -*- C++ -*-
//
// Package:    HTrackAssociator
// Class:      HTrackAssociator
//
/*

 Description: <one line class summary>

 Implementation:
     <Notes on implementation>
*/
//
// Original Author:  Dmytro Kovalskyi
// Modified for ECAL+HCAL by:  Michal Szleper
//

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Common/interface/OrphanHandle.h"

#include "DataFormats/TrackReco/interface/TrackBase.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"

#include "Calibration/Tools/interface/CaloDetIdAssociator.h"
#include "Calibration/Tools/interface/EcalDetIdAssociator.h"
#include "Calibration/Tools/interface/HcalDetIdAssociator.h"
#include "Calibration/Tools/interface/TrackDetMatchInfo.h"

#include "DataFormats/CaloTowers/interface/CaloTower.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/HcalRecHit/interface/HBHERecHit.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/Point3D.h"

#include "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"

class HTrackAssociator {
public:
  HTrackAssociator();
  ~HTrackAssociator();

  class HAssociatorParameters {
  public:
    HAssociatorParameters() {
      // default parameters
      // define match cones, dR=sqrt(dEta^2+dPhi^2)
      dREcal = 0.03;
      dRHcal = 0.07;
      dRCalo = 0.07;

      idREcal = 1;
      idRHcal = 4;
      idRCalo = 4;

      // match all sub-detectors by default
      useEcal = true;
      useHcal = true;
      useCalo = true;
    }
    double dREcal;
    double dRHcal;
    double dRCalo;
    int idREcal;
    int idRHcal;
    int idRCalo;

    bool useEcal;
    bool useHcal;
    bool useCalo;
  };

  HTrackDetMatchInfo associate(const edm::Event&,
                               const edm::EventSetup&,
                               const FreeTrajectoryState&,
                               const HAssociatorParameters&);

  std::vector<EcalRecHit> associateEcal(const edm::Event&,
                                        const edm::EventSetup&,
                                        const FreeTrajectoryState&,
                                        const double dR = -1);

  double getEcalEnergy(const edm::Event&, const edm::EventSetup&, const FreeTrajectoryState&, const double dR = -1);

  std::vector<CaloTower> associateHcal(const edm::Event&,
                                       const edm::EventSetup&,
                                       const FreeTrajectoryState&,
                                       const double dR = -1);

  double getHcalEnergy(const edm::Event&, const edm::EventSetup&, const FreeTrajectoryState&, const double dR = -1);
  void setPropagator(Propagator*);

  void useDefaultPropagator();

  void addDataLabels(const std::string className,
                     const std::string moduleLabel,
                     const std::string productInstanceLabel = "");

  FreeTrajectoryState getFreeTrajectoryState(const edm::EventSetup&, const reco::Track&);
  FreeTrajectoryState getFreeTrajectoryState(const edm::EventSetup&, const SimTrack&, const SimVertex&);

private:
  void fillEcal(const edm::Event&,
                const edm::EventSetup&,
                HTrackDetMatchInfo&,
                const FreeTrajectoryState&,
                const int,
                const double);

  void fillHcal(const edm::Event&,
                const edm::EventSetup&,
                HTrackDetMatchInfo&,
                const FreeTrajectoryState&,
                const int,
                const double);

  void fillHcalTowers(const edm::Event&,
                      const edm::EventSetup&,
                      HTrackDetMatchInfo&,
                      const FreeTrajectoryState&,
                      const int,
                      const double);

  void fillCaloTowers(const edm::Event&,
                      const edm::EventSetup&,
                      HTrackDetMatchInfo&,
                      const FreeTrajectoryState&,
                      const int,
                      const double);

  void init(const edm::EventSetup&);

  math::XYZPoint getPoint(const GlobalPoint& point) { return math::XYZPoint(point.x(), point.y(), point.z()); }

  math::XYZPoint getPoint(const LocalPoint& point) { return math::XYZPoint(point.x(), point.y(), point.z()); }

  math::XYZVector getVector(const GlobalVector& vec) { return math::XYZVector(vec.x(), vec.y(), vec.z()); }

  math::XYZVector getVector(const LocalVector& vec) { return math::XYZVector(vec.x(), vec.y(), vec.z()); }

  Propagator* ivProp_;
  Propagator* defProp_;
  bool useDefaultPropagator_;
  int debug_;
  std::vector<std::vector<std::set<uint32_t> > >* caloTowerMap_;

  HEcalDetIdAssociator ecalDetIdAssociator_;
  HHcalDetIdAssociator hcalDetIdAssociator_;
  HCaloDetIdAssociator caloDetIdAssociator_;

  edm::ESHandle<CaloGeometry> theCaloGeometry_;

  std::vector<std::string> EBRecHitCollectionLabels;
  std::vector<std::string> EERecHitCollectionLabels;
  std::vector<std::string> HBHERecHitCollectionLabels;
  std::vector<std::string> CaloTowerCollectionLabels;
};
#endif