LossyFinalState.hh

// -*- C++ -*-
#ifndef RIVET_LossyFinalState_HH
#define RIVET_LossyFinalState_HH
 
#include "Rivet/Event.hh"
#include "Rivet/Particle.hh"
#include "Rivet/Projection.hh"
#include "Rivet/Config/RivetCommon.hh"
#include "Rivet/Projections/FinalState.hh"
#include "Rivet/Tools/Logging.hh"
 
namespace Rivet {
 

  template <typename FILTER>
  class LossyFinalState : public FinalState {
  public:


    LossyFinalState(const FinalState& fsp, FILTER filter)
        : _filter(filter) {
      setName("LossyFinalState");
      declare(fsp, "FS");
    }

    LossyFinalState(FILTER filter, const Cut& c = Cuts::open())
        : _filter(filter) {
      setName("LossyFinalState");
      declare(FinalState(c), "FS");
    }

    virtual ~LossyFinalState() { }

    RIVET_DEFAULT_PROJ_CLONE(LossyFinalState);


    using Projection::operator=;
 

    void project(const Event& e) {
      const FinalState& fs = apply<FinalState>(e, "FS");
      getLog() << Log::DEBUG << "Pre-loss number of FS particles = " << fs.particles().size() << '\n';
      _theParticles.clear();
      std::remove_copy_if(fs.particles().begin(), fs.particles().end(), std::back_inserter(_theParticles),
                          _filter);
      getLog() << Log::DEBUG << "Filtered number of FS particles = " << _theParticles.size() << '\n';
    }
 

    CmpState compare(const Projection& p) const {
      const LossyFinalState<FILTER>& other = pcast<LossyFinalState<FILTER>>(p);
      const CmpState fscmp = mkNamedPCmp(other, "FS");
      if (fscmp != CmpState::EQ) return fscmp;
      return _filter.compare(other._filter);
    }
 
 
  protected:

    FILTER _filter;
  };
 
 
}
 
#endif