EWP-BplusToKstMuMu-AngAna/Code/Ganga/UsefulScripts/GetEfficiency/get_MCref_eff.py

import os

def get_path(decay = "Kplus", year = "2012", polarity = "Down"):
    sim_version = "Sim09a"
    dst = "DST"
    E = 0
    nu = ""
    trigger = ""
    stripping = ""
    reco = ""

    if (decay=="Kplus"): decay_number = 12143401
    if (decay=="Kshort"): decay_number = 12145102

    if (year=="2011"):
        E = 3500
        nu = 2
        trigger = "0x40760037"
        stripping = "21r1"
        reco = "Reco14c"
    elif (year=="2012"):
        E = 4000
        nu = 2.5
        trigger = "0x409f0045"
        stripping = "21"
        reco = "Reco14c"
    elif (year=="2015"):
        E = 6500
        nu = "1.6-25ns"
        trigger = "0x411400a2"
        stripping = "24"
        reco = "Reco15a/Turbo02"
        sim_version = "Sim09b"
        dst = "MDST"
    elif (year=="2016"):
        E = 6500
        nu ="1.6-25ns"
        trigger = "0x409f0045"
        stripping = "28"
        reco = "Reco15a/Turbo02"
        sim_version = "Sim09c"
        dst = "MDST"

    script_opts = {"YEAR":year,"POL":polarity,"DATA_TYPE":"MC","ENERGY":E,"NU":nu,"TRIG":trigger,"STRIP":stripping,"CHAN":decay, "NUM":decay_number, "RECO":reco, "SIM": sim_version, "DST":dst}
    PATH = "/MC/%(YEAR)s/Beam%(ENERGY)sGeV-%(YEAR)s-Mag%(POL)s-Nu%(NU)s-Pythia8/%(SIM)s/Trig%(TRIG)s/%(RECO)s/Stripping%(STRIP)sNoPrescalingFlagged/%(NUM)s/ALLSTREAMS.%(DST)s" % script_opts

    return PATH

def get_eff(decay = "Kplus", year = "2012", polarity = "Down"):
    command = "lb-run LHCbDirac/prod dirac-bookkeeping-rejection-stats -B " + str(get_path(decay,year,polarity))
    print command
    os.system(command)


if __name__ == '__main__':
    get_eff()