//-
// ==========================================================================
// 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 <maya/MPxNode.h>

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

#include <maya/MFnNumericAttribute.h>

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

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

class transCircle : public MPxNode
{
public:
						transCircle() {};
	virtual				~transCircle() {}; 

	virtual MStatus		compute( const MPlug& plug, MDataBlock& data );
	static  void*		creator();
	static  MStatus		initialize();

	static  MObject		input;            // The input value.
	static  MObject		inputTranslateX;  // The translate X value.  (input)
	static  MObject		inputTranslateY;  // The translate Y value.  (input)
	static  MObject		inputTranslateZ;  // The translate Z value.  (input)
	static  MObject		inputTranslate;
	static  MObject		outputTranslateX; // The translate X value.  (output)
	static  MObject		outputTranslateY; // The translate Y value.  (output)
	static  MObject		outputTranslateZ; // The translate Z value.  (output)
	static  MObject		outputTranslate;
	static  MObject		frames;           // Number of frames for one circle.
	static  MObject		scale;            // Size of circle.
	static  MTypeId		id;
};

MTypeId	transCircle::id( 0x80013 );
MObject	transCircle::input;        
MObject	transCircle::frames;
MObject	transCircle::scale;
MObject	transCircle::inputTranslate;
MObject	transCircle::outputTranslate;
MObject	transCircle::inputTranslateX;
MObject	transCircle::inputTranslateY;
MObject	transCircle::inputTranslateZ;
MObject	transCircle::outputTranslateX;
MObject	transCircle::outputTranslateY;
MObject	transCircle::outputTranslateZ;

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

MStatus transCircle::initialize()
{
	MFnNumericAttribute nAttr;
	MStatus				stat;

	input = nAttr.create( "input", "in", MFnNumericData::kDouble, 0.0 );
	nAttr.setStorable(true);

	inputTranslateX = nAttr.create( "inputTranslateX", "itX", 
									MFnNumericData::kDouble, 0.0 );
	nAttr.setStorable(true);

	inputTranslateY = nAttr.create( "inputTranslateY", "itY", 
									MFnNumericData::kDouble, 0.0 );
	nAttr.setStorable(true);

	inputTranslateZ = nAttr.create( "inputTranslateZ", "itZ", 
									MFnNumericData::kDouble, 0.0 );
	nAttr.setStorable(true);

	inputTranslate = nAttr.create( "inputTranslate", "it", 
									inputTranslateX,
									inputTranslateY,
									inputTranslateZ );
	nAttr.setStorable(true);
	nAttr.setDefault(0.0,0.0,0.0);

	outputTranslateX = nAttr.create( "outputTranslateX", "otX",
									MFnNumericData::kDouble, 0.0 );
	nAttr.setWritable(false);
	nAttr.setStorable(true);

	outputTranslateY = nAttr.create( "outputTranslateY", "otY",
									MFnNumericData::kDouble, 0.0 );
	nAttr.setWritable(false);
	nAttr.setStorable(true);

	outputTranslateZ = nAttr.create( "outputTranslateZ", "otZ",
									MFnNumericData::kDouble, 0.0 );
	nAttr.setWritable(false);
	nAttr.setStorable(true);

	outputTranslate = nAttr.create( "outputTranslate", "ot", 
									outputTranslateX,
									outputTranslateY,
									outputTranslateZ );
	nAttr.setWritable(false);
	nAttr.setStorable(true);
	nAttr.setDefault(0.0,0.0,0.0);

	scale = nAttr.create( "scale", "sc",
									MFnNumericData::kDouble, 10.0 );
	nAttr.setStorable(true);

	frames = nAttr.create( "frames", "fr",
									MFnNumericData::kDouble, 48.0 );
	nAttr.setStorable(true);

	stat = addAttribute( inputTranslate );
		if (!stat) { stat.perror("addAttribute"); return stat;}
	stat = addAttribute( input );
		if (!stat) { stat.perror("addAttribute"); return stat;}
	stat = addAttribute( scale );
		if (!stat) { stat.perror("addAttribute"); return stat;}
	stat = addAttribute( frames );
		if (!stat) { stat.perror("addAttribute"); return stat;}
	stat = addAttribute( outputTranslate );
		if (!stat) { stat.perror("addAttribute"); return stat;}

	stat = attributeAffects( inputTranslateX, outputTranslateX );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslateY, outputTranslateY );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslateZ, outputTranslateZ );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslate, outputTranslateX );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslate, outputTranslateY );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslate, outputTranslateZ );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( inputTranslate, outputTranslate );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( input, outputTranslateX );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( input, outputTranslateY );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( scale, outputTranslateX );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( scale, outputTranslateY );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( frames, outputTranslateX );
		if (!stat) { stat.perror("attributeAffects"); return stat;}
	stat = attributeAffects( frames, outputTranslateY );
		if (!stat) { stat.perror("attributeAffects"); return stat;}

	return MS::kSuccess;
}

MStatus transCircle::compute( const MPlug& plug, MDataBlock& data )
{
    MStatus stat;
 
	bool k = ( plug == outputTranslateX ) |
	         ( plug == outputTranslateY ) |
			 ( plug == outputTranslateZ ) |
			 ( plug == outputTranslate );
	if (!k) return MS::kUnknownParameter;

	MDataHandle inputData = data.inputValue( input, &stat );
	if (!stat) { stat.perror("inputValue: input"); return stat;}

	MDataHandle scaleData = data.inputValue( scale, &stat );
	if (!stat) { stat.perror("inputValue: scale"); return stat;}

	MDataHandle framesData = data.inputValue( frames, &stat );
	if (!stat) { stat.perror("inputValue: frames"); return stat;}

	MDataHandle transData  = data.inputValue( inputTranslate, &stat ); 
	if (!stat) { stat.perror("inputValue: inputTranslate"); return stat;}

	double3& iTranslate = transData.asDouble3();
	double currentFrame = inputData.asDouble();
	double scaleFactor  = scaleData.asDouble();
	double framesPerCircle = framesData.asDouble();
	double angle = 6.2831853 * ( currentFrame/framesPerCircle );
	double3 oTranslate;
	
	oTranslate[0] = iTranslate[0] + (sin( angle ) * scaleFactor);
	oTranslate[1] = iTranslate[1] + 1.0;
	oTranslate[2] = iTranslate[2] + (cos( angle ) * scaleFactor);

	MDataHandle otHandle = data.outputValue( outputTranslate ); 
	otHandle.set( oTranslate[0], oTranslate[1], oTranslate[2] );
	data.setClean(plug);

    return MS::kSuccess;
}

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

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

    return status;
}

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

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

    return status;
}