print "Converted jets: ", besDS.shape[0] - len(jetDF['BES_vars']), " to ", besDS.shape[0]
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# Only plot jet Images for the first iteration (50,000)
if plotJetImages == True and numIter == 0:
print "Plotting Averaged images for the first 50,000 jets "
img.plotAverageBoostedJetImage(jetDF['HiggsFrame_images'], 'ZSample_HiggsFrame', savePNG, savePDF)
img.plotAverageBoostedJetImage(jetDF['TopFrame_images'], 'ZSample_TopFrame', savePNG, savePDF)
img.plotAverageBoostedJetImage(jetDF['WFrame_images'], 'ZSample_WFrame', savePNG, savePDF)
img.plotAverageBoostedJetImage(jetDF['ZFrame_images'], 'ZSample_ZFrame', savePNG, savePDF)
print "Plot the First 3 Images "
img.plotThreeBoostedJetImages(jetDF['HiggsFrame_images'], 'ZSample_HiggsFrame', savePNG, savePDF)
img.plotThreeBoostedJetImages(jetDF['TopFrame_images'], 'ZSample_TopFrame', savePNG, savePDF)
img.plotThreeBoostedJetImages(jetDF['WFrame_images'], 'ZSample_WFrame', savePNG, savePDF)
img.plotThreeBoostedJetImages(jetDF['ZFrame_images'], 'ZSample_ZFrame', savePNG, savePDF)
# increment
numIter += 1
# if the stop iteration option is enabled
if stopAt != None and stopAt <= besDS.shape[0] :
print "This program was told to stop early, please set 'stopAtIter = None' if you want it to run through all files"
break
print "Stored Boosted Event Shape Tagger Inputs"
print "Mischief Managed!!!"
h5f.create_dataset('HH4B', data=jetImagesDF['HH4B'], compression='lzf')
h5f.create_dataset('HH4W', data=jetImagesDF['HH4W'], compression='lzf')
print "Saved Boosted Jet Images"
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# plot with python
if plotJetImages == True:
print "Plotting Average Boosted jet images"
img.plotAverageBoostedJetImage(jetImagesDF['QCD'], 'boost_QCD_10by10',
savePNG, savePDF)
img.plotAverageBoostedJetImage(jetImagesDF['HH4B'], 'boost_HH4B_10by10',
savePNG, savePDF)
img.plotAverageBoostedJetImage(jetImagesDF['HH4W'], 'boost_HH4W_10by10',
savePNG, savePDF)
img.plotThreeBoostedJetImages(jetImagesDF['QCD'], 'boost_QCD_10by10',
savePNG, savePDF)
img.plotThreeBoostedJetImages(jetImagesDF['HH4B'], 'boost_HH4B_10by10',
savePNG, savePDF)
img.plotThreeBoostedJetImages(jetImagesDF['HH4W'], 'boost_HH4W_10by10',
savePNG, savePDF)
#img.plotMolleweideBoostedJetImage(jetImagesDF['QCD'], 'boost_QCD', savePNG, savePDF)
#img.plotMolleweideBoostedJetImage(jetImagesDF['HH4B'], 'boost_HH4B', savePNG, savePDF)
#img.plotMolleweideBoostedJetImage(jetImagesDF['HH4W'], 'boost_HH4W', savePNG, savePDF)
print "Program was a great success!!!"
#==================================================================================
# Store BEST Variables ////////////////////////////////////////////////////////////
#==================================================================================
jetDF['BES_vars'] = upTree.pandas.df([
"jetAK8*", "nSecondaryVertices", "foxwolfram*", "aplanarity*", "thrust*",
"subjet*mass"
])
h5f.create_dataset('BES_vars', data=jetDF['BES_vars'], compression='lzf')
print "Stored Boosted Event Shape variables"
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# plot with python
if plotJetImages == True:
print "Plotting Average Boosted jet images"
img.plotAverageBoostedJetImage(jetDF['jet_images'], 'boost_QCD', savePNG,
savePDF)
img.plotThreeBoostedJetImages(jetDF['jet_images'], 'boost_QCD', savePNG,
savePDF)
img.plotMolleweideBoostedJetImage(jetDF['jet_images'], 'boost_QCD', 31,
savePNG, savePDF)
print "Mischief Managed!!!"
#==================================================================================
# Store Data in h5 file ///////////////////////////////////////////////////////////
#==================================================================================
h5f = h5py.File("images/HH4BphiCosThetaBoostedJetImagesX10.h5", "w")
h5f.create_dataset('HH4B_images',
data=jetImagesDF['HH4B_images'],
compression='lzf')
h5f.create_dataset('HH4B_BES_vars',
data=jetImagesDF['HH4B_BES_vars'],
compression='lzf')
print "Saved HH4B Boosted Jet Images"
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# plot with python
if plotJetImages == True:
print "Plotting Average Boosted jet images"
img.plotAverageBoostedJetImage(jetImagesDF['HH4B_images'], 'boost_HH4B',
savePNG, savePDF)
img.plotThreeBoostedJetImages(jetImagesDF['HH4B_images'], 'boost_HH4B',
savePNG, savePDF)
#img.plotMolleweideBoostedJetImage(jetImagesDF['HH4B'], 'boost_HH4B', savePNG, savePDF)
print "Program was a great success!!!"
if jetDF['Cpp_images'].all() == jetDF['jet_images'].all():
print "The cpp and python jet images match!"
elif jetDF['Cpp_images'].all() != jetDF['jet_images'].all():
print "ERROR: The cpp and python jet images do not match! Something must be wrong with one image making process"
print " 'I stand by what I said ... you would have done well in Slytherin'"
exit(1)
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# plot with python
if plotJetImages == True:
print "Plotting Average Boosted jet images"
img.plotAverageBoostedJetImage(jetDF['jet_images'], 'boost_Test', savePNG,
savePDF)
img.plotAverageBoostedJetImage(jetDF['Cpp_images'], 'cpp_Test', savePNG,
savePDF)
img.plotThreeBoostedJetImages(jetDF['jet_images'], 'boost_Test', savePNG,
savePDF)
img.plotThreeBoostedJetImages(jetDF['Cpp_images'], 'cpp_Test', savePNG,
savePDF)
img.plotMolleweideBoostedJetImage(jetDF['jet_images'], 'boost_Test', 31,
savePNG, savePDF)
print "Mischief Managed!!!"
# Store BEST Variables in DataFrame ///////////////////////////////////////////////
#==================================================================================
jetImagesDF['QCD_BES_vars'] = bestArrayQCD
print "Stored BES variables"
#==================================================================================
# Store Data in h5 file ///////////////////////////////////////////////////////////
#==================================================================================
h5f = h5py.File("images/OpenDataQCDphiCosThetaBoostedJetImagesX10.h5","w")
h5f.create_dataset('QCD_images', data=jetImagesDF['QCD_images'], compression='lzf')
h5f.create_dataset('QCD_BES_vars', data=jetImagesDF['QCD_BES_vars'], compression='lzf')
print "Saved QCD Boosted Jet Images"
#==================================================================================
# Plot Jet Images /////////////////////////////////////////////////////////////////
#==================================================================================
# plot with python
if plotJetImages == True:
print "Plotting Average Boosted jet images"
img.plotAverageBoostedJetImage(jetImagesDF['QCD_images'], 'boost_QCD_OpenData', savePNG, savePDF)
img.plotThreeBoostedJetImages(jetImagesDF['QCD_images'], 'boost_QCD_OpenData', savePNG, savePDF)
#img.plotMolleweideBoostedJetImage(jetImagesDF['QCD'], 'boost_QCD_OpenData', savePNG, savePDF)
print "Program was a great success!!!"
