Event.h

Go to the documentation of this file.

#pragma once
 
// STD
#include <vector>
 
// Darwin
#include <Variation.h>
#include <Weight.h>
 
// ROOT
#include <TRandom3.h>
 
namespace Darwin::Physics {
 
class AbstractEvent {
protected:
    virtual std::string_view weight_group () const = 0;
 
    AbstractEvent () = default;
    virtual ~AbstractEvent () = default;
 
public:
    Weights weights = {{1.,0}}; 
 
    inline double Weight (const Variation& v) const 
    {
        if (v.Group() == weight_group()) {
            namespace DP = Darwin::Physics;
            const DP::Weight& w = weights.at(v.Index());
            if (w.i == v.Bit()) return w;
        }
        return weights.front();
    }
 
    virtual void clear () = 0;
};
 
class GenEvent : public AbstractEvent {
 
    std::string_view weight_group () const override { return WeightVar; }
 
public:
    static inline const char * const WeightVar = "GenEvtWgts"; //< Name of gen weights in variations
 
    float hard_scale = 0; 
 
    GenEvent () = default;
    virtual ~GenEvent () = default;
 
    void clear () override 
    {
        hard_scale = 0;
        weights = {{1.,0}};
    }
};
 
class RecEvent : public AbstractEvent {
 
    std::string_view weight_group () const override { return WeightVar; }
 
public:
    static inline const char * const WeightVar = "RecEvtWgts"; //< Name of rec weights in variations
 
    int fill = 0,  
        runNo = 0, 
        lumi = 0;  
    unsigned long long evtNo = 0; 
 
    RecEvent () = default;
    virtual ~RecEvent () = default;
 
    void clear () override 
    {
        fill = 0;
        runNo = 0;
        lumi = 0;
        evtNo = 0;
        weights = {{1.,0}};
    }
};
 
} // end of Darwin::Physics
 
inline std::ostream& operator<< (std::ostream& s, const Darwin::Physics::GenEvent& event)
{
    return s << event.hard_scale;
}
 
inline std::ostream& operator<< (std::ostream& s, const Darwin::Physics::RecEvent& event)
{
    return s << event.fill << ' ' << event.runNo << ' ' << event.lumi << ' ' << event.evtNo;
}