xAODRootTest Namespace Reference

Classes
class	AllocTestRead
class	Analysis
class	xAODTestClearDecor
class	xAODTestCopy
class	xAODTestDecor
class	xAODTestPDecor
class	xAODTestRead

Functions
def	CHECK (sc)
def	_typename (t)
def	xAODInit ()
def	format_int (x)
def	format_float (x)
def	format_str (x)
def	format_el (x)
def	format_el_vec (v)
def	format_float_vec (v)
def	format_int_vec (v)
def	format_str_vec (v)
def	dump_auxitem (x, auxid, f=sys.stdout)
def	dump_auxdata (x, exclude=[], f=sys.stdout)
def	dump_c (c, f=sys.stdout)
def	dump_xaodobj (h, f=sys.stdout)
def	dump_plinks (p, f=sys.stdout)
def	copy_obj (event, obj, key)
def	copy_vec (event, obj, key)
def	copy_view (event, obj, key)

Variables
	CVec_type = ROOT.DataVector(ROOT.DMTest.C_v1)
dictionary	accessors

Function Documentation

_typename()

def xAODRootTest._typename (   t )

private

Definition at line 34 of file xAODRootTest.py.

def _typename(t):
    return getattr (t, '__cpp_name__', t.__name__)
 

CHECK()

def xAODRootTest.CHECK

(

sc

)

Definition at line 25 of file xAODRootTest.py.

def CHECK(sc):
    if not sc.isSuccess():
        raise Exception ('bad StatusCode')
    return
 
 
 
reg=ROOT.SG.AuxTypeRegistry.instance()
 

copy_obj()

def xAODRootTest.copy_obj	(		event,
			obj,
			key
	)

Definition at line 147 of file xAODRootTest.py.

def copy_obj (event, obj, key):
    copy = obj.__class__()
    copy_aux = obj.getConstStore().__class__()
    copy.setNonConstStore (copy_aux)
    copy.__assign__ (obj)
    CHECK (event.record (copy, key))
    CHECK (event.record (copy_aux, key + 'Aux.'))
    ROOT.SetOwnership (copy, False)
    ROOT.SetOwnership (copy_aux, False)
    return
 
 

copy_vec()

def xAODRootTest.copy_vec	(		event,
			obj,
			key
	)

Definition at line 159 of file xAODRootTest.py.

def copy_vec (event, obj, key):
    copy = obj.__class__()
    copy_aux = obj.getConstStore().__class__()
    copy.setNonConstStore (copy_aux)
    for i in range(obj.size()):
        elt_orig = obj[i]
        if _typename (elt_orig.__class__).startswith ('DataModel_detail::ElementProxy<'):
            elt_orig = elt_orig.__follow__()
        elt = elt_orig.__class__()
        copy.push_back(elt)
        ROOT.SetOwnership (elt, False)
        elt.__assign__ (elt_orig)
    CHECK (event.record (copy, key))
    CHECK (event.record (copy_aux, key + 'Aux.'))
    ROOT.SetOwnership (copy, False)
    ROOT.SetOwnership (copy_aux, False)
    return
 
 

copy_view()

def xAODRootTest.copy_view	(		event,
			obj,
			key
	)

Definition at line 178 of file xAODRootTest.py.

def copy_view (event, obj, key):
    copy = obj.__class__(obj)
    copy.toPersistent()
    CHECK (event.record (copy, key))
    ROOT.SetOwnership (copy, False)
    return
 
 

dump_auxdata()

def xAODRootTest.dump_auxdata	(		x,
			exclude = [],
			f = sys.stdout
	)

Definition at line 110 of file xAODRootTest.py.

def dump_auxdata (x, exclude=[], f = sys.stdout):
    auxids = list(x.getAuxIDs())
    auxids = [(reg.getName(id), id) for id in auxids]
    auxids.sort()
    for name, auxid in auxids:
        if name in exclude: continue
        if reg.isLinked (auxid): continue
        print (name + ': ', file=f, end='')
        dump_auxitem (x, auxid, f)
    return
 
 

dump_auxitem()

def xAODRootTest.dump_auxitem	(		x,
			auxid,
			f = sys.stdout
	)

Definition at line 98 of file xAODRootTest.py.

def dump_auxitem (x, auxid, f = sys.stdout):
    tname = reg.getTypeName (auxid)
    ac_p = accessors.get (tname)
    if not ac_p:
        print ('<unknown %s>'%tname, end='', file=f)
    else:
        (ac_cl, formatter) = ac_p
        val = ac_cl(reg.getName(auxid))(x)
        print (formatter(val) + '; ', end='', file=f)
    return
 
 

dump_c()

def xAODRootTest.dump_c	(		c,
			f = sys.stdout
	)

Definition at line 122 of file xAODRootTest.py.

def dump_c (c, f=sys.stdout):
    if hasattr(c, '__deref__'):
        c = c.__deref__()
    print ('  anInt1: %d; aFloat: %.1f; ' % (c.anInt(), c.aFloat()), end='')
    dump_auxdata (c, exclude = ['anInt1', 'aFloat'])
    print ('')
    return
 
 

dump_plinks()

def xAODRootTest.dump_plinks	(		p,
			f = sys.stdout
	)

Definition at line 139 of file xAODRootTest.py.

def dump_plinks (p, f=sys.stdout):
    if hasattr(p, '__deref__'):
        p = p.__deref__()
    dump_auxdata (p)
    print ('')
    return
 
 

dump_xaodobj()

def xAODRootTest.dump_xaodobj	(		h,
			f = sys.stdout
	)

Definition at line 131 of file xAODRootTest.py.

def dump_xaodobj (h, f=sys.stdout):
    if hasattr(h, '__deref__'):
        h = h.__deref__()
    dump_auxdata (h)
    print ('')
    return
 
 

format_el()

def xAODRootTest.format_el

(

x

)

Definition at line 55 of file xAODRootTest.py.

def format_el(x): return '%s[%s]' % (x.dataID(), ('inv' if x.isDefaultIndex() else x.index()))

format_el_vec()

def xAODRootTest.format_el_vec

(

v

)

Definition at line 56 of file xAODRootTest.py.

def format_el_vec(v): return '[' + ', '.join([format_el(el) for el in v.asVector()]) + ']'

format_float()

def xAODRootTest.format_float

(

x

)

Definition at line 53 of file xAODRootTest.py.

def format_float(x): return '%.1f'%x

format_float_vec()

def xAODRootTest.format_float_vec

(

v

)

Definition at line 57 of file xAODRootTest.py.

def format_float_vec(v):
    l = [format_float(x) for x in v]
    return '[' + ','.join(l) + ']'

format_int()

def xAODRootTest.format_int

(

x

)

Definition at line 52 of file xAODRootTest.py.

def format_int(x): return '%d'%x

format_int_vec()

def xAODRootTest.format_int_vec

(

v

)

Definition at line 60 of file xAODRootTest.py.

def format_int_vec(v):
    l = [format_int(x) for x in v]
    return '[' + ','.join(l) + ']'

format_str()

def xAODRootTest.format_str

(

x

)

Definition at line 54 of file xAODRootTest.py.

def format_str(x): return f"'{x}'"

format_str_vec()

def xAODRootTest.format_str_vec

(

v

)

Definition at line 63 of file xAODRootTest.py.

def format_str_vec(v):
    l = [format_str(x) for x in v]
    return '[' + ','.join(l) + ']'

xAODInit()

def xAODRootTest.xAODInit

(

)

Definition at line 37 of file xAODRootTest.py.

def xAODInit():
    ROOT.xAOD.TEvent
    CHECK(ROOT.xAOD.Init())
    import sys
    if len(sys.argv) >= 2 and sys.argv[1] == '--rntuple':
        return True
    return False
 
 
 
cvec_cls=ROOT.DataVector(ROOT.DMTest.C_v1)
cel_cls=ROOT.ElementLink(cvec_cls)
 
 
 

Variable Documentation

accessors

dictionary xAODRootTest.accessors

Initial value:

=  {
    'int'   :  (ROOT.SG.ConstAccessor(int), format_int),
    'unsigned int'   :  (getattr (ROOT, 'SG::ConstAccessor<unsigned int>'), format_int),
    'float' :  (ROOT.SG.ConstAccessor(float), format_float),
    'std::vector<float>' : (getattr (ROOT, 'SG::ConstAccessor<std::vector<float> >'), format_float_vec),
    'std::vector<int>' : (getattr (ROOT, 'SG::ConstAccessor<std::vector<int> >'), format_int_vec),
    'ElementLink<DataVector<DMTest::C_v1> >' :
               (ROOT.SG.ConstAccessor(cel_cls), format_el),
    'SG::JaggedVecElt<int>' :
    (ROOT.SG.ConstAccessor(ROOT.SG.JaggedVecElt(ROOT.int)),
     format_int_vec),
    'SG::JaggedVecElt<float>' :
    (ROOT.SG.ConstAccessor(ROOT.SG.JaggedVecElt(ROOT.float)),
     format_float_vec),
    'SG::JaggedVecElt<double>' :
    (ROOT.SG.ConstAccessor(ROOT.SG.JaggedVecElt(ROOT.double)),
     format_float_vec),
    'SG::JaggedVecElt<std::string>' :
    (ROOT.SG.ConstAccessor(ROOT.SG.JaggedVecElt(ROOT.std.string)),
     format_str_vec),
    'SG::JaggedVecElt<ElementLink<DataVector<DMTest::C_v1> > >' :
    (ROOT.SG.ConstAccessor(ROOT.SG.JaggedVecElt(ROOT.ElementLink(CVec_type))),
     format_el_vec),
    'SG::PackedLink<DataVector<DMTest::C_v1> >' :
    (ROOT.SG.ConstAccessor(ROOT.SG.PackedLink(ROOT.DataVector(ROOT.DMTest.C_v1))),
     format_el),
    'std::vector<SG::PackedLink<DataVector<DMTest::C_v1> > >' :
    (ROOT.SG.ConstAccessor(ROOT.std.vector(ROOT.SG.PackedLink(ROOT.DataVector(ROOT.DMTest.C_v1)))),
     format_el_vec),
    }

Definition at line 67 of file xAODRootTest.py.

CVec_type

xAODRootTest.CVec_type = ROOT.DataVector(ROOT.DMTest.C_v1)

Definition at line 66 of file xAODRootTest.py.