Merge pull request #135 from Unity-Technologies/Uni-24798-maya-menuitem-icons

Uni-24798-maya-menuitem-icons

Author: vkovec

Assets/Integrations/Autodesk/maya/icons/unity_100.png

@@ -38,6 +38,8 @@ class BaseCommand(OpenMayaMPx.MPxCommand, LoggerMixin):
     """
     Base class for UnityOneClick Plugin Commands.
     """
+    kDefaultIcon = 'unity_100.png'
+    
     def __init__(self):
         OpenMayaMPx.MPxCommand.__init__(self)
         LoggerMixin.__init__(self)
@@ -86,6 +88,14 @@ def loadUnityFbxExportSettings(self):
 
         maya.mel.eval(contents)
         return True
+    
+    @classmethod
+    def iconPath(cls):
+        return ""
+    
+    @classmethod
+    def familyIconPath(cls):
+        return maya.mel.eval('$tempPath = languageResourcePath("{0}");'.format(cls.kDefaultIcon))
 
     def storeAttribute(self, node, attr, attrValue, attrType="string"):
         if not maya.mel.eval('attributeExists "{0}" "{1}"'.format(attr, node)):

Assets/Integrations/Autodesk/maya/icons/unity_150.png

@@ -65,12 +65,39 @@ def installMenu():
     """
     parentMenu = getParentMenu()
 
-    maya.cmds.menuItem(kMenuDivider, divider=True, longDivider=False, insertAfter=kMenuInsertAfter, parent=parentMenu, version=kMayaVersionAdded)
-    maya.cmds.menuItem(kMenuName, parent=parentMenu, insertAfter=kMenuDivider, subMenu=True, label=kMenuLabel, tearOff=True, version=kMayaVersionAdded)
+    maya.cmds.menuItem(kMenuDivider, 
+                       divider=True, 
+                       longDivider=False, 
+                       insertAfter=kMenuInsertAfter, 
+                       parent=parentMenu, 
+                       version=kMayaVersionAdded)
+    maya.cmds.menuItem(kMenuName, 
+                       parent=parentMenu, 
+                       insertAfter=kMenuDivider, 
+                       image=commands.importCmd.familyIconPath(),
+                       subMenu=True, 
+                       label=kMenuLabel, 
+                       tearOff=True, 
+                       version=kMayaVersionAdded)
 
-    maya.cmds.menuItem(parent=kMenuName, label=commands.importCmd.kShortLabel, command=commands.importCmd.kScriptCommand, version=kMayaVersionAdded)
-    maya.cmds.menuItem(parent=kMenuName, label=commands.reviewCmd.kShortLabel, command=commands.reviewCmd.kScriptCommand, version=kMayaVersionAdded)
-    maya.cmds.menuItem(parent=kMenuName, label=commands.publishCmd.kShortLabel, command=commands.publishCmd.kScriptCommand, version=kMayaVersionAdded)
+    maya.cmds.menuItem(parent=kMenuName, 
+                       label=commands.importCmd.kShortLabel, 
+                       annotation=commands.importCmd.kLabel, 
+                       command=commands.importCmd.kScriptCommand,
+                       image=commands.importCmd.iconPath(),
+                       version=kMayaVersionAdded)
+    maya.cmds.menuItem(parent=kMenuName, 
+                       label=commands.reviewCmd.kShortLabel, 
+                       annotation=commands.reviewCmd.kLabel, 
+                       command=commands.reviewCmd.kScriptCommand, 
+                       image=commands.importCmd.iconPath(),
+                       version=kMayaVersionAdded)
+    maya.cmds.menuItem(parent=kMenuName, 
+                       label=commands.publishCmd.kShortLabel, 
+                       annotation=commands.publishCmd.kLabel, 
+                       command=commands.publishCmd.kScriptCommand, 
+                       image=commands.importCmd.iconPath(),
+                       version=kMayaVersionAdded)
 
 def uninstallMenu():
     """