examples/pyvisi/movie.py

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