###################
r.gROOT.ProcessLine('.L ./scripts/tdrStyle.C')
r.setTDRStyle()
#r.TGaxis.SetMaxDigits(3)
r.gStyle.SetOptStat(0)
### Categories to be plotted ###
catList = ['QCD2l', 'ZPlusJet', 'SameSign']
for category in catList:
plotter._directoryList1D = ['Muon', 'Electron', 'GenStudies']
plotter.add_datasets(samples[category], Clear=True)
plotter.set_category(category)
plotter.make_overlays_1D(logScale = doLog, doRatio = doRatio, doEff = doEff)
# Closure plots
#for CR in ['QCD2l', 'ZPlusJet']:
# plotter.make_overlays_diff([(['PROMPT', 'DATA_FAKES'], ['MuNumerEta', 'MuUnevenEtaClosure_{0}'.format(CR)]),
# (['DATA'],['MuNumerEta'])], 'Muon', 'MuClosureEta_{0}'.format(CR))
# plotter.make_overlays_diff([(['PROMPT', 'DATA_FAKES'], ['EleNumerEta', 'EleUnevenEtaClosure_{0}'.format(CR)]),
# (['DATA'],['EleNumerEta'])], 'Electron', 'EleClosureEta_{0}'.format(CR))
# plotter.make_overlays_diff([(['PROMPT', 'DATA_FAKES'], ['MuNumerPt', 'MuUnevenPtClosure_{0}'.format(CR)]),
# (['DATA'],['MuNumerPt'])], 'Muon', 'MuClosurePt_{0}'.format(CR))
# plotter.make_overlays_diff([(['PROMPT', 'DATA_FAKES'], ['EleNumerPt', 'EleUnevenPtClosure_{0}'.format(CR)]),
# (['DATA'],['EleNumerPt'])], 'Electron', 'EleClosurePt_{0}'.format(CR))
# plotter.make_overlays_diff([(['PROMPT', 'DATA_FAKES'], ['MuNumerMet', 'MuMetClosure_{0}'.format(CR)]),
plotter._variableDict['2D_Dilepton'] = ['DileptonM12VsLepPt1', 'DileptonM12VsLepPt2', 'DileptonMVsDeltaPhi12', 'DileptonM12VsQt12',
'DileptonM12VsM23', 'DileptonM12VsM13', 'DileptonM23VsM13', 'DileptonM12VsLepPt3',
'DileptonM13VsLepPt1', 'DileptonM13VsLepPt2', 'DileptonM13VsLepPt3',
'DileptonM23VsLepPt1', 'DileptonM23VsLepPt2', 'DileptonM23VsLepPt3',
'DileptonMos1VsMss', 'DileptonMos2VsMss']
#'DileptonMVsDeltaPhi13', 'DileptonM12VsQt13', 'DileptonMVsDeltaPhi23', 'DileptonM12VsQt23'
###################
### MAKE PLOTS! ###
###################
r.gROOT.SetStyle('Plain')
r.gStyle.SetOptStat(0)
#r.gROOT.ProcessLine('.L ./tdrStyle.C')
#r.setTDRStyle()
for i, cut in enumerate(cutList):
print cut
plotter.set_input_file('histos/' + selection + '_cut' + str(i+1) + '_' + period + '.root')
plotter.set_save_path('../plots/fakes/' + currentDate + '/' + cut)
for category in cat3l:
print '\t' + category
plotter._category = category
plotter.make_overlays_1D(logScale = True, doRatio = True, doEff = False)