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

[backport] Modification to PythiaFilterMotherSister #33815

Merged
merged 4 commits into from Jun 8, 2021
Merged

[backport] Modification to PythiaFilterMotherSister #33815

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
a0bd378
added possibility to cut on pt of the nephews and moved from 3D to 2D…
mgratti May 21, 2021
65b32df
code style checks
mgratti May 21, 2021
d0ca029
while checking pt of decay products of sister particle, allow multipl…
mgratti Jun 1, 2021
2d3472c
ids and pts are now configurable parameters
mgratti Jun 8, 2021
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
7 changes: 3 additions & 4 deletions GeneratorInterface/GenFilters/interface/PythiaFilterMotherSister.h
View file
Open in desktop
Expand Up @@ -62,13 +62,12 @@ class PythiaFilterMotherSister : public edm::global::EDFilter<> {
const double maxrapcut;
const double minphicut;
const double maxphicut;
const double betaBoost;

//const int status;
std::vector<int> motherIDs;
const int sisterID;
//const int processID;

const double betaBoost;
const double maxSisDisplacement;
std::vector<int> nephewIDs;
std::vector<double> minNephewPts;
};
#endif
32 changes: 25 additions & 7 deletions GeneratorInterface/GenFilters/src/PythiaFilterMotherSister.cc
View file
Open in desktop
Expand Up @@ -22,13 +22,17 @@ minrapcut(iConfig.getUntrackedParameter("MinRapidity", -20.)),
maxrapcut(iConfig.getUntrackedParameter("MaxRapidity", 20.)),
minphicut(iConfig.getUntrackedParameter("MinPhi", -3.5)),
maxphicut(iConfig.getUntrackedParameter("MaxPhi", 3.5)),
betaBoost(iConfig.getUntrackedParameter("BetaBoost",0.)),
motherIDs(iConfig.getUntrackedParameter("MotherIDs", std::vector<int>{0})),
sisterID(iConfig.getUntrackedParameter("SisterID", 0)),
betaBoost(iConfig.getUntrackedParameter("BetaBoost",0.)),
maxSisDisplacement(iConfig.getUntrackedParameter("MaxSisterDisplacement", -1.))
maxSisDisplacement(iConfig.getUntrackedParameter("MaxSisterDisplacement", -1.)),
nephewIDs(iConfig.getUntrackedParameter("NephewIDs", std::vector<int>{0})),
minNephewPts(iConfig.getUntrackedParameter("MinNephewPts", std::vector<double>{0.}))
{
//now do what ever initialization is needed

if (nephewIDs.size() != minNephewPts.size()) {
throw cms::Exception("BadConfig") << "PythiaFilterMotherSister: "
<< "'nephewIDs' and 'minNephewPts' need same length.";
}
}


Expand Down Expand Up @@ -86,18 +90,32 @@ bool PythiaFilterMotherSister::filter(edm::StreamID, edm::Event& iEvent, const e
++dau ) {
// find the daugther you're interested in
if(abs((*dau)->pdg_id()) == abs(sisterID)) {

int failNephewPt = 0;

// check pt of the nephews
for ( HepMC::GenVertex::particle_iterator nephew = (*dau)->end_vertex()->particles_begin(HepMC::children);
nephew != (*dau)->end_vertex()->particles_end(HepMC::children);
++nephew ) {
int nephew_pdgId = abs((*nephew)->pdg_id());

for(unsigned int i=0; i<nephewIDs.size(); i++){
if(nephew_pdgId == abs(nephewIDs.at(i)))
failNephewPt += ((*nephew)->momentum().perp() < minNephewPts.at(i));
}
}
if(failNephewPt > 0) return false;

// calculate displacement of the sister particle, from production to decay
HepMC::GenVertex* v1 = (*dau)->production_vertex();
HepMC::GenVertex* v2 = (*dau)->end_vertex();

double lx12 = v1->position().x() - v2->position().x();
double ly12 = v1->position().y() - v2->position().y();
double lz12 = v1->position().z() - v2->position().z();
double lxyz12 = sqrt( lx12*lx12 + ly12*ly12 + lz12*lz12 );
double lxy12 = sqrt( lx12*lx12 + ly12*ly12);

if(maxSisDisplacement!= -1){
if(lxyz12 < maxSisDisplacement){
if(lxy12 < maxSisDisplacement){
return true;
}
} else {
Expand Down