From bd3ef19389cf66a9fb1637c7d78293e1f27b1883 Mon Sep 17 00:00:00 2001
From: "Asad M. Zaman" <masaduzz@users.sourceforge.net>
Date: Tue, 17 Oct 2006 00:24:06 +0000
Subject: [PATCH] added a test program for normals

---
 pandatool/src/mayaprogs/Sources.pp      |   4 +-
 pandatool/src/mayaprogs/normal_test.cxx | 288 ++++++++++++++++++++++++
 2 files changed, 290 insertions(+), 2 deletions(-)
 create mode 100755 pandatool/src/mayaprogs/normal_test.cxx

diff --git a/pandatool/src/mayaprogs/Sources.pp b/pandatool/src/mayaprogs/Sources.pp
index ef488a2265..ebfc27780a 100644
--- a/pandatool/src/mayaprogs/Sources.pp
+++ b/pandatool/src/mayaprogs/Sources.pp
@@ -184,9 +184,9 @@
 
 #begin test_bin_target
   #define USE_PACKAGES maya
-  #define TARGET blend_test
+  #define TARGET normal_test
 
   #define SOURCES \
-    blend_test.cxx
+    normal_test.cxx
 
 #end test_bin_target
diff --git a/pandatool/src/mayaprogs/normal_test.cxx b/pandatool/src/mayaprogs/normal_test.cxx
new file mode 100755
index 0000000000..c6d7e2413b
--- /dev/null
+++ b/pandatool/src/mayaprogs/normal_test.cxx
@@ -0,0 +1,288 @@
+
+#include <iostream>
+#include <fstream>
+#include <stdlib.h>
+#include <string>
+
+#ifndef _BOOL
+#define _BOOL 1
+#endif
+
+#define REQUIRE_IOSTREAM
+
+#include <maya/MGlobal.h>
+#include <maya/MFileIO.h>
+#include <maya/MLibrary.h>
+#include <maya/MStatus.h>
+#include <maya/MItDag.h>
+#include <maya/MDagPath.h>
+#include <maya/MFnDagNode.h>
+#include <maya/MItMeshPolygon.h>
+#include <maya/MItSelectionList.h>
+#include <maya/MFnBlendShapeDeformer.h>
+#include <maya/MFnMesh.h>
+#include <maya/MPointArray.h>
+
+using namespace std;
+
+void
+scan_nodes() {
+  MStatus status;
+
+  MItDag dag_iterator(MItDag::kDepthFirst, MFn::kTransform, &status);
+  if (!status) {
+    status.perror("MItDag constructor");
+    exit(1);
+  }
+
+  while (!dag_iterator.isDone()) {
+    MDagPath dag_path;
+    status = dag_iterator.getPath(dag_path);
+    if (!status) {
+      status.perror("MItDag::getPath");
+      exit(1);
+    }
+
+    MFnDagNode dag_node(dag_path, &status);
+    if (!status) {
+      status.perror("MFnDagNode constructor");
+      exit(1);
+    }
+
+    cerr << dag_node.name() << "\n";
+
+    dag_iterator.next();
+  }
+}
+
+MFnBlendShapeDeformer *
+get_slider(MString slider_name) {
+  MStatus status;
+
+  status = MGlobal::selectByName(slider_name, MGlobal::kReplaceList);
+  if (!status) {
+    status.perror(slider_name);
+    exit(1);
+  }
+  MSelectionList list;
+  status = MGlobal::getActiveSelectionList(list);
+  if (!status) {
+    status.perror("MGLobal::getActiveSelectionList");
+    exit(1);
+  }
+
+  unsigned int num_objects = list.length();
+  if (num_objects != 1) {
+    cerr << "Warning: multiple objects match " << slider_name << "\n";
+  }
+
+  for (unsigned int i = 0; i < num_objects; i++) {
+    MObject obj;
+    status = list.getDependNode(i, obj);
+    if (!status) {
+      cerr << "selected element is not a dependency node.\n";
+      status.perror("getDependNode");
+    } else {
+      if (obj.hasFn(MFn::kBlendShape)) {
+        MFnBlendShapeDeformer *slider = new MFnBlendShapeDeformer(obj, &status);
+        if (!status) {
+          status.perror("MFnBlendShapeDeformer constructor");
+        } else {
+          cerr << "got slider " << slider->name() << "\n";
+          return slider;
+        }
+      }
+
+      cerr << "selected element is not a blend shape\n";
+    }
+  }
+  
+  cerr << "Couldn't find slider " << slider_name << "\n";
+  exit(1);
+}
+
+MFnMesh *
+get_mesh(MString mesh_name) {
+  MStatus status;
+
+  status = MGlobal::selectByName(mesh_name, MGlobal::kReplaceList);
+  if (!status) {
+    status.perror(mesh_name);
+    exit(1);
+  }
+  MSelectionList list;
+  status = MGlobal::getActiveSelectionList(list);
+  if (!status) {
+    status.perror("MGLobal::getActiveSelectionList");
+    exit(1);
+  }
+
+  unsigned int num_objects = list.length();
+  if (num_objects != 1) {
+    cerr << "Warning: multiple objects match " << mesh_name << "\n";
+  }
+
+  for (unsigned int i = 0; i < num_objects; i++) {
+    MObject obj;
+    status = list.getDependNode(i, obj);
+    if (!status) {
+      cerr << "selected element is not a dependency node.\n";
+      status.perror("getDependNode");
+
+    } else {
+      if (obj.hasFn(MFn::kMesh)) {
+        // Maybe it's a mesh object itself.
+        MFnMesh *mesh = new MFnMesh(obj, &status);
+        if (!status) {
+          status.perror("MFnMesh constructor");
+        } else {
+          cerr << "got mesh " << mesh->name() << "\n";
+          return mesh;
+        }
+
+      } else if (obj.hasFn(MFn::kDagNode)) {
+        // Maybe it's a dag node.
+        MDagPath path;
+        status = MDagPath::getAPathTo(obj, path);
+        if (!status) {
+          status.perror("MDagPath::getAPathTo");
+          exit(1);
+        }
+        if (path.hasFn(MFn::kMesh)) {
+          MFnMesh *mesh = new MFnMesh(path, &status);
+          if (!status) {
+            status.perror("MFnMesh constructor");
+          } else {
+            cerr << "got mesh " << mesh->name() << "\n";
+            return mesh;
+          }
+        }
+      }
+
+      cerr << "selected element is not a mesh\n";
+    }
+  }
+  
+  cerr << "Couldn't find mesh " << mesh_name << "\n";
+  exit(1);
+}
+
+void
+output_vertices(const char *filename, MFnMesh &mesh) {
+  MStatus status;
+
+  MPointArray verts;
+  //  status = mesh.getPoints(verts, MSpace::kObject);
+  status = mesh.getPoints(verts, MSpace::kWorld);
+  if (!status) {
+    status.perror("mesh.getPoints");
+    exit(1);
+  }
+
+  ofstream file(filename, ios::out | ios::trunc);
+  if (!file) {
+    cerr << "Couldn't open " << filename << " for output.\n";
+    exit(1);
+  }
+
+  unsigned int num_verts = verts.length();
+  cerr << "writing " << num_verts << " vertices to " << filename << "\n";
+  for (unsigned int i = 0; i < num_verts; i++) {
+    file << i << ". " << verts[i] << "\n";
+  }
+}
+
+void
+output_normals() {
+  MStatus status;
+  MDagPath dagPath, *_dag_path;
+  MObject component;
+
+  MItDag dag_iterator(MItDag::kDepthFirst, MFn::kTransform, &status);
+  if (!status) {
+    status.perror("MItDag constructor");
+    exit(1);
+  }
+
+  while (!dag_iterator.isDone()) {
+    status = dag_iterator.getPath(dagPath);
+    if (!status) {
+      status.perror("MItDag::getPath");
+      exit(1);
+    }
+
+    _dag_path = new MDagPath(dagPath);
+
+    MFnDagNode dag_node(*_dag_path, &status);
+    if (!status) {
+      status.perror("MFnDagNode constructor");
+      exit(1);
+    }
+
+    cerr << dag_node.name() << "\n";
+
+    if (_dag_path->hasFn(MFn::kMesh)) {
+      unsigned int numShapes;
+      status = _dag_path->numberOfShapesDirectlyBelow(numShapes);
+      cerr << "----numShapes: " << numShapes << "\n";
+      /*
+      if (numShapes == 1) {
+        _dag_path->extendToShape();
+      }
+      */
+    }
+    
+    MItMeshPolygon faceIter(*_dag_path, component, &status);
+    if( !status ) cerr << "Error at MItMeshPolygon" << endl;
+  
+    MFnMesh meshFn(*_dag_path);
+  
+    // Traverse the polygonal face
+    for (; !faceIter.isDone();faceIter.next()) {
+      int nVerts = faceIter.polygonVertexCount();
+      
+      // Traverse the vertices to get their indexes and print out the normals
+      for (int i = 0;i<nVerts;i++) {
+        MVector normal;
+        faceIter.getNormal( i, normal, MSpace::kWorld );
+        cerr << "MItMeshPolygon::getNormal for the vertex [" << faceIter.vertexIndex(i) 
+             << "] for face [" << faceIter.index() << "] is " << normal << endl;
+      }
+    }
+
+    dag_iterator.next();
+  }
+}
+
+int
+main(int argc, char *argv[]) {
+  cerr << "Initializing Maya\n";
+  MStatus status;
+  status = MLibrary::initialize(argv[0]);
+  if (!status) {
+    status.perror("Could not initialize");
+    exit(1);
+  }
+
+  cerr << "Using Maya library version " << MGlobal::mayaVersion() << "\n";
+
+  if (argc < 2) {
+    cerr << "\nUsage:\n\nnormal_test.cxx file.mb mesh_name\n\n";
+    exit(1);
+  }
+
+  MString filename = argv[1];
+
+  MFileIO::newFile(true);
+
+  cerr << "Reading " << filename << "\n";
+  status = MFileIO::open(filename);
+  if (!status) {
+    status.perror(filename);
+    exit(1);
+  }
+
+  output_normals();
+
+  return 0;
+}