EventHarvester.h

Go to the documentation of this file.

/*
 *  CepGen: a central exclusive processes event generator
 *  Copyright (C) 2019-2024  Laurent Forthomme
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#include <fstream>
 
#include "CepGen/EventFilter/EventExporter.h"
#include "CepGen/Utils/Histogram.h"
 
namespace cepgen {
  class Event;
  namespace utils {
    class Drawer;
    class EventBrowser;
  }  // namespace utils
  class EventHarvester : public EventExporter {
  public:
    explicit EventHarvester(const ParametersList&);
    virtual ~EventHarvester();
 
    static ParametersDescription description();
 
    inline void setCrossSection(const Value& cross_section) override { cross_section_ = cross_section; }
    bool operator<<(const Event&) override;
 
  private:
    void initialise() override;
    const std::unique_ptr<utils::EventBrowser> browser_;
    //--- variables definition
    const bool show_hists_, save_hists_;
    const std::string filename_;
 
    std::ofstream file_;
    std::unique_ptr<utils::Drawer> drawer_;
 
    Value cross_section_{1., 0.};  
    unsigned long num_evts_{0ul};  
    std::string proc_name_;        
 
    struct Hist1DInfo {
      std::string var;
      utils::Hist1D hist;
      bool log;
    };
    std::vector<Hist1DInfo> hists_;  
    struct Hist2DInfo {
      std::string var1, var2;
      utils::Hist2D hist;
      bool log;
    };
    std::vector<Hist2DInfo> hists2d_;  
  };
}  // namespace cepgen