Nathan
108 lines
2.9 KiB
Python
108 lines
2.9 KiB
Python
import bpy
|
|
from bpy.utils import register_class, unregister_class
|
|
from bpy.types import Operator, PropertyGroup, UIList
|
|
from bpy.props import EnumProperty
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
class SIMPLEBAKE_UL_Presets_List(UIList):
|
|
"""UIList."""
|
|
|
|
def draw_item(self, context, layout, data, item, icon, active_data,
|
|
active_propname, index):
|
|
|
|
# We could write some code to decide which icon to use here...
|
|
custom_icon = 'PACKAGE'
|
|
|
|
# Make sure your code supports all 3 layout types
|
|
if self.layout_type in {'DEFAULT', 'COMPACT'}:
|
|
layout.label(text=item.name, icon = custom_icon)
|
|
|
|
elif self.layout_type in {'GRID'}:
|
|
layout.alignment = 'CENTER'
|
|
layout.label(text="", icon = custom_icon)
|
|
|
|
class SimpleBake_OT_preset_refresh(Operator):
|
|
"""Refresh list of SimpleBake presets"""
|
|
bl_idname = "simplebake.preset_refresh"
|
|
bl_label = "Refresh"
|
|
|
|
def execute(self, context):
|
|
sbp = context.scene.SimpleBake_Props
|
|
|
|
sbp.presets_list.clear()
|
|
|
|
p = Path(bpy.utils.script_path_user())
|
|
p = p.parents[1]
|
|
p = p / "data" / "SimpleBake"
|
|
|
|
try:
|
|
presets = sorted(os.listdir(str(p)))
|
|
presets = [pr for pr in presets if not os.path.isdir(str(p / pr))]
|
|
except Exception as e:
|
|
print(repr(e))
|
|
self.report({"INFO"}, "No presets found")
|
|
return {'CANCELLED'}
|
|
|
|
if len(presets) == 0:
|
|
self.report({"INFO"}, "No presets found")
|
|
return {'CANCELLED'}
|
|
|
|
for preset in presets:
|
|
#List should be clear
|
|
i = sbp.presets_list.add()
|
|
i.name = preset
|
|
|
|
return {'FINISHED'}
|
|
|
|
class SimpleBake_OT_preset_delete(Operator):
|
|
"""Delete selected SimpleBake preset"""
|
|
bl_idname = "simplebake.preset_delete"
|
|
bl_label = "Delete"
|
|
|
|
@classmethod
|
|
def poll(cls,context):
|
|
sbp = context.scene.SimpleBake_Props
|
|
try:
|
|
sbp.presets_list[sbp.presets_list_index].name
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
def execute(self, context):
|
|
sbp = context.scene.SimpleBake_Props
|
|
|
|
p = Path(bpy.utils.script_path_user())
|
|
p = p.parents[1]
|
|
p = p / "data" / "SimpleBake"
|
|
|
|
index = sbp.presets_list_index
|
|
item = sbp.presets_list[index]
|
|
p = p / item.name
|
|
|
|
os.remove(str(p))
|
|
|
|
#Refreh the list
|
|
|
|
bpy.ops.simplebake.preset_refresh()
|
|
|
|
return {'FINISHED'}
|
|
|
|
|
|
classes = ([
|
|
SIMPLEBAKE_UL_Presets_List,
|
|
SimpleBake_OT_preset_refresh,
|
|
SimpleBake_OT_preset_delete
|
|
])
|
|
|
|
def register():
|
|
global classes
|
|
for cls in classes:
|
|
register_class(cls)
|
|
|
|
def unregister():
|
|
global classes
|
|
for cls in classes:
|
|
unregister_class(cls)
|