examples/pyvisi/map.py

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

PYVISI_EXAMPLE_MESHES_PATH = "data_meshes/"
PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images/"
X_SIZE = 800
Y_SIZE = 800

SCALAR_FIELD_POINT_DATA = "temperature"
SCALAR_FIELD_CELL_DATA = "temperature_cell"
FILE_3D = "interior_3D.xml"
IMAGE_NAME = "map.jpg"
JPG_RENDERER = Renderer.ONLINE_JPG


# Create a Scene with four viewports.
s = Scene(renderer = JPG_RENDERER, num_viewport = 4, 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 = PYVISI_EXAMPLE_MESHES_PATH + FILE_3D)
dc1.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)

# Create a  Map for the first viewport.
m1 = Map(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST, 
        lut = Lut.COLOR, cell_to_point = False, outline = True)
m1.setRepresentationToWireframe()

# Create a Camera for the first viewport
c1 = Camera(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST)
c1.isometricView()

# Create a second DataCollector reading from the same XML file but specifying
# a different scalar field.
dc2 = DataCollector(source = Source.XML)
dc2.setFileName(file_name = PYVISI_EXAMPLE_MESHES_PATH + FILE_3D)
dc2.setActiveScalar(scalar = SCALAR_FIELD_CELL_DATA)

# Create a Map for the third viewport.
m2 = Map(scene = s, data_collector = dc2, viewport = Viewport.NORTH_EAST, 
        lut = Lut.COLOR, cell_to_point = True, outline = True)

# Create a Camera for the third viewport
c1 = Camera(scene = s, data_collector = dc1, viewport = Viewport.NORTH_EAST)

# Render the object.
s.render(PYVISI_EXAMPLE_IMAGES_PATH + IMAGE_NAME)
1	# Import the necessary modules.
2	from esys.pyvisi import Scene, DataCollector, Map, Camera
3	from esys.pyvisi.constant import *
4
5	PYVISI_EXAMPLE_MESHES_PATH = "data_meshes/"
6	PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images/"
7	X_SIZE = 800
8	Y_SIZE = 800
9
10	SCALAR_FIELD_POINT_DATA = "temperature"
11	SCALAR_FIELD_CELL_DATA = "temperature_cell"
12	FILE_3D = "interior_3D.xml"
13	IMAGE_NAME = "map.jpg"
14	JPG_RENDERER = Renderer.ONLINE_JPG
15
16
17	# Create a Scene with four viewports.
18	s = Scene(renderer = JPG_RENDERER, num_viewport = 4, x_size = X_SIZE,
19	y_size = Y_SIZE)
20
21	# Create a DataCollector reading from a XML file.
22	dc1 = DataCollector(source = Source.XML)
23	dc1.setFileName(file_name = PYVISI_EXAMPLE_MESHES_PATH + FILE_3D)
24	dc1.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)
25
26	# Create a Map for the first viewport.
27	m1 = Map(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST,
28	lut = Lut.COLOR, cell_to_point = False, outline = True)
29	m1.setRepresentationToWireframe()
30
31	# Create a Camera for the first viewport
32	c1 = Camera(scene = s, data_collector = dc1, viewport = Viewport.SOUTH_WEST)
33	c1.isometricView()
34
35	# Create a second DataCollector reading from the same XML file but specifying
36	# a different scalar field.
37	dc2 = DataCollector(source = Source.XML)
38	dc2.setFileName(file_name = PYVISI_EXAMPLE_MESHES_PATH + FILE_3D)
39	dc2.setActiveScalar(scalar = SCALAR_FIELD_CELL_DATA)
40
41	# Create a Map for the third viewport.
42	m2 = Map(scene = s, data_collector = dc2, viewport = Viewport.NORTH_EAST,
43	lut = Lut.COLOR, cell_to_point = True, outline = True)
44
45	# Create a Camera for the third viewport
46	c1 = Camera(scene = s, data_collector = dc1, viewport = Viewport.NORTH_EAST)
47
48	# Render the object.
49	s.render(PYVISI_EXAMPLE_IMAGES_PATH + IMAGE_NAME)