Changeset 1108 in svn for trunk/classes/DelphesHepMCReader.cc

Timestamp:

May 7, 2013, 3:27:32 PM (12 years ago)

Author:

Pavel Demin

Message:

read momentum and position units from HepMC file

File:

• trunk/classes/DelphesHepMCReader.cc (modified) (5 diffs)

trunk/classes/DelphesHepMCReader.cc

@@ -74 +74 @@
   fWeightSize = 0;
   fWeight.clear();
+  fMomentumCoefficient = 1.0;
+  fPositionCoefficient = 1.0;
   fVertexCounter = -1;
   fInCounter = -1;
@@ -98 +100 @@
   map< int, pair< int, int > >::iterator itMotherMap;
   map< int, pair< int, int > >::iterator itDaughterMap;
-  char key;
+  char key, momentumUnit[4], positionUnit[3];
   int i, rc, state;
   double weight;
@@ -156 +158 @@
     }
   }
+  else if(key == 'U')
+  {
+    rc = sscanf(fBuffer + 1, "%3s %2s", momentumUnit, positionUnit);
+
+    if(rc != 2)
+    {
+      cerr << "** ERROR: " << "invalid units format" << endl;
+      return kFALSE;
+    }
+
+    if(strncmp(momentumUnit, "GEV", 3) == 0)
+    {
+      fMomentumCoefficient = 1.0;
+    }
+    else if(strncmp(momentumUnit, "MEV", 3) == 0)
+    {
+      fMomentumCoefficient = 0.001;
+    }
+    
+    if(strncmp(positionUnit, "MM", 3) == 0)
+    {
+      fPositionCoefficient = 1.0;
+    }
+    else if(strncmp(positionUnit, "CM", 3) == 0)
+    {
+      fPositionCoefficient = 10.0;
+    }
+  }
   else if(key == 'F')
   {
@@ -311 +341 @@
 
   candidate->Momentum.SetPxPyPzE(fPx, fPy, fPz, fE);
+  if(fMomentumCoefficient != 1.0)
+  {
+    candidate->Momentum *= fMomentumCoefficient;
+  }
 
   candidate->M2 = 1;
@@ -323 +357 @@
     candidate->M1 = fOutVertexCode;
     candidate->Position.SetXYZT(fX, fY, fZ, fT);
+    if(fPositionCoefficient != 1.0)
+    {
+      candidate->Position *= fPositionCoefficient;
+    }
   }
   if(fInVertexCode < 0)