besDS = h5f.create_dataset('BES_vars', data=jetDF['BES_vars'], maxshape=(None, len(besKeys)), compression='lzf') else: # append the dataset besDS.resize(besDS.shape[0] + len(jetDF['BES_vars']), axis=0) besDS[-len(jetDF['BES_vars']) :] = jetDF['BES_vars'] 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] :
h5f = h5py.File("data/phiCosThetaBoostedJetImages10by10.h5", "w") h5f.create_dataset('QCD', data=jetImagesDF['QCD'], compression='lzf') 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)
#================================================================================== # 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!!!"