//-
// ==========================================================================
// Copyright (C) 1995 - 2006 Autodesk, Inc. and/or its licensors.  All 
// rights reserved.
//
// The coded instructions, statements, computer programs, and/or related 
// material (collectively the "Data") in these files contain unpublished 
// information proprietary to Autodesk, Inc. ("Autodesk") and/or its 
// licensors, which is protected by U.S. and Canadian federal copyright 
// law and by international treaties.
//
// The Data is provided for use exclusively by You. You have the right 
// to use, modify, and incorporate this Data into other products for 
// purposes authorized by the Autodesk software license agreement, 
// without fee.
//
// The copyright notices in the Software and this entire statement, 
// including the above license grant, this restriction and the 
// following disclaimer, must be included in all copies of the 
// Software, in whole or in part, and all derivative works of 
// the Software, unless such copies or derivative works are solely 
// in the form of machine-executable object code generated by a 
// source language processor.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND. 
// AUTODESK DOES NOT MAKE AND HEREBY DISCLAIMS ANY EXPRESS OR IMPLIED 
// WARRANTIES INCLUDING, BUT NOT LIMITED TO, THE WARRANTIES OF 
// NON-INFRINGEMENT, MERCHANTABILITY OR FITNESS FOR A PARTICULAR 
// PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE, OR 
// TRADE PRACTICE. IN NO EVENT WILL AUTODESK AND/OR ITS LICENSORS 
// BE LIABLE FOR ANY LOST REVENUES, DATA, OR PROFITS, OR SPECIAL, 
// DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES, EVEN IF AUTODESK 
// AND/OR ITS LICENSORS HAS BEEN ADVISED OF THE POSSIBILITY 
// OR PROBABILITY OF SUCH DAMAGES.
//
// ==========================================================================
//+

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

#include <maya/MPxCameraSet.h>
#include <maya/MFnCompoundAttribute.h>
#include <maya/MFnEnumAttribute.h>
#include <maya/MFnPlugin.h>
#include <maya/MString.h> 
#include <maya/MTypeId.h> 
#include <maya/MPlug.h>
#include "exCameraSet.h"
#include "testExCameraSetCmd.h"

MTypeId     exCameraSet::id( 0x80024  );
MObject     exCameraSet::layerTypeAttr;        

exCameraSet::exCameraSet() {}
exCameraSet::~exCameraSet() {}

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

MStatus exCameraSet::initialize()
{
	MFnEnumAttribute   enumAttr;
	MStatus status;

	// Create the layerType attribute and define its enum values.
	layerTypeAttr = enumAttr.create( "layerType", "lt", 0, &status );

	if( !status ) {
		status.perror("ERROR creating layerType enum attribute");
		return status;
	}

	enumAttr.addField("Mono",  0);
	enumAttr.addField("Left",  1);
	enumAttr.addField("Right", 2);

	// Make this attribute a multi so it can store a value per camera layer.
	enumAttr.setArray(true);

	return MS::kSuccess;
}

MStatus exCameraSet::setLayerType( unsigned int index, int layerType )
{
	MStatus status;

	// Get the layerType plug for the given index.
	MPlug layerTypePlug;
	status = getLayerTypePlug(index, layerTypePlug);
	if (status != MS::kSuccess)
		return status;

	// Set the value of the plug to the new value.
	return layerTypePlug.setValue(layerType);
}

MStatus exCameraSet::getLayerType( unsigned int index, int & layerType ) const
{
	MStatus status;

	// Get the layerType plug for the given index.
	MPlug layerTypePlug;
	status = getLayerTypePlug(index, layerTypePlug);
	if (status != MS::kSuccess)
		return status;

	// Get the value of the plug to the new value.
	return layerTypePlug.getValue(layerType);
}

MStatus exCameraSet::getLayerTypePlug(unsigned int index, MPlug &layerTypePlug) const
{
	MStatus status = MS::kSuccess;

	MPlug enumPlug(thisMObject(), layerTypeAttr);
	layerTypePlug = enumPlug.elementByLogicalIndex(index, &status);

	return status;
}

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

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

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

	return status;
}

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

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

	status = plugin.deregisterCommand( kCmdName );
	if (!status) {
		status.perror("deregisterCommand");
		return status;
	}

	return status;
}