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

#include "Rivet/Projections/FinalState.hh"

namespace Rivet {


  class NonPromptFinalState : public FinalState {
  public:



    // Constructor from a final state.
    NonPromptFinalState(const FinalState& fsp, bool accepttaudecays=false, bool acceptmudecays=false);

    NonPromptFinalState(const Cut& c, bool accepttaudecays=false, bool acceptmudecays=false);

    // /// Constructor from a Cut and optional FinalState.
    // NonPromptFinalState(const Cut& c, const FinalState& fsp=FinalState(),
    //                     bool accepttaudecays=false, bool acceptmudecays=false);

    DEFAULT_RIVET_PROJ_CLONE(NonPromptFinalState);


    void acceptMuonDecays(bool acc=true) { _acceptMuDecays = acc; }
    void acceptTauDecays(bool acc=true) { _acceptTauDecays = acc; }


    void project(const Event& e);

    CmpState compare(const Projection& p) const;

  private:

    bool _acceptMuDecays, _acceptTauDecays;

  };


}

#endif