feat: implement type-aware auto-completion utilities and variable frame for smartprop editor

Author: dertwist

src/editors/smartprop_editor/completion_utils.py

@@ -61,6 +61,7 @@ def get_combobox_elements(var_type):
             'RadiusPlacementMode': ['SPHERE', 'CIRCLE'],
             'DistributionMode': ['RANDOM', 'REGULAR'],
             'PathPositions': ['ALL', 'NTH', 'START_AND_END', 'CONTROL_POINTS'],
+            'Direction': ['RANDOM', 'REGULAR'],
         }
         return elements_dict.get(var_type, [])
 
@@ -208,7 +209,7 @@ def _generate_type_specific_completions(var_name, var_type, context):
         # Enum-type variable completions (combobox types)
         elif var_type in ['CoordinateSpace', 'GridPlacementMode', 'GridOriginMode', 'PickMode', 
                          'ScaleMode', 'TraceNoHit', 'ApplyColorMode', 'ChoiceSelectionMode', 
-                         'RadiusPlacementMode', 'DistributionMode', 'PathPositions']:
+                         'RadiusPlacementMode', 'DistributionMode', 'PathPositions', 'Direction']:
             # Get the possible values for this enum type
             enum_values = CompletionUtils.get_combobox_elements(var_type)
             for enum_value in enum_values:

src/editors/smartprop_editor/variable_frame.py

@@ -103,6 +103,7 @@ def __init__(self, name, var_class, var_value, var_visible_in_editor, var_displa
         self.ui.show_child.clicked.connect(self.show_child)
 
         self.init_ui()
+        self.update_colors()
 
     def _initialize_var_instance(self, var_class):
         if var_class == 'Int':
@@ -137,7 +138,7 @@ def _initialize_var_instance(self, var_class):
             )
         elif var_class in ['CoordinateSpace', 'GridPlacementMode', 'GridOriginMode', 'PickMode', 'ScaleMode',
                            'TraceNoHit', 'ApplyColorMode', 'ChoiceSelectionMode', 'RadiusPlacementMode',
-                           'DistributionMode', 'PathPositions']:
+                           'DistributionMode', 'PathPositions', 'Direction']:
             from src.editors.smartprop_editor.variables.combobox import Var_class_combobox
             elements = self._get_combobox_elements(var_class)
             self.var_int_instance = Var_class_combobox(
@@ -161,6 +162,40 @@ def _initialize_var_instance(self, var_class):
     def _get_combobox_elements(self, var_class):
         return CompletionUtils.get_combobox_elements(var_class)
 
+    def update_colors(self):
+        color_map = {
+            'String': '#E67E22',
+            'Model': '#E67E22',
+            'Material': '#E67E22',
+            'MaterialGroup': '#E67E22',
+            'Bool': '#C0392B',
+            'Float': '#2980B9',
+            'Int': '#2471A3',
+            'Vector2D': '#8E44AD',
+            'Vector3D': '#8E44AD',
+            'Vector4D': '#8E44AD',
+            'Angles': '#8E44AD',
+            'Color': '#1B5E20',
+        }
+
+        enum_types = [
+            'Direction', 'CoordinateSpace', 'GridPlacementMode', 'GridOriginMode',
+            'PickMode', 'ScaleMode', 'TraceNoHit', 'ApplyColorMode',
+            'ChoiceSelectionMode', 'RadiusPlacementMode', 'DistributionMode', 'PathPositions'
+        ]
+
+        target_color = color_map.get(self.var_class, '#242424')
+        if self.var_class in enum_types:
+            target_color = '#1D8348'
+
+        style = self.ui.label.styleSheet()
+        import re
+        if 'background-color:' in style:
+            style = re.sub(r'background-color:\s*[^;]+;', f'background-color: {target_color};', style)
+        else:
+            style += f'\nbackground-color: {target_color};'
+        self.ui.label.setStyleSheet(style)
+
     def _setup_hide_expression_completer(self):
         """Setup completer for hide expression input with filtered type-aware completions."""
         # For hide expressions, only show Bool, Int, and Float variables
@@ -216,6 +251,7 @@ def set_class(self, var_class):
         self.var_int_instance.edited.connect(self.on_changed)
         self.ui.layout.insertWidget(2, self.var_int_instance)
         # Emit content_changed directly (pre_change was already emitted above)
+        self.update_colors()
         self.content_changed.emit()
 
     def get_classes(self):