convertVerticesToEdgesCmd/convertVerticesToEdgesCmd.cpp

convertVerticesToEdgesCmd/convertVerticesToEdgesCmd.cpp
//-
// ==========================================================================
// Copyright 1995,2006,2008 Autodesk, Inc. All rights reserved.
//
// Use of this software is subject to the terms of the Autodesk
// license agreement provided at the time of installation or download,
// or which otherwise accompanies this software in either electronic
// or hard copy form.
// ==========================================================================
//+
// DISCLAIMER: THIS PLUGIN IS PROVIDED AS IS. IT IS NOT SUPPORTED BY
// AUTODESK, SO PLEASE USE AND MODIFY AT YOUR OWN RISK.
//
// PLUGIN NAME: Convert Vertices To Contained Edges Command v1.0
// FILE: convertVerticesToEdgesCmd.cpp
// DESCRIPTION: o MEL command that converts a selection of vertices into a selection of edges that interconnect
// the original vertices (i.e. only edges whose composite vertices are contained in the original
// vertex selection). The command's return value is a string array that contains the names of
// all of the new contained edges.
// o This MEL command has no flags, returns a string array, and operates on selected vertices.
// o Example MEL usage:
// select -r pCube1.vtx[2:5];
// string $convertedEdges[] = `convertVerticesToEdges`;
// // Result: pCube1.e[1:2] pCube1.e[6:7] //
// AUTHOR: QT
// LAST UPDATED: Oct. 11th, 2001.
// TESTED ON: Maya 4.0 on Windows
// STEPS TO INSTALL: 1) Unzip the file convertVerticesToEdges_v1_0_Maya4_Windows.zip to any temporary directory.
// 2) Copy the file convertVerticesToEdgesCmd.mll to: /AW/Maya4.0/bin/plug-ins/
// 3) Launch Maya and load the plugin from Windows->Setting/Preferences->Plugin Manager.
// BEGIN PLUGIN:
// =============
// HEADER FILES:
#include <maya/MFnPlugin.h>
#include <maya/MPxCommand.h>
#include <maya/MGlobal.h>
#include <maya/MSelectionList.h>
#include <maya/MItSelectionList.h>
#include <maya/MObject.h>
#include <maya/MDagPath.h>
#include <maya/MItMeshVertex.h>
#include <maya/MItMeshEdge.h>
// MAIN CLASS FOR THE MEL COMMAND:
class convertVerticesToContainedEdgesCommand : public MPxCommand
{
public:
convertVerticesToContainedEdgesCommand();
virtual ~convertVerticesToContainedEdgesCommand();
static void* creator();
bool isUndoable() const;
private:
MSelectionList previousSelectionList;
};
// CONSTRUCTOR:
convertVerticesToContainedEdgesCommand::convertVerticesToContainedEdgesCommand()
{
}
// DESTRUCTOR CLEARS, AND FREES MEMORY OF THE PREVIOUS SELECTION LIST:
convertVerticesToContainedEdgesCommand::~convertVerticesToContainedEdgesCommand()
{
previousSelectionList.clear();
}
// FOR CREATING AN INSTANCE OF THIS COMMAND:
void* convertVerticesToContainedEdgesCommand::creator()
{
return new convertVerticesToContainedEdgesCommand;
}
// MAKE THIS COMMAND UNDOABLE:
bool convertVerticesToContainedEdgesCommand::isUndoable() const
{
return true;
}
// SAVE THE PREVIOUS SELECTION, AND THEN CALL redoIt(), WHICH DOES MOST OF THE WORK:
MStatus convertVerticesToContainedEdgesCommand::doIt(const MArgList& args)
{
MGlobal::getActiveSelectionList(previousSelectionList);
return redoIt();
}
// CONVERTS THE SELECTED VERTICES TO "CONTAINED EDGES", AND RETURN FROM THE MEL COMMAND, A STRING ARRAY OF THE NEW EDGES:
MStatus convertVerticesToContainedEdgesCommand::redoIt()
{
MSelectionList finalEdgesSelection;
MDagPath meshDagPath;
MObject multiVertexComponent, singleVertexComponent;
int dummyIndex;
// ITERATE THROUGH EACH "VERTEX COMPONENT" THAT IS CURRENTLY SELECTED:
for (MItSelectionList vertexComponentIter(previousSelectionList, MFn::kMeshVertComponent); !vertexComponentIter.isDone(); vertexComponentIter.next())
{
// STORE THE DAGPATH, COMPONENT OBJECT AND MESH NAME OF THE CURRENT VERTEX COMPONENT:
vertexComponentIter.getDagPath(meshDagPath, multiVertexComponent);
MString meshName = meshDagPath.fullPathName();
// VERTEX COMPONENT HAS TO CONTAIN AT LEAST ONE VERTEX:
if (!multiVertexComponent.isNull())
{
// ITERATE THROUGH EACH "VERTEX" IN THE CURRENT VERTEX COMPONENT:
for (MItMeshVertex vertexIter(meshDagPath, multiVertexComponent); !vertexIter.isDone(); vertexIter.next())
{
// FOR STORING THE EDGES CONNECTED TO EACH VERTEX:
MIntArray connectedEdgesIndices;
vertexIter.getConnectedEdges(connectedEdgesIndices);
// ITERATE THROUGH EACH EDGE CONNECTED TO THE CURRENT VERTEX:
MItMeshEdge edgeIter(meshDagPath);
for (unsigned i=0; i<connectedEdgesIndices.length(); i++)
{
// FIND AND STORE THE *FIRST* "END VERTEX" OF THE CURRENT EDGE:
edgeIter.setIndex(connectedEdgesIndices[i], dummyIndex);
MSelectionList singleVertexList;
MString vertexName = meshName;
vertexName += ".vtx[";
vertexName += edgeIter.index(0);
vertexName += "]";
singleVertexList.add(vertexName);
singleVertexList.getDagPath(0, meshDagPath, singleVertexComponent);
// SEE WHETHER THE VERTEX BELONGS TO THE ORIGINAL SELECTION, AND IF IT DOES PROCEED TO CHECK NEXT END VERTEX:
if (!singleVertexComponent.isNull() && previousSelectionList.hasItem(meshDagPath, singleVertexComponent))
{
// FIND AND STORE THE *SECOND* "END VERTEX" OF THE CURRENT EDGE:
singleVertexList.clear();
vertexName = meshName;
vertexName += ".vtx[";
vertexName += edgeIter.index(1);
vertexName += "]";
singleVertexList.add(vertexName);
singleVertexList.getDagPath(0, meshDagPath, singleVertexComponent);
// SEE WHETHER THE VERTEX BELONGS TO THE ORIGINAL SELECTION, AND IF IT DOES, ADD THE EDGE TO THE FINAL CONTAINED EDGES LIST:
if (!singleVertexComponent.isNull() && previousSelectionList.hasItem(meshDagPath, singleVertexComponent))
{
MString edgeName = meshName;
edgeName += ".e[";
edgeName += connectedEdgesIndices[i];
edgeName += "]";
finalEdgesSelection.add(edgeName);
}
}
}
}
}
}
// FINALLY, MAKE THE NEW "CONTAINED EDGES", THE CURRENT SELECTION:
// RETURN NEW CONTAINED EDGES LIST FROM THE MEL COMMAND, AS AN ARRAY OF STRINGS:
MStringArray containedEdgesArray;
finalEdgesSelection.getSelectionStrings(containedEdgesArray);
MPxCommand::setResult(containedEdgesArray);
return MS::kSuccess;
}
// TO UNDO THE COMMAND, SIMPLY RESTORE THE ORIGINAL SELECTION BEFORE THE COMMAND WAS INVOKED:
MStatus convertVerticesToContainedEdgesCommand::undoIt()
{
return MS::kSuccess;
}
// INITIALIZES THE PLUGIN:
MStatus initializePlugin(MObject obj)
{
MStatus status;
MFnPlugin plugin(obj, PLUGIN_COMPANY, "4.0", "Any");
status = plugin.registerCommand("convertVerticesToEdges", convertVerticesToContainedEdgesCommand::creator);
if (!status)
{
status.perror("registerCommand");
return status;
}
return status;
}
// UNINITIALIZES THE PLUGIN:
MStatus uninitializePlugin(MObject obj)
{
MStatus status;
MFnPlugin plugin(obj);
status = plugin.deregisterCommand("convertVerticesToEdges");
if (!status)
{
status.perror("deregisterCommand");
return status;
}
return status;
}
// ============================================================================================================== END PLUGIN.