ZFinder.hh

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

#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Projections/DressedLeptons.hh"
#include "Rivet/Projections/VetoedFinalState.hh"

namespace Rivet {


  class ZFinder : public ParticleFinder {
  public:

    enum ChargedLeptons { PROMPTCHLEPTONS=0, ALLCHLEPTONS };
    enum ClusterPhotons { NOCLUSTER=0, CLUSTERNODECAY=1, CLUSTERALL };
    enum PhotonTracking { NOTRACK=0, TRACK=1 };



    ZFinder(const FinalState& inputfs,
            const Cut& cuts,
            PdgId pid,
            double minmass, double maxmass,
            double dRmax=0.1,
            ChargedLeptons chLeptons=PROMPTCHLEPTONS,
            ClusterPhotons clusterPhotons=CLUSTERNODECAY,
            PhotonTracking trackPhotons=NOTRACK,
            double masstarget=91.2*GeV);

    ZFinder(const FinalState& inputfs,
            const Cut& cuts,
            PdgId pid,
            double minmass, double maxmass,
            double dRmax,
            ClusterPhotons clusterPhotons,
            PhotonTracking trackPhotons=NOTRACK,
            double masstarget=91.2*GeV)
      : ZFinder(inputfs, cuts, pid, minmass, maxmass,
                dRmax, PROMPTCHLEPTONS, clusterPhotons, trackPhotons, masstarget)
    {   }


    DEFAULT_RIVET_PROJ_CLONE(ZFinder);



    const Particles& bosons() const { return particles(); }
    const Particle& boson() const { return bosons().front(); }


    const Particles & constituentLeptons() const;
    const Particles & constituents() const { return constituentLeptons(); }

    const VetoedFinalState& remainingFinalState() const;


  protected:

    void project(const Event& e);

    int compare(const Projection& p) const;


  public:

    void clear() { _theParticles.clear(); }


  private:

    double _minmass, _maxmass, _masstarget;

    PhotonTracking _trackPhotons;

    PdgId _pid;

  };


}

#endif