Advanced Search

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)