multicomponents_3d.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	import numpy as np
	from uvw import RectilinearGrid, DataArray

	# Creating coordinates
	x = np.linspace(-0.5, 0.5, 10)
	y = np.linspace(-0.5, 0.5, 20)
	z = np.linspace(-0.9, 0.9, 30)

	# Creating the file
	grid = RectilinearGrid('grid.vtr', (x, y, z))

	# A centered ball
	x, y, z = np.meshgrid(x, y, z, indexing='ij')
	r = np.sqrt(x2 + y2 + z**2)
	ball = r < 0.3

	# Some multi-component multi-dimensional data
	data = np.zeros([10, 20, 30, 3, 3])
	data[ball, ...] = np.array([[0, 1, 0],
	[1, 0, 0],
	[0, 1, 1]])


	# Adding the point data (see help(DataArray) for more info)
	grid.addPointData(DataArray(data, range(3), 'data'))
	grid.write()