Changeset 5c257a5 in git for readers/DelphesCMSFWLite.cpp

Timestamp:

Jul 2, 2021, 3:36:53 PM (3 years ago)

Author:

GitHub <noreply@…>

Branches:

master

Children:

1ad8eca

Parents:

910bd98 (diff), 0799cd1 (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

git-author:

Michele Selvaggi <michele.selvaggi@…> (07/02/21 15:36:53)

git-committer:

GitHub <noreply@…> (07/02/21 15:36:53)

Message:

Merge pull request #100 from jmhogan/master

Implement max/skipEvents in CMSFWLite reader

File:

• readers/DelphesCMSFWLite.cpp (modified) (3 diffs)

readers/DelphesCMSFWLite.cpp

@@ -311 +311 @@
   DelphesFactory *factory = 0;
   TObjArray *allParticleOutputArray = 0, *stableParticleOutputArray = 0, *partonOutputArray = 0;
-  Int_t i;
+  Int_t i, maxEvents, skipEvents;
   Long64_t eventCounter, numberOfEvents;
   Bool_t firstEvent = kTRUE;
@@ -353 +353 @@
     confReader = new ExRootConfReader;
     confReader->ReadFile(argv[1]);
+    
+    maxEvents = confReader->GetInt("::MaxEvents", 0);
+    skipEvents = confReader->GetInt("::SkipEvents", 0);
+    
+    if(maxEvents < 0)
+    {
+      throw runtime_error("MaxEvents must be zero or positive");
+    }
+    
+    if(skipEvents < 0)
+    {
+      throw runtime_error("SkipEvents must be zero or positive");
+    }
 
     modularDelphes = new Delphes("Delphes");
@@ -391 +404 @@
       treeWriter->Clear();
 
-      for(event.toBegin(); !event.atEnd() && !interrupted; ++event)
+      for(event.toBegin(); !event.atEnd() && !interrupted && (maxEvents <= 0 || eventCounter-skipEvents < maxEvents); ++event)
       {
-        ConvertInput(event, eventCounter, branchEvent, branchWeight, factory,
-          allParticleOutputArray, stableParticleOutputArray, partonOutputArray, firstEvent);
-        modularDelphes->ProcessTask();
-
-        firstEvent = kFALSE;
-
-        treeWriter->Fill();
-
-        modularDelphes->Clear();
-        treeWriter->Clear();
+        if(eventCounter > skipEvents){
+          ConvertInput(event, eventCounter, branchEvent, branchWeight, factory,
+            allParticleOutputArray, stableParticleOutputArray, partonOutputArray, firstEvent);
+          modularDelphes->ProcessTask();
+
+          firstEvent = kFALSE;
+
+          treeWriter->Fill();
+
+          modularDelphes->Clear();
+          treeWriter->Clear();
+        }
 
         progressBar.Update(eventCounter, eventCounter);