void arrowLocatorOverride::draw(
const MHWRender::MDrawContext& context,
const MUserData* data)
{
MAngle rotationAngle;
float color [3] ={ 0.0f, 1.0f, 0.0f } ;
// data
MStatus status;
MHWRender::MStateManager* stateMgr = context.getStateManager();
const arrowLocatorData* locatorData =
dynamic_cast<const arrowLocatorData*>(data);
if (!stateMgr || !locatorData) return;
if ( locatorData )
rotationAngle = locatorData->rotateAngle;
// matrices
const MMatrix transform =
context.getMatrix(MHWRender::MFrameContext::kWorldViewMtx, &status);
if (status != MStatus::kSuccess) return;
const MMatrix projection =
context.getMatrix(MHWRender::MFrameContext::kProjectionMtx, &status);
if (status != MStatus::kSuccess) return;
// get renderer
MHWRender::MRenderer *theRenderer =MHWRender::MRenderer::theRenderer () ;
if ( !theRenderer )
return ;
if ( theRenderer->drawAPIIsOpenGL () ) {
glPushAttrib(GL_CURRENT_BIT);
glColor4fv(color);
glPushMatrix();
glRotated(-rotationAngle.asDegrees(), 0.0, 1.0, 0.0);
glBegin( GL_LINE_STRIP);
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glVertex3f(arrow[1][0],arrow[1][1],arrow[1][2]);
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glEnd();
glBegin( GL_LINE_STRIP );
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glVertex3f(arrow[3][0],arrow[3][1],arrow[3][2]);
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glEnd();
glPopMatrix();
}
}
MMatrix ropeGenerator::getMatrixFromParamCurve( MFnNurbsCurve &curveFn, float param, float twist, MAngle divTwist )
{
MPoint pDivPos;
//Here we control the tangent of the rope
curveFn.getPointAtParam( param, pDivPos, MSpace::kWorld );
MVector vTangent( curveFn.tangent( param, MSpace::kWorld ).normal() );
MVector vNormal( curveFn.normal( param, MSpace::kWorld ).normal() );
if ( MAngle( PrevNormal.angle( vNormal ) ).asDegrees() > 90 )
//fprintf(stderr, "Angle = %g\n",MAngle( PrevNormal.angle( vNormal )).asDegrees());
vNormal = vNormal * -1;
PrevNormal = vNormal;
//if ( vNormal.angle( ) )
MQuaternion qTwist( twist * divTwist.asRadians(), vTangent );
vNormal = vNormal.rotateBy( qTwist );
MVector vExtra( vNormal ^ vTangent );
vNormal.normalize();
vTangent.normalize();
vExtra.normalize();
double dTrans[4][4] ={
{vNormal.x, vNormal.y, vNormal.z, 0.0f},
{vTangent.x, vTangent.y, vTangent.z, 0.0f},
{vExtra.x, vExtra.y, vExtra.z, 0.0f},
{pDivPos.x,pDivPos.y,pDivPos.z, 1.0f}};
MMatrix mTrans( dTrans );
return mTrans;
}
//- This method draw this locator in current scene by calling openGL functions
// view -- 3D view that is being drawn into
// path -- path to this locator in the DAG
// style -- style to draw object in
// status -- selection status of object
//
void arrowLocator::draw(M3dView &view,const MDagPath & path,M3dView::DisplayStyle style, M3dView::DisplayStatus status)
{
MObject thisNode = thisMObject();
//- We're in the draw routine, not the compute method
//- therefore it is safe to grab plugs in the following way and
//- get/set values. You would never do something like this in
//- the compute method because it might start a cycle in the
//- graph. Here we just need the value of our winddirection
//- plug so that we can draw our arrow pointing the right way.
MPlug wdPlug(thisNode, windDirection);
MAngle angle;
wdPlug.getValue(angle);
//- Start drawing by OpenGL
view.beginGL();
//- If the drawing style is shaded, set color and draw opaque shape
if ( ( style == M3dView::kFlatShaded ) || ( style == M3dView::kGouraudShaded ) )
{
//- Push to save current color settings
glPushAttrib( GL_CURRENT_BIT );
if ( status == M3dView::kActive )
{
view.setDrawColor( 13, M3dView::kActiveColors );
}
else
{
view.setDrawColor( 13, M3dView::kDormantColors );
}
//- Push the old matrix on the stack, rotate the current one,
//- draw the shape, then pop the old matrix off the stack for
//- maya to use again.
glPushMatrix();
glRotated(-angle.asDegrees(), 0.0, 1.0, 0.0);
glBegin( GL_TRIANGLE_FAN );
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glVertex3f(arrow[1][0],arrow[1][1],arrow[1][2]);
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glEnd();
glBegin( GL_TRIANGLE_FAN );
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glVertex3f(arrow[3][0],arrow[3][1],arrow[3][2]);
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glEnd();
glPopMatrix();
glPopAttrib();
}
//- Draw the outline of the arrow shape
glPushMatrix();
glRotated(-angle.asDegrees(), 0.0, 1.0, 0.0);
glBegin( GL_LINE_STRIP);
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glVertex3f(arrow[1][0],arrow[1][1],arrow[1][2]);
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glEnd();
glBegin( GL_LINE_STRIP );
glVertex3f(arrow[2][0],arrow[2][1],arrow[2][2]);
glVertex3f(arrow[3][0],arrow[3][1],arrow[3][2]);
glVertex3f(arrow[0][0],arrow[0][1],arrow[0][2]);
glEnd();
glPopMatrix();
view.endGL();
}
float convert( const MAngle &from )
{
return from.asRadians();
}
bool animWriter::writeAnimCurve(ofstream &clip,
const MObject *animCurveObj,
bool verboseUnits /* false */)
//
// Description:
// Write out the anim curve from the clipboard item into the
// ofstream. The actual anim curve data is written out.
//
// This method returns true if the write was successful.
//
{
if (NULL == animCurveObj || animCurveObj->isNull() || !clip) {
return true;
}
MStatus status = MS::kSuccess;
MFnAnimCurve animCurve(*animCurveObj, &status);
if (MS::kSuccess != status) {
cerr << "Error: Could not read the anim curve for export." << endl;
return false;
}
clip << kAnimData << kSpaceChar << kBraceLeftChar << endl;
clip << kTwoSpace << kInputString << kSpaceChar <<
boolInputTypeAsWord(animCurve.isUnitlessInput()) <<
kSemiColonChar << endl;
clip << kTwoSpace << kOutputString << kSpaceChar <<
outputTypeAsWord(animCurve.animCurveType()) << kSemiColonChar << endl;
clip << kTwoSpace << kWeightedString << kSpaceChar <<
(animCurve.isWeighted() ? 1 : 0) << kSemiColonChar << endl;
// These units default to the units in the header of the file.
//
if (verboseUnits) {
clip << kTwoSpace << kInputUnitString << kSpaceChar;
if (animCurve.isTimeInput()) {
MString unitName;
animUnitNames::setToShortName(timeUnit, unitName);
clip << unitName;
} else {
// The anim curve has unitless input.
//
clip << kUnitlessString;
}
clip << kSemiColonChar << endl;
clip << kTwoSpace << kOutputUnitString << kSpaceChar;
}
double conversion = 1.0;
MString unitName;
switch (animCurve.animCurveType()) {
case MFnAnimCurve::kAnimCurveTA:
case MFnAnimCurve::kAnimCurveUA:
animUnitNames::setToShortName(angularUnit, unitName);
if (verboseUnits) clip << unitName;
{
MAngle angle(1.0);
conversion = angle.as(angularUnit);
}
break;
case MFnAnimCurve::kAnimCurveTL:
case MFnAnimCurve::kAnimCurveUL:
animUnitNames::setToShortName(linearUnit, unitName);
if (verboseUnits) clip << unitName;
{
MDistance distance(1.0);
conversion = distance.as(linearUnit);
}
break;
case MFnAnimCurve::kAnimCurveTT:
case MFnAnimCurve::kAnimCurveUT:
animUnitNames::setToShortName(timeUnit, unitName);
if (verboseUnits) clip << unitName;
break;
default:
if (verboseUnits) clip << kUnitlessString;
break;
}
if (verboseUnits) clip << kSemiColonChar << endl;
if (verboseUnits) {
MString angleUnitName;
animUnitNames::setToShortName(angularUnit, angleUnitName);
clip << kTwoSpace << kTanAngleUnitString <<
kSpaceChar << angleUnitName << kSemiColonChar << endl;
}
clip << kTwoSpace << kPreInfinityString << kSpaceChar <<
infinityTypeAsWord(animCurve.preInfinityType()) <<
kSemiColonChar << endl;
clip << kTwoSpace << kPostInfinityString << kSpaceChar <<
infinityTypeAsWord(animCurve.postInfinityType()) <<
kSemiColonChar << endl;
clip << kTwoSpace << kKeysString << kSpaceChar << kBraceLeftChar << endl;
// And then write out each keyframe
//
unsigned numKeys = animCurve.numKeyframes();
for (unsigned i = 0; i < numKeys; i++) {
clip << kTwoSpace << kTwoSpace;
if (animCurve.isUnitlessInput()) {
clip << animCurve.unitlessInput(i);
}
else {
clip << animCurve.time(i).value();
}
clip << kSpaceChar << (conversion*animCurve.value(i));
clip << kSpaceChar << tangentTypeAsWord(animCurve.inTangentType(i));
clip << kSpaceChar << tangentTypeAsWord(animCurve.outTangentType(i));
clip << kSpaceChar << (animCurve.tangentsLocked(i) ? 1 : 0);
clip << kSpaceChar << (animCurve.weightsLocked(i) ? 1 : 0);
clip << kSpaceChar << (animCurve.isBreakdown(i) ? 1 : 0);
if (animCurve.inTangentType(i) == MFnAnimCurve::kTangentFixed) {
MAngle angle;
double weight;
animCurve.getTangent(i, angle, weight, true);
clip << kSpaceChar << angle.as(angularUnit);
clip << kSpaceChar << weight;
}
if (animCurve.outTangentType(i) == MFnAnimCurve::kTangentFixed) {
MAngle angle;
double weight;
animCurve.getTangent(i, angle, weight, false);
clip << kSpaceChar << angle.as(angularUnit);
clip << kSpaceChar << weight;
}
clip << kSemiColonChar << endl;
}
clip << kTwoSpace << kBraceRightChar << endl;
clip << kBraceRightChar << endl;
return true;
}
/********************************************************************************************************
* Method to translate a single camera *
********************************************************************************************************/
MStatus OgreExporter::writeCamera(MFnCamera& camera)
{
MPlug plug;
MPlugArray srcplugarray;
double dist;
MAngle angle;
MFnTransform* cameraTransform = NULL;
MFnAnimCurve* animCurve = NULL;
// get camera transform
for (int i=0; i<camera.parentCount(); i++)
{
if (camera.parent(i).hasFn(MFn::kTransform))
{
cameraTransform = new MFnTransform(camera.parent(i));
continue;
}
}
// start camera description
m_params.outCameras << "camera " << cameraTransform->partialPathName().asChar() << "\n";
m_params.outCameras << "{\n";
//write camera type
m_params.outCameras << "\ttype ";
if (camera.isOrtho())
m_params.outCameras << "ortho\n";
else
m_params.outCameras << "persp\n";
// write translation data
m_params.outCameras << "\ttranslation\n";
m_params.outCameras << "\t{\n";
//translateX
m_params.outCameras << "\t\tx ";
plug = cameraTransform->findPlug("translateX");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i < srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to translateX attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(dist);
m_params.outCameras << "= " << dist << "\n";
}
//translateY
m_params.outCameras << "\t\ty ";
plug = cameraTransform->findPlug("translateY");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i< srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to translateY attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(dist);
m_params.outCameras << "= " << dist << "\n";
}
//translateZ
m_params.outCameras << "\t\tz ";
plug = cameraTransform->findPlug("translateZ");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i< srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to translateZ attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(dist);
m_params.outCameras << "= " << dist << "\n";
}
m_params.outCameras << "\t}\n";
// write rotation data
m_params.outCameras << "\trotation\n";
m_params.outCameras << "\t{\n";
m_params.outCameras << "\t\tx ";
//rotateX
plug = cameraTransform->findPlug("rotateX");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i< srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to rotateX attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(angle);
m_params.outCameras << "= " << angle.asDegrees() << "\n";
}
//rotateY
m_params.outCameras << "\t\ty ";
plug = cameraTransform->findPlug("rotateY");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i< srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to rotateY attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(angle);
m_params.outCameras << "= " << angle.asDegrees() << "\n";
}
//rotateZ
m_params.outCameras << "\t\tz ";
plug = cameraTransform->findPlug("rotateZ");
if (plug.isConnected() && m_params.exportCamerasAnim)
{
plug.connectedTo(srcplugarray,true,false,&stat);
for (int i=0; i< srcplugarray.length(); i++)
{
if (srcplugarray[i].node().hasFn(MFn::kAnimCurve))
{
if (animCurve)
delete animCurve;
animCurve = new MFnAnimCurve(srcplugarray[i].node());
continue;
}
else if (i == srcplugarray.length()-1)
{
std::cout << "Invalid link to rotateZ attribute\n";
return MS::kFailure;
}
}
m_params.outCameras << "anim " << animCurve->name().asChar() << "\n";
}
else
{
plug.getValue(angle);
m_params.outCameras << "= " << angle.asDegrees() << "\n";
}
m_params.outCameras << "\t}\n";
// end camera description
m_params.outCameras << "}\n\n";
if (cameraTransform != NULL)
delete cameraTransform;
if (animCurve != NULL)
delete animCurve;
return MS::kSuccess;
}
