From 353c962617c47cefe63bf6c6e02aecb64792e1f5 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Thu, 5 Nov 2020 14:01:35 +0100
Subject: [PATCH] implemented render pass creator

---
 .../tvpaint/create/create_render_pass.py      | 85 +++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 pype/plugins/tvpaint/create/create_render_pass.py

diff --git a/pype/plugins/tvpaint/create/create_render_pass.py b/pype/plugins/tvpaint/create/create_render_pass.py
new file mode 100644
index 0000000000..fa51dcad26
--- /dev/null
+++ b/pype/plugins/tvpaint/create/create_render_pass.py
@@ -0,0 +1,85 @@
+import tempfile
+import avalon.io
+from avalon.tvpaint import pipeline, lib
+
+
+class CreateRenderPass(pipeline.TVPaintCreator):
+    name = "render_pass"
+    label = "Render Pass"
+    family = "renderPass"
+    icon = "cube"
+    defaults = ["Main"]
+
+    def process(self):
+        instances = pipeline.list_instances()
+        layers_data = lib.layers_data()
+        group_ids = set()
+        selected_layers = []
+        for layer in layers_data:
+            if layer["selected"]:
+                selected_layers.append(layer)
+                group_ids.add(layer["group_id"])
+
+        if not selected_layers:
+            raise AssertionError("Nothing is selected.")
+
+        if len(group_ids) != 1:
+            raise AssertionError("More than one group is in selection.")
+
+        group_id = tuple(group_ids)[0]
+
+        beauty_instance = None
+        for instance in instances:
+            if (
+                instance["family"] == "renderLayer"
+                and instance["group_id"] == group_id
+            ):
+                beauty_instance = instance
+                break
+
+        if beauty_instance is None:
+            raise AssertionError("Beauty pass does not exist yet.")
+
+        family = self.data["family"]
+        name = self.data["subset"]
+        # Is this right way how to get name?
+        name = name[len(family):]
+
+        self.data["group_id"] = group_id
+        self.data["name"] = name
+
+        layer_ids = [layer["id"] for layer in selected_layers]
+        self.data["layer_ids"] = layer_ids
+
+        beauty_subset_name = beauty_instance["subset"]
+        beauty_pass_name = "beauty"
+        if not beauty_subset_name.endswith(beauty_pass_name):
+            raise AssertionError(
+                "BUG: Beauty subset name does not end with \"{}\"".format(
+                    beauty_pass_name
+                )
+            )
+        subset_name = beauty_subset_name[:-len(beauty_pass_name)] + name
+        self.data["subset"] = subset_name
+
+        existing_instance = None
+        existing_instance_idx = None
+        for idx, instance in enumerate(instances):
+            if (
+                instance["family"] == family
+                and instance["group_id"] == group_id
+                and instance["name"] == name
+            ):
+                existing_instance = instance
+                existing_instance_idx = idx
+                break
+
+        if existing_instance is not None:
+            if existing_instance == self.data:
+                self.log.info("Instance to create is same. Did nothing.")
+                return
+            instances[existing_instance_idx] = self.data
+        else:
+            instances.append(self.data)
+
+        self.write_instances(instances)