SoftVolumeSphere.cpp

#include "SoftVolumeSphere.h"
#include <maya/MFnDagNode.h>
#include <maya/MColor.h>

const MTypeId SoftVolumeSphere::typeId( 0x70015 );

const MString SoftVolumeSphere::typeName( "softVolumeSphere" );


MObject SoftVolumeSphere::m_Radius;


MObject SoftVolumeSphere::m_Weight;


void SoftVolumeSphere:: OnDraw( M3dView::DisplayStyle style,
                                                        M3dView::DisplayStatus status)
{
        float radius,inner;

        // we need to determine the size of the loocator node by reading back the
        // attributes contained on this node. To do this, we must attach a function
        // set to this node, and then read back the attributes.
        //
        MFnDagNode fn( thisMObject() );

        MPlug RadiusPlug = fn.findPlug( m_Radius );
        RadiusPlug.getValue( radius );

        // the MPxLocator::colorRGB() function returns the user specified colour for
        // a given selection state. For example, by default a selected item is green;
        // the user may however change this within the user preferences and say change
        // this to pink. In order for our node to maintain continuity with the rest
        // of the users settings, we should use this colour value somewhere obvious
        // on our nodes.
        MColor outlineCol = colorRGB(status);

        // if the object is the last selected, there seems to be no colour for this.
        // therefore i check to see if the locator is the last selected item and then
        // just use green. I might just be doing something stupid though ;)
        //
        if(status == M3dView::kLead )
                glColor4f(0.0f,1.0f,0.0f,0.5f);
        else
                // Otherwise i use a slightly varied intensity of the colour
                glColor4f(outlineCol.r,outlineCol.g,outlineCol.b,0.5f);

        glBlendFunc(GL_SRC_ALPHA,GL_ONE_MINUS_SRC_ALPHA);
        glEnable(GL_BLEND);
                DrawCircleXY( radius, 32 );
                DrawCircleXZ( radius, 32 );
                DrawCircleYZ( radius, 32 );
                DrawCrossXY( 2*radius, 2*radius );
                DrawCrossXZ( 2*radius, 2*radius );
                DrawCrossYZ( 2*radius, 2*radius );
        glDisable(GL_BLEND);


}


MBoundingBox SoftVolumeSphere::boundingBox() const
{
        float radius,inner;

        // we need to determine the size of the loocator node by reading back the
        // attributes contained on this node. To do this, we must attach a function
        // set to this node, and then read back the attributes.
        //
        MFnDagNode fn( thisMObject() );

        MPlug RadiusPlug = fn.findPlug( m_Radius );
        RadiusPlug.getValue( radius );

        // expand the bounding box to fit all axes of the locator node
        MBoundingBox bbox;

        bbox.expand( MPoint( -radius, 0.0f, 0.0f ) );
        bbox.expand( MPoint(  radius, 0.0f, 0.0f ) );

        bbox.expand( MPoint(  0.0f,radius,  0.0f ) );
        bbox.expand( MPoint(  0.0f,-radius,  0.0f ) );

        bbox.expand( MPoint(  0.0f, 0.0f, -radius ) );
        bbox.expand( MPoint(  0.0f, 0.0f,  radius ) );

        return bbox;
}

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


MStatus SoftVolumeSphere::initialize()
{
        // in maya, all attributes are individual objects. The nodes within maya
        // maintain their own internal list of attribute objects. Thankfully
        // we don't have to manipulate this list of attributes ourselves. We simply
        // have to create all of the attributes and then add them to the node.
        //
        // A number of function sets allow you to create attributes of various
        // types, You may want attributes of various types, ie time, colour,
        // integer, float, string, bool, array etc, etc...
        //
        // The most common function set for creating an attribute is
        // MFnNumericAttribute since it handles floats, ints etc. In this case
        // however we want to add attributes that represents the nodes size.
        // This should be done using the MFnUnitAttribute function set. The
        // reason for this is that sizes may change since maya's units can
        // be changed to inches, meters, feet etc.
        //
        MFnUnitAttribute fn;
        MFnNumericAttribute nfn;

        MStatus stat;

        // create a size attribute. We specify the long and short name
        // for the attribute, as well as the data type.
        m_Radius = fn.create( "radius", "r", MFnUnitAttribute::kDistance );

        // set a default value of 1 for the size
        fn.setDefault( MDistance(5.0, MDistance::uiUnit()) );

        // An attribute will only appear in the channel box if it is keyable.
        // this means we usually make most attributes keyable, even if we
        // never want to animate them. (since the channel box is handy for
        // setting values).
        //
        // If we have an attribute that we want in the channel box, but we do
        // not want it to be keyed; we can overide the MPxNode::legalConnection()
        // function to say that anim curves should not be attached to the
        // non keyable attribute.
        //
        fn.setKeyable( true );

        // set a minim value for the attribute
        fn.setMin( 0.1f );

        // finally add the attribute to this node.
        stat = addAttribute( m_Radius );

        // check for error
        if( stat != MS::kSuccess ) {
                stat.perror("Unable to attach radius attribute to the softVolumeSphere node");
                return stat;
        }
        
        m_Weight = nfn.create("weight","w",MFnNumericData::kFloat);
        nfn.setDefault(1.0f);
        nfn.setKeyable(true);
        stat = addAttribute(m_Weight);
        
        // check for error
        if( stat != MS::kSuccess ) {
                stat.perror("Unable to attach innerSize attribute to the softVolumeSphere node");
                return stat;
        }
        
        return MS::kSuccess;
}

---