dumpEventsToText.py

Go to the documentation of this file.

from ROOT import *
import cvnie




import argparse

parser = argparse.ArgumentParser(description='Dump a tree to text.')
parser.add_argument('input', type=str, 
      help="""
      Input file.  Will take "cvndump/CVNTrainTree" directory and dump it
       to text.""")

args = parser.parse_args()


infile = args.input

outfileName = infile.replace(".root", ".cvnevents.txt")

outfile = open(outfileName, "w")

rootFile = TFile.Open(infile, "READ")


tree = rootFile.Get("cvndump/CVNTrainTree")

numEntries = tree.GetEntries()
numInputs = None
numOutputs = 11  # Hard code this for now.  Hopefully find a better way later?



for entry in tree:
  if not numInputs:
    numInputs = entry.train.fPMap.fPE.size()
    # Now we know everything, write the first line. 
    outfile.write(" ".join([str(numEntries), str(numInputs), str(numOutputs)]) + "\n")

  inputs = [str(pe) for pe in entry.train.fPMap.fPE]
  outfile.write(" ".join(inputs) + "\n")
  outputs = ["0"] * numOutputs
  outputs[entry.train.fInt] = str(1)
  outfile.write(" ".join(outputs) + "\n")
  

outfile.close()