ZFinder.hh

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

#include "Rivet/Tools/Logging.hh"
#include "Rivet/Rivet.hh"
#include "Rivet/Particle.hh"
#include "Rivet/Event.hh"
#include "Rivet/Projection.hh"
#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Projections/LeptonClusters.hh"

namespace Rivet {


  class ZFinder : public FinalState {

  public:



    ZFinder(const FinalState& inputfs,
            double etaMin, double etaMax,
            double pTmin,
            PdgId pid,
            double minmass, double maxmass,
            double dRmax, bool clusterPhotons, bool trackPhotons,
            double masstarget=91.2*GeV);


    ZFinder(const FinalState& inputfs,
            const std::vector<std::pair<double, double> >& etaRanges,
            double pTmin,
            PdgId pid,
            double minmass, const double maxmass,
            double dRmax, bool clusterPhotons, bool trackPhotons,
            double masstarget=91.2*GeV);


    ZFinder(double, double, double, PdgId, double, double, double,
            bool, bool, double masstarget=91.2*GeV);
    ZFinder(const std::vector<std::pair<double, double> >&, double, PdgId,
            double, double, double, bool, bool, double masstarget=91.2*GeV);


    virtual const Projection* clone() const {
      return new ZFinder(*this);
    }


    const ParticleVector& bosons() const { return _bosons; }

    const vector<Particle>& constituents() const { return _constituents; }

    const FinalState& remainingFinalState() const;


  protected:

    void project(const Event& e);

    int compare(const Projection& p) const;


  public:

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


  private:
    void _init(const FinalState& inputfs,
               const std::vector<std::pair<double, double> >& etaRanges,
               double pTmin,  PdgId pid,
               double minmass, double maxmass,
               double dRmax, bool clusterPhotons, bool trackPhotons,
               double masstarget);

    double _minmass, _maxmass, _masstarget;

    bool _trackPhotons;

    PdgId _pid;

    ParticleVector _bosons;

    vector<Particle> _constituents;

  };


}



#endif