helixCmd.py

之前的helloWorldCmd.py只是一个单独的命令,没有任何命令参数。而helixCmd.py则是一个带参数的命令。
可能有不少人都会用到helix的曲线,但maya里的创建曲线菜单里是没有helix的。其实maya一直都有提供这个插件来创建helix的曲线,至于为什么不把它加入到创建菜单里我就不清楚了。
helix的数学算法
http://en.wikipedia.org/wiki/Helix#cite_note-4




需要注意的是,maya的默认轴向和上面的是不同的,maya是Y向上的,而上面的是Z向上,所以要将Y和Z调换。

helixCmd.py
#!/usr/bin/env python
# -*- coding: UTF-8 -*-

'''
Created on Oct 25, 2009

@author: schi
'''

# 描述:
#     创建一个螺旋状的NURBS曲线
#
# 参数:
#     p=    #螺旋曲线的间距,默认是0.5
#     r=    #螺旋曲线的半径,默认是4.0
#
# 使用例子:
#     python:
#         import maya.cmds as cmds
#         cmds.spHelix(p=0.3,r=7)
#
#     Mel:
#         spHelix -p 0.3 -r 7;

import maya.OpenMaya as om
import maya.OpenMayaMPx as ompx
import sys, math

# 定义命令的名称
kPluginCmdName = 'spHelix'
# 定义参数的名称
kPitchFlag = '-p'
kPitchLongFlag = '-pitch'
kRadiusFlag = '-r'
kRadiusLongFlag = '-radius'

class SpHelixCmd( ompx.MPxCommand ):

    def __init__( self ):
        super( SpHelixCmd, self ).__init__()

    def doIt( self, args ):
        # 定义所需要的数值,如:cv点的数量,曲线的间距,半径等
        deg = 3
        ncvs = 20
        spans = ncvs - deg
        nknots = spans + 2 * deg - 1
        radius = 4.0
        pitch = 0.5
        
        # 剖析参数
        argData = om.MArgDatabase( self.syntax(), args )
        # 查看在使用命令时是否设置-p和-r参数
        # isFlagSet()是MArgParser的方法
        # MArgDatabase继承MArgParser
        # flagArgumentDouble()也一样
        if argData.isFlagSet( kPitchFlag ):
            pitch = argData.flagArgumentDouble( kPitchFlag, 0 )
        if argData.isFlagSet( kRadiusFlag ):
            radius = argData.flagArgumentDouble( kRadiusFlag, 0 )
            
        controlVertices = om.MPointArray()
        knotSequences = om.MDoubleArray()
        
        # 给helix建立cv点
        for i in range( 0, ncvs ):
            controlVertices.append( om.MPoint( radius * math.cos( i ),
                                             pitch * i,
                                             radius * math.sin( i ) ) )
            
        for i in range( 0, nknots ):
            knotSequences.append( i )
            
        # 创建曲线
        curveFn = om.MFnNurbsCurve()
        nullObj = om.MObject()
        
        try:
            # 这个插件是通过剖析cv点来创建曲线的。
            # 通过ep点来创建曲线的方法也已经添加,
            # 只要设置为False就行
            if True:
                curveFn.create( controlVertices,
                               knotSequences,
                               deg,
                               om.MFnNurbsCurve.kOpen,
                               0, 0,
                               nullObj )
            else:
                curveFn.create( controlVertices,
                               3,
                               om.MFnNurbsCurve.kOpen,
                               False, False, False )
        except:
            sys.stderr.write( '创建曲线时发生错误\n' )
            raise 
        
# Creator
def cmdCreator():
    # Create the command
    return ompx.asMPxPtr( SpHelixCmd() )

# Syntax creator
# 命令样式创建
def syntaxCreator():
    syntax = OpenMaya.MSyntax()
    syntax.addFlag( kPitchFlag, kPitchLongFlag, om.MSyntax.kDouble )
    syntax.addFlag( kRadiusFlag, kRadiusLongFlag, om.MSyntax.kDouble )
    return syntax

# 注册插件
def initializePlugin( mobject ):
    mplugin = ompx.MFnPlugin( mobject )
    try:
        mplugin.registerCommand( kPluginCmdName, cmdCreator )
    except:
        sys.stderr.write( '插件 %s 注册失败\n' % kPluginCmdName )
        raise

# 注销插件
def uninitializePlugin( mobject ):
    mplugin = ompx.MFnPlugin( mobject )
    try:
        mplugin.deregisterCommand( kPluginCmdName )
    except:
        sys.stderr.write( '插件 %s 注销失败\n' % kPluginCmdName )
        raise

之前忘了说明initializePlugin和uninitializePlugin这两个方法的作用,maya在加载插件时会调用initializePlugin方法,如果不存在将无法加载,如果加载出错会输出错误信息;同样当你不需要这个插件时可以卸载它,这时就会调用uninitializePlugin方法,否则无法卸载。

你可以在maya安装目录下的devkit/plug-ins/scripted找到helixCmd.py。
在线版
http://download.autodesk.com/us/maya/2010help/API/helix_cmd_8py-example.html

你可能感兴趣的:(算法,python)