JetMatchingTools.h

Go to the documentation of this file.

#ifndef JetMatchingTools_h
#define JetMatchingTools_h

#include <vector>

#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

namespace edm {
  class Event;
}
namespace reco {
  class CaloJet;
  class GenJet;
}

class CaloTower;
class CaloRecHit;
class DetId;
class PCaloHit; 

class JetMatchingTools {
 public:
  struct JetConstituent {
    DetId id;
    double energy;

    JetConstituent() {}
    ~JetConstituent() {}
    JetConstituent(const JetConstituent &j) : id(j.id), energy(j.energy) {}
    JetConstituent(const EcalRecHit &ehit) : id(ehit.detid()), energy(ehit.energy()) {}
    JetConstituent(const CaloRecHit &ehit) : id(ehit.detid()), energy(ehit.energy()) {}
  };

  JetMatchingTools (const edm::Event& fEvent, edm::ConsumesCollector&& iC );
  ~JetMatchingTools ();

  std::vector <const CaloTower*> getConstituents (const reco::CaloJet& fJet ) ;
  std::vector <JetConstituent> getConstituentHits(const CaloTower& fTower);
  std::vector <DetId> getConstituentIds (const CaloTower& fTower) ;
  std::vector <const PCaloHit*> getPCaloHits (DetId fId) ;
  int getTrackId (const PCaloHit& fHit) ;
  const SimTrack* getTrack (unsigned fSimTrackId);
  int generatorId (unsigned fSimTrackId) ;
  const reco::GenParticle* getGenParticle (int fGeneratorId);
  std::vector <const reco::GenParticle*> getGenParticles (const reco::CaloJet& fJet, bool fVerbose = true);
  std::vector <const reco::GenParticle*> getGenParticles (const reco::GenJet& fJet);

  // reverse propagation
  std::vector <const PCaloHit*> getPCaloHits (int fGeneratorId);
  std::vector <const CaloTower*> getCaloTowers (int fGeneratorId);

  double lostEnergyFraction (const reco::CaloJet& fJet );

  double overlapEnergyFraction (const std::vector <const reco::GenParticle*>& fObject, 
                const std::vector <const reco::GenParticle*>& fReference) const;


  const EBRecHitCollection* getEBRecHitCollection ();
  const EERecHitCollection* getEERecHitCollection ();
  const HBHERecHitCollection* getHBHERecHitCollection ();
  const HORecHitCollection* getHORecHitCollection ();
  const HFRecHitCollection* getHFRecHitCollection ();
  const edm::PCaloHitContainer* getEBSimHitCollection ();
  const edm::PCaloHitContainer* getEESimHitCollection ();
  const edm::PCaloHitContainer* getHcalSimHitCollection ();
  const edm::SimTrackContainer* getSimTrackCollection ();
  const edm::SimVertexContainer* getSimVertexCollection ();
  const reco::CandidateCollection* getGenParticlesCollection ();


  
 private:
  const edm::Event* mEvent;
  const EBRecHitCollection* mEBRecHitCollection;
  const EERecHitCollection* mEERecHitCollection;
  const HBHERecHitCollection* mHBHERecHitCollection;
  const HORecHitCollection* mHORecHitCollection;
  const HFRecHitCollection* mHFRecHitCollection;
  const edm::PCaloHitContainer* mEBSimHitCollection;
  const edm::PCaloHitContainer* mEESimHitCollection;
  const edm::PCaloHitContainer* mHcalSimHitCollection;
  const edm::SimTrackContainer* mSimTrackCollection;
  const edm::SimVertexContainer* mSimVertexCollection;
  const reco::CandidateCollection* mGenParticleCollection;

  edm::EDGetTokenT<EBRecHitCollection> input_ebrechits_token_;
  edm::EDGetTokenT<EERecHitCollection> input_eerechits_token_;
  edm::EDGetTokenT<HBHERecHitCollection> input_hbherechits_token_;
  edm::EDGetTokenT<HORecHitCollection> input_horechits_token_;
  edm::EDGetTokenT<HFRecHitCollection> input_hfrechits_token_;
  edm::EDGetTokenT<edm::PCaloHitContainer> input_pcalohits_eecal_token_;
  edm::EDGetTokenT<edm::PCaloHitContainer> input_pcalohits_ebcal_token_;
  edm::EDGetTokenT<edm::PCaloHitContainer> input_pcalohits_hcal_token_;
  edm::EDGetTokenT<edm::SimTrackContainer> input_simtrack_token_;
  edm::EDGetTokenT<edm::SimVertexContainer> input_simvertex_token_;
  edm::EDGetTokenT<reco::CandidateCollection> input_cands_token_;

};

#endif