Implement access to raw MET

include/mensura/PECReader/PECJetMETReader.hpp

@@ -101,6 +101,9 @@ class PECJetMETReader: public JetMETReader
      */
     virtual double GetJetRadius() const override;
 
+    /// Requests reading of raw MET
+    void ReadRawMET(bool enable = true);
+
     /// Specifies name of the plugin that provides generator-level jets
     void SetGenJetReader(std::string const name = "GenJetMET");
 
@@ -134,26 +137,29 @@ class PECJetMETReader: public JetMETReader
     /**
      * \brief An auxiliary pointer to jet buffer
      * 
-     * Need by ROOT to read the object from a tree.
+     * Needed by ROOT to read the object from a tree.
      */
     decltype(bfJets) *bfJetPointer;
 
-    /// Buffer to read the branch with METs
-    std::vector<pec::Candidate> bfMETs;
+    /// Buffers to read the branches with METs
+    std::vector<pec::Candidate> bfMETs, bfUncorrMETs;
 
     /**
-     * \brief An auxiliary pointer to MET buffer
+     * \brief Auxiliary pointers to MET buffers
      * 
-     * Need by ROOT to read the object from a tree.
+     * Needed by ROOT to read the object from a tree.
      */
-    decltype(bfMETs) *bfMETPointer;
+    decltype(bfMETs) *bfMETPointer, *bfUncorrMETPointer;
 
     /// Minimal allowed transverse momentum
     double minPt;
 
     /// Maximal allowed absolute value of pseudorapidity
     double maxAbsEta;
 
+    /// Specifies whether raw MET should be read
+    bool readRawMET;
+
     /**
      * \brief Name of the plugin that produces leptons
      * 

include/mensura/core/JetMETReader.hpp

@@ -45,10 +45,16 @@ class JetMETReader: public ReaderPlugin
     /// Returns corrected MET in the current event
     MET const &GetMET() const;
 
+    /// Returns raw MET in the current event
+    MET const &GetRawMET() const;
+
 protected:
     /// Collection of (corrected) jets in the current event
     std::vector<Jet> jets;
 
     /// Corrected MET in the current event
     MET met;
+
+    /// Raw MET in the current event
+    MET rawMET;
 };

modules/PECReader/src/PECJetMETReader.cpp

@@ -17,8 +17,9 @@ PECJetMETReader::PECJetMETReader(std::string name /*= "JetMET"*/):
     inputDataPluginName("InputData"), inputDataPlugin(nullptr),
     systServiceName("Systematics"),
     treeName("pecJetMET/JetMET"),
-    bfJetPointer(&bfJets), bfMETPointer(&bfMETs),
+    bfJetPointer(&bfJets), bfMETPointer(&bfMETs), bfUncorrMETPointer(&bfUncorrMETs),
     minPt(0.), maxAbsEta(std::numeric_limits<double>::infinity()),
+    readRawMET(false),
     leptonPluginName("Leptons"), leptonPlugin(nullptr),
     genJetPluginName(""), genJetPlugin(nullptr),
     systType(SystType::None), systDirection(0)
@@ -32,8 +33,9 @@ PECJetMETReader::PECJetMETReader(PECJetMETReader const &src) noexcept:
     inputDataPluginName(src.inputDataPluginName), inputDataPlugin(src.inputDataPlugin),
     systServiceName(src.systServiceName),
     treeName(src.treeName),
-    bfJetPointer(&bfJets), bfMETPointer(&bfMETs),
+    bfJetPointer(&bfJets), bfMETPointer(&bfMETs), bfUncorrMETPointer(&bfUncorrMETs),
     minPt(src.minPt), maxAbsEta(src.maxAbsEta),
+    readRawMET(src.readRawMET),
     leptonPluginName(src.leptonPluginName), leptonPlugin(src.leptonPlugin),
     leptonDR2(src.leptonDR2),
     genJetPluginName(src.genJetPluginName), genJetPlugin(src.genJetPlugin),
@@ -97,13 +99,15 @@ void PECJetMETReader::BeginRun(Dataset const &)
 
     tree->SetBranchStatus("jets.bTagCMVA", false);
     tree->SetBranchStatus("jets.secVertexMass", false);
-    // tree->SetBranchStatus("jets.area", false);
     tree->SetBranchStatus("jets.charge", false);
     tree->SetBranchStatus("jets.pullAngle", false);
 
     tree->SetBranchAddress("jets", &bfJetPointer);
     tree->SetBranchAddress("METs", &bfMETPointer);
 
+    if (readRawMET)
+        tree->SetBranchAddress("uncorrMETs", &bfUncorrMETPointer);
+
     ROOTLock::Unlock();
 }
 
@@ -134,6 +138,12 @@ double PECJetMETReader::GetJetRadius() const
 }
 
 
+void PECJetMETReader::ReadRawMET(bool enable /*= true*/)
+{
+    readRawMET = enable;
+}
+
+
 void PECJetMETReader::SetGenJetReader(std::string const name /*= "GenJetMET"*/)
 {
     genJetPluginName = name;
@@ -335,9 +345,17 @@ bool PECJetMETReader::ProcessEvent()
     met.SetPtEtaPhiM(correctedMET.Pt(), 0., correctedMET.Phi(), 0.);
 
 
+    // Copy raw MET if requested
+    if (readRawMET)
+    {
+        pec::Candidate const &pecRawMET = bfUncorrMETs.at(0);
+        rawMET.SetPtEtaPhiM(pecRawMET.Pt(), 0., pecRawMET.Phi(), 0.);
+    }
+
+
     #ifdef DEBUG
     std::cout << "PECJetMETReader[\"" << GetName() << "\"]: MET in the current event:\n";
-    std::cout << " Raw MET (pt, phi): " << bfMETs.at(1).Pt() << ", " << bfMETs.at(1).Phi() << '\n';
+    std::cout << " Raw MET (pt, phi): " << rawMET.Pt() << ", " << rawMET.Phi() << '\n';
     std::cout << " Corrected MET (pt, phi): " << bfMETs.at(0).Pt() << ", "
       << bfMETs.at(0).Phi() << std::endl;
     #endif

modules/core/src/JetMETReader.cpp

@@ -20,3 +20,9 @@ MET const &JetMETReader::GetMET() const
 {
     return met;
 }
+
+
+MET const &JetMETReader::GetRawMET() const
+{
+    return rawMET;
+}