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

#include <maya/MGlobal.h>
#include <maya/MPlugArray.h>
#include <maya/MFnDagNode.h>
#include <maya/MObjectArray.h>

#include "hwPhongShaderBehavior.h"

// Set VERBOSE to 0 to suppress diagnostic outputs. Set it to 1 to
// display when the override methods are called.

#define VERBOSE 0

//
//	Description:
//		Constructor
//
hwPhongShaderBehavior::hwPhongShaderBehavior()
{
}

//
//	Description:
//		Destructor
//
hwPhongShaderBehavior::~hwPhongShaderBehavior()
{
}

//
//	Description:
//		Returns a new instance of this class
//
void *hwPhongShaderBehavior::creator()
{
	return new hwPhongShaderBehavior;
}

//
//	Description:
//		Overloaded function from MPxDragAndDropBehavior
//	this method will return true if it is going to handle the connection
//	between the two nodes given.
//
///////////////////////////////////////////////////////////////////////////////
bool hwPhongShaderBehavior::shouldBeUsedFor(
	MObject & sourceNode, 
	MObject & destinationNode,
	MPlug   & /* sourcePlug */,
	MPlug   & /* destinationPlug */)
{
	bool result = false;

	// Handle dropping a hw shader on a maya shader
	//
	if ((MFnDependencyNode(sourceNode).typeName() == "hwPhongShader") &&
		destinationNode.hasFn(MFn::kLambert))
		result = true;

#if VERBOSE
	cerr << "shouldBeUsedFor "<<MFnDependencyNode(sourceNode).name().asChar() 
		 << " " << MFnDependencyNode(destinationNode).name().asChar()  
		 << (result?" true\n":" false\n");
#endif

	return result;
}

//
// Connect source plug to destination.
//
///////////////////////////////////////////////////////////////////////////////
static MStatus connectAttr(
	const MPlug &srcPlug, 
	const MPlug &destPlug, 
	bool		force)
{
	MStatus result = MS::kFailure;

	if(!srcPlug.isNull() && !destPlug.isNull())
	{
		MString cmd = "connectAttr ";
		if (force) cmd += "-f ";
		cmd += srcPlug.name() + " " + destPlug.name();
		result = MGlobal::executeCommand(cmd);

#if VERBOSE
		cerr << cmd.asChar()<<"\n";
#endif
	}
	return result;
}

//
//	Description:
//		Overloaded function from MPxDragAndDropBehavior
//	this method will handle the connection between the hwPhongShader
//	and the shader it is assigned to as well as any meshes that it is
//	assigned to.
//
///////////////////////////////////////////////////////////////////////////////
MStatus hwPhongShaderBehavior::connectNodeToNode( 
	MObject & sourceNode, 
	MObject & destinationNode, 
	bool      force )
{
	MStatus result = MS::kFailure;
	MFnDependencyNode src(sourceNode);

	if ((src.typeName() == "hwPhongShader") &&
		destinationNode.hasFn(MFn::kLambert))
	{
		MFnDependencyNode dest(destinationNode);
		result = connectAttr(src.findPlug("outColor"), 
							 dest.findPlug("hardwareShader"), force);
	}

#if VERBOSE
	if (result != MS::kSuccess)
		cerr << "connectNodeToNode "<<src.name().asChar() 
			 << " " << MFnDependencyNode(destinationNode).name().asChar()  
			 << " failed\n";
#endif

	return result;
}

//
//	Description:
//		Overloaded function from MPxDragAndDropBehavior
//	this method will assign the correct output from the hwPhong shader 
//	onto the given attribute.
//
///////////////////////////////////////////////////////////////////////////////
MStatus hwPhongShaderBehavior::connectNodeToAttr( 
	MObject & sourceNode,
	MPlug   & destinationPlug,
	bool      force )

{
	MStatus result = MS::kFailure;
	MFnDependencyNode src(sourceNode);

	//if we are dragging from a hwPhongShader
	//to a shader than connect the outColor
	//plug to the plug being passed in
	//	
	if ((src.typeName() == "hwPhongShader") &&
		destinationPlug.node().hasFn(MFn::kLambert))
	{
		result = connectAttr(src.findPlug("outColor"), destinationPlug, force);
	}

#if VERBOSE
	if (result != MS::kSuccess)
		cerr << "connectNodeToAttr "<<src.name().asChar() 
			 << " " << destinationPlug.name().asChar()  
			 << " failed\n";
#endif

	return result;
}

//
//
///////////////////////////////////////////////////////////////////////////////
MStatus hwPhongShaderBehavior::connectAttrToNode( 
	MPlug   & sourcePlug, 
	MObject & destinationNode,
	bool      force )
{
	MStatus result = MS::kFailure;
	MObject sourceNode = sourcePlug.node();
	MFnDependencyNode src(sourceNode);

	if ((src.typeName() == "hwPhongShader") &&
		destinationNode.hasFn(MFn::kLambert))
	{
		MFnDependencyNode dest(destinationNode);
		result = connectAttr(sourcePlug, dest.findPlug("hardwareShader"),
							 force);
	}

#if VERBOSE
	if (result != MS::kSuccess)
		cerr << "connectNodeAttrToNode "<<sourcePlug.name().asChar() 
			 << " " << MFnDependencyNode(destinationNode).name().asChar()  
			 << " failed\n";
#endif

	return result;
}

//
//
///////////////////////////////////////////////////////////////////////////////
MStatus hwPhongShaderBehavior::connectAttrToAttr( 
	MPlug & sourcePlug, 
	MPlug & destinationPlug, 
	bool    force )
{
#if VERBOSE
	cerr << "In connectAttrToAttr "<<sourcePlug.name().asChar() 
		 << " " << destinationPlug.name().asChar()
		 << "\n";
#endif

	return connectAttr(sourcePlug, destinationPlug, force);
}