Changeset 71648c2 in git for modules/TreeWriter.cc

Timestamp:

May 16, 2013, 4:25:05 PM (12 years ago)

Author:

pavel <pavel@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

8608ef8

Parents:

809e3a9

Message:

add Rho branch

File:

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

modules/TreeWriter.cc

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