Merge pull request #16 from takavfx/develop

Merge for release v4.2.0

Author: takavfx

README.md

@@ -9,10 +9,11 @@ Bento is a toolset for Houdini.
 
 ### Shelf tool
 
-![alt tag](docs/img/ss_shelf_tool_0002.png)
+![alt tag](docs/img/ss_shelf_tool.png)
 
 * [Cache Dependency](docs/cacheDependency.md)
 * [Creat Delayed Load Procedural](docs/create-dlp.md)
+* [Color Nodes](docs/color-nodes.md)
 
 ### Python Panel
 

docs/color-nodes.md

@@ -1,7 +1,14 @@
-# Cache Manager
+# Color Nodes
 
 ## Feature
 
+### Shelf Tool
+
+Color nodes when shelf-tool button clicked.
+If you have already selected nodes, this affects only for them.
+
+### OnCreated Script
+
 Color nodes when nodes are created.
 
 ## Detail

docs/img/ss_shelf_tool.png

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+#-------------------------------------------------------------------------------
+## Description
+"""
+This script for color nodes shelf tool and also used in OnCreated script.
+"""
+#-------------------------------------------------------------------------------
+
+import hou
+
+manager_color = (0, 0.4, 1)
+generator_color = (0.8, 0.8, 0.8)
+
+# Common node type colors.  These colors affect these types in all contexts.
+# Node names are matched identically with keys.
+common_type_colors = {
+
+    ### Gray ###
+    "null": (0.36, 0.36, 0.36),
+    "cam": (0.36, 0.36, 0.36),
+    "merge": (0.36, 0.36, 0.36),
+    "output": (0.36, 0.36, 0.36),
+
+    ### Blue: Disk Output ###
+    "df_alembic_export": (0, 0.4, 1),
+    "PRT_ROPDriver": (0, 0.4, 1),
+    "rop_geometry": (0, 0.4, 1),
+    "rop_alembic": (0, 0.4, 1),
+    "ropnet": (0, 0.4, 1),
+
+    ### Yellow: Disk Input ###
+    "file": (1, 0.8, 0),
+    "filecache": (1, 0.8, 0),
+    "filemerge": (1, 0.8, 0),
+    "alembic": (1, 0.8, 0),
+    "alembicarchive": (1, 0.8, 0),
+
+    ### Purple Blue: Dop I/O ###
+    "dopimport": (0.6, 0.6, 1),
+    "dopimportfield": (0.6, 0.6, 1),
+    "dopimportrecords": (0.6, 0.6, 1),
+    "dopio": (0.6, 0.6, 1),
+
+    ### Purple ###
+    "dopnet": (0.4, 0, 0.6),
+
+    ### Green: Geometry Fetch ###
+    "object_merge": (0, 0.4, 0),
+
+    ### Dark Green ###
+    "geo": (0, 0.267, 0)
+
+
+}
+
+def run():
+    if len(hou.selectedNodes()) is 0:
+        nodes = hou.pwd().allSubChildren()
+    else:
+        nodes = hou.selectedNodes()
+
+    for node in nodes:
+        node_type = node.type()
+        node_type_name = node_type.name()
+        type_category_name = node_type.category().name().lower()
+
+        node_color = None
+
+        if node_type.isManager():
+            node_color = manager_color
+
+        elif node_type_name in common_type_colors:
+            node_color = common_type_colors[node_type_name]
+
+        if node_color is not None:
+            node.setColor(hou.Color(node_color))

docs/img/ss_shelf_tool_0001.png

@@ -3,74 +3,27 @@
 ## Description
 """
 This script runs when Houdini nodes are created.
-
-This code is basaed on
 """
 #-------------------------------------------------------------------------------
 
-manager_color = (0, 0.4, 1)
-generator_color = (0.8, 0.8, 0.8)
-
-# Common node type colors.  These colors affect these types in all contexts.
-# Node names are matched identically with keys.
-common_type_colors = {
-
-    ### Gray ###
-    "null": (0.36, 0.36, 0.36),
-    "cam": (0.36, 0.36, 0.36),
-    "merge": (0.36, 0.36, 0.36),
-    "output": (0.36, 0.36, 0.36),
-
-    ### Blue: Disk Output ###
-    "df_alembic_export": (0, 0.4, 1),
-    "PRT_ROPDriver": (0, 0.4, 1),
-    "rop_geometry": (0, 0.4, 1),
-    "rop_alembic": (0, 0.4, 1),
-    "ropnet": (0, 0.4, 1),
-
-    ### Yellow: Disk Input ###
-    "file": (1, 0.8, 0),
-    "filecache": (1, 0.8, 0),
-    "filemerge": (1, 0.8, 0),
-    "alembic": (1, 0.8, 0),
-    "alembicarchive": (1, 0.8, 0),
-
-    ### Purple Blue: Dop I/O ###
-    "dopimport": (0.6, 0.6, 1),
-    "dopimportfield": (0.6, 0.6, 1),
-    "dopimportrecords": (0.6, 0.6, 1),
-    "dopio": (0.6, 0.6, 1),
-
-    ### Purple ###
-    "dopnet": (0.4, 0, 0.6),
-
-    ### Green: Geometry Fetch ###
-    "object_merge": (0, 0.4, 0),
-
-    ### Dark Green ###
-    "geo": (0, 0.267, 0)
-
-
-}
+import colornodes
 
 # Node type information.
 node = kwargs["node"]
 node_type = node.type()
 node_type_name = node_type.name()
-type_category_name = node_type.category().name().lower()
-
 
 # Start with no node color:
 node_color = None
 
 # Manager nodes (sopnet, chopnet, ropnet, etc).
 if node_type.isManager():
-    node_color = manager_color
+    node_color = colornodes.manager_color
 
 # Common node types (nulls, switches, etc) are colored the same across
 # different contexts.
-elif node_type_name in common_type_colors:
-    node_color = common_type_colors[node_type_name]
+elif node_type_name in colornodes.common_type_colors:
+    node_color = colornodes.common_type_colors[node_type_name]
 
 
 # If we found a color mapping, set the color of the node.

docs/img/ss_shelf_tool_0002.png

@@ -8,6 +8,7 @@
   <toolshelf name="bento_shelf" label="Bento">
     <memberTool name="cr_cacheDendency"/>
     <memberTool name="create_dlp"/>
+    <memberTool name="color_nodes"/>
   </toolshelf>
 
   <tool name="cr_cacheDendency" label="cCDP" icon="SOP_file">
@@ -25,4 +26,9 @@ reload(createDlp)
 
 createDlp.main()]]></script>
   </tool>
+
+  <tool name="color_nodes" label="CNodes" icon="COP2_colorwheel">
+    <script scriptType="python"><![CDATA[import colornodes
+colornodes.run()]]></script>
+  </tool>
 </shelfDocument>