display_lib.py

from subprocess import call

import maya.cmds as cmds
import maya.mel as mel
import webrImport as web

#
# web
cn = web.mod( 'constraint_lib' )
ac = web.mod( 'animCurve_lib' )


def message( what = '', maya = True ):
    what = '-- ' + what + ' --'
    if maya:
        mel.eval( 'print \"' + what + '\";' )
    else:
        print( what )


def findControl( ann = '', panelTyp = '', split = 3 ):
    # split = which parent in full path to return
    cntrls = cmds.lsUI( controls = True, long = 1 )
    qualified = []
    result = []
    for con in cntrls:
        if cmds.control( con, exists = True ):
            string = cmds.control( con, q = 1, ann = 1 )
            if ann in string:
                qualified.append( con )
    print( qualified )
    for item in qualified:
        path = cmds.control( item, q = 1, fpn = 1 )
        if panelTyp in path:
            stf = path.split( '|' )
            tmp = stf[len( stf ) - split]
            result.append( cmds.control( tmp, q = 1, fpn = 1 ) )
        else:
            print( '______searching  ', path )
    return result


def findControlParent( control = '', split = 3 ):
    # split = which parent in full path to return
    path = cmds.control( control, q = 1, fpn = 1 )
    stf = path.split( '|' )
    tmp = stf[len( stf ) - split]
    result = cmds.control( tmp, q = 1, fpn = 1 )
    return result


def geButton( name = '', parent = '', attach = None, label = '', cmd = '', gap = 2, w = 26, h = 19, bg = [0.21, 0.21, 0.21], ann = '' ):
    # "boldLabelFont", "smallBoldLabelFont", "tinyBoldLabelFont", "plainLabelFont", "smallPlainLabelFont",
    # "obliqueLabelFont", "smallObliqueLabelFont", "fixedWidthFont" and "smallFixedWidthFont"
    if not cmds.control( name, ex = 1 ):
        cmd = cmd
        # btn = cmds.iconTextButton(name, p=parent, style='textOnly', stp='python', c=cmd, al='center', h=h, bgc=bg, mw=1, w=w, label=label, ann=ann)
        btn = cmds.button( name, p = parent, c = cmd, al = 'center', h = h, bgc = bg, w = w, label = label, ann = ann )
        ac = [( btn, 'left', gap, attach )]
        attachForm = [( btn, 'bottom', 3 )]
        if attach:
            cmds.formLayout( parent, e = True, ac = ac, attachForm = attachForm )
        else:
            attachForm = [( btn, 'left', 2 ), ( btn, 'bottom', 3 )]
            cmds.formLayout( parent, e = True, attachForm = attachForm )
        return btn
    else:
        print( '' )
        cmds.deleteUI( name, control = True )


def geHeading( name = '', parent = '', attach = None, label = '', cmd = '', gap = 10, ann = '', bgc = [0.275, 0.275, 0.275], w = 45, h = 17 ):
    if not cmds.control( name, ex = 1 ):
        lab = cmds.text( name, l = label, fn = 'obliqueLabelFont', al = 'center', bgc = bgc, ebg = True, ann = ann, w = w, h = h )
        ac = [( lab, 'left', gap, attach )]
        attachForm = [( lab, 'bottom', 4 )]
        if attach:
            cmds.formLayout( parent, e = True, ac = ac, attachForm = attachForm )
        else:
            cmds.formLayout( parent, e = True, attachForm = attachForm )
        return lab
    else:
        print( '' )
        cmds.deleteUI( name, control = True )


def geField( name = '', parent = '', attach = None, label = '', cmd = '', w = 100, gap = 10, tx = '', ann = '' ):
    if not cmds.control( name, ex = 1 ):
        # field = cmds.textField( name, tx = tx, h = 21, w = w, cc = cmd, ec = cmd, ann = ann, aie = True )
        field = cmds.textField( name, tx = tx, h = 21, w = w, ec = cmd, ann = ann, aie = True )  # no change command
        ac = [( field, 'left', gap, attach )]
        attachForm = [( field, 'left', 2 ), ( field, 'bottom', 2 )]
        if attach:
            cmds.formLayout( parent, edit = True, ac = ac, attachForm = attachForm )
        else:
            cmds.formLayout( parent, edit = True, attachForm = attachForm )
        return field
    else:
        print( '' )
        cmds.deleteUI( name, control = True )


def geFieldTx( v = '' ):
    ui = GraphEditorButtonNames()
    try:
        tx = cmds.textField( ui.fil, q = 1, tx = 1 )
        if not tx:
            cmds.textField( ui.fil, e = 1, tx = v )
            import graphFilter
            reload( graphFilter )
            graphFilter.graphEditorCMD()
        else:
            cmds.textField( ui.fil, e = 1, tx = '' )
            import graphFilter
            graphFilter.graphEditorCMD()
    except:
        pass


def colorOff():
    return [0.97, 0.92, 0.04]  # yellow


def colorOn():
    return [0.13, 0.77, 0.11]  # green


def graphEditorButtons( *args ):
    # TODO: add button to distribute keys on selected curves
    # TODO: use dif layout, window gets locked to a min size
    # TODO: should work on all graph editors, needs to be smarter
    # TODO: add isolate selected curves
    # ui names
    ui = GraphEditorButtonNames()
    # check where to build
    build = False
    p = findControl( ann = 'Move Nearest Picked Key Tool', panelTyp = 'graphEditor', split = 3 )
    remove = findControl( ann = 'Enter text to limit the listing', panelTyp = 'graphEditor', split = 2 )
    pnl = cmds.getPanel( wf = True )
    if len( p ) == 1:
        build = True
        pnl = 'graphEditor'
    else:
        if 'graphEditor' in pnl:
            build = True
        else:
            cmds.warning( '-- Multiple graph editors... Focus on appropriate graph editor. --' )
            return False
    # build
    if not cmds.control( ui.filD, ex = 1 ):
        bgc = [0.4, 0.4, 0.4]
        # bgc = [0.3, 0.48, 0.57]
        bgcd = [0.97, 0.92, 0.04]  # yellow
        if build:
            p = findControl( ann = 'Move Nearest Picked Key Tool', panelTyp = pnl, split = 3 )[0]
            # print '_______    ', findControl(ann='Indicates that either text filter', panelTyp=pnl, split=2)
            remove = findControl( ann = 'Enter text to limit the listing', panelTyp = pnl, split = 2 )[0]
            cmds.control( remove, e = 1, m = 0 )
            cmds.formLayout( p, e = 1, h = 52 )
            cmds.setParent( p )
            #
            # "import webrImport as web\nds = web.mod('display_lib')\nds.geFieldTx('tx,ty,tz,rx,ry,rz')"
            # build
            item = geButton( name = ui.filD, parent = p, label = '::',
                            cmd = "import webrImport as web\nds = web.mod('display_lib')\nds.geFieldTx('tx,ty,tz,rx,ry,rz')",
                            gap = 2, w = 15, bg = [0.2, 0.3, 0.5], ann = "add default channel filter: tr*,ro*" )
            item = geField( name = ui.fil, parent = p, attach = item,
                           cmd = "import webrImport as web\ngf = web.mod('graphFilter')\ngf.graphEditorCMD()",
                           gap = 2, w = 185, ann = "add channel filters by name and/or wild cards: 't*X'" )
            item = geButton( name = ui.hldPre, parent = p, attach = item, label = '<--',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.holdCrv(postCurrent=False)",
                            gap = 5, bg = [0.4, 0.2, 0.2], ann = 'hold value of selected curve to the left' )
            item = geButton( name = ui.hldAll, parent = p, attach = item, label = 'HOLD',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.holdCrv()",
                            gap = 0, w = 45, bg = [0.5, 0.2, 0.2], ann = 'hold value of selected curve' )
            item = geButton( name = ui.hldPost, parent = p, attach = item, label = '-->',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.holdCrv(preCurrent=False)",
                            gap = 0, bg = [0.4, 0.2, 0.2], ann = 'hold value of selected curve to the right' )
            item = geButton( name = ui.sclDwn, parent = p, attach = item, label = '-',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.scaleCrv(0.99)",
                            gap = 5, bg = [0.1, 0.4, 0.4], ann = 'scale curve:\npivot @ key = keys with same values selected\npivot @ 0     = keys with dif values selected\n' )
            item = geHeading( name = ui.sclTx, parent = p, attach = item, label = 'SCALE', gap = 0, w = 45, bgc = [0.25, 0.4, 0.4] )
            item = geButton( name = ui.sclUp, parent = p, attach = item, label = '+',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.scaleCrv(1.01)",
                            gap = 0, bg = [0.1, 0.4, 0.4], ann = 'scale curve:\npivot @ key = keys with same values selected\npivot @ 0     = keys with dif values selected\n' )
            item = geButton( name = ui.flp, parent = p, attach = item, label = 'FLIP',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.scaleCrv(-1)",
                            gap = 0, w = 50, bg = [0.1, 0.4, 0.4] )
            item = geButton( name = ui.movDwn, parent = p, attach = item, label = '-',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.moveValue(False)",
                            gap = 5, bg = [0.3, 0.3, 0.5], ann = 'nudge keys/curves down in value' )
            item = geHeading( name = ui.movTx, parent = p, attach = item, label = 'VALUE', gap = 0, bgc = [0.35, 0.35, 0.5] )
            item = geButton( name = ui.movUp, parent = p, attach = item, label = '+',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.moveValue(True)",
                            gap = 0, bg = [0.3, 0.3, 0.5], ann = 'nudge keys/curves up in value' )
            item = geButton( name = ui.movTmLf, parent = p, attach = item, label = '<',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.moveTime(True)",
                            gap = 5, bg = [0.2, 0.3, 0.5], ann = 'move keys/curves to the left' )
            item = geHeading( name = ui.movTmTx, parent = p, attach = item, label = 'TIME', gap = 0, bgc = [0.3, 0.35, 0.5] )
            item = geButton( name = ui.movTmRt, parent = p, attach = item, label = '>',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.moveTime(False)",
                            gap = 0, bg = [0.2, 0.3, 0.5], ann = 'move keys/curves to the right' )
            item = geHeading( name = ui.sclTmTx, parent = p, attach = item, label = 'SCALE TIME', w = 65 )
            item = geField( name = ui.sclTmBy, parent = p, attach = item,
                           cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.animScale(cmds.textField('scaleTimeBy',query=True,tx=True))",
                           gap = 2, w = 40, tx = 1.0, ann = 'Scale selected curves from the first frame of playback' )
            w = 105
            item = geButton( name = ui.sftSel, parent = p, attach = item, label = 'SOFT Select',
                            cmd = "import webrImport as web\ncss = web.mod('curveSoftSelect')\ncss.toggleSelJob()",
                            gap = 20, w = w, bg = colorOff() )
            item = geButton( name = ui.sbfrm, parent = p, attach = item, label = 'SUB frame X',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.subframe()",
                            gap = 0, w = w, bg = bgc, ann = 'subframes to whole frames ' )
            item = geButton( name = ui.unfy, parent = p, attach = item, label = 'UNIFY',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.unifyKeys()",
                            gap = 0, w = w, bg = bgcd )
            item = geButton( name = ui.smth, parent = p, attach = item, label = 'SMOOTH',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.smoothKeys()",
                            gap = 0, w = w, bg = bgc )
            item = geButton( name = ui.expTgl, parent = p, attach = item, label = 'COLLAPSE',
                            cmd = "import webrImport as web\ngf = web.mod('graphFilter')\ngf.toggleExpand()",
                            gap = 0, w = w, bg = bgcd )
            item = geButton( name = ui.bkInfty, parent = p, attach = item, label = 'BAKE Results',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.bakeInfinity()",
                            gap = 0, w = w, bg = bgc )
            item = geButton( name = ui.infntyLnr, parent = p, attach = item, label = 'LINEAR Infinity',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.setInfinityLinear()",
                            gap = 0, w = w, bg = bgcd, ann = 'make infinity linear' )
            item = geButton( name = ui.infntyCns, parent = p, attach = item, label = 'CONSTANT Infinity',
                            cmd = "import webrImport as web\nac = web.mod('animCurve_lib')\nac.setInfinityConstant()",
                            gap = 0, w = w, bg = bgc, ann = 'make infinity constant' )
    else:
        # clean UI
        d = ui.__dict__
        try:
            for s in d:
                if cmds.control( str( d[s] ), ex = 1 ):
                    cmds.deleteUI( d[s], control = True )
        except:
            print( 'cant delete all', ui.__dict__ )
        for item in p:
            cmds.formLayout( item, e = 1, h = 28 )
        for item in remove:
            cmds.control( item, e = 1, m = 1 )


class GraphEditorButtonNames():

    def __init__( self ):
        # filter
        self.filD = 'FilterDefault'
        self.fil = 'graphFilter'
        # hold
        self.hldPre = 'holdPre'
        self.hldAll = 'holdAll'
        self.hldPost = 'holdPost'
        # scale
        self.sclDwn = 'scaleDown'
        self.sclTx = 'scaleTxt'
        self.sclUp = 'scaleUp'
        self.flp = 'FlipCurves'
        # value move
        self.movDwn = 'moveDown'
        self.movTx = 'moveValue'
        self.movUp = 'moveUp'
        # time move
        self.movTmLf = 'moveLeft'
        self.movTmTx = 'moveTime'
        self.movTmRt = 'moveRight'
        # scale time
        self.sclTmTx = 'scale'
        self.sclTmBy = 'scaleTimeBy'
        # subframe out
        self.sbfrm = 'subframeOut'
        # unify keys
        self.unfy = 'unifyKeys'
        # bake infinity
        self.bkInfty = 'bakeInfinity'
        # smooth keys
        self.smth = 'smoothKeys'
        # soft seleft
        self.sftSel = 'softSelKeys'
        # expand toggleGeo
        self.expTgl = 'expandAttrToggle'
        # expand toggleGeo
        self.infntyLnr = 'infntyLnrCrvs'
        # expand toggleGeo
        self.infntyCns = 'infntyCnsCrvs'


def toggleObjectDisplay( purpose ):
    '''
    purpose = 'anim'
    purpose = 'cam'
    '''

    # get active and find type of panel
    currentPanel = cmds.getPanel( withFocus = True )
    panelType = cmds.getPanel( to = currentPanel )

    if panelType == 'modelPanel':
        # all off
        cmds.modelEditor( currentPanel, e = True, alo = 0 )
        cmds.modelEditor( currentPanel, e = True, imagePlane = 1 )
        cmds.modelEditor( currentPanel, e = True, pluginShapes = 1 )

        # arguments
        if purpose == 'anim':
            # specific on
            cmds.modelEditor( currentPanel, e = True, nurbsCurves = 1 )
            cmds.modelEditor( currentPanel, e = True, polymeshes = 1 )
            cmds.modelEditor( currentPanel, e = True, locators = 1 )
            cmds.modelEditor( currentPanel, e = True, cameras = 1 )
            cmds.modelEditor( currentPanel, e = True, handles = 1 )
            cmds.modelEditor( currentPanel, e = True, hud = 1 )
            cmds.modelEditor( currentPanel, e = True, grid = 1 )
            message( 'Panel display set for animation' )
        elif purpose == 'cam':
            # specific on
            cmds.modelEditor( currentPanel, e = True, cameras = 1 )
            cmds.modelEditor( currentPanel, e = True, polymeshes = 1 )
            cmds.modelEditor( currentPanel, e = True, hud = 0 )
            cmds.modelEditor( currentPanel, e = True, grid = 0 )
            cmds.modelEditor( currentPanel, e = True, wireframeOnShaded = 0 )
            message( 'Panel display set for camera' )
    else:
        message( 'Current panel is of the wrong type' )


def toggleGeo():
    pnl = cmds.getPanel( withFocus = True )
    if cmds.getPanel( to = pnl ) == 'modelPanel':
        state = cmds.modelEditor( pnl, q = True, polymeshes = True )
        if state:
            cmds.modelEditor( pnl, e = True, polymeshes = 0 )
            cmds.modelEditor( pnl, e = True, nurbsSurfaces = 0 )
            cmds.modelEditor( pnl, e = True, pluginShapes = 0 )
        else:
            cmds.modelEditor( pnl, e = True, polymeshes = 1 )
            cmds.modelEditor( pnl, e = True, nurbsSurfaces = 1 )
            cmds.modelEditor( pnl, e = True, pluginShapes = 1 )


def toggleCurves():
    pnl = cmds.getPanel( withFocus = True )
    if cmds.getPanel( to = pnl ) == 'modelPanel':
        state = cmds.modelEditor( pnl, q = True, nurbsCurves = True )
        if state:
            cmds.modelEditor( pnl, e = True, nurbsCurves = 0 )
            cmds.modelEditor( pnl, e = True, locators = 0 )
        else:
            cmds.modelEditor( pnl, e = True, nurbsCurves = 1 )
            cmds.modelEditor( pnl, e = True, locators = 1 )


def toggleCameras():
    pnl = cmds.getPanel( withFocus = True )
    if cmds.getPanel( to = pnl ) == 'modelPanel':
        state = cmds.modelEditor( pnl, q = True, cameras = True )
        if state:
            cmds.modelEditor( pnl, e = True, cameras = 0 )
            cmds.modelEditor( pnl, e = True, imagePlane = 0 )
            # cmds.modelEditor( pnl, e = True, pluginShapes = 0 )
        else:
            cmds.modelEditor( pnl, e = True, cameras = 1 )
            cmds.modelEditor( pnl, e = True, imagePlane = 1 )
            # cmds.modelEditor( pnl, e = True, pluginShapes = 1 )


def toggleRes():
    sel = cmds.ls( sl = True )
    c = ':c_master_CTRL.'
    attrHi = 'hiResGeoVis'
    attrLo = 'loResGeoVis'
    name = sel[0].split( ':' )[0]
    if cmds.getAttr( name + c + attrHi ) == 1:
        cmds.setAttr( name + c + attrHi, 0 )
        cmds.setAttr( name + c + attrLo, 1 )
        message( 'LO res' )
    else:
        cmds.setAttr( name + c + attrHi, 1 )
        cmds.setAttr( name + c + attrLo, 0 )
        message( 'HI res' )


def shapeSize( obj = None, mltp = 1 ):
    '''\n
    mltp = size multiplier of shape nodes
    '''
    if obj is None:
        # make a list from selection
        obj = cmds.ls( sl = True, l = True )
        if obj:
            shapeSize( obj, mltp )
        else:
            message( 'nothing selected' )
    elif type( obj ) == list:
        # no need to accomodate
        for i in obj:
            shapeSize( i, mltp )
    else:
        # obj must be a single item, make a list
        obj = [obj]
        # run the loop on list
        for item in obj:
            shape = cmds.listRelatives( item, s = True, f = True )
            if shape:
                for node in shape:
                    if 'SharedAttr' not in node:
                        cmds.scale( mltp, mltp, mltp, node + '.cv[*]' )


def locSize( lc, mltp = 1 ):
    axis = ['X', 'Y', 'Z']
    for axs in axis:
        sz = cmds.getAttr( lc + 'Shape.localScale' + axs )
        cmds.setAttr( lc + 'Shape.localScale' + axs, sz * mltp )


def changeColor( color = 17, shapes = False ):
    sel = cmds.ls( sl = True )
    if sel:
        if shapes:
            sel = shapeNodes()
        for item in sel:
            cmds.setAttr( item + '.overrideEnabled', 1 )
            cmds.setAttr( item + '.overrideColor', color )
    else:
        message( 'Select an object' )


def shapeNodes():
    sel = cmds.ls( sl = 1 )
    shapes = []
    if sel:
        for s in sel:
            shape = cmds.listRelatives( s, s = True, f = True )
            if shape:
                for shp in shape:
                    shapes.append( shp )
    return shapes


def worldSpdExp( sel, attr ):
    exp = "float $currentPos[] = `getAttr -t (frame) " + sel + ".worldMatrix`;\nfloat $lastPos[] = `getAttr -t (frame-1) " + sel + ".worldMatrix`;\nfloat $speed = abs(mag (<<0*" + sel + ".tx+$currentPos[12]-$lastPos[12], 0*" + sel + ".ty+$currentPos[13]-$lastPos[13], 0*" + sel + ".tz+$currentPos[14]-$lastPos[14]>>) );\nfloat $inKPH = $speed*24*60*60/100000;\n" + sel + "." + attr + " = $inKPH;"
    return exp


def localSpdExp( sel, attr ):
    exp = "float $lastPosX = `getAttr -t (frame-1) " + sel + ".tx`;\nfloat $lastPosY = `getAttr -t (frame-1) " + sel + ".ty`;\nfloat $lastPosZ = `getAttr -t (frame-1) " + sel + ".tz`;\nfloat $speed = abs(mag (<<" + sel + ".translateX-$lastPosX," + sel + ".translateY-$lastPosY," + sel + ".translateZ-$lastPosZ>>) );\nfloat $inKPH = $speed*24*60*60/100000;\n" + sel + "." + attr + " = $inKPH;"
    return exp


def createSpeedAttr( sel, attr, exp ):
    cmds.addAttr( sel, longName = attr, attributeType = 'float', k = 0 )
    cmds.setAttr( sel + "." + attr, cb = 1 )
    cmds.expression( o = sel, s = exp )


def speed( world = True, local = True ):
    selected = cmds.ls( sl = True )
    if len( selected ) != 0:
        for sel in selected:
            if world:
                attr = 'worldSpeed'
                exp = worldSpdExp( sel, attr )
                if not cmds.attributeQuery( attr, node = sel, ex = True ):
                    createSpeedAttr( sel, attr, exp )
                else:
                    cmds.warning( '-- Speed attr (' + attr + ') already exists - ' + sel + ' ! --' )
                    deleteUserAttr( sel = sel, att = attr )
                    return None
            if local:
                attr = 'localSpeed'
                exp = localSpdExp( sel, attr )
                if not cmds.attributeQuery( attr, node = sel, ex = True ):
                    createSpeedAttr( sel, attr, exp )
                else:
                    cmds.warning( '-- Speed attr (' + attr + ') already exists - ' + sel + ' ! --' )
                    deleteUserAttr( sel = sel, att = attr )
                    return None
    else:
        message( 'Select an object.' )


def rpmExp( sel, attr ):
    '''
    exp = "
    float $lastPosX = `getAttr -t (frame-1) " + sel + ".tx`;\nfloat $lastPosY = `getAttr -t (frame-1) " + sel + ".ty`;\n
    float $lastPosZ = `getAttr -t (frame-1) " + sel + ".tz`;\n
    float $speed = abs(mag (<<" + sel + ".translateX-$lastPosX," + sel + ".translateY-$lastPosY," + sel + ".translateZ-$lastPosZ>>) );\n
    float $inKPH = $speed*24*60*60/100000;\n
    " + sel + "." + attr + " = $inKPH;"
    return exp
    '''
    pass


def createRpmAttr( sel, attr, exp ):
    cmds.addAttr( sel, longName = attr, attributeType = 'float', k = 0 )
    cmds.setAttr( sel + "." + attr, cb = 1 )
    cmds.expression( o = sel, s = exp )


def rpmAvg( obj, attr ):
    '''
    3 frame average
    '''
    # 360/(fps * 60) = degrees
    # 360/(24  * 60) = 0.25
    frames = [cmds.currentTime( q = True ) - 1, cmds.currentTime( q = True ), cmds.currentTime( q = True ) + 1]
    val = []
    rpm = []
    for frame in frames:
        val.append( cmds.getAttr( obj + '.' + attr, t = frame ) )
    list( set( val ) )
    if len( val ) != 1:
        rpm.append( ( val[0] - val[1] ) / 0.25 )
        rpm.append( ( val[1] - val[2] ) / 0.25 )
        # average of 2 values
        return sum( rpm, 0.0 ) / len( rpm )
    else:
        return 0.0


def rpm( world = True, local = True ):
    selected = cmds.ls( sl = True )
    if len( selected ) != 0:
        for sel in selected:
            if world:
                attr = 'worldSpeed'
                exp = worldSpdExp( sel, attr )
                if not cmds.attributeQuery( attr, node = sel, ex = True ):
                    createRpmAttr( sel, attr, exp )
                else:
                    cmds.warning( '-- Speed attr (' + attr + ') already exists - ' + sel + ' ! --' )
                    deleteUserAttr( sel = sel, att = attr )
                    return None
            if local:
                attr = 'localSpeed'
                exp = localSpdExp( sel, attr )
                if not cmds.attributeQuery( attr, node = sel, ex = True ):
                    createRpmAttr( sel, attr, exp )
                else:
                    cmds.warning( '-- Speed attr (' + attr + ') already exists - ' + sel + ' ! --' )
                    deleteUserAttr( sel = sel, att = attr )
                    return None
    else:
        message( 'Select an object.' )


def plotExp( sel, attr ):
    exp = "float $nextPosX = `getAttr -t (frame+1) " + sel + ".rx`;\nfloat $lastPosX = `getAttr -t (frame-1) " + sel + ".rx`;\nfloat $speed = abs(mag (<<" + sel + ".rotateX,$lastPosX,  $nextPosX>>) );\n" + sel + "." + attr + " = $speed;"
    return exp


def createPlotAttr( sel, attr, exp ):
    cmds.addAttr( sel, longName = attr, attributeType = 'float', k = 0 )
    cmds.setAttr( sel + "." + attr, cb = 1 )
    cmds.expression( o = sel, s = exp )


def plotAttr():
    selected = cmds.ls( sl = True, fl = True )
    if len( selected ) != 0:
        sel = selected[0]
        attr = 'plot'
        exp = plotExp( sel, attr )
        if not cmds.attributeQuery( attr, node = sel, ex = True ):
            createPlotAttr( sel, attr, exp )
        else:
            cmds.warning( '-- Plotted attr (' + attr + ') already exists - ' + sel + ' ! --' )
            deleteUserAttr( sel = sel, exp = True, att = attr )
            return None
    else:
        message( 'Select an object.' )


def measureDis( obj1, obj2 ):
    '''
    math for distance between 2 objects
    '''
    p1 = cmds.xform( obj1, q = True, ws = True, t = True )
    p2 = cmds.xform( obj2, q = True, ws = True, t = True )
    v = [0, 0, 0]
    v[0] = p1[0] - p2[0]
    v[1] = p1[1] - p2[1]
    v[2] = p1[2] - p2[2]
    distance = v[0] * v[0] + v[1] * v[1] + v[2] * v[2]
    from math import sqrt
    distance = sqrt( distance )
    return distance


def createDisAttr( sel, attr ):
    '''
    creates attr and applies expression
    '''
    cmds.addAttr( sel, longName = attr, attributeType = 'float', k = True )
    cmds.setAttr( sel + "." + attr, cb = 1 )


def createDisNode( name = '' ):
    dis = cmds.createNode( 'distanceBetween', name = name + '_dis' )
    return dis


def distance( obj1 = None, obj2 = None ):
    '''
    assembles distance relationship
    '''
    selected = None
    attr = 'distance'
    st = []
    disNode = None
    if not obj1:
        selected = cmds.ls( sl = True, fl = True )
    else:
        selected = [obj1, obj2]
    if len( selected ) == 2:
        st.append( cmds.attributeQuery( attr, node = selected[0], ex = True ) )
        st.append( cmds.attributeQuery( attr, node = selected[1], ex = True ) )
        st = list( set( st ) )
        if len( st ) == 1:
            st = st[0]
            # toggle distance attr
            for sel in selected:
                if not st:
                    createDisAttr( sel, attr )
                else:
                    # list
                    if not disNode:
                        disNode = clean( cmds.listConnections( sel + '.worldMatrix[0]', s = False, d = True, type = 'distanceBetween' ) )
                    cmds.deleteAttr( sel + '.' + attr )
            # toggle distance node
            if st:
                if disNode:
                    cmds.delete( disNode )
            else:
                # connect to distance node
                disNode = createDisNode( name = 'dstnc___' + selected[0] + '___' + selected[1] )
                cmds.connectAttr( selected[0] + '.worldMatrix[0]', disNode + '.inMatrix1' )
                cmds.connectAttr( selected[1] + '.worldMatrix[0]', disNode + '.inMatrix2' )
                # feed distance attrs
                cmds.connectAttr( disNode + '.distance', selected[0] + '.' + attr )
                cmds.connectAttr( disNode + '.distance', selected[1] + '.' + attr )
                # reselect objects
                cmds.select( selected )
        else:
            message( 'one object already have a distance attr, aborting.' )
    elif selected:
        if cmds.attributeQuery( attr, node = selected[0], ex = True ):
            # remove distance from pair, delete dis node
            disNode = clean( cmds.listConnections( selected[0] + '.worldMatrix[0]', s = False, d = True, type = 'distanceBetween' ) )
            if disNode:
                nodes = clean( cmds.listConnections( disNode + '.distance', s = False, d = True ) )
                if len( nodes ) == 2:
                    distance( obj1 = nodes[0], obj2 = nodes[1] )
                else:
                    message( 'A pair to the selection was not found' )
    else:
        cmds.warning( '--  Select 2 objects to toggle distance attributes. --' )


def clean( obj ):
    if obj:
        if isinstance( obj, list ):
            if len( obj ) == 1:
                return obj[0]
            else:
                return list( set( obj ) )
        else:
            return obj
    else:
        return None


def deleteUserAttr( sel = None, att = '' ):
    if sel is None:
        sel = cmds.ls( sl = True )
        if len( sel ) != 1:
            message( 'Select one object' )
            return None
        else:
            sel = sel[0]
    selectedAttr = cmds.channelBox( 'mainChannelBox', q = True, sma = True )
    # userAttr = cmds.listAttr(sel, ud=True)
    if att != '':
        cmds.deleteAttr( sel + '.' + att )
        message( 'Attribute ' + att + ' deleted' )
        return None
    if selectedAttr is not None:
        for attr in selectedAttr:
            cmds.deleteAttr( sel + '.' + attr )
    else:
        message( 'Select an attribute(s) in the channelBox.' )


def rotateManip():
    mode = cmds.manipRotateContext( 'Rotate', q = True, mode = True )
    if mode == 0:
        cmds.manipRotateContext( 'Rotate', e = True, mode = 2 )
        message( 'Gimbal' )
    elif mode == 2:
        cmds.manipRotateContext( 'Rotate', e = True, mode = 0 )
        message( 'Local' )
    elif mode == 1:
        cmds.manipRotateContext( 'Rotate', q = True, mode = 2 )
        message( 'Gimbal' )


def translateManip():
    mode = cmds.manipMoveContext( 'Move', q = True, mode = True )
    # print mode
    if mode == 0:
        cmds.manipMoveContext( 'Move', e = True, mode = 1 )
        message( 'Local' )
    elif mode == 1:
        cmds.manipMoveContext( 'Move', e = True, mode = 2 )
        message( 'World' )
    elif mode == 2:
        cmds.manipMoveContext( 'Move', e = True, mode = 0 )
        message( 'Object' )


def clearKey():
    mel.eval( 'timeSliderClearKey;' )


def altFrame( *args ):
    # cmds.nodeType(sel), object type
    # optimize, real slow
    pnl = cmds.getPanel( withFocus = True )
    typ = cmds.getPanel( typeOf = pnl )
    if typ == 'modelPanel':
        sel = cmds.ls( sl = True, fl = True )
        gs = ac.GraphSelection()
        locs = []
        if sel:
            for item in sel:
                if cmds.objectType( item, i = 'transform' ) or cmds.objectType( item, i = 'joint' ):
                    loc = cn.locator( obj = item, ro = 'zxy', X = 0.35, constrain = False, shape = False )[0]
                    locs.append( loc )
                else:
                    try:
                        print( cmds.listRelatives( item, parent = True )[0], '_______________' )
                        loc = cn.locator( obj = cmds.listRelatives( item, parent = True )[0], ro = 'zxy', X = 0.35, constrain = False, shape = False )
                        print( loc, '_____________________' )
                        loc = loc[0]
                        locs.append( loc )
                    except:
                        message( 'didnt frame object: ' + item )
            cmds.select( locs )
            mel.eval( "fitPanel -selected;" )
            cmds.delete( locs )
            gs.reselect()
        else:
            message( 'select an object' )
    else:
        mel.eval( "fitPanel -selected;" )


def polyToggle():
    pnl = cmds.getPanel( wf = 1 )
    st = cmds.modelEditor( pnl, q = 1, polymeshes = 1 )
    cmds.modelEditor( pnl, e = 1, polymeshes = ( not st ) )