examples/pyvisi/streamline.py


########################################################
#
# Copyright (c) 2003-2008 by University of Queensland
# Earth Systems Science Computational Center (ESSCC)
# http://www.uq.edu.au/esscc
#
# Primary Business: Queensland, Australia
# Licensed under the Open Software License version 3.0
# http://www.opensource.org/licenses/osl-3.0.php
#
########################################################

__copyright__="""Copyright (c) 2003-2008 by University of Queensland
Earth Systems Science Computational Center (ESSCC)
http://www.uq.edu.au/esscc
Primary Business: Queensland, Australia"""
__license__="""Licensed under the Open Software License version 3.0
http://www.opensource.org/licenses/osl-3.0.php"""
__url__="http://www.uq.edu.au/esscc/escript-finley"

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

# 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.setTubeRadiusToVaryByVector()
sl1.setPointSourceRadius(0.9)

# 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
2	########################################################
3	#
4	# Copyright (c) 2003-2008 by University of Queensland
5	# Earth Systems Science Computational Center (ESSCC)
6	# http://www.uq.edu.au/esscc
7	#
8	# Primary Business: Queensland, Australia
9	# Licensed under the Open Software License version 3.0
10	# http://www.opensource.org/licenses/osl-3.0.php
11	#
12	########################################################
13
14	__copyright__="""Copyright (c) 2003-2008 by University of Queensland
15	Earth Systems Science Computational Center (ESSCC)
16	http://www.uq.edu.au/esscc
17	Primary Business: Queensland, Australia"""
18	__license__="""Licensed under the Open Software License version 3.0
19	http://www.opensource.org/licenses/osl-3.0.php"""
20	__url__="http://www.uq.edu.au/esscc/escript-finley"
21
22	"""
23	Author: John Ngui, john.ngui@uq.edu.au
24	"""
25
26	# Import the necessary modules.
27	from esys.pyvisi import Scene, DataCollector, StreamLine, Camera
28	from esys.pyvisi.constant import *
29	import os
30
31	PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
32	PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
33	X_SIZE = 400
34	Y_SIZE = 400
35
36	VECTOR_FIELD_CELL_DATA = "temperature"
37	FILE_3D = "interior_3D.xml"
38	IMAGE_NAME = "streamline.jpg"
39	JPG_RENDERER = Renderer.ONLINE_JPG
40
41	# Create a Scene.
42	s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE,
43	y_size = Y_SIZE)
44
45	# Create a DataCollector reading from a XML file.
46	dc1 = DataCollector(source = Source.XML)
47	dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
48
49	# Create a Streamline.
50	sl1 = StreamLine(scene = s, data_collector = dc1,
51	viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.SCALAR,
52	lut = Lut.COLOR, cell_to_point = False, outline = True)
53	sl1.setTubeRadius(radius = 0.02)
54	sl1.setTubeNumberOfSides(3)
55	sl1.setTubeRadiusToVaryByVector()
56	sl1.setPointSourceRadius(0.9)
57
58	# Create a Camera.
59	c1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
60	c1.isometricView()
61
62	# Render the object.
63	s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))