Changeset d203a37 in git

Timestamp:

May 14, 2018, 8:46:22 AM (7 years ago)

Author:

Pavel Demin <pavel-demin@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

2886328

Parents:

00382e9

Message:

add branch Weight to DelphesHepMC.cpp

Files:

• classes/DelphesHepMCReader.cc (modified) (1 diff)
• classes/DelphesHepMCReader.h (modified) (1 diff)
• readers/DelphesHepMC.cpp (modified) (3 diffs)

classes/DelphesHepMCReader.cc

@@ -333 +333 @@
 //---------------------------------------------------------------------------
 
+void DelphesHepMCReader::AnalyzeWeight(ExRootTreeBranch *branch)
+{
+  Weight *element;
+  vector< double >::const_iterator itWeight;
+
+  for(itWeight = fWeight.begin(); itWeight != fWeight.end(); ++itWeight)
+  {
+    element = static_cast<Weight *>(branch->NewEntry());
+
+    element->Weight = *itWeight;
+  }
+}
+
+//---------------------------------------------------------------------------
+
 void DelphesHepMCReader::AnalyzeParticle(DelphesFactory *factory,
   TObjArray *allParticleOutputArray,

classes/DelphesHepMCReader.h

@@ -59 +59 @@
     TStopwatch *readStopWatch, TStopwatch *procStopWatch);
 
+  void AnalyzeWeight(ExRootTreeBranch *branch);
+
 private:
 

readers/DelphesHepMC.cpp

@@ -63 +63 @@
   TStopwatch readStopWatch, procStopWatch;
   ExRootTreeWriter *treeWriter = 0;
-  ExRootTreeBranch *branchEvent = 0;
+  ExRootTreeBranch *branchEvent = 0, *branchWeight = 0;
   ExRootConfReader *confReader = 0;
   Delphes *modularDelphes = 0;
@@ -103 +103 @@
 
     branchEvent = treeWriter->NewBranch("Event", HepMCEvent::Class());
+    branchWeight = treeWriter->NewBranch("Weight", Weight::Class());
 
     confReader = new ExRootConfReader;
@@ -194 +195 @@
 
             reader->AnalyzeEvent(branchEvent, eventCounter, &readStopWatch, &procStopWatch);
+            reader->AnalyzeWeight(branchWeight);
 
             treeWriter->Fill();