Merge branch 'packonce' into '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 bool m_writeComments;
65 public:
66 WriteTokensWalker( TokenWriter& writer, bool ignorePatches, bool writeComments )
67         : m_writer( writer ), m_ignorePatches( ignorePatches ), m_writeComments( writeComments ){
68 }
69
70 bool pre( scene::Node &node ) const {
71         m_stack.push( false );
72
73         Entity* entity = Node_getEntity( node );
74         if ( entity != 0 ) {
75                 if ( m_writeComments ) {
76                         m_writer.writeToken( "//" );
77                         m_writer.writeToken( "entity" );
78                         m_writer.writeUnsigned( g_count_entities++ );
79                         m_writer.nextLine();
80                 }
81
82                 m_writer.writeToken( "{" );
83                 m_writer.nextLine();
84                 m_stack.top() = true;
85
86                 Entity_ExportTokens( *entity, m_writer );
87         }
88         else
89         {
90                 MapExporter* exporter = Node_getMapExporter( node );
91                 if ( exporter != 0
92                         && !( m_ignorePatches && Node_isPatch( node ) ) ) {
93                         if ( m_writeComments ) {
94                                 m_writer.writeToken( "//" );
95                                 m_writer.writeToken( "brush" );
96                                 m_writer.writeUnsigned( g_count_brushes++ );
97                                 m_writer.nextLine();
98                         }
99
100                         exporter->exportTokens( m_writer );
101                 }
102         }
103
104         return true;
105 }
106
107 void post( scene::Node& node ) const {
108         if ( m_stack.top() ) {
109                 m_writer.writeToken( "}" );
110                 m_writer.nextLine();
111         }
112         m_stack.pop();
113 }
114 };
115
116 void Map_Write( scene::Node& root, GraphTraversalFunc traverse, TokenWriter& writer, bool ignorePatches, bool writeComments ){
117         g_count_entities = 0;
118         traverse( root, WriteTokensWalker( writer, ignorePatches, writeComments ) );
119 }