`
schi
  • 浏览: 205231 次
  • 性别: Icon_minigender_1
  • 来自: 上海
社区版块
存档分类
最新评论

scanDagCmd

阅读更多

scanDag命令以depth first(深度优先)或breadth first(广度优先)来迭代整个DAG,输出每个节点的名称和深度等信息,这个命令插件例子中只对相机、灯光、NURBS surfaces进行支持。
在maya的Plug-in Manager中加载scanDagCmd.py,然后你可以随便创建一些相机、灯光、NURBS surfaces,并对它们进行位移等属性调整,然后分别执行

// -d 为深度优先,-b 为广度优先
// 相机
scanDag -d 1 -c 1;
// 灯光
scanDag -d 1 -l 1;
// NURBS surfaces
scanDag -d 1 -l 1;
// 或全部一起
scanDag -d 1 -c 1 -l 1 -n 1;

 或

import maya.cmds as cmds
# 相机
cmds.scanDag(d=1, c=1)
# 灯光
cmds.scanDag(d=1, l=1)

 你会得到类似的输出结果

Output
Starting Depth First scan of the Dag: Filtering for Cameras
perspShape: camera
dagPath: |persp|perspShape
translation: 28.000000 21.000000 28.000000
rotation: [-0.487616, 0.785398, 0.000000]
scale: [1.000000, 1.000000, 1.000000]
eyePoint: 28.000000 21.000000 28.000000
upDirection: -0.331295 0.883452 -0.331295
viewDirection: -0.624695 -0.468521 -0.624695
aspectRatio: 1.500000
horizontalFilmAperture: 1.417320
verticalFilmAperture: 0.944880
topShape: camera
dagPath: |top|topShape
translation: 0.000000 100.100000 0.000000
rotation: [-1.570796, 0.000000, 0.000000]
scale: [1.000000, 1.000000, 1.000000]
eyePoint: 0.000000 100.100000 0.000000
upDirection: 0.000000 0.000000 -1.000000
viewDirection: 0.000000 -1.000000 -0.000000
aspectRatio: 1.500000
horizontalFilmAperture: 1.417320
verticalFilmAperture: 0.944880
frontShape: camera
dagPath: |front|frontShape
translation: 0.000000 0.000000 100.100000
rotation: [0.000000, -0.000000, 0.000000]
scale: [1.000000, 1.000000, 1.000000]
eyePoint: 0.000000 0.000000 100.100000
upDirection: 0.000000 1.000000 0.000000
viewDirection: 0.000000 0.000000 -1.000000
aspectRatio: 1.500000
horizontalFilmAperture: 1.417320
verticalFilmAperture: 0.944880
sideShape: camera
dagPath: |side|sideShape
translation: 100.100000 0.000000 0.000000
rotation: [-0.000000, 1.570796, 0.000000]
scale: [1.000000, 1.000000, 1.000000]
eyePoint: 100.100000 0.000000 0.000000
upDirection: 0.000000 1.000000 0.000000
viewDirection: -1.000000 0.000000 -0.000000
aspectRatio: 1.500000
horizontalFilmAperture: 1.417320
verticalFilmAperture: 0.944880
// Result: 4 //

 scanDagCmd.py

# -*- coding: UTF-8 -*-

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

# 命令名称
kCmdName = 'scanDag'

# 命令参数
kBreadthFlag = '-b'
kBreadthFlagLong = 'breadthFirst'
kDepthFlag = '-d'
kDepthFlagLong = '-depthFirst'
kCameraFlag = '-c'
kCameraFlagLong = '-cameras'
kLightFlag = '-l'
kLightFlagLong = '-lights'
kNurbsSurfaceFlag = '-n'
kNurbsSurfaceFlagLong = '-nurbsSurfaces'
kQuietFlag = '-q'
kQuietFlagLong = '-quiet'

# 定义命令
class ScanDag(ompx.MPxCommand):
    
    def __init__(self):
        super(ScanDag, self).__init__()
        
    def doIt(self, args):
        # 初始化默认变量,防止无参数命令出现
        self._traversalType = om.MItDag.kDepthFirst
        self._filter = om.MFn.kInvalid
        self._quiet = False
        
        self.__parseArgs(args)
        
        return self.__doScan()
    
    def __parseArgs(self, args):
        '''
        检查命令所给的参数,并设置相应的变量
        '''
        argData = om.MArgDatabase(self.syntax(), args)
        if argData.isFlagSet(kBreadthFlag) or argData.isFlagSet(kBreadthFlagLong):
            self._traversalType = om.MItDag.kBreadthFirst
        if argData.isFlagSet(kDepthFlag) or argData.isFlagSet(kDepthFlagLong):
            self._traversalType = om.MItDag.kDepthFirst
        if argData.isFlagSet(kCameraFlag) or argData.isFlagSet(kCameraFlagLong):
            self._filter = om.MFn.kCamera
        if argData.isFlagSet(kLightFlag) or argData.isFlagSet(kLightFlagLong):
            self._filter = om.MFn.kLight
        if argData.isFlagSet(kNurbsSurfaceFlag) or argData.isFlagSet(kNurbsSurfaceFlagLong):
            self._filter = om.MFn.kNurbsSurface
        if argData.isFlagSet(kQuietFlag) or argData.isFlagSet(kQuietFlagLong):
            self._quiet = True
        
        return True
    
    def __doScan(self):
        '''
        完成实际的工作
        '''
        # 创建dag迭代器
        dagIterator = om.MItDag(self._traversalType, self._filter)
        
        # Scan the entire DAG and output the name and depth of each node
        # 扫描整个DAG,输出每个节点的名称和深度
        if self._traversalType == om.MItDag.kBreadthFirst:
            if not self._quiet:
                sys.stdout.write('Starting Breadth First scan of the Dag')
        else:
            if not self._quiet:
                sys.stdout.write('Starting Depth First scan of the Dag')
                
        if self._filter == om.MFn.kCamera:
            if not self._quiet:
                sys.stdout.write(': Filtering for Cameras\n')
        elif self._filter == om.MFn.kLight:
            if not self._quiet:
                sys.stdout.write(': Filtering for Lights\n')
        elif self._filter == om.MFn.kNurbsSurface:
            if not self._quiet:
                sys.stdout.write(': Filtering for Nurbs Surfaces\n')
        else:
            sys.stdout.write('\n')
            
        # 对DAG里的每个节点进行迭代
        objectCount = 0
        while not dagIterator.isDone():
            dagPath = om.MDagPath()
            # 获取节点的DAG路径
            dagIterator.getPath(dagPath)
            # 使用方法集来操作节点
            dagNode = om.MFnDagNode(dagPath)
            
            # 输出节点的名称,类型,全路径
            if not self._quiet:
                sys.stdout.write('%s: %s\n' % (dagNode.name(), dagNode.typeName()))
                
            if not self._quiet:
                sys.stdout.write('  dagPath: %s\n' % dagPath.fullPathName())
                
            # 判别节点的类型,这里只对相机、灯光、NURBS surfaces进行操作
            # 实际上是可以对任意类型,进行相应的操作
            objectCount += 1
            if dagPath.hasFn(om.MFn.kCamera):
                camera = om.MFnCamera(dagPath)
                
                # Get the translation/rotation/scale data
                # 获取translation/rotation/scale等数据
                self.__printTransformData(dagPath)
                
                # Extract some interesting Camera data
                # 分离一些有趣的相机数据
                if not self._quiet:
                    eyePoint = camera.eyePoint(om.MSpace.kWorld)
                    upDirection = camera.upDirection(om.MSpace.kWorld)
                    viewDirection = camera.viewDirection(om.MSpace.kWorld)
                    aspectRatio = camera.aspectRatio()
                    hfa = camera.horizontalFilmAperture()
                    vfa = camera.verticalFilmAperture()
                    
                    sys.stdout.write("  eyePoint: %f %f %f\n" % (eyePoint.x, eyePoint.y, eyePoint.z))
                    sys.stdout.write("  upDirection: %f %f %f\n" % (upDirection.x, upDirection.y, upDirection.z))
                    sys.stdout.write("  viewDirection: %f %f %f\n" % (viewDirection.x, viewDirection.y, viewDirection.z))
                    sys.stdout.write("  aspectRatio: %f\n" % aspectRatio)
                    sys.stdout.write("  horizontalFilmAperture: %f\n" % hfa)
                    sys.stdout.write("  verticalFilmAperture: %f\n" % vfa)
            elif dagPath.hasFn(om.MFn.kLight):
                light = om.MFnLight(dagPath)
                
                # Get the translation/rotation/scale data
                # 获取translation/rotation/scale等数据
                self.__printTransformData(dagPath)
                
                # Extract some interesting Light data
                # 分离一些有趣的灯光数据
                color = om.MColor()
                
                color = light.color()
                if not self._quiet:
                    sys.stdout.write("  color: [%f, %f, %f]\n" % (color.r, color.g, color.b))
                    
                color = light.shadowColor()
                if not self._quiet:
                    sys.stdout.write("  shadowColor: [%f, %f, %f]\n" % (color.r, color.g, color.b))
                    sys.stdout.write("  intensity: %f\n" % light.intensity())
            elif dagPath.hasFn(om.MFn.kNurbsSurface):
                surface = om.MFnNurbsSurface(dagPath)
                
                # Get the translation/rotation/scale data
                # 获取translation/rotation/scale等数据
                self.__printTransformData(dagPath)
                
                # Extract some interesting Surface data
                # 分离一些有趣的NURBS surfaces数据
                if not self._quiet:
                    sys.stdout.write("  numCVs: %d * %d\n" % (surface.numCVsInU(), surface.numCVsInV()))
                    sys.stdout.write("  numKnots: %d * %d\n" % (surface.numKnotsInU(), surface.numKnotsInV()))
                    sys.stdout.write("  numSpans: %d * %d\n" % (surface.numSpansInU(), surface.numSpansInV()))
            else:
                # Get the translation/rotation/scale data
                # 获取translation/rotation/scale等数据
                self.__printTransformData(dagPath)
                
            dagIterator.next()
            
        # 返回命令结果,这个结果为所操作的节点的数量值
        self.setResult(objectCount)
        return om.MStatus.kSuccess
    
    def __printTransformData(self, dagPath):
        '''
        获取translation/rotation/scale等数据
        '''
        # 获取节点的Transform节点
        transformNode = dagPath.transform()
        transform = om.MFnDagNode(transformNode)
        # 获取Transform的Transformation matrix
        matrix = om.MTransformationMatrix(transform.transformationMatrix())
        
        if not self._quiet:
            # 获取节点的位移数值,并输出
            translation = matrix.translation(om.MSpace.kWorld)
            sys.stdout.write('  translation: %f %f %f\n' % (translation.x, translation.y, translation.z))
            
        #rOrder = om.MTransformationMatrix.kXYZ
        #matrix.getRotation(ptr, rOrder)
        
        # 获取节点的旋转数值,并输出
        rotateOrder = 0
        mquat = matrix.rotation()
        rot = mquat.asEulerRotation()
        rot.reorderIt(rotateOrder)
        
        if not self._quiet:
            sys.stdout.write("  rotation: [%f, %f, %f]\n" % (rot.x, rot.y, rot.z))
            
        # 获取节点的缩放数值,并输出
        # 这里需要使用MScriptUtil
        # 因为MTransformationMatrix.getScale()
        # 的第一个参数是C++里的指针
        util = om.MScriptUtil()
        util.createFromDouble(0.0, 0.0, 0.0)
        ptr = util.asDoublePtr()

        matrix.getScale(ptr, om.MSpace.kWorld)
        
        scaleX = util.getDoubleArrayItem(ptr, 0)
        scaleY = util.getDoubleArrayItem(ptr, 1)
        scaleZ = util.getDoubleArrayItem(ptr, 2)
        if not self._quiet:
            sys.stdout.write('  scale: [%f, %f, %f]\n' % (scaleX, scaleY, scaleZ))
            
    
# Creator
def cmdCreator():
    return ompx.asMPxPtr(ScanDag())

# Syntax creator
def syntaxCreator():
    syntax = om.MSyntax()
    syntax.addFlag(kBreadthFlag, kBreadthFlagLong, om.MSyntax.kBoolean)
    syntax.addFlag(kDepthFlag, kDepthFlagLong, om.MSyntax.kBoolean)
    syntax.addFlag(kCameraFlag, kCameraFlagLong, om.MSyntax.kBoolean)
    syntax.addFlag(kLightFlag, kLightFlagLong, om.MSyntax.kBoolean)
    syntax.addFlag(kNurbsSurfaceFlag, kNurbsSurfaceFlagLong, om.MSyntax.kBoolean)
    syntax.addFlag(kQuietFlag, kQuietFlagLong, om.MSyntax.kBoolean)
    return syntax

# Initialize the script plug-in
def initializePlugin(mobj):
    mplugin = ompx.MFnPlugin(mobj, 'Mack Stone', '3.0', 'Any')
    try:
        mplugin.registerCommand(kCmdName, cmdCreator, syntaxCreator)
    except:
        sys.stderr.write('Failed to register command: %s\n' % kCmdName)
        raise
    
# Uninitialize the script plug-in
def uninitializePlugin(mobj):
    mplugin = ompx.MFnPlugin(mobj)
    try:
        mplugin.deregisterCommand(kCmdName)
    except:
        sys.stderr.write('Failed to unregister command: %s\n' % kCmdName)
        raise

 你可以在devkit\plug-ins中找到scanDagCmd.cpp
在线版本

http://download.autodesk.com/us/maya/2010help/API/scan_dag_cmd_8cpp-example.html

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics