Selection

include/AnalysisOutput.h

@@ -0,0 +1,19 @@
+#pragma once
+#include "UHH2/core/include/Event.h"
+#include "UHH2/core/include/AnalysisModule.h"
+
+#include <iostream>
+
+using namespace std;
+
+class WriteOutput: public uhh2::AnalysisModule{
+public:
+
+  explicit WriteOutput(uhh2::Context&);
+  virtual bool process(uhh2::Event & ) override;
+
+private:
+  uhh2::Event::Handle<double>h_weight;
+
+
+};

include/GenSelections.h

@@ -1,21 +1,9 @@
 #pragma once
 
 #include <UHH2/core/include/Event.h>
-#include <UHH2/core/include/AnalysisModule.h>
 #include <UHH2/core/include/Selection.h>
-#include <UHH2/core/include/Utils.h>
 
-#include <UHH2/common/include/NSelections.h>
-#include <UHH2/common/include/ReconstructionHypothesis.h>
-#include <UHH2/common/include/ObjectIdUtils.h>
-#include <UHH2/common/include/TopJetIds.h>
 #include <UHH2/common/include/TTbarGen.h>
-#include <UHH2/common/include/Utils.h>
-
-#include <string>
-#include <vector>
-#include <unordered_map>
-#include <iostream>
 
 using namespace std;
 

include/MTopJetUtils.h

@@ -0,0 +1,9 @@
+#pragma once
+
+#include <UHH2/core/include/Event.h>
+
+////
+
+const Particle* leading_lepton(const uhh2::Event&);
+
+////

include/RecoSelections.h

@@ -12,6 +12,9 @@
 #include <UHH2/common/include/ObjectIdUtils.h>
 #include <UHH2/common/include/TopJetIds.h>
 #include <UHH2/common/include/TTbarGen.h>
+#include <UHH2/common/include/Utils.h>
+
+#include <UHH2/MTopJet_UL/include/MTopJetUtils.h>
 
 #include <iostream>
 #include <fstream>
@@ -35,4 +38,51 @@ namespace uhh2 {
   private:
     float min_met_, max_met_;
   };
+
+  ////////////////////////////////////////////////////////////////
+
+  class ElectronEtaVeto : public Selection {
+
+  public:
+    explicit ElectronEtaVeto(double, double);
+    virtual bool passes(const Event&) override;
+
+  private:
+    double lower, upper;
+  };
+
+  ////////////////////////////////////////////////////////////////
+
+  class TwoDCut : public Selection {
+
+  public:
+    explicit TwoDCut(float min_deltaR, float min_pTrel): min_deltaR_(min_deltaR), min_pTrel_(min_pTrel) {}
+    virtual bool passes(const Event&) override;
+
+  private:
+    float min_deltaR_, min_pTrel_;
+  };
+
+  ////////////////////////////////////////////////////////////////
+
+  class BadHCALSelection: public uhh2::Selection {
+  public:
+    BadHCALSelection(uhh2::Context &ctx, long int seed = 123456789);
+    virtual bool passes(const uhh2::Event &event) override;
+
+  private:
+    TRandom *m_rng;
+    long int m_seed;
+    Year year;
+    int m_runnumber = 319077;
+    double m_lumi_ratio = 0.64844705699; // (Run 319077(17.370008 pb-1) + Run C + Run D) / all 2018
+
+    double m_interval_eta = -1.3;
+    double m_interval_phi_low = -1.57;
+    double m_interval_phi_high = -0.87;
+
+  };
+
+  ////////////////////////////////////////////////////////////////
+
 }

include/RemoveLepton.h

@@ -0,0 +1,31 @@
+#pragma once
+#include "UHH2/core/include/Event.h"
+#include "UHH2/common/include/TTbarGen.h"
+#include "UHH2/core/include/GenTopJet.h"
+
+#include <iostream>
+
+using namespace std;
+using namespace uhh2;
+
+class RemoveLepton: public uhh2::AnalysisModule{
+
+ public:
+   RemoveLepton(uhh2::Context & ctx, const std::string & name_jet);
+   virtual bool process(uhh2::Event & ) override;
+
+ private:
+   uhh2::Event::Handle<std::vector<TopJet>>h_topjets;
+};
+
+
+class RemoveLeptonGen: public uhh2::AnalysisModule{
+
+ public:
+   RemoveLeptonGen(uhh2::Context & ctx, const std::string & name_jet);
+   virtual bool process(uhh2::Event & ) override;
+
+ private:
+   uhh2::Event::Handle<std::vector<GenTopJet>>h_topjets;
+   uhh2::Event::Handle<TTbarGen> h_ttbargen;
+};

src/AnalysisOutput.cxx

@@ -0,0 +1,15 @@
+#include <UHH2/MTopJet_UL/include/AnalysisOutput.h>
+
+using namespace std;
+
+WriteOutput::WriteOutput(uhh2::Context & ctx):
+  h_weight(ctx.declare_event_output<double>("weight")) {}
+
+bool WriteOutput::process(uhh2::Event & event){
+
+  double weight = event.weight;
+
+  event.set(h_weight, weight);
+
+  return true;
+}

src/GenSelections.cxx

@@ -1,5 +1,4 @@
 #include <UHH2/MTopJet_UL/include/GenSelections.h>
-#include <iostream>