analysis.py

# coding: utf-8
# flake8: noqa

"""
Definition of the diHiggs analysis in the bbWW (SL) channel.
"""

import scinum as sn
import order as od
import six
import copy
import re

#
# analysis and config
#
import config.Run2_pp_13TeV_2016 as run_2016
import config.Run2_pp_13TeV_2017 as run_2017
import config.Run2_pp_13TeV_2018 as run_2018
from tasks.corrections.gf_hh import HHReweigthing
from config.util import PrepareConfig
from config.processes import get, rgb
from config.constants import HHres, HHxs

# create the analysis
analysis = od.Analysis("diHiggs_sl", 2)

analysis.set_aux("doFakeNonClosureCorrection", True)

# setup the config for the 2016, 2017, 2018 campaign
# we pass no name, so the config will have the same name as the campaign
config_2016 = run_2016.default_config(analysis=analysis)
config_2017 = run_2017.default_config(analysis=analysis)
config_2018 = run_2018.default_config(analysis=analysis)

procs = [
    od.Process(
        "dc_others",
        999989,
        label=r"Others",
        color=rgb(103, 179, 153),
        processes=[
            get("rare"),
            # get("ttVV"),
            # get("ttVH"),
            # get("ttV"),
            # get("vvv"),
            # get("WJetsToLNu_HT"),
            # get("wjets"),
        ],
    ),
    od.Process(
        "dc_ttVX",
        999990,
        label="ttVX",
        processes=[get("ttV"), get("ttVV")],
    ),
    od.Process(
        "dc_ggHbb",
        23412341,
        label="ggH(bb)",
        processes=[get("GluGluHToBB_M125"), get("GluGluHToTauTau_M125")],
    ),
    od.Process(
        "dc_qqHbb",
        23412351,
        label="qqH(bb)",
        processes=[get("VBFHToBB_M_125"), get("VBFHToTauTau_M125")],
    ),
    od.Process(
        "dc_ZHbb",
        23412361,
        label="ZH(bb)",
        processes=[get("ZH_DY"), get("ggZH"), get("ZHToTauTau_M125")],
    ),
    od.Process(
        "plot_vv(v)",
        999991,
        color=rgb(82, 124, 182),
        label=r"Multi-boson",
        processes=[get("vv"), get("vvv")],
    ),
    od.Process(
        "plot_other",
        999992,
        color=rgb(152, 78, 163),
        label=r"Other",
        processes=[get("rare"), get("ttV"), get("ttVV"), get("ttVH")],
    ),
    od.Process(
        "class_multiboson",
        3789273,
        label="VV(V)",
        processes=[get("vv"), get("vvv")],
    ),
    od.Process(
        "class_ttVX",
        3789274,
        label="ttVX",
        processes=[get("ttV"), get("ttVH"), get("ttVV")],
    ),
    od.Process(
        "fakes",
        999993,
        color=rgb(137, 207, 240),
        label="Fakes",
        is_data=True,
        aux=dict(dynamic=True),  # content is generated on-the-fly, don't whine about no datasets
    ),
]
od.Process(
    "class_HHGluGlu_NLO",
    17789277,
    label="HH(GGF)",
    processes=[
        get("HH_2B2WToLNu2J_GluGlu_NLO"),
        get("HH_2B2VTo2L2Nu_GluGlu_NLO"),
        get("HH_2B2Tau_GluGlu_NLO"),
    ],
)
od.Process(
    "class_HHVBF_NLO",
    17789278,
    label="HH(VBF)",
    processes=[
        get("HH_2B2WToLNu2J_VBF_NLO"),
        get("HH_2B2VTo2L2Nu_VBF_NLO"),
        get("HH_2B2Tau_VBF_NLO"),
    ],
)
od.Process(
    "class_other",
    3789275,
    label="Other",
    processes=[
        get("dy"),
        get("ttV"),
        get("ttVH"),
        get("ttVV"),
        get("vv"),
        get("vvv"),
        get("rare"),
    ],
)

od.Process(
    "plot_H",
    23412371,
    label=r"(SM) Higgs",
    label_short="H",
    color=rgb(120, 33, 80),
    processes=[
        get("ttVH"),
        get("HZJ_HToWW_M125"),
        get("TH"),
        get("ttH"),
        get("WH"),
        get("dc_ggHbb"),
        get("dc_qqHbb"),
        get("dc_ZHbb"),
        get("GluGluHToWWToLNuQQ_M125_NNPDF31"),
        get("VBFHToWWToLNuQQ_M125_NNPDF31"),
    ],
)
od.Process(
    "plot_others",
    23412372,
    color=rgb(152, 78, 163),
    label=r"Others",
    processes=[get("dc_others"), get("ttV"), get("ttVV")],
),

# merge signals
signals = (
    "qqHH_CV_1_C2V_1_kl_0",
    "qqHH_CV_1_C2V_1_kl_1",
    "qqHH_CV_1_C2V_1_kl_2",
    "qqHH_CV_1_C2V_2_kl_1",
    "qqHH_CV_1_5_C2V_1_kl_1",
    "qqHH_CV_0_5_C2V_1_kl_1",
    "qqHH_CV_1_C2V_0_kl_1",
    "ggHH_kl_0_kt_1",
    "ggHH_kl_1_kt_1",
    "ggHH_kl_2p45_kt_1",
    "ggHH_kl_5_kt_1",
)
for i, s in enumerate(signals):
    sl = get(f"{s}_2B2WToLNu2J")
    dl = get(f"{s}_2B2VTo2L2Nu")
    ls = dl.label_short.replace(
        r"HH$, $H \rightarrow b\bar{b}$, $H \rightarrow VV$ (DL)",
        r"HH \rightarrow b\bar{b}VV$" + "\n",
    ).replace("\n ", "\n")
    if s in ("qqHH_CV_1_C2V_1_kl_1", "ggHH_kl_1_kt_1"):
        ls = re.sub(r"\s*\(NLO\) -.+", " (NLO, SM)", ls)
    get("diHiggs").add_process(
        f"{s}_2B2V",
        1900 + i,
        label=dl.label.replace(" (DL)", ""),
        label_short=ls,
        processes=[sl, dl],
    )


for year, cfg in ("2016", config_2016), ("2017", config_2017), ("2018", config_2018):
    # corrections which are only used for distinct analyses
    cfg.aux["non_common_corrections"] = ["Fake", "VJetsCorrections"]
    cfg.aux["multiclass_group"] = "mergedinclusive"

    # categories and files used for the sync of yields
    # fmt: off
    cfg.aux["sync_yield"] = {
        "category": "(all_(boosted|resolved_1b|resolved_2b)_sr_prompt_dnn_node_class_HH.+_NLO|all_(boosted|resolved_1b|resolved_2b)_sr_prompt_dnn_node_H|all_(boosted|resolved)_sr_prompt_dnn_node_tt|all_(boosted|resolved)_sr_prompt_dnn_node_wjets|all_incl_sr_prompt$)",
        "paths": {"TLL": "/net/scratch/cms/dihiggs/sync/tll_sl_2016.json"},
        "lut": {
            "all_incl_sr_prompt": "total",
            "all_resolved_1b_sr_prompt_dnn_node_H": "H_resolved_1b",
            "all_resolved_2b_sr_prompt_dnn_node_H": "H_resolved_2b",
            "all_boosted_sr_prompt_dnn_node_H": "H_boosted",
            "all_resolved_sr_prompt_dnn_node_tt": "TT_resolved",
            "all_boosted_sr_prompt_dnn_node_tt": "TT_boosted",
            "all_resolved_sr_prompt_dnn_node_wjets": "W_resolved",
            "all_boosted_sr_prompt_dnn_node_wjets": "W_boosted",
        }
    }
    # fmt: on
    #
    # process groups
    #
    cfg.set_aux(
        "process_groups",
        {
            # just defines the grouping during plotting/inference
            "plotting": [
                "tt",
                "dy",
                "st",
                "fakes",
                # "QCD",
                "wjets",
                "plot_vv(v)",
                "plot_others",
                "plot_H",
                "qqHH_CV_1_C2V_1_kl_1_2B2V",
                "ggHH_kl_1_kt_1_2B2V",  # "qqHH_CV_1_C2V_1_kl_1_2B2WToLNu2J",  # "ggHH_kl_1_kt_1_2B2WToLNu2J",  # "ggHH_kl_1_kt_1_2B2VTo2L2Nu",  # "GluGluToHHTo2B2VTo2L2Nu_node_SM",
                "data",
            ],
        },
    )

    cfg.set_aux(
        "multiclass_groups",
        {
            "mergedinclusive": {
                "class_HHGluGlu_NLO": {"groups": ["signal", "constrain", "fit"]},
                "class_HHVBF_NLO": {"groups": ["signal", "constrain", "fit"]},
                "tt": {"groups": ["background", "constrain"]},
                "st": {"groups": ["background"]},
                "wjets": {"groups": ["background"]},
                "H": {"groups": ["background"]},
                "class_other": {"groups": ["background"]},
            },
            "ggfttbar": {
                "ggHH_kl_1_kt_1_2B2WToLNu2J": {"groups": ["signal", "constrain", "fit"]},
                "tt": {"groups": ["background", "constrain"]},
            },
        },
    )

    cfg.set_aux(
        "btag_sf_shifts",
        [
            "lf",
            "lfstats1",
            "lfstats2",
            "hf",
            "hfstats1",
            "hfstats2",
            "cferr1",
            "cferr2",
        ],
    )

    cfg.set_aux(
        "sync",
        {
            "lookup_set": {
                ("mu", "m"),
                ("ee", "ee"),
                ("emu", "em"),
                ("mumu", "mm"),
                ("min_dhi_jet", "min_dphi_jet"),
                ("vbf_tag", "VBF_tag"),
                ("electron", "ele"),
                ("muon", "mu"),
                ("jet", "ak4Jet"),
                ("fat", "ak8Jet"),
                ("weight_gen_weight", "MC_weight"),
                ("weight_pileup", "PU_weight"),
                ("weight_trigger_electron_sf", "weight_trigger_el_sf"),
                ("weight_trigger_muon_sf", "weight_trigger_mu_sf"),
            },
            "categories": ["is_all_incl_sr_prompt"],
            "eventnr": "eventnr",
            "int_vars": ["eventnr", "dataset_id"],
        },
    )
    PrepareConfig(
        cfg,
        channels=[],  # leave empty: otherwise we can not access single lepton triggers anymore
        # defines which datasets we will use
        processes=[
            "data",
            "HH_2B2WToLNu2J",
            "HH_2B2VTo2L2Nu",
            "HH_2B2Tau",
            "H",
            "tt",
            "dy_lep_10To50",
            "dy_lep_50ToInf",
            "dy_lep_nj",
            "vv",
            "st",
            "WJetsToLNu",
            "WnJetsToLNu",
            "WJetsToLNu_HT",
            # "WBJetsToLNu_Wpt",
            # "WJetsToLNu_BGenFilter_Wpt",
            "ttV",
            "ttVV",
            "ttVH",
            "vvv",
            "rare",
            "QCD",
        ],
        skip_process=lambda process: process.name.endswith(HHres["resonance"]),
        ignore_datasets=[
            # "TTToHadronic",
            # "THQ_ctcvcp_4f_Hincl",
        ],  # , "ZH_HToBB_ZToLL_M125", "TTWW"],
        allowed_exclusive_processes=[
            "wjets_lep",
            "dy_50to",
        ],
    )

    HHReweigthing.update_config(
        config=cfg,
        root_processes=[
            "HH_2B2WToLNu2J_GluGlu",
            "HH_2B2WToLNu2J_VBF",
            "HH_2B2VTo2L2Nu_GluGlu",
            "HH_2B2VTo2L2Nu_VBF",
        ],
        benchmarks=[dict(kl=x) for x in HHxs["gf_nnlo_ftapprox"].keys()],
    )

    from .categories import setup_categories

    setup_categories(cfg)

    from .variables import setup_variables

    setup_variables(cfg)

    #
    # other configurations
    #

    cfg.set_aux("signal_process", "diHiggs")