Skip to content

Commit 810a5f9

Browse files
committed
Rename for consistency patch(es) -> interactive_template(s)
Signed-off-by: Brianna Major <[email protected]>
1 parent 74abe67 commit 810a5f9

File tree

1 file changed

+40
-36
lines changed

1 file changed

+40
-36
lines changed

hexrd/ui/mask_regions_dialog.py

Lines changed: 40 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ def __init__(self, parent=None):
2525
self.axes = None
2626
self.bg_cache = None
2727
self.press = []
28-
self.added_patches = []
29-
self.patches = {}
28+
self.added_templates = []
29+
self.interactive_templates = {}
3030
self.canvas = parent.image_tab_widget.active_canvas
3131
self.image_mode = None
3232
self.raw_mask_coords = []
@@ -70,24 +70,25 @@ def setup_ui_connections(self):
7070
self.ui.undo.clicked.connect(self.undo_selection)
7171

7272
def update_undo_enable_state(self):
73-
enabled = bool(self.added_patches)
73+
enabled = bool(self.added_templates)
7474
self.ui.undo.setEnabled(enabled)
7575

7676
def select_shape(self):
7777
self.selection = self.ui.shape.currentText()
78-
self.patch = None
78+
self.interactive_template = None
7979

80-
def create_patch(self):
80+
def create_interactive_template(self):
8181
kwargs = {
8282
'fill': False,
8383
'animated': True,
8484
}
85-
self.patch = InteractiveTemplate(self.canvas, self.det)
86-
self.patch.create_polygon([[0,0]], **kwargs)
87-
self.patches.setdefault(self.det, []).append(self.patch)
88-
self.added_patches.append(self.det)
85+
self.interactive_template = InteractiveTemplate(self.canvas, self.det)
86+
self.interactive_template.create_polygon([[0,0]], **kwargs)
87+
self.interactive_templates.setdefault(self.det, []).append(
88+
self.interactive_template)
89+
self.added_templates.append(self.det)
8990

90-
def update_patch(self, event):
91+
def update_interactive_template(self, event):
9192
x0, y0 = self.press
9293
height = event.ydata - y0
9394
width = event.xdata - x0
@@ -99,24 +100,25 @@ def update_patch(self, event):
99100
verts = shape.get_patch_transform().transform(
100101
shape.get_path().vertices[:-1])
101102
verts = add_sample_points(verts, 300)
102-
self.patch.template.set_xy(verts)
103-
self.patch.center = self.patch.get_midpoint()
103+
self.interactive_template.template.set_xy(verts)
104+
self.interactive_template.center = (
105+
self.interactive_template.get_midpoint())
104106

105-
def discard_patch(self):
106-
det = self.added_patches.pop()
107+
def discard_interactive_template(self):
108+
det = self.added_templates.pop()
107109
# If not static mode then the raw coords haven't been saved yet
108-
if self.patch.static_mode:
110+
if self.interactive_template.static_mode:
109111
self.raw_mask_coords.pop()
110-
self.patches[det].pop().template.remove()
112+
self.interactive_templates[det].pop().template.remove()
111113

112114
def undo_selection(self):
113-
if not self.added_patches:
115+
if not self.added_templates:
114116
return
115117

116-
self.discard_patch()
118+
self.discard_interactive_template()
117119
self.canvas.draw_idle()
118120
self.update_undo_enable_state()
119-
self.patch.static_mode = True
121+
self.interactive_template.static_mode = True
120122

121123
def axes_entered(self, event):
122124
self.image_mode = self.canvas.mode
@@ -180,21 +182,22 @@ def button_pressed(self, event):
180182
print('Masking must be done in raw or polar view')
181183
return
182184

183-
if event.button == 3 and self.patch:
184-
self.patch.static_mode = True
185+
if event.button == 3 and self.interactive_template:
186+
self.interactive_template.static_mode = True
185187

186188
if not self.axes:
187189
return
188190

189191
if event.button == 1:
190-
if self.patch and not self.patch.static_mode:
192+
if (self.interactive_template and
193+
not self.interactive_template.static_mode):
191194
return
192195

193196
self.press = [event.xdata, event.ydata]
194197
self.det = self.axes.get_title()
195198
if not self.det:
196199
self.det = self.image_mode
197-
self.create_patch()
200+
self.create_interactive_template()
198201

199202
# For animating the patch
200203
self.bg_cache = self.canvas.copy_from_bbox(self.axes.bbox)
@@ -209,14 +212,14 @@ def drag_motion(self, event):
209212
):
210213
return
211214

212-
if not self.patch.static_mode:
215+
if not self.interactive_template.static_mode:
213216
return
214217

215-
self.update_patch(event)
218+
self.update_interactive_template(event)
216219

217220
# Update animation of patch
218221
self.canvas.restore_region(self.bg_cache)
219-
self.axes.draw_artist(self.patch.template)
222+
self.axes.draw_artist(self.interactive_template.template)
220223
self.canvas.blit(self.axes.bbox)
221224

222225
def save_line_data(self):
@@ -255,15 +258,16 @@ def create_masks(self):
255258
masks_changed_signal[self.image_mode].emit()
256259

257260
def button_released(self, event):
258-
if not self.press or not self.patch.static_mode:
261+
if not self.press or not self.interactive_template.static_mode:
259262
return
260263

261264
# Save it
262-
self.templates.setdefault(self.det, []).append(self.patch.template)
263-
self.patch.static_mode = False
265+
self.templates.setdefault(self.det, []).append(
266+
self.interactive_template.template)
267+
self.interactive_template.static_mode = False
264268

265269
# Turn off animation so the patch will stay
266-
self.patch.template.set_animated(False)
270+
self.interactive_template.template.set_animated(False)
267271

268272
self.press.clear()
269273
self.det = None
@@ -279,13 +283,13 @@ def apply_masks(self):
279283
self.save_line_data()
280284
self.disconnect()
281285
self.create_masks()
282-
while self.added_patches:
283-
self.discard_patch()
286+
while self.added_templates:
287+
self.discard_interactive_template()
284288
self.new_mask_added.emit(self.image_mode)
285289

286290
def cancel(self):
287-
while self.added_patches:
288-
self.discard_patch()
291+
while self.added_templates:
292+
self.discard_interactive_template()
289293

290294
self.disconnect()
291295
if self.canvas is not None:
@@ -294,7 +298,7 @@ def cancel(self):
294298
def canvas_changed(self, canvas):
295299
self.apply_masks()
296300
self.canvas = canvas
297-
if self.patch:
298-
self.patch.canvas_changed(canvas)
301+
if self.interactive_template:
302+
self.interactive_template.canvas_changed(canvas)
299303
if self.ui.isVisible():
300304
self.setup_canvas_connections()

0 commit comments

Comments
 (0)