Unweighter.hh

Go to the documentation of this file.

 
#pragma once
 
#include <functional>
#include <optional>
#include <vector>
 
namespace HEJ {
  class Event;
  struct RNG;
 
  class Unweighter {
  public:
    Unweighter(double cut):
      cut_(cut){}
    Unweighter() = default;
 
    void set_cut(double max_weight){
      cut_ = max_weight;
    }
    void set_cut_to_maxwt(std::vector<Event> const & events){
      set_cut_to_maxwt(events.cbegin(), events.cend());
    }
    template<class ConstIt>
    void set_cut_to_maxwt(ConstIt begin, ConstIt end);
 
 
    void set_cut_to_peakwt(std::vector<Event> const & events, double max_dev){
      set_cut_to_peakwt(events.cbegin(), events.cend(), max_dev);
    }
    template<class ConstIt>
    void set_cut_to_peakwt(ConstIt begin, ConstIt end, double max_dev);
 
    double get_cut() const {
      return cut_;
    }
 
    std::optional<Event> unweight(Event ev, RNG & ran) const;
    std::vector<Event> unweight(
      std::vector<Event> events, RNG & ran
    ) const;
 
    template<class Iterator>
    Iterator unweight(
      Iterator begin, Iterator end, RNG & ran
    ) const;
 
  private:
    double cut_{-1};
    bool discard(RNG & ran, Event & ev) const;
  };
} // namespace HEJ
 
// implementation of template functions
#include "HEJ/detail/Unweighter_impl.hh"