コード例 #1
0
ファイル: stack_dtmc.py プロジェクト: yw5mj/cmgtools-lite
Stack.setLog(LogY)
Stack.doRatio(doRatio)

###
fout = ROOT.TFile(tag + '.root', 'recreate')

# electron
h_Data_el = Data.drawTH3('h3d_data_obs_el',
                         'llnunu_mta:met_pt:llnunu_l1_pt',
                         cuts + '&&' + elChannel,
                         '(1)',
                         250,
                         0,
                         2500,
                         300,
                         0,
                         3000,
                         500,
                         0,
                         5000,
                         titlex="P_{T}(Z)",
                         unitsx="GeV",
                         titley="MET",
                         unitsy="GeV",
                         drawStyle="COLZ")
h_Data_el.Write()

h_VVNonReso_el = WW.drawTH3('h3d_VVNonReso_el',
                            'llnunu_mta:met_pt:llnunu_l1_pt',
                            cuts + '&&' + elChannel,
                            str(lumi * 1000),
                            250,
コード例 #2
0
data = MergedPlotter(dataPlotters)

pvars = options.vars.split(',')
pmins = options.mins.split(',')
pmaxes = options.maxes.split(',')
pbins = options.bins.split(',')

if len(pvars) == 1:
    histo = data.drawTH1(pvars[0], options.cut, "1", int(pbins[0]),
                         float(pmins[0]), float(pmaxes[0]))

if len(pvars) == 2:
    histo = data.drawTH2(pvars[1] + ":" + pvars[0], options.cut, "1",
                         int(pbins[0]), float(pmins[0]), float(pmaxes[0]),
                         int(pbins[1]), float(pmins[1]), float(pmaxes[1]))

if len(pvars) == 3:
    histo = data.drawTH3(pvars[2] + ":" + pvars[1] + ":" + pvars[0],
                         options.cut, "1", int(pbins[0]), float(pmins[0]),
                         float(pmaxes[0]), int(pbins[1]), float(pmins[1]),
                         float(pmaxes[1]), int(pbins[2]), float(pmins[2]),
                         float(pmaxes[2]))

histo.Scale(options.factor)

F = ROOT.TFile(options.output, "RECREATE")
F.cd()
histo.Write("histo")
F.Close()