First MayaPlugin, still need to fix the makefiles and get unique ids for
the Maya nodes. Got rid of boost dependencies. Cleaned up mvl library
This commit is contained in:
nicola.candussi
326
Extras/MayaPlugin/collisionShapeNode.cpp
Normal file
326
Extras/MayaPlugin/collisionShapeNode.cpp
Normal file
@@ -0,0 +1,326 @@
|
||||
/*
|
||||
Bullet Continuous Collision Detection and Physics Library Maya Plugin
|
||||
Copyright (c) 2008 Walt Disney Studios
|
||||
|
||||
This software is provided 'as-is', without any express or implied warranty.
|
||||
In no event will the authors be held liable for any damages arising
|
||||
from the use of this software.
|
||||
Permission is granted to anyone to use this software for any purpose,
|
||||
including commercial applications, and to alter it and redistribute it freely,
|
||||
subject to the following restrictions:
|
||||
|
||||
1. The origin of this software must not be misrepresented; you must
|
||||
not claim that you wrote the original software. If you use this
|
||||
software in a product, an acknowledgment in the product documentation
|
||||
would be appreciated but is not required.
|
||||
2. Altered source versions must be plainly marked as such, and must
|
||||
not be misrepresented as being the original software.
|
||||
3. This notice may not be removed or altered from any source distribution.
|
||||
|
||||
Written by: Nicola Candussi <nicola@fluidinteractive.com>
|
||||
*/
|
||||
|
||||
//collisionShapeNode.cpp
|
||||
|
||||
#include <GL/glut.h>
|
||||
|
||||
#include <maya/MFnMessageAttribute.h>
|
||||
#include <maya/MFnNumericAttribute.h>
|
||||
#include <maya/MFnEnumAttribute.h>
|
||||
#include <maya/MPlugArray.h>
|
||||
#include <maya/MFloatPointArray.h>
|
||||
#include <maya/MFloatVectorArray.h>
|
||||
#include <maya/MIntArray.h>
|
||||
#include <maya/MFnMesh.h>
|
||||
#include <maya/MDagPath.h>
|
||||
|
||||
#include <iterator>
|
||||
#include <vector>
|
||||
#include <assert.h>
|
||||
|
||||
#include "collisionShapeNode.h"
|
||||
#include "mayaUtils.h"
|
||||
#include "solver.h"
|
||||
|
||||
MTypeId collisionShapeNode::typeId( 0x100307 );
|
||||
MString collisionShapeNode::typeName( "dCollisionShape" );
|
||||
|
||||
MObject collisionShapeNode::ia_shape;
|
||||
MObject collisionShapeNode::ia_type;
|
||||
MObject collisionShapeNode::ia_scale;
|
||||
MObject collisionShapeNode::ca_collisionShape;
|
||||
MObject collisionShapeNode::ca_collisionShapeParam;
|
||||
MObject collisionShapeNode::oa_collisionShape;
|
||||
|
||||
MStatus collisionShapeNode::initialize()
|
||||
{
|
||||
MStatus status;
|
||||
MFnMessageAttribute fnMsgAttr;
|
||||
MFnNumericAttribute fnNumericAttr;
|
||||
MFnEnumAttribute fnEnumAttr;
|
||||
|
||||
ia_type = fnEnumAttr.create("type", "tp", 5, &status);
|
||||
MCHECKSTATUS(status, "creating type attribute")
|
||||
fnEnumAttr.addField("Convex Hull", 0);
|
||||
fnEnumAttr.addField("Mesh", 1);
|
||||
fnEnumAttr.addField("Cylinder", 2);
|
||||
fnEnumAttr.addField("Capsule", 3);
|
||||
fnEnumAttr.addField("Box", 4);
|
||||
fnEnumAttr.addField("Sphere", 5);
|
||||
fnEnumAttr.addField("Plane", 6);
|
||||
fnEnumAttr.setKeyable(true);
|
||||
status = addAttribute(ia_type);
|
||||
MCHECKSTATUS(status, "adding type attribute")
|
||||
|
||||
ia_scale = fnNumericAttr.createPoint("scale", "sc", &status);
|
||||
MCHECKSTATUS(status, "creating ia_scale attribute")
|
||||
fnNumericAttr.setDefault(1.0, 1.0, 1.0);
|
||||
fnNumericAttr.setKeyable(true);
|
||||
status = addAttribute(ia_scale);
|
||||
MCHECKSTATUS(status, "adding ia_scale attribute")
|
||||
|
||||
oa_collisionShape = fnMsgAttr.create("outCollisionShape", "oucs", &status);
|
||||
MCHECKSTATUS(status, "creating outCollisionShape attribute")
|
||||
status = addAttribute(oa_collisionShape);
|
||||
MCHECKSTATUS(status, "adding outCollisionShape attribute")
|
||||
|
||||
ia_shape = fnMsgAttr.create("inShape", "insh", &status);
|
||||
MCHECKSTATUS(status, "creating inShape attribute")
|
||||
status = addAttribute(ia_shape);
|
||||
MCHECKSTATUS(status, "adding inShape attribute")
|
||||
|
||||
ca_collisionShape = fnNumericAttr.create("ca_collisionShape", "ccs", MFnNumericData::kBoolean, 0, &status);
|
||||
MCHECKSTATUS(status, "creating ca_collisionShape attribute")
|
||||
fnNumericAttr.setConnectable(false);
|
||||
fnNumericAttr.setHidden(true);
|
||||
fnNumericAttr.setStorable(false);
|
||||
fnNumericAttr.setKeyable(false);
|
||||
status = addAttribute(ca_collisionShape);
|
||||
MCHECKSTATUS(status, "adding ca_collisionShape attribute")
|
||||
|
||||
ca_collisionShapeParam = fnNumericAttr.create("collisionShapeParam", "cspm", MFnNumericData::kBoolean, 0, &status);
|
||||
MCHECKSTATUS(status, "creating ca_collisionShapeParam attribute")
|
||||
fnNumericAttr.setConnectable(false);
|
||||
fnNumericAttr.setHidden(true);
|
||||
fnNumericAttr.setStorable(false);
|
||||
fnNumericAttr.setKeyable(false);
|
||||
status = addAttribute(ca_collisionShapeParam);
|
||||
MCHECKSTATUS(status, "adding ca_collisionShapeParam attribute")
|
||||
|
||||
//
|
||||
status = attributeAffects(ia_shape, oa_collisionShape);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_shape, oa_collisionShape)")
|
||||
|
||||
status = attributeAffects(ia_type, oa_collisionShape);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_type, oa_collisionShape)")
|
||||
|
||||
status = attributeAffects(ia_scale, oa_collisionShape);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_scale, oa_collisionShape)")
|
||||
|
||||
//
|
||||
status = attributeAffects(ia_shape, ca_collisionShape);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_shape, ca_collisionShape)")
|
||||
|
||||
status = attributeAffects(ia_type, ca_collisionShape);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_shape, ca_collisionShape)")
|
||||
|
||||
//
|
||||
status = attributeAffects(ia_shape, ca_collisionShapeParam);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_shape, oa_collisionShapeParam)")
|
||||
|
||||
status = attributeAffects(ia_scale, ca_collisionShapeParam);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_scale, oa_collisionShapeParam)")
|
||||
|
||||
status = attributeAffects(ia_type, ca_collisionShapeParam);
|
||||
MCHECKSTATUS(status, "adding attributeAffects(ia_type, oa_collisionShapeParam)")
|
||||
|
||||
return MS::kSuccess;
|
||||
}
|
||||
|
||||
collisionShapeNode::collisionShapeNode()
|
||||
{
|
||||
// std::cout << "collisionShapeNode::collisionShapeNode" << std::endl;
|
||||
}
|
||||
|
||||
collisionShapeNode::~collisionShapeNode()
|
||||
{
|
||||
// std::cout << "collisionShapeNode::~collisionShapeNode" << std::endl;
|
||||
}
|
||||
|
||||
|
||||
void* collisionShapeNode::creator()
|
||||
{
|
||||
return new collisionShapeNode();
|
||||
}
|
||||
|
||||
|
||||
bool collisionShapeNode::setInternalValueInContext ( const MPlug & plug,
|
||||
const MDataHandle & dataHandle,
|
||||
MDGContext & ctx)
|
||||
{
|
||||
return false; //setInternalValueInContext(plug,dataHandle,ctx);
|
||||
}
|
||||
|
||||
MStatus collisionShapeNode::compute( const MPlug& plug, MDataBlock& data )
|
||||
{
|
||||
if(plug == oa_collisionShape) {
|
||||
computeOutputShape(plug, data);
|
||||
} else if(plug == ca_collisionShape) {
|
||||
computeCollisionShape(plug, data);
|
||||
} else if(plug == ca_collisionShapeParam) {
|
||||
computeCollisionShapeParam(plug, data);
|
||||
} else {
|
||||
return MStatus::kUnknownParameter;
|
||||
}
|
||||
return MStatus::kSuccess;
|
||||
}
|
||||
|
||||
collision_shape_t::pointer collisionShapeNode::collisionShape()
|
||||
{
|
||||
MObject thisObject(thisMObject());
|
||||
MObject update;
|
||||
MPlug(thisObject, oa_collisionShape).getValue(update);
|
||||
MPlug(thisObject, ca_collisionShapeParam).getValue(update);
|
||||
return m_collision_shape;
|
||||
}
|
||||
|
||||
void collisionShapeNode::computeCollisionShape(const MPlug& plug, MDataBlock& data)
|
||||
{
|
||||
// std::cout << "collisionShapeNode::computeCollisionShape" << std::endl;
|
||||
|
||||
MObject thisObject(thisMObject());
|
||||
|
||||
MPlug plgInShape(thisObject, ia_shape);
|
||||
|
||||
MPlug plgType(thisObject, ia_type);
|
||||
int type;
|
||||
plgType.getValue(type);
|
||||
switch(type) {
|
||||
case 0:
|
||||
{
|
||||
//convex hull
|
||||
MPlugArray plgaConnectedTo;
|
||||
plgInShape.connectedTo(plgaConnectedTo, true, true);
|
||||
if(plgaConnectedTo.length() > 0) {
|
||||
MObject node = plgaConnectedTo[0].node();
|
||||
if(node.hasFn(MFn::kMesh)) {
|
||||
MDagPath dagPath;
|
||||
MDagPath::getAPathTo(node, dagPath);
|
||||
MFnMesh fnMesh(dagPath);
|
||||
MFloatPointArray mpoints;
|
||||
MFloatVectorArray mnormals;
|
||||
MIntArray mtrianglecounts;
|
||||
MIntArray mtrianglevertices;
|
||||
fnMesh.getPoints(mpoints, MSpace::kWorld);
|
||||
fnMesh.getNormals(mnormals, MSpace::kWorld);
|
||||
fnMesh.getTriangles(mtrianglecounts, mtrianglevertices);
|
||||
|
||||
std::vector<vec3f> vertices(mpoints.length());
|
||||
std::vector<vec3f> normals(mpoints.length());
|
||||
std::vector<unsigned int> indices(mtrianglevertices.length());
|
||||
|
||||
for(size_t i = 0; i < vertices.size(); ++i) {
|
||||
vertices[i] = vec3f(mpoints[i].x, mpoints[i].y, mpoints[i].z);
|
||||
normals[i] = vec3f(mnormals[i].x, mnormals[i].y, mnormals[i].z);
|
||||
}
|
||||
for(size_t i = 0; i < indices.size(); ++i) {
|
||||
indices[i] = mtrianglevertices[i];
|
||||
}
|
||||
m_collision_shape = solver_t::create_convex_hull_shape(&(vertices[0]), vertices.size(), &(normals[0]),
|
||||
&(indices[0]), indices.size());
|
||||
}
|
||||
}
|
||||
}
|
||||
break;
|
||||
case 1:
|
||||
//mesh
|
||||
{
|
||||
//convex hull
|
||||
MPlugArray plgaConnectedTo;
|
||||
plgInShape.connectedTo(plgaConnectedTo, true, true);
|
||||
if(plgaConnectedTo.length() > 0) {
|
||||
MObject node = plgaConnectedTo[0].node();
|
||||
if(node.hasFn(MFn::kMesh)) {
|
||||
MDagPath dagPath;
|
||||
MDagPath::getAPathTo(node, dagPath);
|
||||
MFnMesh fnMesh(dagPath);
|
||||
MFloatPointArray mpoints;
|
||||
MFloatVectorArray mnormals;
|
||||
MIntArray mtrianglecounts;
|
||||
MIntArray mtrianglevertices;
|
||||
fnMesh.getPoints(mpoints, MSpace::kWorld);
|
||||
fnMesh.getNormals(mnormals, MSpace::kWorld);
|
||||
fnMesh.getTriangles(mtrianglecounts, mtrianglevertices);
|
||||
|
||||
std::vector<vec3f> vertices(mpoints.length());
|
||||
std::vector<vec3f> normals(mpoints.length());
|
||||
std::vector<unsigned int> indices(mtrianglevertices.length());
|
||||
|
||||
for(size_t i = 0; i < vertices.size(); ++i) {
|
||||
vertices[i] = vec3f(mpoints[i].x, mpoints[i].y, mpoints[i].z);
|
||||
normals[i] = vec3f(mnormals[i].x, mnormals[i].y, mnormals[i].z);
|
||||
}
|
||||
for(size_t i = 0; i < indices.size(); ++i) {
|
||||
indices[i] = mtrianglevertices[i];
|
||||
}
|
||||
m_collision_shape = solver_t::create_mesh_shape(&(vertices[0]), vertices.size(), &(normals[0]),
|
||||
&(indices[0]), indices.size());
|
||||
}
|
||||
}
|
||||
}
|
||||
break;
|
||||
case 2:
|
||||
//cylinder
|
||||
break;
|
||||
case 3:
|
||||
//capsule
|
||||
break;
|
||||
case 4:
|
||||
//box
|
||||
m_collision_shape = solver_t::create_box_shape();
|
||||
break;
|
||||
case 5:
|
||||
//sphere
|
||||
m_collision_shape = solver_t::create_sphere_shape();
|
||||
break;
|
||||
case 6:
|
||||
//plane
|
||||
m_collision_shape = solver_t::create_plane_shape();
|
||||
break;
|
||||
}
|
||||
|
||||
assert(m_collision_shape);
|
||||
|
||||
data.setClean(plug);
|
||||
}
|
||||
|
||||
|
||||
void collisionShapeNode::computeCollisionShapeParam(const MPlug& plug, MDataBlock& data)
|
||||
{
|
||||
// std::cout << "collisionShapeNode::computeCollisionShapeParam" << std::endl;
|
||||
|
||||
MObject thisObject(thisMObject());
|
||||
|
||||
MObject update;
|
||||
MPlug(thisObject, ia_shape).getValue(update);
|
||||
MPlug(thisObject, ia_type).getValue(update);
|
||||
|
||||
float3& scale = data.inputValue(ia_scale).asFloat3();
|
||||
|
||||
m_collision_shape->set_scale(vec3f(scale[0], scale[1], scale[2]));
|
||||
|
||||
data.setClean(plug);
|
||||
}
|
||||
|
||||
void collisionShapeNode::computeOutputShape(const MPlug& plug, MDataBlock& data)
|
||||
{
|
||||
// std::cout << "collisionShapeNode::computeOutputShape" << std::endl;
|
||||
|
||||
MObject thisObject(thisMObject());
|
||||
|
||||
MObject update;
|
||||
MPlug(thisObject, ca_collisionShape).getValue(update);
|
||||
MPlug(thisObject, ca_collisionShapeParam).getValue(update);
|
||||
|
||||
data.setClean(plug);
|
||||
}
|
||||
Reference in New Issue
Block a user