pythonmelautils.MultiDimensionalCppArray Class Reference

Inheritance diagram for pythonmelautils.MultiDimensionalCppArray:

Collaboration diagram for pythonmelautils.MultiDimensionalCppArray:

Public Member Functions
def	__init__ (self, uniqueid, cppvariable, includes, othercppargs, *dimensions)
def	writecpp (self, f=None)
def	compilecpp (self, f)
def	__getitem__ (self, item)
def	__setitem__ (self, item, value)

Public Attributes
	uniqueid
	cppvariable
	dimensions
	ndim
	subarrays
	othercppargs
	functionfilecontents
	includes
	functionfile
	getitem
	setitem

Static Public Attributes
string	functionfiletemplate
list	uniqueids = []
dictionary	functionfiles = {}
dictionary	getitems = {}
dictionary	setitems = {}

Detailed Description

Definition at line 57 of file pythonmelautils.py.

Constructor & Destructor Documentation

__init__()

def pythonmelautils.MultiDimensionalCppArray.__init__	(		self,
			uniqueid,
			cppvariable,
			includes,
			othercppargs,
		*	dimensions
	)

Definition at line 70 of file pythonmelautils.py.

  def __init__(self, uniqueid, cppvariable, includes, othercppargs, *dimensions):
 
    self.uniqueid = uniqueid
    for i in self.uniqueids:
      if i == uniqueid:
        raise ValueError("Two MultiDimensionalCppArrays can't have the same id\n{}".format(i, self.uniqueid))
    self.uniqueids.append(uniqueid)
 
    self.cppvariable = cppvariable
    self.dimensions = dimensions
    self.ndim = len(self.dimensions)
    if self.ndim == 0:
      raise TypeError("Can't have a 0 dimensional array!")
 
    if self.ndim > 1:
      self.subarrays = []
      for i in range(dimensions[0]):
          othercppargs["int index{}".format(len(dimensions))] = i
          self.subarrays.append(
                                MultiDimensionalCppArray(
                                                         "{}_{}".format(self.uniqueid, i),
                                                         "{}[index{}]".format(cppvariable, len(dimensions)),
                                                         includes,
                                                         othercppargs,
                                                         *dimensions[1:]
                                                        )
                               )
    else:
      self.othercppargs = OrderedDict(othercppargs)
      self.functionfilecontents = self.functionfiletemplate.format(
                                                                   name="NAME",
                                                                   cppvariable=self.cppvariable,
                                                                   cppargs=",".join([key for key in self.othercppargs]+["int item"]),
                                                                   includes="\n".join("#include <{}>".format(_) for _ in includes),
                                                                  )
      self.includes = includes
      self.functionfile = self.getitem = self.setitem = None
 

Member Function Documentation

__getitem__()

def pythonmelautils.MultiDimensionalCppArray.__getitem__	(		self,
			item
	)

Definition at line 140 of file pythonmelautils.py.

  def __getitem__(self, item):
    if self.ndim > 1:
      return self.subarrays[item]
    else:
      if self.getitem is None: self.compilecpp(self.writecpp())
      if item >= self.dimensions[0]:
        raise IndexError("Index {} out of range (0-{})".format(item, self.dimensions[0]))
      return self.getitem(*(list(self.othercppargs.values())+[item]))
 

__setitem__()

def pythonmelautils.MultiDimensionalCppArray.__setitem__	(		self,
			item,
			value
	)

Definition at line 149 of file pythonmelautils.py.

  def __setitem__(self, item, value):
    if self.ndim > 1:
      raise TypeError("Need to specify all indices to write to the array.")
    else:
      if self.setitem is None: self.compilecpp()
      if item >= self.dimensions[0]:
        raise IndexError("Index {} out of range (0-{})".format(item, self.dimensions[0]))
      self.setitem(*(list(self.othercppargs.values())+[item, value]))
 

compilecpp()

def pythonmelautils.MultiDimensionalCppArray.compilecpp	(		self,
			f
	)

Definition at line 123 of file pythonmelautils.py.

  def compilecpp(self, f):
    if self.ndim > 1:
      for subarray in self.subarrays:
        subarray.compilecpp(f)
      return
 
    if self.functionfilecontents not in self.getitems:
      f.compile()
      self.functionfiles[self.functionfilecontents] = f
      self.getitems[self.functionfilecontents] = getattr(ROOT, "{}_getitem".format(self.uniqueid))
      self.setitems[self.functionfilecontents] = getattr(ROOT, "{}_setitem".format(self.uniqueid))
 
    self.functionfile = self.functionfiles[self.functionfilecontents]
    self.getitem = self.getitems[self.functionfilecontents]
    self.setitem = self.setitems[self.functionfilecontents]
 
 

writecpp()

def pythonmelautils.MultiDimensionalCppArray.writecpp	(		self,
			f = None
	)

Definition at line 108 of file pythonmelautils.py.

  def writecpp(self, f=None):
    if self.ndim > 1:
      for subarray in self.subarrays:
        f = subarray.writecpp(f)
      return f
 
    if self.functionfilecontents not in self.functionfiles:
      if f is None:
        f = NamedTemporaryMacro()
      self.functionfiles[self.functionfilecontents] = f
      f.write(self.functionfilecontents.replace("NAME", self.uniqueid))
      return f
    else:
      return self.functionfiles[self.functionfilecontents]
 

Member Data Documentation

cppvariable

pythonmelautils.MultiDimensionalCppArray.cppvariable

Definition at line 78 of file pythonmelautils.py.

dimensions

pythonmelautils.MultiDimensionalCppArray.dimensions

Definition at line 79 of file pythonmelautils.py.

functionfile

pythonmelautils.MultiDimensionalCppArray.functionfile

Definition at line 106 of file pythonmelautils.py.

functionfilecontents

pythonmelautils.MultiDimensionalCppArray.functionfilecontents

Definition at line 99 of file pythonmelautils.py.

functionfiles

dictionary pythonmelautils.MultiDimensionalCppArray.functionfiles = {}

static

Definition at line 66 of file pythonmelautils.py.

functionfiletemplate

string pythonmelautils.MultiDimensionalCppArray.functionfiletemplate

static

Initial value:

=  """
  #include <type_traits>
  {includes}
  auto {name}_getitem({cppargs}) -> std::remove_reference<decltype({cppvariable}[item])>::type {{return {cppvariable}[item];}}
  void {name}_setitem({cppargs}, std::remove_reference<decltype({cppvariable}[item])>::type value) {{{cppvariable}[item] = value;}}
"""

Definition at line 58 of file pythonmelautils.py.

getitem

pythonmelautils.MultiDimensionalCppArray.getitem

Definition at line 106 of file pythonmelautils.py.

getitems

dictionary pythonmelautils.MultiDimensionalCppArray.getitems = {}

static

Definition at line 67 of file pythonmelautils.py.

includes

pythonmelautils.MultiDimensionalCppArray.includes

Definition at line 105 of file pythonmelautils.py.

ndim

pythonmelautils.MultiDimensionalCppArray.ndim

Definition at line 80 of file pythonmelautils.py.

othercppargs

pythonmelautils.MultiDimensionalCppArray.othercppargs

Definition at line 98 of file pythonmelautils.py.

setitem

pythonmelautils.MultiDimensionalCppArray.setitem

Definition at line 106 of file pythonmelautils.py.

setitems

dictionary pythonmelautils.MultiDimensionalCppArray.setitems = {}

static

Definition at line 68 of file pythonmelautils.py.

subarrays

pythonmelautils.MultiDimensionalCppArray.subarrays

Definition at line 85 of file pythonmelautils.py.

uniqueid

pythonmelautils.MultiDimensionalCppArray.uniqueid

Definition at line 72 of file pythonmelautils.py.

uniqueids

list pythonmelautils.MultiDimensionalCppArray.uniqueids = []

static

Definition at line 65 of file pythonmelautils.py.

---

The documentation for this class was generated from the following file:

• MELA/python/pythonmelautils.py