1 files changed, 165 insertions, 0 deletions
diff --git a/space_view3d_brush_menus/utils_core.py b/space_view3d_brush_menus/utils_core.py
new file mode 100644
index 00000000..ecac6073
--- /dev/null
+++ b/space_view3d_brush_menus/utils_core.py
@@ -0,0 +1,165 @@
+# gpl author: Ryan Inch (Imaginer)
+
+import bpy
+
+object_mode = 'OBJECT'
+edit = 'EDIT'
+sculpt = 'SCULPT'
+vertex_paint = 'VERTEX_PAINT'
+weight_paint = 'WEIGHT_PAINT'
+texture_paint = 'TEXTURE_PAINT'
+particle_edit = 'PARTICLE_EDIT'
+pose = 'POSE'
+gpencil_edit = 'GPENCIL_EDIT'
+get_addon_name = 'space_view3d_brush_menus'
+
+PIW = '       '
+
+
+# check for (currently) brushes being linked
+def get_brush_link(context, types="brush"):
+    tool_settings = context.tool_settings
+    has_brush = None
+
+    if get_mode() == sculpt:
+        datapath = tool_settings.sculpt
+
+    elif get_mode() == vertex_paint:
+        datapath = tool_settings.vertex_paint
+
+    elif get_mode() == weight_paint:
+        datapath = tool_settings.weight_paint
+
+    elif get_mode() == texture_paint:
+        datapath = tool_settings.image_paint
+    else:
+        datapath = None
+
+    if types == "brush":
+        has_brush = getattr(datapath, "brush", None)
+
+    return has_brush
+
+
+# Addon settings
+def addon_settings(lists=True):
+    # separate function just for more convience
+    addon = bpy.context.user_preferences.addons[get_addon_name]
+    colum_n = addon.preferences.column_set if addon else 1
+    use_list = addon.preferences.use_brushes_menu_type
+
+    return use_list if lists else colum_n
+
+
+def error_handlers(self, op_name, error, reports="ERROR", func=False):
+    if self and reports:
+        self.report({'WARNING'}, reports + " (See Console for more info)")
+
+    is_func = "Function" if func else "Operator"
+    print("\n[Sculpt/Paint Brush Menus]\n{}: {}\nError: {}\n".format(is_func, op_name, error))
+
+
+class Menu():
+    def __init__(self, menu):
+        self.layout = menu.layout
+        self.items = {}
+        self.current_item = None
+
+    def add_item(self, ui_type="row", parent=None, name=None, **kwargs):
+        # set the parent layout
+        if parent:
+            layout = parent
+        else:
+            layout = self.layout
+
+        # set unique identifier for new items
+        if not name:
+            name = len(self.items) + 1
+
+        # create and return a ui layout
+        if ui_type == "row":
+            self.current_item = self.items[name] = layout.row(**kwargs)
+
+            return self.current_item
+
+        elif ui_type == "column":
+            self.current_item = self.items[name] = layout.column(**kwargs)
+
+            return self.current_item
+
+        elif ui_type == "column_flow":
+            self.current_item = self.items[name] = layout.column_flow(**kwargs)
+
+            return self.current_item
+
+        elif ui_type == "box":
+            self.current_item = self.items[name] = layout.box(**kwargs)
+
+            return self.current_item
+
+        elif ui_type == "split":
+            self.current_item = self.items[name] = layout.split(**kwargs)
+
+            return self.current_item
+        else:
+            print("Unknown Type")
+
+
+def get_mode():
+    try:
+        if bpy.context.gpencil_data and \
+        bpy.context.object.mode == object_mode and \
+        bpy.context.scene.grease_pencil.use_stroke_edit_mode:
+            return gpencil_edit
+        else:
+            return bpy.context.object.mode
+    except:
+        return None
+
+
+def menuprop(item, name, value, data_path,
+             icon='NONE', disable=False, disable_icon=None,
+             custom_disable_exp=None, method=None, path=False):
+
+    # disable the ui
+    if disable:
+        disabled = False
+
+        # used if you need a custom expression to disable the ui
+        if custom_disable_exp:
+            if custom_disable_exp[0] == custom_disable_exp[1]:
+                item.enabled = False
+                disabled = True
+
+        # check if the ui should be disabled for numbers
+        elif isinstance(eval("bpy.context.{}".format(data_path)), float):
+            if round(eval("bpy.context.{}".format(data_path)), 2) == value:
+                item.enabled = False
+                disabled = True
+
+        # check if the ui should be disabled for anything else
+        else:
+            if eval("bpy.context.{}".format(data_path)) == value:
+                item.enabled = False
+                disabled = True
+
+        # change the icon to the disable_icon if the ui has been disabled
+        if disable_icon and disabled:
+            icon = disable_icon
+
+    # creates the menu item
+    prop = item.operator("wm.context_set_value", text=name, icon=icon)
+
+    # sets what the menu item changes
+    if path:
+        prop.value = value
+        value = eval(value)
+
+    elif type(value) == str:
+        prop.value = "'{}'".format(value)
+
+    else:
+        prop.value = '{}'.format(value)
+
+    # sets the path to what is changed
+    prop.data_path = data_path