def plot_TstEdges():
plot_Tst()
for tracksterCollection in trackstersIters:
hgctracksters = [
hgcalPlots.create_hgcalTrackstersPlotter(
sample.files(), tracksterCollection, tracksterCollection)
]
val.doPlots(hgctracksters, plotterDrawArgs=drawArgs)
def main(opts):
drawArgs={}
extendedFlag = False
if opts.no_ratio:
drawArgs["ratio"] = False
if opts.separate:
drawArgs["separate"] = True
if opts.png:
drawArgs["saveFormat"] = ".png"
if opts.extended:
extendedFlag = True
if opts.verbose:
plotting.verbose = True
filenames = [(f, f.replace(".root", "")) for f in opts.files]
sample = SimpleSample(opts.subdirprefix[0], opts.html_sample, filenames)
val = SimpleValidation([sample], opts.outputDir[0])
if opts.separate:
val = SeparateValidation([sample], opts.outputDir[0])
htmlReport = val.createHtmlReport(validationName=opts.html_validation_name[0])
#layerClusters
if (opts.collection == layerClustersGeneralLabel):
hgclayclus = [hgcalPlots.hgcalLayerClustersPlotter]
hgcalPlots.append_hgcalLayerClustersPlots("hgcalLayerClusters", "Layer Clusters", extendedFlag)
val.doPlots(hgclayclus, plotterDrawArgs=drawArgs)
#simClusters
elif (opts.collection == simClustersGeneralLabel):
hgcsimclus = [hgcalPlots.hgcalSimClustersPlotter]
for i_iter in simClustersIters:
hgcalPlots.append_hgcalSimClustersPlots(i_iter, i_iter)
val.doPlots(hgcsimclus, plotterDrawArgs=drawArgs)
#multiClusters
elif (opts.collection == multiclustersGeneralLabel):
hgcmulticlus = [hgcalPlots.hgcalMultiClustersPlotter]
hgcalPlots.append_hgcalMultiClustersPlots(multiclustersGeneralLabel, "MultiClusters")
val.doPlots(hgcmulticlus, plotterDrawArgs=drawArgs)
#ticlTracksters
elif (opts.collection == trackstersGeneralLabel):
hgcmulticlus = [hgcalPlots.hgcalMultiClustersPlotter]
for i_iter in trackstersIters :
tracksterCollection = i_iter.replace("ticlMultiClustersFromTracksters","ticlTracksters")
hgcalPlots.append_hgcalMultiClustersPlots(i_iter, tracksterCollection)
val.doPlots(hgcmulticlus, plotterDrawArgs=drawArgs)
# TICLTrackstersEdges plots
for i_iter in trackstersIters :
tracksterCollection = i_iter.replace("ticlMultiClustersFromTracksters","ticlTracksters")
hgctracksters = [hgcalPlots.create_hgcalTrackstersPlotter(sample.files(), tracksterCollection, tracksterCollection)]
val.doPlots(hgctracksters, plotterDrawArgs=drawArgs)
elif (opts.collection == caloParticlesLabel):
particletypes = {"pion-":"-211", "pion+":"211", "pion0": "111",
"muon-": "-13", "muon+":"13",
"electron-": "-11", "electron+": "11", "photon": "22",
"kaon0L": "310", "kaon0S": "130",
"kaon-": "-321", "kaon+": "321"}
hgcaloPart = [hgcalPlots.hgcalCaloParticlesPlotter]
for i_part, i_partID in particletypes.iteritems() :
hgcalPlots.append_hgcalCaloParticlesPlots(sample.files(), i_partID, i_part)
val.doPlots(hgcaloPart, plotterDrawArgs=drawArgs)
#hitValidation
elif (opts.collection == hitValidationLabel):
hgchit = [hgcalPlots.hgcalHitPlotter]
hgcalPlots.append_hgcalHitsPlots('HGCalSimHitsV', "Simulated Hits")
hgcalPlots.append_hgcalHitsPlots('HGCalRecHitsV', "Reconstruced Hits")
hgcalPlots.append_hgcalDigisPlots('HGCalDigisV', "Digis")
val.doPlots(hgchit, plotterDrawArgs=drawArgs)
#hitCalibration
elif (opts.collection == hitCalibrationLabel):
hgchitcalib = [hgcalPlots.hgcalHitCalibPlotter]
val.doPlots(hgchitcalib, plotterDrawArgs=drawArgs)
elif (opts.collection == allLabel):
#caloparticles
particletypes = {"pion-":"-211", "pion+":"211", "pion0": "111",
"muon-": "-13", "muon+":"13",
"electron-": "-11", "electron+": "11", "photon": "22",
"kaon0L": "310", "kaon0S": "130",
"kaon-": "-321", "kaon+": "321"}
hgcaloPart = [hgcalPlots.hgcalCaloParticlesPlotter]
for i_part, i_partID in particletypes.iteritems() :
hgcalPlots.append_hgcalCaloParticlesPlots(sample.files(), i_partID, i_part)
val.doPlots(hgcaloPart, plotterDrawArgs=drawArgs)
#hits
hgchit = [hgcalPlots.hgcalHitPlotter]
hgcalPlots.append_hgcalHitsPlots('HGCalSimHitsV', "Simulated Hits")
hgcalPlots.append_hgcalHitsPlots('HGCalRecHitsV', "Reconstruced Hits")
hgcalPlots.append_hgcalDigisPlots('HGCalDigisV', "Digis")
val.doPlots(hgchit, plotterDrawArgs=drawArgs)
#calib
hgchitcalib = [hgcalPlots.hgcalHitCalibPlotter]
val.doPlots(hgchitcalib, plotterDrawArgs=drawArgs)
#simClusters
hgcsimclus = [hgcalPlots.hgcalSimClustersPlotter]
for i_iter in simClustersIters :
hgcalPlots.append_hgcalSimClustersPlots(i_iter, i_iter)
val.doPlots(hgcsimclus, plotterDrawArgs=drawArgs)
#layer clusters
hgclayclus = [hgcalPlots.hgcalLayerClustersPlotter]
hgcalPlots.append_hgcalLayerClustersPlots("hgcalLayerClusters", "Layer Clusters", extendedFlag)
val.doPlots(hgclayclus, plotterDrawArgs=drawArgs)
#multiclusters
hgcmulticlus = [hgcalPlots.hgcalMultiClustersPlotter]
for i_iter in trackstersIters :
tracksterCollection = i_iter.replace("ticlMultiClustersFromTracksters","ticlTracksters")
hgcalPlots.append_hgcalMultiClustersPlots(i_iter, tracksterCollection)
val.doPlots(hgcmulticlus, plotterDrawArgs=drawArgs)
#TICLTrackstersEdges plots
for i_iter in trackstersIters :
tracksterCollection = i_iter.replace("ticlMultiClustersFromTracksters","ticlTracksters")
hgctracksters = [hgcalPlots.create_hgcalTrackstersPlotter(sample.files(), tracksterCollection, tracksterCollection)]
val.doPlots(hgctracksters, plotterDrawArgs=drawArgs)
if opts.no_html:
print("Plots created into directory '%s'." % opts.outputDir)
else:
htmlReport.write()
print("Plots and HTML report created into directory '%s'. You can just move it to some www area and access the pages via web browser" % (','.join(opts.outputDir)))