TSH&R-PC

1 files changed, 222 insertions, 0 deletions

diff --git a/tools/trackeditor/code/commands/trackeditorcommands.cpp b/tools/trackeditor/code/commands/trackeditorcommands.cpp
new file mode 100644
index 0000000..29aeb76
--- /dev/null
+++ b/tools/trackeditor/code/commands/trackeditorcommands.cpp
@@ -0,0 +1,222 @@
+#include "precompiled/PCH.h"
+
+#include "trackeditorcommands.h"
+#include "main/trackeditor.h"
+#include "utility/mext.h"
+#include "main/constants.h"
+
+//TEStateChange
+const char* TEStateChangeCommand::stringId = "TE_StateChange";
+
+//==============================================================================
+// TEStateChangeCommand::creator
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ()
+//
+// Return:      void
+//
+//==============================================================================
+void* TEStateChangeCommand::creator()
+{
+    return new TEStateChangeCommand();
+}
+
+//==============================================================================
+// TEStateChangeCommand::doIt
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ( const MArgList& args )
+//
+// Return:      MStatus 
+//
+//==============================================================================
+MStatus TEStateChangeCommand::doIt( const MArgList& args )
+{
+    assert( args.length() == 1 );
+
+    int arg;
+    args.get( 0, arg );
+
+    TrackEditor::SetEditMode( (TrackEditor::EditMode) arg );
+
+    return MStatus::kSuccess;
+}
+
+//TEGetSelectedVertexPosition
+
+const char* TEGetSelectedVertexPosition::stringId = "TE_GetSelectedVertexPosition";
+
+//==============================================================================
+// TEGetSelectedVertexPosition::creator
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ()
+//
+// Return:      void
+//
+//==============================================================================
+void* TEGetSelectedVertexPosition::creator()
+{
+    return new TEGetSelectedVertexPosition(); 
+}
+
+//==============================================================================
+// TEGetSelectedVertexPosition::doIt
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ( const MArgList& args )
+//
+// Return:      MStatus 
+//
+//==============================================================================
+MStatus TEGetSelectedVertexPosition::doIt( const MArgList& args )
+{
+    MStatus status;
+    MDoubleArray returnVal( 3, 0 );
+
+    assert( args.length() == 1 );  //Only one arg.
+
+    MString argString;
+    args.get( 0, argString );
+
+    MSelectionList activeList;
+    MGlobal::getActiveSelectionList(activeList);
+
+    MItSelectionList iter( activeList,  MFn::kMeshVertComponent, &status);
+
+    //We're only going to deal with the first selected item.  Don't select more
+    //Than one please.
+    
+    if ( !iter.isDone() )
+    {
+        MDagPath item;
+        MObject component;
+        iter.getDagPath( item, component );
+        // do something with it
+
+        MStatus isMeshIT;
+        MItMeshVertex mITVert( item , component, &isMeshIT );
+
+        if(isMeshIT == MS::kSuccess)
+        {
+            MPoint vertPos;
+            double x, y, z;
+            
+            if ( argString == MString( "local" ) )
+            {
+                vertPos = mITVert.position( MSpace::kObject, &status );
+                x = vertPos[0];
+                y = vertPos[1];
+                z = vertPos[2];
+            }
+            else  //"world"
+            {
+                vertPos = mITVert.position( MSpace::kWorld, &status );
+                x = vertPos[0];
+                y = vertPos[1];
+                z = vertPos[2];
+            }
+            
+            returnVal[0] = x / TEConstants::Scale;
+            returnVal[1] = y / TEConstants::Scale;
+            returnVal[2] = z / TEConstants::Scale;
+
+        }
+
+        iter.next(); //This is to test if there are more verts than needed.
+    }
+    else
+    {
+        MExt::DisplayWarning("No vertices selected!");
+    }
+
+    if ( !iter.isDone() )
+    {
+        MExt::DisplayWarning("Too many vertices selected!");
+    }
+
+
+    MPxCommand::setResult( returnVal );
+
+    return MStatus::kSuccess;
+}
+
+
+//TEGetSelectedVertexIndex
+
+const char* TEGetSelectedVertexIndex::stringId = "TE_GetSelectedVertexIndex";
+
+//==============================================================================
+// TEGetSelectedVertexIndex::creator
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ()
+//
+// Return:      void
+//
+//==============================================================================
+void* TEGetSelectedVertexIndex::creator()
+{
+    return new TEGetSelectedVertexIndex();
+}
+
+//==============================================================================
+// TEGetSelectedVertexIndex::doIt
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ( const MArgList& args )
+//
+// Return:      MStatus 
+//
+//==============================================================================
+MStatus TEGetSelectedVertexIndex::doIt( const MArgList& args )
+{
+    int returnVal = -1;
+
+    MStatus status;
+    MSelectionList activeList;
+    MGlobal::getActiveSelectionList(activeList);
+
+    MItSelectionList iter( activeList,  MFn::kMeshVertComponent, &status);
+
+    //We're only going to deal with the first selected item.  Don't select more
+    //Than one please.
+    
+    if ( !iter.isDone() )
+    {
+        MDagPath item;
+        MObject component;
+        iter.getDagPath( item, component );
+        // do something with it
+
+        MStatus isMeshIT;
+        MItMeshVertex mITVert( item , component, &isMeshIT );
+
+        if(isMeshIT == MS::kSuccess)
+        {
+            returnVal = mITVert.index();
+        }
+
+        iter.next(); //This is to test if there are more verts than needed.
+    }
+    else
+    {
+        MExt::DisplayWarning("No vertices selected!");
+    }
+
+    if ( !iter.isDone() )
+    {
+        MExt::DisplayWarning("Too many vertices selected!");
+    }
+
+    MPxCommand::setResult( returnVal );
+
+    return MStatus::kSuccess;
+}