#include <stdio.h>
#include <OpenGLViewportRenderer.h>
#include <maya/MGlobal.h>
#include <maya/MString.h>
#include <maya/MRenderingInfo.h>
#include <maya/MFnCamera.h>
#include <maya/MAngle.h>
#include <maya/MPoint.h>
#include <maya/MVector.h>
#include <maya/MItDag.h>
#include <maya/MMatrix.h>
#include <maya/MDagPath.h>
#include <maya/MFnDagNode.h>
#include <maya/MFnMesh.h>
#include <maya/MBoundingBox.h>
#include <maya/MHardwareRenderer.h>
#include <stdio.h>
#include <maya/MIOStream.h>
#include <maya/MDrawTraversal.h>
#include <maya/MGeometryManager.h>
#include <maya/MGeometry.h>
#include <maya/MGeometryData.h>
#include <maya/MGeometryPrimitive.h>
#include <maya/MHwTextureManager.h>
#include <maya/MImageFileInfo.h>
#include <maya/MFnSingleIndexedComponent.h>
#if MAYA_API_VERSION > 800
#define _USE_MGL_FT_
#include <maya/MGLFunctionTable.h>
static MGLFunctionTable *gGLFT = NULL;
#else
#if defined(OSMac_MachO_)
#include <OpenGL/gl.h>
#else
#include <GL/gl.h>
#endif
#endif
OpenGLViewportRenderer::OpenGLViewportRenderer()
: MViewportRenderer("OpenGLViewportRenderer")
{
fUIName.set( "Plugin OpenGL Renderer");
fRenderingOverride = MViewportRenderer::kOverrideAllDrawing;
m_API = MViewportRenderer::kOpenGL;
m_Version = 2.0f;
}
OpenGLViewportRenderer::~OpenGLViewportRenderer()
{
uninitialize();
}
MStatus
OpenGLViewportRenderer::initialize()
{
#if defined(_USE_MGL_FT)
Tboolean useMGLFT = false;
MHardwareRenderer *rend = MHardwareRenderer::theRenderer();
if (rend)
gGLFT = rend->glFunctionTable();
if (!gGLFT)
return MStatus::kFailure;
#endif
return MStatus::kSuccess;
}
MStatus
OpenGLViewportRenderer::uninitialize()
{
gGLFT = NULL;
return MStatus::kSuccess;
}
MStatus
OpenGLViewportRenderer::render(const MRenderingInfo &renderInfo)
{
if (renderInfo.renderingAPI() == MViewportRenderer::kOpenGL)
{
renderToTarget( renderInfo );
}
return MStatus::kSuccess;
}
bool
OpenGLViewportRenderer::nativelySupports( MViewportRenderer::RenderingAPI api,
float version )
{
return (api == m_API);
}
bool
OpenGLViewportRenderer::override( MViewportRenderer::RenderingOverride override )
{
return (override == fRenderingOverride);
}
bool
OpenGLViewportRenderer::drawSurface( const MDagPath &dagPath, bool active, bool templated )
{
bool drewSurface = false;
if ( dagPath.hasFn( MFn::kMesh ))
{
MObject object = dagPath.node();
MFnMesh mesh(object);
bool haveTexture = false;
int numUVsets = mesh.numUVSets();
MString uvSetName;
MObjectArray textures;
if (numUVsets > 0)
{
mesh.getCurrentUVSetName( uvSetName );
MStatus status = mesh.getAssociatedUVSetTextures(uvSetName, textures);
if (status == MS::kSuccess && textures.length())
{
haveTexture = true;
}
}
bool haveColors = false;
int numColors = mesh.numColorSets();
MString colorSetName;
if (numColors > 0)
{
haveColors = true;
mesh.getCurrentColorSetName(colorSetName);
}
bool useNormals = false;
MGeometryRequirements requirements;
requirements.addPosition();
if (useNormals)
requirements.addNormal();
if (haveTexture)
requirements.addTexCoord( uvSetName );
if (haveColors)
requirements.addColor( colorSetName );
bool testBinormal = false;
if (testBinormal)
requirements.addBinormal( uvSetName );
bool testTangent= false;
if (testTangent)
requirements.addTangent( uvSetName );
#if defined(_TEST_OGL_RENDERER_COMPONENT_FILTER_)
MFnSingleIndexedComponent components;
MObject comp = components.create( MFn::kMeshPolygonComponent );
components.addElement( 1 );
components.addElement( 200 );
components.addElement( 5000 );
MGeometry geom = MGeometryManager::getGeometry( dagPath, requirements, &comp );
#else
MGeometry geom = MGeometryManager::getGeometry( dagPath, requirements, NULL );
#endif
unsigned int numPrims = geom.primitiveArrayCount();
if (numPrims)
{
const MGeometryPrimitive prim = geom.primitiveArray(0);
unsigned int numElem = prim.elementCount();
if (numElem)
{
unsigned int *idx = (unsigned int *) prim.data();
const MGeometryData pos = geom.position();
float * posPtr = (float * )pos.data();
bool haveData = idx && posPtr;
float * normPtr = NULL;
if (useNormals)
{
const MGeometryData norm = geom.normal();
normPtr = (float * )norm.data();
}
float *uvPtr = NULL;
if (haveTexture)
{
const MGeometryData uvs = geom.texCoord( uvSetName );
uvPtr = (float *)uvs.data();
}
unsigned int numColorComponents = 4;
float *clrPtr = NULL;
if (haveColors)
{
const MGeometryData clrs = geom.color( colorSetName );
clrPtr = (float *)clrs.data();
}
else if (testBinormal)
{
const MGeometryData binorm = geom.binormal( uvSetName );
clrPtr = (float *)binorm.data();
numColorComponents = 3;
}
else if (testTangent)
{
const MGeometryData tang = geom.tangent( uvSetName );
clrPtr = (float *)tang.data();
numColorComponents = 3;
}
if (haveData)
{
drewSurface = true;
bool drawWire = false;
MMatrix matrix = dagPath.inclusiveMatrix();
#if defined(_USE_MGL_FT_)
gGLFT->glMatrixMode( MGL_MODELVIEW );
gGLFT->glPushMatrix();
gGLFT->glMultMatrixd( &(matrix.matrix[0][0]) );
#else
::glMatrixMode( GL_MODELVIEW );
::glPushMatrix();
::glMultMatrixd( &(matrix.matrix[0][0]) );
#endif
bool boundTexture = false;
if (uvPtr)
{
MImageFileInfo::MHwTextureType hwType;
if (MS::kSuccess == MHwTextureManager::glBind( textures[0], hwType ))
{
boundTexture = true;
#if defined(_USE_MGL_FT_)
gGLFT->glTexParameteri(MGL_TEXTURE_2D, MGL_TEXTURE_WRAP_S, MGL_REPEAT);
gGLFT->glTexParameteri(MGL_TEXTURE_2D, MGL_TEXTURE_WRAP_T, MGL_REPEAT);
gGLFT->glTexParameteri(MGL_TEXTURE_2D, MGL_TEXTURE_MIN_FILTER, MGL_LINEAR);
gGLFT->glTexParameteri(MGL_TEXTURE_2D, MGL_TEXTURE_MAG_FILTER, MGL_LINEAR);
if (!clrPtr)
gGLFT->glColor4f( 1.0f, 1.0f, 1.0f, 1.0f );
gGLFT->glEnableClientState( MGL_TEXTURE_COORD_ARRAY );
gGLFT->glTexCoordPointer( 2, MGL_FLOAT, 0, uvPtr );
#else
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
if (!clrPtr)
glColor4f( 1.0f, 1.0f, 1.0f, 1.0f );
glEnableClientState( GL_TEXTURE_COORD_ARRAY );
glTexCoordPointer( 2, GL_FLOAT, 0, uvPtr );
#endif
drawWire = false;
}
}
if (clrPtr)
{
#if defined(_USE_MGL_FT_)
gGLFT->glEnableClientState(MGL_COLOR_ARRAY);
gGLFT->glColorPointer( numColorComponents, MGL_FLOAT, 0, clrPtr );
#else
glEnableClientState(GL_COLOR_ARRAY);
glColorPointer( numColorComponents, GL_FLOAT, 0, clrPtr );
#endif
}
#if defined(_USE_MGL_FT_)
gGLFT->glEnableClientState( MGL_VERTEX_ARRAY );
gGLFT->glVertexPointer ( 3, MGL_FLOAT, 0, posPtr );
#else
glEnableClientState( GL_VERTEX_ARRAY );
glVertexPointer ( 3, GL_FLOAT, 0, posPtr );
#endif
if (normPtr)
{
#if defined(_USE_MGL_FT_)
gGLFT->glEnableClientState( MGL_NORMAL_ARRAY );
gGLFT->glNormalPointer ( MGL_FLOAT, 0, normPtr );
#else
glEnableClientState( GL_NORMAL_ARRAY );
glNormalPointer ( GL_FLOAT, 0, normPtr );
#endif
drawWire = false;
}
if (drawWire)
{
#if defined(_USE_MGL_FT_)
gGLFT->glPolygonMode(MGL_FRONT_AND_BACK, MGL_LINE);
gGLFT->glDrawElements ( MGL_TRIANGLES, numElem, MGL_UNSIGNED_INT, idx );
gGLFT->glPolygonMode(MGL_FRONT_AND_BACK, MGL_FILL);
if (clrPtr)
{
gGLFT->glDisableClientState(MGL_COLOR_ARRAY);
}
#else
glPolygonMode(GL_FRONT_AND_BACK, GL_LINE);
glDrawElements ( GL_TRIANGLES, numElem, GL_UNSIGNED_INT, idx );
glPolygonMode(GL_FRONT_AND_BACK, GL_FILL);
if (clrPtr)
{
glDisableClientState(GL_COLOR_ARRAY);
}
#endif
}
else
{
#if defined(_USE_MGL_FT_)
if (active)
{
gGLFT->glEnable(MGL_POLYGON_OFFSET_FILL);
gGLFT->glPolygonOffset( 0.95f, 1.0f );
}
gGLFT->glDrawElements ( MGL_TRIANGLES, numElem, MGL_UNSIGNED_INT, idx );
if (active)
gGLFT->glDisable(MGL_POLYGON_OFFSET_FILL);
if (normPtr)
gGLFT->glDisableClientState( MGL_NORMAL_ARRAY );
if (boundTexture)
{
gGLFT->glDisableClientState( MGL_TEXTURE_COORD_ARRAY );
gGLFT->glDisable(MGL_TEXTURE_2D);
gGLFT->glBindTexture(MGL_TEXTURE_2D, 0);
}
if (clrPtr)
{
gGLFT->glDisableClientState(MGL_COLOR_ARRAY);
}
if (active)
{
gGLFT->glColor3f( 1.0f, 1.0f, 1.0f );
gGLFT->glPolygonMode(MGL_FRONT_AND_BACK, MGL_LINE);
gGLFT->glDrawElements ( MGL_TRIANGLES, numElem, MGL_UNSIGNED_INT, idx );
gGLFT->glPolygonMode(MGL_FRONT_AND_BACK, MGL_FILL);
}
#else
if (active)
{
glEnable(GL_POLYGON_OFFSET_FILL);
glPolygonOffset( 0.95f, 1.0f );
}
glDrawElements ( GL_TRIANGLES, numElem, GL_UNSIGNED_INT, idx );
if (active)
glDisable(GL_POLYGON_OFFSET_FILL);
if (normPtr)
glDisableClientState( GL_NORMAL_ARRAY );
if (boundTexture)
{
glDisableClientState( GL_TEXTURE_COORD_ARRAY );
glDisable(GL_TEXTURE_2D);
glBindTexture(GL_TEXTURE_2D, 0);
}
if (clrPtr)
{
glDisableClientState(GL_COLOR_ARRAY);
}
if (active)
{
::glColor3f( 1.0f, 1.0f, 1.0f );
glPolygonMode(GL_FRONT_AND_BACK, GL_LINE);
glDrawElements ( GL_TRIANGLES, numElem, GL_UNSIGNED_INT, idx );
glPolygonMode(GL_FRONT_AND_BACK, GL_FILL);
}
#endif
}
#if defined(_USE_MGL_FT_)
gGLFT->glDisableClientState( MGL_VERTEX_ARRAY );
gGLFT->glPopMatrix();
#else
glDisableClientState( GL_VERTEX_ARRAY );
::glPopMatrix();
#endif
}
}
}
}
return drewSurface;
}
bool
OpenGLViewportRenderer::drawBounds( const MDagPath &dagPath,
const MBoundingBox &box)
{
MMatrix matrix = dagPath.inclusiveMatrix();
{
MPoint minPt = box.min();
MPoint maxPt = box.max();
double bottomLeftFront[3] = { minPt.x, minPt.y, minPt.z };
double topLeftFront[3] = { minPt.x, maxPt.y, minPt.z };
double bottomRightFront[3] = { maxPt.x, minPt.y, minPt.z };
double topRightFront[3] = { maxPt.x, maxPt.y, minPt.z };
double bottomLeftBack[3] = { minPt.x, minPt.y, maxPt.z };
double topLeftBack[3] = { minPt.x, maxPt.y, maxPt.z };
double bottomRightBack[3] = { maxPt.x, minPt.y, maxPt.z };
double topRightBack[3] = { maxPt.x, maxPt.y, maxPt.z };
#if defined(_USE_MGL_FT_)
gGLFT->glMatrixMode( MGL_MODELVIEW );
gGLFT->glPushMatrix();
gGLFT->glMultMatrixd( &(matrix.matrix[0][0]) );
gGLFT->glBegin(MGL_LINE_STRIP);
gGLFT->glVertex3dv( bottomLeftFront );
gGLFT->glVertex3dv( bottomLeftBack );
gGLFT->glVertex3dv( topLeftBack );
gGLFT->glVertex3dv( topLeftFront );
gGLFT->glVertex3dv( bottomLeftFront );
gGLFT->glVertex3dv( bottomRightFront );
gGLFT->glVertex3dv( bottomRightBack);
gGLFT->glVertex3dv( topRightBack );
gGLFT->glVertex3dv( topRightFront );
gGLFT->glVertex3dv( bottomRightFront );
gGLFT->glEnd();
gGLFT->glBegin(MGL_LINES);
gGLFT->glVertex3dv(bottomLeftBack);
gGLFT->glVertex3dv(bottomRightBack);
gGLFT->glVertex3dv(topLeftBack);
gGLFT->glVertex3dv(topRightBack);
gGLFT->glVertex3dv(topLeftFront);
gGLFT->glVertex3dv(topRightFront);
gGLFT->glEnd();
gGLFT->glPopMatrix();
#else
::glMatrixMode( GL_MODELVIEW );
::glPushMatrix();
::glMultMatrixd( &(matrix.matrix[0][0]) );
::glBegin(GL_LINE_STRIP);
::glVertex3dv( bottomLeftFront );
::glVertex3dv( bottomLeftBack );
::glVertex3dv( topLeftBack );
::glVertex3dv( topLeftFront );
::glVertex3dv( bottomLeftFront );
::glVertex3dv( bottomRightFront );
::glVertex3dv( bottomRightBack);
::glVertex3dv( topRightBack );
::glVertex3dv( topRightFront );
::glVertex3dv( bottomRightFront );
::glEnd();
::glBegin(GL_LINES);
::glVertex3dv(bottomLeftBack);
::glVertex3dv(bottomRightBack);
::glVertex3dv(topLeftBack);
::glVertex3dv(topRightBack);
::glVertex3dv(topLeftFront);
::glVertex3dv(topRightFront);
::glEnd();
::glPopMatrix();
#endif
}
return true;
}
bool
OpenGLViewportRenderer::setupLighting()
{
return false;
}
class MsurfaceDrawTraversal : public MDrawTraversal
{
virtual bool filterNode( const MDagPath &traversalItem )
{
bool prune = false;
#if defined(_DEBUG_TRAVERSAL_PRUNING)
MString pname = traversalItem.fullPathName() ;
#endif
if ( traversalItem.childCount() == 0)
{
if ( !traversalItem.hasFn( MFn::kMesh) &&
!traversalItem.hasFn( MFn::kNurbsSurface) &&
!traversalItem.hasFn( MFn::kSubdiv) &&
!traversalItem.hasFn( MFn::kSketchPlane ) &&
!traversalItem.hasFn( MFn::kGroundPlane )
)
{
#if defined(_DEBUG_TRAVERSAL_PRUNING)
printf("Prune path [ %s ]\n", pname.asChar());
#endif
prune = true;
}
}
else
{
#if defined(_DEBUG_TRAVERSAL_PRUNING)
printf("Don't prune path [ %s ]\n", pname.asChar());
#endif
}
return prune;
}
};
bool OpenGLViewportRenderer::renderToTarget( const MRenderingInfo &renderInfo )
{
const MRenderTarget &renderTarget = renderInfo.renderTarget();
#if defined(_USE_MGL_FT_)
gGLFT->glDisable(MGL_LIGHTING);
gGLFT->glBegin(MGL_LINES);
gGLFT->glColor3f( 1.0f, 0.0f, 0.0f );
gGLFT->glVertex3f( 0.0f, 0.0f, 0.0f );
gGLFT->glVertex3f( 3.0f, 0.0f, 0.0f );
gGLFT->glColor3f( 0.0f, 1.0f, 0.0f );
gGLFT->glVertex3f( 0.0f, 0.0f, 0.0f );
gGLFT->glVertex3f( 0.0f, 3.0f, 0.0f );
gGLFT->glColor3f( 0.0f, 0.0f, 1.0f );
gGLFT->glVertex3f( 0.0f, 0.0f, 0.0f );
gGLFT->glVertex3f( 0.0f, 0.0f, 3.0f );
gGLFT->glEnd();
gGLFT->glEnable(MGL_DEPTH_TEST);
#else
::glDisable(GL_LIGHTING);
::glBegin(GL_LINES);
::glColor3f( 1.0f, 0.0f, 0.0f );
::glVertex3f( 0.0f, 0.0f, 0.0f );
::glVertex3f( 3.0f, 0.0f, 0.0f );
::glColor3f( 0.0f, 1.0f, 0.0f );
::glVertex3f( 0.0f, 0.0f, 0.0f );
::glVertex3f( 0.0f, 3.0f, 0.0f );
::glColor3f( 0.0f, 0.0f, 1.0f );
::glVertex3f( 0.0f, 0.0f, 0.0f );
::glVertex3f( 0.0f, 0.0f, 3.0f );
::glEnd();
glEnable(GL_DEPTH_TEST);
#endif
bool useDrawTraversal = true;
if (useDrawTraversal)
{
#if defined(_DEBUG_TRAVERSAL_PRUNING)
printf("==========================\n");
#endif
const MDagPath &cameraPath = renderInfo.cameraPath();
if (cameraPath.isValid())
{
bool pruneDuringTraversal = true;
MDrawTraversal *trav = NULL;
if (pruneDuringTraversal)
{
trav = new MsurfaceDrawTraversal;
trav->enableFiltering( true );
}
else
{
trav = new MDrawTraversal;
trav->enableFiltering( false );
}
if (!trav)
{
printf("Warning: failed to create a traversal class !\n");
return true;
}
trav->setFrustum( cameraPath, renderTarget.width(),
renderTarget.height() );
if (!trav->frustumValid())
printf("Warning : Frustum is invalid !\n");
trav->traverse();
unsigned int numItems = trav->numberOfItems();
#if defined(_DEBUG_TRAVERSAL_PRUNING)
printf("There are %d items on the traversal list\n", numItems );
#endif
if (numItems)
{
setupLighting();
}
unsigned int i;
for (i=0; i<numItems; i++)
{
MDagPath path;
trav->itemPath(i, path);
if (path.isValid())
{
bool drawIt = false;
if (!pruneDuringTraversal)
if ( path.hasFn( MFn::kViewManip ))
continue;
#if defined(_DEBUG_TRAVERSAL_PRUNING)
MString pname = path.fullPathName() ;
printf("Draw path [%d][ %s ]\n", i, pname.asChar());
#endif
bool active = false;
bool templated = false;
if ( path.hasFn( MFn::kMesh) ||
path.hasFn( MFn::kNurbsSurface) ||
path.hasFn( MFn::kSubdiv) )
{
drawIt = true;
if (trav->itemHasStatus( i, MDrawTraversal::kActiveItem ))
{
active = true;
#if defined(_USE_MGL_FT_)
gGLFT->glColor3f( 1.0f, 1.0f, 1.0f );
#else
::glColor3f( 1.0f, 1.0f, 1.0f );
#endif
}
else if (trav->itemHasStatus( i, MDrawTraversal::kTemplateItem ))
{
#if defined(_USE_MGL_FT_)
gGLFT->glColor3f( 0.2f, 0.2f, 0.2f );
#else
::glColor3f( 0.2f, 0.2f, 0.2f );
#endif
templated = true;
}
else
{
#if defined(_USE_MGL_FT_)
if (path.hasFn( MFn::kMesh ))
gGLFT->glColor3f( 0.286f, 0.706f, 1.0f );
else if (path.hasFn( MFn::kNurbsSurface))
gGLFT->glColor3f( 0.486f, 0.306f, 1.0f );
else
gGLFT->glColor3f( 0.886f, 0.206f, 1.0f );
#else
if (path.hasFn( MFn::kMesh ))
::glColor3f( 0.286f, 0.706f, 1.0f );
else if (path.hasFn( MFn::kNurbsSurface))
::glColor3f( 0.486f, 0.306f, 1.0f );
else
::glColor3f( 0.886f, 0.206f, 1.0f );
#endif
}
}
else if (path.hasFn( MFn::kSketchPlane ) ||
path.hasFn( MFn::kGroundPlane ))
{
drawIt = true;
#if defined(_USE_MGL_FT_)
gGLFT->glColor3f( 0.0f, 1.0f, 0.0f );
#else
::glColor3f( 0.0f, 1.0f, 0.0f );
#endif
}
if (drawIt)
{
MFnDagNode dagNode(path);
if (!drawSurface( path, active, templated ))
{
MBoundingBox box = dagNode.boundingBox();
drawBounds( path, box );
}
}
}
}
if (trav)
delete trav;
}
}
else
{
MItDag::TraversalType traversalType = MItDag::kDepthFirst;
MFn::Type filter = MFn::kMesh;
MStatus status;
MItDag dagIterator( traversalType, filter, &status);
for ( ; !dagIterator.isDone(); dagIterator.next() )
{
MDagPath dagPath;
status = dagIterator.getPath(dagPath);
if ( !status ) {
status.perror("MItDag::getPath");
continue;
}
MFnDagNode dagNode(dagPath, &status);
if ( !status ) {
status.perror("MFnDagNode constructor");
continue;
}
MBoundingBox box = dagNode.boundingBox();
#if defined(_USE_MGL_FT_)
gGLFT->glColor3f( 1.0f, 0.0f, 0.0f );
#else
::glColor3f( 1.0f, 0.0f, 0.0f );
#endif
drawBounds( dagPath, box );
}
}
return true;
}