MStatus initializePlugin( MObject obj )

{

	MStatus status;

	const MString& swatchName =	MHWShaderSwatchGenerator::initialize();

	const MString UserClassify( "shader/surface/utility/:swatch/"+swatchName );

	MFnPlugin plugin( obj, "Autodesk", "4.5", "Any");

	status = plugin.registerNode( "hwPhongShader", hwPhongShader::id,

				hwPhongShader::creator, hwPhongShader::initialize,

				MPxNode::kHwShaderNode, &UserClassify );

	if (!status) {

		status.perror("registerNode");

		return status;

	}

	plugin.registerDragAndDropBehavior("hwPhongShaderBehavior", 

						hwPhongShaderBehavior::creator);

	return MS::kSuccess;

}

MStatus uninitializePlugin( MObject obj )

{

	MStatus status;

	MFnPlugin plugin( obj );

	// Unregister all chamelion shader nodes

	plugin.deregisterNode( hwPhongShader::id );

	if (!status) {

		status.perror("deregisterNode");

		return status;

	}

	plugin.deregisterDragAndDropBehavior("hwPhongShaderBehavior");

	return MS::kSuccess;

}

MStatus hwPhongShader::initialize()

{

 MFnNumericAttribute nAttr; 

 // Create input attributes

 aColor = nAttr.createColor( "color", "c");

 nAttr.setStorable(true);

 nAttr.setKeyable(true);

 nAttr.setDefault(0.1f, 0.1f, 0.1f);

 nAttr.setCached( true );

 nAttr.setInternal( true );

 aDiffuseColor = nAttr.createColor( "diffuseColor", "dc" );

 nAttr.setStorable(true);

 nAttr.setKeyable(true);

 nAttr.setDefault(1.f, 0.5f, 0.5f);

 nAttr.setCached( true );

 nAttr.setInternal( true );

 aSpecularColor = nAttr.createColor( "specularColor", "sc" );

 nAttr.setStorable(true);

 nAttr.setKeyable(true);

 nAttr.setDefault(0.5f, 0.5f, 0.5f);

 nAttr.setCached( true );

 nAttr.setInternal( true );

 // This is defined as a point, so that users can easily enter

 // values beyond 1.

 aShininess = nAttr.createPoint( "shininess", "sh" );

 nAttr.setStorable(true);

 nAttr.setKeyable(true);

 nAttr.setDefault(100.0f, 100.0f, 100.0f);

 nAttr.setCached( true );

 nAttr.setInternal( true );

 aGeometryShape = nAttr.create( "geometryShape", "gs", MFnNumericData::kInt );

 nAttr.setStorable(true);

 nAttr.setKeyable(true);

 nAttr.setDefault(0);

 nAttr.setCached( true );

 nAttr.setInternal( true );

 // create output attributes here

 // outColor is the only output attribute and it is inherited

 // so we do not need to create or add it.

 //

 // Add the attributes here

 addAttribute(aColor);

 addAttribute(aDiffuseColor);

 addAttribute(aSpecularColor);

 addAttribute(aShininess);

 addAttribute(aGeometryShape);

 attributeAffects (aColor,			outColor);

 attributeAffects (aDiffuseColor,	outColor);

 attributeAffects (aSpecularColor,	outColor);

 attributeAffects (aShininess,		outColor);

 return MS::kSuccess;

}

MStatus hwPhongShader::compute(

const MPlug& plug,

 MDataBlock& block ) 

{ 

 if ((plug != outColor) && (plug.parent() != outColor))

		return MS::kUnknownParameter;

 MFloatVector & color = block.inputValue( aDiffuseColor ).asFloatVector();

 // set output color attribute

 MDataHandle outColorHandle = block.outputValue( outColor );

 MFloatVector& outColor = outColorHandle.asFloatVector();

 outColor = color;

 outColorHandle.setClean();

 return MS::kSuccess;

}

MStatus	hwPhongShader::bind(const MDrawRequest& request, M3dView& view)

{

	if (mAttributesChanged || (phong_map_id == 0))

	{

		init_Phong_texture ();

	}

	return MS::kSuccess;

}

MStatus	hwPhongShader::glBind(const MDagPath&)

{

	if ( mAttributesChanged || (phong_map_id == 0))

	{

		init_Phong_texture ();

	}

	return MS::kSuccess;

}

MStatus	hwPhongShader::unbind(const MDrawRequest& request, M3dView& view)

{

 // The texture may have been allocated by the draw; it's kept

 // around for use again. When scene new or open is performed this

 // texture will be released in releaseEverything().

 return MS::kSuccess;

}

MStatus	hwPhongShader::glUnbind(const MDagPath&)

{

 // The texture may have been allocated by the draw; it's kept

 // around for use again. When scene new or open is performed this

 // texture will be released in releaseEverything().

 return MS::kSuccess;

}

MStatus	hwPhongShader::geometry( const MDrawRequest& request,

				M3dView& view,

 int prim,

 unsigned int writable,

 int indexCount,

 const unsigned int * indexArray,

 int vertexCount,

 const int * vertexIDs,

 const float * vertexArray,

 int normalCount,

 const float ** normalArrays,

 int colorCount,

 const float ** colorArrays,

 int texCoordCount,

 const float ** texCoordArrays)

{

	MStatus stat = MStatus::kSuccess;

	if (mGeometryShape != 0)

		drawDefaultGeometry();

	else

		stat = draw( prim, writable, indexCount, indexArray, vertexCount,

				vertexIDs, vertexArray, normalCount, normalArrays,

				colorCount, colorArrays, 

				texCoordCount, texCoordArrays);

	return stat;

}

MStatus	hwPhongShader::glGeometry(const MDagPath & path,

 int prim,

 unsigned int writable,

 int indexCount,

 const unsigned int * indexArray,

 int vertexCount,

 const int * vertexIDs,

 const float * vertexArray,

 int normalCount,

 const float ** normalArrays,

 int colorCount,

 const float ** colorArrays,

 int texCoordCount,

 const float ** texCoordArrays)

{

	MStatus stat = MStatus::kSuccess;

	if (mGeometryShape != 0)

		drawDefaultGeometry();

	else

		stat = draw( prim, writable, indexCount, indexArray, vertexCount,

				vertexIDs, vertexArray, normalCount, normalArrays,

				colorCount, colorArrays, 

				texCoordCount, texCoordArrays);

	return stat;

}

MStatus	hwPhongShader::draw(int prim,

			unsigned int writable,

			int indexCount,

			const unsigned int * indexArray,

			int vertexCount,

			const int * vertexIDs,

			const float * vertexArray,

			int normalCount,

			const float ** normalArrays,

			int colorCount,

			const float ** colorArrays,

			int texCoordCount,

			const float ** texCoordArrays)

{

	if ( prim != GL_TRIANGLES && prim != GL_TRIANGLE_STRIP)	{

 return MS::kFailure;

 }

 {

		glPushAttrib ( GL_ENABLE_BIT );

		glDisable ( GL_LIGHTING );

		glDisable ( GL_TEXTURE_1D );

		glDisable ( GL_TEXTURE_2D );

		// Setup cube map generation

		glEnable ( GL_TEXTURE_CUBE_MAP_EXT );

		glBindTexture ( GL_TEXTURE_CUBE_MAP_EXT, phong_map_id );

		glEnable ( GL_TEXTURE_GEN_S );

		glEnable ( GL_TEXTURE_GEN_T );

		glEnable ( GL_TEXTURE_GEN_R );

		glTexGeni ( GL_S, GL_TEXTURE_GEN_MODE, GL_NORMAL_MAP_EXT );

		glTexGeni ( GL_T, GL_TEXTURE_GEN_MODE, GL_NORMAL_MAP_EXT );

		glTexGeni ( GL_R, GL_TEXTURE_GEN_MODE, GL_NORMAL_MAP_EXT );

		glTexParameteri(GL_TEXTURE_CUBE_MAP_EXT, GL_TEXTURE_WRAP_S, GL_CLAMP);

		glTexParameteri(GL_TEXTURE_CUBE_MAP_EXT, GL_TEXTURE_WRAP_T, GL_CLAMP);

		glTexParameteri(GL_TEXTURE_CUBE_MAP_EXT, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

		glTexParameteri(GL_TEXTURE_CUBE_MAP_EXT, GL_TEXTURE_MIN_FILTER, GL_LINEAR);

		glTexEnvi ( GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE );

		// Could modify the texture matrix here to do light tracking...

		glMatrixMode ( GL_TEXTURE );

		glPushMatrix ();

		glLoadIdentity ();

		glMatrixMode ( GL_MODELVIEW );

 }

 // Draw the surface.

 //

 {

		glPushClientAttrib ( GL_CLIENT_VERTEX_ARRAY_BIT );

		glEnableClientState( GL_VERTEX_ARRAY );

		glEnableClientState( GL_NORMAL_ARRAY );

		glVertexPointer ( 3, GL_FLOAT, 0, &vertexArray[0] );

		glNormalPointer ( GL_FLOAT, 0, &normalArrays[0][0] );

		glDrawElements ( prim, indexCount, GL_UNSIGNED_INT, indexArray );

		// The client attribute is already being popped. You

		glPopClientAttrib();

 }

 {

		glMatrixMode ( GL_TEXTURE );

		glPopMatrix ();

		glMatrixMode ( GL_MODELVIEW );

		glDisable ( GL_TEXTURE_CUBE_MAP_EXT );

		glDisable ( GL_TEXTURE_GEN_S );

		glDisable ( GL_TEXTURE_GEN_T );

		glDisable ( GL_TEXTURE_GEN_R );

		glPopAttrib();

 }

	return MS::kSuccess;

}