【Maya Python】 選択したパーティクルの軌跡をカーブで作成。

createCurveOfParticleTrail.py

# -*- coding:utf-8 -*-

import maya.cmds as cmds


def createCurveOfParticleTrail(startF, endF):
    u"""
    概要:
        選択したパーティクルの軌跡をカーブで作成。
        particle, nParticle どちらでもOK。
    引数:
        startF(int) : 開始フレーム。
        endF(int)   : 終了フレーム。
    戻り値:
        string : 作成されたカーブのグループ名。
    """

    sel = cmds.ls(sl=True)
    nodeType = cmds.nodeType(cmds.listRelatives(sel[0], s=True))
    if nodeType != "particle":
        cmds.error("Please select a particle node.")

    ptkey = {}
    for i in range(endF - startF + 1):
        cmds.currentTime(startF + i)
        ptnum = cmds.getAttr(sel[0] + ".count")
        if ptnum == 0:
            pass
        else:
            for ptnums in range(ptnum):
                if sel[0] + ".pt[%d]" % ptnums in ptkey.keys():
                    ptkey[sel[0] + ".pt[%d]" % ptnums].append(
                        cmds.pointPosition(sel[0] + ".pt[%d]" % ptnums, w=True))
                else:
                    ptkey[sel[0] + ".pt[%d]" %ptnums] = [cmds.pointPosition(sel[0] + ".pt[%d]" % ptnums, w=True)]

    curveList = []
    for curves in range(len(ptkey.keys())):
        curve = cmds.curve(
            p=ptkey[sel[0] + ".pt[%d]" % curves], d=1, n="%s_curve%d" % (sel[0], curves + 1))
        curveList.append(curve)

    return cmds.group(curveList, n="%s_curveGrp" % sel[0])


createCurveOfParticleTrail(1, 50)