circleNode.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.
// ==========================================================================
//+

// Example Plugin: circleNode.cpp
//
// This plug-in is an example of a user-defined dependency graph node.
// It takes a number as input (such as time) and generates two output
// numbers one which describes a sine curve as the input varies and
// one that generates a cosine curve. If these two are hooked up to
// the x and z translation attributes of an object the object will describe
// move through a circle in the xz plane as time is changed.
//
// Executing the command "source circleNode" will run the MEL script which will
// create a new "Circle" menu with a single item. Selecting this will build
// a simple model (a sphere which follows a circular path) which can be played back,
// by clicking on the "play" icon on the time slider.  Note: the circleNode
// plugin needs to be loaded before the "Circle" menu item can be executed
// properly.
//
// The node has two additional attributes which can be changed to affect
// the animation, "scale" which defines the size of the circular path, and
// "frames" which defines the number of frames required for a complete circuit
// of the path. Either of these can be hooked up to other nodes, or can
// be simply set via the MEL command "setAttr" operating on the circle node
// "circleNode1" created by the MEL script. For example, "setAttr circleNode1.scale #"
// will change the size of the circle and "setAttr circleNode1.frames #"
// will cause objects to complete a circle in indicated number of frames.


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

#include <maya/MPxNode.h> 

#include <maya/MString.h> 
#include <maya/MTypeId.h> 
#include <maya/MPlug.h>

#include <maya/MFnNumericAttribute.h>
#include <maya/MVector.h>

#include <maya/MFnPlugin.h>
#include <maya/MDataBlock.h>
#include <maya/MDataHandle.h>

#include <math.h>

// The circle class defines the attributes
// and methods necessary for the circleNode plugin
//
class circle : public MPxNode
{
public:
                                                circle();
        virtual                         ~circle(); 

        virtual MStatus         compute( const MPlug& plug, MDataBlock& data );

        static  void*           creator();
        static  MStatus         initialize();
 
public:
        static  MObject         input;          // The input value.
        static  MObject         sOutput;        // The sinusoidal output value.
        static  MObject         cOutput;        // The cosinusoidal output value.
        static  MObject         frames;         // Number of frames for one circle.
        static  MObject         scale;          // Size of circle.
        static  MTypeId         id;
};

MTypeId     circle::id( 0x80005 );
MObject     circle::input;        
MObject     circle::sOutput;       
MObject     circle::cOutput;       
MObject     circle::frames;
MObject         circle::scale;

// The creator method creates an instance of the circleNode class
// and is the first method called by Maya
// when a circleNode needs to be created.
//
void* circle::creator()
{
        return new circle();
}

// The initialize routine is called after the node has been created.
// It sets up the input and output attributes and adds them to the node.
// Finally the dependencies are arranged so that when the inputs
// change Maya knowns to call compute to recalculate the output values.
// The inputs are: input, scale, frames
// The outputs are: sineOutput, cosineOutput
//
MStatus circle::initialize()
{
        MFnNumericAttribute nAttr;
        MStatus                         stat;

        // Setup the input attributes
        //
        input = nAttr.create( "input", "in", MFnNumericData::kFloat, 0.0,
                        &stat );
        CHECK_MSTATUS( stat );
        CHECK_MSTATUS( nAttr.setStorable( true ) );

        scale = nAttr.create( "scale", "sc", MFnNumericData::kFloat, 10.0,
                        &stat );
        CHECK_MSTATUS( stat );
        CHECK_MSTATUS( nAttr.setStorable( true ) );

        frames = nAttr.create( "frames", "fr", MFnNumericData::kFloat, 48.0,
                        &stat );
        CHECK_MSTATUS( stat );
        CHECK_MSTATUS( nAttr.setStorable( true ) );

        // Setup the output attributes
        //
        sOutput = nAttr.create( "sineOutput", "so", MFnNumericData::kFloat,
                        0.0, &stat );
        CHECK_MSTATUS( stat );
        CHECK_MSTATUS( nAttr.setWritable( false ) );
        CHECK_MSTATUS( nAttr.setStorable( false ) );

        cOutput = nAttr.create( "cosineOutput", "co", MFnNumericData::kFloat,
                        0.0, &stat );
        CHECK_MSTATUS( stat );
        CHECK_MSTATUS( nAttr.setWritable( false ) );
        CHECK_MSTATUS( nAttr.setStorable( false ) );

        // Add the attributes to the node
        //
        CHECK_MSTATUS( addAttribute( input ) );
        CHECK_MSTATUS( addAttribute( scale ) );
        CHECK_MSTATUS( addAttribute( frames ) );
        CHECK_MSTATUS( addAttribute( sOutput ) );
        CHECK_MSTATUS( addAttribute( cOutput ) );

        // Set the attribute dependencies
        //
        CHECK_MSTATUS( attributeAffects( input, sOutput ) );
        CHECK_MSTATUS( attributeAffects( input, cOutput ) );
        CHECK_MSTATUS( attributeAffects( scale, sOutput ) );
        CHECK_MSTATUS( attributeAffects( scale, cOutput ) );
        CHECK_MSTATUS( attributeAffects( frames, sOutput ) );
        CHECK_MSTATUS( attributeAffects( frames, cOutput ) );

        return MS::kSuccess;
} 

// The constructor does nothing
//
circle::circle() {}

// The destructor does nothing
//
circle::~circle() {}

// The compute method is called by Maya when the input values
// change and the output values need to be recomputed.
// The input values are read then using sinf() and cosf()
// the output values are stored on the output plugs.
//
MStatus circle::compute (const MPlug& plug, MDataBlock& data)
{
        
        MStatus returnStatus;
 
        // Check that the requested recompute is one of the output values
        //
        if (plug == sOutput || plug == cOutput) {
                // Read the input values
                //
                MDataHandle inputData = data.inputValue (input, &returnStatus);
                CHECK_MSTATUS( returnStatus );
                MDataHandle scaleData = data.inputValue (scale, &returnStatus);
                CHECK_MSTATUS( returnStatus );
                MDataHandle framesData = data.inputValue (frames, &returnStatus);
                CHECK_MSTATUS( returnStatus );

                // Compute the output values
                //
                float currentFrame = inputData.asFloat();
                float scaleFactor  = scaleData.asFloat();
                float framesPerCircle = framesData.asFloat();
                float angle = 6.2831853f * (currentFrame/framesPerCircle);
                float sinResult = sinf (angle) * scaleFactor;
                float cosResult = cosf (angle) * scaleFactor;

                // Store them on the output plugs
                //
                MDataHandle sinHandle = data.outputValue( circle::sOutput,
                                &returnStatus );
                CHECK_MSTATUS( returnStatus );
                MDataHandle cosHandle = data.outputValue( circle::cOutput,
                                &returnStatus );
                CHECK_MSTATUS( returnStatus );
                sinHandle.set( sinResult );
                cosHandle.set( cosResult );
                CHECK_MSTATUS( data.setClean( plug ) );
        } else {
                return MS::kUnknownParameter;
        }

        return MS::kSuccess;
}

// The initializePlugin method is called by Maya when the circleNode
// plugin is loaded.  It registers the circleNode which provides
// Maya with the creator and initialize methods to be called when
// a circleNode is created.
//
MStatus initializePlugin ( MObject obj )
{ 
        MStatus   status;
        MFnPlugin plugin( obj, PLUGIN_COMPANY, "4.5", "Any");

        status = plugin.registerNode( "circle", circle::id,
                                                  circle::creator, circle::initialize );
        if (!status) {
                status.perror("registerNode");
                return status;
        }

        return status;
}

// The unitializePlugin is called when Maya needs to unload the plugin.
// It basically does the opposite of initialize by calling
// the deregisterCommand to remove it.
//
MStatus uninitializePlugin( MObject obj)
{
        MStatus   status;
        MFnPlugin plugin( obj );

        status = plugin.deregisterNode( circle::id );
        if (!status) {
                status.perror("deregisterNode");
                return status;
        }

        return status;
}