examples/pyvisi/streamline.py

# Import the necessary modules.
from esys.pyvisi import Scene, DataCollector, StreamLine, 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 = "temperature"
FILE_3D = "interior_3D.xml"
IMAGE_NAME = "streamline.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))

# Create a Streamline.
sl1 = StreamLine(scene = s, data_collector = dc1,
        viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.SCALAR, 
        lut = Lut.COLOR, cell_to_point = False, outline = True)
sl1.setTubeRadius(radius = 0.02)
sl1.setTubeNumberOfSides(3)
sl1.setTubeRadiusToVaryByScalar()

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

# Render the object.
s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))
1	ksteube	1147	# Import the necessary modules.
2			from esys.pyvisi import Scene, DataCollector, StreamLine, Camera
3			from esys.pyvisi.constant import *
4	jongui	1148	import os
5	ksteube	1147
6	jongui	1148	PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
7			PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
8	ksteube	1147	X_SIZE = 400
9			Y_SIZE = 400
10
11			VECTOR_FIELD_CELL_DATA = "temperature"
12			FILE_3D = "interior_3D.xml"
13			IMAGE_NAME = "streamline.jpg"
14			JPG_RENDERER = Renderer.ONLINE_JPG
15
16			# Create a Scene.
17			s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE,
18			y_size = Y_SIZE)
19
20			# Create a DataCollector reading from a XML file.
21			dc1 = DataCollector(source = Source.XML)
22	jongui	1148	dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
23	ksteube	1147
24			# Create a Streamline.
25			sl1 = StreamLine(scene = s, data_collector = dc1,
26			viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.SCALAR,
27			lut = Lut.COLOR, cell_to_point = False, outline = True)
28			sl1.setTubeRadius(radius = 0.02)
29	jongui	1158	sl1.setTubeNumberOfSides(3)
30			sl1.setTubeRadiusToVaryByScalar()
31	ksteube	1147
32			# Create a Camera.
33	jongui	1148	c1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
34	ksteube	1147	c1.isometricView()
35
36			# Render the object.
37	jongui	1148	s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))