TSH&R-PC

1 files changed, 282 insertions, 0 deletions

diff --git a/tools/worldbuilder/code/commands/export.cpp b/tools/worldbuilder/code/commands/export.cpp
new file mode 100644
index 0000000..dda3c16
--- /dev/null
+++ b/tools/worldbuilder/code/commands/export.cpp
@@ -0,0 +1,282 @@
+#include "main/toolhack.h"
+
+#include "precompiled/PCH.h"
+
+#include "export.h"
+#include "main/constants.h"
+#include "main/worldbuilder.h"
+#include "nodes/eventlocatornode.h"
+#include "nodes/scriptlocatornode.h"
+#include "nodes/genericlocatornode.h"
+#include "nodes/carstartlocatornode.h"
+#include "nodes/triggervolumenode.h"
+#include "nodes/splinelocatornode.h"
+#include "nodes/zoneeventlocatornode.h"
+#include "nodes/occlusionlocatornode.h"
+#include "nodes/railcamlocatornode.h"
+#include "nodes/interiorentrancelocatornode.h"
+#include "nodes/directionallocatornode.h"
+#include "nodes/actioneventlocatornode.h"
+#include "nodes/fovlocatornode.h"
+#include "nodes/BreakableCameraLocatorNode.h"
+#include "nodes/StaticCameraLocatorNode.h"
+#include "nodes/PedGroupLocator.h"
+#include "nodes/wbspline.h"
+#include "utility/mui.h"
+#include "utility/mext.h"
+
+#include <toollib.hpp>
+
+#include "../../../game/code/meta/locatortypes.h"
+
+const char*  ExportCommand::stringId = "WB_Export";
+bool ExportCommand::sRegisteredChunks = false;
+
+ExportCommand::ExportCommand() {};
+ExportCommand::~ExportCommand() {};
+
+//==============================================================================
+// ExportCommand::creator
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ()
+//
+// Return:      void
+// 
+//==============================================================================
+void* ExportCommand::creator()
+{
+    return new ExportCommand();
+}
+
+//==============================================================================
+// ExportCommand::doIt
+//==============================================================================
+// Description: Comment
+//
+// Parameters:  ( const MArgList& args )
+//
+// Return:      MStatus 
+//
+//==============================================================================
+MStatus ExportCommand::doIt( const MArgList& args )
+{
+    if ( !sRegisteredChunks )
+    {
+        tlDataChunk::RegisterDefaultChunks();
+        sRegisteredChunks = true;
+    }
+   
+    MStatus status;
+
+    //Go through all the chunks looking for the types we want and exporting them.
+    //Alternatively, we could go thtough all the chunks and attempt to access the
+    //IExportable interface and then export if the interface exists...
+
+    //Args are all, or selected.
+
+    const unsigned char FILE_NAME_SIZE = 255;
+    char filePath[FILE_NAME_SIZE];
+    filePath[0] = '\0';
+
+    if ( MUI::FileDialog( filePath,  
+                          FILE_NAME_SIZE, 
+                          "World Builder Export", 
+                          "Pure3D(*.p3d)|*.p3d|All Files(*.*)|*.*||", 
+                          "p3d", 
+                          MUI::SAVE ) )
+    {
+        MDagPath pathToWorldBuilder;
+        if ( !MExt::FindDagNodeByName( &pathToWorldBuilder, MString( WorldBuilder::sName ) ) )
+        {
+            return MStatus::kSuccess;
+        }        
+
+        MItDag dagIt;
+        dagIt.reset( pathToWorldBuilder, MItDag::kBreadthFirst );
+
+        tlDataChunk* outChunk = new tlDataChunk;
+
+        //Put in a history chunk.
+        tlHistory history;
+        
+        char hist[256];
+//        sprintf(hist, "World Builder version: 2.0, toollib version: %s", tlversion);
+
+//        history.AddLine( hist );
+
+        outChunk->AppendSubChunk( history.Chunk(), 0 );
+
+        bool deleteLast = false;
+        MFnDependencyNode fnNode;
+        MObject lastObj;
+        MTypeId id;
+
+        while ( !dagIt.isDone() )
+        { 
+            fnNode.setObject( dagIt.item() );
+            id = fnNode.typeId();
+
+            tlDataChunk* newChunk = NULL;
+
+            if ( id == EventLocatorNode::id )
+            {
+                //Export an event locator;
+                newChunk = EventLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == ScriptLocatorNode::id )
+            {
+                //export a script locator
+                newChunk = ScriptLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == GenericLocatorNode::id )
+            {
+                //export a script locator
+                newChunk = GenericLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == CarStartLocatorNode::id )
+            {
+                //export a script locator
+                newChunk = CarStartLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == SplineLocatorNode::id )
+            {
+                //Export a spline locator
+                newChunk = SplineLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == ZoneEventLocatorNode::id )
+            {
+                //Export a zone event locator
+                newChunk = ZoneEventLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == OcclusionLocatorNode::id )
+            {
+                //Export a zone event locator
+                newChunk = OcclusionLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == InteriorEntranceLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = InteriorEntranceLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == DirectionalLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = DirectionalLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == ActionEventLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = ActionEventLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == FOVLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = FOVLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == BreakableCameraLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = BreakableCameraLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == StaticCameraLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = StaticCameraLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == PedGroupLocatorNode::id )
+            {
+                //Export an interior entrance
+                newChunk = PedGroupLocatorNode::Export( dagIt.item() );
+            }
+            else if ( id == RailCamLocatorNode::id )
+            {
+                //Don't export, but don't delete either.
+            }
+            else
+            {
+                //Is it a nurbs curve or a trigger volume?
+                MFnNurbsCurve( dagIt.item(), &status );
+
+                if ( !status && id != TriggerVolumeNode::id )
+                {
+                    deleteLast = true;
+                }
+            }
+
+            if ( newChunk )
+            {
+                tlWBLocatorChunk* newLoc = dynamic_cast<tlWBLocatorChunk*>(newChunk);
+
+                if ( newLoc )
+                {
+                    if ( newLoc->Type() == LocatorType::GENERIC )
+                    {
+                        unsigned int i;
+                        for ( i = 0; i < newChunk->SubChunkCount(); ++i )
+                        {
+                            outChunk->AppendSubChunk( newChunk->GetSubChunk( i ) );
+                        
+                        }
+
+                        if ( i == 0 )
+                        {
+                            //Append to output file
+                            outChunk->AppendSubChunk( newChunk );
+                        }
+                    }
+                    else
+                    {
+                        outChunk->AppendSubChunk( newChunk );
+                    }
+                }
+                else
+                {
+
+                    //Append to output file
+                    outChunk->AppendSubChunk( newChunk );
+                }
+            }
+            
+            if ( deleteLast )
+            {
+                lastObj = dagIt.item();
+            }
+            
+            dagIt.next();
+
+            if ( deleteLast )
+            {
+                MFnTransform fnTransform( lastObj, &status );
+                if ( !status )
+                {
+                    MExt::DisplayWarning( "Deleting useless node: %s", fnNode.name().asChar() );
+                    MExt::DeleteNode( lastObj, true );
+                }
+                deleteLast = false;
+            }
+        }
+
+        tlFile output(new tlFileByteStream(filePath, omWRITE), tlFile::CHUNK32);
+
+        if(!output.IsOpen()) 
+        {
+
+            MGlobal::displayError("Unable to write file!");
+
+            delete outChunk;
+            return MS::kFailure;
+        }
+
+        //Sort it out..
+        outChunk->SortSubChunks();
+        outChunk->Write(&output);
+
+        delete outChunk;
+
+        MGlobal::executeCommand( MString("flushUndo") );
+    }
+
+    return MS::kSuccess;
+}