multicomponents_3d.py
e12fd2f4762emaster

multicomponents_3d.py
View Options

			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()