WFinder.hh

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

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

namespace Rivet {


  class WFinder : public ParticleFinder {
  public:

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




    WFinder(const FinalState& inputfs,
            const Cut& leptoncuts,
            PdgId pid,
            double minmass, double maxmass,
            double missingET,
            double dRmax=0.1,
            ChargedLeptons chLeptons=PROMPTCHLEPTONS,
            ClusterPhotons clusterPhotons=CLUSTERNODECAY,
            PhotonTracking trackPhotons=NOTRACK,
            MassWindow masstype=MASS,
            double masstarget=80.4*GeV);

    WFinder(const FinalState& inputfs,
            const Cut& leptoncuts,
            PdgId pid,
            double minmass, double maxmass,
            double missingET,
            double dRmax,
            ClusterPhotons clusterPhotons,
            PhotonTracking trackPhotons=NOTRACK,
            MassWindow masstype=MASS,
            double masstarget=80.4*GeV)
      : WFinder(inputfs, leptoncuts, pid, minmass, maxmass, missingET,
                dRmax, PROMPTCHLEPTONS, clusterPhotons, trackPhotons, masstype, masstarget)
    {   }

    // /// Constructor with more convenient argument ordering and default args
    // ///
    // /// @todo Revisit NOTRACK as default?
    // WFinder(const FinalState& inputfs,
    //         const Cut& leptoncuts,
    //         PdgId pid,
    //         double minmass, double maxmass,
    //         double missingET,
    //         MassWindow masstype,
    //         double masstarget=80.4*GeV,
    //         ClusterPhotons clusterPhotons=CLUSTERNODECAY,
    //         double dRmax=0.1,
    //         PhotonTracking trackPhotons=NOTRACK)
    //   : WFinder(inputfs, leptoncuts, pid, minmass, maxmass, missingET,
    //             dRmax, clusterPhotons, trackPhotons, masstype, masstarget)
    // {    }

    DEFAULT_RIVET_PROJ_CLONE(WFinder);



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


    const Particles& constituentLeptons() const { return _leptons; }
    const Particle& constituentLepton() const { return _leptons.front(); }


    const Particles& constituentNeutrinos() const { return _neutrinos; }
    const Particle& constituentNeutrino() const { return _neutrinos.front(); }


    const VetoedFinalState& remainingFinalState() const;

    const MissingMomentum& missingMom() const;

    double mT() const {
      if (bosons().empty()) return -1;
      return Rivet::mT(constituentLepton().mom(), constituentNeutrino().mom());
    }


  protected:

    void project(const Event& e);

    int compare(const Projection& p) const;


  public:

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


  private:

    double _minmass, _maxmass, _masstarget;

    bool _useTransverseMass;

    double _etMissMin;

    PhotonTracking _trackPhotons;

    PdgId _pid;

    Particles _leptons, _neutrinos;

  };


}


#endif