Ryven udates

PythonOCC/gui.py

@@ -0,0 +1,254 @@
+from ryven.gui_env import *
+
+from qtpy.QtCore import Signal
+from qtpy.QtWidgets import QLineEdit
+
+
+# 
+# General Base Classes
+# 
+
+
+class PythonOCCNodeGuiBase(NodeGUI):
+
+    input_widget_classes = {
+        'DataSmall': inp_widgets.Builder.evaled_line_edit(size='s', resizing=True),
+    }
+
+    @staticmethod
+    def builder():
+        return GuiBuilder
+
+
+class GuiBuilder:
+
+    @staticmethod
+    def attach_input_widgets(input_widgets):
+        """
+        A decorator which automatically attaches input widgets specified
+        in the input_widgets list to the gui of the node.
+        """
+
+        # make sure all input widget names provided are valid
+        assert all([
+            wn in PythonOCCNodeGuiBase.input_widget_classes 
+            for wn in input_widgets if wn is not None
+        ])
+
+        def decorator(cls):
+            inp_widgets = {
+                i: {'name': widget_name, 'pos': 'besides'}
+                for i, widget_name in enumerate(input_widgets)
+                if widget_name is not None
+            }
+            if not isinstance(cls.GUI, PythonOCCNodeGuiBase):
+                # override GUI class
+                class AutoInputWidgets_NodeGui(PythonOCCNodeGuiBase):
+                    init_input_widgets = inp_widgets
+                cls.GUI = AutoInputWidgets_NodeGui
+            else:
+                # add input widgets to existing GUI class
+                cls.GUI.init_input_widgets = inp_widgets
+            return cls
+
+        return decorator
+
+
+# 
+# Specific Node GUIs
+# 
+
+
+class PyOCCBase_DynamicImportsGui(PythonOCCNodeGuiBase):
+    def __init__(self, params):
+        super().__init__(params)
+
+        self.num_inputs = 0
+        self.actions['add input'] = {'method': self.add_operand_input}
+        self.actions['remove input'] = {}
+
+    def initialized(self):
+        # if the node was loaded from a state, catch up with 
+        # the number of inputs
+        self.num_inputs = len(self.node.inputs)
+        # the actions are restrored automatically
+
+        # for debugging only
+        self.actions['remove input'] = {
+            i: {
+                'method': self.remove_operand_input,
+                'data': i,
+            } for i in range(self.num_inputs)
+        }
+
+    def add_operand_input(self):
+        self.node.add_operand_input()
+
+        # update actions
+        new_index = self.num_inputs
+        self.actions['remove input'][new_index] = {
+            'method': self.remove_operand_input,
+            'data': new_index,
+        }
+        self.num_inputs += 1
+
+    def remove_operand_input(self, index):
+        self.node.remove_operand_input(index)
+        del self.actions['remove input'][self.num_inputs-1]
+        self.num_inputs -= 1
+
+
+class GpNodeGui(PythonOCCNodeGuiBase):
+    color = '#5e0a91'
+
+
+class BrepBuilderAPINodeGui(PythonOCCNodeGuiBase):
+    color = '#DAA520'
+
+
+class BrepOffsetAPINodeGui(PythonOCCNodeGuiBase):
+    color = '#aabb44'
+
+
+class BrepPrimAPINodeBase(PythonOCCNodeGuiBase):
+    color = '#aabb44'
+
+
+class BrepAlgoAPINodeGui(PythonOCCNodeGuiBase):
+    color = '#ab0c36'
+
+
+class BrepFilletAPINodeGui(PythonOCCNodeGuiBase):
+    color = '#e0149c'
+
+
+class GeomNodeGui(PythonOCCNodeGuiBase):
+    color = '#c91604'
+
+
+class GeomAPINodeGui(PythonOCCNodeGuiBase):
+    color = '#ff4633'
+
+
+class TopExplorerGui(PythonOCCNodeGuiBase):
+    color = '#FF00FF'
+
+
+class BoundingBoxGui(PythonOCCNodeGuiBase):
+    color = '#FF00FF'
+
+
+class DisplayNodeGui(PythonOCCNodeGuiBase):
+    color = '#3355dd'
+
+
+class ListGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class ListLengthGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class FlattenListGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class ListItemGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class RepeatDataGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class SerieGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class ShiftListGui(PythonOCCNodeGuiBase):
+    color = '#000000'
+
+
+class DataExchangeNodeGui(PythonOCCNodeGuiBase):
+    color = '#6b6767'
+
+
+# 
+# inport file node
+# 
+
+
+class ImportFileNode_MainWidget(NodeMainWidget, QLineEdit):
+
+    value_changed = Signal(object)
+
+    def __init__(self, params):
+        NodeMainWidget.__init__(self, params)
+        QLineEdit.__init__(self)
+
+        # self.setFixedWidth(80)
+        # self.setMinimumWidth(80)
+        self.resize(120, 31)
+        self.editingFinished.connect(self.editing_finished)
+
+    def editing_finished(self):
+        # self.node.update()
+        self.value_changed.emit(self.get_val())
+
+    def get_val(self):
+        val = None
+        try:
+            val = eval(self.text())
+        except Exception as e:
+            val = self.text()
+        return val
+
+    def get_state(self):
+        data = {'text': self.text()}
+        return data
+
+    def set_state(self, data):
+        self.setText(data['text'])
+
+
+class ImportFileNode_Gui(NodeGUI):
+    main_widget_class = ImportFileNode_MainWidget
+    main_widget_pos = 'between ports'
+
+
+# ---------------------------------------------------------------------------------------------------------------------------------
+
+
+# 
+# Export
+# 
+
+
+export_guis([
+    PythonOCCNodeGuiBase,
+    PyOCCBase_DynamicImportsGui,
+    GuiBuilder,
+
+    GpNodeGui,
+    BrepBuilderAPINodeGui,
+    BrepOffsetAPINodeGui,
+    BrepPrimAPINodeBase,
+    BrepAlgoAPINodeGui,
+    BrepFilletAPINodeGui,
+    GeomNodeGui,
+    GeomAPINodeGui,
+    TopExplorerGui,
+    BoundingBoxGui,
+    DisplayNodeGui,
+    ListGui,
+    ListLengthGui,
+    FlattenListGui,
+    ListItemGui,
+    RepeatDataGui,
+    SerieGui,
+    ShiftListGui,
+    DataExchangeNodeGui,
+
+    ImportFileNode_Gui,
+])