Maya API转换curve为螺旋曲线

创建一条curve，输入命令，使其转会为螺旋曲线

 

#include <stdio.h>
#include <math.h>
#include <maya/MFnPlugin.h>
#include <maya/MFnNurbsCurve.h>
#include <maya/MPointArray.h>
#include <maya/MDoubleArray.h>
#include <maya/MPoint.h>
#include <maya/MSelectionList.h>
#include <maya/MItSelectionList.h>
#include <maya/MItCurveCV.h>
#include <maya/MGlobal.h>
#include <maya/MDagPath.h>
#include <maya/MString.h>
#include <maya/MPxCommand.h>
#include <maya/MArgList.h>
class helix2 : public MPxCommand
{
public:
    helix2();
    virtual ~helix2();
    MStatus doIt( const MArgList& args );
    MStatus redoIt();
    MStatus undoIt();
    bool isUndoable() ;
    static void * creator();
private:
    MDagPath    fdagPath;
    MPointArray fCVs;
    double        radius;
    double        pitch;

};
void *helix2::creator()
{
    return new helix2;
}
helix2::helix2():radius(4.0),pitch(0.5){}
helix2::~helix2(){}
MStatus helix2::doIt( const MArgList& args )
{
    MStatus status;
    int i ;
    for ( i = 0 ; i< args.length(); i++)
        if (MString("-p") == args.asString(i,&status)&&MS::kSuccess == status)
        {
            double tmp = args.asDouble(++i,&status);
            if(MS::kSuccess == status)
                pitch = tmp;
        }
        else if (MString("-r") == args.asString(i,&status)&&MS::kSuccess == status)
        {
            double tmp = args.asDouble(++i,&status);
            if(MS::kSuccess == status )
                pitch = tmp;

        }
        else
        {
            MString msg = "Invalid flag: ";
            msg += args.asString(i);
            displayError(msg);
            return MS::kSuccess;
        }
        MSelectionList slist;
        MGlobal::getActiveSelectionList(slist);
        MItSelectionList list(slist,MFn::kNurbsCurve,&status);
        if(MS::kSuccess != status )
        {
            displayError("Could not create selection list iterator");
            return status;
        }
        if(list.isDone())
        {
            displayError("No curve selected");
            return MS::kFailure;
        }
        MObject component;
        list.getDagPath(fdagPath,component);
        return redoIt();
}

MStatus helix2::redoIt()
{
    unsigned i,numCVs;
    MStatus status;
    MFnNurbsCurve curveFn(fdagPath);
    numCVs = curveFn.numCVs();
    status = curveFn.getCVs(fCVs);
    if (MS::kSuccess != status )
    {
        displayError("Could not get curve`s CVs");
        return MS::kFailure;
    }
    MPointArray points(fCVs);
    for(i=0;i<numCVs;i++)
        points[i] = MPoint(radius*cos((double )i),pitch*(double)i,radius*sin((double)i));
    status = curveFn.setCVs(points);
    if (MS::kSuccess != status)
    {
        displayError("Could not set new CV infomation");
        fCVs.clear();
        return status;
    }
    status = curveFn.updateCurve();
    if(MS::kSuccess!=status)
    {
        displayError("Could not update curve");
        return status;
    }
    return MS::kSuccess;
}
MStatus helix2::undoIt()
{
    MStatus status;
    MFnNurbsCurve curveFn(fdagPath);
    status = curveFn.setCVs(fCVs);
    if(MS::kSuccess!=status)
    {
        displayError("Could not set old CV information");
        return status;
    }
    status = curveFn.updateCurve();
    if(MS::kSuccess!=status)
    {
        displayError("Could not update curve");
        return status;
    }
    return MS::kSuccess;
}
MStatus initializePlugin(MObject obj)
{
    MFnPlugin plugin(obj,"lijy","1.0","any");
    plugin.registerCommand("helix2",helix2::creator);
    return MS::kSuccess;
}
MStatus uninitializePlugin(MObject obj)
{
    MFnPlugin plugin(obj);
    plugin.deregisterCommand("helix2");
    return MS::kSuccess;
}