examples/pyvisi/velocityonplanecut.py

"""
Author: John Ngui, john.ngui@uq.edu.au
"""

# Import the necessary modules
from esys.pyvisi import Scene, DataCollector, VelocityOnPlaneCut, Camera
from esys.pyvisi.constant import *
import os

PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
X_SIZE = 400
Y_SIZE = 400

VECTOR_FIELD_CELL_DATA = "velocity"
FILE_3D = "interior_3D.xml"
IMAGE_NAME = "velocity.jpg"
JPG_RENDERER = Renderer.ONLINE_JPG

# Create a Scene.
s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE, 
        y_size = Y_SIZE)

# Create a DataCollector reading from a XML file.
dc1 = DataCollector(source = Source.XML)
dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
dc1.setActiveVector(vector = VECTOR_FIELD_CELL_DATA)

# Create VelocityOnPlaneCut.
vopc1 = VelocityOnPlaneCut(scene = s, data_collector = dc1, 
        viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.VECTOR, 
        arrow = Arrow.THREE_D, lut = Lut.COLOR, cell_to_point = False, 
        outline = True)
vopc1.setScaleFactor(scale_factor = 0.5)
vopc1.setPlaneToXY(offset = 0.5)
vopc1.setRatio(2)
vopc1.randomOn()

# Create a Camera.
c1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
c1.isometricView()
c1.elevation(angle = -20)

# Render the object.
s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))

1	jongui	1203	"""
2			Author: John Ngui, john.ngui@uq.edu.au
3			"""
4
5	ksteube	1147	# Import the necessary modules
6			from esys.pyvisi import Scene, DataCollector, VelocityOnPlaneCut, Camera
7			from esys.pyvisi.constant import *
8	jongui	1148	import os
9	ksteube	1147
10	jongui	1148	PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
11			PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
12	ksteube	1147	X_SIZE = 400
13			Y_SIZE = 400
14
15			VECTOR_FIELD_CELL_DATA = "velocity"
16			FILE_3D = "interior_3D.xml"
17			IMAGE_NAME = "velocity.jpg"
18			JPG_RENDERER = Renderer.ONLINE_JPG
19
20	jongui	1210	# Create a Scene.
21	ksteube	1147	s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE,
22			y_size = Y_SIZE)
23
24			# Create a DataCollector reading from a XML file.
25			dc1 = DataCollector(source = Source.XML)
26	jongui	1148	dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
27	ksteube	1147	dc1.setActiveVector(vector = VECTOR_FIELD_CELL_DATA)
28
29			# Create VelocityOnPlaneCut.
30			vopc1 = VelocityOnPlaneCut(scene = s, data_collector = dc1,
31			viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.VECTOR,
32			arrow = Arrow.THREE_D, lut = Lut.COLOR, cell_to_point = False,
33			outline = True)
34			vopc1.setScaleFactor(scale_factor = 0.5)
35			vopc1.setPlaneToXY(offset = 0.5)
36			vopc1.setRatio(2)
37			vopc1.randomOn()
38
39			# Create a Camera.
40	jongui	1148	c1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
41	ksteube	1147	c1.isometricView()
42			c1.elevation(angle = -20)
43
44			# Render the object.
45	jongui	1148	s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))
46	ksteube	1147