- Main Page
- Related Pages
- Modules
- Classes
- Examples

# 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