selection.hpp

Go to the documentation of this file.

#ifndef SELECTION_HPP
#define SELECTION_HPP

#include <vector>

#include "filval/filval.hpp"
#include "filval_root/filval_root.hpp"

#include "analysis/common/obj_types.hpp"

struct EventSelection{
    ObsFilter* trilepton;
    ObsFilter* b_jet3;
    ObsFilter* z_mass_veto;

    ObsFilter* base_sel;

    ObsFilter* J4;
    ObsFilter* J5;
    ObsFilter* J6;

    ObsFilter* SR4j;
    ObsFilter* SR5j;
    ObsFilter* SR6j;
};

EventSelection event_selection;

void init_selection(){
    auto leptons = lookup<std::vector<Particle>>("leptons");
    auto jets    = lookup<std::vector<Particle>>("jets");

    // Require *exactly* three charged leptons
    event_selection.trilepton = obs_filter("trilepton",GenFunction::register_function<bool()>("trilepton",
        FUNC(([leptons](){
            return leptons->get_value().size() == 3;
        }))));

    // Require at least three b-jets
    event_selection.b_jet3 = obs_filter("b_jet3",GenFunction::register_function<bool()>("b_jet3",
        FUNC(([jets](){
            int n_b_jets = 0;
            for(auto jet : jets->get_value()){
                if(jet.tag == Particle::JET && jet.jet.b_cmva > 0)
                    n_b_jets++;
            }
            return n_b_jets >= 3;
        }))));

    // Require all same-flavour OS dilepton combinations are outside the Z mass
    // window (70,105)
    event_selection.z_mass_veto = obs_filter("z_mass_veto",GenFunction::register_function<bool()>("z_mass_veto",
        FUNC(([leptons](){
            auto& leps = leptons->get_value();
            int n = leps.size();
            for(int i = 0; i < n; i++){
                for(int j = i+1; j < n; j++){
                    const Particle& p1 = leps[i];
                    const Particle& p2 = leps[j];
                    if(abs(p1.lepton.pdg_id) != abs(p2.lepton.pdg_id)) continue;
                    if(p1.lepton.charge == p2.lepton.charge) continue;
                    double m = (p1.v + p2.v).M();
                    if(70 < m && m < 105)
                        return false;
                }
            }
            return true;
        }))));


    event_selection.J4 = obs_filter("4jet_selection",GenFunction::register_function<bool()>("4jet_selection",
        FUNC(([jets](){
            return jets->get_value().size() >= 4;
        }))));

    event_selection.J5 = obs_filter("5jet_selection",GenFunction::register_function<bool()>("5jet_selection",
        FUNC(([jets](){
            return jets->get_value().size() >= 5;
        }))));

    event_selection.J6 = obs_filter("6jet_selection",GenFunction::register_function<bool()>("6jet_selection",
        FUNC(([jets](){
            return jets->get_value().size() >= 6;
        }))));

    event_selection.base_sel = ObsFilter::conj(event_selection.z_mass_veto, ObsFilter::conj(event_selection.trilepton, event_selection.b_jet3));
    event_selection.SR4j = ObsFilter::conj(event_selection.base_sel, event_selection.J4);
    event_selection.SR5j = ObsFilter::conj(event_selection.base_sel, event_selection.J5);
    event_selection.SR6j = ObsFilter::conj(event_selection.base_sel, event_selection.J6);
}
#endif // SELECTION_HPP