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 Velocity.
vopc1 = Velocity(scene = s, data_collector = dc1, 
        viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.VECTOR, 
        arrow = Arrow.TWO_D, lut = Lut.COLOR, cell_to_point = False, 
        outline = True)
vopc1.setScaleFactor(scale_factor = 0.07)
vopc1.setRatio(ratio = 8)
vopc1.setColor(color = Color.BLACK)

# Create a scalar Legend.
sb = Legend(scene = s, data_collector= dc1, viewport = Viewport.SOUTH_WEST,
        lut = Lut.COLOR, legend = LegendType.SCALAR)
sb.setOrientationToHorizontal()
sb.setPosition(LocalPosition(85,5))
sb.setScalarRange(0, 1) 
sb.setLabelColor(color = Color.BLACK)
sb.setTitleColor(Color.BLACK)
sb.setTitle("Temperature")

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

# Create a movie.
mov = Movie()

# 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)

# Generate the movie from the rendered images.
mov.makeMovie(input_directory = PYVISI_EXAMPLE_IMAGES_PATH, 
        first_image = IMAGE_NAME + "000938.jpg", 
        last_image = IMAGE_NAME + "000949.jpg", 
        movie = 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 Velocity.
31	vopc1 = Velocity(scene = s, data_collector = dc1,
32	viewport = Viewport.SOUTH_WEST, color_mode = ColorMode.VECTOR,
33	arrow = Arrow.TWO_D, lut = Lut.COLOR, cell_to_point = False,
34	outline = True)
35	vopc1.setScaleFactor(scale_factor = 0.07)
36	vopc1.setRatio(ratio = 8)
37	vopc1.setColor(color = Color.BLACK)
38
39	# Create a scalar Legend.
40	sb = Legend(scene = s, data_collector= dc1, viewport = Viewport.SOUTH_WEST,
41	lut = Lut.COLOR, legend = LegendType.SCALAR)
42	sb.setOrientationToHorizontal()
43	sb.setPosition(LocalPosition(85,5))
44	sb.setScalarRange(0, 1)
45	sb.setLabelColor(color = Color.BLACK)
46	sb.setTitleColor(Color.BLACK)
47	sb.setTitle("Temperature")
48
49	# Create a Camera.
50	cam1 = Camera(scene = s, viewport = Viewport.SOUTH_WEST)
51
52	# Create a movie.
53	mov = Movie()
54
55	# Read in one file one after another and render the object.
56	for i in range(938, 949):
57	dc1.setFileName(file_name = os.path.join(PYVISI_EXAMPLE_MESHES_PATH, \
58	FILE_2D + "%06d.vtu") % i)
59
60	s.render(image_name = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, \
61	IMAGE_NAME + "%06d.jpg") % i)
62
63	# Generate the movie from the rendered images.
64	mov.makeMovie(input_directory = PYVISI_EXAMPLE_IMAGES_PATH,
65	first_image = IMAGE_NAME + "000938.jpg",
66	last_image = IMAGE_NAME + "000949.jpg",
67	movie = os.path.join(PYVISI_EXAMPLE_IMAGES_PATH, "movie.mpg"))
68