METManager.h

Go to the documentation of this file.

#ifndef __RecoParticleFlow_PFRootEvent_METManager__
#define __RecoParticleFlow_PFRootEvent_METManager__

#include "RecoParticleFlow/Benchmark/interface/GenericBenchmark.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"

#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/METReco/interface/GenMETCollection.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"

class METManager {

 public:

  METManager(std::string outmetfilename);
  
  void addGenBenchmark(std::string GenBenchmarkName);

  void FillHisto(std::string);

  void write();

  void setMET1(const reco::MET*);
  void setMET2(const reco::MET*);

  void setMET1(const reco::GenParticleCollection*);
  void setMET2(const reco::GenParticleCollection*);

  void setMET1(const reco::PFCandidateCollection&);
  void setMET2(const reco::PFCandidateCollection&);

  void coutTailEvents(const int entry, const double DeltaMETcut,
              const double DeltaPhicut, const double MET1cut) const;

  void propagateJECtoMET1(const std::vector<reco::CaloJet>& caloJets,
            const std::vector<reco::CaloJet>& corr_caloJets);
  void propagateJECtoMET2(const std::vector<reco::CaloJet>& caloJets,
            const std::vector<reco::CaloJet>& corr_caloJets);

  void SetIgnoreParticlesIDs(const std::vector<unsigned int>*);
  void SetSpecificIdCut(const std::vector<unsigned int>*, const std::vector<double>*);

  // Addition to have recomputed METs : by S. Dutta 
  reco::MET getMET1(){return MET1_;}
  reco::MET getMET2(){return MET2_;}
  //-----------------------------------------------

 private:
 
  reco::MET computeGenMET( const reco::GenParticleCollection* ) const;
  reco::MET recomputePFMET( const reco::PFCandidateCollection&) const;
  reco::MET propagateJEC(const reco::MET&, const std::vector<reco::CaloJet>& caloJets,
             const std::vector<reco::CaloJet>& corr_caloJets) const;

  reco::MET MET1_;
  reco::MET MET2_;
  std::map<std::string,GenericBenchmark> GenBenchmarkMap_;
  std::string outmetfilename_;
  TFile* outfile_;
  std::vector<unsigned int> vIgnoreParticlesIDs_;
  std::vector<unsigned int> trueMetSpecificIdCut_;
  std::vector<double> trueMetSpecificEtaCut_;

};

#endif