return w, c1
if __name__ == '__main__':
##########################
if args.DTF:
DTF_label = '_DTF'
x_var = 'Tau_DTF_Tau_M'
addTitlePlot = 'DTF'
else:
DTF_label = ''
x_var = 'Tau_M'
addTitlePlot = ''
##########################
# Make Dataset
if args.dataset:
print 'Making RooDataSet'
dataSet = makeRooDataset('/afs/cern.ch/work/g/gdujany/LHCb/LFV/store/data2012.root')
dataSet.SaveAs('RooDataSets/rooDataSet.root')
# Make fit
w = TFile('pickles/signalShape'+DTF_label+'.root').Get('w')
dataSet = TFile('RooDataSets/rooDataSet.root').Get('taus')
cuts_str = ''#'(Mu_ProbNNmu - 4./3.*Mu_ProbNNpi) > 0.6'
dataSet = dataSet.reduce( RooFit.Cut(cuts_str) )
w, c1 = doDataFit(dataSet, w, x_var, addTitlePlot)
c1.Print('plots/plotDataFit'+DTF_label+'.pdf')
