Merge pull request #1325 from pypeclub/feature/3.0_tvpaint_initial_project_settings

Set initial project settings

openpype/hosts/tvpaint/api/__init__.py

@@ -1,14 +1,15 @@
 import os
 import logging
 
-from avalon.tvpaint.communication_server import register_localization_file
-from avalon.tvpaint import pipeline
 import avalon.api
 import pyblish.api
+from avalon.tvpaint import pipeline
+from avalon.tvpaint.communication_server import register_localization_file
+from .lib import set_context_settings
 
 from openpype.hosts import tvpaint
 
-log = logging.getLogger("openpype.hosts.tvpaint")
+log = logging.getLogger(__name__)
 
 HOST_DIR = os.path.dirname(os.path.abspath(tvpaint.__file__))
 PLUGINS_DIR = os.path.join(HOST_DIR, "plugins")
@@ -34,6 +35,18 @@ def on_instance_toggle(instance, old_value, new_value):
         pipeline._write_instances(current_instances)
 
 
+def initial_launch():
+    # Setup project settings if its the template that's launched.
+    # TODO also check for template creation when it's possible to define
+    #   templates
+    last_workfile = os.environ.get("AVALON_LAST_WORKFILE")
+    if not last_workfile or os.path.exists(last_workfile):
+        return
+
+    log.info("Setting up project...")
+    set_context_settings()
+
+
 def install():
     log.info("OpenPype - Installing TVPaint integration")
     localization_file = os.path.join(HOST_DIR, "resources", "avalon.loc")
@@ -49,6 +62,8 @@ def install():
     if on_instance_toggle not in registered_callbacks:
         pyblish.api.register_callback("instanceToggled", on_instance_toggle)
 
+    avalon.api.on("application.launched", initial_launch)
+
 
 def uninstall():
     log.info("OpenPype - Uninstalling TVPaint integration")

openpype/hosts/tvpaint/api/lib.py

@@ -1,5 +1,8 @@
 from PIL import Image
 
+import avalon.io
+from avalon.tvpaint.lib import execute_george
+
 
 def composite_images(input_image_paths, output_filepath):
     """Composite images in order from passed list.
@@ -18,3 +21,64 @@ def composite_images(input_image_paths, output_filepath):
         else:
             img_obj.alpha_composite(_img_obj)
     img_obj.save(output_filepath)
+
+
+def set_context_settings(asset_doc=None):
+    """Set workfile settings by asset document data.
+
+    Change fps, resolution and frame start/end.
+    """
+    if asset_doc is None:
+        # Use current session asset if not passed
+        asset_doc = avalon.io.find_one({
+            "type": "asset",
+            "name": avalon.io.Session["AVALON_ASSET"]
+        })
+
+    project_doc = avalon.io.find_one({"type": "project"})
+
+    framerate = asset_doc["data"].get("fps")
+    if framerate is None:
+        framerate = project_doc["data"].get("fps")
+
+    if framerate is not None:
+        execute_george(
+            "tv_framerate {} \"timestretch\"".format(framerate)
+        )
+    else:
+        print("Framerate was not found!")
+
+    width_key = "resolutionWidth"
+    height_key = "resolutionHeight"
+
+    width = asset_doc["data"].get(width_key)
+    height = asset_doc["data"].get(height_key)
+    if width is None or height is None:
+        width = project_doc["data"].get(width_key)
+        height = project_doc["data"].get(height_key)
+
+    if width is None or height is None:
+        print("Resolution was not found!")
+    else:
+        execute_george("tv_resizepage {} {} 0".format(width, height))
+
+    frame_start = asset_doc["data"].get("frameStart")
+    frame_end = asset_doc["data"].get("frameEnd")
+
+    if frame_start is None or frame_end is None:
+        print("Frame range was not found!")
+        return
+
+    handles = asset_doc["data"].get("handles") or 0
+    handle_start = asset_doc["data"].get("handleStart")
+    handle_end = asset_doc["data"].get("handleEnd")
+
+    if handle_start is None or handle_end is None:
+        handle_start = handles
+        handle_end = handles
+
+    frame_start -= int(handle_start)
+    frame_end += int(handle_end)
+
+    execute_george("tv_markin {} set".format(frame_start - 1))
+    execute_george("tv_markout {} set".format(frame_end - 1))