root2sqlite Namespace Reference

Functions
def	columnescape (s)
def	read_objects_root (rootfile)
def	save_keyvalue (dictionary, name)
def	tosqlite (x)
def	treetotable (ttree, name)

Variables
	args
	basic_objects
	db
	default
	description
	f
	help
	inf
	nan
	parser

Function Documentation

columnescape()

def root2sqlite.columnescape

(

s

)

Definition at line 46 of file root2sqlite.py.

Referenced by treetotable().

def columnescape(s):
    # add whatever is not a valid column name here
    SQLKWDs = ["index"]
    if s.lower() in SQLKWDs:
        return s + "_"
    else:
        return s

read_objects_root()

def root2sqlite.read_objects_root

(

rootfile

)

Definition at line 70 of file root2sqlite.py.

Referenced by save_keyvalue().

def read_objects_root(rootfile):
    xml_re = re.compile(r"^<(.+)>(.+)=(.+)<\/\1>$")
    def parse_directory(di):
        directory = rootfile.GetDirectory(di)
        for key in directory.GetListOfKeys():
            entry = key.GetName()
            rtype = key.GetClassName()
            fullpath = "%s/%s" % (di, entry) if di != "" else entry
            if (rtype == "TDirectoryFile"):
                for k, v, t in parse_directory(fullpath):
                    yield (k, v, t)
            else:
                obj = rootfile.Get(fullpath)
                if obj:
                    yield (fullpath, obj, rtype)
                else:
                    # special case to parse the xml abomination
                    m = xml_re.search(entry)
                    if m:
                        name = m.group(1)
                        typecode = m.group(2)
                        value = m.group(3)
                        fp = "%s/%s" % (di, name)
                        yield (fp, value, rtype)
                    else:
                        raise Exception("Invalid xml:" + entry)
    path_fix = re.compile(r"^\/Run \d+")
    for fullname, obj, rtype in parse_directory(""):
        yield fullname, obj, rtype

save_keyvalue()

def root2sqlite.save_keyvalue	(		dictionary,
			name
	)

Definition at line 100 of file root2sqlite.py.

References print(), read_objects_root(), tosqlite(), treetotable(), and betterConfigParser.unicode.

def save_keyvalue(dictionary, name):
    name = name.replace("/", "_")
    create =  "CREATE TABLE %s(key, value);"  % name
    print(create)
    db.execute(create)
    data = []
    for k, v in dictionary.iteritems():
        vals = (unicode(k), tosqlite(v))
        data.append(vals)
    insert = "INSERT INTO %s(key, value) VALUES (?,?);" % name
    print(insert)
    db.executemany(insert, data)
    db.commit()

tosqlite()

def root2sqlite.tosqlite

(

x

)

Definition at line 23 of file root2sqlite.py.

References edmScanValgrind.buffer, and betterConfigParser.unicode.

Referenced by save_keyvalue(), and treetotable().

def tosqlite(x):
    if isinstance(x, ROOT.string):
        try:
            return unicode(x.data())
        except:
            return buffer(x.data())
    if isinstance(x, int):
          return x
    if isinstance(x, float):
        return x
    if isinstance(x, int):
        return x
    else:
        try: 
            rootobj = unicode(ROOT.TBufferJSON.ConvertToJSON(x))
            # turns out ROOT does not generate valid JSON for NaN/inf
            clean = nan.sub('\\g<1>0\\g<2>', inf.sub('\\g<1>1e38\\g<2>', rootobj))
            obj = json.loads(clean)
            jsonobj = json.dumps(obj, allow_nan=False)
            return jsonobj
        except Exception as e:
            return json.dumps({"root2sqlite_error": e.__repr__(), "root2sqlite_object": x.__repr__()})

treetotable()

def root2sqlite.treetotable	(		ttree,
			name
	)

Definition at line 54 of file root2sqlite.py.

References columnescape(), join(), print(), isotrackApplyRegressor.range, tosqlite(), and mkLumiAveragedPlots.tuple.

Referenced by save_keyvalue().

def treetotable(ttree, name):
    name = name.replace("/", "_")
    branches = [b.GetName() for b in ttree.GetListOfBranches()]
    colnames = ", ".join(columnescape(b) for b in branches)
    create =  "CREATE TABLE %s(%s);"  % (name, colnames)
    print(create)
    db.execute(create)
    data = []
    for i in range(ttree.GetEntries()):
        ttree.GetEntry(i)
        vals = tuple([tosqlite(getattr(ttree, b)) for b in branches])
        data.append(vals)
    insert = "INSERT INTO %s(%s) VALUES (%s);" % (name, colnames, ",".join(["?"] * len(branches)))
    print(insert)
    db.executemany(insert, data)

Variable Documentation

args

root2sqlite.args

Definition at line 13 of file root2sqlite.py.

basic_objects

root2sqlite.basic_objects

Definition at line 18 of file root2sqlite.py.

db

root2sqlite.db

Definition at line 16 of file root2sqlite.py.

default

root2sqlite.default

Definition at line 12 of file root2sqlite.py.

description

root2sqlite.description

Definition at line 9 of file root2sqlite.py.

f

root2sqlite.f

Definition at line 15 of file root2sqlite.py.

help

root2sqlite.help

Definition at line 11 of file root2sqlite.py.

inf

root2sqlite.inf

Definition at line 20 of file root2sqlite.py.

nan

root2sqlite.nan

Definition at line 21 of file root2sqlite.py.

parser

root2sqlite.parser

Definition at line 9 of file root2sqlite.py.