From 49fe164c12ae52c11815d4a06d6fad9ed2b7e841 Mon Sep 17 00:00:00 2001
From: iLLiCiTiT <jakub.trllo@gmail.com>
Date: Fri, 26 Jun 2020 14:55:46 +0200
Subject: [PATCH] implemented view for instances

---
 pype/tools/pyblish_pype/view.py | 68 ++++++++++++++++++++++++++++++++-
 1 file changed, 67 insertions(+), 1 deletion(-)

diff --git a/pype/tools/pyblish_pype/view.py b/pype/tools/pyblish_pype/view.py
index 0d129c6dff..e4bc3a483e 100644
--- a/pype/tools/pyblish_pype/view.py
+++ b/pype/tools/pyblish_pype/view.py
@@ -1,6 +1,6 @@
 from Qt import QtCore, QtWidgets
 from . import model
-from .constants import Roles
+from .constants import Roles, EXPANDER_WIDTH
 # Imported when used
 widgets = None
 
@@ -155,6 +155,72 @@ class PluginView(OverviewView):
         return super(PluginView, self).mouseReleaseEvent(event)
 
 
+class InstanceView(OverviewView):
+    def __init__(self, parent=None):
+        super(InstanceView, self).__init__(parent)
+        self.viewport().setMouseTracking(True)
+
+    def mouseMoveEvent(self, event):
+        index = self.indexAt(event.pos())
+        if index.data(Roles.TypeRole) == model.GroupType:
+            self.update(index)
+
+    def item_expand(self, index, expand=None):
+        if expand is None:
+            expand = not self.isExpanded(index)
+
+        if expand:
+            self.expand(index)
+        else:
+            self.collapse(index)
+
+    def group_toggle(self, index):
+        model = index.model()
+
+        chilren_indexes_checked = []
+        chilren_indexes_unchecked = []
+        for idx in range(model.rowCount(index)):
+            child_index = model.index(idx, 0, index)
+            if not child_index.data(Roles.IsEnabledRole):
+                continue
+
+            if child_index.data(QtCore.Qt.CheckStateRole):
+                chilren_indexes_checked.append(child_index)
+            else:
+                chilren_indexes_unchecked.append(child_index)
+
+        if chilren_indexes_checked:
+            to_change_indexes = chilren_indexes_checked
+            new_state = False
+        else:
+            to_change_indexes = chilren_indexes_unchecked
+            new_state = True
+
+        for index in to_change_indexes:
+            model.setData(index, new_state, QtCore.Qt.CheckStateRole)
+            self.toggled.emit(index, new_state)
+
+    def mouseReleaseEvent(self, event):
+        if event.button() == QtCore.Qt.LeftButton:
+            indexes = self.selectionModel().selectedIndexes()
+            if len(indexes) == 1:
+                index = indexes[0]
+                pos_index = self.indexAt(event.pos())
+                if index == pos_index:
+                    # If instance or Plugin
+                    if index.data(Roles.TypeRole) == model.InstanceType:
+                        if event.pos().x() < 20:
+                            self.toggled.emit(index, None)
+                        elif event.pos().x() > self.width() - 20:
+                            self.show_perspective.emit(index)
+                    else:
+                        if event.pos().x() < EXPANDER_WIDTH:
+                            self.item_expand(index)
+                        else:
+                            self.group_toggle(index)
+                            self.item_expand(index, True)
+        return super(InstanceView, self).mouseReleaseEvent(event)
+
 
 class TerminalView(QtWidgets.QTreeView):
     # An item is requesting to be toggled, with optional forced-state