yt_idv/examples/plane_textures_multiple.py at 8a4935efc03585f8044c15398ef81153e1803cd2 · yt-project/yt_idv · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import numpy as np
import yt

from yt_idv import render_context
from yt_idv.cameras.trackball_camera import TrackballCamera
from yt_idv.scene_components.plane import Plane  # NOQA
from yt_idv.scene_data.plane import PlaneData  # NOQA
from yt_idv.scene_graph import SceneGraph

rc = render_context(height=800, width=800, gui=True)
c = TrackballCamera(position=[3.5, 3.5, 3.5], focus=[0.0, 0.0, 0.0])
rc.scene = SceneGraph(camera=c)

ds = yt.load("Enzo_64/RD0005/RedshiftOutput0005")

# add a projection
proj = ds.proj(("enzo", "Density"), 1.0)
proj_data = PlaneData(data_source=proj)
proj_data.add_data(("enzo", "Density"), 1.0, (400, 400))
proj_render = Plane(data=proj_data, cmap_log=True)
rc.scene.data_objects.append(proj_data)
rc.scene.components.append(proj_render)

# add another projection
proj_2 = ds.proj(("enzo", "Density"), 0.0)
proj_data_2 = PlaneData(data_source=proj_2)
proj_data_2.add_data(("enzo", "Density"), 1.0, (400, 400))
proj_render_2 = Plane(data=proj_data_2, cmap_log=True)
rc.scene.data_objects.append(proj_data_2)
rc.scene.components.append(proj_render_2)

# add another projection, shifted to other axis
proj_3 = ds.proj(("enzo", "Density"), 1.0)
proj_data_3 = PlaneData(data_source=proj_3)
proj_data_3.add_data(("enzo", "Density"), 1.0, (400, 400), translate=-1.0)
proj_render_3 = Plane(data=proj_data_3, cmap_log=True)
rc.scene.data_objects.append(proj_data_3)
rc.scene.components.append(proj_render_3)

# add an axis-normal slice
slc = ds.slice(0, 0.5)
slice_data = PlaneData(data_source=slc)
slice_data.add_data(("enzo", "Density"), 1.0, (400, 400))
slice_render = Plane(data=slice_data, cmap_log=True)
rc.scene.data_objects.append(slice_data)
rc.scene.components.append(slice_render)

# add a cutting plane
normal = np.array([1.0, 1.0, 0.0], dtype="float64")
center = ds.domain_center.to("code_length").value
cut = ds.cutting(normal, center)
cut_data = PlaneData(data_source=cut)
cut_data.add_data(("enzo", "Density"), 1.0, (400, 400))
cut_render = Plane(data=cut_data, cmap_log=True)
rc.scene.data_objects.append(cut_data)
rc.scene.components.append(cut_render)

rc.run()