Changeset 0dc0eeb in git for classes

Timestamp:

Aug 16, 2013, 4:29:25 PM (11 years ago)

Author:

pavel <pavel@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

82e07cf

Parents:

0a0a5ef

Message:

read reweighting information from LHEF

Location:

classes

Files:

• DelphesLHEFReader.cc (modified) (8 diffs)
• DelphesLHEFReader.h (modified) (4 diffs)

classes/DelphesLHEFReader.cc

@@ -41 +41 @@
 DelphesLHEFReader::DelphesLHEFReader() :
   fInputFile(0), fBuffer(0), fPDG(0),
-  fEventCounter(-1), fParticleCounter(-1)
+  fEventReady(kFALSE), fEventCounter(-1), fParticleCounter(-1)
 {
   fBuffer = new char[kBufferSize];
@@ -66 +66 @@
 void DelphesLHEFReader::Clear()
 {
+  fEventReady = kFALSE;
   fEventCounter = -1;
   fParticleCounter = -1;
+  fRwgtList.clear();
 }
 
@@ -74 +76 @@
 bool DelphesLHEFReader::EventReady()
 {
-  return (fEventCounter == 0) && (fParticleCounter == 0);
+  return fEventReady;
 }
 
@@ -85 +87 @@
 {
   int rc;
+  char *pch;
+  double weight;
 
   if(!fgets(fBuffer, kBufferSize, fInputFile)) return kFALSE;
-
-  DelphesStream bufferStream(fBuffer);
 
   if(strstr(fBuffer, "<event>"))
@@ -97 +99 @@
   else if(fEventCounter > 0)
   {
+    DelphesStream bufferStream(fBuffer);
+
     rc = bufferStream.ReadInt(fParticleCounter)
       && bufferStream.ReadInt(fProcessID)
@@ -114 +118 @@
   else if(fParticleCounter > 0)
   {
+    DelphesStream bufferStream(fBuffer);
+
     rc = bufferStream.ReadInt(fPID)
       && bufferStream.ReadInt(fStatus)
@@ -137 +143 @@
     --fParticleCounter;
   }
+  else if(strstr(fBuffer, "<wgt"))
+  {
+    pch = strstr(fBuffer, ">");
+    if(!pch)
+    {
+      cerr << "** ERROR: " << "invalid weight format" << endl;
+      return kFALSE;
+    }
+
+    DelphesStream bufferStream(pch + 1);
+    rc = bufferStream.ReadDbl(weight);
+
+    if(!rc)
+    {
+      cerr << "** ERROR: " << "invalid weight format" << endl;
+      return kFALSE;
+    }
+
+    fRwgtList.push_back(weight);
+  }
+  else if(strstr(fBuffer, "</event>"))
+  {
+    fEventReady = kTRUE;
+  }
 
   return kTRUE;
@@ -163 +193 @@
 //---------------------------------------------------------------------------
 
+void DelphesLHEFReader::AnalyzeRwgt(ExRootTreeBranch *branch)
+{
+  Weight *element;
+  vector<double>::const_iterator itRwgtList;
+
+  for(itRwgtList = fRwgtList.begin(); itRwgtList != fRwgtList.end(); ++itRwgtList)
+  {
+    element = static_cast<Weight *>(branch->NewEntry());
+
+    element->Weight = *itRwgtList;
+  }
+}
+
+//---------------------------------------------------------------------------
+
 void DelphesLHEFReader::AnalyzeParticle(DelphesFactory *factory,
   TObjArray *allParticleOutputArray,

classes/DelphesLHEFReader.h

@@ -16 +16 @@
 
 #include <stdio.h>
+
+#include <vector>
 
 class TObjArray;
@@ -43 +45 @@
     TStopwatch *readStopWatch, TStopwatch *procStopWatch);
 
+  void AnalyzeRwgt(ExRootTreeBranch *branch);
+
 private:
 
@@ -56 +60 @@
   TDatabasePDG *fPDG;
 
+  bool fEventReady;
+
   int fEventCounter;
 
@@ -63 +69 @@
   int fPID, fStatus, fM1, fM2, fC1, fC2;
   double fPx, fPy, fPz, fE, fMass;
+  
+  std::vector<double> fRwgtList;
 };