I have a vtk version of vismach that offers a feature to show/hide certain objects. Maybe that helps to come up with an idea:
Attachment not found
github.com/Sigma1912/vtk-vismach
Model groups:
Each object of a model can optionally be assigned to a group. The assignment method can be called
on an existing object (2 lines of code)
Example 1: 'work_piece = Box(600,600,600)'
'work_piece.set_group('work')'
or the object can be created and the 'set_group' method called in one line.
Example 2: 'ReadPolyData('work_piece.stl', path_stl).set_group('work')'
Vismach will automatically create a checkbutton to show or hide each group used in the model.
Outline of my solution:
Each object in the model has a simple class variable 'group'.
When the window is created a checkbox is created for each group found in the model.
In each update cycle the objects with the matching 'group' variable value are shown or hidden according to the relevant checkbox state
Note that the vtk 'SetVisibility()' method did not seem to work for me so I use 'item.SetScale(0,0,0)' to hide an object:
Code:
class ModelGroups(object):
def __init__(self, model):
self.groups_in_model = {}
self.get_groups(model)
self.has_groups = len(self.groups_in_model.keys()) > 0
def get_groups(self, objects):
for item in objects.GetParts():
if hasattr(item, 'group'):
if item.group:
if item.group in self.groups_in_model.keys():
self.groups_in_model[item.group].append(item)
else:
self.groups_in_model[item.group] = [item]
if isinstance(item, vtk.vtkAssembly):
self.get_groups(item)
def update(self, checkboxes):
# SetVisibility does not seem to work on modifier objects so we use Scale
for group in self.groups_in_model.keys():
for checkbox in checkboxes:
if checkbox.objectName() == group:
if checkbox.isChecked():
for item in self.groups_in_model[group]:
item.restore_scale()
else:
for item in self.groups_in_model[group]:
item.store_scale()
item.SetScale(0,0,0)
