examples/pyvisi/contour.py

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

# Import the necessary modules
from esys.pyvisi import Scene, DataCollector, Contour, 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

SCALAR_FIELD_POINT_DATA = "temperature"
FILE_3D = "interior_3D.xml"
IMAGE_NAME = "contour.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 a XML file.
dc1 = DataCollector(source = Source.XML)
dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
dc1.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)

# Create a Contour.
ctr1 = Contour(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST,
        lut = Lut.COLOR, cell_to_point = False, outline = True)
ctr1.generateContours(contours = 3)

# Create a Camera.
cam1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
cam1.elevation(angle = -40)

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

1	"""
2	Author: John Ngui, john.ngui@uq.edu.au
3	"""
4
5	# Import the necessary modules
6	from esys.pyvisi import Scene, DataCollector, Contour, Camera
7	from esys.pyvisi.constant import *
8	import os
9
10	PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
11	PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
12	X_SIZE = 400
13	Y_SIZE = 400
14
15	SCALAR_FIELD_POINT_DATA = "temperature"
16	FILE_3D = "interior_3D.xml"
17	IMAGE_NAME = "contour.jpg"
18	JPG_RENDERER = Renderer.ONLINE_JPG
19
20	# Create a Scene.
21	s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE,
22	y_size = Y_SIZE)
23
24	# Create a DataCollector reading a XML file.
25	dc1 = DataCollector(source = Source.XML)
26	dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, FILE_3D))
27	dc1.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)
28
29	# Create a Contour.
30	ctr1 = Contour(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST,
31	lut = Lut.COLOR, cell_to_point = False, outline = True)
32	ctr1.generateContours(contours = 3)
33
34	# Create a Camera.
35	cam1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
36	cam1.elevation(angle = -40)
37
38	# Render the object.
39	s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, IMAGE_NAME))
40