1#include "Rivet/Analysis.hh"
  2#include "Rivet/Projections/FinalState.hh"
  3#include "Rivet/Projections/FastJets.hh"
  4
  5namespace Rivet {
  6
  7  class ATLAS_2016_I1419070 : public Analysis {
  8  public:
  9
 10    /// Constructor
 11    ATLAS_2016_I1419070() : Analysis("ATLAS_2016_I1419070")
 12    {  }
 13
 14  public:
 15
 16    void init() {
 17
 18      declare(FastJets(FinalState(), JetAlg::ANTIKT, 0.4), "Jets");
 19
 20      book(forward_500MeV ,1, 1, 1);
 21      book(forward_2GeV   ,2, 1, 1);
 22      book(forward_5GeV   ,3, 1, 1);
 23
 24      book(central_500MeV ,4, 1, 1);
 25      book(central_2GeV   ,5, 1, 1);
 26      book(central_5GeV   ,6, 1, 1);
 27
 28      book(diff_500MeV, "d07-x01-y01");
 29      book(diff_2GeV  , "d08-x01-y01");
 30      book(diff_5GeV  , "d09-x01-y01");
 31
 32      book(sum_500MeV, "d10-x01-y01");
 33      book(sum_2GeV  , "d11-x01-y01");
 34      book(sum_5GeV  , "d12-x01-y01");
 35
 36    }
 37
 38    /// Perform the per-event analysis
 39    void analyze(const Event& event) {
 40      Jets m_goodJets = apply<JetFinder>(event, "Jets").jetsByPt(Cuts::pT > 25*GeV && Cuts::abseta < 2.1);
 41
 42      if (m_goodJets.size() < 2)        vetoEvent;
 43      if (m_goodJets[0].pT() < 50*GeV)  vetoEvent;
 44      if (m_goodJets[1].pT() < 50*GeV)  vetoEvent;
 45      if (fabs(1.0 - m_goodJets[0].pT()/m_goodJets[1].pT()) > 0.5)  vetoEvent;
 46
 47      bool check = m_goodJets[0].abseta() < m_goodJets[1].abseta();
 48      int pos_f = int(check);
 49      int pos_c = int(!check);
 50
 51      double pt500MeV_f = CalculateNCharge(m_goodJets[pos_f], 0.5);
 52      double pt2GeV_f   = CalculateNCharge(m_goodJets[pos_f], 2.0);
 53      double pt5GeV_f   = CalculateNCharge(m_goodJets[pos_f], 5.0);
 54      double pT_f = m_goodJets[pos_f].pT();
 55
 56      double pt500MeV_c = CalculateNCharge(m_goodJets[pos_c], 0.5);
 57      double pt2GeV_c   = CalculateNCharge(m_goodJets[pos_c], 2.0);
 58      double pt5GeV_c   = CalculateNCharge(m_goodJets[pos_c], 5.0);
 59      double pT_c = m_goodJets[pos_c].pT();
 60
 61      forward_500MeV->fill(pT_f, pt500MeV_f);
 62      forward_2GeV->fill(  pT_f, pt2GeV_f);
 63      forward_5GeV->fill(  pT_f, pt5GeV_f);
 64
 65      central_500MeV->fill(pT_c, pt500MeV_c);
 66      central_2GeV->fill(  pT_c, pt2GeV_c);
 67      central_5GeV->fill(  pT_c, pt5GeV_c);
 68    }
 69
 70    double CalculateNCharge(Jet& jet, double pTcut=0.5) {
 71      unsigned int ncharge = 0;
 72      for (const Particle& p : jet.particles()) {
 73        if (p.pT() < pTcut)  continue;
 74        if (p.charge3())  ++ncharge;
 75      }
 76      if (ncharge > 60)  ncharge = 60;
 77      return double(ncharge);
 78    }
 79
 80    /// Normalise histograms etc., after the run
 81    void finalize() {
 82
 83      if (numEvents() > 2) {
 84        for (unsigned int i = 1; i < forward_500MeV->numBins()+1; ++i) {
 85          const YODA::Dbn2D& bsum  = central_500MeV->bin(i) + forward_500MeV->bin(i);
 86          const YODA::Dbn2D& bsum2 = central_2GeV->bin(i) + forward_2GeV->bin(i);
 87          const YODA::Dbn2D& bsum5 = central_5GeV->bin(i) + forward_5GeV->bin(i);
 88
 89          double ydiff  = central_500MeV->bin(i).effNumEntries()? central_500MeV->bin(i).yMean() : 0.0;
 90          double ydiff2 = central_2GeV->bin(i).effNumEntries()?   central_2GeV->bin(i).yMean()   : 0.0;
 91          double ydiff5 = central_5GeV->bin(i).effNumEntries()?   central_5GeV->bin(i).yMean()   : 0.0;
 92          ydiff  -= forward_500MeV->bin(i).effNumEntries()? forward_500MeV->bin(i).yMean() : 0.0;
 93          ydiff2 -= forward_2GeV->bin(i).effNumEntries()?   forward_2GeV->bin(i).yMean()   : 0.0;
 94          ydiff5 -= forward_5GeV->bin(i).effNumEntries()?   forward_5GeV->bin(i).yMean()   : 0.0;
 95
 96          double yerr  = bsum.effNumEntries()  > 1.0 ?  bsum.yStdErr() : 0.0;
 97          double yerr2 = bsum2.effNumEntries() > 1.0 ? bsum2.yStdErr() : 0.0;
 98          double yerr5 = bsum5.effNumEntries() > 1.0 ? bsum5.yStdErr() : 0.0;
 99
100          diff_500MeV->bin(i).set(ydiff, yerr);
101          diff_2GeV->bin(i).set(ydiff2, yerr2);
102          diff_5GeV->bin(i).set(ydiff5, yerr5);
103
104          sum_500MeV->bin(i).set(bsum.effNumEntries()? bsum.yMean() : 0.0, yerr);
105          sum_2GeV->bin(i).set(bsum2.effNumEntries()? bsum2.yMean() : 0.0, yerr2);
106          sum_5GeV->bin(i).set(bsum5.effNumEntries()? bsum5.yMean() : 0.0, yerr5);
107        }
108      }
109
110    }
111
112
113  private:
114
115    Profile1DPtr forward_500MeV;
116    Profile1DPtr forward_2GeV;
117    Profile1DPtr forward_5GeV;
118
119    Profile1DPtr central_500MeV;
120    Profile1DPtr central_2GeV;
121    Profile1DPtr central_5GeV;
122
123    Estimate1DPtr sum_500MeV;
124    Estimate1DPtr sum_2GeV;
125    Estimate1DPtr sum_5GeV;
126
127    Estimate1DPtr diff_500MeV;
128    Estimate1DPtr diff_2GeV;
129    Estimate1DPtr diff_5GeV;
130
131  };
132
133  RIVET_DECLARE_PLUGIN(ATLAS_2016_I1419070);
134}