Event.hh

Go to the documentation of this file.

#pragma once

#include <map>

#include "boca/generic/Mutable.hh"
#include "boca/generic/Flag.hh"
#include "boca/fastjet/Jet.hh"
#include "boca/fastjet/Particle.hh"

namespace boca
{

enum class JetDetail
{
    none = 0,
    plain = 1 << 0,
    structure = 1 << 1,
    isolation = 1 << 2,
};

template<>
struct Flag<JetDetail> {
    static const bool enable = true;
};

std::string Name(JetDetail jet_detail);

enum class Decay
{
    leptonic,
    hadronic
};

std::string Name(Decay decay);

enum class Status
{
    none = 0,
    stable = 1,
    unstable = 2,
    generator = 3
};

class TreeReader;

class Event
{

public:

    Event(boca::TreeReader const& tree_reader);

    std::vector<Particle> GenParticles() const;

    std::vector<Particle> Particles() const;

    std::vector<Lepton> Leptons() const;

    std::vector<Lepton> Electrons() const;

    std::vector<Lepton> Muons() const;

    std::vector<Photon> Photons() const;

    std::vector<Jet> Jets() const;

    std::vector<Jet> EFlow(JetDetail jet_detail) const;

    boca::MissingEt MissingEt() const;

    Momentum ScalarHt() const;

    Event(Event const&) = default;

    Event(Event &&) = default;

    Event &operator=(Event const&) & = default;

    Event &operator=(Event &&) & = default;

protected:

   ~Event() {}

    boca::TreeReader const& TreeReader() const;

    virtual std::vector<Particle> GetParticles(Status max_status) const = 0;

    virtual std::vector<Lepton> GetElectrons() const = 0;

    virtual std::vector<Lepton> GetMuons() const = 0;

    virtual std::vector<Photon> GetPhotons() const = 0;

    virtual std::vector<Jet> GetJets() const = 0;

    virtual std::vector<Jet> GetEFlow(JetDetail jet_detail) const = 0;

    virtual boca::MissingEt GetMissingEt() const = 0;

    virtual Momentum GetScalarHt() const = 0;

private:

    boca::TreeReader const* tree_reader_;

    mutable std::map<Status, Mutable<std::vector<Particle>>> particles_;

    Mutable<std::vector<Lepton>> electrons_;

    Mutable<std::vector<Lepton>> muons_;

    Mutable<std::vector<Photon>> photons_;

    Mutable<std::vector<Jet>> jets_;

    Mutable<std::vector<Jet>> eflow_;

    Mutable<boca::MissingEt> missing_et_;

    Mutable<Momentum> scalar_ht_;

};

}