Python add_argument Examples

Example #1

import ROOT
from PlotParser import parser

ROOT.gROOT.SetBatch(ROOT.kTRUE)
parser.add_argument('-s',
                    metavar="select",
                    dest='select',
                    action="store",
                    required=True,
                    help='prefix for selecting plots',
                    type=str)
parser.add_argument('-d',
                    metavar="option",
                    dest='option',
                    action="store",
                    required=True,
                    help='draw option',
                    type=str)

args = parser.parse_args()

inFileName = args.inputPath
outFileName = args.outputPath
hist_prefix = args.select
drawOption = args.option

inputFile = ROOT.TFile(inFileName, "READ")
histNames = [
    k.GetName() for k in inputFile.GetListOfKeys()
    if hist_prefix in k.GetName()
]

Example #2

File: plot_allHist.py Project: tmrhombus/HcalTupleAnalyzer

import ROOT
from PlotParser import parser
from Utils.rootTools import SetPalette

setPalette = SetPalette()
setPalette("kBird")

parser.add_argument('-d',
                    metavar="drawOption",
                    dest='drawOption',
                    action="store",
                    required=False,
                    help='draw option',
                    type=str,
                    default="colz")
parser.add_argument('-l',
                    dest='logz',
                    action="store_true",
                    required=False,
                    help='draw option',
                    default=False)

ROOT.gROOT.SetBatch(ROOT.kTRUE)
args = parser.parse_args()
inFileName = args.inputPath
outFileName = args.outputPath
drawOption = args.drawOption
setLogZ = args.logz

inputFile = ROOT.TFile(inFileName, "READ")
histNames = [k.GetName() for k in inputFile.GetListOfKeys()]

Example #3

File: plot_PulseShape.py Project: DryRun/HcalTupleAnalyzer

import ROOT
from PlotParser import parser

ROOT.gROOT.SetBatch(ROOT.kTRUE)
parser.add_argument('-s', metavar="select", dest='select',action="store" , required=True, help='prefix for selecting plots',type=str)
parser.add_argument('-d', metavar="option", dest='option',action="store" , required=True, help='draw option',type=str)

args = parser.parse_args()

inFileName = args.inputPath
outFileName = args.outputPath
hist_prefix = args.select
drawOption = args.option

inputFile = ROOT.TFile(inFileName,"READ")
histNames = [k.GetName() for k in inputFile.GetListOfKeys() if hist_prefix in k.GetName() ] 
hists =  [ inputFile.Get(histName) for histName in histNames ]

print "Total Number of Hist: %s"%len(hists)
c = ROOT.TCanvas()
c.Print(outFileName+"[")
for i,hist in enumerate(hists):
	hist.SetStats(0)
	hist.GetYaxis().SetRangeUser(0.,1.2*hist.GetMaximum())
	# hist.SetTitle(hist.GetName().replace("_"," ").replace("EndCap","Central")+" ; TS ; Average fC per rechit")
	hist.Draw(drawOption)
	c.Print(outFileName)
c.Print(outFileName+"]")

Example #4

File: plot_allHist.py Project: DryRun/HcalTupleAnalyzer

import ROOT
from PlotParser import parser
from Utils.rootTools import SetPalette

setPalette = SetPalette()
setPalette("kBird")

parser.add_argument('-d', metavar="drawOption", dest='drawOption',action="store" , required=False, help='draw option',type=str,default="colz")
parser.add_argument('-l', dest='logz',action="store_true" , required=False, help='draw option',default=False)

ROOT.gROOT.SetBatch(ROOT.kTRUE)
args = parser.parse_args()
inFileName = args.inputPath
outFileName = args.outputPath
drawOption = args.drawOption
setLogZ = args.logz

inputFile = ROOT.TFile(inFileName,"READ")
histNames = [k.GetName() for k in inputFile.GetListOfKeys()]
hists = [ inputFile.Get(histName) for histName in histNames]

hists.sort(key=lambda x: x.GetName())

c = ROOT.TCanvas()
if setLogZ:
	c.SetLogz()
c.Print(outFileName+"[")
for i,hist in enumerate(hists):
	hist.SetStats(0)
	hist.Draw(drawOption)