RecoVertex/test/vertexCompare.py

0001 #!/usr/bin/env python3
0002
0003 # This is an example of plotting the standard vertex validation
0004 # plots from an explicit set of DQM root files.
0005
0006 import Validation.RecoTrack.plotting.plotting as plotting
0007 from Validation.RecoTrack.plotting.validation import SimpleValidation
0008 import Validation.RecoVertex.plotting.vertexPlots as vertexPlots
0009
0010
0011 # Example of file - label pairs
0012 filesLabels = [
0013     ("DQM_V0001_R000000001__Global__CMSSW_X_Y_Z__RECO_1.root", "Option 1"),
0014     ("DQM_V0001_R000000001__Global__CMSSW_X_Y_Z__RECO_2.root", "Option 2"),
0015 ]
0016
0017 outputDir = "plots"
0018
0019 ### Track algorithm name and quality. Can be a list.
0020 Collections = ["offlinePrimaryVertices", "selectedOfflinePrimaryVertices"]
0021 Qualities=None
0022
0023 def newdirname(algo, quality):
0024     ret = ""
0025     if algo is not None:
0026         ret += "_"+algo
0027     return ret
0028
0029
0030 val = SimpleValidation([x[0] for x in filesLabels], [x[1] for x in filesLabels], outputDir)
0031 val.doPlots(Collections, Qualities, vertexPlots.plotter, algoDirMap=lambda a, q: a, newdirFunc=newdirname)