Merge remote-tracking branch 'ttimo/master'
[xonotic/netradiant.git] / plugins / mapq3 / write.cpp
1 /*
2    Copyright (C) 2001-2006, William Joseph.
3    All Rights Reserved.
4
5    This file is part of GtkRadiant.
6
7    GtkRadiant is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11
12    GtkRadiant is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16
17    You should have received a copy of the GNU General Public License
18    along with GtkRadiant; if not, write to the Free Software
19    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
20  */
21
22 #include "write.h"
23
24 #include "ientity.h"
25 #include "iscriplib.h"
26 #include "scenelib.h"
27
28 inline MapExporter* Node_getMapExporter( scene::Node& node ){
29         return NodeTypeCast<MapExporter>::cast( node );
30 }
31
32
33 static std::size_t g_count_entities;
34 static std::size_t g_count_brushes;
35
36
37 void Entity_ExportTokens( const Entity& entity, TokenWriter& writer ){
38         g_count_brushes = 0;
39
40         class WriteKeyValue : public Entity::Visitor
41         {
42         TokenWriter& m_writer;
43 public:
44         WriteKeyValue( TokenWriter& writer )
45                 : m_writer( writer ){
46         }
47
48         void visit( const char* key, const char* value ){
49                 m_writer.writeString( key );
50                 m_writer.writeString( value );
51                 m_writer.nextLine();
52         }
53
54         } visitor( writer );
55
56         entity.forEachKeyValue( visitor );
57 }
58
59 class WriteTokensWalker : public scene::Traversable::Walker
60 {
61 mutable Stack<bool> m_stack;
62 TokenWriter& m_writer;
63 bool m_ignorePatches;
64 public:
65 WriteTokensWalker( TokenWriter& writer, bool ignorePatches )
66         : m_writer( writer ), m_ignorePatches( ignorePatches ){
67 }
68 bool pre( scene::Node& node ) const {
69         m_stack.push( false );
70
71         Entity* entity = Node_getEntity( node );
72         if ( entity != 0 ) {
73                 m_writer.writeToken( "//" );
74                 m_writer.writeToken( "entity" );
75                 m_writer.writeUnsigned( g_count_entities++ );
76                 m_writer.nextLine();
77
78                 m_writer.writeToken( "{" );
79                 m_writer.nextLine();
80                 m_stack.top() = true;
81
82                 Entity_ExportTokens( *entity, m_writer );
83         }
84         else
85         {
86                 MapExporter* exporter = Node_getMapExporter( node );
87                 if ( exporter != 0
88                          && !( m_ignorePatches && Node_isPatch( node ) ) ) {
89                         m_writer.writeToken( "//" );
90                         m_writer.writeToken( "brush" );
91                         m_writer.writeUnsigned( g_count_brushes++ );
92                         m_writer.nextLine();
93
94                         exporter->exportTokens( m_writer );
95                 }
96         }
97
98         return true;
99 }
100 void post( scene::Node& node ) const {
101         if ( m_stack.top() ) {
102                 m_writer.writeToken( "}" );
103                 m_writer.nextLine();
104         }
105         m_stack.pop();
106 }
107 };
108
109 void Map_Write( scene::Node& root, GraphTraversalFunc traverse, TokenWriter& writer, bool ignorePatches ){
110         g_count_entities = 0;
111         traverse( root, WriteTokensWalker( writer, ignorePatches ) );
112 }