HT module from Sal Rappoccio

Author: fgolf

interface/htProducerCppWorker.h

@@ -0,0 +1,27 @@
+#ifndef PhysicsTools_NanoAODTools_htProducerCppWorker_h
+#define PhysicsTools_NanoAODTools_htProducerCppWorker_h
+
+#include <utility>
+#include <TTreeReaderValue.h>
+#include <TTreeReaderArray.h>
+#include "DataFormats/Math/interface/LorentzVector.h"
+
+class htProducerCppWorker {
+
+public:
+
+  htProducerCppWorker(){}
+
+  void setJets(TTreeReaderValue<unsigned> *nJet_, TTreeReaderArray<float> *Jet_pt_){
+    nJet = nJet_; Jet_pt = Jet_pt_; 
+  }
+
+  float getHT();
+
+private:
+  TTreeReaderValue<unsigned> *nJet = nullptr;
+  TTreeReaderArray<float> *Jet_pt = nullptr;
+
+};
+
+#endif

python/postprocessing/examples/htProducerCpp.py

@@ -0,0 +1,51 @@
+import ROOT
+import os
+ROOT.PyConfig.IgnoreCommandLineOptions = True
+
+from PhysicsTools.NanoAODTools.postprocessing.framework.eventloop import Module
+
+class htProducerCpp(Module): # HT producer, unclean jets only (no lepton overlap cleaning, no jet selection)
+    def __init__(self):
+        if "/htProducerCppWorker_cc.so" not in ROOT.gSystem.GetLibraries():
+            print "Load C++ htProducerCppWorker worker module"
+            base = os.getenv("NANOAODTOOLS_BASE")
+            if base:
+                ROOT.gROOT.ProcessLine(".L %s/src/htProducerCppWorker.cc+O"%base)
+            else:
+                base = "%s/src/PhysicsTools/NanoAODTools"%os.getenv("CMSSW_BASE")
+                ROOT.gSystem.Load("libPhysicsToolsNanoAODTools.so")
+                ROOT.gROOT.ProcessLine(".L %s/interface/htProducerCppWorker.h"%base)
+        self.worker = ROOT.htProducerCppWorker()
+        pass
+    def beginJob(self):
+        pass
+    def endJob(self):
+        pass
+    def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
+        self.initReaders(inputTree) # initReaders must be called in beginFile
+        self.out = wrappedOutputTree
+        self.out.branch("HT_pt",  "F");
+    def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
+        pass
+
+    def initReaders(self,tree): # this function gets the pointers to Value and ArrayReaders and sets them in the C++ worker class
+        self.nJet = tree.valueReader("nJet")
+        self.Jet_pt = tree.arrayReader("Jet_pt")
+        self.worker.setJets(self.nJet,self.Jet_pt)
+        self._ttreereaderversion = tree._ttreereaderversion # self._ttreereaderversion must be set AFTER all calls to tree.valueReader or tree.arrayReader
+
+    def analyze(self, event):
+        """process event, return True (go to next module) or False (fail, go to next event)"""
+
+        if event._tree._ttreereaderversion > self._ttreereaderversion: # do this check at every event, as other modules might have read further branches
+            self.initReaders(event._tree)
+        # do NOT access other branches in python between the check/call to initReaders and the call to C++ worker code
+        output = self.worker.getHT()
+
+        self.out.fillBranch("HT_pt", output)
+        return True
+
+
+# define modules using the syntax 'name = lambda : constructor' to avoid having them loaded when not needed
+
+ht = lambda : htProducerCpp()

src/htProducerCppWorker.cc

@@ -0,0 +1,14 @@
+#include "../interface/htProducerCppWorker.h"
+
+float htProducerCppWorker::getHT(){
+
+    float ht(0.0);
+
+    unsigned n = (*nJet).Get()[0];
+    for (unsigned i=0; i<n; i++){
+      ht += (*Jet_pt)[i];
+    }
+
+    return ht; 
+
+};