You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
174 lines
19 KiB
174 lines
19 KiB
|
|
from compareUltimateSource import *
|
|
|
|
|
|
'''
|
|
Tool for comparing two variables from two datasets.
|
|
Yes, writting it in Python3 would be smarter,
|
|
but default is Python2 on all servers so far
|
|
'''
|
|
def compareUltimateAllPolarities(variable1, variable2,
|
|
year1, year2, cut1, cut2 ,
|
|
MC1, TM1, ReferenceChannel1, PHSP1, Preselected1, BDTed1,
|
|
MC2, TM2, ReferenceChannel2, PHSP2, Preselected2, BDTed2,
|
|
sWeighted1, bWeighted1, b2Dweighted1,
|
|
sWeighted2, bWeighted2, b2Dweighted2,
|
|
bPrint, KshortDecaysInVelo1, KshortDecaysInVelo2, verbose):
|
|
for magnet in polarityList():
|
|
compareUltimate(variable1 = variable1, variable2 = variable2,
|
|
year1 = year1, year2 = year2, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet, MC1 = MC1, TM1 = TM1, ReferenceChannel1 = ReferenceChannel1, PHSP1 = PHSP1, Preselected1 = Preselected1, BDTed1 = BDTed1,
|
|
magnet2 = magnet, MC2 = MC2, TM2 = TM2, ReferenceChannel2 = ReferenceChannel2, PHSP2 = PHSP2, Preselected2 = Preselected2, BDTed2 = BDTed2,
|
|
sWeighted1 = sWeighted1, bWeighted1 = bWeighted1, b2Dweighted1 = b2Dweighted1, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = sWeighted2, bWeighted2 = bWeighted2, b2Dweighted2 = b2Dweighted2, weightBranch2 = firstMCweight(),
|
|
bPrint = bPrint, KshortDecaysInVelo1 = KshortDecaysInVelo1, KshortDecaysInVelo2 = KshortDecaysInVelo2, verbose = verbose)
|
|
|
|
def compareUltimateAllPolAndYears( variable1, variable2,
|
|
cut1, cut2 ,
|
|
MC1, TM1, ReferenceChannel1, PHSP1, Preselected1, BDTed1,
|
|
MC2, TM2, ReferenceChannel2, PHSP2, Preselected2, BDTed2,
|
|
sWeighted1, bWeighted1, b2Dweighted1,
|
|
sWeighted2, bWeighted2, b2Dweighted2,
|
|
bPrint, KshortDecaysInVelo1, KshortDecaysInVelo2, verbose):
|
|
for year in polarityList():
|
|
compareUltimateAllPolarities(variable1 = variable1, variable2 = variable2,
|
|
year1 = year, year2 = year, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet, MC1 = MC1, TM1 = TM1, ReferenceChannel1 = ReferenceChannel1, PHSP1 = PHSP1, Preselected1 = Preselected1, BDTed1 = BDTed1,
|
|
magnet2 = magnet, MC2 = MC2, TM2 = TM2, ReferenceChannel2 = ReferenceChannel2, PHSP2 = PHSP2, Preselected2 = Preselected2, BDTed2 = BDTed2,
|
|
sWeighted1 = sWeighted1, bWeighted1 = bWeighted1, b2Dweighted1 = b2Dweighted1, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = sWeighted2, bWeighted2 = bWeighted2, b2Dweighted2 = b2Dweighted2, weightBranch2 = firstMCweight(),
|
|
bPrint = bPrint, KshortDecaysInVelo1 = KshortDecaysInVelo1, KshortDecaysInVelo2 = KshortDecaysInVelo2, verbose = verbose)
|
|
|
|
#Two samples, same variable
|
|
def compareUltimateDataMC(variable, year, cut, magnet,
|
|
TM = True, ReferenceChannel = False, PHSP = False,
|
|
sWeighted = True, bWeighted = False, b2Dweighted = True, KshortDecaysInVelo = False, verbose = False):
|
|
return compareUltimate(variable1 = variable, variable2 = variable,
|
|
year1 = year, year2 = year, cut1 = cut, cut2 = cut,
|
|
magnet1 = magnet, MC1 = False, TM1 = False, ReferenceChannel1 = False, PHSP1 = False, Preselected1 = True, BDTed1 = False,
|
|
magnet2 = magnet, MC2 = True, TM2 = TM, ReferenceChannel2 = ReferenceChannel, PHSP2 = PHSP, Preselected2 = True, BDTed2 = False,
|
|
sWeighted1 = sWeighted, bWeighted1 = False , b2Dweighted1 = False, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = False, bWeighted2 = bWeighted, b2Dweighted2 = b2Dweighted, weightBranch2 = firstMCweight(),
|
|
bPrint = False, KshortDecaysInVelo1 = KshortDecaysInVelo, KshortDecaysInVelo2 = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#Two MC versions, same variable
|
|
def compareUltimateMCMC(variable, year1, year2, cut1, cut2, magnet1, magnet2,
|
|
TM = True, ReferenceChannel = False, PHSP = False,
|
|
bWeighted1 = False, b2Dweighted1 = False, bWeighted2 = False, b2Dweighted2 = False, KshortDecaysInVelo = False, verbose = False):
|
|
return compareUltimate(variable1 = variable, variable2 = variable,
|
|
year1 = year1, year2 = year2, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet1, MC1 = True, TM1 = TM, ReferenceChannel1 = ReferenceChannel, PHSP1 = PHSP, Preselected1 = True, BDTed1 = False,
|
|
magnet2 = magnet2, MC2 = True, TM2 = TM, ReferenceChannel2 = ReferenceChannel, PHSP2 = PHSP, Preselected2 = True, BDTed2 = False,
|
|
sWeighted1 = False, bWeighted1 = bWeighted1, b2Dweighted1 = b2Dweighted1, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = False, bWeighted2 = bWeighted2, b2Dweighted2 = b2Dweighted2, weightBranch2 = firstMCweight(),
|
|
bPrint = False, KshortDecaysInVelo1 = KshortDecaysInVelo, KshortDecaysInVelo2 = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#Two data versions, same variable
|
|
def compareUltimateDataData(variable, year1, year2, cut1, cut2, magnet1, magnet2, sWeighted1, sWeighted2, KshortDecaysInVelo1, KshortDecaysInVelo2, verbose):
|
|
|
|
return compareUltimate(variable1 = variable, variable2 = variable,
|
|
year1 = year1, year2 = year2, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet1, MC1 = False, TM1 = False, ReferenceChannel1 = False, PHSP1 = False, Preselected1 = True, BDTed1 = False,
|
|
magnet2 = magnet2, MC2 = True, TM2 = False, ReferenceChannel2 = False, PHSP2 = False, Preselected2 = True, BDTed2 = False,
|
|
sWeighted1 = sWeighted1, bWeighted1 = False, b2Dweighted1 = False, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = sWeighted2, bWeighted2 = False, b2Dweighted2 = False, weightBranch2 = firstMCweight(),
|
|
bPrint = False, KshortDecaysInVelo1 = KshortDecaysInVelo, KshortDecaysInVelo2 = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#Two variables, same data sample, different cuts
|
|
def compareUltimateTwoVars(variable1, variable2, year, cut1, cut2, magnet, MC, TM, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose = False):
|
|
return compareUltimate(variable1 = variable1, variable2 = variable2,
|
|
year1 = year, year2 = year, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet, MC1 = MC, TM1 = TM, ReferenceChannel1 = ReferenceChannel, PHSP1 = PHSP, Preselected1 = Preselected, BDTed1 = BDTed,
|
|
magnet2 = magnet, MC2 = MC, TM2 = TM, ReferenceChannel2 = ReferenceChannel, PHSP2 = PHSP, Preselected2 = Preselected, BDTed2 = BDTed,
|
|
sWeighted1 = sWeighted, bWeighted1 = bWeighted, b2Dweighted1 = b2Dweighted, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = sWeighted, bWeighted2 = bWeighted, b2Dweighted2 = b2Dweighted, weightBranch2 = firstMCweight(),
|
|
bPrint = False, KshortDecaysInVelo1 = KshortDecaysInVelo, KshortDecaysInVelo2 = KshortDecaysInVelo, verbose = verbose)
|
|
#Two variables, data sample, different cuts
|
|
def compareUltimateTwoVarsData(variable1, variable2, year, cut1, cut2, magnet, Preselected, BDTed, sWeighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimateTwoVars(variable1=variable1, variable2=variable2, year=year, cut1=cut1,cut2=cut2, magnet=magnet, MC=False, TM=False, ReferenceChannel=False, PHSP=False, Preselected=Preselected, BDTed = BDTed, sWeighted=sWeighted, bWeighted=False, b2Dweighted=False, KshortDecaysInVelo=KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#Two variables, MC sample, different cuts
|
|
def compareUltimateTwoVarsMC(variable1, variable2, year, cut1, cut2, magnet, TM, ReferenceChannel, PHSP, Preselected, BDTed, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimateTwoVars(variable1=variable1, variable2=variable2, year=year, cut1=cut1,cut2=cut2, magnet=magnet, MC=True, TM=TM, ReferenceChannel=ReferenceChannel, PHSP=PHSP, Preselected=Preselected, BDTed= BDTed, sWeighted=False, bWeighted=bWeighted, b2Dweighted=b2Dweighted, KshortDecaysInVelo=KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#One variable, same data sample, different cuts
|
|
def compareUltimateTwoCuts(variable, year, magnet, cut1, cut2, MC, TM, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimate(variable1 = variable, variable2 = variable,
|
|
year1 = year, year2 = year, cut1 = cut1, cut2 = cut2,
|
|
magnet1 = magnet, MC1 = MC, TM1 = TM, ReferenceChannel1 = ReferenceChannel, PHSP1 = PHSP, Preselected1 = Preselected, BDTed1 = BDTed,
|
|
magnet2 = magnet, MC2 = MC, TM2 = TM, ReferenceChannel2 = ReferenceChannel, PHSP2 = PHSP, Preselected2 = Preselected, BDTed2 = BDTed,
|
|
sWeighted1 = sWeighted, bWeighted1 = bWeighted, b2Dweighted1 = b2Dweighted, weightBranch1 = firstMCweight(),
|
|
sWeighted2 = sWeighted, bWeighted2 = bWeighted, b2Dweighted2 = b2Dweighted, weightBranch2 = firstMCweight(),
|
|
bPrint = False, KshortDecaysInVelo1 = KshortDecaysInVelo, KshortDecaysInVelo2 = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#One variable, data sample, different cuts
|
|
def compareUltimateTwoCutsData(variable, year, magnet, cut1, cut2, Preselected, BDTed, sWeighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimateTwoCuts(variable=variable, year= year, magnet= magnet, cut1 = cut1, cut2 = cut2,MC=False, TM=False, ReferenceChannel = False, PHSP = False, Preselected= Preselected, BDTed = BDTed, sWeighted = sWeighted, bWeighted = False, b2Dweighted = False, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
#One variable, MC sample, different cuts
|
|
def compareUltimateTwoCutsMC(variable, year, magnet, cut1, cut2, TM, ReferenceChannel, PHSP, Preselected, BDTed, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimateTwoCuts(variable=variable, year= year, magnet= magnet, cut1 = cut1, cut2 = cut2,MC=True, TM=TM, ReferenceChannel = ReferenceChannel, PHSP = False, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
|
|
def compareUltimateBchargeAllSamples(variable, year, magnet, MC, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
if (not MC):
|
|
compareUltimateTwoCutsData(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = -521", Preselected= Preselected, BDTed = BDTed, sWeighted = sWeighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
else:
|
|
compareUltimateTwoCutsMC(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = -521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
compareUltimateTwoCutsMC(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = 521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
compareUltimateTwoCutsMC(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_ID = -521", cut2 = "B_plus_ID = -521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
compareUltimateTwoCutsMC(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_TRUEID = 521", cut2 = "B_plus_TRUEID = -521", TM=True, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
compareUltimateTwoCutsMC(variable=variable, year= year, magnet= magnet, cut1 = "B_plus_TRUEID = 521", cut2 = "B_plus_TRUEID = -521", TM=True, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = False, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
|
|
def compareUltimateBchargeAllSamplesTwoVars(variable1, variable2, year, magnet, MC, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
if (not MC):
|
|
compareUltimateTwoVarsData(variable1=variable1, variable2=variable2, year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = -521", Preselected= Preselected, BDTed = BDTed, sWeighted = sWeighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
else:
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = -521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_ID = -521", cut2 = "B_plus_ID = 521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_ID = 521", cut2 = "B_plus_ID = 521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_ID = -521", cut2 = "B_plus_ID = -521", TM=False, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_TRUEID = 521", cut2 = "B_plus_TRUEID = -521", TM=True, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
compareUltimateTwoVarsMC(variable1=variable1, variable2=variable2,year= year, magnet= magnet, cut1 = "B_plus_TRUEID = 521", cut2 = "B_plus_TRUEID = -521", TM=True, ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = KshortDecaysInVelo, verbose = verbose)
|
|
|
|
|
|
|
|
def compareUltimateBchargeAllPolarities(variable, year,MC, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
for pol in polarityList():
|
|
compareUltimateBchargeAllSamples(variable=variable, year=year, magnet = pol,MC = MC, ReferenceChannel= ReferenceChannel, PHSP=PHSP, Preselected = Preselected, BDTed= BDTed, sWeighted=sWeighted, bWeighted=bWeighted, b2Dweighted=b2Dweighted, KshortDecaysInVelo=KshortDecaysInVelo, verbose=verbose)
|
|
|
|
|
|
def compareUltimateBchargeAllYears(variable,MC, ReferenceChannel, PHSP, Preselected, BDTed, sWeighted, bWeighted, b2Dweighted, KshortDecaysInVelo, verbose):
|
|
for year in yearList():
|
|
compareUltimateBchargeAllPolarities(variable = variable, year = year, MC = MC,ReferenceChannel = ReferenceChannel, PHSP = PHSP, Preselected= Preselected, BDTed = BDTed, sWeighted = sWeighted, bWeighted= bWeighted, b2Dweighted = b2Dweighted,KshortDecaysInVelo =KshortDecaysInVelo, verbose= verbose)
|
|
|
|
def compareUltimateMuonIPCHI2All(verbose): #TODO possibly change parameters such as preselected
|
|
compareUltimateBchargeAllYears(variable = "log(mu_plus_IPCHI2_OWNPV)", MC = False, ReferenceChannel = False, PHSP = False, Preselected= True, BDTed = False, sWeighted = False, bWeighted= False, b2Dweighted = False,KshortDecaysInVelo =False, verbose= verbose)
|
|
compareUltimateBchargeAllYears(variable = "log(mu_minus_IPCHI2_OWNPV)", MC = True, ReferenceChannel = False, PHSP = False, Preselected= True, BDTed = False, sWeighted = False, bWeighted= False, b2Dweighted = False,KshortDecaysInVelo =False, verbose= verbose)
|
|
compareUltimateBchargeAllYears(variable = "log(mu_minus_IPCHI2_OWNPV)", MC = True, ReferenceChannel = True, PHSP = False, Preselected= True, BDTed = False, sWeighted = False, bWeighted= False, b2Dweighted = False,KshortDecaysInVelo =False, verbose= verbose)
|
|
compareUltimateBchargeAllYears(variable = "log(mu_minus_IPCHI2_OWNPV)", MC = True, ReferenceChannel = False, PHSP = True, Preselected= True, BDTed = False, sWeighted = False, bWeighted= False, b2Dweighted = False,KshortDecaysInVelo =False, verbose= verbose)
|
|
|
|
|
|
|
|
|
|
def compareUltimateBDTcut(variable, year, magnet, sWeighted, KshortDecaysInVelo, verbose):
|
|
return compareUltimateTwoCuts(variable = variable, year =year, magnet =magnet, cut1 ="MLPresponse < 0.1", cut2 = "MLPresponse > 0.9", MC=False, TM=False, ReferenceChannel=False, PHSP=False, Preselected=True, BDTed=True, sWeighted = sWeighted, bWeighted=False, b2Dweighted=False, KshortDecaysInVelo=KshortDecaysInVelo, verbose=verbose)
|
|
|
|
|
|
def compareUltimateAllKplusDataMC(year = 2011, magnet = "both", cut = "", ReferenceChannel = False, PHSP = False, bWeighted = False, b2Dweighted = True, whichParticle = 1, verbose = False):
|
|
for var in ListOfKplusVars(whichParticle=whichParticle):
|
|
compareUltimateDataMC(var, year, cut, magnet,
|
|
TM = True, ReferenceChannel = ReferenceChannel, PHSP = PHSP,
|
|
sWeighted = True, bWeighted = bWeighted, b2Dweighted = b2Dweighted, KshortDecaysInVelo = False, verbose = verbose)
|
|
|
|
return 0 #TODO
|
|
|
|
def compareUltimateAllKplusBDTcut(year = 2011, magnet = "both", whichParticle = 1, verbose = False):
|
|
for var in ListOfKplusVars(whichParticle=whichParticle):
|
|
compareUltimateBDTcut(var, year, year, magnet, magnet, KshortDecaysInVelo =False)
|
|
return 0 #TODO
|