NoPileUpMEtUtilities.h

Go to the documentation of this file.

#ifndef RecoMET_METPUSubtraction_noPileUpMEtUtilities_h
#define RecoMET_METPUSubtraction_noPileUpMEtUtilities_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/METReco/interface/CommonMETData.h"
#include "DataFormats/METReco/interface/PUSubMETData.h"
#include "DataFormats/METReco/interface/PUSubMETDataFwd.h"

#include <vector>

class NoPileUpMEtUtilities
{

 public:

  enum {kOutsideJet=0,kWithin, kAll};
  enum {kChHSMET=0, kChPUMET, kNeutralUncMET, kHadronicHSMET, kHadronicPUMET};

  NoPileUpMEtUtilities();
  ~NoPileUpMEtUtilities();

 
  // general auxiliary functions
  void finalizeMEtData(CommonMETData&);
  
  //-------------------------------------------------------------------------------
  // auxiliary functions for jets
  reco::PUSubMETCandInfoCollection cleanJets(const reco::PUSubMETCandInfoCollection&, 
                         const std::vector<reco::Candidate::LorentzVector>&, 
                         double, bool);
  
  // auxiliary functions for PFCandidates
  reco::PUSubMETCandInfoCollection cleanPFCandidates(const reco::PUSubMETCandInfoCollection&,
                             const std::vector<reco::Candidate::LorentzVector>&,
                             double, bool);

 
  // common internal functions for jets and pfCandidates
  void computeAllSums( const reco::PUSubMETCandInfoCollection& jets,
               const reco::PUSubMETCandInfoCollection& pfCandidates);

  CommonMETData computeRecoil(int metType, double& sumAbsPx, double& sumAbsPy);
  //-------------------------------------------------------------------------------

 private:
 
  // common internal functions for jets and pfCandidates, to compute the different object sums
  CommonMETData computeCandidateSum(const reco::PUSubMETCandInfoCollection& cands,
                    bool neutralFracOnly, double& sumAbsPx, double& sumAbsPy);
  
  reco::PUSubMETCandInfoCollection selectCandidates(const reco::PUSubMETCandInfoCollection& cands, 
                            double minPt, double maxPt, int type, 
                            bool isCharged, int isWithinJet);


 private:


  double minPtDef_;
  double maxPtDef_;

  CommonMETData chHSPfcSum_;
  CommonMETData chPUPfcSum_;
  CommonMETData nUncPfcSum_;
  CommonMETData nHSJetSum_;
  CommonMETData nPUJetSum_;

  double chHSPfcSumAbsPx_;
  double chPUPfcSumAbsPx_;
  double nUncPfcSumAbsPx_;
  double nHSJetSumAbsPx_;
  double nPUJetSumAbsPx_;

  double chHSPfcSumAbsPy_;
  double chPUPfcSumAbsPy_;
  double nUncPfcSumAbsPy_;
  double nHSJetSumAbsPy_;
  double nPUJetSumAbsPy_;

};

#endif