Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

allow to ask for a mother ID in MCMultiParticleFilter #16270

Merged
merged 1 commit into from Oct 23, 2016
Merged

allow to ask for a mother ID in MCMultiParticleFilter #16270

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
3 changes: 2 additions & 1 deletion GeneratorInterface/GenFilters/interface/MCMultiParticleFilter.h
View file
Open in desktop
Expand Up @@ -51,9 +51,10 @@ class MCMultiParticleFilter : public edm::EDFilter {
bool acceptMore_; // if true (default), accept numRequired or more.
// if false, accept events with exactly equal to numRequired.
std::vector<int> particleID_; // vector of particle IDs to look for
// the three next variables can either be a vector of length 1 (in which case the same
// the four next variables can either be a vector of length 1 (in which case the same
// value is used for all particle IDs) or of length equal to the length of ParticleID (in which
// case the corresponding value is used for each).
std::vector<int> motherID_; // mother ID of particles (optional)
std::vector<double> ptMin_; // minimum Pt of particles
std::vector<double> etaMax_; // maximum fabs(eta) of particles
std::vector<int> status_; // status of particles
Expand Down
30 changes: 26 additions & 4 deletions GeneratorInterface/GenFilters/src/MCMultiParticleFilter.cc
View file
Open in desktop
Expand Up @@ -17,12 +17,17 @@ MCMultiParticleFilter::MCMultiParticleFilter(const edm::ParameterSet& iConfig) :
std::vector<double> defptmin(1, 0);
std::vector<double> defetamax(1, 999.0);
std::vector<int> defstat(1, 0);
std::vector<int> defmother;
defmother.push_back(0);
motherID_ = iConfig.getUntrackedParameter< std::vector<int> >("MotherID", defstat);

// check for same size
if ( (ptMin_.size() > 1 && particleID_.size() != ptMin_.size())
|| (etaMax_.size() > 1 && particleID_.size() != etaMax_.size())
|| (status_.size() > 1 && particleID_.size() != status_.size()) ) {
edm::LogWarning("MCMultiParticleFilter") << "WARNING: MCMultiParticleFilter: size of PtMin, EtaMax, and/or Status does not match ParticleID size!" << std::endl;
|| (status_.size() > 1 && particleID_.size() != status_.size())
|| (motherID_.size() > 1 && particleID_.size() != motherID_.size())
) {
edm::LogWarning("MCMultiParticleFilter") << "WARNING: MCMultiParticleFilter: size of PtMin, EtaMax, motherID, and/or Status does not match ParticleID size!" << std::endl;
}

// Fill arrays with defaults if necessary
Expand All @@ -32,6 +37,8 @@ MCMultiParticleFilter::MCMultiParticleFilter(const edm::ParameterSet& iConfig) :
etaMax_.push_back(defetamax[0]);
while (status_.size() < particleID_.size())
status_.push_back(defstat[0]);
while (motherID_.size() < particleID_.size())
motherID_.push_back(defmother[0]);
}

MCMultiParticleFilter::~MCMultiParticleFilter()
Expand Down Expand Up @@ -62,8 +69,23 @@ bool MCMultiParticleFilter::filter(edm::Event& iEvent, const edm::EventSetup& iS
(*p)->momentum().perp() > ptMin_[i] &&
fabs((*p)->momentum().eta()) < etaMax_[i] &&
(status_[i] == 0 || (*p)->status() == status_[i])) {
nFound++;
break; // only match a given particle once!
if(motherID_[i] == 0 ){ // do not check for mother ID if not sepcified
nFound++;
break; // only match a given particle once!
}
else{
bool hascorrectmother=false;
for ( HepMC::GenVertex::particles_in_const_iterator mo = (*p)->production_vertex()->particles_in_const_begin(); mo != (*p)->production_vertex()->particles_in_const_end(); ++mo){
if( (*mo)->pdg_id() == motherID_[i]){
hascorrectmother = true;
break;
}
}
if(hascorrectmother){
nFound++;
break; // only match a given particle once!
}
}
}
} // loop over targets

Expand Down