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 "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

namespace edm {
  class Event;
}
namespace reco {
  class CaloJet;
  class GenJet;
}
class CaloTower;
class CaloRecHit;
class DetId;
class PCaloHit; 

class JetMatchingTools {
 public:
  JetMatchingTools (const edm::Event& fEvent);
  ~JetMatchingTools ();

  std::vector <const CaloTower*> getConstituents (const reco::CaloJet& fJet ) ;
  std::vector <const CaloRecHit*> getConstituents (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 CaloRecHit*> getCaloRecHits (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;
};

#endif