surfaceCreateCmd/surfaceCreateCmd.cpp

//-
// ==========================================================================
// Copyright 1995,2006,2008 Autodesk, Inc. All rights reserved.
//
// Use of this software is subject to the terms of the Autodesk
// license agreement provided at the time of installation or download,
// or which otherwise accompanies this software in either electronic
// or hard copy form.
// ==========================================================================
//+

// DESCRIPTION:
// 
// Produces the MEL command "surfaceCreate".
//
// This plug-in creates a NURBS surface by supplying its own CVs and knots.
// To use it, enter the command "surfaceCreate". 
//

#include <maya/MStatus.h>
#include <maya/MObject.h>
#include <maya/MFnPlugin.h>
#include <maya/MDoubleArray.h>
#include <maya/MPointArray.h>
#include <maya/MPoint.h>
#include <maya/MFnNurbsSurface.h>
#include <maya/MString.h>
#include <maya/MPxCommand.h>

#include <maya/MIOStream.h> 
#include <math.h> 

class surfaceCreate : public MPxCommand {
public:

                    surfaceCreate();
    virtual         ~surfaceCreate();

    virtual MStatus doIt( const MArgList& args );

    static void*    creator();
};

#define NUM_SPANS        30
#define WIDTH            10.0
#define VERTICAL_SCALING 4.0

surfaceCreate::~surfaceCreate() {}

surfaceCreate::surfaceCreate() {}

void* surfaceCreate::creator()
{
    return new surfaceCreate();
}

MStatus surfaceCreate::doIt( const MArgList& )
//
//  Description:
//      Plugin command to test Selection List Iterator.
//
//
{  
    // Set up knots
    //
    MDoubleArray knotArray;
    int i;
    // Add extra starting knots so that the first CV matches
    // the curve start point
    //
    knotArray.append( 0.0 );
    knotArray.append( 0.0 );
    for ( i = 0; i <= NUM_SPANS; i++ ) {
        knotArray.append( (double)i );
    }
    // Add extra ending knots so that the last CV matches the curve end point
    //
    knotArray.append( (double)i );
    knotArray.append( (double)i );
 
    // Now, Set up CVs
    //
    MPointArray cvArray;
    
    // We need a 2D array of CVs with NUM_SPANS + 3 CVs on a side
    //
    int last = NUM_SPANS + 3;
    for ( i = 0; i < last; i++ ) {
        for ( int j = 0; j < last; j++ ) {
            MPoint cv;
            cv.x = (((double)(j))/((double)(NUM_SPANS + 3)) * WIDTH) 
                - (WIDTH/2.0);
            cv.z = (((double)(i))/((double)(NUM_SPANS + 3)) * WIDTH) 
                - (WIDTH/2.0);
            double dist = sqrt( cv.x*cv.x + cv.z*cv.z );
            cv.y = cos( dist ) * VERTICAL_SCALING;
            cvArray.append( cv );
        }
    }

    // Create the surface
    // 
    MFnNurbsSurface mfnNurbsSurf;

    MStatus stat;
    mfnNurbsSurf.create( cvArray, knotArray, knotArray, 3, 3, 
                         MFnNurbsSurface::kOpen, MFnNurbsSurface::kOpen,
                         true, MObject::kNullObj, &stat );

    if ( MS::kSuccess != stat )
        cerr << "surfaceCreate failed: status " << stat << endl;

    return stat;
}

MStatus initializePlugin( MObject obj )
{
    MStatus   status;
    MFnPlugin plugin( obj, PLUGIN_COMPANY, "3.0", "Any");

    status = plugin.registerCommand( "surfaceCreate", surfaceCreate::creator );
    if (!status) {
        status.perror("registerCommand");
        return status;
    }
    
    return status;
}

MStatus uninitializePlugin( MObject obj )
{
    MStatus   status;
    MFnPlugin plugin( obj );
    status = plugin.deregisterCommand( "surfaceCreate" );
    if (!status) {
        status.perror("registerCommand");
        return status;
    }
    
    return status;
}