selection.hpp

Go to the documentation of this file.

#ifndef SELECTION_HPP
#define SELECTION_HPP

#include <iostream>
#include <vector>
#include <utility>
#include <numeric>
#include <limits>

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

#include <TSystem.h>

#define JET_REQUIREMENT   6
#define B_JET_REQUIREMENT 3
#define B_JET_WP          0.  // lower bound on CMVA value
using namespace fv;
struct Selection {
public:

    static ObsFilter* trilepton_filter;

    static ObsFilter* jet_multiplicity_filter;
    static ObsFilter* b_jet_multiplicity_filter;

    static void init(){
        auto nLep = lookup<int>("nLepGood");
        trilepton_filter = obs_filter("trilepton_filter",GenFunction::register_function<bool()>("trilepton_filter",
            FUNC(([nLep=nLep](){
                return nLep->get_value() == 3;
            }))));


        auto nJet = lookup<int>("nJet");
        jet_multiplicity_filter = obs_filter("jet_multiplicity_filter",GenFunction::register_function<bool()>("jet_multiplicity_filter",
            FUNC(([nJet=nJet](){
                return nJet->get_value() >= JET_REQUIREMENT;
            }))));

        auto bJet_MVA = lookup<std::vector<float>>("Jet_btagCMVA");
        b_jet_multiplicity_filter = obs_filter("b_jet_multiplicity_filter",GenFunction::register_function<bool()>("b_jet_multiplicity_filter",
            FUNC(([bJet_MVA=bJet_MVA](){
                int n_b_jet = 0;
                for(auto j : bJet_MVA->get_value()){
                    if(j > B_JET_WP){
                        n_b_jet++;
                    }
                }
                return n_b_jet >= B_JET_REQUIREMENT;
            }))));
    }
};
#endif // SELECTION_HPP