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

# mathTableControl.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.
    #
    # ==========================================================================
    #+

    #
    # Creation Date:   13 October 2006
    #
    # Author: mlausch
    #
    # Example Plugin: mathTableControl.py
    #
    # Creates a simple table control
    #
    # Example:
    #   import maya.cmds as cmds
    #   window = cmds.window(title="Math Table",widthHeight=(400,150))
    #   cmds.paneLayout()
    #   cmds.showWindow()
    #   cmds.spMathTableControl()
    #

    import math, sys

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

    kPluginCmdName = "spMathTableControl"

    kNopFlag = "-nop"
    kNopLongFlag = "-noOperation"
    kMultFlag = "-mul"
    kMultLongFlag = "-multiplyVals"
    kAddFlag = "-add"
    kAddLongFlag = "-addVals"
    kRedrawFlag = "-rd"
    kRedrawFlagLong = "-redraw"

    # operations on cell coordinates, must return strings
    kNop = lambda x,y: "cell(%d,%d)" % (x,y)
    kMult = lambda x,y: str(x*y)
    kAdd = lambda x,y: str(x+y)

    kPythonPtrTable = {}

    class MathTableControlCmd(OpenMayaMPx.MPxControlCommand):
        def __init__(self):
            OpenMayaMPx.MPxControlCommand.__init__(self)

        def makeControl(self):
            control = MathTableControl(self)
            control.setOperation(kNop)
            control.setNumberOfRows(5)
            control.setNumberOfColumns(3)
            return OpenMayaMPx.asMPxPtr(control)

        def doEditFlags(self):
            theParser = self._parser()
            theControl = kPythonPtrTable.get(OpenMayaMPx.asHashable(self._control()), None)

            if theParser.isFlagSet(kNopFlag):
                theControl.setOperation(kNop)
            elif theParser.isFlagSet(kMultFlag):
                theControl.setOperation(kMult)
            elif theParser.isFlagSet(kAddFlag):
                theControl.setOperation(kAdd)
            elif theParser.isFlagSet(kRedrawFlag):
                theControl.redrawCells()
                theControl.redrawLabels()
            else:
                OpenMayaMPx.MPxControlCommand.doEditFlags(self)

        def doQueryFlags(self):
            return OpenMayaMPx.MPxControlCommand.doQueryFlags(self)

        def appendSyntax(self):
            theSyntax = self._syntax()
            theSyntax.addFlag(kNopFlag, kNopLongFlag)
            theSyntax.addFlag(kMultFlag, kMultLongFlag)
            theSyntax.addFlag(kAddFlag, kAddLongFlag)
            theSyntax.addFlag(kRedrawFlag, kRedrawFlagLong)

    class MathTableControl(OpenMayaMPx.MPxUITableControl):
        def __init__(self, command):
            OpenMayaMPx.MPxUITableControl.__init__(self, command)
            self.__myOperation = None
            kPythonPtrTable[OpenMayaMPx.asHashable(self)] = self

        def __del__(self):
            del kPythonPtrTable[OpenMayaMPx.asHashable(self)]

        def cellString(self, row, column, isValidCell):
            result = ""
            if callable(self.__myOperation):
                result = self.__myOperation(row, column)
            OpenMaya.MScriptUtil.setBool(isValidCell, bool(result))
            return result

        def labelString(self, labelType, index):
            value = ""
            if labelType == OpenMayaMPx.MPxUITableControl.kRowLabel:
                value = "[Row %d]" % index
            elif labelType == OpenMayaMPx.MPxUITableControl.kColumnLabel:
                value = "[Col %d]" % index
            return value

        def setOperation(self, op):
            self.__myOperation = op
            self.redrawCells()

    ################################################################

    def cmdCreator():
        return OpenMayaMPx.asMPxPtr(MathTableControlCmd())

    def initializePlugin(mobject):
        mplugin = OpenMayaMPx.MFnPlugin(mobject, "Autodesk", "1.0", "Any")
        try:
            mplugin.registerControlCommand(kPluginCmdName, cmdCreator)
        except:
            sys.stderr.write( "Failed to register command: %s\n" % kPluginCmdName)
            raise

    def uninitializePlugin(mobject):
        mplugin = OpenMayaMPx.MFnPlugin(mobject)
        try:
            mplugin.deregisterControlCommand(kPluginCmdName)
        except:
            sys.stderr.write("Failed to unregister command: %s\n" % kPluginCmdName)
            raise