compareCafs.py

Go to the documentation of this file.

from ROOT import *
import sys, os
import cafe
import canMan
import novaStyle
from collections import OrderedDict





vars = OrderedDict()
"""
vars["nhit"]            =  "trk.kalman.nhit"
vars["nhitx"]           =  "trk.kalman.nhitx"
vars["nhity"]           =  "trk.kalman.nhity"
vars["nplane"]          =  "trk.kalman.nplane"
vars["maxplanegap"]     =  "trk.kalman.maxplanegap"
vars["maxplanecont"]    =  "trk.kalman.maxplanecont"

vars["nhitc"]            =  "trk.cosmic.nhit"
vars["nhitxc"]           =  "trk.cosmic.nhitx"
vars["nhityc"]           =  "trk.cosmic.nhity"
vars["nplanec"]          =  "trk.cosmic.nplane"
vars["maxplanegapc"]     =  "trk.cosmic.maxplanegap"
vars["maxplanecontc"]    =  "trk.cosmic.maxplanecont"

"""
leaves = None

histSets = []
files = []
canMan.newCan("bogus")
for filePath in [filePath for filePath in sys.argv[1:] if ".root" in filePath]:
  histSet = OrderedDict()
  histSets.append(histSet)
  fileId = canMan.getFileId(filePath)
  tFile = TFile(filePath, "READ")
  files.append(tFile)
  tree = tFile.Get("recTree")
  if not leaves:
    leaves = tree.GetListOfLeaves()
    for leaf in leaves:
      if "_" in leaf.GetName():
        continue
      if "fUnique" in leaf.GetName():
        continue
      if "fBits" in leaf.GetName():
        continue
      vars[leaf.GetName()] = "rec." + leaf.GetName()

  for varKey in vars.keys():

    histName = fileId + varKey
    
    tree.Draw(vars[varKey] + ">>" + histName, "", "")
    
    histSet[varKey] = gDirectory.Get(histName)

differences = open("caf_differences.txt", 'w')

texts = []
for varKey in vars.keys():
  hists = [histSet[varKey] for histSet in histSets]
  canMan.drawOverlay(hists)
  boxHeight = 0.075 * (len(hists) + 1)
  text = TPaveText()
  text.SetX1NDC(0.7)
  text.SetY1NDC(0.8 - boxHeight)
  text.SetX2NDC(0.85)
  text.SetY2NDC(0.8)
  text.AddText("Means")
  lastMeanStr = None 
  for hist in hists:
    meanStr = str(round(hist.GetMean(), 4))
    text.AddText(meanStr)
    if not lastMeanStr == None:
      if not meanStr == lastMeanStr:
        print vars[varKey], meanStr, lastMeanStr
        differences.write(vars[varKey]+"   "+meanStr+"   "+lastMeanStr+'\n')
    lastMeanStr = meanStr
    
    
  text.Draw()
  texts.append(text)
novaStyle.prettyStat("emr")
canMan.canUp()  
del canMan.cans[0]