From 9258d860dac295ed3bb0cd63bede7862925a1b62 Mon Sep 17 00:00:00 2001
From: wikoreman <wikoreman@gmail.com>
Date: Mon, 10 Sep 2018 16:24:04 +0200
Subject: [PATCH] Added set_project_settings and set_scene_resolution functions

---
 colorbleed/maya/lib.py | 53 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/colorbleed/maya/lib.py b/colorbleed/maya/lib.py
index bb41b4a738..5371f5abc6 100644
--- a/colorbleed/maya/lib.py
+++ b/colorbleed/maya/lib.py
@@ -1371,6 +1371,7 @@ def get_id_from_history(node):
             return _id
 
 
+# Project settings
 def set_scene_fps(fps, update=True):
     """Set FPS from project configuration
 
@@ -1399,6 +1400,58 @@ def set_scene_fps(fps, update=True):
     cmds.file(modified=True)
 
 
+def set_scene_resolution(resolution, renderer):
+    """Set the render resolution
+
+    Args:
+        resolution(str): <width>x<heigth>, eg: '1920x1080'
+        renderer(str): name of the curren renderer; vray / redshift / arnold
+
+    Returns:
+        bool: True if successful
+
+    """
+
+    control_node = "defaultResolution"
+    cmds.setAttr("defaultRenderGlobals.currentRenderer",
+                 renderer,
+                 type="string")
+
+    width, height = resolution.split("x")
+
+    # Give VRay a helping hand as it is slightly different from the rest
+    if renderer == "vray":
+        control_node = "vraySettings"
+        if not cmds.objExists(type=control_node):
+            cmds.createNode("VRaySettingsNode", name=control_node)
+
+    cmds.setAttr("%s.width" % control_node, int(width))
+    cmds.setAttr("%s.height" % control_node, int(height))
+
+    log.info("Set project resolution to: %s" % resolution)
+
+    return True
+
+
+def set_project_settings():
+    """Apply the project settings from the project definition
+
+    Returns:
+        None
+    """
+
+    # Todo (Wijnand): apply renderer and resolution of project
+
+    # Get project settings
+    data = lib.get_project_data()
+    fps = data.get("fps", None)
+
+    if fps is None:
+        return
+
+    set_scene_fps(fps)
+
+
 # Valid FPS
 def validate_fps():
     """Validate current scene FPS and show pop-up when it is incorrect