Changeset 0799cd1 in git for readers/DelphesCMSFWLite.cpp

Timestamp:

Jun 24, 2021, 9:02:43 PM (3 years ago)

Author:

GitHub <noreply@…>

Branches:

master

Children:

5c257a5

Parents:

71eba5e

git-author:

Julie Hogan <jmhogan@…> (06/24/21 21:02:43)

git-committer:

GitHub <noreply@…> (06/24/21 21:02:43)

Message:

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);