Commit 64f82741323b39609d9196fb2b0f8a4bfec61548
1 parent
a1115f29
Exists in
master
and in
68 other branches
ENH: Visibility of surfaces generated by seeds using connectivity
Showing
3 changed files
with
75 additions
and
9 deletions
Show diff stats
invesalius/data/polydata_utils.py
... | ... | @@ -173,7 +173,9 @@ def SplitDisconectedParts(polydata): |
173 | 173 | polydata_collection = [] |
174 | 174 | |
175 | 175 | # Update progress value in GUI |
176 | - UpdateProgress = vu.ShowProgress(nregions - 1) | |
176 | + progress = nregions -1 | |
177 | + if progress: | |
178 | + UpdateProgress = vu.ShowProgress(progress) | |
177 | 179 | |
178 | 180 | for region in xrange(nregions): |
179 | 181 | conn.InitializeSpecifiedRegionList() |
... | ... | @@ -185,6 +187,7 @@ def SplitDisconectedParts(polydata): |
185 | 187 | p.Update() |
186 | 188 | |
187 | 189 | polydata_collection.append(p) |
188 | - UpdateProgress(region, _("Splitting disconected parts")) | |
190 | + if progress: | |
191 | + UpdateProgress(region, _("Splitting disconected parts")) | |
189 | 192 | |
190 | 193 | return polydata_collection | ... | ... |
invesalius/data/surface.py
... | ... | @@ -126,6 +126,10 @@ class SurfaceManager(): |
126 | 126 | ps.Publisher().subscribe(self.OnSeedSurface, "Create surface from seeds") |
127 | 127 | |
128 | 128 | def OnSeedSurface(self, pubsub_evt): |
129 | + """ | |
130 | + Create a new surface, based on the last selected surface, | |
131 | + using as reference seeds user add to surface of reference. | |
132 | + """ | |
129 | 133 | points_id_list = pubsub_evt.data |
130 | 134 | index = self.last_surface_index |
131 | 135 | proj = prj.Project() |
... | ... | @@ -133,27 +137,42 @@ class SurfaceManager(): |
133 | 137 | |
134 | 138 | new_polydata = pu.JoinSeedsParts(surface.polydata, |
135 | 139 | points_id_list) |
136 | - self.CreateSurfaceFromPolydata(new_polydata) | |
137 | - self.ShowActor(index, False) | |
140 | + index = self.CreateSurfaceFromPolydata(new_polydata) | |
141 | + ps.Publisher().sendMessage('Show single surface', (index, True)) | |
142 | + #self.ShowActor(index, True) | |
143 | + | |
138 | 144 | |
139 | 145 | def OnSplitSurface(self, pubsub_evt): |
146 | + """ | |
147 | + Create n new surfaces, based on the last selected surface, | |
148 | + according to their connectivity. | |
149 | + """ | |
140 | 150 | index = self.last_surface_index |
141 | 151 | proj = prj.Project() |
142 | 152 | surface = proj.surface_dict[index] |
143 | 153 | |
154 | + index_list = [] | |
144 | 155 | new_polydata_list = pu.SplitDisconectedParts(surface.polydata) |
145 | 156 | for polydata in new_polydata_list: |
146 | - self.CreateSurfaceFromPolydata(polydata) | |
147 | - self.ShowActor(index, False) | |
157 | + index = self.CreateSurfaceFromPolydata(polydata) | |
158 | + index_list.append(index) | |
159 | + #self.ShowActor(index, True) | |
160 | + | |
161 | + ps.Publisher().sendMessage('Show multiple surfaces', (index_list, True)) | |
162 | + | |
148 | 163 | |
149 | 164 | def OnLargestSurface(self, pubsub_evt): |
165 | + """ | |
166 | + Create a new surface, based on largest part of the last | |
167 | + selected surface. | |
168 | + """ | |
150 | 169 | index = self.last_surface_index |
151 | 170 | proj = prj.Project() |
152 | 171 | surface = proj.surface_dict[index] |
153 | 172 | |
154 | 173 | new_polydata = pu.SelectLargestPart(surface.polydata) |
155 | - self.CreateSurfaceFromPolydata(new_polydata) | |
156 | - self.ShowActor(index, False) | |
174 | + new_index = self.CreateSurfaceFromPolydata(new_polydata) | |
175 | + ps.Publisher().sendMessage('Show single surface', (new_index, True)) | |
157 | 176 | |
158 | 177 | def CreateSurfaceFromPolydata(self, polydata, overwrite=False): |
159 | 178 | |
... | ... | @@ -208,6 +227,8 @@ class SurfaceManager(): |
208 | 227 | surface.colour, surface.volume, |
209 | 228 | surface.transparency)) |
210 | 229 | |
230 | + return surface.index | |
231 | + | |
211 | 232 | |
212 | 233 | def OnCloseProject(self, pubsub_evt): |
213 | 234 | self.CloseProject() |
... | ... | @@ -481,7 +502,6 @@ class SurfaceManager(): |
481 | 502 | |
482 | 503 | def OnShowSurface(self, pubsub_evt): |
483 | 504 | index, value = pubsub_evt.data |
484 | - #print "OnShowSurface", index, value | |
485 | 505 | self.ShowActor(index, value) |
486 | 506 | |
487 | 507 | def ShowActor(self, index, value): | ... | ... |
invesalius/gui/data_notebook.py
... | ... | @@ -341,6 +341,11 @@ class MasksListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin): |
341 | 341 | if i > index: |
342 | 342 | new_dict[i-1] = old_dict[i] |
343 | 343 | self.mask_list_index = new_dict |
344 | + | |
345 | + if new_dict: | |
346 | + self.SetItemImage(0, 1) | |
347 | + ps.Publisher().sendMessage('Show mask', (0, 1)) | |
348 | + | |
344 | 349 | self.DeleteItem(index) |
345 | 350 | |
346 | 351 | #------------------------------------------------- |
... | ... | @@ -372,6 +377,10 @@ class SurfacesListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin): |
372 | 377 | 'Set surface colour') |
373 | 378 | ps.Publisher().subscribe(self.OnCloseProject, 'Close project data') |
374 | 379 | |
380 | + | |
381 | + ps.Publisher().subscribe(self.OnShowSingle, 'Show single surface') | |
382 | + ps.Publisher().subscribe(self.OnShowMultiple, 'Show multiple surfaces') | |
383 | + | |
375 | 384 | def __bind_events_wx(self): |
376 | 385 | self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnItemActivated) |
377 | 386 | self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEditLabel) |
... | ... | @@ -436,6 +445,40 @@ class SurfacesListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin): |
436 | 445 | def OnCheckItem(self, index, flag): |
437 | 446 | ps.Publisher().sendMessage('Show surface', (index, flag)) |
438 | 447 | |
448 | + def OnShowSingle(self, pubsub_evt): | |
449 | + index, visibility = pubsub_evt.data | |
450 | + print "----------------------" | |
451 | + print "OnShowSingle" | |
452 | + print "index", index | |
453 | + print "visibility", visibility | |
454 | + print "----------------------" | |
455 | + for key in self.surface_list_index.keys(): | |
456 | + if key != index: | |
457 | + self.SetItemImage(key, not visibility) | |
458 | + ps.Publisher().sendMessage('Show surface', | |
459 | + (key, not visibility)) | |
460 | + self.SetItemImage(index, visibility) | |
461 | + ps.Publisher().sendMessage('Show surface', | |
462 | + (index, visibility)) | |
463 | + | |
464 | + def OnShowMultiple(self, pubsub_evt): | |
465 | + index_list, visibility = pubsub_evt.data | |
466 | + print "----------------------" | |
467 | + print "OnShowMultiple" | |
468 | + print "index", index_list | |
469 | + print "visibility", visibility | |
470 | + print "----------------------" | |
471 | + | |
472 | + for key in self.surface_list_index.keys(): | |
473 | + if key not in index_list: | |
474 | + self.SetItemImage(key, not visibility) | |
475 | + ps.Publisher().sendMessage('Show surface', | |
476 | + (key, not visibility)) | |
477 | + for index in index_list: | |
478 | + self.SetItemImage(index, visibility) | |
479 | + ps.Publisher().sendMessage('Show surface', | |
480 | + (index, visibility)) | |
481 | + | |
439 | 482 | def AddSurface(self, pubsub_evt): |
440 | 483 | |
441 | 484 | ... | ... |