LeptonFinder.hh

// -*- C++ -*-
#ifndef RIVET_LeptonFinder_HH
#define RIVET_LeptonFinder_HH
 
#include "Rivet/DressedLepton.hh"
#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Projections/PromptFinalState.hh"
#include "Rivet/Projections/IdentifiedFinalState.hh"
#include "Rivet/Config/RivetCommon.hh"
 
namespace Rivet {
  
 
  class LeptonFinder : public FinalState {
  public:
 
    LeptonFinder(const FinalState& allfs,
                 double dRdress,
                 const Cut& cut=Cuts::OPEN,
         DressingType dressing=DressingType::CONE)
      : LeptonFinder(allfs, allfs, dRdress, cut, dressing)
    {     }
 
 
    LeptonFinder(const FinalState& allfs,
         const Cut& cut,
                 double dRdress,
         DressingType dressing=DressingType::CONE)
      : LeptonFinder(allfs, dRdress, cut, dressing)
    {   }
 
 
    LeptonFinder(const Cut& cut,
         double dRdress,
         LeptonOrigin whichleptons=LeptonOrigin::NODECAY,
         PhotonOrigin whichphotons=PhotonOrigin::NODECAY,
                 TauDecaysAs tauDecays=TauDecaysAs::PROMPT,
                 MuDecaysAs muDecays=MuDecaysAs::PROMPT,
         DressingType dressing=DressingType::CONE)
      : LeptonFinder(dRdress, cut, whichleptons, whichphotons, tauDecays, muDecays, dressing)
    {   }
 
 
    LeptonFinder(const Cut& cut,
         double dRdress,
         DressingType dressing,
         LeptonOrigin whichleptons=LeptonOrigin::NODECAY,
         PhotonOrigin whichphotons=PhotonOrigin::NODECAY,
                 TauDecaysAs tauDecays=TauDecaysAs::PROMPT,
                 MuDecaysAs muDecays=MuDecaysAs::PROMPT)
      : LeptonFinder(dRdress, cut, whichleptons, whichphotons, tauDecays, muDecays, dressing)
    {   }
 
 
    LeptonFinder(double dRdress, const Cut& cut=Cuts::OPEN,
         LeptonOrigin whichleptons=LeptonOrigin::NODECAY,
         PhotonOrigin whichphotons=PhotonOrigin::NODECAY,
                 TauDecaysAs tauDecays=TauDecaysAs::PROMPT,
                 MuDecaysAs muDecays=MuDecaysAs::PROMPT,
         DressingType dressing=DressingType::CONE);
 
 
    LeptonFinder(double dRdress, const Cut& cut,
         DressingType dressing,
         LeptonOrigin whichleptons=LeptonOrigin::NODECAY,
         PhotonOrigin whichphotons=PhotonOrigin::NODECAY,
                 TauDecaysAs tauDecays=TauDecaysAs::PROMPT,
                 MuDecaysAs muDecays=MuDecaysAs::PROMPT)
      : LeptonFinder(dRdress, cut, whichleptons, whichphotons, tauDecays, muDecays, dressing)
    {   }
 
 
    LeptonFinder(const FinalState& leptonfs, const FinalState& photonfs,
         double dRdress, const Cut& cut=Cuts::OPEN,
         DressingType dressing=DressingType::CONE);
 
 
    LeptonFinder(const FinalState& leptonfs, const FinalState& photonfs,
         const Cut& cut, double dRdress, DressingType dressing=DressingType::CONE)
      : LeptonFinder(leptonfs, photonfs, dRdress, cut, dressing)
    {   }
 
 
    RIVET_DEFAULT_PROJ_CLONE(LeptonFinder);
 
    using Projection::operator =;
 
 
    DressedLeptons dressedLeptons() const {
      DressedLeptons rtn;
      for (const Particle& p : particles(cmpMomByPt))
        rtn += DressedLepton(p);  //static_cast<const DressedLepton>(p);
      return rtn;
    }
 
    DressedLeptons dressedLeptons(const ParticleSorter& sorter) const {
      DressedLeptons rtn;
      for (const Particle& p : particles(sorter))
        rtn += DressedLepton(p);  //static_cast<const DressedLepton>(p);
      return rtn;
    }
 
 
  protected:
 
    void project(const Event& e);
 
    CmpState compare(const Projection& p) const;
 
 
  protected:
 
    double _dRdress;
 
    DressingType _dressMode;
 
  };
 
 
}
 
#endif