examples/pyvisi/movie.py

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

# Import the necessary modules.
from esys.pyvisi import Scene, DataCollector, Map, Camera, Velocity, Legend 
from esys.pyvisi import Movie, LocalPosition
from esys.pyvisi.constant import *
import os

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

SCALAR_FIELD_POINT_DATA = "temp"
FILE_2D = "tempvel-"
IMAGE_NAME = "movie"
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.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)

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

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

# Create a movie.
mov = Movie()
#lst = []

# Read in one file one after another and render the object. 
for i in range(938, 949):
    dc1.setFileName(file_name =  os.path.join(PYVISI_EXAMPLE_MESHES_PATH, \
            FILE_2D + "%06d.vtu") % i)

    s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, \
            IMAGE_NAME + "%06d.jpg") % i)

    #lst.append(IMAGE_NAME + "%06d.jpg" % i)

# Images (first and last inclusive) from which the movie is to be generated.
mov.imageRange(input_directory = PYVISI_EXAMPLE_IMAGES_PATH, 
        first_image = IMAGE_NAME + "000938.jpg", 
        last_image = IMAGE_NAME + "000948.jpg")

# Alternatively, a list of images can be specified.
#mov.imageList(input_directory = PYVISI_EXAMPLE_IMAGES_PATH, image_list = lst)

# Generate the movie.
mov.makeMovie(os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, "movie.mpg"))

1	jongui	1203	"""
2			Author: John Ngui, john.ngui@uq.edu.au
3			"""
4
5	jongui	1191	# Import the necessary modules.
6			from esys.pyvisi import Scene, DataCollector, Map, Camera, Velocity, Legend
7			from esys.pyvisi import Movie, LocalPosition
8			from esys.pyvisi.constant import *
9			import os
10
11			PYVISI_EXAMPLE_MESHES_PATH = "data_meshes"
12			PYVISI_EXAMPLE_IMAGES_PATH = "data_sample_images"
13			X_SIZE = 800
14			Y_SIZE = 800
15
16			SCALAR_FIELD_POINT_DATA = "temp"
17			FILE_2D = "tempvel-"
18			IMAGE_NAME = "movie"
19			JPG_RENDERER = Renderer.ONLINE_JPG
20
21			# Create a Scene.
22			s = Scene(renderer = JPG_RENDERER, num_viewport = 1, x_size = X_SIZE,
23			y_size = Y_SIZE)
24
25			# Create a DataCollector reading from a XML file.
26			dc1 = DataCollector(source = Source.XML)
27			dc1.setActiveScalar(scalar = SCALAR_FIELD_POINT_DATA)
28
29			# Create a Map.
30			m1 = Map(scene = s, data_collector = dc1,
31			viewport = Viewport.SOUTH_WEST, lut = Lut.COLOR, cell_to_point = False,
32			outline = True)
33
34			# Create a Camera.
35			cam1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
36
37			# Create a movie.
38	jongui	1202	mov = Movie()
39	jongui	1197	#lst = []
40	jongui	1191
41			# Read in one file one after another and render the object.
42			for i in range(938, 949):
43			dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, \
44			FILE_2D + "%06d.vtu") % i)
45
46			s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, \
47			IMAGE_NAME + "%06d.jpg") % i)
48
49	jongui	1197	#lst.append(IMAGE_NAME + "%06d.jpg" % i)
50
51			# Images (first and last inclusive) from which the movie is to be generated.
52			mov.imageRange(input_directory = PYVISI_EXAMPLE_IMAGES_PATH,
53	jongui	1191	first_image = IMAGE_NAME + "000938.jpg",
54	jongui	1197	last_image = IMAGE_NAME + "000948.jpg")
55	jongui	1191
56	jongui	1197	# Alternatively, a list of images can be specified.
57			#mov.imageList(input_directory = PYVISI_EXAMPLE_IMAGES_PATH, image_list = lst)
58
59			# Generate the movie.
60			mov.makeMovie(os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, "movie.mpg"))
61