diff --git a/main/python/cmdLineUtils.py b/main/python/cmdLineUtils.py
index c1997f4ae932832bea2520bf27322ac0f5390af4..8f9065354c58de7e6709c3803bb552a9ce481fee 100644
--- a/main/python/cmdLineUtils.py
+++ b/main/python/cmdLineUtils.py
@@ -1,9 +1,9 @@
-#!/usr/bin/python
+#!/usr/bin/env python
# ROOT command line tools module: cmdLineUtils
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Contain utils for ROOT command line tools"""
@@ -240,7 +240,7 @@ def dirListSort(dirList):
def keyClassSpliter(rootFile,pathSplitList):
"""
Return a list of directories and a list of keys corresponding
- to the other objects, for rools and rooprint use
+ to the other objects, for rootLs and rooprint use
"""
keyList = []
dirList = []
@@ -264,18 +264,17 @@ def openROOTFile(fileName, mode="read"):
logging.warning("File %s does not exist", fileName)
return theFile
-def openROOTFileCompress(fileName, optDict):
+def openROOTFileCompress(fileName, compress, recreate):
"""
Open a ROOT file (like openROOTFile) with the possibility
to change compression settings
"""
- compressOptionValue = optDict["compress"]
- if compressOptionValue != None and os.path.isfile(fileName):
+ if compress != None and os.path.isfile(fileName):
logging.warning("can't change compression settings on existing file")
return None
- mode = "recreate" if optDict["recreate"] else "update"
+ mode = "recreate" if recreate else "update"
theFile = openROOTFile(fileName, mode)
- if compressOptionValue != None: theFile.SetCompressionSettings(compressOptionValue)
+ if compress != None: theFile.SetCompressionSettings(compress)
return theFile
def joinPathSplit(pathSplit):
@@ -467,13 +466,13 @@ OMITTING_FILE_ERROR = "omitting file '{0}'"
OMITTING_DIRECTORY_ERROR = "omitting directory '{0}'"
OVERWRITE_ERROR = "cannot overwrite non-directory '{0}' with directory '{1}'"
-def copyRootObject(sourceFile,sourcePathSplit,destFile,destPathSplit,optDict,oneSource=False):
+def copyRootObject(sourceFile,sourcePathSplit,destFile,destPathSplit,oneSource,recursive,replace):
"""
Initialize the recursive function 'copyRootObjectRecursive', written to be as unix-like as possible
"""
retcode = 0
isMultipleInput = not (oneSource and sourcePathSplit != [])
- recursiveOption = optDict["recursive"]
+ recursiveOption = recursive
# Multiple input and unexisting or non-directory destination
# TARGET_ERROR
if isMultipleInput and destPathSplit != [] \
@@ -496,7 +495,7 @@ def copyRootObject(sourceFile,sourcePathSplit,destFile,destPathSplit,optDict,one
# to follow the unix copy behaviour
if sourcePathSplit == []:
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit,optDict)
+ destFile,destPathSplit,replace)
else:
setName = ""
if not isMultipleInput and (destPathSplit != [] \
@@ -507,13 +506,13 @@ def copyRootObject(sourceFile,sourcePathSplit,destFile,destPathSplit,optDict,one
if setName != "":
createDirectory(destFile,destPathSplit[:-1]+[setName])
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit[:-1]+[setName],optDict)
+ destFile,destPathSplit[:-1]+[setName],replace)
elif isDirectory(destFile,destPathSplit):
if not isExisting(destFile,destPathSplit+[objectName]):
createDirectory(destFile,destPathSplit+[objectName])
if isDirectory(destFile,destPathSplit+[objectName]):
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit+[objectName],optDict)
+ destFile,destPathSplit+[objectName],replace)
else:
logging.warning(OVERWRITE_ERROR.format( \
objectName,objectName))
@@ -525,14 +524,14 @@ def copyRootObject(sourceFile,sourcePathSplit,destFile,destPathSplit,optDict,one
else:
if setName != "":
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit[:-1],optDict,setName)
+ destFile,destPathSplit[:-1],replace,setName)
elif isDirectory(destFile,destPathSplit):
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit,optDict)
+ destFile,destPathSplit,replace)
else:
setName = destPathSplit[-1]
retcode += copyRootObjectRecursive(sourceFile,sourcePathSplit, \
- destFile,destPathSplit[:-1],optDict,setName)
+ destFile,destPathSplit[:-1],replace,setName)
return retcode
DELETE_ERROR = "object {0} was not existing, so it is not deleted"
@@ -551,7 +550,7 @@ def deleteObject(rootFile,pathSplit):
retcode += 1
return retcode
-def copyRootObjectRecursive(sourceFile,sourcePathSplit,destFile,destPathSplit,optDict,setName=""):
+def copyRootObjectRecursive(sourceFile,sourcePathSplit,destFile,destPathSplit,replace,setName=""):
"""
Copy objects from a file or directory (sourceFile,sourcePathSplit)
to an other file or directory (destFile,destPathSplit)
@@ -561,7 +560,7 @@ def copyRootObjectRecursive(sourceFile,sourcePathSplit,destFile,destPathSplit,op
$ROOTSYS/tutorials/io/copyFiles.C
"""
retcode = 0
- replaceOption = optDict["replace"]
+ replaceOption = replace
for key in getKeyList(sourceFile,sourcePathSplit):
objectName = key.GetName()
if isDirectoryKey(key):
@@ -570,7 +569,7 @@ def copyRootObjectRecursive(sourceFile,sourcePathSplit,destFile,destPathSplit,op
if isDirectory(destFile,destPathSplit+[objectName]):
retcode +=copyRootObjectRecursive(sourceFile, \
sourcePathSplit+[objectName], \
- destFile,destPathSplit+[objectName],optDict)
+ destFile,destPathSplit+[objectName],replace)
else:
logging.warning(OVERWRITE_ERROR.format( \
objectName,objectName))
@@ -620,14 +619,14 @@ DIRECTORY_REMOVE_ERROR = "cannot remove '{0}': Is a directory"
ASK_FILE_REMOVE = "remove '{0}' ? (y/n) : "
ASK_OBJECT_REMOVE = "remove '{0}' from '{1}' ? (y/n) : "
-def deleteRootObject(rootFile,pathSplit,optDict):
+def deleteRootObject(rootFile, pathSplit, interactive, recursive):
"""
Remove the object (rootFile,pathSplit)
-interactive : prompt before every removal
-recursive : allow directory, and ROOT file, removal
"""
retcode = 0
- if not optDict["recursive"] and isDirectory(rootFile,pathSplit):
+ if not recursive and isDirectory(rootFile,pathSplit):
if pathSplit == []:
logging.warning(FILE_REMOVE_ERROR.format(rootFile.GetName()))
retcode += 1
@@ -635,7 +634,7 @@ def deleteRootObject(rootFile,pathSplit,optDict):
logging.warning(DIRECTORY_REMOVE_ERROR.format(pathSplit[-1]))
retcode += 1
else:
- if optDict['interactive']:
+ if interactive:
if pathSplit != []:
answer = raw_input(ASK_OBJECT_REMOVE \
.format("/".join(pathSplit),rootFile.GetName()))
@@ -675,3 +674,708 @@ REPLACE_HELP = "replace object if already existing"
# End of help strings
##########
+
+##########
+# ROOTBROWSE
+
+def _openBrowser(rootFile=None):
+ browser = ROOT.TBrowser()
+ if rootFile: rootFile.Browse(browser)
+ ROOT.PyROOT.TPyROOTApplication.Run(ROOT.gApplication)
+
+def rootBrowse(fileName=None):
+ if fileName:
+ rootFile = openROOTFile(fileName)
+ if not rootFile: return 1
+ _openBrowser(rootFile)
+ rootFile.Close()
+ else:
+ _openBrowser()
+ return 0
+
+# End of ROOTBROWSE
+##########
+
+##########
+# ROOTCP
+
+def _copyObjects(fileName, pathSplitList, destFile, destPathSplit, oneFile, \
+ recursive, replace):
+ retcode = 0
+ destFileName = destFile.GetName()
+ rootFile = openROOTFile(fileName) \
+ if fileName != destFileName else \
+ destFile
+ if not rootFile: return 1
+ ROOT.gROOT.GetListOfFiles().Remove(rootFile) # Fast copy necessity
+ for pathSplit in pathSplitList:
+ oneSource = oneFile and len(pathSplitList)==1
+ retcode += copyRootObject(rootFile, pathSplit, destFile, destPathSplit, \
+ oneSource, recursive, replace)
+ if fileName != destFileName: rootFile.Close()
+ return retcode
+
+def rootCp(sourceList, destFileName, destPathSplit, \
+ compress=None, recreate=False, recursive=False, replace=False):
+ # Check arguments
+ if sourceList == [] or destFileName == "": return 1
+ if recreate and destFileName in [n[0] for n in sourceList]:
+ logging.error("cannot recreate destination file if this is also a source file")
+ return 1
+
+ # Open destination file
+ destFile = openROOTFileCompress(destFileName, compress, recreate)
+ if not destFile: return 1
+ ROOT.gROOT.GetListOfFiles().Remove(destFile) # Fast copy necessity
+
+ # Loop on the root files
+ retcode = 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _copyObjects(fileName, pathSplitList, destFile, destPathSplit, \
+ len(sourceList)==1, recursive, replace)
+ destFile.Close()
+ return retcode
+
+# End of ROOTCP
+##########
+
+##########
+# ROOTEVENTSELECTOR
+
+def _copyTreeSubset(sourceFile,sourcePathSplit,destFile,destPathSplit,firstEvent,lastEvent):
+ """Copy a subset of the tree from (sourceFile,sourcePathSplit)
+ to (destFile,destPathSplit) according to options in optDict"""
+ retcode = changeDirectory(sourceFile,sourcePathSplit[:-1])
+ if retcode != 0: return retcode
+ bigTree = getFromDirectory(sourcePathSplit[-1])
+ nbrEntries = bigTree.GetEntries()
+ # changeDirectory for the small tree not to be memory-resident
+ retcode = changeDirectory(destFile,destPathSplit)
+ if retcode != 0: return retcode
+ smallTree = bigTree.CloneTree(0)
+ if lastEvent == -1:
+ lastEvent = nbrEntries-1
+ for i in xrange(firstEvent, lastEvent+1):
+ bigTree.GetEntry(i)
+ smallTree.Fill()
+ smallTree.Write()
+ return retcode
+
+def _copyTreeSubsets(fileName, pathSplitList, destFile, destPathSplit, first, last):
+ retcode = 0
+ destFileName = destFile.GetName()
+ rootFile = openROOTFile(fileName) \
+ if fileName != destFileName else \
+ destFile
+ if not rootFile: return 1
+ for pathSplit in pathSplitList:
+ if isTree(rootFile,pathSplit):
+ retcode += _copyTreeSubset(rootFile,pathSplit, \
+ destFile,destPathSplit,first,last)
+ if fileName != destFileName: rootFile.Close()
+ return retcode
+
+def rootEventselector(sourceList, destFileName, destPathSplit, \
+ compress=None, recreate=False, first=0, last=-1):
+ # Check arguments
+ if sourceList == [] or destFileName == "": return 1
+ if recreate and destFileName in sourceList:
+ logging.error("cannot recreate destination file if this is also a source file")
+ return 1
+
+ # Open destination file
+ destFile = openROOTFileCompress(destFileName, compress, recreate)
+ if not destFile: return 1
+
+ # Loop on the root file
+ retcode = 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _copyTreeSubsets(fileName, pathSplitList, destFile, destPathSplit, \
+ first, last)
+ destFile.Close()
+ return retcode
+
+# End of ROOTEVENTSELECTOR
+##########
+
+##########
+# ROOTLS
+
+# Ansi characters
+ANSI_BOLD = "\x1B[1m"
+ANSI_BLUE = "\x1B[34m"
+ANSI_GREEN = "\x1B[32m"
+ANSI_END = "\x1B[0m"
+
+# Needed for column width calculation
+ANSI_BOLD_LENGTH = len(ANSI_BOLD+ANSI_END)
+ANSI_BLUE_LENGTH = len(ANSI_BLUE+ANSI_END)
+ANSI_GREEN_LENGTH = len(ANSI_GREEN+ANSI_END)
+
+# Terminal and platform booleans
+IS_TERMINAL = sys.stdout.isatty()
+IS_WIN32 = sys.platform == 'win32'
+
+def isSpecial(ansiCode,string):
+ """Use ansi code on 'string' if the output is the
+ terminal of a not Windows platform"""
+ if IS_TERMINAL and not IS_WIN32: return ansiCode+string+ANSI_END
+ else: return string
+
+def write(string,indent=0,end=""):
+ """Use sys.stdout.write to write the string with an indentation
+ equal to indent and specifying the end character"""
+ sys.stdout.write(" "*indent+string+end)
+
+TREE_TEMPLATE = "{0:{nameWidth}}"+"{1:{titleWidth}}{2:{memoryWidth}}"
+
+def _recursifTreePrinter(tree,indent):
+ """Print recursively tree informations"""
+ listOfBranches = tree.GetListOfBranches()
+ if len(listOfBranches) > 0: # Width informations
+ maxCharName = max([len(branch.GetName()) \
+ for branch in listOfBranches])
+ maxCharTitle = max([len(branch.GetTitle()) \
+ for branch in listOfBranches])
+ dic = { \
+ "nameWidth":maxCharName+2, \
+ "titleWidth":maxCharTitle+4, \
+ "memoryWidth":1}
+ for branch in listOfBranches: # Print loop
+ rec = \
+ [branch.GetName(), \
+ "\""+branch.GetTitle()+"\"", \
+ str(branch.GetTotBytes())]
+ write(TREE_TEMPLATE.format(*rec,**dic),indent,end="\n")
+ _recursifTreePrinter(branch,indent+2)
+
+def _prepareTime(time):
+ """Get time in the proper shape
+ ex : 174512 for 17h 45m 12s
+ ex : 094023 for 09h 40m 23s"""
+ time = str(time)
+ time = '000000'+time
+ time = time[len(time)-6:]
+ return time
+
+MONTH = {1:'Jan',2:'Feb',3:'Mar',4:'Apr',5:'May',6:'Jun', \
+ 7:'Jul',8:'Aug',9:'Sep',10:'Oct',11:'Nov',12:'Dec'}
+LONG_TEMPLATE = \
+ isSpecial(ANSI_BOLD,"{0:{classWidth}}")+"{1:{timeWidth}}" + \
+ "{2:{nameWidth}}{3:{titleWidth}}"
+
+def _rootLsPrintLongLs(keyList,indent,treeListing):
+ """Print a list of Tkey in columns
+ pattern : classname, datetime, name and title"""
+ if len(keyList) > 0: # Width informations
+ maxCharClass = max([len(key.GetClassName()) for key in keyList])
+ maxCharTime = 12
+ maxCharName = max([len(key.GetName()) for key in keyList])
+ dic = { \
+ "classWidth":maxCharClass+2, \
+ "timeWidth":maxCharTime+2, \
+ "nameWidth":maxCharName+2, \
+ "titleWidth":1}
+ date = ROOT.Long(0)
+ for key in keyList:
+ time = ROOT.Long(0)
+ datime = key.GetDatime()
+ datime.GetDateTime(datime.Get(),date,time)
+ time = _prepareTime(time)
+ rec = \
+ [key.GetClassName(), \
+ MONTH[int(str(date)[4:6])]+" " +str(date)[6:]+ \
+ " "+time[:2]+":"+time[2:4], \
+ key.GetName(), \
+ "\""+key.GetTitle()+"\""]
+ write(LONG_TEMPLATE.format(*rec,**dic),indent,end="\n")
+ if treeListing and isTreeKey(key):
+ tree = key.ReadObj()
+ _recursifTreePrinter(tree,indent+2)
+
+##
+# The code of the getTerminalSize function can be found here :
+# https://gist.github.com/jtriley/1108174
+# Thanks jtriley !!
+
+import os
+import shlex
+import struct
+import platform
+import subprocess
+
+def getTerminalSize():
+ """ getTerminalSize()
+ - get width and height of console
+ - works on linux,os x,windows,cygwin(windows)
+ originally retrieved from:
+ http://stackoverflow.com/questions/566746/how-to-get-console-window-width-in-python"""
+ current_os = platform.system()
+ tuple_xy = None
+ if current_os == 'Windows':
+ tuple_xy = _get_terminal_size_windows()
+ if tuple_xy is None:
+ tuple_xy = _get_terminal_size_tput()
+ # needed for window's python in cygwin's xterm!
+ if current_os in ['Linux', 'Darwin'] or current_os.startswith('CYGWIN'):
+ tuple_xy = _get_terminal_size_linux()
+ if tuple_xy is None:
+ #print "default"
+ #_get_terminal_size_windows() or _get_terminal_size_tput don't work
+ tuple_xy = (80, 25) # default value
+ return tuple_xy
+
+def _get_terminal_size_windows():
+ try:
+ from ctypes import windll, create_string_buffer
+ # stdin handle is -10
+ # stdout handle is -11
+ # stderr handle is -12
+ h = windll.kernel32.GetStdHandle(-12)
+ csbi = create_string_buffer(22)
+ res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
+ if res:
+ (bufx, bufy, curx, cury, wattr,
+ left, top, right, bottom,
+ maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
+ sizex = right - left + 1
+ sizey = bottom - top + 1
+ return sizex, sizey
+ except:
+ pass
+
+def _get_terminal_size_tput():
+ # get terminal width
+ # src: http://stackoverflow.com/questions/263890/how-do-i-find-the-width-height-of-a-terminal-window
+ try:
+ cols = int(subprocess.check_call(shlex.split('tput cols')))
+ rows = int(subprocess.check_call(shlex.split('tput lines')))
+ return (cols, rows)
+ except:
+ pass
+
+def _get_terminal_size_linux():
+ def ioctl_GWINSZ(fd):
+ try:
+ import fcntl
+ import termios
+ cr = struct.unpack('hh',
+ fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
+ return cr
+ except:
+ pass
+ cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
+ if not cr:
+ try:
+ fd = os.open(os.ctermid(), os.O_RDONLY)
+ cr = ioctl_GWINSZ(fd)
+ os.close(fd)
+ except:
+ pass
+ if not cr:
+ try:
+ cr = (os.environ['LINES'], os.environ['COLUMNS'])
+ except:
+ return None
+ return int(cr[1]), int(cr[0])
+
+# End of getTerminalSize code
+##
+
+def _rootLsPrintSimpleLs(keyList,indent,oneColumn):
+ """Print list of strings in columns
+ - blue for directories
+ - green for trees"""
+ # This code is adaptated from the pprint_list function here :
+ # http://stackoverflow.com/questions/25026556/output-list-like-ls
+ # Thanks hawkjo !!
+ if len(keyList) == 0: return
+ (term_width, term_height) = getTerminalSize()
+ term_width = term_width - indent
+ min_chars_between = 2
+ min_element_width = min( len(key.GetName()) for key in keyList ) \
+ + min_chars_between
+ max_element_width = max( len(key.GetName()) for key in keyList ) \
+ + min_chars_between
+ if max_element_width >= term_width: ncol,col_widths = 1,[1]
+ else:
+ # Start with max possible number of columns and reduce until it fits
+ ncol = 1 if oneColumn else min( len(keyList), term_width / min_element_width )
+ while True:
+ col_widths = \
+ [ max( len(key.GetName()) + min_chars_between \
+ for j, key in enumerate(keyList) if j % ncol == i ) \
+ for i in range(ncol) ]
+ if sum( col_widths ) <= term_width: break
+ else: ncol -= 1
+
+ for i, key in enumerate(keyList):
+ if i%ncol == 0: write("",indent) # indentation
+ # Don't add spaces after the last element of the line or of the list
+ if (i+1)%ncol != 0 and i != len(keyList)-1:
+ if not IS_TERMINAL: write( \
+ key.GetName().ljust(col_widths[i%ncol]))
+ elif isDirectoryKey(keyList[i]): write( \
+ isSpecial(ANSI_BLUE,key.GetName()).ljust( \
+ col_widths[i%ncol] + ANSI_BLUE_LENGTH))
+ elif isTreeKey(keyList[i]): write( \
+ isSpecial(ANSI_GREEN,key.GetName()).ljust( \
+ col_widths[i%ncol] + ANSI_GREEN_LENGTH))
+ else: write(key.GetName().ljust(col_widths[i%ncol]))
+ else: # No spaces after the last element of the line or of the list
+ if not IS_TERMINAL: write(key.GetName())
+ elif isDirectoryKey(keyList[i]):
+ write(isSpecial(ANSI_BLUE, key.GetName()))
+ elif isTreeKey(keyList[i]):
+ write(isSpecial(ANSI_GREEN, key.GetName()))
+ else: write(key.GetName())
+ write('\n')
+
+def _rootLsPrint(keyList, indent, oneColumn, \
+ longListing, treeListing):
+ """Print informations given by keyList with a rootLs
+ style choosen with the options"""
+ if longListing or treeListing: \
+ _rootLsPrintLongLs(keyList, indent, treeListing)
+ else:
+ _rootLsPrintSimpleLs(keyList, indent, oneColumn)
+
+def _rootLsProcessFile(fileName, pathSplitList, manySources, indent, \
+ oneColumn, longListing, treeListing):
+ retcode = 0
+ rootFile = openROOTFile(fileName)
+ if not rootFile: return 1
+
+ keyList,dirList = keyClassSpliter(rootFile,pathSplitList)
+ if manySources: write("{0} :".format(fileName)+"\n")
+ _rootLsPrint(keyList, indent, oneColumn, longListing, treeListing)
+
+ # Loop on the directories
+ manyPathSplits = len(pathSplitList) > 1
+ indentDir = 2 if manyPathSplits else 0
+ for pathSplit in dirList:
+ keyList = getKeyList(rootFile,pathSplit)
+ keyListSort(keyList)
+ if manyPathSplits: write("{0} :".format("/".join(pathSplit)),indent,end="\n")
+ _rootLsPrint(keyList, indent+indentDir, oneColumn, longListing, treeListing)
+
+ rootFile.Close()
+ return retcode
+
+def rootLs(sourceList, oneColumn=False, longListing=False, treeListing=False):
+ # Check arguments
+ if sourceList == []: return 1
+ tupleListSort(sourceList)
+
+ # Loop on the ROOT files
+ retcode = 0
+ manySources = len(sourceList) > 1
+ indent = 2 if manySources else 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _rootLsProcessFile(fileName, pathSplitList, manySources, indent, \
+ oneColumn, longListing, treeListing)
+ return retcode
+
+# End of ROOTLS
+##########
+
+##########
+# ROOTMKDIR
+
+MKDIR_ERROR = "cannot create directory '{0}'"
+
+def _createDirectories(rootFile,pathSplit,parents):
+ """Same behaviour as createDirectory but allows the possibility
+ to build an whole path recursively with the option \"parents\" """
+ retcode = 0
+ lenPathSplit = len(pathSplit)
+ if lenPathSplit == 0:
+ pass
+ elif parents:
+ for i in xrange(lenPathSplit):
+ currentPathSplit = pathSplit[:i+1]
+ if not (isExisting(rootFile,currentPathSplit) \
+ and isDirectory(rootFile,currentPathSplit)):
+ retcode += createDirectory(rootFile,currentPathSplit)
+ else:
+ doMkdir = True
+ for i in xrange(lenPathSplit-1):
+ currentPathSplit = pathSplit[:i+1]
+ if not (isExisting(rootFile,currentPathSplit) \
+ and isDirectory(rootFile,currentPathSplit)):
+ doMkdir = False
+ break
+ if doMkdir:
+ retcode += createDirectory(rootFile,pathSplit)
+ else:
+ logging.warning(MKDIR_ERROR.format("/".join(pathSplit)))
+ retcode += 1
+ return retcode
+
+def _rootMkdirProcessFile(fileName, pathSplitList, parents):
+ retcode = 0
+ rootFile = openROOTFile(fileName,"update")
+ if not rootFile: return 1
+ for pathSplit in pathSplitList:
+ retcode+=_createDirectories(rootFile,pathSplit,parents)
+ rootFile.Close()
+ return retcode
+
+def rootMkdir(sourceList, parents=False):
+ # Check arguments
+ if sourceList == []: return 1
+
+ # Loop on the ROOT files
+ retcode = 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _rootMkdirProcessFile(fileName, pathSplitList, parents)
+ return retcode
+
+# End of ROOTMKDIR
+##########
+
+##########
+# ROOTMV
+
+MOVE_ERROR = "error during copy of {0}, it is not removed from {1}"
+
+def _moveObjects(fileName, pathSplitList, destFile, destPathSplit, \
+ oneFile, interactive):
+ retcode = 0
+ recursive = True
+ replace = True
+ destFileName = destFile.GetName()
+ rootFile = openROOTFile(fileName,"update") \
+ if fileName != destFileName else \
+ destFile
+ if not rootFile: return 1
+ ROOT.gROOT.GetListOfFiles().Remove(rootFile) # Fast copy necessity
+ for pathSplit in pathSplitList:
+ oneSource = oneFile and len(pathSplitList)==1
+ retcodeTemp = copyRootObject(rootFile,pathSplit, \
+ destFile,destPathSplit,oneSource,recursive,replace)
+ if not retcodeTemp:
+ retcode += deleteRootObject(rootFile, pathSplit, interactive, recursive)
+ else:
+ logging.warning(MOVE_ERROR.format("/".join(pathSplit),rootFile.GetName()))
+ retcode += retcodeTemp
+ if fileName != destFileName: rootFile.Close()
+ return retcode
+
+def rootMv(sourceList, destFileName, destPathSplit, compress=None, \
+ interactive=False, recreate=False):
+ # Check arguments
+ if sourceList == [] or destFileName == "": return 1
+ if recreate and destFileName in sourceList:
+ logging.error("cannot recreate destination file if this is also a source file")
+ return 1
+
+ # Open destination file
+ destFile = openROOTFileCompress(destFileName,compress,recreate)
+ if not destFile: return 1
+ ROOT.gROOT.GetListOfFiles().Remove(destFile) # Fast copy necessity
+
+ # Loop on the root files
+ retcode = 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _moveObjects(fileName, pathSplitList, destFile, destPathSplit, \
+ len(sourceList)==1, interactive)
+ destFile.Close()
+ return retcode
+
+# End of ROOTMV
+##########
+
+##########
+# ROOTPRINT
+
+def _keyListExtended(rootFile,pathSplitList):
+ keyList,dirList = keyClassSpliter(rootFile,pathSplitList)
+ for pathSplit in dirList: keyList.extend(getKeyList(rootFile,pathSplit))
+ keyList = [key for key in keyList if not isDirectoryKey(key)]
+ keyListSort(keyList)
+ return keyList
+
+def rootPrint(sourceList, directory = None, divide = None, draw = "", outputFormat = None, \
+ output = None, size = None, style = None, verbose = False):
+ # Check arguments
+ if sourceList == []: return 1
+ tupleListSort(sourceList)
+
+ # Option values
+ directoryOptionValue = directory
+ drawOptionValue = draw
+ formatOptionValue = outputFormat
+ outputOptionValue = output
+ sizeOptionValue = size
+ styleOptionValue = style
+
+ # Verbose option
+ if not verbose:
+ ROOT.gErrorIgnoreLevel = 9999
+
+ # Don't open windows
+ ROOT.gROOT.SetBatch()
+
+ # Style option
+ if styleOptionValue:
+ ROOT.gInterpreter.ProcessLine(".x {0}".format(styleOptionValue))
+
+ # Initialize the canvas
+ if sizeOptionValue:
+ try:
+ width,height = sizeOptionValue.split("x")
+ width = int(width)
+ height = int(height)
+ except ValueError:
+ logging.warning("canvas size is on a wrong format")
+ return 1
+ canvas = ROOT.TCanvas("canvas","canvas",width,height)
+ else:
+ canvas = ROOT.TCanvas("canvas")
+
+ # Divide the canvas
+ if divide:
+ try:
+ x,y = divide.split(".")
+ x = int(x)
+ y = int(y)
+ except ValueError:
+ logging.warning("divide is on a wrong format")
+ return 1
+ canvas.Divide(x,y)
+ caseNumber = x*y
+
+ # Take the format of the output file (format option)
+ if not formatOptionValue and outputOptionValue:
+ fileName = outputOptionValue
+ fileFormat = fileName.split(".")[-1]
+ formatOptionValue = fileFormat
+
+ # Use pdf as default format
+ if not formatOptionValue: formatOptionValue = "pdf"
+
+ # Create the output directory (directory option)
+ if directoryOptionValue:
+ if not os.path.isdir(os.path.join(os.getcwd(),directoryOptionValue)):
+ os.mkdir(directoryOptionValue)
+
+ # Make the output name, begin to print (output option)
+ if outputOptionValue:
+ if formatOptionValue in ['ps','pdf']:
+ outputFileName = outputOptionValue
+ if directoryOptionValue: outputFileName = \
+ directoryOptionValue + "/" + outputFileName
+ canvas.Print(outputFileName+"[",formatOptionValue)
+ else:
+ logging.warning("can't merge pictures, only postscript or pdf files")
+ outputOptionValue = None
+
+ # Loop on the root files
+ retcode = 0
+ objDrawnNumber = 0
+ for fileName, pathSplitList in sourceList:
+ rootFile = openROOTFile(fileName)
+ if not rootFile:
+ retcode += 1
+ continue
+
+ # Fill the key list (almost the same as in rools)
+ keyList = _keyListExtended(rootFile,pathSplitList)
+ for key in keyList:
+ if isTreeKey(key):
+ pass
+ #obj = key.ReadObj()
+ #for branch in obj.GetListOfBranches():
+ # if not outputOptionValue:
+ # outputFileName = \
+ # key.GetName() + "_" + branch.GetName() + "." +formatOptionValue
+ # if directoryOptionValue:
+ # outputFileName = os.path.join( \
+ # directoryOptionValue,outputFileName)
+ # if divide:
+ # canvas.cd(objDrawnNumber%caseNumber + 1)
+ # objDrawnNumber += 1
+ # branch.Draw(drawOptionValue)
+ # if divide:
+ # if objDrawnNumber%caseNumber == 0:
+ # outputFileName = str(objDrawnNumber//caseNumber)+"."+formatOptionValue
+ # if directoryOptionValue:
+ # outputFileName = os.path.join( \
+ # directoryOptionValue,outputFileName)
+ # canvas.Print(outputFileName,formatOptionValue)
+ # canvas.Clear()
+ # canvas.Divide(x,y)
+ # elif (outputOptionValue or formatOptionValue == 'pdf') and not divide:
+ # objTitle = "Title:"+branch.GetName()+" : "+branch.GetTitle()
+ # canvas.Print(outputFileName,objTitle)
+ # else:
+ # canvas.Print(outputFileName,formatOptionValue)
+ else:
+ if not outputOptionValue:
+ outputFileName = key.GetName() + "." +formatOptionValue
+ if directoryOptionValue:
+ outputFileName = os.path.join( \
+ directoryOptionValue,outputFileName)
+ obj = key.ReadObj()
+ if divide:
+ canvas.cd(objDrawnNumber%caseNumber + 1)
+ objDrawnNumber += 1
+ obj.Draw(drawOptionValue)
+ if divide:
+ if objDrawnNumber%caseNumber == 0:
+ outputFileName = str(objDrawnNumber//caseNumber)+"."+formatOptionValue
+ if directoryOptionValue:
+ outputFileName = os.path.join( \
+ directoryOptionValue,outputFileName)
+ canvas.Print(outputFileName,formatOptionValue)
+ canvas.Clear()
+ canvas.Divide(x,y)
+ elif outputOptionValue or formatOptionValue == 'pdf':
+ objTitle = "Title:"+key.GetClassName()+" : "+key.GetTitle()
+ canvas.Print(outputFileName,objTitle)
+ else:
+ canvas.Print(outputFileName,formatOptionValue)
+ rootFile.Close()
+
+ ## End to print (divide option)
+ #if divide:
+ # if objDrawnNumber%caseNumber != 0:
+ # outputFileName = str(objDrawnNumber//caseNumber + 1)+"."+formatOptionValue
+ # if directoryOptionValue:
+ # outputFileName = os.path.join(directoryOptionValue,outputFileName)
+ # canvas.Print(outputFileName,formatOptionValue)
+
+ # End to print (output option)
+ if outputOptionValue:
+ canvas.Print(outputFileName+"]",objTitle)
+
+ return retcode
+
+# End of ROOTPRINT
+##########
+
+##########
+# ROOTRM
+
+def _removeObjects(fileName, pathSplitList, interactive=False, recursive=False):
+ retcode = 0
+ rootFile = openROOTFile(fileName,"update")
+ if not rootFile: return 1
+ for pathSplit in pathSplitList:
+ retcode += deleteRootObject(rootFile, pathSplit, interactive, recursive)
+ rootFile.Close()
+ return retcode
+
+def rootRm(sourceList, interactive=False, recursive=False):
+ # Check arguments
+ if sourceList == []: return 1
+
+ # Loop on the root files
+ retcode = 0
+ for fileName, pathSplitList in sourceList:
+ retcode += _removeObjects(fileName, pathSplitList, interactive, recursive)
+ return retcode
+
+# End of ROOTRM
+##########
diff --git a/main/python/rootbrowse.py b/main/python/rootbrowse.py
index 6f04515cb373a0a7c2a28619bec2ab85db89449f..1db2e7c458996fa8e2c11c31c224a2e7afb90a69 100755
--- a/main/python/rootbrowse.py
+++ b/main/python/rootbrowse.py
@@ -3,12 +3,12 @@
# ROOT command line tools: rootbrowse
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to open a ROOT file on a TBrowser"""
-import sys
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Open a ROOT file in a TBrowser"
@@ -21,11 +21,6 @@ EPILOG = """Examples:
Open the ROOT file 'file.root' in a TBrowser
"""
-def openBrowser(rootFile=None):
- browser = cmdLineUtils.ROOT.TBrowser()
- if rootFile: rootFile.Browse(browser)
- cmdLineUtils.ROOT.PyROOT.TPyROOTApplication.Run(cmdLineUtils.ROOT.gApplication)
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserSingleFile(COMMAND_HELP, EPILOG)
@@ -33,14 +28,7 @@ def execute():
# Put arguments in shape
args = cmdLineUtils.getArgs(parser)
- if args.FILE:
- rootFile = cmdLineUtils.openROOTFile(args.FILE)
- if not rootFile:
- return 1
- openBrowser(rootFile)
- rootFile.Close()
- else :
- openBrowser()
- return 0
+ # Process rootBrowse
+ return cmdLineUtils.rootBrowse(args.FILE)
sys.exit(execute())
diff --git a/main/python/rootcp.py b/main/python/rootcp.py
index 3a45d539875bce20b3cb5a13519b0ae303be46a2..e34ab38a85ac606d21eafbb6e43a02b1b6182e66 100755
--- a/main/python/rootcp.py
+++ b/main/python/rootcp.py
@@ -3,15 +3,12 @@
# ROOT command line tools: rootcp
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to copy objects from ROOT files into an other"""
-import sys
-import os
-import logging
-import ROOT
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Copy objects from ROOT files into an other"
@@ -30,21 +27,6 @@ EPILOG = """Examples:
Change the compression factor of the destination file 'dest.root' if not existing and copy the histogram named 'hist' from 'source.root' into it.
"""
-def copyObjects(fileName, pathSplitList, destFile, destPathSplit, optDict, oneFile):
- retcode = 0
- destFileName = destFile.GetName()
- rootFile = cmdLineUtils.openROOTFile(fileName) \
- if fileName != destFileName else \
- destFile
- if not rootFile: return 1
- ROOT.gROOT.GetListOfFiles().Remove(rootFile) # Fast copy necessity
- for pathSplit in pathSplitList:
- oneSource = oneFile and len(pathSplitList)==1
- retcode += cmdLineUtils.copyRootObject(rootFile,pathSplit, \
- destFile,destPathSplit,optDict,oneSource)
- if fileName != destFileName: rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserSourceDest(COMMAND_HELP, EPILOG)
@@ -55,22 +37,10 @@ def execute():
# Put arguments in shape
sourceList, destFileName, destPathSplit, optDict = cmdLineUtils.getSourceDestListOptDict(parser)
- if sourceList == [] or destFileName == "": return 1
- if optDict["recreate"] and destFileName in [n[0] for n in sourceList]:
- logging.error("cannot recreate destination file if this is also a source file")
- return 1
-
- # Open destination file
- destFile = cmdLineUtils.openROOTFileCompress(destFileName,optDict)
- if not destFile: return 1
- ROOT.gROOT.GetListOfFiles().Remove(destFile) # Fast copy necessity
- # Loop on the root files
- retcode = 0
- for fileName, pathSplitList in sourceList:
- retcode += copyObjects(fileName, pathSplitList, destFile, \
- destPathSplit, optDict, len(sourceList)==1)
- destFile.Close()
- return retcode
+ # Process rootCp
+ return cmdLineUtils.rootCp(sourceList, destFileName, destPathSplit, \
+ compress=optDict["compress"], recreate=optDict["recreate"], \
+ recursive=optDict["recursive"], replace=optDict["replace"])
sys.exit(execute())
diff --git a/main/python/rooteventselector.py b/main/python/rooteventselector.py
index 55b238411e5d3b9b8bdef5af209a17dc8430099c..e89a79c5b4666c75cf7bca77a573e8e3e1a82819 100755
--- a/main/python/rooteventselector.py
+++ b/main/python/rooteventselector.py
@@ -3,13 +3,12 @@
# ROOT command line tools: rooteventselector
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
-"""Command line to copy subsets of trees from source
-ROOT files to new trees on a destination ROOT file"""
+"""Command line to copy subsets of trees from source ROOT files to new trees on a destination ROOT file"""
-import sys
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Copy subsets of trees from source ROOT files"
@@ -34,63 +33,20 @@ EPILOG="""Examples:
Change the compression factor of the destination file 'dest.root' and copy the tree 'tree' from 'source.root' to 'dest.root'. For more information about compression settings of ROOT file, please look at the reference guide available on the ROOT site.
"""
-def copyTreeSubset(sourceFile,sourcePathSplit,destFile,destPathSplit,firstEvent,lastEvent):
- """Copy a subset of the tree from (sourceFile,sourcePathSplit)
- to (destFile,destPathSplit) according to options in optDict"""
- retcode = cmdLineUtils.changeDirectory(sourceFile,sourcePathSplit[:-1])
- if retcode != 0: return retcode
- bigTree = cmdLineUtils.getFromDirectory(sourcePathSplit[-1])
- nbrEntries = bigTree.GetEntries()
- # changeDirectory for the small tree not to be memory-resident
- retcode = cmdLineUtils.changeDirectory(destFile,destPathSplit)
- if retcode != 0: return retcode
- smallTree = bigTree.CloneTree(0)
- if lastEvent == -1:
- lastEvent = nbrEntries-1
- for i in xrange(firstEvent, lastEvent+1):
- bigTree.GetEntry(i)
- smallTree.Fill()
- smallTree.Write()
- return retcode
-
-def copyTreeSubsets(fileName, pathSplitList, destFile, destPathSplit, optDict):
- retcode = 0
- destFileName = destFile.GetName()
- rootFile = cmdLineUtils.openROOTFile(fileName) \
- if fileName != destFileName else \
- destFile
- if not rootFile: return 1
- for pathSplit in pathSplitList:
- if cmdLineUtils.isTree(rootFile,pathSplit):
- retcode += copyTreeSubset(rootFile,pathSplit, \
- destFile,destPathSplit,optDict["first"],optDict["last"])
- if fileName != destFileName: rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserSourceDest(COMMAND_HELP, EPILOG)
parser.add_argument("-c","--compress", type=int, help=cmdLineUtils.COMPRESS_HELP)
parser.add_argument("--recreate", help=cmdLineUtils.RECREATE_HELP, action="store_true")
- parser.add_argument("-f","--first", type=int, default=0,help=FIRST_EVENT_HELP)
+ parser.add_argument("-f","--first", type=int, default=0, help=FIRST_EVENT_HELP)
parser.add_argument("-l","--last", type=int, default=-1, help=LAST_EVENT_HELP)
# Put arguments in shape
sourceList, destFileName, destPathSplit, optDict = cmdLineUtils.getSourceDestListOptDict(parser)
- if sourceList == [] or destFileName == "": return 1
- if optDict["recreate"] and destFileName in sourceList:
- logging.error("cannot recreate destination file if this is also a source file")
- return 1
-
- # Open destination file
- destFile = cmdLineUtils.openROOTFileCompress(destFileName,optDict)
- if not destFile: return 1
- # Loop on the root file
- retcode = 0
- for fileName, pathSplitList in sourceList:
- retcode += copyTreeSubsets(fileName, pathSplitList, destFile, destPathSplit, optDict)
- destFile.Close()
- return retcode
+ # Process rootEventselector
+ return cmdLineUtils.rootEventselector(sourceList, destFileName, destPathSplit, \
+ compress=optDict["compress"], recreate=optDict["recreate"], \
+ first=optDict["first"], last=optDict["last"])
sys.exit(execute())
diff --git a/main/python/rootls.py b/main/python/rootls.py
index a553cc80008d2484a4e15c2dd9a9f1adbeee3e19..5454d8a8c86445d388402c1684a2cac7b8742e09 100755
--- a/main/python/rootls.py
+++ b/main/python/rootls.py
@@ -3,13 +3,12 @@
# ROOT command line tools: rootls
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to dump ROOT files contents to terminal"""
-import sys
-import ROOT
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = """Display ROOT files contents in the terminal."""
@@ -19,309 +18,43 @@ LONG_PRINT_HELP = "use a long listing format."
TREE_PRINT_HELP = "print tree recursively and use a long listing format."
EPILOG = """Examples:
-- rools example.root
+- rootls example.root
Display contents of the ROOT file 'example.root'.
-- rools example.root:dir
+- rootls example.root:dir
Display contents of the directory 'dir' from the ROOT file 'example.root'.
-- rools example.root:*
+- rootls example.root:*
Display contents of the ROOT file 'example.root' and his subdirectories.
-- rools file1.root file2.root
+- rootls file1.root file2.root
Display contents of ROOT files 'file1.root' and 'file2.root'.
-- rools *.root
+- rootls *.root
Display contents of ROOT files whose name ends with '.root'.
-- rools -1 example.root
+- rootls -1 example.root
Display contents of the ROOT file 'example.root' in one column.
-- rools -l example.root
+- rootls -l example.root
Display contents of the ROOT file 'example.root' and use a long listing format.
-- rools -t example.root
+- rootls -t example.root
Display contents of the ROOT file 'example.root', use a long listing format and print trees recursively.
"""
-# Ansi characters
-ANSI_BOLD = "\x1B[1m"
-ANSI_BLUE = "\x1B[34m"
-ANSI_GREEN = "\x1B[32m"
-ANSI_END = "\x1B[0m"
-
-# Needed for column width calculation
-ANSI_BOLD_LENGTH = len(ANSI_BOLD+ANSI_END)
-ANSI_BLUE_LENGTH = len(ANSI_BLUE+ANSI_END)
-ANSI_GREEN_LENGTH = len(ANSI_GREEN+ANSI_END)
-
-# Terminal and platform booleans
-IS_TERMINAL = sys.stdout.isatty()
-IS_WIN32 = sys.platform == 'win32'
-
-def isSpecial(ansiCode,string):
- """Use ansi code on 'string' if the output is the
- terminal of a not Windows platform"""
- if IS_TERMINAL and not IS_WIN32: return ansiCode+string+ANSI_END
- else: return string
-
-def write(string,indent=0,end=""):
- """Use sys.stdout.write to write the string with an indentation
- equal to indent and specifying the end character"""
- sys.stdout.write(" "*indent+string+end)
-
-TREE_TEMPLATE = "{0:{nameWidth}}"+"{1:{titleWidth}}{2:{memoryWidth}}"
-
-def recursifTreePrinter(tree,indent):
- """Print recursively tree informations"""
- listOfBranches = tree.GetListOfBranches()
- if len(listOfBranches) > 0: # Width informations
- maxCharName = max([len(branch.GetName()) \
- for branch in listOfBranches])
- maxCharTitle = max([len(branch.GetTitle()) \
- for branch in listOfBranches])
- dic = { \
- "nameWidth":maxCharName+2, \
- "titleWidth":maxCharTitle+4, \
- "memoryWidth":1}
- for branch in listOfBranches: # Print loop
- rec = \
- [branch.GetName(), \
- "\""+branch.GetTitle()+"\"", \
- str(branch.GetTotBytes())]
- write(TREE_TEMPLATE.format(*rec,**dic),indent,end="\n")
- recursifTreePrinter(branch,indent+2)
-
-def prepareTime(time):
- """Get time in the proper shape
- ex : 174512 for 17h 45m 12s
- ex : 094023 for 09h 40m 23s"""
- time = str(time)
- time = '000000'+time
- time = time[len(time)-6:]
- return time
-
-MONTH = {1:'Jan',2:'Feb',3:'Mar',4:'Apr',5:'May',6:'Jun', \
- 7:'Jul',8:'Aug',9:'Sep',10:'Oct',11:'Nov',12:'Dec'}
-LONG_TEMPLATE = \
- isSpecial(ANSI_BOLD,"{0:{classWidth}}")+"{1:{timeWidth}}" + \
- "{2:{nameWidth}}{3:{titleWidth}}"
-
-def roolsPrintLongLs(keyList,optDict,indent):
- """Print a list of Tkey in columns
- pattern : classname, datetime, name and title"""
- if len(keyList) > 0: # Width informations
- maxCharClass = max([len(key.GetClassName()) for key in keyList])
- maxCharTime = 12
- maxCharName = max([len(key.GetName()) for key in keyList])
- dic = { \
- "classWidth":maxCharClass+2, \
- "timeWidth":maxCharTime+2, \
- "nameWidth":maxCharName+2, \
- "titleWidth":1}
- date = ROOT.Long(0)
- for key in keyList:
- time = ROOT.Long(0)
- datime = key.GetDatime()
- datime.GetDateTime(datime.Get(),date,time)
- time = prepareTime(time)
- rec = \
- [key.GetClassName(), \
- MONTH[int(str(date)[4:6])]+" " +str(date)[6:]+ \
- " "+time[:2]+":"+time[2:4], \
- key.GetName(), \
- "\""+key.GetTitle()+"\""]
- write(LONG_TEMPLATE.format(*rec,**dic),indent,end="\n")
- if optDict['tree'] and cmdLineUtils.isTreeKey(key):
- tree = key.ReadObj()
- recursifTreePrinter(tree,indent+2)
-
-##
-# The code of the getTerminalSize function can be found here :
-# https://gist.github.com/jtriley/1108174
-# Thanks jtriley !!
-
-import os
-import shlex
-import struct
-import platform
-import subprocess
-
-def getTerminalSize():
- """ getTerminalSize()
- - get width and height of console
- - works on linux,os x,windows,cygwin(windows)
- originally retrieved from:
- http://stackoverflow.com/questions/566746/how-to-get-console-window-width-in-python"""
- current_os = platform.system()
- tuple_xy = None
- if current_os == 'Windows':
- tuple_xy = _get_terminal_size_windows()
- if tuple_xy is None:
- tuple_xy = _get_terminal_size_tput()
- # needed for window's python in cygwin's xterm!
- if current_os in ['Linux', 'Darwin'] or current_os.startswith('CYGWIN'):
- tuple_xy = _get_terminal_size_linux()
- if tuple_xy is None:
- #print "default"
- #_get_terminal_size_windows() or _get_terminal_size_tput don't work
- tuple_xy = (80, 25) # default value
- return tuple_xy
-
-def _get_terminal_size_windows():
- try:
- from ctypes import windll, create_string_buffer
- # stdin handle is -10
- # stdout handle is -11
- # stderr handle is -12
- h = windll.kernel32.GetStdHandle(-12)
- csbi = create_string_buffer(22)
- res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
- if res:
- (bufx, bufy, curx, cury, wattr,
- left, top, right, bottom,
- maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
- sizex = right - left + 1
- sizey = bottom - top + 1
- return sizex, sizey
- except:
- pass
-
-def _get_terminal_size_tput():
- # get terminal width
- # src: http://stackoverflow.com/questions/263890/how-do-i-find-the-width-height-of-a-terminal-window
- try:
- cols = int(subprocess.check_call(shlex.split('tput cols')))
- rows = int(subprocess.check_call(shlex.split('tput lines')))
- return (cols, rows)
- except:
- pass
-
-def _get_terminal_size_linux():
- def ioctl_GWINSZ(fd):
- try:
- import fcntl
- import termios
- cr = struct.unpack('hh',
- fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
- return cr
- except:
- pass
- cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
- if not cr:
- try:
- fd = os.open(os.ctermid(), os.O_RDONLY)
- cr = ioctl_GWINSZ(fd)
- os.close(fd)
- except:
- pass
- if not cr:
- try:
- cr = (os.environ['LINES'], os.environ['COLUMNS'])
- except:
- return None
- return int(cr[1]), int(cr[0])
-
-# End of getTerminalSize code
-##
-
-def roolsPrintSimpleLs(keyList,indent,oneColumn):
- """Print list of strings in columns
- - blue for directories
- - green for trees"""
- # This code is adaptated from the pprint_list function here :
- # http://stackoverflow.com/questions/25026556/output-list-like-ls
- # Thanks hawkjo !!
- if len(keyList) == 0: return
- (term_width, term_height) = getTerminalSize()
- term_width = term_width - indent
- min_chars_between = 2
- min_element_width = min( len(key.GetName()) for key in keyList ) \
- + min_chars_between
- max_element_width = max( len(key.GetName()) for key in keyList ) \
- + min_chars_between
- if max_element_width >= term_width: ncol,col_widths = 1,[1]
- else:
- # Start with max possible number of columns and reduce until it fits
- ncol = 1 if oneColumn else min( len(keyList), term_width / min_element_width )
- while True:
- col_widths = \
- [ max( len(key.GetName()) + min_chars_between \
- for j, key in enumerate(keyList) if j % ncol == i ) \
- for i in range(ncol) ]
- if sum( col_widths ) <= term_width: break
- else: ncol -= 1
-
- for i, key in enumerate(keyList):
- if i%ncol == 0: write("",indent) # indentation
- # Don't add spaces after the last element of the line or of the list
- if (i+1)%ncol != 0 and i != len(keyList)-1:
- if not IS_TERMINAL: write( \
- key.GetName().ljust(col_widths[i%ncol]))
- elif cmdLineUtils.isDirectoryKey(keyList[i]): write( \
- isSpecial(ANSI_BLUE,key.GetName()).ljust( \
- col_widths[i%ncol] + ANSI_BLUE_LENGTH))
- elif cmdLineUtils.isTreeKey(keyList[i]): write( \
- isSpecial(ANSI_GREEN,key.GetName()).ljust( \
- col_widths[i%ncol] + ANSI_GREEN_LENGTH))
- else: write(key.GetName().ljust(col_widths[i%ncol]))
- else: # No spaces after the last element of the line or of the list
- if not IS_TERMINAL: write(key.GetName())
- elif cmdLineUtils.isDirectoryKey(keyList[i]):
- write(isSpecial(ANSI_BLUE, key.GetName()))
- elif cmdLineUtils.isTreeKey(keyList[i]):
- write(isSpecial(ANSI_GREEN, key.GetName()))
- else: write(key.GetName())
- write('\n')
-
-def roolsPrint(keyList,optDict,indent=0):
- """Print informations given by keyList with a rools
- style choosen with optDict"""
- if optDict['long'] or optDict['tree']: \
- roolsPrintLongLs(keyList,optDict,indent)
- else:
- oneColumn = True if optDict['one'] else False
- roolsPrintSimpleLs(keyList,indent, oneColumn)
-
-def processFile(fileName, pathSplitList, optDict, manySources, indent):
- retcode = 0
- rootFile = cmdLineUtils.openROOTFile(fileName)
- if not rootFile: return 1
-
- keyList,dirList = cmdLineUtils.keyClassSpliter(rootFile,pathSplitList)
- if manySources: write("{0} :".format(fileName)+"\n")
- roolsPrint(keyList,optDict,indent)
-
- # Loop on the directories
- manyPathSplits = len(pathSplitList) > 1
- indentDir = 2 if manyPathSplits else 0
- for pathSplit in dirList:
- keyList = cmdLineUtils.getKeyList(rootFile,pathSplit)
- cmdLineUtils.keyListSort(keyList)
- if manyPathSplits: write("{0} :".format("/".join(pathSplit)),indent,end="\n")
- roolsPrint(keyList,optDict,indent+indentDir)
-
- rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserFile(COMMAND_HELP, EPILOG)
- parser.add_argument("-1", "--one", help=ONE_HELP, action="store_true")
- parser.add_argument("-l", "--long", help=LONG_PRINT_HELP, action="store_true")
- parser.add_argument("-t", "--tree", help=TREE_PRINT_HELP, action="store_true")
+ parser.add_argument("-1", "--oneColumn", help=ONE_HELP, action="store_true")
+ parser.add_argument("-l", "--longListing", help=LONG_PRINT_HELP, action="store_true")
+ parser.add_argument("-t", "--treeListing", help=TREE_PRINT_HELP, action="store_true")
# Put arguments in shape
sourceList, optDict = cmdLineUtils.getSourceListOptDict(parser)
- if sourceList == []: return 1
- cmdLineUtils.tupleListSort(sourceList)
- # Loop on the ROOT files
- retcode = 0
- manySources = len(sourceList) > 1
- indent = 2 if manySources else 0
- for fileName, pathSplitList in sourceList:
- retcode += processFile(fileName, pathSplitList, optDict, manySources, indent)
- return retcode
+ # Process rootLs
+ return cmdLineUtils.rootLs(sourceList, oneColumn=optDict["oneColumn"], \
+ longListing=optDict["longListing"], treeListing=optDict["treeListing"])
sys.exit(execute())
diff --git a/main/python/rootmkdir.py b/main/python/rootmkdir.py
index f67d68480472e16f03f84a84142a91924cabb2ec..1e3fb25f7c5bc067b7a248bcf0173b7b95e06f32 100755
--- a/main/python/rootmkdir.py
+++ b/main/python/rootmkdir.py
@@ -3,13 +3,12 @@
# ROOT command line tools: rootmkdir
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to add directories in ROOT files"""
-import sys
-import logging
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Add directories in ROOT files"
@@ -30,45 +29,6 @@ EPILOG="""Examples:
Create an empty ROOT file named 'example.root'
"""
-MKDIR_ERROR = "cannot create directory '{0}'"
-
-def createDirectories(rootFile,pathSplit,optDict):
- """Same behaviour as createDirectory but allows the possibility
- to build an whole path recursively with opt_dict["parents"]"""
- retcode = 0
- lenPathSplit = len(pathSplit)
- if lenPathSplit == 0:
- pass
- elif optDict["parents"]:
- for i in xrange(lenPathSplit):
- currentPathSplit = pathSplit[:i+1]
- if not (cmdLineUtils.isExisting(rootFile,currentPathSplit) \
- and cmdLineUtils.isDirectory(rootFile,currentPathSplit)):
- retcode += cmdLineUtils.createDirectory(rootFile,currentPathSplit)
- else:
- doMkdir = True
- for i in xrange(lenPathSplit-1):
- currentPathSplit = pathSplit[:i+1]
- if not (cmdLineUtils.isExisting(rootFile,currentPathSplit) \
- and cmdLineUtils.isDirectory(rootFile,currentPathSplit)):
- doMkdir = False
- break
- if doMkdir:
- retcode += cmdLineUtils.createDirectory(rootFile,pathSplit)
- else:
- logging.warning(MKDIR_ERROR.format("/".join(pathSplit)))
- retcode += 1
- return retcode
-
-def processFile(fileName, pathSplitList, optDict):
- retcode = 0
- rootFile = cmdLineUtils.openROOTFile(fileName,"update")
- if not rootFile: return 1
- for pathSplit in pathSplitList:
- retcode+=createDirectories(rootFile,pathSplit,optDict)
- rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserFile(COMMAND_HELP, EPILOG)
@@ -76,12 +36,8 @@ def execute():
# Put arguments in shape
sourceList, optDict = cmdLineUtils.getSourceListOptDict(parser, wildcards = False)
- if sourceList == []: return 1
- # Loop on the ROOT files
- retcode = 0
- for fileName, pathSplitList in sourceList:
- retcode += processFile(fileName, pathSplitList,optDict)
- return retcode
+ # Process rootMkdir
+ return cmdLineUtils.rootMkdir(sourceList, parents=optDict["parents"])
sys.exit(execute())
diff --git a/main/python/rootmv.py b/main/python/rootmv.py
index 907b5072ad18381cf110e97e02f67898cd07f4a7..35d8a088ca9c8561febc3125cd750ca151120336 100755
--- a/main/python/rootmv.py
+++ b/main/python/rootmv.py
@@ -3,15 +3,12 @@
# ROOT command line tools: rootmv
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to move objects from ROOT files to another"""
-import sys
-import os
-import logging
-import ROOT
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Move objects from ROOT files to another"
@@ -30,28 +27,6 @@ EPILOG = """Examples:
Change the compression level of the destination file 'dest.root' and move the histogram named 'hist' from 'source.root' into it. For more information about compression settings of ROOT file, please look at the reference guide available on the ROOT site.
"""
-MOVE_ERROR = "error during copy of {0}, it is not removed from {1}"
-
-def moveObjects(fileName, pathSplitList, destFile, destPathSplit, optDict, oneFile):
- retcode = 0
- destFileName = destFile.GetName()
- rootFile = cmdLineUtils.openROOTFile(fileName,"update") \
- if fileName != destFileName else \
- destFile
- if not rootFile: return 1
- ROOT.gROOT.GetListOfFiles().Remove(rootFile) # Fast copy necessity
- for pathSplit in pathSplitList:
- oneSource = oneFile and len(pathSplitList)==1
- retcodeTemp = cmdLineUtils.copyRootObject(rootFile,pathSplit, \
- destFile,destPathSplit,optDict,oneSource)
- if not retcodeTemp:
- retcode += cmdLineUtils.deleteRootObject(rootFile,pathSplit,optDict)
- else:
- logging.warning(MOVE_ERROR.format("/".join(pathSplit),rootFile.GetName()))
- retcode += retcodeTemp
- if fileName != destFileName: rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserSourceDest(COMMAND_HELP, EPILOG)
@@ -61,24 +36,10 @@ def execute():
# Put arguments in shape
sourceList, destFileName, destPathSplit, optDict = cmdLineUtils.getSourceDestListOptDict(parser)
- if sourceList == [] or destFileName == "": return 1
- if optDict["recreate"] and destFileName in sourceList:
- logging.error("cannot recreate destination file if this is also a source file")
- return 1
- optDict["recursive"] = True
- optDict["replace"] = True
-
- # Open destination file
- destFile = cmdLineUtils.openROOTFileCompress(destFileName,optDict)
- if not destFile: return 1
- ROOT.gROOT.GetListOfFiles().Remove(destFile) # Fast copy necessity
- # Loop on the root files
- retcode = 0
- for fileName, pathSplitList in sourceList:
- retcode += moveObjects(fileName, pathSplitList, destFile, \
- destPathSplit, optDict, len(sourceList)==1)
- destFile.Close()
- return retcode
+ # Process rootMv
+ return cmdLineUtils.rootMv(sourceList, destFileName, destPathSplit, \
+ compress=optDict["compress"], interactive=optDict["interactive"], \
+ recreate=optDict["recreate"])
sys.exit(execute())
diff --git a/main/python/rootprint.py b/main/python/rootprint.py
index 27385470f6969c8b56ffa577e669bd8cd5e107ce..cb418cc1e2bf48450c13d0a427d652d6b6e0ff3d 100755
--- a/main/python/rootprint.py
+++ b/main/python/rootprint.py
@@ -1,26 +1,25 @@
-#!/usr/bin/python
+#!/usr/bin/env python
# ROOT command line tools: rootprint
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to print ROOT files contents on ps,pdf or png,gif..."""
-import sys
-import ROOT
-import logging
-import os
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Print ROOT files contents on ps,pdf or pictures files"
DIRECTORY_HELP = "put output files in a subdirectory named DIRECTORY."
+DIVIDE_HELP = "divide the canvas ont the format 'x'.'y' (ex: 2.2)"
DRAW_HELP = "specify draw option"
FORMAT_HELP = "specify output format (ex: pdf, png)."
OUTPUT_HELP = "merge files in a file named OUTPUT (only for ps and pdf)."
SIZE_HELP = "specify canvas size on the format 'width'x'height' (ex: 600x400)"
+STYLE_HELP = "specify a C file name which define a style"
VERBOSE_HELP = "print informations about the running"
EPILOG = """Examples:
@@ -37,125 +36,26 @@ EPILOG = """Examples:
Create a pdf file named 'histograms.pdf' which contain all histograms whose name starts with 'hist'. It works also with postscript.
"""
-def keyListExtended(rootFile,pathSplitList):
- keyList,dirList = cmdLineUtils.keyClassSpliter(rootFile,pathSplitList)
- for pathSplit in dirList: keyList.extend(cmdLineUtils.getKeyList(rootFile,pathSplit))
- keyList = [key for key in keyList if not cmdLineUtils.isDirectoryKey(key)]
- cmdLineUtils.keyListSort(keyList)
- return keyList
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserFile(COMMAND_HELP, EPILOG)
parser.add_argument("-d", "--directory", help=DIRECTORY_HELP)
+ parser.add_argument("--divide", help=DIVIDE_HELP)
parser.add_argument("-D", "--draw", default="", help=DRAW_HELP)
- parser.add_argument("-f", "--format", help=FORMAT_HELP)
+ parser.add_argument("-f", "--outputFormat", help=FORMAT_HELP)
parser.add_argument("-o", "--output", help=OUTPUT_HELP)
parser.add_argument("-s", "--size", help=SIZE_HELP)
+ parser.add_argument("-S", "--style", help=STYLE_HELP)
parser.add_argument("-v", "--verbose", action="store_true", help=VERBOSE_HELP)
# Put arguments in shape
sourceList, optDict = cmdLineUtils.getSourceListOptDict(parser)
- if sourceList == []: return 1
- cmdLineUtils.tupleListSort(sourceList)
-
- # Option values
- directoryOptionValue = optDict["directory"]
- drawOptionValue = optDict["draw"]
- formatOptionValue = optDict["format"]
- outputOptionValue = optDict["output"]
- sizeOptionValue = optDict["size"]
-
- # Verbose option
- if not optDict["verbose"]:
- ROOT.gErrorIgnoreLevel = 9999
-
- # Don't open windows
- ROOT.gROOT.SetBatch()
-
- # Initialize the canvas
- if sizeOptionValue:
- try:
- width,height = sizeOptionValue.split("x")
- width = int(width)
- height = int(height)
- except ValueError:
- logging.warning("canvas size is on a wrong format")
- return 1
- canvas = ROOT.TCanvas("canvas","canvas",width,height)
- else:
- canvas = ROOT.TCanvas("canvas")
-
- # Take the format of the output file (format option)
- if not formatOptionValue and outputOptionValue:
- fileName = outputOptionValue
- fileFormat = fileName.split(".")[-1]
- formatOptionValue = fileFormat
-
- # Use pdf as default format
- if not formatOptionValue: formatOptionValue = "pdf"
-
- # Create the output directory (directory option)
- if directoryOptionValue:
- if not os.path.isdir(os.path.join(os.getcwd(),directoryOptionValue)):
- os.mkdir(directoryOptionValue)
-
- # Make the output name, begin to print (output option)
- if outputOptionValue:
- if formatOptionValue in ['ps','pdf']:
- outputFileName = outputOptionValue
- if directoryOptionValue: outputFileName = \
- directoryOptionValue + "/" + outputFileName
- canvas.Print(outputFileName+"[",formatOptionValue)
- else:
- logging.warning("can't merge pictures, only postscript or pdf files")
- outputOptionValue = None
-
- # Loop on the root files
- retcode = 0
- for fileName, pathSplitList in sourceList:
- rootFile = cmdLineUtils.openROOTFile(fileName)
- if not rootFile:
- retcode += 1
- continue
-
- # Fill the key list (almost the same as in rools)
- keyList = keyListExtended(rootFile,pathSplitList)
- for key in keyList:
- if cmdLineUtils.isTreeKey(key):
- obj = key.ReadObj()
- for branch in obj.GetListOfBranches():
- if not outputOptionValue:
- outputFileName = \
- key.GetName() + "_" + branch.GetName() + "." +formatOptionValue
- if directoryOptionValue:
- outputFileName = os.path.join( \
- directoryOptionValue,outputFileName)
- obj.Draw(drawOptionValue)
- if outputOptionValue or formatOptionValue == 'pdf':
- objTitle = "Title:"+branch.GetName()+" : "+branch.GetTitle()
- canvas.Print(outputFileName,objTitle)
- else:
- canvas.Print(outputFileName,formatOptionValue)
- else:
- if not outputOptionValue:
- outputFileName = key.GetName() + "." +formatOptionValue
- if directoryOptionValue:
- outputFileName = os.path.join( \
- directoryOptionValue,outputFileName)
- obj = key.ReadObj()
- obj.Draw(drawOptionValue)
- if outputOptionValue or formatOptionValue == 'pdf':
- objTitle = "Title:"+key.GetClassName()+" : "+key.GetTitle()
- canvas.Print(outputFileName,objTitle)
- else:
- canvas.Print(outputFileName,formatOptionValue)
- rootFile.Close()
-
- # End to print (output option)
- if outputOptionValue:
- canvas.Print(outputFileName+"]",objTitle)
- return retcode
+ # Process rootPrint
+ return cmdLineUtils.rootPrint(sourceList, directory = optDict["directory"], \
+ divide = optDict["divide"], draw = optDict["draw"], \
+ outputFormat = optDict["outputFormat"], \
+ output = optDict["output"], size = optDict["size"], \
+ style = optDict["style"], verbose = optDict["verbose"])
sys.exit(execute())
diff --git a/main/python/rootrm.py b/main/python/rootrm.py
index 0485f5fd5880a359538ab14b855f6623acf54365..ef8e6e16f3c854ebbd9a74f249e858c0e4635a2b 100755
--- a/main/python/rootrm.py
+++ b/main/python/rootrm.py
@@ -3,12 +3,12 @@
# ROOT command line tools: rootrm
# Author: Julien Ripoche
# Mail: julien.ripoche@u-psud.fr
-# Date: 13/08/15
+# Date: 20/08/15
"""Command line to remove objects from ROOT files"""
-import sys
import cmdLineUtils
+import sys
# Help strings
COMMAND_HELP = "Remove objects from ROOT files"
@@ -27,15 +27,6 @@ EPILOG = """Examples:
Display a confirmation request before deleting: 'remove 'hist' from 'example.root' ? (y/n) :'
"""
-def removeObjects(fileName, pathSplitList, optDict):
- retcode = 0
- rootFile = cmdLineUtils.openROOTFile(fileName,"update")
- if not rootFile: return 1
- for pathSplit in pathSplitList:
- retcode += cmdLineUtils.deleteRootObject(rootFile,pathSplit,optDict)
- rootFile.Close()
- return retcode
-
def execute():
# Collect arguments with the module argparse
parser = cmdLineUtils.getParserFile(COMMAND_HELP, EPILOG)
@@ -44,12 +35,9 @@ def execute():
# Put arguments in shape
sourceList, optDict = cmdLineUtils.getSourceListOptDict(parser)
- if sourceList == []: return 1
- # Loop on the root files
- retcode = 0
- for fileName, pathSplitList in sourceList:
- retcode += removeObjects(fileName, pathSplitList,optDict)
- return retcode
+ # Process rootRm
+ return cmdLineUtils.rootRm(sourceList, interactive=optDict["interactive"], \
+ recursive=optDict["recursive"])
sys.exit(execute())