gp_render/ui.py

203 lines
7.9 KiB
Python

import bpy
from bpy.types import Panel
# from .preferences import get_addon_prefs
# Node view panel
class GPEXP_PT_gp_node_ui(Panel):
bl_space_type = "NODE_EDITOR"
bl_region_type = "UI"
# bl_category = "View"
bl_category = "GP render"
bl_label = "Gpencil Render Manager"
def draw(self, context):
layout = self.layout
layout.operator('gp.render_scene_switch', icon='SCENE_DATA', text='Switch Scene')
if not context.scene.use_nodes or not context.scene.node_tree:
return
# TODO : add advanced bool checkbox to hide some options from the user
col = layout.column(align=True)
ct = len([n for n in context.scene.node_tree.nodes if n.type == 'R_LAYERS' and n.select])
txt = f'Merge {ct} Layer Nodes'
col.operator('gp.merge_selected_viewlayer_nodes', icon='NODETREE', text=txt).disconnect = True
col.operator('gp.merge_selected_viewlayer_nodes', icon='NODETREE', text='Merge (keep connect)').disconnect = False
col.enabled = ct > 1
layout.separator()
layout.label(text='All Outputs:')
row=layout.row()
row.operator('gp.mute_toggle_output_nodes', icon='NODE_INSERT_ON', text='Mute').mute = True
row.operator('gp.mute_toggle_output_nodes', icon='NODE_INSERT_OFF', text='Unmute').mute = False
layout.separator()
col=layout.column()
col.label(text='Clean and updates:')
row = col.row()
n = context.scene.node_tree.nodes.active
if n:
row.enabled = n and n.type == 'R_LAYERS' and not n.outputs[0].is_linked
else:
row.enabled = False
row.operator('gp.reconnect_render_layer', icon='ANIM', text='Reconnect')
col.operator('gp.clean_compo_tree', icon='BRUSHES_ALL', text='Clean Nodes') # NODE_CORNER
col.separator()
## (re)number exports
ct = len([n for n in context.scene.node_tree.nodes if n.type == 'OUTPUT_FILE' and n.select])
txt = f'Renumber {ct} Selected outputs'
subcol = col.column()
subcol.enabled = bool(ct)
subcol.operator('gp.number_outputs', icon='LINENUMBERS_ON', text=txt).mode = 'SELECTED'
# col.operator('gp.number_outputs', icon='LINENUMBERS_ON', text='Renumber all outputs').mode = 'ALL'
subcol.operator('gp.set_output_node_format', icon='OUTPUT', text='Copy Active Output Format')
layout.separator()
col=layout.column()
col.label(text='Delete Options:')
col.operator('gp.clear_render_tree', icon='X', text='Clear Framed Nodes')
col.operator('gp.clear_render_tree', icon='X', text='Clear & Delete Render Scene').mode = "COMPLETE"
# layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text='Layer To Render').mode = 'ALL'
# layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text='Layer To Render').mode = 'SELECTED'
# layout.operator('gp.merge_layers', icon='X', text='Merge selected nodes')
class GPEXP_PT_gp_dopesheet_ui(Panel):
bl_space_type = 'DOPESHEET_EDITOR'
bl_region_type = 'UI'
bl_category = "GP Render"
# bl_parent_id='DOPESHEET_PT_gpencil_mode'
bl_label = "Gpencil Render Manager"
@classmethod
def poll(cls, context):
return context.object and context.object.type == 'GPENCIL'
def draw(self, context):
layout = self.layout
if context.object:
layout.label(text=f'Object: {context.object.name}')
if context.object.data.users > 1:
row = layout.row()
row.label(text=f'Multiple users ({context.object.data.users})', icon='ERROR')
row.operator("wm.call_menu", text="", icon='QUESTION').name = "GPEXP_MT_multi_user_doc"
layout.label(text=f'viewlayer: {context.object.data.layers.active.viewlayer_render}')
## On layers
if context.object and context.object.type == 'GPENCIL':
txt = f'{len([l for l in context.object.data.layers if l.select])} Layer(s) To Render'
else:
txt = 'Layer To Render'
layout.operator('gp.add_layer_to_render', icon='RENDERLAYERS', text=txt)
# merge (only accessible if multiple layers selected)
row = layout.row()
ct = len([l for l in context.object.data.layers if l.select])
txt = f'Merge {ct} layers'
# merge layers from dopesheet
row.operator('gp.merge_viewlayers_to_active', text=txt, icon='SELECT_EXTEND')
row.enabled= ct > 1
## all and objects
layout.separator()
layout.label(text='Whole objects:')
if context.scene.name != 'Render':
txt = f'{len([o for o in context.selected_objects if o.type == "GPENCIL" and o.select_get()])} Selected Object(s) To Render'
layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text=txt).mode='SELECTED'
layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text='All Visible GP To Render').mode='ALL'
layout.separator()
layout.operator('gp.layers_state', icon='CHECKMARK', text='Check layers')
layout.operator('gp.lower_layers_name', icon='SYNTAX_OFF', text='Rename Lowercase')
# row = layout.row()
layout.prop(bpy.context.preferences.edit, 'use_anim_channel_group_colors')
class GPEXP_MT_multi_user_doc(bpy.types.Menu):
# bl_idname = "OBJECT_MT_custom_menu"
bl_label = "Case of multiuser objects"
def draw(self, context):
layout = self.layout
# call another menu
#layout.operator("wm.call_menu", text="Unwrap").name = "VIEW3D_MT_uv_map"
#**Behavior from context mode**
col = layout.column()
col.label(text='Multi user data will be rendered all together on last generated viewlayers.', icon='INFO')
col.label(text='Make them single user if needed to render separately.')
# col.label(text='Select objects > call search pop-up (F3) > "Make single user" > tick "object data" > ok')
col.label(text='Procedure:')
col.label(text='- select concerned objects')
col.label(text='- call search pop-up (F3)')
col.label(text='- search "Make single user"')
col.label(text='- tick only "object data"')
## not registered for now (better to place render menu in GP dopesheet)
def manager_ui(self, context):
'''appended to DATA_PT_gpencil_layers'''
layout = self.layout
## On layers
if context.object and context.object.type == 'GPENCIL':
txt = f'{len([l for l in context.object.data.layers if l.select])} Layer(s) To Render'
else:
txt = 'Layer To Render'
layout.operator('gp.add_layer_to_render', icon='RENDERLAYERS', text=txt)
## On objects
# txt = 'Selected Object To Render'
if context.scene.name != 'Render':
txt = f'{len([o for o in context.selected_objects if o.type == "GPENCIL" and o.select_get()])} Selected Object(s) To Render'
layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text=txt).mode='SELECTED'
layout.operator('gp.add_object_to_render', icon='RENDERLAYERS', text='All GP at once').mode='ALL'
# ## function to append in a menu
# def palette_manager_menu(self, context):
# """Palette menu to append in existing menu"""
# # GPENCIL_MT_material_context_menu
# layout = self.layout
# # {'EDIT_GPENCIL', 'PAINT_GPENCIL','SCULPT_GPENCIL','WEIGHT_GPENCIL', 'VERTEX_GPENCIL'}
# layout.separator()
# prefs = get_addon_prefs()
# layout.operator("", text='do stuff from material submenu', icon='MATERIAL')
#-# REGISTER
classes=(
GPEXP_MT_multi_user_doc,
GPEXP_PT_gp_node_ui,
GPEXP_PT_gp_dopesheet_ui,
)
def register():
for cls in classes:
bpy.utils.register_class(cls)
# bpy.types.DATA_PT_gpencil_layers.prepend(manager_ui)
def unregister():
# bpy.types.DATA_PT_gpencil_layers.remove(manager_ui)
for cls in reversed(classes):
bpy.utils.unregister_class(cls)