AddZJetsWeights Namespace Reference

Classes
class	CustomFormatter

Functions
	generateZpTMapping (dictionary)
	main (infile, weightsFile, treename, attr_mcID, attr_nTruthJets)

Variables
str	__author__ = "John Anders & Jack Lindon"
str	__doc__ = """Script to add Sherpa Systematic Weight branches to existing file."""
	parser
	description
	usage
	formatter_class
	type
	str
	help
	required
	False
	default
	metavar
	args = parser.parse_args()

Function Documentation

generateZpTMapping()

AddZJetsWeights.generateZpTMapping

(

dictionary

)

Definition at line 13 of file AddZJetsWeights.py.

def generateZpTMapping(dictionary):
 
    #Sherpa 2.1
    # 0-70, bin 1, 70-140 bin 2 etc.
    print "Adding Z--> nunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(361444, 361447):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361447, 361450):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361450, 361453):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361453, 361456):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361456, 361459):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361459, 361462):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361462, 361465):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361465, 361468):
        dictionary[i] = 8
        print i,
 
 
 
    print "\nAdding Z--> ee to the dictionary"
    print "\tRange 0-70:",
    for i in range(361372, 361375):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361375, 361378):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361378, 361381):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361381, 361384):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361384, 361387):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361387, 361390):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361390, 361393):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361393, 361396):
        dictionary[i] = 8
        print i,
 
 
 
    print "\nAdding Z--> mumu to the dictionary"
    print "\tRange 0-70:",
    for i in range(361396, 361399):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361399, 361402):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361402, 361405):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361405, 361408):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361408, 361411):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361411, 361414):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361414, 361417):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361417, 361420):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding Z--> tautau to the dictionary"
    print "\tRange 0-70:",
    for i in range(361420, 361423):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361423, 361426):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361426, 361429):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361429, 361432):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361432, 361435):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361435, 361438):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361438, 361441):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361441, 361444):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding W+Jets MCIDs",
    print "Adding W--> enu to the dictionary"
    print "\tRange 0-70:",
    for i in range(361300, 361303):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361303, 361306):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361306, 361309):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361309, 361312):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361312, 361315):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361315, 361318):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361318, 361321):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361321, 361324):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding W--> munu to the dictionary"
    print "\tRange 0-70:",
    for i in range(361324, 361327):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361327, 361330):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361330, 361333):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361333, 361336):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361336, 361339):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361339, 361342):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361342, 361345):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361345, 361348):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding W--> taunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(361348, 361351):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(361351, 361354):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(361354, 361357):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(361357, 361360):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(361360, 361363):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(361363, 361366):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(361366, 361369):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(361369, 361372):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding the Sherpa 2.2 nominal to the dictionary"
 
    # 0-70, bin 1, 70-140 bin 2 etc.
    print "Adding Z--> nunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(363412, 363415):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363415, 363418):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363418, 363421):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363421, 363424):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363424, 363427):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363427, 363430):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363430, 363433):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363433, 363436):
        dictionary[i] = 8
        print i,
 
 
 
    print "\nAdding Z--> ee to the dictionary"
    print "\tRange 0-70:",
    for i in range(363388, 363391):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363391, 363394):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363394, 363397):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363397, 363400):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363400, 363403):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363403, 363406):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363406, 363409):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363409, 363412):
        dictionary[i] = 8
        print i,
 
 
 
    print "\nAdding Z--> mumu to the dictionary"
    print "\tRange 0-70:",
    for i in range(363364, 363367):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363367, 363370):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363370, 363373):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363373, 363376):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363376, 363379):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363379, 363382):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363382, 363385):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363385, 363388):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding Z--> tautau to the dictionary"
    print "\tRange 0-70:",
    for i in range(363099, 363102):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363102, 363105):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363105, 363108):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363108, 363111):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363111, 363114):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363114, 363117):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363117, 363120):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363120, 363123):
        dictionary[i] = 8
        print i,
 
 
 
 
    print "\nAdding W+Jets MCIDs for Sherpa 2.2"
    print "Adding W--> enu to the dictionary"
    print "\tRange 0-70:",
    for i in range(363460, 363463):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363463, 363466):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363466, 363469):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363469, 363472):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363472, 363475):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363475, 363478):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363478, 363481):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363481, 363484):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding W--> munu to the dictionary"
    print "\tRange 0-70:",
    for i in range(363436, 363439):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363439, 363442):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363442, 363445):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363445, 363448):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363448, 363451):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363451, 363454):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363454, 363457):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363447, 363460):
        dictionary[i] = 8
        print i,
 
 
    print "\nAdding W--> taunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(363331, 363334):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(363334, 363337):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(363337, 363340):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(363340, 363343):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-700:",
    for i in range(363343, 363346):
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 700-1000:",
    for i in range(363346, 363349):
        dictionary[i] = 6
        print i,
 
    print "\n\tRange 1000-2000:",
    for i in range(363349, 363352):
        dictionary[i] = 7
        print i,
 
    print "\n\tRange 2000-ECMS:",
    for i in range(363352, 363355):
        dictionary[i] = 8
        print i,
 
 
 
    
    print "\nAdding the Sherpa 2.2.1 nominal to the dictionary"
 
    # 0-70, bin 1, 70-140 bin 2 etc.
    print "Adding Z--> nunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(364142, 364145):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364145, 364148):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364148, 364151):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364151, 364154):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    for i in [364154]:
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 1000-ECMS:",
    for i in [364155]:
        dictionary[i] = 7
        print i,
 
 
 
    print "\nAdding Z--> ee to the dictionary"
    print "\tRange 0-70:",
    for i in range(364114, 364117):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364117, 364120):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364120, 364123):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364123, 364126):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364126] = 5
    dictionary[364218] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364127] = 7
    dictionary[364219] = 7
    print i,
 
 
 
    print "\nAdding Z--> mumu to the dictionary"
    print "\tRange 0-70:",
    for i in range(364100, 364103):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364103, 364106):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364106, 364109):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364109, 364112):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364112] = 5
    dictionary[364216] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364113] = 7
    dictionary[364217] = 7
    print i,
 
 
    print "\nAdding Z--> tautau to the dictionary"
    print "\tRange 0-70:",
    for i in range(364128, 364131):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364131, 364134):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364134, 364137):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364137, 364140):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364140] = 5
    dictionary[364220] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364141] = 7
    dictionary[364221] = 7
    print i,
 
 
    #NEW ZNUNU
    print "\nAdding Z--> nunu (2019 PTV filtered samples) to the dictionary"
 
    print "\tRange 70-100:",
    for i in [366010,366019,366028]:
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 100-140:",
    for i in [366011,366012,366013,366020,366021,366022,366029,366030,366031]:
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in [366014,366015,366016,366023,366024,366025,366032,366033,366034]:
        dictionary[i] = 3
        print i,
 
    print "\n\tRange 280-500:",
    for i in [366017,366026,366035]:
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    for i in [364222]:
        dictionary[i] = 5
        print i,
 
    print "\n\tRange 1000-ECMS:",
    for i in [364223]:
        dictionary[i] = 7
        print i,
    # clear from previous print
    print ""
 
    print "\nAdding W--> enu to the dictionary"
    print "\tRange 0-70:",
    for i in range(364170, 364173):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364173, 364176):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364176, 364179):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364179, 364182):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364182] = 5
    dictionary[364226] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364183] = 7
    dictionary[364227] = 7
    print i,
 
 
 
    print "\nAdding W--> munu to the dictionary"
    print "\tRange 0-70:",
    for i in range(364156, 364159):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364159, 364162):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364162, 364165):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364165, 364168):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364168] = 5
    dictionary[364224] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364169] = 7
    dictionary[364225] = 7
    print i,
 
 
    print "\nAdding W--> taunu to the dictionary"
    print "\tRange 0-70:",
    for i in range(364184, 364187):
        dictionary[i] = 1
        print i,
 
    print "\n\tRange 70-140:",
    for i in range(364187, 364190):
        dictionary[i] = 2
        print i,
 
    print "\n\tRange 140-280:",
    for i in range(364190, 364193):
        dictionary[i] = 3
        print i,
 
 
    print "\n\tRange 280-500:",
    for i in range(364193, 364196):
        dictionary[i] = 4
        print i,
 
    print "\n\tRange 500-1000:",
    dictionary[364196] = 5
    dictionary[364228] = 5
    print i,
 
    print "\n\tRange 1000-ECMS:",
    dictionary[364197] = 7
    dictionary[364229] = 7
    print i,
    # clear from previous print
    print ""
 
 
    return dictionary
 

main()

AddZJetsWeights.main	(		infile,
			weightsFile,
			treename,
			attr_mcID,
			attr_nTruthJets )

Definition at line 908 of file AddZJetsWeights.py.

def main(infile, weightsFile, treename, attr_mcID, attr_nTruthJets):
    print "Creating MCID, ZpT mapping"
    ZpTMapping = dict()
    generateZpTMapping(ZpTMapping)
 
    print "Loading histograms with weights"
    ZeeHistograms = []
    ZnunuHistograms = []
    WenuHistograms = []
 
    g = ROOT.TFile(weightsFile)
    systList = ["ckkw15", "ckkw30", "fac025", "fac4", "renorm025", "renorm4", "qsf025", "qsf4"]
 
    for syst in systList:
        h = ROOT.TH2F(g.Get("Zee"+syst))
        ZeeHistograms.append(h)
        h = ROOT.TH2F(g.Get("Znunu"+syst))
        ZnunuHistograms.append(h)
        h = ROOT.TH2F(g.Get("Wenu"+syst))
        WenuHistograms.append(h)
 
    print "Loading File"
    f = ROOT.TFile(infile,"update")
 
    print "Loading Tree"
 
    # empty arrays for adding branches
    ckkw15 = array('f', [0.])
    ckkw30 = array('f', [0.])
    fac025 = array('f', [0.])
    fac4 = array('f', [0.])
    renorm025 = array('f', [0.])
    renorm4 = array('f', [0.])
    qsf025 = array('f', [0.])
    qsf4 = array('f', [0.])
 
    # Add branches
    T = f.Get(treename)
    ckkw15branch = T.Branch("ckkw15_Weight", ckkw15,"ckkw15_Weight/F")
    ckkw30branch = T.Branch("ckkw30_Weight", ckkw30,"ckkw30_Weight/F")
    fac025branch = T.Branch("fac025_Weight", fac025,"fac025_Weight/F")
    fac4branch = T.Branch("fac4_Weight", fac4,"fac4_Weight/F")
    renorm025branch = T.Branch("renorm025_Weight", renorm025,"renorm025_Weight/F")
    renorm4branch = T.Branch("renorm4_Weight", renorm4,"renorm4_Weight/F")
    qsf025branch = T.Branch("qsf025_Weight", qsf025,"qsf025_Weight/F")
    qsf4branch = T.Branch("qsf4_Weight", qsf4,"qsf4_Weight/F")
 
    print "Adding Weights"
    nEvts = T.GetEntries()
 
    for iEvt in range(nEvts):
 
        if iEvt%1000 == 0:
            print '\x1b[2K\r', "Current Event being Processed is: {0:15d}".format(iEvt),
            sys.stdout.flush()
 
        T.GetEntry(iEvt)
 
        MCID = getattr(T, attr_mcID)
        nTruthJets = getattr(T, attr_nTruthJets)
 
        decayType = "none"
 
        # Find the Decay type
        if (
               (361372 <= MCID <= 361443) or
               (363102 <= MCID <= 363122) or
               (363361 <= MCID <= 363363) or
               (363364 <= MCID <= 363411) or
               (364114 <= MCID <= 364127) or
               (364100 <= MCID <= 364113) or
               (364128 <= MCID <= 364141) or
           (364216 <= MCID <= 364217) or
           (364220 <= MCID <= 364221)
            ):
            decayType = "dilepton"
 
        elif (
               (361444 <= MCID <= 361467) or
               (363412 <= MCID <= 363435) or
               (364142 <= MCID <= 364155) or
           (366010 <= MCID <= 366035) or
           (364222 <= MCID <= 364223)
            ):
            decayType = "nolepton"
 
        elif (
               (361300 <= MCID <= 361371) or
               (363331 <= MCID <= 363354) or
               (363436 <= MCID <= 363459) or
               (363460 <= MCID <= 363483) or
               (364156 <= MCID <= 364169) or
               (364170 <= MCID <= 364183) or
               (364184 <= MCID <= 364197) or
           (364226 <= MCID <= 364227) or
               (364224 <= MCID <= 364225) or
           (364228 <= MCID <= 364229)
            ):
            decayType = "onelepton"
 
        else:
            decayType = "none"
 
        # find which ZpT bin we want
        ZpTBin = ZpTMapping.get(MCID)
        TruthJetBin = nTruthJets+1
        if nTruthJets >= 11:
            TruthJetBin = 12
        if ZpTBin == None:
            #this MCID isn't in the mapping, so skip and default to 1
            ckkw15[0] = 1
            ckkw30[0] = 1
            fac025[0] = 1
            fac4[0] = 1
            renorm025[0] = 1
            renorm4[0] = 1
            qsf025[0] = 1
            qsf4[0] = 1
 
        else:
            if decayType == "dilepton":
                ckkw15[0] = ZeeHistograms[0].GetBinContent(ZpTBin,TruthJetBin)
                ckkw30[0] =  ZeeHistograms[1].GetBinContent(ZpTBin,TruthJetBin)
                fac025[0] =  ZeeHistograms[2].GetBinContent(ZpTBin,TruthJetBin)
                fac4[0] =  ZeeHistograms[3].GetBinContent(ZpTBin,TruthJetBin)
                renorm025[0] =  ZeeHistograms[4].GetBinContent(ZpTBin,TruthJetBin)
                renorm4[0] =  ZeeHistograms[5].GetBinContent(ZpTBin,TruthJetBin)
                qsf025[0] =  ZeeHistograms[6].GetBinContent(ZpTBin,TruthJetBin)
                qsf4[0] =  ZeeHistograms[7].GetBinContent(ZpTBin,TruthJetBin)
            elif decayType == "nolepton":
                ckkw15[0] = ZnunuHistograms[0].GetBinContent(ZpTBin,TruthJetBin)
                ckkw30[0] =  ZnunuHistograms[1].GetBinContent(ZpTBin,TruthJetBin)
                fac025[0] =  ZnunuHistograms[2].GetBinContent(ZpTBin,TruthJetBin)
                fac4[0] =  ZnunuHistograms[3].GetBinContent(ZpTBin,TruthJetBin)
                renorm025[0] =  ZnunuHistograms[4].GetBinContent(ZpTBin,TruthJetBin)
                renorm4[0] =  ZnunuHistograms[5].GetBinContent(ZpTBin,TruthJetBin)
                qsf025[0] =  ZnunuHistograms[6].GetBinContent(ZpTBin,TruthJetBin)
                qsf4[0] =  ZnunuHistograms[7].GetBinContent(ZpTBin,TruthJetBin)
            elif decayType == "onelepton":
                ckkw15[0] = WenuHistograms[0].GetBinContent(ZpTBin,TruthJetBin)
                ckkw30[0] =  WenuHistograms[1].GetBinContent(ZpTBin,TruthJetBin)
                fac025[0] =  WenuHistograms[2].GetBinContent(ZpTBin,TruthJetBin)
                fac4[0] =  WenuHistograms[3].GetBinContent(ZpTBin,TruthJetBin)
                renorm025[0] =  WenuHistograms[4].GetBinContent(ZpTBin,TruthJetBin)
                renorm4[0] =  WenuHistograms[5].GetBinContent(ZpTBin,TruthJetBin)
                qsf025[0] =  WenuHistograms[6].GetBinContent(ZpTBin,TruthJetBin)
                qsf4[0] =  WenuHistograms[7].GetBinContent(ZpTBin,TruthJetBin)
            else:
                print '\x1b[2K\r', "Missing entry for MCID={0}, nTruthJets={1}".format(MCID, nTruthJets)
                exit(1)
 
        ckkw15branch.Fill()
        ckkw30branch.Fill()
        fac025branch.Fill()
        fac4branch.Fill()
        renorm025branch.Fill()
        renorm4branch.Fill()
        qsf025branch.Fill()
        qsf4branch.Fill()
 
    print '\x1b[2K\r', "All entries processed"
    T.Write()
 
    print "Done with {0:s}/{1:s}.".format(infile, treename)
    f.Close()
 

Variable Documentation

__author__

str AddZJetsWeights.__author__ = "John Anders & Jack Lindon"

private

Definition at line 11 of file AddZJetsWeights.py.

__doc__

AddZJetsWeights.__doc__ = """Script to add Sherpa Systematic Weight branches to existing file."""

private

Definition at line 12 of file AddZJetsWeights.py.

args

AddZJetsWeights.args = parser.parse_args()

Definition at line 1089 of file AddZJetsWeights.py.

default

AddZJetsWeights.default

Definition at line 1084 of file AddZJetsWeights.py.

description

AddZJetsWeights.description

Definition at line 1081 of file AddZJetsWeights.py.

False

AddZJetsWeights.False

Definition at line 1084 of file AddZJetsWeights.py.

formatter_class

AddZJetsWeights.formatter_class

Definition at line 1082 of file AddZJetsWeights.py.

help

AddZJetsWeights.help

Definition at line 1083 of file AddZJetsWeights.py.

metavar

AddZJetsWeights.metavar

Definition at line 1084 of file AddZJetsWeights.py.

parser

AddZJetsWeights.parser

Definition at line 1081 of file AddZJetsWeights.py.

required

AddZJetsWeights.required

Definition at line 1084 of file AddZJetsWeights.py.

str

AddZJetsWeights.str

Definition at line 1083 of file AddZJetsWeights.py.

type

AddZJetsWeights.type

Definition at line 1083 of file AddZJetsWeights.py.

usage

AddZJetsWeights.usage

Definition at line 1081 of file AddZJetsWeights.py.