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