Changeset 1123 in svn for trunk/modules

Timestamp:

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

Author:

Pavel Demin

Message:

add Weight branch and Weighter module

Location:

trunk/modules

Files:

• ModulesLinkDef.h (modified) (4 diffs)
• TreeWriter.cc (modified) (2 diffs)
• TreeWriter.h (modified) (1 diff)

trunk/modules/ModulesLinkDef.h

@@ -7 +7 @@
  *  $Revision$
  *
- *  
+ *
  *  \author P. Demin - UCL, Louvain-la-Neuve
  *
@@ -13 +13 @@
 
 #include "modules/Delphes.h"
- 
+
 #include "modules/FastJetFinder.h"
 #include "modules/ParticlePropagator.h"
@@ -33 +33 @@
 #include "modules/StatusPidFilter.h"
 #include "modules/Cloner.h"
+#include "modules/Weighter.h"
 #include "modules/ExampleModule.h"
 
@@ -62 +63 @@
 #pragma link C++ class StatusPidFilter+;
 #pragma link C++ class Cloner+;
+#pragma link C++ class Weighter+;
 #pragma link C++ class ExampleModule+;
 

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()
 {

trunk/modules/TreeWriter.h

@@ -51 +51 @@
   void ProcessScalarHT(ExRootTreeBranch *branch, TObjArray *array);
   void ProcessRho(ExRootTreeBranch *branch, TObjArray *array);
+  void ProcessWeight(ExRootTreeBranch *branch, TObjArray *array);
 
 #ifndef __CINT__