GenSpecificAlgo Class Reference

#include <RecoMET/METAlgorithms/interface/GenSpecificAlgo.h>

Public Member Functions
reco::GenMET	addInfo (edm::Handle< edm::View< reco::Candidate > > particles, CommonMETData *met, double globalThreshold=0, bool onlyFiducial=false, bool applyFiducialThresholdForFractions=false, bool usePt=false)

Private Types
typedef math::XYZTLorentzVector	LorentzVector
typedef math::XYZPoint	Point

Private Member Functions
void	fillCommonMETData (CommonMETData *met, edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool usePt)
SpecificGenMETData	mkSpecificGenMETData (edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool applyFiducialThresholdForFractions, bool usePt)

Detailed Description

Description: Adds generator level HEPMC specific information to MET

Implementation: [Notes on implementation]

Definition at line 29 of file GenSpecificAlgo.h.

Member Typedef Documentation

typedef math::XYZTLorentzVector GenSpecificAlgo::LorentzVector

private

Definition at line 36 of file GenSpecificAlgo.h.

typedef math::XYZPoint GenSpecificAlgo::Point

private

Definition at line 37 of file GenSpecificAlgo.h.

Member Function Documentation

reco::GenMET GenSpecificAlgo::addInfo	(	edm::Handle< edm::View< reco::Candidate > >	particles,
		CommonMETData *	met,
		double	globalThreshold = 0,
		bool	onlyFiducial = false,
		bool	applyFiducialThresholdForFractions = false,
		bool	usePt = false
	)

Definition at line 17 of file GenSpecificAlgo.cc.

References fillCommonMETData(), HepMCValidationHelper::genMet(), CommonMETData::met, CommonMETData::mex, CommonMETData::mey, CommonMETData::mez, mkSpecificGenMETData(), p4, HadronAndPartonSelector_cfi::particles, CommonMETData::sumet, and badGlobalMuonTaggersAOD_cff::vtx.

Referenced by cms::GenMETProducer::produce().

{ 
  fillCommonMETData(met, particles, globalThreshold, onlyFiducial, usePt);

  SpecificGenMETData specific = mkSpecificGenMETData(particles, globalThreshold, onlyFiducial, applyFiducialThresholdForFractions, usePt);

  const LorentzVector p4( met->mex, met->mey, met->mez, met->met );
  const Point vtx( 0.0, 0.0, 0.0 );

  reco::GenMET genMet(specific, met->sumet, p4, vtx );
  return genMet;
}

void GenSpecificAlgo::fillCommonMETData ( CommonMETData *  met, edm::Handle< edm::View< reco::Candidate > > &  particles, double  globalThreshold, bool  onlyFiducial, bool  usePt  )

private

Definition at line 31 of file GenSpecificAlgo.cc.

References Abs(), begin, funct::cos(), MillePedeFileConverter_cfg::e, end, stringResolutionProvider_cfi::et, CaloMET_cfi::globalThreshold, CommonMETData::met, CommonMETData::mex, CommonMETData::mey, CommonMETData::mez, HadronAndPartonSelector_cfi::particles, phi, funct::sin(), mathSSE::sqrt(), CommonMETData::sumet, and theta().

Referenced by addInfo().

{
  double sum_et = 0.0;
  double sum_ex = 0.0;
  double sum_ey = 0.0;
  double sum_ez = 0.0;

  for( edm::View<reco::Candidate>::const_iterator iParticle = (particles.product())->begin(); iParticle != (particles.product())->end(); ++iParticle)
    {
    if( (onlyFiducial && TMath::Abs(iParticle->eta()) >= 5.0)) continue;

    if( iParticle->et() <= globalThreshold  ) continue;

    if(usePt)
      {
        double phi = iParticle->phi();
        double et = iParticle->pt();
        sum_ez += iParticle->pz();
        sum_et += et;
        sum_ex += et*cos(phi);
        sum_ey += et*sin(phi);
      }
    else
      {
        double phi = iParticle->phi();
        double theta = iParticle->theta();
        double e = iParticle->energy();
        double et = e*sin(theta);
        sum_ez += e*cos(theta);
        sum_et += et;
        sum_ex += et*cos(phi);
        sum_ey += et*sin(phi);
      }
    }
  
  met->mex   = -sum_ex;
  met->mey   = -sum_ey;
  met->mez   = -sum_ez;
  met->met   = sqrt( sum_ex*sum_ex + sum_ey*sum_ey );
  met->sumet = sum_et;
}

SpecificGenMETData GenSpecificAlgo::mkSpecificGenMETData ( edm::Handle< edm::View< reco::Candidate > > &  particles, double  globalThreshold, bool  onlyFiducial, bool  applyFiducialThresholdForFractions, bool  usePt  )

private

Definition at line 74 of file GenSpecificAlgo.cc.

References Abs(), begin, SpecificGenMETData::ChargedEMEtFraction, SpecificGenMETData::ChargedHadEtFraction, end, CaloMET_cfi::globalThreshold, SpecificGenMETData::InvisibleEtFraction, SpecificGenMETData::MuonEtFraction, SpecificGenMETData::NeutralEMEtFraction, SpecificGenMETData::NeutralHadEtFraction, HadronAndPartonSelector_cfi::particles, cosmictrackingParticleSelector_cfi::pdgId, EnergyCorrector::pt, and genMetCalo_cfi::usePt.

Referenced by addInfo().

{
  const static int neutralEMpdgId[] = { 22 /* photon */ };
  const static std::set<int> neutralEMpdgIdSet(neutralEMpdgId, neutralEMpdgId + sizeof(neutralEMpdgId)/sizeof(int));

  const static int chargedEMpdgId[] = { 11 /* e */ };
  const static std::set<int> chargedEMpdgIdSet(chargedEMpdgId, chargedEMpdgId + sizeof(chargedEMpdgId)/sizeof(int));

  const static int muonpdgId[] = { 13 /* muon */ };
  const static std::set<int> muonpdgIdSet(muonpdgId, muonpdgId + sizeof(muonpdgId)/sizeof(int));

  const static int neutralHADpdgId[] = {
    130  /* K_long          */,
    310  /* K_short         */,
    3122 /* Lambda          */,
    2112 /* n               */,
    3222 /* Neutral Cascade */
  };
  const static std::set<int> neutralHADpdgIdSet(neutralHADpdgId, neutralHADpdgId + sizeof(neutralHADpdgId)/sizeof(int));

  const static int chargedHADpdgId[] = {
    211  /* pi        */,
    321  /* K+/K-     */,
    2212 /* p         */,
    3312 /* Cascade - */,
    3112 /* Sigma -   */,
    3322 /* Sigma +   */,
    3334 /* Omega -   */
  };
  const static std::set<int> chargedHADpdgIdSet(chargedHADpdgId, chargedHADpdgId + sizeof(chargedHADpdgId)/sizeof(int));

  const static int invisiblepdgId[] = {
    12      /* e_nu            */,
    14      /* mu_nu           */,
    16      /* tau_nu          */,
    1000022 /* Neutral ~Chi_0  */,
    1000012 /* LH ~e_nu        */,
    1000014 /* LH ~mu_nu       */,
    1000016 /* LH ~tau_nu      */,
    2000012 /* RH ~e_nu        */,
    2000014 /* RH ~mu_nu       */,
    2000016 /* RH ~tau_nu      */,
    39      /* G               */,
    1000039 /* ~G              */,
    5100039 /* KK G            */,
    4000012 /* excited e_nu    */,
    4000014 /* excited mu_nu   */,
    4000016 /* excited tau_nu  */,
    9900012 /* Maj e_nu        */,
    9900014 /* Maj mu_nu       */,
    9900016 /* Maj tau_nu      */,
  };
  const static std::set<int> invisiblepdgIdSet(invisiblepdgId, invisiblepdgId + sizeof(invisiblepdgId)/sizeof(int));
  
  SpecificGenMETData specific;
  double Et_unclassified = 0.0;
  
  for(edm::View<reco::Candidate>::const_iterator iParticle = (particles.product())->begin(); iParticle != (particles.product())->end(); ++iParticle)
    {
      if(applyFiducialThresholdForFractions) if( onlyFiducial && (TMath::Abs(iParticle->eta()) >= 5.0) ) continue;
      if(applyFiducialThresholdForFractions) if( iParticle->et() <= globalThreshold ) continue;

      int pdgId = TMath::Abs( iParticle->pdgId() ) ;
      double pt = (usePt) ? iParticle->pt() : iParticle->et();
      if(neutralEMpdgIdSet.count(pdgId))       specific.NeutralEMEtFraction  += pt;
      else if(chargedEMpdgIdSet.count(pdgId))  specific.ChargedEMEtFraction  += pt;
      else if(muonpdgIdSet.count(pdgId))       specific.MuonEtFraction       += pt;
      else if(neutralHADpdgIdSet.count(pdgId)) specific.NeutralHadEtFraction += pt;
      else if(chargedHADpdgIdSet.count(pdgId)) specific.ChargedHadEtFraction += pt;
      else if(invisiblepdgIdSet.count(pdgId))  specific.InvisibleEtFraction  += pt;
      else Et_unclassified += pt;
    }
  
  double Et_Total = specific.NeutralEMEtFraction + specific.NeutralHadEtFraction + specific.ChargedEMEtFraction + 
    specific.ChargedHadEtFraction + specific.MuonEtFraction + specific.InvisibleEtFraction + Et_unclassified;
  
  if(Et_Total) 
    {
      specific.NeutralEMEtFraction /= Et_Total;
      specific.NeutralHadEtFraction /= Et_Total;
      specific.ChargedEMEtFraction /= Et_Total;
      specific.ChargedHadEtFraction /= Et_Total;
      specific.MuonEtFraction /= Et_Total;
      specific.InvisibleEtFraction /= Et_Total;
    }

  return specific;
}