Release v0.0.1

Author: DaBaptisteFraboul

Piston_UI.py

@@ -27,52 +27,51 @@ def abspath(path):
     'generatePiston': abspath('icons/generatePiston.png'),
     'deleteNodes': abspath('icons/deleteNodes.png'),
     'pistonSolverNode': abspath('icons/pistonSolverNode.png'),
-    'pistonVectorLength': abspath('icons/pistonVectorLengthNode.png')
+    'pistonVectorLength': abspath('icons/pistonVectorLengthNode.png'),
+    'flushUndo':abspath('icons/flushUndo.png')
 }
 
 # Buttons commands for the shelf
 
-def generateCommand():
-    cmds.generatePiston()
+def generateCommand() :
+    pm.generatePiston()
 
 
 def deletePluginNodes():
-    """
-    Delete all the custom nodetypes created by the plugin
-    :return:
-    """
     plugin_nodetypes = [
         'pistonVectorLength',
         'pistonNode'
     ]
-
     for type in plugin_nodetypes:
         nodes = pm.ls(type=type)
         if nodes:
             for i in range(len(nodes) - 1, -1, -1):
                 if pm.objExists(nodes[i]):
                     pm.delete(nodes[i])
             print(type, ' : {} nodes deleted'.format(len(nodes)))
+    pm.flushUndo()
+
 
+def createSolver():
+    pm.createNode('pistonNode')
 
-def createSolver() :
-    cmds.createNode('pistonNode')
+def flushUndo():
+    pm.flushUndo()
 
 def createVectorLength():
-    cmds.createNode('pistonVectorLength')
+    pm.createNode('pistonVectorLength')
 
 # Generate the plugin Shelf
 
 def generate_shelf():
-
     # If the shelf exists delete it
     if pm.shelfLayout('Piston plugin', exists=True):
         print('Refreshing old shelf')
         if pm.shelfLayout('Piston plugin', query=True, ca=True):
             for each in pm.shelfLayout('Piston plugin', query=True, ca=True):
                 print(each)
                 pm.deleteUI(each)
-        pm.deleteUI('Piston plugin')
+        pm.deleteUI('Piston_plugin')
 
     # Create the shelf
     plugin_shelf = pm.shelfLayout('Piston plugin', parent='ShelfLayout')
@@ -93,6 +92,13 @@ def generate_shelf():
                    iol ='CLEAN',
                    parent=plugin_shelf)
 
+    pm.shelfButton(annotation='flushUndo() command shortcut,\n'
+                              'in order to avoid crash when reloading plugin',
+                   command = flushUndo,
+                   image=icons['flushUndo'],
+                   iol='FLUSH',
+                   parent=plugin_shelf)
+
     pm.shelfButton(annotation='Create pistonSolver in node editor',
                    command=createSolver,
                    image=icons['pistonSolverNode'],
@@ -106,5 +112,3 @@ def generate_shelf():
     # button that create the rig
 
     # button that delete all plugin-type nodes
-generate_shelf()
-

Piston_plugin.py

@@ -3,10 +3,6 @@
 import pymel.core as pm
 import importlib
 import sys
-
-for module in sys.modules.copy() :
-    if module.startswith('Piston_UI') :
-        del sys.modules[module]
 import Piston_UI
 
 maya_useNewAPI = True
@@ -353,6 +349,10 @@ def initializePlugin(plugin):
     :return:
     '''
     plugin_fn = OpenMaya.MFnPlugin(plugin, 'Baptiste Fraboul', '0.0.1')
+    try :
+        Piston_UI.generate_shelf()
+    except :
+        print(r'Couldn\'t create UI')
 
     try:
         plugin_fn.registerCommand(generatePiston.kPluginCmdName,
@@ -394,6 +394,17 @@ def uninitializePlugin(plugin):
 
     plugin_fn = OpenMaya.MFnPlugin(plugin)
 
+    # We completely unload modules to avoid error
+    try :
+        for module in sys.modules.copy():
+            if module.startswith('Piston_UI'):
+                del sys.modules[module]
+    except :
+        print('Cant\'t delete module Piston_UI')
+    try :
+        pm.deleteUI('Piston_plugin')
+    except :
+        print("Can't delete UI")
     try:
         plugin_fn.deregisterCommand(generatePiston.kPluginCmdName)
         plugin_fn.deregisterNode(pistonVectorLengthNode.type_id)