change ui

Author: inwoods

Assets/Integrations/Autodesk/maya/scripts/unityOneClick/commands.py

@@ -38,7 +38,8 @@ class BaseCommand(OpenMayaMPx.MPxCommand, LoggerMixin):
     """
     Base class for UnityOneClick Plugin Commands.
     """
-    kDefaultIcon = 'unity_100.png'
+    kIconPath = ""
+    kFamilyIcon = 'unity.png'
     kFamilyLabel = "The UnityOneClick plugin allows you to reliably exchange and review your work between Maya and Unity."
 
     def __init__(self):
@@ -92,11 +93,11 @@ def loadUnityFbxExportSettings(self):
 
     @classmethod
     def iconPath(cls):
-        return ""
+        return cls.kIconPath
 
     @classmethod
     def familyIconPath(cls):
-        return maya.mel.eval('$tempPath = languageResourcePath("{0}");'.format(cls.kDefaultIcon))
+        return maya.mel.eval('$tempPath = languageResourcePath("{0}");'.format(cls.kFamilyIcon))
 
     def storeAttribute(self, node, attr, attrValue, attrType="string"):
         if not maya.mel.eval('attributeExists "{0}" "{1}"'.format(attr, node)):
@@ -117,6 +118,7 @@ class importCmd(BaseCommand):
     
     @ingroup UnityCommands
     """
+    kIconPath = "import.png"
     kLabel = 'Import FBX file from Unity Project and auto-configure for publishing'
     kShortLabel = 'Import'
     kCmdName = "{}Import".format(version.pluginPrefix())
@@ -145,7 +147,7 @@ def syntaxCreator(cls):
     @classmethod
     def scriptCmd(cls):
         return
-    
+
     def beforeImport(self, retCode, file, clientData):
         # store path and filename
         self._tempPath = file.resolvedPath()
@@ -242,6 +244,7 @@ class reviewCmd(BaseCommand):
         
     @ingroup UnityCommands
     """
+    kIconPath = "preview.png"
     kLabel = 'Review Model in Unity'
     kShortLabel = 'Review'
     kCmdName = "{}Review".format(version.pluginPrefix())
@@ -332,6 +335,7 @@ class publishCmd(BaseCommand):
         
     @ingroup UnityCommands
     """
+    kIconPath = "export.png"
     kLabel = 'Publish Model to Unity'
     kShortLabel = 'Publish'
     kCmdName = "{}Publish".format(version.pluginPrefix())

Assets/Integrations/Autodesk/maya/scripts/unityOneClick/ui.py

@@ -99,13 +99,13 @@ def installMenu():
                        label=commands.reviewCmd.kShortLabel, 
                        annotation=commands.reviewCmd.kLabel, 
                        command=commands.reviewCmd.kScriptCommand, 
-                       image=commands.importCmd.iconPath(),
+                       image=commands.reviewCmd.iconPath(),
                        version=whatsNewVersion())
     maya.cmds.menuItem(parent=kMenuName, 
                        label=commands.publishCmd.kShortLabel, 
                        annotation=commands.publishCmd.kLabel, 
                        command=commands.publishCmd.kScriptCommand, 
-                       image=commands.importCmd.iconPath(),
+                       image=commands.publishCmd.iconPath(),
                        version=whatsNewVersion())
 
 def uninstallMenu():