Editorial_pkg: publishing sequence into OTIO timeline with rendered media

client/ayon_unreal/api/lib.py

@@ -5,7 +5,8 @@
 )
 from ayon_unreal.api.pipeline import (
     get_camera_tracks,
-    ls
+    ls,
+    UNREAL_VERSION
 )
 from ayon_core.pipeline.context_tools import get_current_folder_entity
 import ayon_api
@@ -306,3 +307,42 @@ def import_animation(
                     sequence.get_playback_end())
                 sec_params = section.get_editor_property('params')
                 sec_params.set_editor_property('animation', animation)
+
+
+def get_shot_track_names(sel_objects=None, get_name=True):
+    selection = [
+        a for a in sel_objects
+        if a.get_class().get_name() == "LevelSequence"
+    ]
+    if (
+        UNREAL_VERSION.major == 5
+        and UNREAL_VERSION.minor > 4
+    ):
+        sub_sequence_tracks = [
+            track for sel in selection for track in
+            sel.find_tracks_by_type(unreal.MovieSceneSubTrack)
+        ]
+    else:
+        sub_sequence_tracks = [
+            track for sel in selection for track in
+            sel.find_master_tracks_by_type(unreal.MovieSceneSubTrack)
+        ]
+
+    if get_name:
+        return [shot_tracks.get_display_name() for shot_tracks in
+                sub_sequence_tracks]
+    else:
+        return [shot_tracks for shot_tracks in sub_sequence_tracks]
+
+
+def get_shot_tracks(members):
+    ar = unreal.AssetRegistryHelpers.get_asset_registry()
+    selected_sequences = [
+        ar.get_asset_by_object_path(member).get_asset() for member in members
+    ]
+    return get_shot_track_names(selected_sequences, get_name=False)
+
+
+def get_screen_resolution():
+    game_user_settings = unreal.GameUserSettings.get_game_user_settings()
+    return game_user_settings.get_screen_resolution()

client/ayon_unreal/api/pipeline.py

@@ -526,6 +526,30 @@ def get_subsequences(sequence: unreal.LevelSequence):
     return []
 
 
+def get_movie_shot_tracks(sequence: unreal.LevelSequence):
+    """Get list of movie shot tracks from sequence.
+
+    Args:
+        sequence (unreal.LevelSequence): Sequence
+
+    Returns:
+        list(unreal.LevelSequence): List of movie shot tracks
+
+    """
+    tracks = sequence.find_master_tracks_by_type(unreal.MovieSceneSubTrack)
+    subscene_track = next(
+        (
+            t
+            for t in tracks
+            if t.get_class() == unreal.MovieSceneCinematicShotTrack.static_class()
+        ),
+        None,
+    )
+    if subscene_track is not None and subscene_track.get_sections():
+        return subscene_track.get_sections()
+    return []
+
+
 def set_sequence_hierarchy(
     seq_i, seq_j, max_frame_i, min_frame_j, max_frame_j, map_paths
 ):
@@ -1000,7 +1024,7 @@ def get_sequence(files):
         Optional[list[str]]: file sequence.
 
     """
-    collections, _remainder = clique.assemble(
+    collections, _ = clique.assemble(
         files,
         patterns=[clique.PATTERNS["frames"]],
         minimum_items=1)
@@ -1013,6 +1037,38 @@ def get_sequence(files):
     return [os.path.basename(filename) for filename in collections[0]]
 
 
+def get_sequence_for_otio(files):
+    """Get sequence from filename.
+
+    This will only return files if they exist on disk as it tries
+    to collect the sequence using the filename pattern and searching
+    for them on disk.
+
+    Supports negative frame ranges like -001, 0000, 0001 and -0001,
+    0000, 0001.
+
+    Arguments:
+        files (str): List of files
+
+    Returns:
+        Optional[list[str]]: file sequence.
+        Optional[str]: file head.
+
+    """
+    base_filenames = [os.path.basename(filename) for filename in files]
+    collections, _remainder = clique.assemble(
+        base_filenames,
+        patterns=[clique.PATTERNS["frames"]],
+        minimum_items=1)
+
+    if len(collections) > 1:
+        raise ValueError(
+            f"Multiple collections found for {collections}. "
+            "This is a bug.")
+    filename_padding = collections[0].padding
+    return filename_padding
+
+
 def find_camera_actors_in_camera_tracks(sequence) -> list[Any]:
     """Find the camera actors in the tracks from the Level Sequence
 

client/ayon_unreal/api/rendering.py

@@ -169,7 +169,7 @@ def start_rendering():
 
     for i in instances:
         data = pipeline.parse_container(i.get_path_name())
-        if data["productType"] == "render":
+        if data["productType"] == "render" or "editorial_pkg":
             inst_data.append(data)
 
     try:
@@ -199,6 +199,8 @@ def start_rendering():
     current_level_name = current_level.get_outer().get_path_name()
 
     for i in inst_data:
+        if i["productType"] == "editorial_pkg":
+            render_dir = f"{root}/{project_name}/editorial_pkg"
         # for some reason the instance data has strings, convert them
         # back to their original types
         render_preset = ast.literal_eval(i["creator_attributes"]).get(
@@ -226,7 +228,7 @@ def start_rendering():
         for seq in sequences:
             subscenes = pipeline.get_subsequences(seq.get('sequence'))
 
-            if subscenes:
+            if subscenes and i["productType"] != "editorial_pkg":
                 sequences.extend(
                     {
                         "sequence": sub_seq.get_sequence(),
@@ -241,6 +243,24 @@ def start_rendering():
                     }
                     for sub_seq in subscenes
                 )
+            # remove all the codes unnecssary for the editorial package
+            elif subscenes:
+                for sub_seq in subscenes:
+                    sub_seq_obj = sub_seq.get_sequence()
+                    if sub_seq_obj is None:
+                        continue
+                    # exclude camera-related sequence
+                    elif "_camera" in sub_seq_obj.get_name():
+                        continue
+                    render_list.append({
+                        "sequence": seq.get('sequence'),
+                        "output": (f"{seq.get('output')}"),
+                        "frame_range": (
+                            sub_seq.get_start_frame(),
+                            sub_seq.get_end_frame(),
+                        )
+                    }
+                )
             elif "_camera" not in seq.get('sequence').get_name():
                 render_list.append(seq)
 
@@ -269,8 +289,7 @@ def start_rendering():
             # read in the job's OnJobFinished callback. We could,
             # for instance, pass the AyonPublishInstance's path to the job.
             # job.user_data = ""
-
-            output_dir = render_setting.get('output')
+            output_name = render_setting.get('output')
             shot_name = render_setting.get('sequence').get_name()
 
             settings = job_config.find_or_add_setting_by_class(
@@ -279,8 +298,9 @@ def start_rendering():
             settings.custom_start_frame = render_setting.get("frame_range")[0]
             settings.custom_end_frame = render_setting.get("frame_range")[1]
             settings.use_custom_playback_range = True
+            # make sure all sequences share the same filename
             settings.file_name_format = f"{shot_name}" + ".{frame_number}"
-            settings.output_directory.path = f"{render_dir}/{output_dir}"
+            settings.output_directory.path = f"{render_dir}/{output_name}"
 
             job_config.find_or_add_setting_by_class(
                 unreal.MoviePipelineDeferredPassBase)