PatCandidates/test/dump_parameterization.py

0001 #!/bin/env python
0002 from builtins import range
0003 import ROOT
0004 from ROOT import gROOT, TFile, TIter
0005 import sys
0006 if len(sys.argv) < 2 :
0007   print("Usage: dump_parameteriza.py filename")
0008   exit()
0009 fileToRead= TFile(sys.argv[1], "read")
0010
0011 fileToRead.cd()
0012
0013
0014
0015 SchemasMeaningFile = open("SchemasMeaning.txt", "w")
0016 SchemasMeaningFile.write("------------------------LEGEND-------------------------  \n")
0017 SchemasMeaningFile.write("METHOD:  \n")
0018 SchemasMeaningFile.write("\t0 \t float16  \n")
0019 SchemasMeaningFile.write("\t1 \t reduceMantissa  \n")
0020 SchemasMeaningFile.write("\t2 \t logPack  \n")
0021 SchemasMeaningFile.write("\t3 \t tanLogPack  \n")
0022 SchemasMeaningFile.write("\t4 \t zero  \n")
0023 SchemasMeaningFile.write("\t5 \t one  \n")
0024
0025
0026 SchemasMeaningFile.write("TARGET:  \n")
0027 SchemasMeaningFile.write("\t0 \t realValue  \n")
0028 SchemasMeaningFile.write("\t1 \t ratioToRef  \n")
0029 SchemasMeaningFile.write("\t2 \t differenceToRef  \n\n")
0030
0031
0032
0033 next = TIter ((fileToRead.Get("schemas")).GetListOfKeys())
0034
0035
0036 key = next()
0037 while key != None  :
0038
0039     cl = gROOT.GetClass(key.GetClassName())
0040     schemaNumber = key.ReadObj().GetName();
0041     schemaN = int(schemaNumber);
0042
0043     SchemasMeaningFile.write("SCHEMA " + schemaNumber + "\n")
0044     SchemasMeaningFile.write("ELEMENT \t METHOD \tTARGET  \tBIT USED \t  Lmin \t\t  Lmax" + "\n")
0045
0046
0047     for i in range(0,5) :
0048         for j in range(i,5) :
0049             folder = "schemas/"+schemaNumber+"/"+str(i)+str(j)
0050
0051
0052
0053             method = str(fileToRead.Get(folder+"/method").GetVal())
0054             target = str(fileToRead.Get(folder+"/target").GetVal())
0055             bit    = str(fileToRead.Get(folder+"/bit").GetVal())
0056             param0 = str(fileToRead.Get(folder+"/param")[0])
0057             param1 = str(fileToRead.Get(folder+"/param")[1])
0058
0059
0060             SchemasMeaningFile.write(str(i)+str(j) + "\t\t " + method + "\t\t  " + target + "\t\t  " + bit + "\t\t   " + param0 + "  \t  " + param1 + "\n")
0061
0062
0063
0064     key = next()
0065
0066
0067 fileToRead.Close()
0068 SchemasMeaningFile.close()
0069
0070
0071
0072
0073