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