115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
import bpy
|
|
from . import gen_vlayer, fn
|
|
|
|
class GPEXP_OT_add_layer_to_render(bpy.types.Operator):
|
|
bl_idname = "gp.add_layer_to_render"
|
|
bl_label = "Add Gp Layer as render nodes"
|
|
bl_description = "Create a view_layer for GP > add node > connect to output"
|
|
bl_options = {"REGISTER"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.object and context.object.type == 'GPENCIL'
|
|
|
|
# mode : bpy.props.StringProperty(options={'SKIP_SAVE'})
|
|
|
|
def execute(self, context):
|
|
ob = context.object
|
|
layer = ob.data.layers.active
|
|
if not layer:
|
|
self.report({'ERROR'}, 'No active layer')
|
|
return {"CANCELLED"}
|
|
|
|
ct = 0
|
|
# send scene ?
|
|
hided = 0
|
|
for l in ob.data.layers:
|
|
if not l.select:
|
|
if not l.viewlayer_render:
|
|
# TODO : need to link, can raise error if object is not linked in Render scene yet
|
|
l.viewlayer_render = fn.get_view_layer('exclude').name
|
|
continue
|
|
gen_vlayer.get_set_viewlayer_from_gp(ob, l)
|
|
|
|
if l.hide:
|
|
hided += 1
|
|
ct += 1
|
|
|
|
if hided:
|
|
self.report({'WARNING'}, f'{hided}/{ct} layers are hided !')
|
|
|
|
else:
|
|
self.report({'INFO'}, f'{ct} layer(s) added to scene "Render"')
|
|
|
|
return {"FINISHED"}
|
|
|
|
## send operator with mode ALL or SELECTED to batch build
|
|
class GPEXP_OT_add_objects_to_render(bpy.types.Operator):
|
|
bl_idname = "gp.add_object_to_render"
|
|
bl_label = "Add all Gp Layer of active object as render nodes"
|
|
bl_description = "Setup GP object in render scene\
|
|
\nNote: 'send all' mode skip hided objects"
|
|
bl_options = {"REGISTER"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.object and context.object.type == 'GPENCIL'
|
|
|
|
mode : bpy.props.StringProperty(default='ALL', options={'SKIP_SAVE'})
|
|
|
|
def execute(self, context):
|
|
# create render scene
|
|
if context.scene.name == 'Scene':
|
|
scn = fn.get_render_scene()
|
|
else:
|
|
scn = context.scene
|
|
|
|
|
|
excludes = [] # ['MA', 'IN'] # Get list dynamically
|
|
if self.mode == 'SELECTED':
|
|
gen_vlayer.export_gp_objects([o for o in context.selected_objects if o.type == 'GPENCIL'], exclude_list=excludes, scene=scn)
|
|
|
|
elif self.mode == 'ALL':
|
|
# scn = bpy.data.scenes.get('Scene')
|
|
# if not scn:
|
|
# self.report({'ERROR'}, 'Could not found default scene')
|
|
# return {"CANCELLED"}
|
|
gen_vlayer.export_gp_objects([o for o in context.scene.objects if o.type == 'GPENCIL' and not o.hide_get() and fn.is_valid_name(o.name)], exclude_list=excludes, scene=scn)
|
|
|
|
return {"FINISHED"}
|
|
|
|
|
|
class GPEXP_OT_split_to_scene(bpy.types.Operator):
|
|
bl_idname = "gp.split_to_scene"
|
|
bl_label = "Split Objects To Scene"
|
|
bl_description = "Take selected objects and send them to separate scene\n(new scene is named after active object)"
|
|
bl_options = {"REGISTER"}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.object and context.object.type == 'GPENCIL'
|
|
|
|
mode : bpy.props.StringProperty(default='ALL', options={'SKIP_SAVE'})
|
|
|
|
def execute(self, context):
|
|
err = fn.split_object_to_scene()
|
|
if err:
|
|
self.report({'ERROR'}, err)
|
|
return {"CANCELLED"}
|
|
return {"FINISHED"}
|
|
|
|
|
|
|
|
classes=(
|
|
GPEXP_OT_add_layer_to_render,
|
|
GPEXP_OT_add_objects_to_render,
|
|
GPEXP_OT_split_to_scene,
|
|
)
|
|
|
|
def register():
|
|
for cls in classes:
|
|
bpy.utils.register_class(cls)
|
|
|
|
def unregister():
|
|
for cls in reversed(classes):
|
|
bpy.utils.unregister_class(cls) |