Changeset 1123 in svn for trunk/modules/TreeWriter.cc

Timestamp:

May 26, 2013, 2:00:39 AM (11 years ago)

Author:

Pavel Demin

Message:

add Weight branch and Weighter module

File:

• trunk/modules/TreeWriter.cc (modified) (2 diffs)

trunk/modules/TreeWriter.cc

@@ -65 +65 @@
   fClassMap[ScalarHT::Class()] = &TreeWriter::ProcessScalarHT;
   fClassMap[Rho::Class()] = &TreeWriter::ProcessRho;
+  fClassMap[Weight::Class()] = &TreeWriter::ProcessWeight;
 
   TBranchMap::iterator itBranchMap;
@@ -538 +539 @@
 //------------------------------------------------------------------------------
 
+void TreeWriter::ProcessWeight(ExRootTreeBranch *branch, TObjArray *array)
+{
+  Candidate *candidate = 0;
+  Weight *entry = 0;
+
+  // get the first entry
+  if((candidate = static_cast<Candidate*>(array->At(0))))
+  {
+    const TLorentzVector &momentum = candidate->Momentum;
+
+    entry = static_cast<Weight*>(branch->NewEntry());
+
+    entry->Weight = momentum.E();
+  }
+}
+
+//------------------------------------------------------------------------------
+
 void TreeWriter::Process()
 {