autour_de_minuit/vse_toolbox
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

import edl, import movie, import audio

Browse Source
This commit is contained in:
clement.ducarteron
2023-03-16 18:32:17 +01:00
parent 2973fd85d8
commit c75c5469f9
5 changed files with 298 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
operators/operators.py
+152 -18
View File
@@ -17,10 +17,21 @@ from bpy.types import (
Operator,
)
from pathlib import Path
from vse_toolbox.constants import (
EDITS,
MOVIES,
SOUNDS,
EDIT_SUFFIXES,
MOVIE_SUFFIXES,
SOUND_SUFFIXES,
)
from vse_toolbox.sequencer_utils import (
clean_sequencer,
get_shot_sequence,
get_strips,
import_edl,
import_edit,
import_movie,
import_sound,
rename_strips,
render_strips,
set_channels,
@@ -44,10 +55,55 @@ class VSETB_OT_export_csv(Operator):
return {"CANCELLED"}
class VSETB_OT_auto_select_files(Operator):
bl_idname = "import.auto_select_files"
bl_label = "Auto Select"
bl_description = "Auto Select Files"
bl_options = {"REGISTER", "UNDO"}
@classmethod
def poll(cls, context):
return True
def get_items(self, items=[]):
if not items:
return [('NONE', 'None', '', 0)]
return [(e, e, '', i) for i, e in enumerate(sorted(items))]
def execute(self, context):
params = context.space_data.params
directory = Path(params.directory.decode())
EDITS.clear()
MOVIES.clear()
SOUNDS.clear()
edits = []
movies = []
sounds = []
for file_entry in directory.glob('*'):
if file_entry.is_dir():
continue
if file_entry.suffix in EDIT_SUFFIXES:
edits.append(file_entry.name)
elif file_entry.suffix in MOVIE_SUFFIXES:
movies.append(file_entry.name)
elif file_entry.suffix in SOUND_SUFFIXES:
sounds.append(file_entry.name)
EDITS.extend(self.get_items(items=edits))
MOVIES.extend(self.get_items(items=movies))
SOUNDS.extend(self.get_items(items=sounds))
return {'FINISHED'}
class VSETB_OT_import(Operator):
bl_idname = "sequencer.import"
bl_label = "Set Scene"
bl_description = "Set Scene for Breakdown"
bl_label = "Import"
bl_description = "Import Edit"
bl_options = {"REGISTER", "UNDO"}
directory : StringProperty(subtype='DIR_PATH')
@@ -58,29 +114,103 @@ class VSETB_OT_import(Operator):
subtype='FILE_PATH',
)
files : CollectionProperty(type=bpy.types.OperatorFileListElement)
filter_glob: StringProperty(
default='*.edl;*.mov;*.wav',
options={'HIDDEN'}
clean_sequencer : BoolProperty(
name="Clean Sequencer",
default=False,
description="Clean all existing strips in sequencer",
)
import_edit : BoolProperty(name='', default=True)
edit: EnumProperty(name='', items=lambda s, c: EDITS)
import_movie : BoolProperty(name='', default=False)
movie: EnumProperty(name='', items=lambda s, c: MOVIES)
import_sound : BoolProperty(name='', default=False)
sound: EnumProperty(name='', items=lambda s, c: SOUNDS)
@classmethod
def poll(cls, context):
return True
def draw(self, context):
scn = context.scene
settings = get_settings()
layout = self.layout
layout.use_property_split = True
layout.use_property_decorate = False
col = layout.column(align=True)
col.operator('import.auto_select_files', text='Auto Select')
row = self.layout.row(heading="Import Edit", align=True)
row.prop(self, 'import_edit')
sub = row.row()
sub.active = self.import_edit
sub.prop(self, 'edit')
row = self.layout.row(heading="Import Movie", align=True)
row.prop(self, 'import_movie')
sub = row.row()
sub.active = self.import_movie
sub.prop(self, 'movie')
row = self.layout.row(heading="Import Sound", align=True)
row.prop(self, 'import_sound')
sub = row.row()
sub.active = self.import_sound
sub.prop(self, 'sound')
col = layout.column()
col.separator()
col.prop(self, 'clean_sequencer')
def invoke(self, context, event):
context.window_manager.fileselect_add(self)
return {'RUNNING_MODAL'}
def execute(self, context):
otio = install_module('opentimelineio')
for filepath in self.files:
filepath = Path(self.directory, filepath.name)
if filepath.suffix == '.edl':
edl = import_edl(filepath, adapter="cmx_3600")
print('edl: ', edl)
return {"CANCELLED"}
edit_filepath = Path(self.directory, self.edit)
if not edit_filepath.exists():
self.import_edit = False
movie_filepath = Path(self.directory, self.movie)
if not movie_filepath.exists():
self.import_movie = False
sound_filepath = Path(self.directory, self.sound)
if not sound_filepath.exists():
self.import_sound = False
if self.clean_sequencer:
clean_sequencer(
edit=self.import_edit,
movie=self.import_movie,
sound=self.import_sound,
)
if self.import_edit:
print(f'[>.] Loading Edit from: {str(edit_filepath)}')
import_edit(edit_filepath, adapter="cmx_3600")
if self.import_movie:
print(f'[>.] Loading Movie from: {str(movie_filepath)}')
import_movie(movie_filepath)
if self.import_sound:
print(f'[>.] Loading Audio from: {str(sound_filepath)}')
import_sound(sound_filepath)
elif not self.import_sound and self.import_movie:
print(f'[>.] Loading Audio from Movie: {str(movie_filepath)}')
import_sound(movie_filepath)
context.scene.sequence_editor.sequences.update()
return {"FINISHED"}
class VSETB_OT_load_projects(Operator):
@@ -111,7 +241,7 @@ class VSETB_OT_load_projects(Operator):
episode.name = episode_data['name']
episode.id = episode_data['id']
return {'FINISHED'}
return {'FINISHED'}
class VSETB_OT_new_episode(Operator):
@@ -202,7 +332,8 @@ class VSETB_OT_rename(Operator):
scn = context.scene
settings = get_settings()
col = self.layout
layout = self.layout
col = layout.column()
col.use_property_split = True
col.prop(self, 'template')
col.prop(self, 'start_number')
@@ -245,15 +376,17 @@ class VSETB_OT_render(Operator):
scn = context.scene
settings = get_settings()
col = self.layout
layout = self.layout
col = layout.column()
col.use_property_split = True
col.use_property_decorate = False
col.prop(settings, 'channels', text='Channel')
col.prop(settings, 'channel', text='Channel')
col.prop(self, 'selected_only')
def execute(self, context):
scn = context.scene
strips = get_strips(channel=self.channel_name, selected_only=self.selected_only)
scn = context.scene
settings = get_settings()
strips = get_strips(channel=settings.channel, selected_only=self.selected_only)
render_strips(strips)
@@ -288,6 +421,7 @@ class VSETB_OT_set_scene(Operator):
classes=(
VSETB_OT_auto_select_files,
VSETB_OT_export_csv,
VSETB_OT_import,
VSETB_OT_load_projects,