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

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

#include <maya/MPxNode.h> 

#include <maya/MFnNumericAttribute.h>
#include <maya/MFnGenericAttribute.h>
#include <maya/MFnPlugin.h>

#include <maya/MTypeId.h> 
#include <maya/MPlug.h>
#include <maya/MDataBlock.h>
#include <maya/MDataHandle.h>
#include <maya/MGlobal.h>
 
//
// Node definition
//
class genericAttributeNode : public MPxNode
{
public:
						genericAttributeNode();
	virtual				~genericAttributeNode(); 

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

	static  void*		creator();
	static  MStatus		initialize();

	static MStatus	addComplexFloatGenericAttribute( MObject& attrObject, const char *longName, const char *shortName );
	
public:
	// Numeric attribute
	static MObject		gInputInt;
	// Generic attribute that can be a float, 2float, 3float
	static MObject		gOutputFloat_2Float_3Float;
	// Type id
	static	MTypeId		id;
};

// Static initialization
MTypeId genericAttributeNode::id( 0x81031 );
MObject genericAttributeNode::gInputInt;
MObject genericAttributeNode::gOutputFloat_2Float_3Float;

// No-op constructor and destructor
genericAttributeNode::genericAttributeNode() {}
genericAttributeNode::~genericAttributeNode() {}

//
// Compute that will update the generic attribute
// when requested.
//
MStatus genericAttributeNode::compute( const MPlug& plug, MDataBlock& data )
{
	MStatus returnStatus;
 
	if ( plug == gOutputFloat_2Float_3Float )
	{
		// attribute affecting generic attribute case.  Based on the
		// input attribute, we modify the output generic attribute
		MDataHandle inputDataInt = data.inputValue( gInputInt );
		int inputInt = inputDataInt.asInt();
		
		// Get the output handle
		MDataHandle outputData = data.outputValue( plug );	
		bool isGenericNumeric = false;
		bool isGenericNull = false;
		
		// Is the output handle generic data
		if ( outputData.isGeneric( isGenericNumeric, isGenericNull ) )
		{
			// Based on the inputHandle, update the generic
			// output handle
			if ( inputInt == 1 )
				outputData.setGenericBool( false, true );
			else if ( inputInt == 2 )
				outputData.setGenericBool( true, true );
			else if ( inputInt == 3 )
				outputData.setGenericChar( 127, true );
			else if ( inputInt == 4 )
				outputData.setGenericDouble( 3.145, true );
			else if ( inputInt == 5 )
				outputData.setGenericFloat( (float)9.98, true );	
			else if ( inputInt == 6 )
				outputData.setGenericShort( 3245, true );
			else if ( inputInt == 7 )
				outputData.setGenericInt( 32768, true );
			else if ( inputInt == 8 )
			{
				MFnNumericData numericData;
				MObject obj = numericData.create( MFnNumericData::k2Float, &returnStatus );
				CHECK_MSTATUS( returnStatus );
				
				returnStatus = numericData.setData( (float)1.5, (float)6.7 );
				CHECK_MSTATUS( returnStatus );

				outputData.set( obj );
			}
			else if ( inputInt == 9 )
			{
				MFnNumericData numericData;
				MObject obj = numericData.create( MFnNumericData::k3Float, &returnStatus );
				CHECK_MSTATUS( returnStatus );

				returnStatus = numericData.setData( (float)2.5, (float)8.7, (float)2.3345 );
				CHECK_MSTATUS( returnStatus );

				outputData.set( obj );
			}
			else if ( inputInt == 10 )
			{
				outputData.setGenericInt( 909, true );
			}							

			// Mark the data clean
			outputData.setClean();
			data.setClean( gOutputFloat_2Float_3Float );
		}
	} 
	else 
	{
		return MS::kUnknownParameter;
	}

	return MS::kSuccess;
}

// Creator
void* genericAttributeNode::creator()
{
	return new genericAttributeNode();
}

// Adds a generic attribute that accepts a float, float2, float3
MStatus	genericAttributeNode::addComplexFloatGenericAttribute( 
	MObject& attrObject,
	const char *longName, const char *shortName )
{
	// Create the generic attribute and set the 3 accepts types
	MFnGenericAttribute gAttr; 
	attrObject = gAttr.create( longName, shortName ); 
	MStatus status = gAttr.addAccept(MFnNumericData::kFloat); 
	CHECK_MSTATUS( status );

	status = gAttr.addAccept(MFnNumericData::k2Float); 
	CHECK_MSTATUS( status );

	status = gAttr.addAccept(MFnNumericData::k3Float); 
	CHECK_MSTATUS( status );

	gAttr.setWritable(false);
	gAttr.setStorable( false );
	
	// Add the attribute to the node
	status = addAttribute( attrObject );
	CHECK_MSTATUS( status );

	return MS::kSuccess;
}

// Create and add attributes
MStatus genericAttributeNode::initialize()
{
	MFnNumericAttribute nAttr;
	MStatus status;

	// single float attribute affecting a generic attribute
	gInputInt = nAttr.create( "gInputInt", "gii",
            	   MFnNumericData::kInt, 0, &status );
	nAttr.setStorable(true);
	nAttr.setKeyable(true);
	CHECK_MSTATUS( status );

	status = addAttribute( gInputInt );
	CHECK_MSTATUS( status );

	if ( !addComplexFloatGenericAttribute( gOutputFloat_2Float_3Float,
					 "gOutputFloat_2Float_3Float", "gof2f3f" ) )
		return MS::kFailure;

	status = attributeAffects( gInputInt, gOutputFloat_2Float_3Float );
	CHECK_MSTATUS( status );
	
	return MS::kSuccess;
}

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

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

	return status;
}

MStatus uninitializePlugin( MObject obj)
{
	MStatus   status;
	MFnPlugin plugin( obj );

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

	return status;
}