Page MenuHomeHEPForge
Files F10881536

No OneTemporary
Actions

View Options

diff --git a/src/Core/Jet.cc b/src/Core/Jet.cc
--- a/src/Core/Jet.cc
+++ b/src/Core/Jet.cc
@@ -1,210 +1,210 @@
#include "Rivet/Jet.hh"
#include "Rivet/Cuts.hh"
#include "Rivet/ParticleName.hh"
#include "Rivet/Tools/Logging.hh"
#include "Rivet/Tools/ParticleIdUtils.hh"
namespace Rivet {
Jet& Jet::clear() {
_momentum = FourMomentum();
_pseudojet.reset(0,0,0,0);
_particles.clear();
return *this;
}
Jet& Jet::setState(const FourMomentum& mom, const Particles& particles, const Particles& tags) {
clear();
_momentum = mom;
_pseudojet = fastjet::PseudoJet(mom.px(), mom.py(), mom.pz(), mom.E());
_particles = particles;
_tags = tags;
return *this;
}
Jet& Jet::setState(const fastjet::PseudoJet& pj, const vector<Particle>& particles, const Particles& tags) {
clear();
_pseudojet = pj;
_momentum = FourMomentum(pj.e(), pj.px(), pj.py(), pj.pz());
_particles = particles;
_tags = tags;
// if (_particles.empty()) {
// foreach (const fastjet::PseudoJet pjc, _pseudojet.constituents()) {
// // If there is no attached user info, we can't create a meaningful particle, so skip
// if (!pjc.has_user_info<RivetFJInfo>()) continue;
// const RivetFJInfo& fjinfo = pjc.user_info<RivetFJInfo>();
// // Don't add ghosts to the particles list
// if (fjinfo.isGhost) continue;
// // Otherwise construct a Particle from the PseudoJet, preferably from an associated GenParticle
// ?if (fjinfo.genParticle != NULL) {
// _particles.push_back(Particle(fjinfo.genParticle));
// } else {
// if (fjinfo.pid == 0) continue; // skip if there is a null PID entry in the FJ info
// const FourMomentum pjcmom(pjc.e(), pjc.px(), pjc.py(), pjc.pz());
// _particles.push_back(Particle(fjinfo.pid, pjcmom));
// }
// }
// }
return *this;
}
Jet& Jet::setParticles(const vector<Particle>& particles) {
_particles = particles;
return *this;
}
bool Jet::containsParticle(const Particle& particle) const {
const int barcode = particle.genParticle()->barcode();
foreach (const Particle& p, particles()) {
if (p.genParticle()->barcode() == barcode) return true;
}
return false;
}
bool Jet::containsParticleId(PdgId pid) const {
foreach (const Particle& p, particles()) {
if (p.pid() == pid) return true;
}
return false;
}
bool Jet::containsParticleId(const vector<PdgId>& pids) const {
foreach (const Particle& p, particles()) {
foreach (PdgId pid, pids) {
if (p.pid() == pid) return true;
}
}
return false;
}
/// @todo Jet::containsMatch(Matcher m) { ... if m(pid) return true; ... }
double Jet::neutralEnergy() const {
double e_neutral = 0.0;
foreach (const Particle& p, particles()) {
const PdgId pid = p.pid();
if (PID::threeCharge(pid) == 0) {
e_neutral += p.E();
}
}
return e_neutral;
}
double Jet::hadronicEnergy() const {
double e_hadr = 0.0;
foreach (const Particle& p, particles()) {
const PdgId pid = p.pid();
if (PID::isHadron(pid)) {
e_hadr += p.E();
}
}
return e_hadr;
}
bool Jet::containsCharm(bool include_decay_products) const {
foreach (const Particle& p, particles()) {
const PdgId pid = p.pid();
if (abs(pid) == PID::CQUARK) return true;
if (PID::isHadron(pid) && PID::hasCharm(pid)) return true;
if (include_decay_products) {
const HepMC::GenVertex* gv = p.genParticle()->production_vertex();
if (gv) {
foreach (const GenParticle* pi, Rivet::particles(gv, HepMC::ancestors)) {
const PdgId pid2 = pi->pdg_id();
if (PID::isHadron(pid2) && PID::hasCharm(pid2)) return true;
}
}
}
}
return false;
}
bool Jet::containsBottom(bool include_decay_products) const {
foreach (const Particle& p, particles()) {
const PdgId pid = p.pid();
if (abs(pid) == PID::BQUARK) return true;
if (PID::isHadron(pid) && PID::hasBottom(pid)) return true;
if (include_decay_products) {
const HepMC::GenVertex* gv = p.genParticle()->production_vertex();
if (gv) {
foreach (const GenParticle* pi, Rivet::particles(gv, HepMC::ancestors)) {
const PdgId pid2 = pi->pdg_id();
if (PID::isHadron(pid2) && PID::hasBottom(pid2)) return true;
}
}
}
}
return false;
}
Particles Jet::tags(const Cut& c) const {
Particles rtn;
foreach (const Particle& p, tags()) {
if (c->accept(p)) rtn.push_back(p);
}
return rtn;
}
Particles Jet::bTags(const Cut& c) const {
Particles rtn;
foreach (const Particle& tp, tags()) {
if (hasBottom(tp) && c->accept(tp)) rtn.push_back(tp);
}
return rtn;
}
Particles Jet::cTags(const Cut& c) const {
Particles rtn;
foreach (const Particle& tp, tags()) {
/// @todo Is making b and c tags exclusive the right thing to do?
if (hasCharm(tp) && !hasBottom(tp) && c->accept(tp)) rtn.push_back(tp);
}
return rtn;
}
Particles Jet::tauTags(const Cut& c) const {
Particles rtn;
foreach (const Particle& tp, tags()) {
if (isTau(tp) && c->accept(tp)) rtn.push_back(tp);
}
return rtn;
}
/// Filter a jet collection in-place to the subset that passes the supplied Cut
Jets& filterBy(Jets& jets, const Cut& c) {
if (c != Cuts::OPEN) {
const auto newend = std::remove_if(jets.begin(), jets.end(), [&](const Jet& j){ return !c->accept(j); });
jets.erase(newend, jets.end());
}
return jets;
}
/// Get a subset of the supplied jets that passes the supplied Cut
Jets filterBy(const Jets& jets, const Cut& c) {
// Just return a copy if the cut is open
if (c == Cuts::OPEN) return jets;
// But if there is a non-trivial cut...
Jets rtn;
- std::copy_if(jets.begin(), jets.end(), rtn.begin(), [&](const Jet& j){ return c->accept(j); });
+ std::copy_if(jets.begin(), jets.end(), back_inserter(rtn), [&](const Jet& j){ return c->accept(j); });
return rtn;
}
}

File Metadata

Mime Type
text/x-diff
Expires
Sat, May 3, 6:30 AM (1 d, 17 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4983050
Default Alt Text
(6 KB)

Event Timeline