Inspector.py

Go to the documentation of this file.

from __future__ import print_function
#____________________________________________________________
#
#  cuy
#
# A very simple way to make plots with ROOT via an XML file
#
# Francisco Yumiceva
# yumiceva@fnal.gov
#
# Fermilab, 2008
#
#____________________________________________________________
 
import sys
import ROOT
from ROOT import TFile
 
 
class Inspector:
 
    def SetFilename(self, value):
        self.Filename = value
    def Verbose(self, value):
        self.Verbose = value
 
    def createXML(self, value):
        self.XML = value
 
    def SetTag(self,value):
        self.tag = value
        self.TagOption = True
 
    def Loop(self):
 
        afile = TFile(self.Filename)
        afilename = self.Filename
        stripfilename = afilename
 
        try:
            if self.TagOption:
                stripfilename = self.tag
        except:
            stripfilename = afilename.split('/')[len(afilename.split('/')) -1]
            stripfilename = stripfilename[0:(len(stripfilename)-5)]
 
        alist = self.dir.GetListOfKeys()
 
        for i in alist:
            aobj = i.ReadObj()
            if aobj.IsA().InheritsFrom("TDirectory"):
                if self.Verbose:
                    print(' found directory: '+i.GetName())
 
                if self.XML:
                    print('   <!-- '+i.GetName()+' -->')
 
                bdir = self.dir
                afile.GetObject(i.GetName(),bdir)
                blist = bdir.GetListOfKeys()
                for j in blist:
                    bobj = j.ReadObj()
                    if bobj.IsA().InheritsFrom(ROOT.TH1.Class()):
                        if self.Verbose:
                            print('  --> found TH1: name = '+j.GetName() + ' title = '+j.GetTitle())
                        if self.XML:
                            print('   <TH1 name=\"'+stripfilename+'_'+j.GetName()+'\" source=\"'+'/'+i.GetName()+'/'+j.GetName()+'\"/>')
 
    def GetListObjects(self):
 
        afile = TFile(self.Filename)
 
        if afile.IsZombie():
            print(" error trying to open file: " + self.Filename)
            sys.exit()
 
        if self.XML:
 
            print('''
<cuy>
''')    
            print('  <validation type=\"'+afile.GetName()+'\" file=\"'+self.Filename+'\" release=\"x.y.z\">')
 
        self.dir = ROOT.gDirectory
        self.Loop()
 
        if self.XML:
 
            print('''
  </validation>
 
</cuy>
''')