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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
|
"""A grid plane component.
"""
# Author: Prabhu Ramachandran <prabhu_r@users.sf.net>
# Copyright (c) 2005-2020, Enthought, Inc.
# License: BSD Style.
# Enthought library imports.
from traits.api import Instance, Int, Range
from traitsui.api import View, Group, Item
from tvtk.api import tvtk
# Local imports.
from mayavi.core.component import Component
from mayavi.core.common import error
######################################################################
# `CustomGridPlane` class.
######################################################################
class CustomGridPlane(Component):
# The version of this class. Used for persistence.
__version__ = 0
# The TVTK object that extracts the grid plane. This is created
# dynamically based on the input data type.
plane = Instance(tvtk.Object)
# Minimum x value.
x_min = Range(value=0, low='_x_low', high='_x_high',
enter_set=True, auto_set=False,
desc='minimum x value of the domain')
# Maximum x value.
x_max = Range(value=10000, low='_x_low', high='_x_high',
enter_set=True, auto_set=False,
desc='maximum x value of the domain')
# Minimum y value.
y_min = Range(value=0, low='_y_low', high='_y_high',
enter_set=True, auto_set=False,
desc='minimum y value of the domain')
# Maximum y value.
y_max = Range(value=10000, low='_y_low', high='_y_high',
enter_set=True, auto_set=False,
desc='maximum y value of the domain')
# Minimum z value.
z_min = Range(value=0, low='_z_low', high='_z_high',
enter_set=True, auto_set=False,
desc='minimum z value of the domain')
# Maximum z value.
z_max = Range(value=10000, low='_z_low', high='_z_high',
enter_set=True, auto_set=False,
desc='maximum z value of the domain')
########################################
# Private traits.
# Determines the lower/upper limit of the axes for the sliders.
_x_low = Int(0)
_x_high = Int(10000)
_y_low = Int(0)
_y_high = Int(10000)
_z_low = Int(0)
_z_high = Int(10000)
########################################
# View related traits.
# The View for this object.
view = View(Group(Item(name='x_min'),
Item(name='x_max'),
Item(name='y_min'),
Item(name='y_max'),
Item(name='z_min'),
Item(name='z_max'),
),
resizable=True
)
######################################################################
# `object` interface
######################################################################
def __get_pure_state__(self):
d = super(CustomGridPlane, self).__get_pure_state__()
# These traits are dynamically created.
for axis in ('x', 'y', 'z'):
for name in ('_min', '_max'):
d.pop(axis + name, None)
d.pop('_' + axis + '_low', None)
d.pop('_' + axis + '_high', None)
d.pop('plane', None)
return d
######################################################################
# `Component` interface
######################################################################
def update_pipeline(self):
"""Override this method so that it *updates* the tvtk pipeline
when data upstream is known to have changed.
This method is invoked (automatically) when the input fires a
`pipeline_changed` event.
"""
if len(self.inputs) == 0:
return
input = self.inputs[0].get_output_dataset()
plane = None
if input.is_a('vtkStructuredGrid'):
plane = tvtk.StructuredGridGeometryFilter()
elif input.is_a('vtkStructuredPoints') or input.is_a('vtkImageData'):
plane = tvtk.ImageDataGeometryFilter ()
elif input.is_a('vtkRectilinearGrid'):
plane = tvtk.RectilinearGridGeometryFilter ()
else:
msg = "The GridPlane component does not support the %s dataset."\
%(input.class_name)
error(msg)
raise TypeError(msg)
self.configure_connection(plane, self.inputs[0])
self.plane = plane
self._update_limits()
self._update_voi()
self.outputs = [plane]
def update_data(self):
"""Override this method to do what is necessary when upstream
data changes.
This method is invoked (automatically) when any of the inputs
sends a `data_changed` event.
"""
self._update_limits()
self._update_voi()
# Propagate the data_changed event.
self.data_changed = True
def has_output_port(self):
""" The contour filter has an output port."""
return True
def get_output_object(self):
""" Returns the output port."""
return self.plane.output_port
######################################################################
# Non-public methods.
######################################################################
def _update_limits(self):
extents = self.plane.input.whole_extent
self._x_low, self._x_high = extents[:2]
self._y_low, self._y_high = extents[2:4]
self._z_low, self._z_high = extents[4:]
def _x_min_changed(self, val):
if val > self.x_max:
self.x_max = val
else:
self._update_voi()
def _x_max_changed(self, val):
if val < self.x_min:
self.x_min = val
else:
self._update_voi()
def _y_min_changed(self, val):
if val > self.y_max:
self.y_max = val
else:
self._update_voi()
def _y_max_changed(self, val):
if val < self.y_min:
self.y_min = val
else:
self._update_voi()
def _z_min_changed(self, val):
if val > self.z_max:
self.z_max = val
else:
self._update_voi()
def _z_max_changed(self, val):
if val < self.z_min:
self.z_min = val
else:
self._update_voi()
def _update_voi(self):
if len(self.inputs) == 0:
return
plane = self.plane
extents = (self.x_min, self.x_max,
self.y_min, self.y_max,
self.z_min, self.z_max)
try:
plane.set_extent(extents)
except AttributeError:
plane.extent = tuple(extents)
plane.update_whole_extent()
plane.update()
self.data_changed = True
|
