import time import re from pathlib import Path import os import bpy from bpy.types import Operator from bpy.props import BoolProperty from vse_toolbox.sequencer_utils import (get_strips, render_strip, render_sound) from vse_toolbox.bl_utils import (get_scene_settings, background_render) from vse_toolbox.file_utils import install_module class VSETB_OT_render(Operator): bl_idname = "vse_toolbox.strips_render" bl_label = "Render Shots Strips" bl_description = "Render Shots Strips" bl_options = {"REGISTER", "UNDO"} #selected_only : BoolProperty(name="Selected Only", default=False) @classmethod def poll(cls, context): settings = get_scene_settings() return settings.active_project def invoke(self, context, event): scn = context.scene settings = get_scene_settings() return context.window_manager.invoke_props_dialog(self, width=350) def draw(self, context): scn = context.scene settings = get_scene_settings() layout = self.layout layout.use_property_split = True layout.use_property_decorate = False box = layout.box() col = box.column() col.label(text='Single File') #col.use_property_split = True #col.use_property_decorate = False #col.prop(settings.active_project, "render_single_file") row = col.row() row.prop(settings.active_project, "render_video") row = col.row() row.enabled = settings.active_project.render_video row.prop(settings.active_project, "render_video_template", text='Video Path') row = col.row() row.prop(settings.active_project, "render_audio") row = col.row() row.enabled = settings.active_project.render_audio row.prop(settings.active_project, "render_audio_template", text='Audio Path') #layout.separator() box = layout.box() col = box.column() col.label(text='File per strip') col.prop(settings.active_project, "render_video_per_strip", text='Render Video') row = col.row() row.enabled = settings.active_project.render_video_per_strip row.prop(settings.active_project, "render_video_strip_template", text='Video Path') col.prop(settings.active_project, "render_audio_per_strip", text='Render Audio') row = col.row() row.enabled = settings.active_project.render_audio_per_strip row.prop(settings.active_project, "render_audio_strip_template", text='Audio Path') #row.prop(settings.active_project, "render_sound_format", expand=True) #col.prop(settings, 'channel', text='Channel') #col.prop(self, 'selected_only') def execute(self, context): scn = context.scene settings = get_scene_settings() project = settings.active_project format_data = {**settings.format_data, **project.format_data} start_time = time.perf_counter() if project.render_video: video_path = project.render_video_template.format(**format_data) background_render(output=video_path) if project.render_audio: audio_path = project.render_audio_template.format(**format_data) bpy.ops.sound.mixdown(filepath=audio_path) for strip in get_strips(channel='Shots', selected_only=True): strip_settings = strip.vsetb_strip_settings strip_data = {**format_data, **strip_settings.format_data} if project.render_video_per_strip: strip_render_path = project.render_video_strip_template.format(**strip_data) render_strip(strip, strip_render_path) if project.render_audio_per_strip: audio_render_path = project.render_audio_strip_template.format(**strip_data) render_sound(strip, audio_render_path) self.report({"INFO"}, f'Strips rendered in {time.perf_counter()-start_time} seconds') return {"FINISHED"} class VSETB_OT_export_edl(Operator): bl_idname = "vse_toolbox.export_edl" bl_label = "Export Edl" bl_description = "Export Edl" bl_options = {"REGISTER", "UNDO"} #selected_only : BoolProperty(name="Selected Only", default=False) @classmethod def poll(cls, context): settings = get_scene_settings() return settings.active_project def invoke(self, context, event): scn = context.scene settings = get_scene_settings() return context.window_manager.invoke_props_dialog(self) def draw(self, context): scn = context.scene settings = get_scene_settings() layout = self.layout col = layout.column() col.use_property_split = True col.use_property_decorate = False #col.prop(settings, 'channel', text='Channel') #col.prop(self, 'selected_only') col.prop(settings.active_project, "export_edl_template") def execute(self, context): opentimelineio = install_module('opentimelineio') from opentimelineio.schema import (Clip, Timeline, Track, ExternalReference) from opentimelineio.opentime import (RationalTime, TimeRange) start_time = time.perf_counter() scn = context.scene settings = get_scene_settings() fps = scn.render.fps project = settings.active_project format_data = {**settings.format_data, **project.format_data} output_edl = project.export_edl_template.format(**format_data) output_edl = os.path.abspath(bpy.path.abspath(output_edl)) output_edl = Path(output_edl) timeline = Timeline(output_edl.stem) track = Track(f"Track 1") track.kind = "Video" timeline.tracks.append(track) for strip in get_strips(channel='Shots', selected_only=False): strip_data = {**format_data, **strip.vsetb_strip_settings.format_data} strip_render_path = project.render_strip_template.format(**strip_data) clip = Clip(Path(strip_render_path).name) clip.metadata.setdefault("cmx_3600", {}) clip.metadata['cmx_3600']['reel'] = strip.name #clip.metadata['cmx_3600']['flags'] = 'IS_CLIP' #clip.media_reference = ExternalReference(f'//render/{strip.name}.mov') #clip.media_reference.name = strip.name #clip.available_range = TimeRange(RationalTime(0, fps), RationalTime(strip.frame_duration, fps)) clip.source_range = TimeRange(RationalTime(strip.frame_offset_start, fps), RationalTime(strip.frame_final_duration, fps)) track.append(clip) output_edl.parent.mkdir(exist_ok=True, parents=True) edl = opentimelineio.adapters.write_to_string(timeline, 'cmx_3600') output_edl.write_text(edl) #opentimelineio.adapters.write_to_file(timeline, str(output_edl), 'cmx_3600') self.report({"INFO"}, f'Edl Exported in {time.perf_counter()-start_time} seconds') return {"FINISHED"} classes = ( VSETB_OT_render, VSETB_OT_export_edl ) def register(): for cls in classes: bpy.utils.register_class(cls) def unregister(): for cls in reversed(classes): bpy.utils.unregister_class(cls)