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
 ):
@@ -951,6 +975,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

@@ -1,5 +1,4 @@
 import os
-
 import unreal
 
 from ayon_core.settings import get_project_settings
@@ -131,7 +130,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:
@@ -161,6 +160,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"
         sequence = ar.get_asset_by_object_path(i["sequence"]).get_asset()
 
         sequences = [{
@@ -178,12 +179,15 @@ def start_rendering():
         for seq in sequences:
             subscenes = pipeline.get_subsequences(seq.get('sequence'))
 
-            if subscenes:
+            if subscenes and i["productType"] != "editorial_pkg":
                 for sub_seq in subscenes:
+                    sub_seq_obj = sub_seq.get_sequence()
+                    if sub_seq_obj is None:
+                        continue
                     sequences.append({
-                        "sequence": sub_seq.get_sequence(),
+                        "sequence": sub_seq_obj,
                         "output": (f"{seq.get('output')}/"
-                                   f"{sub_seq.get_sequence().get_name()}"),
+                                f"{sub_seq_obj.get_name()}"),
                         "frame_range": (
                             sub_seq.get_start_frame(), sub_seq.get_end_frame())
                     })
@@ -215,7 +219,6 @@ 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')
             shot_name = render_setting.get('sequence').get_name()