data_array.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	import numpy as np
	import functools
	import operator

	class DataArray:
	"""Class holding information on ndarray"""
	def __init__(self, data, spatial_axes, name='', components_order='C'):
	"""
	Data array constructor

	:param data: the numpy array containing the data (possibly a view)
	:param spatial_axes: a container of ints that indicate which axes of the
	array correspond to space dimensions (in order)
	:param name: the name of the data
	:param components_order: the order of the non-spatial axes of the array
	"""
	self.data = data
	axes = list(range(data.ndim))
	for ax in spatial_axes:
	axes.remove(ax)

	nb_components = functools.reduce(lambda x, y: x * data.shape[y], axes, 1)

	if components_order == 'C':
	axes.reverse()
	else:
	raise Exception('Unrecognized components order')

	axes += spatial_axes

	# Hopefully this is a view
	self.flat_data = self.data.transpose(*axes).reshape(-1, order='F')

	self.attributes = {
	"Name": name,
	"type": str(self.flat_data.dtype).capitalize(),
	"NumberOfComponents": str(nb_components)
	}

	def __str__(self):
	return self.attributes.__str__()