STAR_2006_S6860818.cc

Go to the documentation of this file.

// -*- C++ -*-
#include "Rivet/Analysis.hh"
#include "Rivet/Tools/Logging.hh"
#include "Rivet/Projections/ChargedFinalState.hh"
#include "Rivet/Projections/IdentifiedFinalState.hh"
#include "Rivet/Projections/UnstableFinalState.hh"
#include "Rivet/RivetAIDA.hh"

namespace Rivet {

  /// @brief strange particle spectra in pp at 200 GeV
  class STAR_2006_S6860818 : public Analysis {
  public:

    /// Constructor
    STAR_2006_S6860818()
      : Analysis("STAR_2006_S6860818"),
        _sumWeightSelected(0.0)
    {
      setBeams(PROTON, PROTON);
      for (size_t i=0 ; i<4 ; i++) {
        _nBaryon[i]=0;
        _nAntiBaryon[i]=0;
        _nWeightedBaryon[i]=0.;
        _nWeightedAntiBaryon[i]=0.;
      }
    }

    /// Book projections and histograms
    void init() {
      ChargedFinalState bbc1(-5.0,-3.5, 0.0*GeV); // beam-beam-counter trigger
      ChargedFinalState bbc2( 3.5, 5.0, 0.0*GeV); // beam-beam-counter trigger
      addProjection(bbc1, "BBC1");
      addProjection(bbc2, "BBC2");

      UnstableFinalState ufs(-2.5, 2.5, 0.0*GeV);
      addProjection(ufs, "UFS");

      _h_pT_k0s        = bookHistogram1D(1, 1, 1);
      _h_pT_kminus     = bookHistogram1D(1, 2, 1);
      _h_pT_kplus      = bookHistogram1D(1, 3, 1);
      _h_pT_lambda     = bookHistogram1D(1, 4, 1);
      _h_pT_lambdabar  = bookHistogram1D(1, 5, 1);
      _h_pT_ximinus    = bookHistogram1D(1, 6, 1);
      _h_pT_xiplus     = bookHistogram1D(1, 7, 1);
      //_h_pT_omega      = bookHistogram1D(1, 8, 1);
      _h_antibaryon_baryon_ratio = bookDataPointSet(2, 1, 1);
      _h_pT_vs_mass    = bookProfile1D(3, 1, 1);
    }


    /// Do the analysis
    void analyze(const Event& event) {
      const ChargedFinalState& bbc1 = applyProjection<ChargedFinalState>(event, "BBC1");
      const ChargedFinalState& bbc2 = applyProjection<ChargedFinalState>(event, "BBC2");
      if (bbc1.size()<1 || bbc2.size()<1) {
        getLog() << Log::DEBUG << "Failed beam-beam-counter trigger" << std::endl;
        vetoEvent;
      }

      const double weight = event.weight();

      const UnstableFinalState& ufs = applyProjection<UnstableFinalState>(event, "UFS");
      foreach (const Particle& p, ufs.particles()) {
        if (fabs(p.momentum().rapidity()) < 0.5) {
          const PdgId pid = p.pdgId();
          const double pT = p.momentum().pT() / GeV;
          switch (abs(pid)) {
            case PIPLUS:
              if (pid < 0) _h_pT_vs_mass->fill(0.1396, pT, weight);
              break;
            case PROTON:
              if (pid < 0) _h_pT_vs_mass->fill(0.9383, pT, weight);
              if (pT > 0.4) {
                pid > 0 ? _nBaryon[0]++ : _nAntiBaryon[0]++;
                pid > 0 ? _nWeightedBaryon[0]+=weight : _nWeightedAntiBaryon[0]+=weight;
              }
              break;
            case K0S:
              if (pT > 0.2) {
                _h_pT_k0s->fill(pT, weight/pT);
              }
              _h_pT_vs_mass->fill(0.5056, pT, weight);
              break;
            case K0L:
              _h_pT_vs_mass->fill(0.5056, pT, weight);
              break;
            case 113: // rho0(770)
              _h_pT_vs_mass->fill(0.7755, pT, weight);
              break;
            case 313: // K0*(892)
              _h_pT_vs_mass->fill(0.8960, pT, weight);
              break;
            case 333: // phi(1020)
              _h_pT_vs_mass->fill(1.0190, pT, weight);
              break;
            case 3214: // Sigma(1385)
              _h_pT_vs_mass->fill(1.3840, pT, weight);
              break;
            case 3124: // Lambda(1520)
              _h_pT_vs_mass->fill(1.5200, pT, weight);
              break;
            case KPLUS:
              if (pid < 0) _h_pT_vs_mass->fill(0.4856, pT, weight);
              if (pT > 0.2) {
                pid > 0 ? _h_pT_kplus->fill(pT, weight/pT) : _h_pT_kminus->fill(pT, weight/pT);
              }
              break;
            case LAMBDA:
              pid > 0 ? _h_pT_vs_mass->fill(1.1050, pT, weight) : _h_pT_vs_mass->fill(1.1250, pT, weight);
              if (pT > 0.3) {
                pid > 0 ? _h_pT_lambda->fill(pT, weight/pT) : _h_pT_lambdabar->fill(pT, weight/pT);
                pid > 0 ? _nBaryon[1]++ : _nAntiBaryon[1]++;
                pid > 0 ? _nWeightedBaryon[1]+=weight : _nWeightedAntiBaryon[1]+=weight;
              }
              break;
            case XIMINUS:
              pid > 0 ? _h_pT_vs_mass->fill(1.3120, pT, weight) : _h_pT_vs_mass->fill(1.3320, pT, weight);
              if (pT > 0.5) {
                pid > 0 ? _h_pT_ximinus->fill(pT, weight/pT) : _h_pT_xiplus->fill(pT, weight/pT);
                pid > 0 ? _nBaryon[2]++ : _nAntiBaryon[2]++;
                pid > 0 ? _nWeightedBaryon[2]+=weight : _nWeightedAntiBaryon[2]+=weight;
              }
              break;
            case OMEGAMINUS:
              _h_pT_vs_mass->fill(1.6720, pT, weight);
              if (pT > 0.5) {
                //_h_pT_omega->fill(pT, weight/pT);
                pid > 0 ? _nBaryon[3]++ : _nAntiBaryon[3]++;
                pid > 0 ? _nWeightedBaryon[3]+=weight : _nWeightedAntiBaryon[3]+=weight;
              }
              break;
          }

        }
      }

      _sumWeightSelected += event.weight();
    }


    /// Finalize
    void finalize() {
      std::vector<double> xval;
      std::vector<double> xerr;
      std::vector<double> yval;
      std::vector<double> yerr;
      for (size_t i=0 ; i<4 ; i++) {
        xval.push_back(i);
        xerr.push_back(0.5);
        if (_nWeightedBaryon[i]==0 || _nWeightedAntiBaryon[i]==0) {
          yval.push_back(0);
          yerr.push_back(0);
        }
        else {
          double y  = _nWeightedAntiBaryon[i]/_nWeightedBaryon[i];
          double dy = sqrt( 1./_nAntiBaryon[i] + 1./_nBaryon[i] );
          yval.push_back(y);
          yerr.push_back(y*dy);
        }
      }
      _h_antibaryon_baryon_ratio->setCoordinate(0, xval, xerr);
      _h_antibaryon_baryon_ratio->setCoordinate(1, yval, yerr);

      AIDA::IHistogramFactory& hf = histogramFactory();
      const string dir = histoDir();
      hf.divide(dir + "/d02-x02-y01", *_h_pT_lambdabar, *_h_pT_lambda);
      hf.divide(dir + "/d02-x03-y01", *_h_pT_xiplus, *_h_pT_ximinus);

      scale(_h_pT_k0s,       1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_kminus,    1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_kplus,     1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_lambda,    1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_lambdabar, 1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_ximinus,   1./(2*M_PI*_sumWeightSelected));
      scale(_h_pT_xiplus,    1./(2*M_PI*_sumWeightSelected));
      //scale(_h_pT_omega,     1./(2*M_PI*_sumWeightSelected));
      getLog() << Log::DEBUG << "sumOfWeights()     = " << sumOfWeights() << std::endl;
      getLog() << Log::DEBUG << "_sumWeightSelected = " << _sumWeightSelected << std::endl;
    }

  private:

    double _sumWeightSelected;
    int _nBaryon[4];
    int _nAntiBaryon[4];
    double _nWeightedBaryon[4];
    double _nWeightedAntiBaryon[4];

    AIDA::IHistogram1D * _h_pT_k0s;
    AIDA::IHistogram1D * _h_pT_kminus;
    AIDA::IHistogram1D * _h_pT_kplus;
    AIDA::IHistogram1D * _h_pT_lambda;
    AIDA::IHistogram1D * _h_pT_lambdabar;
    AIDA::IHistogram1D * _h_pT_ximinus;
    AIDA::IHistogram1D * _h_pT_xiplus;
    //AIDA::IHistogram1D * _h_pT_omega;
    AIDA::IDataPointSet* _h_antibaryon_baryon_ratio;
    AIDA::IProfile1D*    _h_pT_vs_mass;
  };



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

}