Переглянути джерело

Merge pull request #13266 from wkpark/xyz-prepare

xyz_grid: add prepare
AUTOMATIC1111 1 рік тому
батько
коміт
ee8e98711b
1 змінених файлів з 4 додано та 1 видалено
  1. 4 1
      scripts/xyz_grid.py

+ 4 - 1
scripts/xyz_grid.py

@@ -205,13 +205,14 @@ def csv_string_to_list_strip(data_str):
 
 
 
 
 class AxisOption:
 class AxisOption:
-    def __init__(self, label, type, apply, format_value=format_value_add_label, confirm=None, cost=0.0, choices=None):
+    def __init__(self, label, type, apply, format_value=format_value_add_label, confirm=None, cost=0.0, choices=None, prepare=None):
         self.label = label
         self.label = label
         self.type = type
         self.type = type
         self.apply = apply
         self.apply = apply
         self.format_value = format_value
         self.format_value = format_value
         self.confirm = confirm
         self.confirm = confirm
         self.cost = cost
         self.cost = cost
+        self.prepare = prepare
         self.choices = choices
         self.choices = choices
 
 
 
 
@@ -536,6 +537,8 @@ class Script(scripts.Script):
 
 
             if opt.choices is not None and not csv_mode:
             if opt.choices is not None and not csv_mode:
                 valslist = vals_dropdown
                 valslist = vals_dropdown
+            elif opt.prepare is not None:
+                valslist = opt.prepare(vals)
             else:
             else:
                 valslist = csv_string_to_list_strip(vals)
                 valslist = csv_string_to_list_strip(vals)