moveManip.py

#-
# ==========================================================================
# Copyright (C) 1995 - 2006 Autodesk, Inc. and/or its licensors.  All 
# rights reserved.
#
# The coded instructions, statements, computer programs, and/or related 
# material (collectively the "Data") in these files contain unpublished 
# information proprietary to Autodesk, Inc. ("Autodesk") and/or its 
# licensors, which is protected by U.S. and Canadian federal copyright 
# law and by international treaties.
#
# The Data is provided for use exclusively by You. You have the right 
# to use, modify, and incorporate this Data into other products for 
# purposes authorized by the Autodesk software license agreement, 
# without fee.
#
# The copyright notices in the Software and this entire statement, 
# including the above license grant, this restriction and the 
# following disclaimer, must be included in all copies of the 
# Software, in whole or in part, and all derivative works of 
# the Software, unless such copies or derivative works are solely 
# in the form of machine-executable object code generated by a 
# source language processor.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND. 
# AUTODESK DOES NOT MAKE AND HEREBY DISCLAIMS ANY EXPRESS OR IMPLIED 
# WARRANTIES INCLUDING, BUT NOT LIMITED TO, THE WARRANTIES OF 
# NON-INFRINGEMENT, MERCHANTABILITY OR FITNESS FOR A PARTICULAR 
# PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE, OR 
# TRADE PRACTICE. IN NO EVENT WILL AUTODESK AND/OR ITS LICENSORS 
# BE LIABLE FOR ANY LOST REVENUES, DATA, OR PROFITS, OR SPECIAL, 
# DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES, EVEN IF AUTODESK 
# AND/OR ITS LICENSORS HAS BEEN ADVISED OF THE POSSIBILITY 
# OR PROBABILITY OF SUCH DAMAGES.
#
# ==========================================================================
#+

# import maya
# maya.cmds.loadPlugin("moveManip.py")
# maya.cmds.spMoveManipCtxCmd( 'spMoveManipContext1' )
# maya.cmds.setParent( 'Shelf1' )
# maya.cmds.toolButton( 'spMoveManip1', cl='toolCluster', t='spMoveManipContext1', i1="moveManip.xpm" )

import sys
import maya.OpenMaya as OpenMaya
import maya.OpenMayaUI as OpenMayaUI
import maya.OpenMayaMPx as OpenMayaMPx

moveManipId = OpenMaya.MTypeId(0x87009)
contextCmdName = "spMoveManipCtxCmd"
nodeName = "spMoveManip"

class moveManip(OpenMayaMPx.MPxManipContainer):
    fDistanceManip = OpenMaya.MDagPath()
    fFreePointManip = OpenMaya.MDagPath()

    def __init__(self):
        OpenMayaMPx.MPxManipContainer.__init__(self)

    def createChildren(self):
        self.fDistanceManip = self.addDistanceManip("distanceManip", "distance")
        distanceManipFn = OpenMayaUI.MFnDistanceManip(self.fDistanceManip)
        startPoint = OpenMaya.MPoint(0.0, 0.0, 0.0)
        direction = OpenMaya.MVector(0.0, 1.0, 0.0)
        distanceManipFn.setStartPoint(startPoint)
        distanceManipFn.setDirection(direction)
        self.fFreePointManip = self.addFreePointTriadManip("pointManip", "freePoint")

    def connectToDependNode(self, node):
        nodeFn = OpenMaya.MFnDependencyNode(node)

        try:
            syPlug = nodeFn.findPlug("scaleY")
            tPlug = nodeFn.findPlug("translate")
            distanceManipFn = OpenMayaUI.MFnDistanceManip(self.fDistanceManip)
            distanceManipFn.connectToDistancePlug(syPlug)
            freePointManipFn = OpenMayaUI.MFnFreePointTriadManip(self.fFreePointManip)
            freePointManipFn.connectToPointPlug(tPlug)
            OpenMayaMPx.MPxManipContainer.finishAddingManips(self)
            OpenMayaMPx.MPxManipContainer.connectToDependNode(self,node)
        except:
            sys.stderr.write( "Error finding and connecting plugs\n" )
            raise

def moveManipCreator():
    return OpenMayaMPx.asMPxPtr( moveManip() )

def moveManipInitialize():
    OpenMayaMPx.MPxManipContainer.initialize()

class moveManipContext(OpenMayaMPx.MPxSelectionContext):
    def __init__(self):
        OpenMayaMPx.MPxSelectionContext.__init__(self)

    def toolOnSetup(self,event):
        updateManipulators(self)
        OpenMaya.MModelMessage.addCallback(OpenMaya.MModelMessage.kActiveListModified, updateManipulators, self)

def updateManipulators(clientData):
    clientData.deleteManipulators()
    selectionList = OpenMaya.MSelectionList()

    OpenMaya.MGlobal.getActiveSelectionList(selectionList)
    selectionIter = OpenMaya.MItSelectionList(selectionList, OpenMaya.MFn.kInvalid)
    while not selectionIter.isDone():
        dependNode = OpenMaya.MObject()
        selectionIter.getDependNode(dependNode)
        if dependNode.isNull() or not dependNode.hasFn(OpenMaya.MFn.kDependencyNode):
            print "depend node is null"
            continue
            
        dependNodeFn = OpenMaya.MFnDependencyNode(dependNode)
        rPlug = dependNodeFn.findPlug("translate", False)
        sPlug = dependNodeFn.findPlug("scaleY", False)
        if rPlug.isNull() or sPlug.isNull():
            print "translate and/or scale plugs are null"
            selectionIter.next()
            continue

        manipObject = OpenMaya.MObject()
        manipulator = OpenMayaMPx.MPxManipContainer.newManipulator(nodeName, manipObject)
        if manipulator is not None:
            clientData.addManipulator(manipObject)
            manipulator.connectToDependNode(dependNode)
        selectionIter.next()

class moveManipCtxCmd(OpenMayaMPx.MPxContextCommand):
    def __init__(self):
        OpenMayaMPx.MPxContextCommand.__init__(self)

    def makeObj(self):
        return OpenMayaMPx.asMPxPtr( moveManipContext() )


def contextCmdCreator():
    return OpenMayaMPx.asMPxPtr( moveManipCtxCmd() )


# initialize the script plug-in
def initializePlugin(mobject):
    mplugin = OpenMayaMPx.MFnPlugin(mobject)

    try:
        mplugin.registerContextCommand( contextCmdName, contextCmdCreator )
    except:
        print "Failed to register context command: %s" % contextCmdName
        raise
    
    try:
        mplugin.registerNode(nodeName, moveManipId, moveManipCreator, moveManipInitialize, OpenMayaMPx.MPxNode.kManipContainer)
    except:
        print "Failed to register node: %s" % nodeName
        raise

# uninitialize the script plug-in
def uninitializePlugin(mobject):
    mplugin = OpenMayaMPx.MFnPlugin(mobject)
    try:
        mplugin.deregisterContextCommand(contextCmdName)
    except:
        print "Failed to deregister context command: %s" % contextCmdName
        raise
        
    try:
        mplugin.deregisterNode(moveManipId)
    except:
        print "Failed to deregister node: %s" % nodeName
        raise