LHEEventProduct.h

Go to the documentation of this file.

#ifndef SimDataFormats_GeneratorProducts_LHEEventProduct_h
#define SimDataFormats_GeneratorProducts_LHEEventProduct_h

#include <memory>
#include <vector>
#include <string>

#include "SimDataFormats/GeneratorProducts/interface/LesHouches.h"
#include "SimDataFormats/GeneratorProducts/interface/PdfInfo.h"
#include "SimDataFormats/GeneratorProducts/interface/WeightsInfo.h"

class LHEEventProduct {
    public:
    typedef gen::PdfInfo PDF;
    typedef gen::WeightsInfo WGT;

    typedef std::vector<std::string>::const_iterator
                        comments_const_iterator;
    typedef std::vector<std::string>::size_type size_type;

    LHEEventProduct() {}
        LHEEventProduct(const lhef::HEPEUP &hepeup) : 
      hepeup_(hepeup), originalXWGTUP_(0) {}
    LHEEventProduct(const lhef::HEPEUP &hepeup,
            const double originalXWGTUP) : 
      hepeup_(hepeup), originalXWGTUP_(originalXWGTUP) {}
    LHEEventProduct(LHEEventProduct&& other) = default;

    LHEEventProduct& operator=(LHEEventProduct&& other) =default;

    ~LHEEventProduct() =default;

    void setPDF(const PDF &pdf) { pdf_.reset(new PDF(pdf)); }
    void addWeight(const WGT& wgt) {      
      weights_.push_back(wgt);
    }
    void addComment(const std::string &line) { comments_.push_back(line); }

    double originalXWGTUP() const { return originalXWGTUP_; }
    const std::vector<WGT>& weights() const { return weights_; }

    const std::vector<float> &scales() const { return scales_; }
    void setScales(const std::vector<float> &scales) { scales_ = scales; }
    
        int npLO() const { return npLO_; }
        int npNLO() const { return npNLO_; }
        
        void setNpLO(int n) { npLO_ = n; }
        void setNpNLO(int n) { npNLO_ = n; }    
    
    const lhef::HEPEUP &hepeup() const { return hepeup_; }
    const PDF *pdf() const { return pdf_.get(); }

    size_type comments_size() const { return comments_.size(); }
    comments_const_iterator comments_begin() const { return comments_.begin(); }
    comments_const_iterator comments_end() const { return comments_.end(); }
  
  const char* getComment(unsigned i) const {
    if(comments_.empty() || i>=comments_.size()) return "";
    else return (const char*) comments_[i].c_str();
  }

    class const_iterator {
        public:
        typedef std::forward_iterator_tag   iterator_category;
        typedef std::string         value_type;
        typedef std::ptrdiff_t          difference_type;
        typedef std::string         *pointer;
        typedef std::string         &reference;

        const_iterator() : line(npos) {}
        ~const_iterator() {}

        inline bool operator == (const const_iterator &other) const
        { return line == other.line; }
        inline bool operator != (const const_iterator &other) const
        { return !operator == (other); }

        inline const_iterator &operator ++ ()
        { next(); return *this; }
        inline const_iterator operator ++ (int dummy)
        { const_iterator orig = *this; next(); return orig; }

        const std::string &operator * () const { return tmp; }
        const std::string *operator -> () const { return &tmp; }

        private:
        friend class LHEEventProduct;

        void next();

        const LHEEventProduct   *event;
        unsigned int        line;
        std::string     tmp;

        static const unsigned int npos = 99999;
    };

    const_iterator begin() const;
    inline const_iterator end() const { return const_iterator(); }

    private:
    lhef::HEPEUP            hepeup_;
    std::vector<std::string>    comments_;
    std::unique_ptr<PDF>        pdf_;
    std::vector<WGT>                weights_;
    double                          originalXWGTUP_;
        std::vector<float>              scales_; //scale value used to exclude EWK-produced partons from matching
        int                             npLO_; //number of partons for LO process (used to steer matching/merging)
        int                             npNLO_; //number of partons for NLO process (used to steer matching/merging)
};

#endif // GeneratorEvent_LHEInterface_LHEEventProduct_h