publishedinterface.cpp

使用stl技术,(还没看,是听说的)

#include "PublishedInterface.h"
#include "modstack.h"

#define VIEW_DEFAULT_WIDTH ((float)400.0)

// FP interface descriptor - FnPub-published methods & properties
//****************************************************************
static IPhysiqueInterface iPhysiqueInterface (IPHYSIQUEINTERFACE, _T("physiqueOps"), 0, NULL, FP_CORE + FP_STATIC_METHODS,
	//methods
	IPhysiqueInterface::checkMatrix, _T("checkMatrix"), 0, TYPE_BOOL, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG,f_keyArgDefault, NULL,

	IPhysiqueInterface::getPhysique, _T("getPhysiqueModifier"), 0, TYPE_REFTARG, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("index"), 0, TYPE_INDEX, f_keyArgDefault, 0,
	IPhysiqueInterface::getAPIVersion, _T("getAPIVersion"), 0, TYPE_INT, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,

	IPhysiqueInterface::getBoneList, _T("getBones"), 0, TYPE_INODE_TAB_BV, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getBoneCount, _T("getBoneCount"), 0, TYPE_INT, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexCount, _T("getVertexCount"), 0, TYPE_INT, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,

	IPhysiqueInterface::getInitialNodeTM, _T("getInitialNodeTM"), 0, TYPE_MATRIX3_BV, 0, 2,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getInitialBoneTM, _T("getInitialBoneTM"), 0, TYPE_MATRIX3_BV, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("boneIndex"), 0, TYPE_INDEX,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	
	IPhysiqueInterface::getVertexType, _T("getVertexType"), 0, TYPE_ENUM, IPhysiqueInterface::blendingTypeEnum, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexBones, _T("getVertexBones"), 0, TYPE_INODE_TAB_BV, 0, 5,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("rigid"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("blending"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexBoneCount, _T("getVertexBoneCount"), 0, TYPE_INT, 0, 4,
		_T("physiquedNode"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("rigid"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexBone, _T("getVertexBone"), 0, TYPE_INODE, 0, 6,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("boneIndex"), 0, TYPE_INDEX,
		_T("rigid"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("blending"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexOffset, _T("getVertexOffset"), 0, TYPE_POINT3_BV, 0, 6,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("boneIndex"), 0, TYPE_INDEX,
		_T("rigid"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("blending"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexDeformableOffset, _T("getVertexDeformableOffset"), 0, TYPE_POINT3_BV, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::getVertexWeight, _T("getVertexWeight"), 0, TYPE_FLOAT, 0, 6,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("boneIndex"), 0, TYPE_INDEX,
		_T("rigid"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("blending"), 0, TYPE_bool, f_keyArgDefault, true,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,

	IPhysiqueInterface::setInitialPose, _T("setInitialPose"), 0, TYPE_VOID, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("state"), 0, TYPE_bool,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::setFigureMode, _T("setFigureMode"), 0, TYPE_bool, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("state"), 0, TYPE_bool,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,

	IPhysiqueInterface::attachToNode, _T("attachToNode"), 0, TYPE_bool, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("rootNode"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::initialize, _T("initialize"), 0, TYPE_bool, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("rootNode"), 0, TYPE_INODE,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	
	IPhysiqueInterface::lockVertex, _T("lockVertex"), 0, TYPE_VOID, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::unlockVertex, _T("unlockVertex"), 0, TYPE_VOID, 0, 3,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,
	IPhysiqueInterface::setVertexBone, _T("setVertexBone"), 0, TYPE_VOID, 0, 6,
		_T("physiqued_Node"), 0, TYPE_INODE,
		_T("vertexIndex"), 0, TYPE_INDEX,
		_T("bone"), 0, TYPE_INODE,
		_T("weight"), 0, TYPE_FLOAT, f_keyArgDefault, 1.0f,
		_T("clear"), 0, TYPE_bool, f_keyArgDefault, false,
		_T("modifier"), 0, TYPE_REFTARG, f_keyArgDefault, NULL,

//	properties,
//	ICoreTestInterface::iobjectInterfaceServer, FP_NO_FUNCTION, _T("interfaceServer"), 0, TYPE_IOBJECT,
	enums,
	IPhysiqueInterface::blendingTypeEnum, 8,
					_T("rigid_type"), RIGID_TYPE,
					_T("deformable_type"), DEFORMABLE_TYPE,
					_T("blended_type"), BLENDED_TYPE,
					_T("floating_type"), FLOATING_TYPE,
					_T("rigid_non_blended_type"), RIGID_NON_BLENDED_TYPE,
					_T("deformable_non_blended_type"), DEFORMABLE_NON_BLENDED_TYPE,
					_T("rigid_blended_type"), RIGID_BLENDED_TYPE,
					_T("deformable_blended_type"), DEFORMABLE_BLENDED_TYPE,
	IPhysiqueInterface::interfaceTypeEnum, 1,
					_T("floatingInterface"), FLOATING_TYPE,

	end
	); 

//SDK level methods
//****************************************************************
Modifier* IPhysiqueInterface::FindPhysiqueModifier (INode* nodePtr, int modIndex)
{
	// Get object from node. Abort if no object.
	if (!nodePtr) return NULL;

	Object* ObjectPtr = nodePtr->GetObjectRef();

	if (!ObjectPtr ) return NULL;
	
	while (ObjectPtr->SuperClassID() == GEN_DERIVOB_CLASS_ID && ObjectPtr)
	{
		// Yes -> Cast.
		IDerivedObject* DerivedObjectPtr = (IDerivedObject *)(ObjectPtr);				
			
		// Iterate over all entries of the modifier stack.
		int modStackIndex = 0, phyModIndex = 0;
		
		while (modStackIndex < DerivedObjectPtr->NumModifiers())
		{
			// Get current modifier.
			Modifier* ModifierPtr = DerivedObjectPtr->GetModifier(modStackIndex);

			// Is this Physique ?
			if (ModifierPtr->ClassID() == Class_ID(PHYSIQUE_CLASS_ID_A, PHYSIQUE_CLASS_ID_B))
			{
				if (phyModIndex >= modIndex)
				{ 
					return ModifierPtr;
				}
				phyModIndex++;
			}
			modStackIndex++;
		}
		ObjectPtr = DerivedObjectPtr->GetObjRef();
	}

	// Not found.
	throw MAXException("No physique modifier found");
	return NULL;
}

BOOL IPhysiqueInterface::CheckMatrix(INode* node, ReferenceTarget* mod)
{
	if (!mod) mod = FindPhysiqueModifier(node, 0);
	if (!mod) throw MAXException("No Physique modifier was found on this node");
	
	//get a pointer to the export interface
	IPhysiqueExport *phyExport = (IPhysiqueExport *)mod->GetInterface(I_PHYINTERFACE);


	int version = phyExport->Version();
	
	if(version == 30)
	{
	//get the node's initial transformation matrix and store it in a matrix3
		Matrix3 initTM = Matrix3(TRUE);
		int msg = phyExport->GetInitNodeTM(node, initTM);

		//Release the physique interface


		if (msg == NO_MATRIX_SAVED)
			return false;
	}

	mod->ReleaseInterface(I_PHYINTERFACE, phyExport);
	return true;


}

Matrix3 IPhysiqueInterface::GetInitialNodeTM(INode* node, ReferenceTarget* mod)
{

	if (!mod) mod = FindPhysiqueModifier(node, 0);
	if (!mod) throw MAXException("No Physique modifier was found on this node");
	
	//get a pointer to the export interface
	IPhysiqueExport *phyExport = (IPhysiqueExport *)mod->GetInterface(I_PHYINTERFACE);

	//get the node's initial transformation matrix and store it in a matrix3
	Matrix3 initTM = Matrix3(TRUE);
	int msg = phyExport->GetInitNodeTM(node, initTM);

	//Release the physique interface
	mod->ReleaseInterface(I_PHYINTERFACE, phyExport);
	if (msg != MATRIX_RETURNED)
	{
		switch (msg)
		{
			case NODE_NOT_FOUND:
				throw MAXException ("Node not found");
			case NO_MATRIX_SAVED:
				throw MAXException("No matrix saved");
			case INVALID_MOD_POINTER:
				throw MAXException("Invalid modifier pointer");
			default: 
				break;
		}
	}


	return initTM;
}

Matrix3 IPhysiqueInterface::GetInitialBoneTM(INode* node, int index, ReferenceTarget* mod)
{

	if (!mod) mod = FindPhysiqueModifier(node, 0);
	if (!mod) throw MAXException("No Physique modifier was found on this node");
	
	//get a pointer to the export interface
	IPhysiqueExport *phyExport = (IPhysiqueExport *)mod->GetInterface(I_PHYINTERFACE);

	//get the list of bones
	Tab<INode*> bones = GetBoneList(node);
	if (index <0 || index >= bones.Count()) throw MAXException("The Bone index is not in the valid range");

	//get the node's initial transformation matrix and store it in a matrix3
	Matrix3 initTM = Matrix3(TRUE);
	int msg = phyExport->GetInitNodeTM(bones[index], initTM);
	//Release the physique interface
	mod->ReleaseInterface(I_PHYINTERFACE, phyExport);
	
	if (msg != MATRIX_RETURNED)
	{
		switch (msg)
		{
			case NODE_NOT_FOUND:
				throw MAXException ("Node not found");
			case NO_MATRIX_SAVED:
				throw MAXException("No matrix saved");
			case INVALID_MOD_POINTER:
				throw MAXException("Invalid modifier pointer");
			default: 
				break;
		}
	}
	return initTM;
}

int IPhysiqueInterface::GetBoneCount(INode* node, ReferenceTarget* mod)
{
	//get the list of bones
	Tab<INode*> bones = GetBoneList(node, mod);
	return bones.Count();
}



// This function can be used to gather the list of bones used by the modifier
//****************************************************************************
Tab<INode*> IPhysiqueInterface::GetBoneList(INode* node, ReferenceTarget* mod)
{
	int i = 0, x = 0;
	Tab<INode*> bones;
	INode* bone;

	if (!mod) mod = FindPhysiqueModifier(node, 0);
	if (!mod) throw MAXException("No Physique modifier was found on this node");

	//get a pointer to the export interface
	IPhysiqueExport *phyExport = (IPhysiqueExport *)mod->GetInterface(I_PHYINTERFACE);

	//get a pointer to the export context interface
	IPhyContextExport *mcExport = (IPhyContextExport *)phyExport->GetContextInterface(node);

	//convert to rigid for time independent vertex assignment
	mcExport->ConvertToRigid(true);

	//allow blending to export multi-link assignments
	mcExport->AllowBlending(true);

	//These are the different types of vertex classes 
	IPhyBlendedRigidVertex *rb_vtx;
	IPhyRigidVertex *r_vtx;
	IPhyFloatingVertex *f_vtx;

	//get the vertex count from the export interface
	int numverts = mcExport->GetNumberVertices();
	
	//iterate through all vertices and gather the bone list
	for (i = 0; i<numverts; i++) 
	{
		BOOL exists = false;
		
		//get the hierarchial vertex interface
		IPhyVertexExport* vi = mcExport->GetVertexInterface(i);
		if (vi) {

			//check the vertex type and process accordingly
			int type = vi->GetVertexType();
			switch (type) 
			{
				//The vertex is rigid, blended vertex.  It's assigned to multiple links
				case RIGID_BLENDED_TYPE:
					//type-cast the node to the proper class		
					rb_vtx = (IPhyBlendedRigidVertex*)vi;
					
					//iterate through the bones assigned to this vertex