Merge pull request #33815 from mgratti/from-CMSSW_10_2_X_2021-05-16-0000

[backport] Modification to PythiaFilterMotherSister
cms-sw · Jun 8, 2021 · 16498a8 · 16498a8
2 parents c02fdbb + 2d3472c
commit 16498a8
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 11 deletions.
diff --git a/GeneratorInterface/GenFilters/interface/PythiaFilterMotherSister.h b/GeneratorInterface/GenFilters/interface/PythiaFilterMotherSister.h
@@ -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
diff --git a/GeneratorInterface/GenFilters/src/PythiaFilterMotherSister.cc b/GeneratorInterface/GenFilters/src/PythiaFilterMotherSister.cc
@@ -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.";
+   }
 }
 
 
@@ -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 {