2016/09/04

[Maya][Python]デコレータでmayaバージョンごとに処理分け

pythonドキュメントにpropertyデコレータの挙動を解説するために同じ処理を再現したクラスが載っていてすごく勉強になりました。

で、mayaのバージョンごとに処理を変える方法ないかなーと思っていたので↑を参考にデコレータで実装してみました。
最後の行でdoSomething関数を実行していますが、同じ関数名でもmayaのバージョンによって処理が変わります。

import maya.cmds as cmds

maya_version = cmds.about(version=1) # mayaのバージョン


## バージョンごとに関数(メソッド)の処理を変えるデコレータ
class DoOnVertion(object):
    
    ## 初期化
    def __init__(self, func):
        self.func = func
        
        
        
    ## 関数のように呼び出された時の処理
    #  self.funcを実行する
    def __call__(self, *args, **kwargs):
        return self.func(*args, **kwargs)
        
        
        
    ## 指定されたバージョンとmayaバージョンが合っていれば関数(メソッド)を置き換える
    def version(self, version):
        print 'Replace funcition if version is [{0}].'.format(version)
        
        def new_func(func):
            if maya_version == version:
                rtn = func
            else:
                rtn = self.func
            return type(self)(rtn)
            
        return new_func
            
        

# 指定バージョン以外のときの処理
@DoOnVertion
def doSomething(name=None):
    print 'do something with argument [{0}].'.format(name)
    
    
# バージョン2015で実行したときの処理
@doSomething.version('2015')
def doSomething(name='cube'):
    cmds.polyCube(name=name)
 
 
# バージョン2016で実行したときの処理
@doSomething.version('2016')
def doSomething(name='joint'):
    cmds.createNode('joint', name=name)
   
   
# バージョン2017で実行したときの処理
@doSomething.version('2017')
def doSomething(name='transform'):
    cmds.createNode('transform', name=name)
    
    
# doSomething関数の実行
# 実行するmayaのバージョンによって処理が変わる
doSomething()

2016/08/28

[Maya]OpenMayaでNurbsCurve作成2

api1.0使用
ポイント位置のリストを受け取りカーブを作成
選択がTransformを継承しているオブジェクトだったらその下に、そうでなければ新しいオブジェクトを作ってその下にカーブシェイプを作る
import maya.OpenMaya as om

pntPoss = [
(0, 0, -1.0),
(-1, 0, 0.5),
(1, -0, 0.5),
(0, 0, -1.0),
]

def getMPoints(pntPoss):
    pnts = om.MPointArray()
    for pntPos in pntPoss:
        pnts.append(om.MPoint(*pntPos))
    return pnts
    
def getSelection():
    sl = om.MSelectionList()
    om.MGlobal.getActiveSelectionList(sl)
    if not sl.length():
        return
    dagPath = om.MDagPath()
    sl.getDagPath(0, dagPath)
    if not dagPath.hasFn(om.MFn.kTransform):
        return
    node = dagPath.node()
    return node
        
def createCurve(pntPoss):
    crv = om.MFnNurbsCurve()
    pnts = getMPoints(pntPoss)
    deg = 1
    span = pnts.length() - deg
    knotCnt = span + 2 * deg -1
    knots = om.MDoubleArray()
    for i in range(knotCnt):
        knots.append(i)
    sl = getSelection()
    if not sl:
        sl = om.MObject()
    crv.create(pnts, knots, deg, om.MFnNurbsCurve.kOpen, False, False, sl)

createCurve(pntPoss)

2016/08/23

[Maya]OpenMayaでカーブのポイントを取得

api2.0(2016以降)
import maya.api.OpenMaya as om2

def main():
    sl = om2.MGlobal.getActiveSelectionList()
    dagPath = sl.getDagPath(0)
    try:
        dagPath.extendToShape()
    except:
        return
    if not dagPath.hasFn(om.MFn.kNurbsCurve):
        return
    crv = om.MFnNurbsCurve(dagPath)
    pnts = fn_crv.cvPositions()
    for pnt in pnts:
        print pnt
        
        
main()
api1.0
import maya.OpenMaya as om
util = om.MScriptUtil()

def main():
    sl = om.MSelectionList()
    om.MGlobal.getActiveSelectionList(sl)
    dagPath = om.MDagPath()
    sl.getDagPath(0, dagPath)
    try:
        dagPath.extendToShape()
    except: return
    if not dagPath.hasFn(om.MFn.kNurbsCurve):
        return
    crv = om.MFnNurbsCurve(dagPath)
    pnts = om.MPointArray()
    crv.getCVs(pnts)
    for i in range(pnts.length()):
        pnt = pnts[i]
        print pnt.x, pnt.y, pnt.z
main()

[Maya]OpenMayaでNurbsCurve

2016以降(api2.0使用)

import maya.api.OpenMaya as om

curveFn = om.MFnNurbsCurve()
deg = 1
vts = om.MPointArray()
vts.append( om.MPoint(1,0,-1) )
vts.append( om.MPoint(-1,0,-1) )
vts.append( om.MPoint(-1,0,1) )
vts.append( om.MPoint(1,0,1) )
vts.append( om.MPoint(1,0,-1) )
span = len(vts) - deg
nknots    = span + 2*deg - 1
knots = om.MDoubleArray()
for i in xrange(nknots):
    knots.append(i)

sl = om.MGlobal.getActiveSelectionList()
dagPath = sl.getDagPath(0)
node = dagPath.node()
curve = curveFn.create( vts, knots, deg, om.MFnNurbsCurve.kOpen, False, False, node)


2015以前

import maya.OpenMaya as om

curveFn = om.MFnNurbsCurve()
deg = 1
vts = om.MPointArray()
vts.append( om.MPoint(1,0,-1) )
vts.append( om.MPoint(-1,0,-1) )
vts.append( om.MPoint(-1,0,1) )
vts.append( om.MPoint(1,0,1) )
vts.append( om.MPoint(1,0,-1) )
span = vts.length() - deg
nknots    = span + 2*deg - 1
knots = om.MDoubleArray()
for i in xrange(nknots):
    knots.append(i)

sl = om.MSelectionList()
om.MGlobal.getActiveSelectionList(sl)
dagPath = om.MDagPath()
sl.getDagPath(0, dagPath)
node = dagPath.node()
curve = curveFn.create( vts, knots, deg, om.MFnNurbsCurve.kOpen, False, False, node)