CMSSW_4_4_3_patch1/src/DQMOffline/JetMET/interface/JPTJetAnalyzer.h

Go to the documentation of this file.

#ifndef JPTJetAnalyzer_H
#define JPTJetAnalyzer_H

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "DQMOffline/JetMET/interface/JetAnalyzerBase.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DQMServices/Core/interface/MonitorElement.h"
#include "DataFormats/JetReco/interface/JPTJet.h"
#include "DataFormats/JetReco/interface/JPTJetCollection.h"
#include <memory>
// forward declare classes which do not need to be defined for interface
class DQMStore;
namespace reco {
  namespace helper {
    class JetIDHelper;
  }
}
namespace jptJetAnalysis {
  class TrackPropagatorToCalo;
  class StripSignalOverNoiseCalculator;
}
namespace jpt {
  class MatchedTracks;
}
class JetPlusTrackCorrector;
class JetCorrector;
class TrackingRecHit;
class SiStripRecHit2D;


class JPTJetAnalyzer : public JetAnalyzerBase {
 public:
  JPTJetAnalyzer(const edm::ParameterSet& config);
  
  virtual ~JPTJetAnalyzer();
  
  void beginJob(DQMStore * dbe);
  
  void analyze(const edm::Event& event, const edm::EventSetup& eventSetup, const reco::JPTJet& jptJet, double& pt1, double& pt2, double& pt3);
  void analyze(const edm::Event& event, const edm::EventSetup& eventSetup, const reco::JPTJetCollection& jptJets);
  
  virtual void endJob();
  
 private:
   
  // Helper classes
  struct HistogramConfig {
    bool enabled;
    unsigned int nBins;
    double min;
    double max;
    unsigned int nBinsY;
    double minY;
    double maxY;
    HistogramConfig();
    HistogramConfig(const unsigned int theNBins, const double theMin, const double theMax);
    HistogramConfig(const unsigned int theNBinsX, const double theMinX, const double theMaxX,
                    const unsigned int theNBinsY, const double theMinY, const double theMaxY);
  };
  struct TrackHistograms {
    MonitorElement* nTracksHisto;
    MonitorElement* ptHisto;
    MonitorElement* phiHisto;
    MonitorElement* etaHisto;
    MonitorElement* nHitsHisto;
    MonitorElement* nLayersHisto;
    MonitorElement* ptVsEtaHisto;
    MonitorElement* dzHisto;
    MonitorElement* dxyHisto;
    MonitorElement* trackDirectionJetDRHisto;
    MonitorElement* trackImpactPointJetDRHisto;
    TrackHistograms();
    TrackHistograms(MonitorElement* theNTracksHisto, MonitorElement* thePtHisto, MonitorElement* thePhiHisto, MonitorElement* theEtaHisto,
                    MonitorElement* theNHitsHisto, MonitorElement* theNLayersHisto, MonitorElement* thePtVsEtaHisto,
                    MonitorElement* dzHisto, MonitorElement* dxyHisto,
                    MonitorElement* theTrackDirectionJetDRHisto, MonitorElement* theTrackImpactPointJetDRHisto);
  };
  
  // Private methods
  void getConfigForHistogram(const std::string& configName, const edm::ParameterSet& psetContainingConfigPSet, std::ostringstream* pDebugStream = NULL);
  void getConfigForTrackHistograms(const std::string& tag, const edm::ParameterSet& psetContainingConfigPSet,std::ostringstream* pDebugStream = NULL);
  MonitorElement* bookHistogram(const std::string& name, const std::string& title, const std::string& xAxisTitle, DQMStore* dqm);
  MonitorElement* book2DHistogram(const std::string& name, const std::string& title, const std::string& xAxisTitle, const std::string& yAxisTitle, DQMStore* dqm);
  MonitorElement* bookProfile(const std::string& name, const std::string& title, const std::string& xAxisTitle, const std::string& yAxisTitle, DQMStore* dqm);
  void bookHistograms(DQMStore* dqm);
  void bookTrackHistograms(TrackHistograms* histos, const std::string& tag, const std::string& titleTag,
                           MonitorElement* trackDirectionJetDRHisto, MonitorElement* trackImpactPointJetDRHisto, DQMStore* dqm);
  void fillHistogram(MonitorElement* histogram, const double value);
  void fillHistogram(MonitorElement* histogram, const double valueX, const double valueY);
  void fillTrackHistograms(TrackHistograms& allTracksHistos, TrackHistograms& inCaloInVertexHistos,
                           TrackHistograms& inCaloOutVertexHistos, TrackHistograms& outCaloInVertexHistos,
                           const reco::TrackRefVector& inVertexInCalo,
                           const reco::TrackRefVector& outVertexInCalo,
                           const reco::TrackRefVector& inVertexOutCalo,
                           const reco::Jet& rawJet);
  void fillTrackHistograms(TrackHistograms& histos, const reco::TrackRefVector& tracks, const reco::Jet& rawJet);
  void fillSiStripSoNForTracks(const reco::TrackRefVector& tracks);
  void fillSiStripHitSoN(const TrackingRecHit& hit);
  void fillSiStripHitSoNForSingleHit(const SiStripRecHit2D& hit);
  static double findPtFractionInCone(const reco::TrackRefVector& inConeTracks, const reco::TrackRefVector& outOfConeTracks);
  
  static const char* messageLoggerCatregory;
  
  // Config
  const std::string histogramPath_;
  const bool verbose_;
  std::map<std::string,HistogramConfig> histogramConfig_;
  
  const bool writeDQMStore_;
  std::string dqmStoreFileName_;
  
  const int n90HitsMin_;
  const double fHPDMax_;
  const double resEMFMin_;
  const double correctedPtMin_;
  
  std::auto_ptr<jptJetAnalysis::TrackPropagatorToCalo> trackPropagator_;
  std::auto_ptr<jptJetAnalysis::StripSignalOverNoiseCalculator> sOverNCalculator_;
  std::auto_ptr<reco::helper::JetIDHelper> jetID_;
  
  // Histograms
  MonitorElement *JetE_, *JetEt_, *JetP_, *JetMass_, *JetPt_;
  MonitorElement *JetPt1_, *JetPt2_, *JetPt3_;
  MonitorElement *JetPx_, *JetPy_, *JetPz_;
  MonitorElement *JetEta_, *JetPhi_, *JetDeltaEta_, *JetDeltaPhi_, *JetPhiVsEta_;
  MonitorElement *JetN90Hits_, *JetfHPD_, *JetResEMF_, *JetfRBX_;
  MonitorElement *TrackSiStripHitStoNHisto_;
  MonitorElement *InCaloTrackDirectionJetDRHisto_, *OutCaloTrackDirectionJetDRHisto_;
  MonitorElement *InVertexTrackImpactPointJetDRHisto_, *OutVertexTrackImpactPointJetDRHisto_;
  MonitorElement *NTracksPerJetHisto_, *NTracksPerJetVsJetEtHisto_, *NTracksPerJetVsJetEtaHisto_;
  MonitorElement *PtFractionInConeHisto_, *PtFractionInConeVsJetRawEtHisto_, *PtFractionInConeVsJetEtaHisto_;
  MonitorElement *CorrFactorHisto_, *CorrFactorVsJetEtHisto_, *CorrFactorVsJetEtaHisto_;
  MonitorElement *ZSPCorrFactorHisto_, *ZSPCorrFactorVsJetEtHisto_, *ZSPCorrFactorVsJetEtaHisto_;
  MonitorElement *JPTCorrFactorHisto_, *JPTCorrFactorVsJetEtHisto_, *JPTCorrFactorVsJetEtaHisto_;
  TrackHistograms allPionHistograms_, inCaloInVertexPionHistograms_, inCaloOutVertexPionHistograms_, outCaloInVertexPionHistograms_;
  TrackHistograms allMuonHistograms_, inCaloInVertexMuonHistograms_, inCaloOutVertexMuonHistograms_, outCaloInVertexMuonHistograms_;
  TrackHistograms allElectronHistograms_, inCaloInVertexElectronHistograms_, inCaloOutVertexElectronHistograms_, outCaloInVertexElectronHistograms_;
  
  DQMStore* dqm_;
};

inline void JPTJetAnalyzer::fillHistogram(MonitorElement* histogram, const double value)
{
  if (histogram) histogram->Fill(value);
}

inline void JPTJetAnalyzer::fillHistogram(MonitorElement* histogram, const double valueX, const double valueY)
{
  if (histogram) histogram->Fill(valueX,valueY);
}

#endif