Changeset 1157 in svn for trunk/examples/DelphesCMSFWLite.cpp

Timestamp:

Jul 2, 2013, 5:45:31 PM (11 years ago)

Author:

Pavel Demin

Message:

fix daughters indices in DelphesCMSFWLite

File:

• trunk/examples/DelphesCMSFWLite.cpp (modified) (5 diffs)

trunk/examples/DelphesCMSFWLite.cpp

@@ +1 @@
+#include <algorithm>
 #include <stdexcept>
 #include <iostream>
@@ -5 +6 @@
 
 #include <map>
+#include <vector>
 
 #include <stdlib.h>
@@ -41 +43 @@
 void ConvertInput(fwlite::Event &event, DelphesFactory *factory, TObjArray *allParticleOutputArray, TObjArray *stableParticleOutputArray, TObjArray *partonOutputArray)
 {
-  size_t i;
   fwlite::Handle< vector< reco::GenParticle > > handleParticle;
+  vector< reco::GenParticle >::const_iterator itParticle;
+
+  vector< const reco::Candidate * > vectorCandidate;
+  vector< const reco::Candidate * >::iterator itCandidate;
+
   handleParticle.getByLabel(event, "genParticles");
 
@@ -56 +62 @@
   pdg = TDatabasePDG::Instance();
 
-  for(i = 0; i < handleParticle->size(); ++i)
-  {
-    const reco::GenParticle &particle = (*handleParticle)[i];
+  for(itParticle = handleParticle->begin(); itParticle != handleParticle->end(); ++itParticle)
+  {
+    vectorCandidate.push_back(&*itParticle);
+  }
+
+  for(itParticle = handleParticle->begin(); itParticle != handleParticle->end(); ++itParticle)
+  {
+    const reco::GenParticle &particle = *itParticle;
 
     pid = particle.pdgId();
@@ -71 +82 @@
 
     candidate->Status = status;
+
+    itCandidate = find(vectorCandidate.begin(), vectorCandidate.end(), particle.daughter(0));
+    if(itCandidate != vectorCandidate.end()) candidate->D1 = distance(vectorCandidate.begin(), itCandidate);
+
+    itCandidate = find(vectorCandidate.begin(), vectorCandidate.end(), particle.daughter(particle.numberOfDaughters() - 1));
+    if(itCandidate != vectorCandidate.end()) candidate->D2 = distance(vectorCandidate.begin(), itCandidate);
 
     pdgParticle = pdg->GetParticle(pid);