RootMap.py

Go to the documentation of this file.

#!/usr/bin/env python

import os
import sys


def _getPath(pathstring=""):
    pthlist = []
    if pathstring == "":
        if sys.platform == "linux2":
            pathstring = os.environ["LD_LIBRARY_PATH"]
        else:
            pathstring = os.environ["PATH"]
    pthlist += pathstring.split(os.pathsep)
    return pthlist


def _getEntry(line):
    ll = line.split()
    entry = ll[0]
    entry = entry.replace("Library.", "")
    entry = entry.replace(":", "")
    entry = entry.replace("@", ":")
    entry = entry.replace("-", " ")
    lib = ll[1]
    return entry, lib


def _getBlock(line):
    if line.find("Begin"):
        block = line.split()[-1]
    return block


def _procRootMap(rtmpfile, rtmapdict):
    block = ""
    for line in open(rtmpfile, 'r'):
        line = line[:-1]
        if line:
            if line[0] != '#':
                entry, lib = _getEntry(line)
                if not rtmapdict.has_key(entry):
                    rtmapdict[entry] = []
                rtmapdict[entry].append(
                    (os.path.join(os.path.dirname(rtmpfile), lib), block))
            else:
                block = _getBlock(line)


def _procSysRootMap(rtmapdict):
    if os.environ.has_key("ROOTSYS"):
        rtmpfile = os.path.join(os.environ["ROOTSYS"], "etc", "system.rootmap")
        block = ""
        for line in open(rtmpfile, 'r'):
            line = line[:-1]
            if line:
                if line[0] != '#':
                    entry, lib = _getEntry(line)
                    if not rtmapdict.has_key(entry):
                        rtmapdict[entry] = []
                    rtmapdict[entry].append(
                        (os.path.join(os.environ["ROOTSYS"], "lib", lib), block))
                else:
                    block = _getBlock(line)
    else:
        print "WARNING: No ROOTSYS defined!"


def _isRootMap(filename):
    # The file must begin with "rootmap"
    if (filename.find("rootmap") == 0):
        return True
    return False


def getMaps(pathstring="", sysrtmap=False):
    rtmapdict = dict()
    pthlist = _getPath(pathstring)
    if sysrtmap:
        _procSysRootMap(rtmapdict)
    for p in pthlist:
        try:
            for f in filter(_isRootMap, os.listdir(p)):
                rtmpfile = os.path.join(p, f)
                if (os.path.exists(rtmpfile)):
                    _procRootMap(rtmpfile, rtmapdict)
        except:
            pass
    return rtmapdict


def _getLongestEntry(maps):
    sz = 0
    for k in maps.keys():
        if len(k) > sz:
            sz = len(k)
    return sz


def printMaps(maps, recomp=None):
    linelen = _getLongestEntry(maps)
    frmat = r"%-" + str(linelen) + "s\t"
    kys = maps.keys()
    kys.sort()
    if recomp:
        kys = filter(recomp.search, kys)
    for k in kys:
        if len(maps[k]) > 1:
            print "!!!!!!!!!!!! WARNING - More than one entry !!!!!!!!!!"
        for l in maps[k]:
            print frmat % k,
            for m in l:
                print m,
            print " "
    return


def shortPrintMaps(maps, recomp=None):
    kys = maps.keys()
    kys.sort()
    if recomp:
        kys = filter(recomp.search, kys)
    for k in kys:
        if len(maps[k]) > 1:
            print k, "!!!!!!!!!!!! WARNING - More than one entry !!!!!!!!!!"
        else:
            print k
        for l in maps[k]:
            for m in l:
                print "\t%s" % m,
            print " "
    return


def printKeys(maps, recomp=None):
    kys = maps.keys()
    kys.sort()
    if recomp:
        kys = filter(recomp.search, kys)
    for k in kys:
        if len(maps[k]) > 1:
            print "!!!!!!!!!!!! WARNING - More than one entry !!!!!!!!!!"
        for l in maps[k]:
            print k
    return


def checkDict(maps, recomp=None):
    kys = maps.keys()
    kys.sort()
    if recomp:
        kys = filter(recomp.search, kys)
    for k in kys:
        if len(maps[k]) > 1:
            print "!!!!!!!!!!!! WARNING - More than one entry !!!!!!!!!!"
            print k
            for l in maps[k]:
                for m in l:
                    print "\t%s" % m,
                print " "
    return