cafExposure.py

Go to the documentation of this file.

#! /usr/bin/env python 
import argparse
import os
from collections import OrderedDict

parser = argparse.ArgumentParser(description='Count the events/POT/exposure in the files specified by a set of constraints')
parser.add_argument('--dir', '-d', help='Directory of files', type=str)
parser.add_argument('--match', '-m', help='Match pattern for files', type=str, action='append')
parser.add_argument('--printName', '-p', help='Prints filename of counted files', action='store_true')
parser.add_argument('files', nargs = '*',
                    help = 'List of files to operate on. Or use --dir and --match.')
args = parser.parse_args()

# ROOT hijacks the help text, so delay importing it until argparse has done its
# thing.
import ROOT
ROOT.gErrorIgnoreLevel=100000

if args.files and args.dir:
  print 'Error: --dir is not compatible with explicitly listing files. Use one or the other'
  exit(1)

if not args.files and not args.dir:
  print 'Must specify either --dir or a list of files. See --help text'
  exit(1)

if not args.match:
  args.match = []

fileList = []

if args.files:
  fileList = args.files

if args.dir:
  fileList = [os.path.join(args.dir, file) for file in os.listdir(args.dir)]


if not fileList:
  raise Exception("No file list found. Need to pass in a directory with --dir")

hists = OrderedDict()
hists["TotalPOT"] = ROOT.TH1F("TotalPOT", ";Arbitrary;Total POT", 1, 0,1)
hists["TotalEvents"] = ROOT.TH1F("TotalEvents", ";Arbitrary; Total Events", 1,0,1)
hists["TotalLivetime"] = ROOT.TH1F("TotalLivetime", ";Arbitrary; Total Livetime", 1,0,1)

nFiles = 0
for fileName in fileList: 
  if not ".caf." in os.path.basename(fileName):
    continue 

  match = True
  for string in args.match:
    if not string in os.path.basename(fileName):
      match = False 
      break
  if not match:
    continue
  file = ROOT.TFile.Open(fileName, "READ")
  if args.printName:
    print os.path.basename(fileName)
  if file and not (file.IsZombie() or file.TestBit(ROOT.TFile.kRecovered)):
    nFiles += 1
    for histKey in hists.keys():
      hists[histKey].Add(file.Get(histKey))

print 'Number of files :', nFiles
for histKey in hists.keys():
  print histKey, ":", hists[histKey].Integral()