Merge pull request #249 from tokejepsen/2.x/feature/harmony_ensure_scene_settings

Ensure scene settings on application launch.

pype/hosts/harmony/__init__.py

@@ -1,7 +1,87 @@
 import os
+import sys
 
 from avalon import api, harmony
+from avalon.vendor import Qt
 import pyblish.api
+from pype import lib
+
+
+def ensure_scene_settings():
+    asset_data = lib.get_asset()["data"]
+    fps = asset_data.get("fps")
+    frame_start = asset_data.get("frameStart")
+    frame_end = asset_data.get("frameEnd")
+    resolution_width = asset_data.get("resolutionWidth")
+    resolution_height = asset_data.get("resolutionHeight")
+
+    settings = {
+        "fps": fps,
+        "frameStart": frame_start,
+        "frameEnd": frame_end,
+        "resolutionWidth": resolution_width,
+        "resolutionHeight": resolution_height
+    }
+
+    invalid_settings = []
+    valid_settings = {}
+    for key, value in settings.items():
+        if value is None:
+            invalid_settings.append(key)
+        else:
+            valid_settings[key] = value
+
+    # Warn about missing attributes.
+    print("Starting new QApplication..")
+    app = Qt.QtWidgets.QApplication(sys.argv)
+
+    message_box = Qt.QtWidgets.QMessageBox()
+    message_box.setIcon(Qt.QtWidgets.QMessageBox.Warning)
+    msg = "Missing attributes:"
+    if invalid_settings:
+        for item in invalid_settings:
+            msg += f"\n{item}"
+        message_box.setText(msg)
+        message_box.exec_()
+
+    # Garbage collect QApplication.
+    del app
+
+    func = """function func(args)
+    {
+        if (args[0]["fps"])
+        {
+            scene.setFrameRate(args[0]["fps"]);
+        }
+        if (args[0]["frameStart"] && args[0]["frameEnd"])
+        {
+            var duration = args[0]["frameEnd"] - args[0]["frameStart"] + 1
+            if (frame.numberOf() > duration)
+            {
+                frame.remove(
+                    duration, frame.numberOf() - duration
+                );
+            }
+            if (frame.numberOf() < duration)
+            {
+                frame.insert(
+                    duration, duration - frame.numberOf()
+                );
+            }
+
+            scene.setStartFrame(1);
+            scene.setStopFrame(duration);
+        }
+        if (args[0]["resolutionWidth"] && args[0]["resolutionHeight"])
+        {
+            scene.setDefaultResolution(
+                args[0]["resolutionWidth"], args[0]["resolutionHeight"], 41.112
+            )
+        }
+    }
+    func
+    """
+    harmony.send({"function": func, "args": [valid_settings]})
 
 
 def install():
@@ -23,10 +103,13 @@ def install():
         api.Creator, os.path.join(plugins_directory, "create")
     )
 
+    # Register callbacks.
     pyblish.api.register_callback(
         "instanceToggled", on_pyblish_instance_toggled
     )
 
+    api.on("application.launched", ensure_scene_settings)
+
 
 def on_pyblish_instance_toggled(instance, old_value, new_value):
     """Toggle node enabling on instance toggles."""