def test_arr2cell_array(self):
"""Test Numeric array to vtkCellArray conversion."""
# Test list of lists.
a = [[0], [1, 2], [3, 4, 5], [6, 7, 8, 9]]
cells = array_handler.array2vtkCellArray(a)
z = numpy.array([1, 0, 2, 1,2, 3, 3,4,5, 4, 6,7,8,9])
arr = array_handler.vtk2array(cells.GetData())
self.assertEqual(numpy.sum(arr - z), 0)
self.assertEqual(len(arr.shape), 1)
self.assertEqual(len(arr), 14)
# Test if optional argument stuff also works.
cells = vtk.vtkCellArray()
ident = id(cells)
cells = array_handler.array2vtkCellArray(a, cells)
self.assertEqual(id(cells), ident)
arr = array_handler.vtk2array(cells.GetData())
self.assertEqual(numpy.sum(arr - z), 0)
self.assertEqual(cells.GetNumberOfCells(), 4)
# Make sure this resets the cell array and does not add to the
# existing list!
cells = array_handler.array2vtkCellArray(a, cells)
self.assertEqual(cells.GetNumberOfCells(), 4)
# Test Numeric array handling.
N = 3
a = numpy.zeros((N,3), numpy.int)
a[:,1] = 1
a[:,2] = 2
cells = array_handler.array2vtkCellArray(a)
arr = array_handler.vtk2array(cells.GetData())
expect = numpy.array([3, 0, 1, 2]*3, numpy.int)
self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)),
True)
self.assertEqual(cells.GetNumberOfCells(), N)
# Test if a list of Numeric arrays of different cell lengths works.
l_a = [a[:,:1], a, a[:2,:2]]
cells = array_handler.array2vtkCellArray(l_a)
arr = array_handler.vtk2array(cells.GetData())
expect = numpy.array([1, 0]*3 + [3, 0, 1, 2]*3 + [2, 0,1]*2, numpy.int)
self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)),
True)
self.assertEqual(cells.GetNumberOfCells(), N*2 + 2)
# This should not take a long while. This merely tests if a
# million cells can be created rapidly.
N = int(1e6)
a = numpy.zeros((N,3), numpy.int)
a[:,1] = 1
a[:,2] = 2
cells = array_handler.array2vtkCellArray(a)
self.assertEqual(cells.GetNumberOfCells(), N)
def test_arr2cell_array(self):
"""Test Numeric array to vtkCellArray conversion."""
# Test list of lists.
a = [[0], [1, 2], [3, 4, 5], [6, 7, 8, 9]]
cells = array_handler.array2vtkCellArray(a)
z = numpy.array([1, 0, 2, 1, 2, 3, 3, 4, 5, 4, 6, 7, 8, 9])
arr = array_handler.vtk2array(cells.GetData())
self.assertEqual(numpy.sum(arr - z), 0)
self.assertEqual(len(arr.shape), 1)
self.assertEqual(len(arr), 14)
# Test if optional argument stuff also works.
cells = vtk.vtkCellArray()
ident = id(cells)
cells = array_handler.array2vtkCellArray(a, cells)
self.assertEqual(id(cells), ident)
arr = array_handler.vtk2array(cells.GetData())
self.assertEqual(numpy.sum(arr - z), 0)
self.assertEqual(cells.GetNumberOfCells(), 4)
# Make sure this resets the cell array and does not add to the
# existing list!
cells = array_handler.array2vtkCellArray(a, cells)
self.assertEqual(cells.GetNumberOfCells(), 4)
# Test Numeric array handling.
N = 3
a = numpy.zeros((N, 3), numpy.int)
a[:, 1] = 1
a[:, 2] = 2
cells = array_handler.array2vtkCellArray(a)
arr = array_handler.vtk2array(cells.GetData())
expect = numpy.array([3, 0, 1, 2] * 3, numpy.int)
self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True)
self.assertEqual(cells.GetNumberOfCells(), N)
# Test if a list of Numeric arrays of different cell lengths works.
l_a = [a[:, :1], a, a[:2, :2]]
cells = array_handler.array2vtkCellArray(l_a)
arr = array_handler.vtk2array(cells.GetData())
expect = numpy.array([1, 0] * 3 + [3, 0, 1, 2] * 3 + [2, 0, 1] * 2,
numpy.int)
self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True)
self.assertEqual(cells.GetNumberOfCells(), N * 2 + 2)
# This should not take a long while. This merely tests if a
# million cells can be created rapidly.
N = int(1e6)
a = numpy.zeros((N, 3), numpy.int)
a[:, 1] = 1
a[:, 2] = 2
cells = array_handler.array2vtkCellArray(a)
self.assertEqual(cells.GetNumberOfCells(), N)
def test_deref_array(self):
"""Test if dereferencing array args works correctly."""
sigs = [
[["vtkDataArray"]],
[["vtkFloatArray"]],
[["vtkCellArray"]],
[["vtkPoints"]],
[["int", "vtkIdList"]],
[["int", ("float", "float"), "vtkDataArray"]],
[["Prop", "int", "vtkDataArray"]],
[["Points", ("float", "float", "float")]],
]
args = [
[[1, 2, 3]],
[[0, 0, 0]],
[[[1, 2, 3], [4, 5, 6]]],
[[[0.0, 0.0, 0.0], [1.0, 1.0, 1.0]]],
[1, [1, 2, 3]],
[1, (0.0, 0.0), [1.0, 1.0, 1.0]],
[Prop(), 1, numpy.array([1.0, 1.0, 1.0])],
[[[1, 2, 3]], [1, 2, 3]],
]
r = array_handler.deref_array(args[0], sigs[0])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[0]), 0)
r = array_handler.deref_array(args[1], sigs[1])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[1]), 0)
r = array_handler.deref_array(args[2], sigs[2])
self.assertEqual(r[0].GetNumberOfCells(), 2)
r = array_handler.deref_array(args[3], sigs[3])
self.assertEqual(mysum(array_handler.vtk2array(r[0].GetData()) - numpy.array(args[3], "f")), 0)
r = array_handler.deref_array(args[4], sigs[4])
self.assertEqual(r[0], 1)
self.assertEqual(r[1].__class__.__name__, "vtkIdList")
r = array_handler.deref_array(args[5], sigs[5])
self.assertEqual(r[0], 1)
self.assertEqual(r[1], (0.0, 0.0))
self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[5][2]), 0)
r = array_handler.deref_array(args[6], sigs[6])
self.assertEqual(r[0].IsA("vtkProperty"), True)
self.assertEqual(r[1], 1)
self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[6][2]), 0)
r = array_handler.deref_array(args[7], sigs[7])
def test_deref_array(self):
"""Test if dereferencing array args works correctly."""
sigs = [[['vtkDataArray']],
[['vtkFloatArray']],
[['vtkCellArray']],
[['vtkPoints']],
[['int', 'vtkIdList']],
[['int', ('float', 'float'), 'vtkDataArray']],
[['Prop', 'int', 'vtkDataArray']],
[['Points', ('float', 'float', 'float')]]
]
args = [[[1,2,3]],
[[0,0,0]],
[[[1,2,3],[4,5,6]]],
[[[0.,0.,0.], [1.,1.,1.]]],
[1, [1,2,3]],
[1, (0.0, 0.0), [1.0, 1.0, 1.0]],
[Prop(), 1, numpy.array([1.0, 1.0, 1.0])],
[[[1,2,3]], [1,2,3]]
]
r = array_handler.deref_array(args[0], sigs[0])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) -args[0]), 0)
r = array_handler.deref_array(args[1], sigs[1])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[1]), 0)
r = array_handler.deref_array(args[2], sigs[2])
self.assertEqual(r[0].GetNumberOfCells(), 2)
r = array_handler.deref_array(args[3], sigs[3])
self.assertEqual(mysum(array_handler.vtk2array(r[0].GetData()) -
numpy.array(args[3], 'f')), 0)
r = array_handler.deref_array(args[4], sigs[4])
self.assertEqual(r[0], 1)
self.assertEqual(r[1].__class__.__name__, 'vtkIdList')
r = array_handler.deref_array(args[5], sigs[5])
self.assertEqual(r[0], 1)
self.assertEqual(r[1], (0.0, 0.0))
self.assertEqual(mysum(array_handler.vtk2array(r[2]) -args[5][2]), 0)
r = array_handler.deref_array(args[6], sigs[6])
self.assertEqual(r[0].IsA('vtkProperty'), True)
self.assertEqual(r[1], 1)
self.assertEqual(mysum(array_handler.vtk2array(r[2]) -args[6][2]), 0)
r = array_handler.deref_array(args[7], sigs[7])
def test_vtk2array_appended_array(self):
"""Test the vtk2array can tolerate appending a cached array."""
# array is cached upon array2vtk is called
arr = numpy.arange(8).reshape(2, 4)
vtk_arr = array_handler.array2vtk(arr)
arr1 = array_handler.vtk2array(vtk_arr)
# the vtk array is appended, shapes don't match cached array anymore
extra_row = (1, 2, 3, 4)
vtk_arr.InsertTuple4(2, *extra_row)
# arr2 has a different shape
arr2 = array_handler.vtk2array(vtk_arr)
self.assertEqual(arr2.shape, (3, 4))
# check values
expected = numpy.vstack((arr, numpy.array(extra_row)))
self.assertEqual(numpy.sum(arr2 - expected), 0)
def test_vtk2array_appended_array(self):
"""Test the vtk2array can tolerate appending a cached array."""
# array is cached upon array2vtk is called
arr = numpy.arange(8).reshape(2, 4)
vtk_arr = array_handler.array2vtk(arr)
arr1 = array_handler.vtk2array(vtk_arr)
# the vtk array is appended, shapes don't match cached array anymore
extra_row = (1, 2, 3, 4)
vtk_arr.InsertTuple4(2, *extra_row)
# arr2 has a different shape
arr2 = array_handler.vtk2array(vtk_arr)
self.assertEqual(arr2.shape, (3, 4))
# check values
expected = numpy.vstack((arr, numpy.array(extra_row)))
self.assertEqual(numpy.sum(arr2 - expected), 0)
def test_id_array(self):
"""Test if a vtkIdTypeArray is converted correctly."""
arr = vtk.vtkIdTypeArray()
arr.SetNumberOfTuples(10)
for i in range(10):
arr.SetValue(i, i)
np = array_handler.vtk2array(arr)
self.assertEqual(numpy.all(np == list(range(10))), True)
def test_id_array(self):
"""Test if a vtkIdTypeArray is converted correctly."""
arr = vtk.vtkIdTypeArray()
arr.SetNumberOfTuples(10)
for i in range(10):
arr.SetValue(i, i)
np = array_handler.vtk2array(arr)
self.assertEqual(numpy.all(np == range(10)), True)
def test_reference_to_array(self):
"""Does to_array return an existing array instead of a new copy."""
arr = numpy.arange(0.0, 10.0, 0.1)
arr = numpy.reshape(arr, (25, 4))
vtk_arr = array_handler.array2vtk(arr)
arr1 = array_handler.vtk2array(vtk_arr)
# Now make sure these are using the same memory.
arr[0][0] = 100.0
self.assertEqual(arr[0][0], arr1[0][0])
self.assertEqual(arr.shape, arr1.shape)
def test_reference_to_array(self):
"""Does to_array return an existing array instead of a new copy."""
arr = numpy.arange(0.0, 10.0, 0.1)
arr = numpy.reshape(arr, (25, 4))
vtk_arr = array_handler.array2vtk(arr)
arr1 = array_handler.vtk2array(vtk_arr)
# Now make sure these are using the same memory.
arr[0][0] = 100.0
self.assertEqual(arr[0][0], arr1[0][0])
self.assertEqual(arr.shape, arr1.shape)
def test_deref_array(self):
"""Test if dereferencing array args works correctly."""
sigs = [[['vtkDataArray']], [['vtkFloatArray']], [['vtkCellArray']],
[['vtkPoints']], [['int', 'vtkIdList']],
[['int', ('float', 'float'), 'vtkDataArray']],
[['Prop', 'int', 'vtkDataArray']],
[['Points', ('float', 'float', 'float')]]]
args = [[[1, 2, 3]], [[0, 0, 0]], [[[1, 2, 3], [4, 5, 6]]],
[[[0., 0., 0.], [1., 1., 1.]]], [1, [1, 2, 3]],
[1, (0.0, 0.0), [1.0, 1.0, 1.0]],
[Prop(), 1, numpy.array([1.0, 1.0, 1.0])],
[[[1, 2, 3]], [1, 2, 3]]]
r = array_handler.deref_array(args[0], sigs[0])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[0]), 0)
r = array_handler.deref_array(args[1], sigs[1])
self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[1]), 0)
r = array_handler.deref_array(args[2], sigs[2])
self.assertEqual(r[0].GetNumberOfCells(), 2)
r = array_handler.deref_array(args[3], sigs[3])
self.assertEqual(
mysum(
array_handler.vtk2array(r[0].GetData()) -
numpy.array(args[3], 'f')), 0)
r = array_handler.deref_array(args[4], sigs[4])
self.assertEqual(r[0], 1)
self.assertEqual(r[1].__class__.__name__, 'vtkIdList')
r = array_handler.deref_array(args[5], sigs[5])
self.assertEqual(r[0], 1)
self.assertEqual(r[1], (0.0, 0.0))
self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[5][2]), 0)
r = array_handler.deref_array(args[6], sigs[6])
self.assertEqual(r[0].IsA('vtkProperty'), True)
self.assertEqual(r[1], 1)
self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[6][2]), 0)
r = array_handler.deref_array(args[7], sigs[7])
def readData(self):
if self.type == 'vtu':
filename=''.join([self.filenameout,repr(self.offset).zfill(4),'.vtu'])
datareader = v.vtkXMLUnstructuredGridReader()
elif self.type == 'pvtu':
filename=''.join([self.filenameout,repr(self.offset).zfill(4),'.pvtu'])
datareader = v.vtkXMLPUnstructuredGridReader()
print '=== Reading ',filename,' ==='
datareader.SetFileName(filename)
datareader.Update()
self.data = datareader.GetOutput()
vtk_points=self.data.GetPoints().GetData()
self.points=ah.vtk2array(vtk_points)
def getVar(data, var='rho'):
var = (ah.vtk2array(data.GetCellData().GetArray(var))).astype(np.float64)
return var
def test_array2vtk(self):
"""Test Numeric array to VTK array conversion and vice-versa."""
# Put all the test arrays here.
t_z = []
# Test the different types of arrays.
t_z.append(numpy.array([-128, 0, 127], numpy.int8))
# FIXME: character arrays are a problem since there is no
# unique mapping to a VTK data type and back.
#t_z.append(numpy.array([-128, 0, 127], numpy.character))
t_z.append(numpy.array([-32768, 0, 32767], numpy.int16))
t_z.append(numpy.array([-2147483648, 0, 2147483647], numpy.int32))
t_z.append(numpy.array([0, 255], numpy.uint8))
t_z.append(numpy.array([0, 65535], numpy.uint16))
t_z.append(numpy.array([0, 4294967295L], numpy.uint32))
t_z.append(numpy.array([-1.0e38, 0, 1.0e38], 'f'))
t_z.append(numpy.array([-1.0e299, 0, 1.0e299], 'd'))
# Check multi-component arrays.
t_z.append(numpy.array([[1], [2], [300]], 'd'))
t_z.append(numpy.array([[1, 20], [300, 4000]], 'd'))
t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'f'))
t_z.append(numpy.array([[1, 2, 3],[4, 5, 6]], 'd'))
t_z.append(numpy.array([[1, 2, 3, 400],[4, 5, 6, 700]],
'd'))
t_z.append(numpy.array([range(9),range(10,19)], 'f'))
# Test if a Python list also works.
t_z.append(numpy.array([[1., 2., 3., 400.],[4, 5, 6, 700]],
'd'))
# Test if arrays with number of components not in [1,2,3,4,9] work.
t_z.append(numpy.array([[1, 2, 3, 400, 5000],
[4, 5, 6, 700, 8000]], 'd'))
t_z.append(numpy.array([range(10), range(10,20)], 'd'))
for z in t_z:
vtk_arr = array_handler.array2vtk(z)
# Test for memory leaks.
self.assertEqual(vtk_arr.GetReferenceCount(),
array_handler.BASE_REFERENCE_COUNT)
self._check_arrays(z, vtk_arr)
z1 = array_handler.vtk2array(vtk_arr)
if len(z.shape) == 1:
self.assertEqual(len(z1.shape), 1)
if z.dtype.char != 'c':
#print z1
self.assertEqual(sum(numpy.ravel(z) - numpy.ravel(z1)), 0)
else:
#print z1.astype('c')
self.assertEqual(z, z1.astype('c'))
# Check if type conversion works correctly.
z = numpy.array([-128, 0, 127], numpy.int8)
vtk_arr = vtk.vtkDoubleArray()
ident = id(vtk_arr)
vtk_arr = array_handler.array2vtk(z, vtk_arr)
# Make sure this is the same array!
self.assertEqual(ident, id(vtk_arr))
self._check_arrays(z, vtk_arr)
# Check the vtkBitArray.
vtk_arr = vtk.vtkBitArray()
vtk_arr.InsertNextValue(0)
vtk_arr.InsertNextValue(1)
vtk_arr.InsertNextValue(0)
vtk_arr.InsertNextValue(1)
arr = array_handler.vtk2array(vtk_arr)
self.assertEqual(numpy.sum(arr - [0,1,0,1]), 0)
vtk_arr = array_handler.array2vtk(arr, vtk_arr)
self.assertEqual(vtk_arr.GetValue(0), 0)
self.assertEqual(vtk_arr.GetValue(1), 1)
self.assertEqual(vtk_arr.GetValue(2), 0)
self.assertEqual(vtk_arr.GetValue(3), 1)
# ----------------------------------------
# Test if the array is copied or not.
a = numpy.array([[1, 2, 3],[4, 5, 6]], 'd')
vtk_arr = array_handler.array2vtk(a)
# Change the numpy array and see if the changes are
# reflected in the VTK array.
a[0] = [10.0, 20.0, 30.0]
self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.))
# Make sure the cache is doing its job.
key = vtk_arr.__this__
z = array_handler._array_cache.get(vtk_arr)
self.assertEqual(numpy.sum(z - numpy.ravel(a)), 0.0)
l1 = len(array_handler._array_cache)
# del the Numeric array and see if this still works.
del a
self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.))
# Check the cache -- just making sure.
self.assertEqual(len(array_handler._array_cache), l1)
# Delete the VTK array and see if the cache is cleared.
del vtk_arr
self.assertEqual(len(array_handler._array_cache), l1-1)
self.assertEqual(array_handler._array_cache._cache.has_key(key),
False)
# Make sure bit arrays are copied.
vtk_arr = vtk.vtkBitArray()
a = numpy.array([0,1,0,1], numpy.int32)
vtk_arr = array_handler.array2vtk(a, vtk_arr)
del a
self.assertEqual(vtk_arr.GetValue(0), 0)
self.assertEqual(vtk_arr.GetValue(1), 1)
self.assertEqual(vtk_arr.GetValue(2), 0)
self.assertEqual(vtk_arr.GetValue(3), 1)
# Make sure the code at least runs for all the non-complex
# numerical dtypes in numpy.
for dtype in (numpy.sctypes['int'] + numpy.sctypes['uint'] +
numpy.sctypes['float']):
array_handler.array2vtk(numpy.zeros((1,), dtype=dtype))
def getVar(data,var='rho'):
var=(ah.vtk2array(data.GetCellData().GetArray(var))).astype(np.float64)
return var
def getRho(data):
rho=(ah.vtk2array(data.GetCellData().GetArray('rho'))).astype(np.float64)
return rho
def getRho(data):
rho=ah.vtk2array(data.GetCellData().GetArray('rho'))
return rho
def test_array2vtk(self):
"""Test Numeric array to VTK array conversion and vice-versa."""
# Put all the test arrays here.
t_z = []
# Test the different types of arrays.
t_z.append(numpy.array([-128, 0, 127], numpy.int8))
# FIXME: character arrays are a problem since there is no
# unique mapping to a VTK data type and back.
#t_z.append(numpy.array([-128, 0, 127], numpy.character))
t_z.append(numpy.array([-32768, 0, 32767], numpy.int16))
t_z.append(numpy.array([-2147483648, 0, 2147483647], numpy.int32))
t_z.append(numpy.array([0, 255], numpy.uint8))
t_z.append(numpy.array([0, 65535], numpy.uint16))
t_z.append(numpy.array([0, 4294967295], numpy.uint32))
t_z.append(numpy.array([-1.0e38, 0, 1.0e38], 'f'))
t_z.append(numpy.array([-1.0e299, 0, 1.0e299], 'd'))
# Check multi-component arrays.
t_z.append(numpy.array([[1], [2], [300]], 'd'))
t_z.append(numpy.array([[1, 20], [300, 4000]], 'd'))
t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'f'))
t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'd'))
t_z.append(numpy.array([[1, 2, 3, 400], [4, 5, 6, 700]], 'd'))
t_z.append(numpy.array([list(range(9)), list(range(10, 19))], 'f'))
# Test if a Python list also works.
t_z.append(numpy.array([[1., 2., 3., 400.], [4, 5, 6, 700]], 'd'))
# Test if arrays with number of components not in [1,2,3,4,9] work.
t_z.append(
numpy.array([[1, 2, 3, 400, 5000], [4, 5, 6, 700, 8000]], 'd'))
t_z.append(numpy.array([list(range(10)), list(range(10, 20))], 'd'))
for z in t_z:
vtk_arr = array_handler.array2vtk(z)
# Test for memory leaks.
self.assertEqual(vtk_arr.GetReferenceCount(),
array_handler.BASE_REFERENCE_COUNT)
self._check_arrays(z, vtk_arr)
z1 = array_handler.vtk2array(vtk_arr)
if len(z.shape) == 1:
self.assertEqual(len(z1.shape), 1)
if z.dtype.char != 'c':
#print z1
self.assertEqual(sum(numpy.ravel(z) - numpy.ravel(z1)), 0)
else:
#print z1.astype('c')
self.assertEqual(z, z1.astype('c'))
# Check if type conversion works correctly.
z = numpy.array([-128, 0, 127], numpy.int8)
vtk_arr = vtk.vtkDoubleArray()
ident = id(vtk_arr)
vtk_arr = array_handler.array2vtk(z, vtk_arr)
# Make sure this is the same array!
self.assertEqual(ident, id(vtk_arr))
self._check_arrays(z, vtk_arr)
# Check the vtkBitArray.
vtk_arr = vtk.vtkBitArray()
vtk_arr.InsertNextValue(0)
vtk_arr.InsertNextValue(1)
vtk_arr.InsertNextValue(0)
vtk_arr.InsertNextValue(1)
arr = array_handler.vtk2array(vtk_arr)
self.assertEqual(numpy.sum(arr - [0, 1, 0, 1]), 0)
vtk_arr = array_handler.array2vtk(arr, vtk_arr)
self.assertEqual(vtk_arr.GetValue(0), 0)
self.assertEqual(vtk_arr.GetValue(1), 1)
self.assertEqual(vtk_arr.GetValue(2), 0)
self.assertEqual(vtk_arr.GetValue(3), 1)
# ----------------------------------------
# Test if the array is copied or not.
a = numpy.array([[1, 2, 3], [4, 5, 6]], 'd')
vtk_arr = array_handler.array2vtk(a)
# Change the numpy array and see if the changes are
# reflected in the VTK array.
a[0] = [10.0, 20.0, 30.0]
self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.))
# Make sure the cache is doing its job.
key = vtk_arr.__this__
z = array_handler._array_cache.get(vtk_arr)
self.assertEqual(numpy.sum(z - numpy.ravel(a)), 0.0)
l1 = len(array_handler._array_cache)
# del the Numeric array and see if this still works.
del a
self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.))
# Check the cache -- just making sure.
self.assertEqual(len(array_handler._array_cache), l1)
# Delete the VTK array and see if the cache is cleared.
del vtk_arr
self.assertEqual(len(array_handler._array_cache), l1 - 1)
self.assertEqual(key in array_handler._array_cache._cache, False)
# Make sure bit arrays are copied.
vtk_arr = vtk.vtkBitArray()
a = numpy.array([0, 1, 0, 1], numpy.int32)
vtk_arr = array_handler.array2vtk(a, vtk_arr)
del a
self.assertEqual(vtk_arr.GetValue(0), 0)
self.assertEqual(vtk_arr.GetValue(1), 1)
self.assertEqual(vtk_arr.GetValue(2), 0)
self.assertEqual(vtk_arr.GetValue(3), 1)
# Make sure the code at least runs for all the non-complex
# numerical dtypes in numpy.
for dtype in (numpy.sctypes['int'] + numpy.sctypes['uint'] +
numpy.sctypes['float']):
array_handler.array2vtk(numpy.zeros((1, ), dtype=dtype))
def getVar(data,var='rho'):
var=ah.vtk2array(data.GetCellData().GetArray(var))
return var
def getRho(data):
rho = ah.vtk2array(data.GetCellData().GetArray('rho'))
return rho
def getRho(data):
rho = (ah.vtk2array(data.GetCellData().GetArray('rho'))).astype(np.float64)
return rho