diff options
Diffstat (limited to 'tools/trackeditor/code/main/trackeditor.cpp')
| -rw-r--r-- | tools/trackeditor/code/main/trackeditor.cpp | 350 |
1 files changed, 350 insertions, 0 deletions
diff --git a/tools/trackeditor/code/main/trackeditor.cpp b/tools/trackeditor/code/main/trackeditor.cpp new file mode 100644 index 0000000..0b0ccc2 --- /dev/null +++ b/tools/trackeditor/code/main/trackeditor.cpp @@ -0,0 +1,350 @@ +#include "precompiled/PCH.h" + +#include "trackeditor.h" +#include "utility/mext.h" + +#include "nodes/tiledisplay.h" +#include "nodes/fenceline.h" +#include "nodes/intersection.h" +#include "nodes/walllocator.h" +#include "nodes/road.h" +#include "nodes/treelineshapenode.h" +#include "nodes/pedpath.h" + +const char* TrackEditor::Name = "TrackEditorNode"; +TrackEditor::EditMode TrackEditor::sEditMode = TrackEditor::OFF; +unsigned int TrackEditor::sNodeAddedbackID = 0; +unsigned int TrackEditor::sWindowClosedCallbackID = 0; +bool TrackEditor::sDeleteTreelines = true; + + +//============================================================================== +// TrackEditor::TrackEditor +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: TrackEditor +// +//============================================================================== +TrackEditor::TrackEditor() +{ +// sNodeAddedbackID = MDGMessage::addNodeAddedCallback ( NodeAddedCB, +// MString( "surfaceShape" ) ); +}; + +//============================================================================== +// TrackEditor::~TrackEditor +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: TrackEditor +// +//============================================================================== +TrackEditor::~TrackEditor() +{ + if ( sNodeAddedbackID ) + { + MDGMessage::removeCallback( sNodeAddedbackID ); + } + + if ( sWindowClosedCallbackID ) + { + MUiMessage::removeCallback( sWindowClosedCallbackID ); + } + + RemoveTileDisplayNode(); +}; + +//============================================================================== +// TrackEditor::Exists +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: bool +// +//============================================================================== +bool TrackEditor::Exists() +{ + MDagPath pathToTrackEditor; + return MExt::FindDagNodeByName( &pathToTrackEditor, MString( TrackEditor::Name ) ); +} + +//============================================================================== +// TrackEditor::AddChild +//============================================================================== +// Description: Comment +// +// Parameters: ( MObject& obj ) +// +// Return: MStatus +// +//============================================================================== +MStatus TrackEditor::AddChild( MObject& obj ) +{ + //Make sure this exists. + CreateTrackEditorNode(); + + MDagPath pathToTrackEditor; + + bool good = false; + + if ( MExt::FindDagNodeByName( &pathToTrackEditor, MString( TrackEditor::Name ) ) ) + { + good = true; + } + else + { + MGlobal::sourceFile( MString( "te_setup.mel" ) ); + + if ( MExt::FindDagNodeByName( &pathToTrackEditor, MString( TrackEditor::Name ) ) ) + { + good = true; + } + } + + if ( good ) + { + MFnDagNode fnDagNodeTE; + + //Which type? + MFnDagNode fnDagNodeObj( obj ); + + if ( fnDagNodeObj.typeId() == FenceLineNode::id ) + { + //This is a fenceline, parent to the "Fences" node. + MDagPath dagPath; + if ( MExt::FindDagNodeByName( &dagPath, MString("Fences"), pathToTrackEditor.node() ) ) + { + fnDagNodeTE.setObject( dagPath.node() ); + } + else + { + MExt::DisplayError( "Someone has deleted Terrain Edit nodes!!" ); + } + } + if ( fnDagNodeObj.typeId() == PedPathNode::id ) + { + //This is a ped path, parent to the "PedPaths" node. + MDagPath dagPath; + if ( MExt::FindDagNodeByName( &dagPath, MString("PedPaths"), pathToTrackEditor.node() ) ) + { + fnDagNodeTE.setObject( dagPath.node() ); + } + else + { + MExt::DisplayError( "Someone has deleted Terrain Edit nodes!!" ); + } + } + else if ( fnDagNodeObj.typeId() == IntersectionLocatorNode::id ) + { + //This is a fenceline, parent to the "Intersections" node. + MDagPath dagPath; + if ( MExt::FindDagNodeByName( &dagPath, MString("Intersections"), pathToTrackEditor.node() ) ) + { + fnDagNodeTE.setObject( dagPath.node() ); + } + else + { + MExt::DisplayError( "Someone has deleted Terrain Edit nodes!!" ); + } + } + else if ( fnDagNodeObj.typeId() == RoadNode::id ) + { + //This is a fenceline, parent to the "Roads" node. + MDagPath dagPath; + if ( MExt::FindDagNodeByName( &dagPath, MString("Roads"), pathToTrackEditor.node() ) ) + { + fnDagNodeTE.setObject( dagPath.node() ); + } + else + { + MExt::DisplayError( "Someone has deleted Terrain Edit nodes!!" ); + } + } + else if ( fnDagNodeObj.typeId() == TETreeLine::TreelineShapeNode::id ) + { + //This is a tree line, add to the "Treelines" node + MDagPath dagPath; + if ( MExt::FindDagNodeByName( &dagPath, MString("Treelines"), pathToTrackEditor.node() ) ) + { + fnDagNodeTE.setObject( dagPath.node() ); + } + else + { + MExt::DisplayError( "Someone has deleted Terrain Edit nodes!!" ); + } + } + else + { + fnDagNodeTE.setObject( pathToTrackEditor.node() ); + } + + MObject objT = fnDagNodeObj.parent( 0 ); + + return fnDagNodeTE.addChild( objT ); + } + + return MS::kFailure; +} + +//Edit mode stuff. +//============================================================================== +// TrackEditor::GetEditMode +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: TrackEditor +// +//============================================================================== +TrackEditor::EditMode TrackEditor::GetEditMode() +{ + //Test the track editor radio buttons for their state. + + return sEditMode; +}; + +//============================================================================== +// TrackEditor::SetEditMode +//============================================================================== +// Description: Comment +// +// Parameters: ( TrackEditor::EditMode mode ) +// +// Return: void +// +//============================================================================== +void TrackEditor::SetEditMode( TrackEditor::EditMode mode ) +{ + //Setup whatever needs setting up for the given mode. + switch( mode ) + { + case OFF: + { + RemoveTileDisplayNode(); + + MGlobal::executeCommand( MString("teCloseEditorWindow()") ); + } + break; + case EDIT: + { + //Make sure this exists; + CreateTileDisplayNode(); + + MGlobal::executeCommand( MString("teOpenEditorWindow()") ); + + //We should register a callback for when the window is closed. + sWindowClosedCallbackID = MUiMessage::addUiDeletedCallback( MString( "TE_TileEditor" ), WindowClosedCB ); + } + break; + case DISPLAY: + { + //Make sure this exists; + CreateTileDisplayNode(); + + MGlobal::executeCommand( MString("teCloseEditorWindow()") ); + } + break; + default: + { + break; + } + } + + sEditMode = mode; +} + +//============================================================================== +// TrackEditor::NodeAddedCB +//============================================================================== +// Description: Comment +// +// Parameters: ( MObject& node, void* data ) +// +// Return: void +// +//============================================================================== +void TrackEditor::NodeAddedCB( MObject& node, void* data ) +{ +// assert( false ); +} + +//============================================================================== +// TrackEditor::WindowClosedCB +//============================================================================== +// Description: Comment +// +// Parameters: ( void* data ) +// +// Return: void +// +//============================================================================== +void TrackEditor::WindowClosedCB( void* data ) +{ + SetEditMode( OFF ); + + MUiMessage::removeCallback( sWindowClosedCallbackID ); +} + +//============================================================================== +// TrackEditor::CreateTrackEditorNode +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: void +// +//============================================================================== +void TrackEditor::CreateTrackEditorNode() +{ + MGlobal::executeCommand( "te_Create_TrackEditorNode()" ); +} + +//============================================================================== +// TrackEditor::CreateTileDisplayNode +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: void +// +//============================================================================== +void TrackEditor::CreateTileDisplayNode() +{ + if ( !MExt::FindDagNodeByName( NULL, MString( TileDisplayNode::stringId ) ) ) + { + MExt::CreateNode( 0, 0, MString( TileDisplayNode::stringId ) ); + } +} + +//============================================================================== +// TrackEditor::RemoveTileDisplayNode +//============================================================================== +// Description: Comment +// +// Parameters: () +// +// Return: void +// +//============================================================================== +void TrackEditor::RemoveTileDisplayNode() +{ + MDagPath dagPath; + + if ( MExt::FindDagNodeByName( &dagPath, MString( TileDisplayNode::stringId ) ) ) + { + MFnDagNode fnDagNode( dagPath ); + + MExt::DeleteNode( fnDagNode.object(), true ); + } +} + |