"""
Wrappers for basic types that are compatible with ROOT TTrees
"""
from array import array
from .registry import register
import ROOT
[docs]class Column(object):
def __init__(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
def __call__(self):
return self.type(*self.args, **self.kwargs)
def __repr__(self):
arg_params = ', '.join([str(a) for a in self.args])
kwd_params = ', '.join(['%s=%s' % (name, value)
for name, value in self.kwargs.items()])
params = []
if arg_params:
params.append(arg_params)
if kwd_params:
params.append(kwd_params)
return "%s(%s)" % (self.__class__.__name__, ', '.join(params))
def __str__(self):
return repr(self)
[docs]class ObjectCol(Column):
def __init__(self, cls, *args, **kwargs):
self.type = cls
Column.__init__(self, *args, **kwargs)
[docs]class Variable(array):
"""This is the base class for all variables"""
def __init__(self, resetable=True):
array.__init__(self)
self.resetable = resetable
[docs] def reset(self):
"""Reset the value to the default"""
if self.resetable:
self[0] = self.default
[docs] def clear(self):
"""Supplied to match the interface of ROOT.vector"""
self.reset()
@property
[docs] def value(self):
"""The current value"""
return self[0]
[docs] def set(self, value):
"""Set the value"""
if isinstance(value, Variable):
self[0] = self.convert(value.value)
else:
self[0] = self.convert(value)
def __str__(self):
return self.__repr__()
def __repr__(self):
return "%s(%s) at %s" % \
(self.__class__.__name__, repr(self.value), id(self).__hex__())
def __getitem__(self, i):
return array.__getitem__(self, 0)
def __setitem__(self, i, value):
if isinstance(value, Variable):
array.__setitem__(self, 0, value.value)
else:
array.__setitem__(self, 0, value)
def __lt__(self, value):
if isinstance(value, Variable):
return self.value < value.value
return self.value < value
def __le__(self, value):
if isinstance(value, Variable):
return self.value <= value.value
return self.value <= value
def __eq__(self, value):
if isinstance(value, Variable):
return self.value == value.value
return self.value == value
def __ne__(self, value):
if isinstance(value, Variable):
return self.value != value.value
return self.value != value
def __gt__(self, value):
if isinstance(value, Variable):
return self.value > value.value
return self.value > value
def __ge__(self, value):
if isinstance(value, Variable):
return self.value >= value.value
return self.value >= value
def __nonzero__(self):
return self.value != 0
def __add__(self, other):
if isinstance(other, Variable):
return self.value + other.value
return self.value + other
def __radd__(self, other):
return self + other
def __sub__(self, other):
if isinstance(other, Variable):
return self.value - other.value
return self.value - other
def __rsub__(self, other):
return other - self.value
def __mul__(self, other):
if isinstance(other, Variable):
return self.value * other.value
return self.value * other
def __rmul__(self, other):
return self * other
def __div__(self, other):
if isinstance(other, Variable):
return self.value / other.value
return self.value / other
def __rdiv__(self, other):
return other / self.value
[docs]class VariableArray(array):
"""This is the base class for all array variables"""
def __init__(self, resetable=True):
array.__init__(self)
self.resetable = resetable
[docs] def reset(self):
"""Reset the value to the default"""
if self.resetable:
for i in xrange(len(self)):
self[i] = self.default
[docs] def clear(self):
"""Supplied to match the interface of ROOT.vector"""
self.reset()
[docs] def set(self, other):
for i, thing in enumerate(other):
self[i] = thing
def __str__(self):
return self.__repr__()
def __repr__(self):
return "%s[%s] at %s" % \
(self.__class__.__name__,
', '.join(map(str, self)),
id(self).__hex__())
@register(names=('B', 'BOOL_T'), builtin=True)
[docs]class Bool(Variable):
"""
This is a variable containing a Boolean type
"""
# The ROOT character representation of the Boolean type
type = 'O'
typename = 'Bool_t'
def __new__(cls, default=False, **kwargs):
return Variable.__new__(cls, 'B', [Bool.convert(default)])
def __init__(self, default=False, **kwargs):
Variable.__init__(self, **kwargs)
self.default = Bool.convert(default)
@classmethod
[docs] def convert(cls, value):
return int(bool(value))
[docs]class BoolCol(Column):
type = Bool
@register(names=('B[]', 'BOOL_T[]'), builtin=True)
[docs]class BoolArray(VariableArray):
"""
This is an array of Booleans
"""
# The ROOT character representation of the Boolean type
type = 'O'
typename = 'Bool_t'
def __new__(cls, length, default=False, **kwargs):
return VariableArray.__new__(cls, 'B',
[Bool.convert(default)] * length)
def __init__(self, length, default=False, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Bool.convert(default)
[docs]class BoolArrayCol(Column):
type = BoolArray
@register(names=('C', 'CHAR_T'), builtin=True)
[docs]class Char(Variable):
"""
This is a variable containing a character type
"""
# The ROOT character representation of the char type
type = 'C'
typename = 'Char_t'
def __new__(cls, default='\x00', **kwargs):
return Variable.__new__(cls, 'b', [Char.convert(default)])
def __init__(self, default=False, **kwargs):
Variable.__init__(self, **kwargs)
self.default = Char.convert(default)
@classmethod
[docs] def convert(cls, value):
if type(value) is int:
return chr(value)
return value
[docs]class CharCol(Column):
type = Char
@register(names=('C[]', 'CHAR_T[]'), builtin=True)
[docs]class CharArray(VariableArray):
"""
This is an array of characters
"""
# The ROOT character representation of the char type
type = 'C'
typename = 'Char_t'
def __new__(cls, length, default='\x00', **kwargs):
return VariableArray.__new__(cls, 'b',
[Char.convert(default)] * length)
def __init__(self, length, default=False, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Char.convert(default)
[docs]class CharArrayCol(Column):
type = CharArray
@register(names=('UC', 'UCHAR_T'), builtin=True)
[docs]class UChar(Variable):
"""
This is a variable containing an unsigned character type
"""
# The ROOT character representation of the unsigned char type
type = 'c'
typename = 'UChar_t'
def __new__(cls, default='\x00', **kwargs):
return Variable.__new__(cls, 'B', [UChar.convert(default)])
def __init__(self, default=False, **kwargs):
Variable.__init__(self, **kwargs)
self.default = UChar.convert(default)
@classmethod
[docs] def convert(cls, value):
if type(value) is int:
return chr(value)
return value
[docs]class UCharCol(Column):
type = UChar
@register(names=('UC[]', 'UCHAR_T[]'), builtin=True)
[docs]class UCharArray(VariableArray):
"""
This is an array of unsigned characters
"""
# The ROOT character representation of the unsigned char type
type = 'c'
typename = 'UChar_t'
def __new__(cls, length, default='\x00', **kwargs):
return VariableArray.__new__(cls, 'B',
[UChar.convert(default)] * length)
def __init__(self, length, default=False, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = UChar.convert(default)
[docs]class UCharArrayCol(Column):
type = UCharArray
@register(names=('S', 'SHORT_T'), builtin=True)
[docs]class Short(Variable):
"""
This is a variable containing an integer
"""
# The ROOT character representation of the short type
type = 'S'
typename = 'Short_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'h', [Short.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = Short.convert(default)
@classmethod
[docs] def convert(cls, value):
return int(value)
[docs]class ShortCol(Column):
type = Short
@register(names=('S[]', 'SHORT_T[]'), builtin=True)
[docs]class ShortArray(VariableArray):
"""
This is an array of integers
"""
# The ROOT character representation of the short type
type = 'S'
typename = 'Short_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'h',
[Short.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Short.convert(default)
[docs]class ShortArrayCol(Column):
type = ShortArray
@register(names=('US', 'USHORT_T'), builtin=True)
[docs]class UShort(Variable):
"""
This is a variable containing a short
"""
# The ROOT character representation of the unsigned short type
type = 's'
typename = 'UShort_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'H', [UShort.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = UShort.convert(default)
@classmethod
[docs] def convert(cls, value):
if value < 0:
raise ValueError(
"Assigning negative value (%i) to unsigned type" % value)
return int(value)
[docs]class UShortCol(Column):
type = UShort
@register(names=('US[]', 'USHORT_T[]'), builtin=True)
[docs]class UShortArray(VariableArray):
"""
This is an array of unsigned shorts
"""
# The ROOT character representation of the unsigned short type
type = 's'
typename = 'UShort_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'H',
[UShort.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = UShort.convert(default)
[docs]class UShortArrayCol(Column):
type = UShortArray
@register(names=('I', 'INT_T'), builtin=True)
[docs]class Int(Variable):
"""
This is a variable containing an integer
"""
# The ROOT character representation of the integer type
type = 'I'
typename = 'Int_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'i', [Int.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = Int.convert(default)
@classmethod
[docs] def convert(cls, value):
return int(value)
[docs]class IntCol(Column):
type = Int
@register(names=('I[]', 'INT_T[]'), builtin=True)
[docs]class IntArray(VariableArray):
"""
This is an array of integers
"""
# The ROOT character representation of the integer type
type = 'I'
typename = 'Int_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'i',
[Int.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Int.convert(default)
[docs]class IntArrayCol(Column):
type = IntArray
@register(names=('UI', 'UINT_T'), builtin=True)
[docs]class UInt(Variable):
"""
This is a variable containing an unsigned integer
"""
# The ROOT character representation of the unsigned integer type
type = 'i'
typename = 'UInt_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'I', [UInt.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = UInt.convert(default)
@classmethod
[docs] def convert(cls, value):
if value < 0:
raise ValueError(
"Assigning negative value (%i) to unsigned type" % value)
return long(value)
[docs]class UIntCol(Column):
type = UInt
@register(names=('UI[]', 'UINT_T[]'), builtin=True)
[docs]class UIntArray(VariableArray):
"""
This is an array of unsigned integers
"""
# The ROOT character representation of the unsigned integer type
type = 'i'
typename = 'UInt_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'I',
[UInt.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = UInt.convert(default)
[docs]class UIntArrayCol(Column):
type = UIntArray
@register(names=('L', 'LONG64_T'), builtin=True)
[docs]class Long(Variable):
"""
This is a variable containing a long
"""
# The ROOT character representation of the long type
type = 'L'
typename = 'Long64_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'l', [Long.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = Long.convert(default)
@classmethod
[docs] def convert(cls, value):
return long(value)
[docs]class LongCol(Column):
type = Long
@register(names=('L[]', 'LONG64_T[]'), builtin=True)
[docs]class LongArray(VariableArray):
"""
This is an array of longs
"""
# The ROOT character representation of the long type
type = 'L'
typename = 'Long64_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'l',
[Long.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Long.convert(default)
[docs]class LongArrayCol(Column):
type = LongArray
@register(names=('UL', 'ULONG64_T'), builtin=True)
[docs]class ULong(Variable):
"""
This is a variable containing an unsigned long
"""
# The ROOT character representation of the long type
type = 'l'
typename = 'ULong64_t'
def __new__(cls, default=0, **kwargs):
return Variable.__new__(cls, 'L', [ULong.convert(default)])
def __init__(self, default=0, **kwargs):
Variable.__init__(self, **kwargs)
self.default = ULong.convert(default)
@classmethod
[docs] def convert(cls, value):
if value < 0:
raise ValueError(
"Assigning negative value (%i) to unsigned type" % value)
return long(value)
[docs]class ULongCol(Column):
type = ULong
@register(names=('UL[]', 'ULONG64_T[]'), builtin=True)
[docs]class ULongArray(VariableArray):
"""
This is of unsigned longs
"""
# The ROOT character representation of the long type
type = 'l'
typename = 'ULong64_t'
def __new__(cls, length, default=0, **kwargs):
return VariableArray.__new__(cls, 'L',
[ULong.convert(default)] * length)
def __init__(self, length, default=0, **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = ULong.convert(default)
[docs]class ULongArrayCol(Column):
type = ULongArray
@register(names=('F', 'FLOAT_T'), builtin=True)
[docs]class Float(Variable):
"""
This is a variable containing a float
"""
# The ROOT character representation of the float type
type = 'F'
typename = 'Float_t'
def __new__(cls, default=0., **kwargs):
return Variable.__new__(cls, 'f', [Float.convert(default)])
def __init__(self, default=0., **kwargs):
Variable.__init__(self, **kwargs)
self.default = Float.convert(default)
@classmethod
[docs] def convert(cls, value):
return float(value)
[docs]class FloatCol(Column):
type = Float
@register(names=('F[]', 'FLOAT_T[]'), builtin=True)
[docs]class FloatArray(VariableArray):
"""
This is an array of floats
"""
# The ROOT character representation of the float type
type = 'F'
typename = 'Float_t'
def __new__(cls, length, default=0., **kwargs):
return VariableArray.__new__(cls, 'f',
[Float.convert(default)] * length)
def __init__(self, length, default=0., **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Float.convert(default)
[docs]class FloatArrayCol(Column):
type = FloatArray
@register(names=('D', 'DOUBLE_T'), builtin=True)
[docs]class Double(Variable):
"""
This is a variable containing a double
"""
# The ROOT character representation of the double type
type = 'D'
typename = 'Double_t'
def __new__(cls, default=0., **kwargs):
return Variable.__new__(cls, 'd', [Double.convert(default)])
def __init__(self, default=0., **kwargs):
Variable.__init__(self, **kwargs)
self.default = Double.convert(default)
@classmethod
[docs] def convert(cls, value):
return float(value)
[docs]class DoubleCol(Column):
type = Double
@register(names=('D[]', 'DOUBLE_T[]'), builtin=True)
[docs]class DoubleArray(VariableArray):
"""
This is an array of doubles
"""
# The ROOT character representation of the double type
type = 'D'
typename = 'Double_t'
def __new__(cls, length, default=0., **kwargs):
return VariableArray.__new__(cls, 'd',
[Double.convert(default)] * length)
def __init__(self, length, default=0., **kwargs):
VariableArray.__init__(self, **kwargs)
self.default = Double.convert(default)
[docs]class DoubleArrayCol(Column):
type = DoubleArray
"""
generate dictionaries for commonly used types not included in ROOT
"""
from .classfactory import generate
if ROOT.gROOT.GetVersionInt() >= 53000:
generate('vector<vector<bool> >', '<vector>')
generate('vector<vector<vector<bool> > >', '<vector>')
generate('vector<vector<char> >', '<vector>')
generate('vector<vector<unsigned char> >', '<vector>')
generate('vector<vector<short> >', '<vector>')
generate('vector<vector<unsigned short> >', '<vector>')
generate('vector<vector<int> >', '<vector>')
generate('vector<vector<unsigned int> >', '<vector>')
generate('vector<vector<long> >', '<vector>')
generate('vector<vector<unsigned long> >', '<vector>')
generate('vector<vector<float> >', '<vector>')
generate('vector<vector<double> >', '<vector>')
generate('vector<vector<string> >', '<vector>;<string>')
generate('vector<vector<vector<char> > >', '<vector>')
generate('vector<vector<vector<unsigned char> > >', '<vector>')
generate('vector<vector<vector<short> > >', '<vector>')
generate('vector<vector<vector<unsigned short> > >', '<vector>')
generate('vector<vector<vector<int> > >', '<vector>')
generate('vector<vector<vector<unsigned int> > >', '<vector>')
generate('vector<vector<vector<long> > >', '<vector>')
generate('vector<vector<vector<unsigned long> > >', '<vector>')
generate('vector<vector<vector<float> > >', '<vector>')
generate('vector<vector<vector<double> > >', '<vector>')
generate('vector<vector<vector<string> > >', '<vector>;<string>')
generate('vector<TLorentzVector>', 'TLorentzVector.h')
generate('vector<vector<TLorentzVector> >', 'TLorentzVector.h')
generate('vector<vector<vector<TLorentzVector> > >', 'TLorentzVector.h')
generate('map<string,string>', '<map>;<string>')
"""
Register builtin types:
"""
if ROOT.gROOT.GetVersionInt() >= 53000:
register(builtin=True, names=('VB', 'VECTOR<BOOL>'))(ROOT.vector('bool'))
register(builtin=True, names=('VVB', 'VECTOR<VECTOR<BOOL> >'), demote='VB')(ROOT.vector('vector<bool>'))
register(builtin=True, names=('VVVB', 'VECTOR<VECTOR<VECTOR<BOOL> > >'), demote='VVB')(ROOT.vector('vector<vector<bool> >'))
register(builtin=True, names=('VC', 'VECTOR<CHAR>'))(ROOT.vector('char'))
register(builtin=True, names=('VUC', 'VECTOR<UNSIGNED CHAR>'))(ROOT.vector('unsigned char'))
register(builtin=True, names=('VS', 'VECTOR<SHORT>'))(ROOT.vector('short'))
register(builtin=True, names=('VUS', 'VECTOR<UNSIGNED SHORT>'))(ROOT.vector('unsigned short'))
register(builtin=True, names=('VI', 'VECTOR<INT>'), demote='I')(ROOT.vector('int'))
register(builtin=True, names=('VUI', 'VECTOR<UNSIGNED INT>'), demote='UI')(ROOT.vector('unsigned int'))
register(builtin=True, names=('VL', 'VECTOR<LONG>'), demote='L')(ROOT.vector('long'))
register(builtin=True, names=('VUL', 'VECTOR<UNSIGNED LONG>'), demote='UL')(ROOT.vector('unsigned long'))
register(builtin=True, names=('VF', 'VECTOR<FLOAT>'), demote='F')(ROOT.vector('float'))
register(builtin=True, names=('VD', 'VECTOR<DOUBLE>'), demote='D')(ROOT.vector('double'))
register(builtin=True, names=('VVC', 'VECTOR<VECTOR<CHAR> >'), demote='VC')(ROOT.vector('vector<char>'))
register(builtin=True, names=('VVUC', 'VECTOR<VECTOR<UNSIGNED CHAR> >'), demote='VUC')(ROOT.vector('vector<unsigned char>'))
register(builtin=True, names=('VVS', 'VECTOR<VECTOR<SHORT> >'), demote='VS')(ROOT.vector('vector<short>'))
register(builtin=True, names=('VVUS', 'VECTOR<VECTOR<UNSIGNED SHORT> >'), demote='VUS')(ROOT.vector('vector<unsigned short>'))
register(builtin=True, names=('VVI', 'VECTOR<VECTOR<INT> >'), demote='VI')(ROOT.vector('vector<int>'))
register(builtin=True, names=('VVUI', 'VECTOR<VECTOR<UNSIGNED INT> >'), demote='VUI')(ROOT.vector('vector<unsigned int>'))
register(builtin=True, names=('VVL', 'VECTOR<VECTOR<LONG> >'), demote='VL')(ROOT.vector('vector<long>'))
register(builtin=True, names=('VVUL', 'VECTOR<VECTOR<UNSIGNED LONG> >'), demote='VUL')(ROOT.vector('vector<unsigned long>'))
register(builtin=True, names=('VVF', 'VECTOR<VECTOR<FLOAT> >'), demote='VF')(ROOT.vector('vector<float>'))
register(builtin=True, names=('VVD', 'VECTOR<VECTOR<DOUBLE> >'), demote='VD')(ROOT.vector('vector<double>'))
register(builtin=True, names=('VVVC', 'VECTOR<VECTOR<VECTOR<CHAR> > >'), demote='VVC')(ROOT.vector('vector<vector<char> >'))
register(builtin=True, names=('VVVUC', 'VECTOR<VECTOR<VECTOR<UNSIGNED CHAR> > >'), demote='VVUC')(ROOT.vector('vector<vector<unsigned char> >'))
register(builtin=True, names=('VVVS', 'VECTOR<VECTOR<VECTOR<SHORT> > >'), demote='VVS')(ROOT.vector('vector<vector<short> >'))
register(builtin=True, names=('VVVUS', 'VECTOR<VECTOR<VECTOR<UNSIGNED SHORT> > >'), demote='VVUS')(ROOT.vector('vector<vector<unsigned short> >'))
register(builtin=True, names=('VVVI', 'VECTOR<VECTOR<VECTOR<INT> > >'), demote='VVI')(ROOT.vector('vector<vector<int> >'))
register(builtin=True, names=('VVVUI', 'VECTOR<VECTOR<VECTOR<UNSIGNED INT> > >'), demote='VVUI')(ROOT.vector('vector<vector<unsigned int> >'))
register(builtin=True, names=('VVVL', 'VECTOR<VECTOR<VECTOR<LONG> > >'), demote='VVL')(ROOT.vector('vector<vector<long> >'))
register(builtin=True, names=('VVVUL', 'VECTOR<VECTOR<VECTOR<UNSIGNED LONG> > >'), demote='VVUL')(ROOT.vector('vector<vector<unsigned long> >'))
register(builtin=True, names=('VVVF', 'VECTOR<VECTOR<VECTOR<FLOAT> > >'), demote='VVF')(ROOT.vector('vector<vector<float> >'))
register(builtin=True, names=('VVVD', 'VECTOR<VECTOR<VECTOR<DOUBLE> > >'), demote='VVD')(ROOT.vector('vector<vector<double> >'))
register(builtin=True, names=('VSTR', 'VECTOR<STRING>'))(ROOT.vector('string'))
register(builtin=True, names=('VVSTR', 'VECTOR<VECTOR<STRING> >'), demote='VSTR')(ROOT.vector('vector<string>'))
register(builtin=True, names=('VVVSTR', 'VECTOR<VECTOR<VECTOR<STRING> > >'), demote='VVSTR')(ROOT.vector('vector<vector<string> >'))
register(builtin=True, names=('MSTRI', 'MAP<STRING,INT>'))(ROOT.map('string,int'))
register(builtin=True, names=('MSTRF', 'MAP<STRING,FLOAT>'))(ROOT.map('string,float'))
register(builtin=True, names=('MSTRS', 'MAP<STRING,STRING>'))(ROOT.map('string,string'))
register(builtin=True, names=('VECTOR<TLORENTZVECTOR>',))(ROOT.vector('TLorentzVector'))
register(builtin=True, names=('VECTOR<VECTOR<TLORENTZVECTOR> >',))(ROOT.vector('vector<TLorentzVector>'))
register(builtin=True, names=('VECTOR<VECTOR<VECTOR<TLORENTZVECTOR> > >',))(ROOT.vector('vector<vector<TLorentzVector> >'))
# ROOT type codes:
root_type_codes = \
'''\
O a boolean (Bool_t) (see note 1)
B an 8 bit signed integer (Char_t)
b an 8 bit unsigned integer (UChar_t)
S a 16 bit signed integer (Short_t)
s a 16 bit unsigned integer (UShort_t)
I a 32 bit signed integer (Int_t)
i a 32 bit unsigned integer (UInt_t)
L a 64 bit signed integer (Long64_t)
l a 64 bit unsigned integer (ULong64_t)
F a 32 bit floating point (Float_t)
D a 64 bit floating point (Double_t)\
'''
root_type_codes = [line.split()[0] for line in root_type_codes.split('\n')]
# ROOT type names:
root_type_names = \
'''\
Bool_t
Char_t
UChar_t
Short_t
UShort_t
Int_t
UInt_t
Long64_t
ULong64_t
Float_t
Double_t\
'''
root_type_names = [line.split()[0] for line in root_type_names.split('\n')]
# Python array:
python_codes = \
'''\
B unsigned char int 1 (used as boolean)
b signed char int 1
B unsigned char int 1
h signed short int 2
H unsigned short int 2
i signed int int 2
I unsigned int long 2
l signed long int 4
L unsigned long long 4
f float float 4
d double float 8\
'''
python_codes = [line.split()[0] for line in python_codes.split('\n')]
# Python NumPy array:
numpy_codes = \
'''\
b Boolean
i1 Char
u1 Unsigned Char
i2 Short Integer
u2 Unsigned Short integer
i4 Integer
u4 Unsigned integer
i8 Long Integer
u8 Unsigned Long integer
f4 Floating point
f8 Double Floating point\
'''
numpy_codes = [line.split()[0] for line in numpy_codes.split('\n')]
[docs]def convert(origin, target, type):
"""
convert type from origin to target
origin/target must be ROOTCODE, ROOTNAME, ARRAY, or NUMPY
"""
_origin = origin.upper()
if _origin == 'ROOTCODE':
_origin = root_type_codes
elif _origin == 'ROOTNAME':
_origin = root_type_names
elif _origin == 'ARRAY':
_origin = python_codes
elif _origin == 'NUMPY':
_origin = numpy_codes
else:
raise ValueError("%s is not a valid type" % origin)
_target = target.upper()
if _target == 'ROOTCODE':
_target = root_type_codes
elif _target == 'ROOTNAME':
_target = root_type_names
elif _target == 'ARRAY':
_target = python_codes
elif _target == 'NUMPY':
_target = numpy_codes
else:
raise ValueError("%s is not a valid type" % target)
try:
index = _origin.index(type)
except:
raise ValueError("%s is not a valid %s type" % (type, origin))
return _target[index]