ATLAS_2010_S8918562.cc

Go to the documentation of this file.

// -*- C++ -*-

#include "Rivet/Analysis.hh"
#include "Rivet/RivetAIDA.hh"
#include "Rivet/Projections/ChargedFinalState.hh"
#include "Rivet/Tools/Logging.hh"

namespace Rivet {


  /// Rivet analysis class for ATLAS_2010_S8918562 dataset
  class ATLAS_2010_S8918562 : public Analysis {
  public:

    /// Helper for collectively filling Nch, pT, eta, and pT vs. Nch histograms
    void fillPtEtaNch(const ChargedFinalState& cfs, int nchcut, double weight,
                      AIDA::IHistogram1D* h_nch, AIDA::IHistogram1D* h_pt,
                      AIDA::IHistogram1D* h_eta, AIDA::IProfile1D* h_ptnch = 0) {

      // Get number of particles and skip if event fails cut
      const int nch = cfs.size();
      if (nch < nchcut) return;

      // Fill nch
      h_nch->fill(nch, weight);
      // Loop over particles, fill pT, eta and ptnch
      foreach (const Particle& p, cfs.particles()) {
        const double pt = p.momentum().pT();
        h_pt->fill(pt/GeV, weight/pt);
        h_eta->fill(p.momentum().eta(), weight);
        if (h_ptnch != 0) h_ptnch->fill(nch, pt/GeV, weight);
      }
    }


    /// Default constructor
    ATLAS_2010_S8918562() : Analysis("ATLAS_2010_S8918562") {
      _sumW_pt100_nch2 = 0;
      _sumW_pt100_nch20 = 0;
      _sumW_pt500_nch1 = 0;
      _sumW_pt500_nch6 = 0;
      _sumW_pt2500_nch1 = 0;
    }


    /// Initialization, called once before running
    void init() {
      // Projections
      const ChargedFinalState cfs100(-2.5, 2.5, 100.0*MeV);
      addProjection(cfs100, "CFS100");
      const ChargedFinalState cfs500(-2.5, 2.5, 500.0*MeV);
      addProjection(cfs500, "CFS500");
      const ChargedFinalState cfs2500(-2.5, 2.5, 2500.0*MeV);
      addProjection(cfs2500, "CFS2500");

      // Book histograms
      if (fuzzyEquals(sqrtS()/GeV, 900)) {
        _hist_pt100_nch2_nch = bookHistogram1D(18, 1, 1);
        _hist_pt100_nch2_pt = bookHistogram1D(11, 1, 1);
        _hist_pt100_nch2_eta = bookHistogram1D(4, 1, 1);
        _hist_pt100_nch2_ptnch = bookProfile1D(24, 1, 1);

        _hist_pt100_nch20_nch = bookHistogram1D(34, 1, 1);
        _hist_pt100_nch20_pt = bookHistogram1D(30, 1, 1);
        _hist_pt100_nch20_eta = bookHistogram1D(26, 1, 1);

        _hist_pt500_nch1_nch = bookHistogram1D(15, 1, 1);
        _hist_pt500_nch1_pt = bookHistogram1D(8, 1, 1);
        _hist_pt500_nch1_eta = bookHistogram1D(1, 1, 1);
        _hist_pt500_nch1_ptnch = bookProfile1D(22, 1, 1);

        _hist_pt500_nch6_nch = bookHistogram1D(20, 1, 1);
        _hist_pt500_nch6_pt = bookHistogram1D(13, 1, 1);
        _hist_pt500_nch6_eta = bookHistogram1D(6, 1, 1);

        _hist_pt2500_nch1_nch = bookHistogram1D(36, 1, 1);
        _hist_pt2500_nch1_pt = bookHistogram1D(32, 1, 1);
        _hist_pt2500_nch1_eta = bookHistogram1D(28, 1, 1);
        _hist_pt2500_nch1_ptnch = bookProfile1D(38, 1, 1);

      } else if (fuzzyEquals(sqrtS()/GeV, 2360)) {

        _hist_pt500_nch1_nch = bookHistogram1D(16, 1, 1);
        _hist_pt500_nch1_pt = bookHistogram1D(9, 1, 1);
        _hist_pt500_nch1_eta = bookHistogram1D(2, 1, 1);
        // This one histogram might be called while unbooked, so ensure its pointer is null!
        _hist_pt500_nch1_ptnch = 0;

      } else if (fuzzyEquals(sqrtS()/GeV, 7000)) {

        _hist_pt100_nch2_nch = bookHistogram1D(19, 1, 1);
        _hist_pt100_nch2_pt = bookHistogram1D(12, 1, 1);
        _hist_pt100_nch2_eta = bookHistogram1D(5, 1, 1);
        _hist_pt100_nch2_ptnch = bookProfile1D(25, 1, 1);

        _hist_pt100_nch20_nch = bookHistogram1D(35, 1, 1);
        _hist_pt100_nch20_pt = bookHistogram1D(31, 1, 1);
        _hist_pt100_nch20_eta = bookHistogram1D(27, 1, 1);

        _hist_pt500_nch1_nch = bookHistogram1D(17, 1, 1);
        _hist_pt500_nch1_pt = bookHistogram1D(10, 1, 1);
        _hist_pt500_nch1_eta = bookHistogram1D(3, 1, 1);
        _hist_pt500_nch1_ptnch = bookProfile1D(23, 1, 1);

        _hist_pt500_nch6_nch = bookHistogram1D(21, 1, 1);
        _hist_pt500_nch6_pt = bookHistogram1D(14, 1, 1);
        _hist_pt500_nch6_eta = bookHistogram1D(7, 1, 1);

        _hist_pt2500_nch1_nch = bookHistogram1D(37, 1, 1);
        _hist_pt2500_nch1_pt = bookHistogram1D(33, 1, 1);
        _hist_pt2500_nch1_eta = bookHistogram1D(29, 1, 1);
        _hist_pt2500_nch1_ptnch = bookProfile1D(39, 1, 1);

      } else {
        throw LogicError("The ATLAS_2010_S8918562 analysis is only valid for sqrt(s) = 900, 2360 and 7000 GeV!");
      }

    }



    void analyze(const Event& event) {
      const double weight = event.weight();

      // 100 GeV final states
      if (!fuzzyEquals(sqrtS()/GeV, 2360)) {
        const ChargedFinalState& cfs100 = applyProjection<ChargedFinalState>(event, "CFS100");
        // nch>=2
        if (cfs100.size() >= 2) _sumW_pt100_nch2 += weight;
        fillPtEtaNch(cfs100, 2, weight, _hist_pt100_nch2_nch, _hist_pt100_nch2_pt, _hist_pt100_nch2_eta, _hist_pt100_nch2_ptnch);
        // nch>=20
        if (cfs100.size() >= 20) _sumW_pt100_nch20 += weight;
        fillPtEtaNch(cfs100, 20, weight, _hist_pt100_nch20_nch, _hist_pt100_nch20_pt, _hist_pt100_nch20_eta);
      }

      // 500 GeV final states
      const ChargedFinalState& cfs500 = applyProjection<ChargedFinalState>(event, "CFS500");
      // nch>=1
      if (cfs500.size() >= 1) _sumW_pt500_nch1 += weight;
      fillPtEtaNch(cfs500, 1, weight, _hist_pt500_nch1_nch, _hist_pt500_nch1_pt, _hist_pt500_nch1_eta, _hist_pt500_nch1_ptnch);
      // nch>=6
      if (!fuzzyEquals(sqrtS()/GeV, 2360)) {
        if (cfs500.size() >= 6) _sumW_pt500_nch6 += weight;
        fillPtEtaNch(cfs500, 6, weight, _hist_pt500_nch6_nch, _hist_pt500_nch6_pt, _hist_pt500_nch6_eta);
      }

      // 2500 GeV final states
      if (!fuzzyEquals(sqrtS()/GeV, 2360)) {
        const ChargedFinalState& cfs2500 = applyProjection<ChargedFinalState>(event, "CFS2500");
        // nch>=1
        if (cfs2500.size() >= 1) _sumW_pt2500_nch1 += weight;
        fillPtEtaNch(cfs2500, 1, weight, _hist_pt2500_nch1_nch, _hist_pt2500_nch1_pt, _hist_pt2500_nch1_eta, _hist_pt2500_nch1_ptnch);
      }

    }



    void finalize() {

      if (!fuzzyEquals(sqrtS()/GeV, 2360)) {
        if (_sumW_pt100_nch2 > 0) {
          _hist_pt100_nch2_nch->scale(1.0/_sumW_pt100_nch2);
          _hist_pt100_nch2_pt->scale(1.0/_sumW_pt100_nch2/TWOPI/5);
          _hist_pt100_nch2_eta->scale(1.0/_sumW_pt100_nch2);
        }

        if (_sumW_pt100_nch20 > 0) {
          _hist_pt100_nch20_nch->scale(1.0/_sumW_pt100_nch20);
          _hist_pt100_nch20_pt->scale(1.0/_sumW_pt100_nch20/TWOPI/5);
          _hist_pt100_nch20_eta->scale(1.0/_sumW_pt100_nch20);
        }

        if (_sumW_pt500_nch6 > 0) {
          _hist_pt500_nch6_nch->scale(1.0/_sumW_pt500_nch6);
          _hist_pt500_nch6_pt->scale(1.0/_sumW_pt500_nch6/TWOPI/5);
          _hist_pt500_nch6_eta->scale(1.0/_sumW_pt500_nch6);
        }

        if (_sumW_pt2500_nch1 > 0) {
          _hist_pt2500_nch1_nch->scale(1.0/_sumW_pt2500_nch1);
          _hist_pt2500_nch1_pt->scale(1.0/_sumW_pt2500_nch1/TWOPI/5);
          _hist_pt2500_nch1_eta->scale(1.0/_sumW_pt2500_nch1);
        }
      }

      if (_sumW_pt500_nch1 > 0) {
        _hist_pt500_nch1_nch->scale(1.0/_sumW_pt500_nch1);
        _hist_pt500_nch1_pt->scale(1.0/_sumW_pt500_nch1/TWOPI/5);
        _hist_pt500_nch1_eta->scale(1.0/_sumW_pt500_nch1);
      }
    }


  private:

    double _sumW_pt100_nch2, _sumW_pt100_nch20, _sumW_pt500_nch1,
      _sumW_pt500_nch6, _sumW_pt2500_nch1;

    AIDA::IHistogram1D *_hist_pt100_nch2_nch,
      *_hist_pt100_nch20_nch, *_hist_pt500_nch1_nch,
      *_hist_pt500_nch6_nch, *_hist_pt2500_nch1_nch;

    AIDA::IHistogram1D *_hist_pt100_nch2_pt,
      *_hist_pt100_nch20_pt, *_hist_pt500_nch1_pt,
      *_hist_pt500_nch6_pt, *_hist_pt2500_nch1_pt;

    AIDA::IHistogram1D *_hist_pt100_nch2_eta,
      *_hist_pt100_nch20_eta, *_hist_pt500_nch1_eta,
      *_hist_pt500_nch6_eta, *_hist_pt2500_nch1_eta;

    AIDA::IProfile1D *_hist_pt100_nch2_ptnch,
      *_hist_pt500_nch1_ptnch, *_hist_pt2500_nch1_ptnch;

  };



  // This global object acts as a hook for the plugin system
  AnalysisBuilder<ATLAS_2010_S8918562> plugin_ATLAS_2010_S8918562;

}