# SPDX-License-Identifier: GPL-2.0-or-later from pathlib import Path import bpy from bpy.types import Panel, Menu from bl_ui.utils import PresetPanel from vse_toolbox.bl_utils import (get_addon_prefs, get_scene_settings, get_strip_settings) from vse_toolbox.constants import ASSET_PREVIEWS, REVIEW_TEMPLATE_BLEND from vse_toolbox.sequencer_utils import (set_active_strip, get_channel_name, get_strips) from vse_toolbox.file_utils import norm_str class VSETB_main: bl_space_type = "SEQUENCE_EDITOR" bl_region_type = "UI" bl_category = "VSE ToolBox" bl_label = "VSE ToolBox" class VSETB_PT_main(VSETB_main, Panel): bl_options = {"HIDE_HEADER"} # def draw_header(self, context): # settings = get_scene_settings() # prefs = get_addon_prefs() # row = self.layout.row(align=True) # row.label('VSE Toolbox') # row.prop(settings, 'project_name', text='') # project = settings.active_project # if project and project.type == 'TVSHOW': # row.prop(project, 'episode_name', text='') # row.operator('vse_toolbox.load_projects', icon='FILE_REFRESH', text='', emboss=False) def draw(self, context): wm = context.window_manager scn = context.scene settings = get_scene_settings() prefs = get_addon_prefs() row = self.layout.row(align=False) row.operator('vse_toolbox.load_projects', icon='FILE_REFRESH', text='', emboss=False) row.prop(settings, 'project_name', text='') project = settings.active_project if project and project.type == 'TVSHOW': row.prop(project, 'episode_name', text='') # settings = get_scene_settings() # prefs = get_addon_prefs() # project = settings.active_project # layout = self.layout # col = layout.column() # col.prop(settings, 'project_name', text='Project') # if project: # if project.type == 'TVSHOW': # col.prop(project, 'episode_name', text='Episodes') #col.separator() #row = col.row(align=True) #row.prop(settings, 'toogle_prefs', text='', icon='PREFERENCES', toggle=True) ''' if settings.toogle_prefs: box = col.box() col = box.column(align=True) col.use_property_split = True col.use_property_decorate = False col.prop(settings, 'project_name', text='Projects') if project: if project.type == 'TV Shows': col.prop(project, 'episode_name', text='Episodes') #col.prop(project, 'sequence_template') #col.prop(project, 'shot_template') # col.separator() # col.operator('vse_toolbox.new_episode', text='Add Episode', icon='IMPORT') ''' # Rename class VSETB_PT_strip(Panel): bl_space_type = "SEQUENCE_EDITOR" bl_region_type = "UI" bl_category = "Strip" bl_label = "VSE ToolBox" #bl_order = 0 @classmethod def poll(cls, context): strip = context.active_sequence_strip return strip and get_channel_name(strip) == 'Shots' def draw(self, context): prefs = get_addon_prefs() layout = self.layout settings = get_strip_settings() layout.use_property_split = True layout.use_property_decorate = False layout.prop(settings, 'source_name', text='Source Name') class VSETB_PT_sequencer(VSETB_main, Panel): bl_label = "Sequencer" #bl_parent_id = "VSETB_PT_main" def draw_header_preset(self, context): settings = get_scene_settings() audio_strips = get_strips('Audio') depress = any(s.show_waveform for s in audio_strips) self.layout.operator('vse_toolbox.show_waveform', text="", icon="IPO_ELASTIC", depress=depress).enabled = not depress ico = ("RESTRICT_SELECT_OFF" if settings.auto_select_strip else "RESTRICT_SELECT_ON") self.layout.prop(settings, "auto_select_strip", text="", icon=ico) def draw(self, context): prefs = get_addon_prefs() layout = self.layout settings = get_scene_settings() project = settings.active_project strip = context.active_sequence_strip channel = get_channel_name(strip) col = layout.column() col.operator('vse_toolbox.set_sequencer', text='Set-Up Sequencer', icon='SEQ_SEQUENCER') col.operator('vse_toolbox.strips_rename', text=f'Rename {channel}', icon='SORTALPHA') col.operator('vse_toolbox.set_stamps', text='Set Stamps', icon='COLOR') col.operator("vse_toolbox.collect_files", text='Collect Files', icon='PACKAGE') class VSETB_PT_settings(VSETB_main, Panel): bl_label = "Settings" #bl_parent_id = "VSETB_PT_main" bl_options = {'DEFAULT_CLOSED'} #def draw_header_preset(self, context): # self.layout.operator('vse_toolbox.import_files', icon='IMPORT', text='', emboss=False) def draw(self, context): prefs = get_addon_prefs() layout = self.layout settings = get_scene_settings() project = settings.active_project col = layout.column() #row = col.row(align=True) col.prop(project, 'sequence_template') col.prop(project, 'shot_template') col.prop(project, 'render_template') class VSETB_PT_imports(VSETB_main, Panel): bl_label = "Imports" #bl_parent_id = "VSETB_PT_main" bl_options = {'DEFAULT_CLOSED'} def draw_header_preset(self, context): self.layout.operator('vse_toolbox.import_files', icon='IMPORT', text='', emboss=False) def draw(self, context): prefs = get_addon_prefs() layout = self.layout settings = get_scene_settings() project = settings.active_project col = layout.column() #row = col.row(align=True) #col.operator('vse_toolbox.import_files', text='Import', icon='IMPORT') col.operator('vse_toolbox.import_spreadsheet', text='Import Spreadsheet', icon='SPREADSHEET') col.operator("vse_toolbox.import_shots", text='Import Shots', icon="FILE_MOVIE") class VSETB_PT_presets(PresetPanel, Panel): bl_label = 'Spreadsheet Presets' preset_subdir = 'vse_toolbox' preset_operator = 'script.execute_preset' preset_add_operator = "vse_toolbox.add_spreadsheet_preset" class VSETB_PT_exports(VSETB_main, Panel): bl_label = "Exports" #bl_parent_id = "VSETB_PT_main" bl_options = {'DEFAULT_CLOSED'} def draw_header_preset(self, context): self.layout.operator('vse_toolbox.export_spreadsheet', icon='EXPORT', text='', emboss=False) def draw(self, context): prefs = get_addon_prefs() layout = self.layout settings = get_scene_settings() project = settings.active_project col = layout.column(align=False) # TODO FAIRE DES VRAIS OPS col.operator('vse_toolbox.strips_render', text='Render Strips', icon='SEQUENCE') tracker_label = settings.tracker_name.title().replace('_', ' ') col.operator('vse_toolbox.upload_to_tracker', text=f'Upload to {tracker_label}', icon='EXPORT') col.operator('vse_toolbox.export_spreadsheet', text='Export Spreadsheet', icon='SPREADSHEET') col.operator('vse_toolbox.export_edl', text='Export edl', icon='SEQ_SEQUENCER') class VSETB_PT_tracker(VSETB_main, Panel): bl_label = "Tracker" #bl_parent_id = "VSETB_PT_main" bl_options = {'DEFAULT_CLOSED'} @classmethod def poll(cls, context): return context.active_sequence_strip def draw_header_preset(self, context): active_strip = context.active_sequence_strip self.layout.label(text=active_strip.name) def draw(self, context): return class VSETB_PT_casting(VSETB_main, Panel): bl_label = "Casting" bl_parent_id = "VSETB_PT_tracker" bl_options = {'DEFAULT_CLOSED'} @classmethod def poll(cls, context): strip = context.active_sequence_strip return strip and get_channel_name(strip) == 'Shots' def draw(self, context): layout = self.layout settings = get_scene_settings() strip_settings = get_strip_settings() project = settings.active_project if not project: return if not project.assets: row = layout.row(align=True) row.label(text='No Assets in this Project') else: row = layout.row() col = row.column() col.template_list("VSETB_UL_casting", "shot_casting", strip_settings, "casting", strip_settings, "casting_index", rows=6) col_tool = row.column(align=True) col_tool.operator('vse_toolbox.casting_add', icon='ADD', text="") col_tool.operator('vse_toolbox.casting_remove', icon='REMOVE', text="") col_tool.separator() col_tool.operator('vse_toolbox.casting_move', icon='TRIA_UP', text="").direction = 'UP' col_tool.operator('vse_toolbox.casting_move', icon='TRIA_DOWN', text="").direction = 'DOWN' col_tool.separator() col_tool.operator('vse_toolbox.copy_casting', icon='COPYDOWN', text="") col_tool.operator('vse_toolbox.paste_casting', icon='PASTEDOWN', text="") col_tool.separator() col_tool.operator('vse_toolbox.casting_replace', icon='ZOOM_ALL', text="") if strip_settings.casting: casting_item = strip_settings.casting[strip_settings.casting_index] asset = casting_item.asset if asset: if asset.icon_id: row = col.row(align=True) #row.scale_y = 0.5 # box = col.box() # box.template_icon(icon_value=ico.icon_id, scale=7.5) row.template_icon_view(asset, "previews", show_labels=False) class VSETB_PT_metadata(VSETB_main, Panel): bl_label = "Shot Metadata" bl_parent_id = "VSETB_PT_tracker" bl_options = {'DEFAULT_CLOSED'} @classmethod def poll(cls, context): return context.active_sequence_strip and get_scene_settings().active_project def draw(self, context): strip_settings = get_strip_settings() project = get_scene_settings().active_project layout = self.layout row = layout.row(align=True) label_col = row.column(align=True) label_col.alignment = 'RIGHT' field_col = row.column(align=True) for metadata_type in project.metadata_types: if metadata_type.entity_type != 'SHOT': continue metadata_key = metadata_type.field_name metadata_label = metadata_key.title() label_col.label(text=metadata_label) field_row = field_col.row(align=True) field_row.separator() if metadata_type.choices: metadata_value = getattr(strip_settings.metadata, metadata_key) icon = 'LAYER_USED' if metadata_value: if metadata_value in metadata_type.choices: icon = 'DOT' else: icon = 'ADD' field_row.prop_search(strip_settings.metadata, metadata_key, metadata_type, 'choices', results_are_suggestions=True, icon=icon, text='') else: field_row.prop(strip_settings.metadata, metadata_key, text='') class VSETB_PT_comments(VSETB_main, Panel): bl_label = "Comments" bl_parent_id = "VSETB_PT_tracker" bl_options = {'DEFAULT_CLOSED'} @classmethod def poll(cls, context): return context.active_sequence_strip and get_scene_settings().active_project def draw(self, context): layout = self.layout row = layout.row(align=True) label_col = row.column(align=True) label_col.alignment = 'RIGHT' field_col = row.column(align=True) strip_settings = get_strip_settings() project = get_scene_settings().active_project for task_type in project.task_types: norm_task_name = norm_str(task_type.name) if not hasattr(strip_settings.tasks, norm_task_name): continue label_col.label(text=task_type.name) row = field_col.row(align=True) row.separator() sub = row.row(align=True) sub.alignment = 'LEFT' sub.scale_x = 0.15 sub.prop(task_type, 'color', text='') sub.enabled = False row.prop(getattr(strip_settings.tasks, norm_task_name), 'comment', text='') def context_menu_prop(self, context): if not hasattr(context, 'button_prop') or context.space_data.type != 'SEQUENCE_EDITOR': return settings = get_strip_settings() if not settings: return button_prop = context.button_prop if button_prop not in settings.metadata.bl_rna.properties.values(): return layout = self.layout layout.separator() layout.operator('vse_toolbox.copy_metadata', icon='PASTEDOWN', text='Copy metadata to selected').metadata = button_prop.name class VSETB_MT_main_menu(Menu): bl_label = "Vse Toolbox" def draw(self, context): layout = self.layout op = layout.operator('workspace.append_activate', text='Set Review Workspace', icon="WORKSPACE") op.idname = 'Review' op.filepath = str(REVIEW_TEMPLATE_BLEND) layout.operator("wm.split_view", icon="ARROW_LEFTRIGHT") layout.operator('vse_toolbox.open_shot_folder', text='Open Shot Folder', icon='FILE_FOLDER') layout.operator('vse_toolbox.open_shot_on_tracker', text='Open Shot on Tracker', icon='URL') def draw_vse_toolbox_menu(self, context): self.layout.menu("VSETB_MT_main_menu") classes = ( VSETB_PT_main, VSETB_PT_imports, VSETB_PT_sequencer, VSETB_PT_tracker, VSETB_PT_casting, VSETB_PT_metadata, VSETB_PT_comments, VSETB_PT_presets, VSETB_PT_exports, VSETB_PT_strip, VSETB_MT_main_menu ) def register(): for cls in classes: bpy.utils.register_class(cls) bpy.types.UI_MT_button_context_menu.append(context_menu_prop) bpy.types.SEQUENCER_MT_editor_menus.append(draw_vse_toolbox_menu) def unregister(): for cls in reversed(classes): bpy.utils.unregister_class(cls) bpy.types.UI_MT_button_context_menu.remove(context_menu_prop) bpy.types.SEQUENCER_MT_editor_menus.remove(draw_vse_toolbox_menu)