338 lines
12 KiB
Python
338 lines
12 KiB
Python
|
import bpy
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
from bpy_extras.io_utils import ImportHelper, ExportHelper
|
|||
|
|
from pathlib import Path
|
|||
|
|
from . import utils
|
|||
|
|
# from . import blendfile
|
|||
|
|
|
|||
|
|
from bpy.types import (
|
|||
|
|
Panel,
|
|||
|
|
Operator,
|
|||
|
|
PropertyGroup,
|
|||
|
|
UIList,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
from bpy.props import (
|
|||
|
|
IntProperty,
|
|||
|
|
BoolProperty,
|
|||
|
|
StringProperty,
|
|||
|
|
FloatProperty,
|
|||
|
|
EnumProperty,
|
|||
|
|
PointerProperty,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
class GPTB_UL_blend_list(UIList):
|
|||
|
|
# order_by_distance : BoolProperty(default=True)
|
|||
|
|
|
|||
|
|
def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
|
|||
|
|
layout.label(text=item.blend_name)
|
|||
|
|
|
|||
|
|
def draw_filter(self, context, layout):
|
|||
|
|
row = layout.row()
|
|||
|
|
subrow = row.row(align=True)
|
|||
|
|
subrow.prop(self, "filter_name", text="") # Only show items matching this name (use ‘*’ as wildcard)
|
|||
|
|
|
|||
|
|
# reverse order
|
|||
|
|
icon = 'SORT_DESC' if self.use_filter_sort_reverse else 'SORT_ASC'
|
|||
|
|
subrow.prop(self, "use_filter_sort_reverse", text="", icon=icon) # built-in reverse
|
|||
|
|
|
|||
|
|
def filter_items(self, context, data, propname):
|
|||
|
|
# example : https://docs.blender.org/api/blender_python_api_current/bpy.types.UIList.html
|
|||
|
|
# This function gets the collection property (as the usual tuple (data, propname)), and must return two lists:
|
|||
|
|
# * The first one is for filtering, it must contain 32bit integers were self.bitflag_filter_item marks the
|
|||
|
|
# matching item as filtered (i.e. to be shown), and 31 other bits are free for custom needs. Here we use the
|
|||
|
|
# * The second one is for reordering, it must return a list containing the new indices of the items (which
|
|||
|
|
# gives us a mapping org_idx -> new_idx).
|
|||
|
|
# Please note that the default UI_UL_list defines helper functions for common tasks (see its doc for more info).
|
|||
|
|
# If you do not make filtering and/or ordering, return empty list(s) (this will be more efficient than
|
|||
|
|
# returning full lists doing nothing!).
|
|||
|
|
|
|||
|
|
collec = getattr(data, propname)
|
|||
|
|
helper_funcs = bpy.types.UI_UL_list
|
|||
|
|
|
|||
|
|
# Default return values.
|
|||
|
|
flt_flags = []
|
|||
|
|
flt_neworder = []
|
|||
|
|
|
|||
|
|
|
|||
|
|
# Filtering by name #not working damn !
|
|||
|
|
if self.filter_name:
|
|||
|
|
flt_flags = helper_funcs.filter_items_by_name(self.filter_name, self.bitflag_filter_item, collec, "name",
|
|||
|
|
reverse=self.use_filter_sort_reverse)#self.use_filter_name_reverse)
|
|||
|
|
|
|||
|
|
return flt_flags, flt_neworder
|
|||
|
|
|
|||
|
|
class GPTB_UL_object_list(UIList):
|
|||
|
|
# order_by_distance : BoolProperty(default=True)
|
|||
|
|
|
|||
|
|
def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
|
|||
|
|
layout.label(text=item.name)
|
|||
|
|
|
|||
|
|
def draw_filter(self, context, layout):
|
|||
|
|
row = layout.row()
|
|||
|
|
subrow = row.row(align=True)
|
|||
|
|
subrow.prop(self, "filter_name", text="") # Only show items matching this name (use ‘*’ as wildcard)
|
|||
|
|
# reverse order
|
|||
|
|
icon = 'SORT_DESC' if self.use_filter_sort_reverse else 'SORT_ASC'
|
|||
|
|
subrow.prop(self, "use_filter_sort_reverse", text="", icon=icon) # built-in reverse
|
|||
|
|
|
|||
|
|
def filter_items(self, context, data, propname):
|
|||
|
|
collec = getattr(data, propname)
|
|||
|
|
helper_funcs = bpy.types.UI_UL_list
|
|||
|
|
# Default return values.
|
|||
|
|
flt_flags = []
|
|||
|
|
flt_neworder = []
|
|||
|
|
if self.filter_name:
|
|||
|
|
flt_flags = helper_funcs.filter_items_by_name(self.filter_name, self.bitflag_filter_item, collec, "name",
|
|||
|
|
reverse=self.use_filter_sort_reverse)
|
|||
|
|
return flt_flags, flt_neworder
|
|||
|
|
|
|||
|
|
|
|||
|
|
def reload_blends(self, context):
|
|||
|
|
scn = context.scene
|
|||
|
|
pl_prop = scn.bl_palettes_props
|
|||
|
|
uilist = scn.bl_palettes_props.blends
|
|||
|
|
uilist.clear()
|
|||
|
|
pl_prop['bl_idx'] = 0
|
|||
|
|
|
|||
|
|
prefs = utils.get_addon_prefs()
|
|||
|
|
|
|||
|
|
if pl_prop.use_project_path:
|
|||
|
|
palette_fp = prefs.palette_path
|
|||
|
|
else:
|
|||
|
|
palette_fp = pl_prop.custom_dir
|
|||
|
|
|
|||
|
|
if not palette_fp: # singular
|
|||
|
|
item = uilist.add()
|
|||
|
|
item.blend_name = 'No Palette Path Specified'
|
|||
|
|
reload_objects(self, context)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
palettes_dir = Path(palette_fp)
|
|||
|
|
if not palettes_dir.exists():
|
|||
|
|
item = uilist.add()
|
|||
|
|
item.blend_name = 'Palette Path not found'
|
|||
|
|
reload_objects(self, context)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
blends = [(o.path, Path(o).stem, "") for o in os.scandir(palettes_dir)
|
|||
|
|
if o.is_file()
|
|||
|
|
and o.name.endswith('.blend')
|
|||
|
|
and re.search(r'v\d{3}', o.name, re.I)]
|
|||
|
|
|
|||
|
|
blends.sort(key=lambda x: x[1], reverse=False)
|
|||
|
|
# print('blends found', len(blends))
|
|||
|
|
|
|||
|
|
for bl in blends: # populate list
|
|||
|
|
item = uilist.add()
|
|||
|
|
|
|||
|
|
scn.bl_palettes_props['bl_idx'] = len(uilist) - 1 # don't trigger updates
|
|||
|
|
item.blend_path = bl[0]
|
|||
|
|
item.blend_name = bl[1]
|
|||
|
|
|
|||
|
|
scn.bl_palettes_props.bl_idx = len(uilist) - 1 # trigger update ()
|
|||
|
|
# reload_objects(self, context) # triggered by above assignation
|
|||
|
|
|
|||
|
|
# return len(blends) # return value must be None
|
|||
|
|
|
|||
|
|
class GPTB_OT_palettes_reload_blends(Operator):
|
|||
|
|
bl_idname = "gp.palettes_reload_blends"
|
|||
|
|
bl_label = "Reload Palette Blends"
|
|||
|
|
bl_description = "Reload the blends in UI list of palettes linker"
|
|||
|
|
bl_options = {"REGISTER"} # , "INTERNAL"
|
|||
|
|
|
|||
|
|
def execute(self, context):
|
|||
|
|
reload_blends(self, context)
|
|||
|
|
# ret = reload_blends(self, context)
|
|||
|
|
# if ret is None:
|
|||
|
|
# self.report({'ERROR'}, 'No blend scanned, check palette path')
|
|||
|
|
# else:
|
|||
|
|
# self.report({'INFO'}, f'{ret} blends found')
|
|||
|
|
return {"FINISHED"}
|
|||
|
|
|
|||
|
|
def reload_objects(self, context):
|
|||
|
|
scn = context.scene
|
|||
|
|
prefs = utils.get_addon_prefs()
|
|||
|
|
pal_prop = scn.bl_palettes_props
|
|||
|
|
blend_uil = pal_prop.blends
|
|||
|
|
obj_uil = pal_prop.objects
|
|||
|
|
obj_uil.clear()
|
|||
|
|
pal_prop['ob_idx'] = 0
|
|||
|
|
|
|||
|
|
if not len(blend_uil) or (len(blend_uil) == 1 and not bool(blend_uil[0].blend_path)):
|
|||
|
|
item = obj_uil.add()
|
|||
|
|
item.name = 'No blend to list object'
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if not blend_uil[pal_prop.bl_idx].blend_path:
|
|||
|
|
item = obj_uil.add()
|
|||
|
|
item.name = 'Selected blend has no path'
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
path_to_blend = Path(blend_uil[pal_prop.bl_idx].blend_path)
|
|||
|
|
|
|||
|
|
ob_list = utils.check_objects_in_blend(str(path_to_blend)) # get list of string of all object except camera
|
|||
|
|
|
|||
|
|
ob_list.sort(reverse=False) # filter object by name ?
|
|||
|
|
# remove camera
|
|||
|
|
# print('blends found', len(blends))
|
|||
|
|
|
|||
|
|
for ob_name in ob_list: # populate list
|
|||
|
|
item = obj_uil.add()
|
|||
|
|
item.name = ob_name
|
|||
|
|
item.path = str(path_to_blend / 'Object' / ob_name)
|
|||
|
|
|
|||
|
|
pal_prop.ob_idx = len(obj_uil) - 1
|
|||
|
|
# return len(ob_list) # must return None if used in update
|
|||
|
|
|
|||
|
|
|
|||
|
|
## PROPS
|
|||
|
|
|
|||
|
|
class GPTB_PG_blend_prop(PropertyGroup):
|
|||
|
|
blend_name : StringProperty() # stem of the path
|
|||
|
|
blend_path : StringProperty() # ful path
|
|||
|
|
# select: BoolProperty(update=update_select) # use and update to set the plane selection
|
|||
|
|
|
|||
|
|
class GPTB_PG_object_prop(PropertyGroup):
|
|||
|
|
name : StringProperty() # stem of the path
|
|||
|
|
path : StringProperty() # Object / Material ?
|
|||
|
|
|
|||
|
|
class GPTB_PG_palette_settings(PropertyGroup):
|
|||
|
|
bl_idx : IntProperty(update=reload_objects) # update_on_index_change to reload object
|
|||
|
|
blends : bpy.props.CollectionProperty(type=GPTB_PG_blend_prop)
|
|||
|
|
|
|||
|
|
ob_idx : IntProperty()
|
|||
|
|
objects : bpy.props.CollectionProperty(type=GPTB_PG_object_prop)
|
|||
|
|
|
|||
|
|
use_project_path : BoolProperty(name='Use Project Palettes',
|
|||
|
|
default=True, description='Use palettes directory specified in gp toolbox addon preferences',
|
|||
|
|
update=reload_blends)
|
|||
|
|
|
|||
|
|
show_path : BoolProperty(name='Show path',
|
|||
|
|
default=True, description='Show Palette directoty filepath')
|
|||
|
|
|
|||
|
|
custom_dir : StringProperty(name='Custom Palettes Directory', subtype='DIR_PATH',
|
|||
|
|
description='Use choosen directory to load blend palettes',
|
|||
|
|
update=reload_blends)
|
|||
|
|
|
|||
|
|
import_type : EnumProperty(
|
|||
|
|
name="Import Type", description="Choose inmport type: link, append, append reuse (keep existing materials)",
|
|||
|
|
default='LINK', options={'ANIMATABLE'}, update=None, get=None, set=None,
|
|||
|
|
items=(
|
|||
|
|
('LINK', 'Link', 'Link materials to selected object', 0),
|
|||
|
|
('APPEND', 'Append', 'Append materials to selected objects', 1),
|
|||
|
|
('APPEND_REUSE', 'Append (Reuse)', 'Append materials to selected objects\nkeep those already there', 2),
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# fav_blend: StringProperty() ## mark a blend as prefered ? (need to be stored in prefereneces to restore in other blend...)
|
|||
|
|
|
|||
|
|
class GPTB_OT_import_obj_palette(Operator):
|
|||
|
|
bl_idname = "gp.import_obj_palette"
|
|||
|
|
bl_label = "Import Object Palette"
|
|||
|
|
bl_description = "Import object palette from blend"
|
|||
|
|
bl_options = {"REGISTER", "INTERNAL"}
|
|||
|
|
|
|||
|
|
def execute(self, context):
|
|||
|
|
pl_prop = context.scene.bl_palettes_props
|
|||
|
|
path = pl_prop.objects[pl_prop.ob_idx].path
|
|||
|
|
self.report({'INFO'}, f'Path to object: {path}')
|
|||
|
|
return {"FINISHED"}
|
|||
|
|
|
|||
|
|
## PANEL
|
|||
|
|
|
|||
|
|
class GPTB_PT_palettes_linker_ui(Panel):
|
|||
|
|
bl_idname = 'GPTB_PT_palettes_linker_ui'
|
|||
|
|
bl_space_type = "VIEW_3D"
|
|||
|
|
bl_region_type = "UI"
|
|||
|
|
bl_category = "Gpencil"#Dev
|
|||
|
|
bl_label = "Palettes Mat Linker"
|
|||
|
|
|
|||
|
|
def draw(self, context):
|
|||
|
|
layout = self.layout
|
|||
|
|
scn = bpy.context.scene
|
|||
|
|
pl_prop = scn.bl_palettes_props
|
|||
|
|
col= layout.column()
|
|||
|
|
prefs = utils.get_addon_prefs()
|
|||
|
|
## Here put the path thing (only to use a non-library)
|
|||
|
|
|
|||
|
|
# maybe in submenu...
|
|||
|
|
row = col.row()
|
|||
|
|
expand_icon = 'TRIA_DOWN' if pl_prop.show_path else 'TRIA_RIGHT'
|
|||
|
|
row.prop(pl_prop, 'show_path', text='', icon=expand_icon, emboss=False)
|
|||
|
|
row.prop(pl_prop, 'use_project_path', text='Use Project Palettes')
|
|||
|
|
row.operator("gp.palettes_reload_blends", icon="FILE_REFRESH", text="")
|
|||
|
|
|
|||
|
|
if pl_prop.use_project_path:
|
|||
|
|
## gp toolbox addon prefs path
|
|||
|
|
if not prefs.palette_path:
|
|||
|
|
col.label(text='Gp Toolbox Palette Directory Needed')
|
|||
|
|
col.label(text='(saved with preferences)')
|
|||
|
|
if pl_prop.show_path or not prefs.palette_path:
|
|||
|
|
col.prop(prefs, 'palette_path', text='Project Dir')
|
|||
|
|
else:
|
|||
|
|
## local path
|
|||
|
|
if not pl_prop.custom_dir:
|
|||
|
|
col.label(text='Need to specify directory')
|
|||
|
|
if pl_prop.show_path or not pl_prop.custom_dir:
|
|||
|
|
col.prop(pl_prop, 'custom_dir', text='Custom Dir')
|
|||
|
|
|
|||
|
|
|
|||
|
|
row = col.row()
|
|||
|
|
row.template_list("GPTB_UL_blend_list", "", pl_prop, "blends", pl_prop, "bl_idx",
|
|||
|
|
rows=2)
|
|||
|
|
# side panel
|
|||
|
|
# subcol = row.column(align=True)
|
|||
|
|
# subcol.operator("gp.palettes_reload_blends", icon="FILE_REFRESH", text="")
|
|||
|
|
|
|||
|
|
## Show object UI list only once blend Uilist is filled ?
|
|||
|
|
if not len(pl_prop.blends) or (len(pl_prop.blends) == 1 and not bool(pl_prop.blends[0].blend_path)):
|
|||
|
|
col.label(text='Select a blend to list available object')
|
|||
|
|
|
|||
|
|
row = col.row()
|
|||
|
|
row.template_list("GPTB_UL_object_list", "", pl_prop, "objects", pl_prop, "ob_idx",
|
|||
|
|
rows=4)
|
|||
|
|
|
|||
|
|
## Show link button in the border of the UI list ?
|
|||
|
|
# col.prop(pl_prop, 'import_type')
|
|||
|
|
split = col.split(align=True, factor=0.4 )
|
|||
|
|
split.prop(pl_prop, 'import_type', text='')
|
|||
|
|
|
|||
|
|
split.enabled = len(pl_prop.objects) and bool(pl_prop.objects[pl_prop.ob_idx].path)
|
|||
|
|
split.operator('gp.import_obj_palette', text='Palette')
|
|||
|
|
|
|||
|
|
# button to launch link with combined props (active only if the two items are valids)
|
|||
|
|
# str(Path(self.blends) / 'Object' / self.objects
|
|||
|
|
|
|||
|
|
|
|||
|
|
classes = (
|
|||
|
|
# blend list
|
|||
|
|
GPTB_PG_blend_prop,
|
|||
|
|
GPTB_UL_blend_list,
|
|||
|
|
GPTB_OT_palettes_reload_blends,
|
|||
|
|
|
|||
|
|
# object in blend list
|
|||
|
|
GPTB_PG_object_prop,
|
|||
|
|
GPTB_UL_object_list,
|
|||
|
|
|
|||
|
|
# prop containing two above
|
|||
|
|
GPTB_PG_palette_settings,
|
|||
|
|
|
|||
|
|
GPTB_OT_import_obj_palette,
|
|||
|
|
GPTB_PT_palettes_linker_ui,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def register():
|
|||
|
|
for cls in classes:
|
|||
|
|
bpy.utils.register_class(cls)
|
|||
|
|
bpy.types.Scene.bl_palettes_props = bpy.props.PointerProperty(type=GPTB_PG_palette_settings)
|
|||
|
|
|
|||
|
|
def unregister():
|
|||
|
|
for cls in reversed(classes):
|
|||
|
|
bpy.utils.unregister_class(cls)
|
|||
|
|
|
|||
|
|
del bpy.types.Scene.bl_palettes_props
|