ExampleTree.hh

Go to the documentation of this file.

// -*- C++ -*-
#ifndef RIVET_ExampleTree_HH
#define RIVET_ExampleTree_HH

#include "Rivet/Analysis.hh"
#include "Rivet/Projections/ChargedLeptons.hh"
#include "Rivet/Projections/TotalVisibleMomentum.hh"
#include "Rivet/Projections/KtJets.hh"
#include "Rivet/Projections/WZandh.hh"
#include "Rivet/RivetAIDA.fhh"

// Root stuff
#ifdef HAVE_ROOT
#include "TTree.h"
#include "TFile.h"
#include "TString.h"
#endif


namespace Rivet {

  /// This Analysis books and fills a ROOT tree with simulated data.
  /// Based initially on the ntuples used in Phys. Rev. D65; 096014 (2002)
  /// and JHEP05 (2007) 033.
  class ExampleTree : public Analysis {

  public:

    /// Default constructor
    inline ExampleTree()
      : _fsproj(-4.0, 4.0, 0.0), _chgleptonsproj(_fsproj), 
        _ktjetsproj(_fsproj), _wzandhproj(), _vfsproj(_fsproj),
        _totvismomproj(_fsproj)
    {
      /// Particle IDs for neutrinos and antineutrinos and LSP
      _vfsproj
        .addVetoDetail(12, 10.0, 50.0)
        .addVetoId(14)
        .addVetoId(16)
        .addVetoId(-12)
        .addVetoId(-14)
        .addVetoId(-16)
        .addVetoId(1000022);
      _totvismomproj = TotalVisibleMomentum(_vfsproj);

      addProjection(_fsproj);
      addProjection(_chgleptonsproj);
      addProjection(_ktjetsproj);
      addProjection(_wzandhproj);
      addProjection(_vfsproj);
      addProjection(_totvismomproj);
    }

    inline ~ExampleTree() {}

  public:

    /// Factory method
    static Analysis* create() { return new ExampleTree(); }

    /// Return the name of this analysis
    inline string getName() const {
      return "ExampleTree";
    }

  public:

    /// Initialise the analysis.
    void init();
    
    /// Perform the analysis for this event.
    void analyze(const Event& event);
    
    /// Finish off the analysis.
    void finalize();


  private:

    /// The FinalState projector used by this analysis.
    FinalState _fsproj;

    /// The Charged Lepton projector used by this analysis.
    ChargedLeptons _chgleptonsproj;

    /// The jet projector.
    KtJets _ktjetsproj;

    /// The vector boson projector.
    WZandh _wzandhproj;

    /// The VetoedFinalState projector used by this analysis.
    VetoedFinalState _vfsproj;

    /// The total visible momentum projector.
    TotalVisibleMomentum _totvismomproj;


#ifdef HAVE_ROOT
  private:

    /// The tree
    TTree* _rivetTree;

    /// The file for the Tree
    TFile* _treeFile;

    /// The filename
    TString _treeFileName;
#endif


  private:

    /// @name The ntuple variables.
    //@{
    /// Event number
    int _nevt;            

    /// Number of W bosons
    int _nvb;             
    /// 4 momentum of W bosons.
    float _vbvec[8][4];
    /// Type (i.e. decay mode) of W bosons.
    int _vbtype[8]; 

    /// Number of jets
    int _njet; 
    /// Four momentum of the jets
    float _vjet[50][4]; 

    /// Number of jets for which the subjet analysis was performed.
    int _nsub; 
    /// Four vector of jets for which we found subjets.
    float _sjet3[200][4];
    /// y 1->2, 2->3, 3->4, 4->5 for the above jets.
    float _ysubsj[200][4];

    /// Number of leptons
    int _nlep;
    /// Lepton types
    int _leptype[150][3];
    float _vlep[150][4];

    /// Number of partons
    int _npart; 
    float _ppart[4000][4];
    int _pid[4000];
    int _mo[4000];

    /// Total visible momentum
    float _esumr[4];
    //@}


  private:

    /// Hide the assignment operator
    ExampleTree& operator=(const ExampleTree&);

    /// Minimum pt of jets which will go into the tree.
    int _jet_pt_cut;

    /// Minimum pt of jets which will have y evaluated and stored.
    int _subj_pt_cut;

    /// Minimum pt of charged leptons which will go into the tree.
    int _lepton_pt_cut;

    /// Store the partons or not?
    bool _store_partons;
  };

}

#endif