CaloParticleSelector Class Reference

#include <CaloParticleSelector.h>

Public Member Functions
	CaloParticleSelector ()
	CaloParticleSelector (double ptMin, double ptMax, double minRapidity, double maxRapidity, double lip, double tip, int minHit, unsigned int maxSimClusters, bool signalOnly, bool intimeOnly, bool chargedOnly, bool stableOnly, bool notConvertedOnly, const std::vector< int > &pdgId=std::vector< int >(), double minPhi=-3.2, double maxPhi=3.2)
bool	operator() (const CaloParticle &cp, std::vector< SimVertex > const &simVertices) const

Private Attributes
bool	chargedOnly_
bool	intimeOnly_
double	lip_
float	maxRapidity_
unsigned int	maxSimClusters_
float	meanPhi_
int	minHit_
float	minRapidity_
bool	notConvertedOnly_
std::vector< int >	pdgId_
double	ptMax2_
double	ptMin2_
float	rangePhi_
bool	signalOnly_
bool	stableOnly_
double	tip2_

Detailed Description

Definition at line 10 of file CaloParticleSelector.h.

Constructor & Destructor Documentation

CaloParticleSelector::CaloParticleSelector ( )

inline

Definition at line 12 of file CaloParticleSelector.h.

{}

CaloParticleSelector::CaloParticleSelector ( double  ptMin, double  ptMax, double  minRapidity, double  maxRapidity, double  lip, double  tip, int  minHit, unsigned int  maxSimClusters, bool  signalOnly, bool  intimeOnly, bool  chargedOnly, bool  stableOnly, bool  notConvertedOnly, const std::vector< int > &  pdgId = std::vector<int>(), double  minPhi = -3.2, double  maxPhi = 3.2  )

inline

Definition at line 13 of file CaloParticleSelector.h.

References Exception, M_PI, HLT_FULL_cff::maxPhi, and HLT_FULL_cff::minPhi.

      : ptMin2_(ptMin * ptMin),
        ptMax2_(ptMax * ptMax),
        minRapidity_(minRapidity),
        maxRapidity_(maxRapidity),
        lip_(lip),
        tip2_(tip * tip),
        meanPhi_((minPhi + maxPhi) / 2.),
        rangePhi_((maxPhi - minPhi) / 2.),
        minHit_(minHit),
        maxSimClusters_(maxSimClusters),
        signalOnly_(signalOnly),
        intimeOnly_(intimeOnly),
        chargedOnly_(chargedOnly),
        stableOnly_(stableOnly),
        notConvertedOnly_(notConvertedOnly),
        pdgId_(pdgId) {
    if (minPhi >= maxPhi) {
      throw cms::Exception("Configuration")
          << "CaloParticleSelector: minPhi (" << minPhi << ") must be smaller than maxPhi (" << maxPhi
          << "). The range is constructed from minPhi to maxPhi around their average.";
    }
    if (minPhi >= M_PI) {
      throw cms::Exception("Configuration")
          << "CaloParticleSelector: minPhi (" << minPhi
          << ") must be smaller than PI. The range is constructed from minPhi to maxPhi around their average.";
    }
    if (maxPhi <= -M_PI) {
      throw cms::Exception("Configuration")
          << "CaloParticleSelector: maxPhi (" << maxPhi
          << ") must be larger than -PI. The range is constructed from minPhi to maxPhi around their average.";
    }
  }

Member Function Documentation

bool CaloParticleSelector::operator() ( const CaloParticle &  cp, std::vector< SimVertex > const &  simVertices  ) const

inline

Definition at line 64 of file CaloParticleSelector.h.

References funct::abs(), EncodedEventId::bunchCrossing(), CaloParticle::charge(), chargedOnly_, srCondWrite_cfg::deltaPhi, PVValHelper::eta, EncodedEventId::event(), CaloParticle::eventId(), CaloParticle::g4Tracks(), CaloParticle::genParticle_begin(), CaloParticle::genParticle_end(), intimeOnly_, dqmiolumiharvest::j, maxRapidity_, maxSimClusters_, meanPhi_, minRapidity_, notConvertedOnly_, AlCaHLTBitMon_ParallelJobs::p, CaloParticle::p4(), CaloParticle::pdgId(), pdgId_, ntuple::pdgids, ptMax2_, ptMin2_, rangePhi_, signalOnly_, CaloParticle::simClusters(), edm::RefVector< C, T, F >::size(), stableOnly_, and CaloParticle::status().

                                                                                         {
    // signal only means no PU particles
    if (signalOnly_ && !(cp.eventId().bunchCrossing() == 0 && cp.eventId().event() == 0))
      return false;
    // intime only means no OOT PU particles
    if (intimeOnly_ && !(cp.eventId().bunchCrossing() == 0))
      return false;

    if (cp.simClusters().size() > maxSimClusters_)
      return false;

    auto pdgid = cp.pdgId();
    if (!pdgId_.empty()) {
      bool testId = false;
      for (auto id : pdgId_) {
        if (id == pdgid) {
          testId = true;
          break;
        }
      }
      if (!testId)
        return false;
    }

    if (chargedOnly_ && cp.charge() == 0)
      return false;  //select only if charge!=0

    // select only stable particles
    if (stableOnly_) {
      for (CaloParticle::genp_iterator j = cp.genParticle_begin(); j != cp.genParticle_end(); ++j) {
        if (j->get() == nullptr || j->get()->status() != 1) {
          return false;
        }
      }

      // test for remaining unstabled due to lack of genparticle pointer
      std::vector<int> pdgids{11, 13, 211, 321, 2212, 3112, 3222, 3312, 3334};
      if (cp.status() == -99 && (!std::binary_search(pdgids.begin(), pdgids.end(), std::abs(pdgid)))) {
        return false;
      }
    }

    // select only particles which did not convert/decay before the calorimeter
    // in case of electrons, bremsstrahlung is usually the cause, thus this selection is skipped
    if (std::abs(pdgid) != 11) {
      if (notConvertedOnly_) {
        if (cp.g4Tracks()[0].getPositionAtBoundary() == math::XYZTLorentzVectorF(0, 0, 0, 0)) {
          return false;
        }
        if (cp.g4Tracks()[0].getMomentumAtBoundary() == math::XYZTLorentzVectorF(0, 0, 0, 0)) {
          return false;
        }
      }
    }

    auto etaOk = [&](const CaloParticle& p) -> bool {
      float eta = etaFromXYZ(p.px(), p.py(), p.pz());
      return (eta >= minRapidity_) & (eta <= maxRapidity_);
    };
    auto phiOk = [&](const CaloParticle& p) {
      float dphi = deltaPhi(atan2f(p.py(), p.px()), meanPhi_);
      return dphi >= -rangePhi_ && dphi <= rangePhi_;
    };
    auto ptOk = [&](const CaloParticle& p) {
      double pt2 = cp.p4().perp2();
      return pt2 >= ptMin2_ && pt2 <= ptMax2_;
    };

    return (ptOk(cp) && etaOk(cp) && phiOk(cp));
  }

Member Data Documentation

bool CaloParticleSelector::chargedOnly_

private

Definition at line 148 of file CaloParticleSelector.h.

Referenced by operator()().

bool CaloParticleSelector::intimeOnly_

private

Definition at line 147 of file CaloParticleSelector.h.

Referenced by operator()().

double CaloParticleSelector::lip_

private

Definition at line 140 of file CaloParticleSelector.h.

float CaloParticleSelector::maxRapidity_

private

Definition at line 139 of file CaloParticleSelector.h.

Referenced by operator()().

unsigned int CaloParticleSelector::maxSimClusters_

private

Definition at line 145 of file CaloParticleSelector.h.

Referenced by operator()().

float CaloParticleSelector::meanPhi_

private

Definition at line 142 of file CaloParticleSelector.h.

Referenced by operator()().

int CaloParticleSelector::minHit_

private

Definition at line 144 of file CaloParticleSelector.h.

float CaloParticleSelector::minRapidity_

private

Definition at line 138 of file CaloParticleSelector.h.

Referenced by operator()().

bool CaloParticleSelector::notConvertedOnly_

private

Definition at line 150 of file CaloParticleSelector.h.

Referenced by operator()().

std::vector<int> CaloParticleSelector::pdgId_

private

Definition at line 151 of file CaloParticleSelector.h.

Referenced by operator()().

double CaloParticleSelector::ptMax2_

private

Definition at line 137 of file CaloParticleSelector.h.

Referenced by operator()().

double CaloParticleSelector::ptMin2_

private

Definition at line 136 of file CaloParticleSelector.h.

Referenced by operator()().

float CaloParticleSelector::rangePhi_

private

Definition at line 143 of file CaloParticleSelector.h.

Referenced by operator()().

bool CaloParticleSelector::signalOnly_

private

Definition at line 146 of file CaloParticleSelector.h.

Referenced by operator()().

bool CaloParticleSelector::stableOnly_

private

Definition at line 149 of file CaloParticleSelector.h.

Referenced by operator()().

double CaloParticleSelector::tip2_

private

Definition at line 141 of file CaloParticleSelector.h.