variations.py

from code_generation.systematics import SystematicShift
from .producers import scalefactors as scalefactors
from .producers import pairselection as pairselection
from .producers import muons as muons
from .producers import electrons as electrons
from .producers import taus as taus
from .producers import event as event 

def add_PUweightsShifts(configuration):
    configuration.add_shift(
        SystematicShift(
            name="PUUp",
            scopes=["global"],
            shift_config={
                ("global"): {"PU_reweighting_variation": "up"},
            },
            producers={
                "global": [
                    event.PUweights,
                ],
            },
        )
    )

    configuration.add_shift(
        SystematicShift(
            name="PUDown",
            scopes=["global"],
            shift_config={
                ("global"): {"PU_reweighting_variation": "down"},
            },
            producers={
                "global": [
                    event.PUweights,
                ],
            },
        )
    )

    return configuration

def add_leptonSFShifts(configuration):
    configuration.add_shift(
        SystematicShift(
            name="MuonIDUp",
            scopes = ["mm", "mmet"],
            shift_config={
                ("mm", "mmet"): {"muon_sf_varation": "systup"},
            },
            producers={
                "mm": [
                    scalefactors.Muon_1_ID_SF,
                    scalefactors.Muon_2_ID_SF,
                ],
                "mmet": [
                    scalefactors.Muon_1_ID_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="MuonIDDown",
            scopes = ["mm", "mmet"],
            shift_config={
                ("mm", "mmet"): {"muon_sf_varation": "systdown"},
            },
            producers={
                "mm": [
                    scalefactors.Muon_1_ID_SF,
                    scalefactors.Muon_2_ID_SF,
                ],
                "mmet": [
                    scalefactors.Muon_1_ID_SF,
                ],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="MuonIsoUp",
            scopes = ["mm", "mmet"],
            shift_config={
                ("mm", "mmet"): {"muon_sf_varation": "systup"},
            },
            producers={
                "mm": [
                    scalefactors.Muon_1_Iso_SF,
                    scalefactors.Muon_2_Iso_SF,
                ],
                "mmet": [
                    scalefactors.Muon_1_Iso_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="MuonIsoDown",
            scopes = ["mm", "mmet"],
            shift_config={
                ("mm", "mmet"): {"muon_sf_varation": "systdown"},
            },
            producers={
                "mm": [
                    scalefactors.Muon_1_Iso_SF,
                    scalefactors.Muon_2_Iso_SF,
                ],
                "mmet": [
                    scalefactors.Muon_1_Iso_SF,
                ],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="ElectronIDUp",
            scopes = ["ee", "emet"],
            shift_config={
                ("ee", "emet"): {"ele_sf_varation": "sfup"},
            },
            producers={
                "ee": [
                    scalefactors.Ele_1_IDWPMedium_SF,
                    scalefactors.Ele_2_IDWPMedium_SF,
                ],
                "emet": [
                    scalefactors.Ele_1_IDWPMedium_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="ElectronIDDown",
            scopes = ["ee", "emet"],
            shift_config={
                ("ee", "emet"): {"ele_sf_varation": "sfdown"},
            },
            producers={
                "ee": [
                    scalefactors.Ele_1_IDWPMedium_SF,
                    scalefactors.Ele_2_IDWPMedium_SF,
                ],
                "emet": [
                    scalefactors.Ele_1_IDWPMedium_SF,
                ],
            },
        )
    )

    return configuration

def add_tauVariations(configuration):
    #########################
    # TauvsMuID scale factor shifts
    #########################
    # vsJet shifts et/mt, tau pt dependent
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau30to35Down",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau30to35": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau30to35Up",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau30to35": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau35to40Down",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau35to40": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau35to40Up",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau35to40": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau40to500Down",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau40to500": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau40to500Up",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau40to500": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau500to1000Down",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau500to1000": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau500to1000Up",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau500to1000": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau1000toInfDown",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau1000toinf": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTau1000toInfUp",
            shift_config={("et", "mt"): {"tau_sf_vsjet_tau1000toinf": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsJetTauID_lt_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM0Down",
            shift_config={"tt": {"tau_sf_vsjet_tauDM0": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    # vsJet shifts tt, tau dm dependent
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM0Up",
            shift_config={"tt": {"tau_sf_vsjet_tauDM0": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM1Down",
            shift_config={"tt": {"tau_sf_vsjet_tauDM1": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM1Up",
            shift_config={"tt": {"tau_sf_vsjet_tauDM1": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM10Down",
            shift_config={"tt": {"tau_sf_vsjet_tauDM10": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM10Up",
            shift_config={"tt": {"tau_sf_vsjet_tauDM10": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM11Down",
            shift_config={"tt": {"tau_sf_vsjet_tauDM11": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsJetTauDM11Up",
            shift_config={"tt": {"tau_sf_vsjet_tauDM11": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsJetTauID_SF,
                    scalefactors.Tau_2_VsJetTauID_tt_SF,
                ]
            },
        )
    )
    #########################
    # TauvsEleID scale factor shifts
    #########################
    configuration.add_shift(
        SystematicShift(
            name="vsEleBarrelDown",
            shift_config={("et", "mt"): {"tau_sf_vsele_barrel": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsEleTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleBarrelUp",
            shift_config={("et", "mt"): {"tau_sf_vsele_barrel": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsEleTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleEndcapDown",
            shift_config={("et", "mt"): {"tau_sf_vsele_endcap": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsEleTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleEndcapUp",
            shift_config={("et", "mt"): {"tau_sf_vsele_endcap": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsEleTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleBarrelDown",
            shift_config={"tt": {"tau_sf_vsele_barrel": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsEleTauID_SF,
                    scalefactors.Tau_2_VsEleTauID_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleBarrelUp",
            shift_config={"tt": {"tau_sf_vsele_barrel": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsEleTauID_SF,
                    scalefactors.Tau_2_VsEleTauID_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleEndcapDown",
            shift_config={"tt": {"tau_sf_vsele_endcap": "down"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsEleTauID_SF,
                    scalefactors.Tau_2_VsEleTauID_SF,
                ]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsEleEndcapUp",
            shift_config={"tt": {"tau_sf_vsele_endcap": "up"}},
            producers={
                "tt": [
                    scalefactors.Tau_1_VsEleTauID_SF,
                    scalefactors.Tau_2_VsEleTauID_SF,
                ]
            },
        )
    )
    #########################
    # TauvsMuID scale factor shifts
    #########################
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel1Down",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel1": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel1Up",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel1": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel2Down",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel2": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel2Up",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel2": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel3Down",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel3": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel3Up",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel3": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel4Down",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel4": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel4Up",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel4": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel5Down",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel5": "down"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel5Up",
            shift_config={("et", "mt"): {"tau_sf_vsmu_wheel5": "up"}},
            producers={("et", "mt"): scalefactors.Tau_2_VsMuTauID_SF},
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel1Down",
            shift_config={"tt": {"tau_sf_vsmu_wheel1": "down"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel1Up",
            shift_config={"tt": {"tau_sf_vsmu_wheel1": "up"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel2Down",
            shift_config={"tt": {"tau_sf_vsmu_wheel2": "down"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel2Up",
            shift_config={"tt": {"tau_sf_vsmu_wheel2": "up"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel3Down",
            shift_config={"tt": {"tau_sf_vsmu_wheel3": "down"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel3Up",
            shift_config={"tt": {"tau_sf_vsmu_wheel3": "up"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel4Down",
            shift_config={"tt": {"tau_sf_vsmu_wheel4": "down"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel4Up",
            shift_config={"tt": {"tau_sf_vsmu_wheel4": "up"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel5Down",
            shift_config={"tt": {"tau_sf_vsmu_wheel5": "down"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="vsMuWheel5Up",
            shift_config={"tt": {"tau_sf_vsmu_wheel5": "up"}},
            producers={
                "tt": [scalefactors.Tau_1_VsMuTauID_SF, scalefactors.Tau_2_VsMuTauID_SF]
            },
        )
    )
    #########################
    # TES Shifts
    #########################
    configuration.add_shift(
        SystematicShift(
            name="tauEs1prong0pizeroDown",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM0": "down"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs1prong0pizeroUp",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM0": "up"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs1prong1pizeroDown",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM1": "down"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs1prong1pizeroUp",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM1": "up"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs3prong0pizeroDown",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM10": "down"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs3prong0pizeroUp",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM10": "up"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs3prong1pizeroDown",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM11": "down"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )
    configuration.add_shift(
        SystematicShift(
            name="tauEs3prong1pizeroUp",
            shift_config={("et", "mt", "tt"): {"tau_ES_shift_DM11": "up"}},
            producers={("et", "mt", "tt"): taus.TauPtCorrection_genTau},
            ignore_producers={
                "et": [pairselection.LVEl1, electrons.VetoElectrons],
                "mt": [pairselection.LVMu1, muons.VetoMuons],
                "tt": [],
            },
        )
    )

    return configuration