makeCutflowTables.py

import sys
from subprocess import call
from datetime import date
import CatFormats

def write_syst(file,value):
    file.write(value[0]+' &')
    for val in value[1:]:
      file.write(val+" &")
    file.write('\\\\ \n')

def write_foot(file):
    file.write('\\hline\n')
    file.write('\end{tabular}\n')
    file.write('\\end{center}\n')

def write_head(file,columns):
    #print columns
    file.write('\\begin{center}\n')
    file.write('\\begin{tabular}{l')
    for entry in columns[1:]:
        file.write('c')
    file.write('}\n')
    file.write('\\hline\n')
    for entry in columns[:-1]:
        file.write(entry+' &')
    file.write(columns[-1]+' \\\\ \n')
    file.write('\\hline\n')

def isfloat(value):
  try:
    float(value)
    return True
  except ValueError:
    return False



##begin script 
if len(sys.argv)<=2:
  print "usage:"
  print "python makeCutflowTables.py OUPUTFILE_WITHOUT_EXTENSION LIST_OF_INPUT_CUTFLOWS"
  exit(0)

first=True
samples=[]
suffixes=["MC_MadGraph_","MC_aMCatNLO_","MC_powheg_","MC_Powheg_","MC_PowHeg_"]
affixes=["_nominal_Cutflow.txt","_JESDOWN_Cutflow.txt","_JESUP_Cutflow.txt"]
outfile = sys.argv[1]
if "/" not in outfile:
  outfile="./"+outfile
for filename in sys.argv[2:]:
  sample=[]
  steps=[]
  nGen=[]
  yields=[]
  lines=[]
  infile=open(filename,"r")
  rawlines=list(infile)
  for rawline in rawlines:
    rawline=rawline.replace("\n","")
    lines.append(rawline)
  print filename
  if "/" in filename:
    samplename=filename.rsplit("/",1)[1]
  else:
    samplename=filename
  for suffix in suffixes:
    if suffix in samplename:
      samplename=samplename.replace(suffix,"")
  for affix in affixes:
    if affix in samplename:
      samplename=samplename.replace(affix,"")
  #print samplename
  sample.append(samplename)
  for line in lines:
    print line
    splitline=line.split(" : ")
    steps.append(splitline[1])
  for line in lines:
    splitline=line.split(" : ")
    nGen.append(splitline[2])
  for line in lines:
    splitline=line.split(" : ")
    buffval=splitline[3]
    if isfloat(buffval):
      buffval=round(float(buffval),2)
    yields.append(str(buffval))
    print "here", yields
  sample.append(steps)
  sample.append(nGen)
  sample.append(yields)
  print sample
  samples.append(sample)

#print samples

for sample in samples:
  if len(sample[1])!=len(samples[0][1]):
    print sample[0], " has the wrong number of cutflow steps"
    exit(1)

out=open(outfile+"_events.tex","w")
out.write( '\\documentclass{article}\n')
out.write( '\\begin{document}')
s=[]
for sample in samples:
  print sample[0]
  printedname=sample[0]
  for sn in CatFormats.sampleNames:
    if printedname==sn[0]:
      printedname=sn[1]
      break
  s.append(printedname)
#print s
write_head(out,["cutflow steps"]+s)
for step in samples[0][1]:
  line=[]
  line.append(step.replace(">","$>$").replace("<","$<$"))
  i=samples[0][1].index(step)
  for s in samples:
    line.append(s[2][i])
  write_syst(out,line)
write_foot(out)
out.write("\\end{document}")
out.close()

call(["pdflatex","-interaction","batchmode","-output-directory",outfile.rsplit("/",1)[0],outfile+"_events.tex"])
call(["pdftops",outfile+"_events.pdf",outfile+"_events.eps"])
print "did the table for the number of events"


out=open(outfile+"_Cutflows.tex","w")
out.write( '\\documentclass{article}\n')
out.write( '\\begin{document}')
s=[]
for sample in samples:
  printedname=sample[0]
  for sn in CatFormats.sampleNames:
    if printedname==sn[0]:
      printedname=sn[1]
      break
  s.append(printedname)
#print s
write_head(out,["cutflow steps"]+s)
for step in samples[0][1]:
  line=[]
  line.append(step.replace(">","$>$").replace("<","$<$"))
  i=samples[0][1].index(step)
  for s in samples:
    print s
    line.append(s[3][i])
  write_syst(out,line)
write_foot(out)
out.write("\\end{document}")
out.close()

call(["pdflatex","-interaction","batchmode","-output-directory",outfile.rsplit("/",1)[0],outfile+"_Cutflows.tex"])
call(["pdftops",outfile+"_Cutflows.pdf",outfile+"_Cutflows.eps"])
print "did the table for the yields"