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 {
  30     return NodeTypeCast<MapExporter>::cast(node);
  31 }
  32
  33
  34 static std::size_t g_count_entities;
  35 static std::size_t g_count_brushes;
  36
  37
  38 void Entity_ExportTokens(const Entity &entity, TokenWriter &writer)
  39 {
  40     g_count_brushes = 0;
  41
  42     class WriteKeyValue : public Entity::Visitor {
  43         TokenWriter &m_writer;
  44     public:
  45         WriteKeyValue(TokenWriter &writer)
  46                 : m_writer(writer)
  47         {
  48         }
  49
  50         void visit(const char *key, const char *value)
  51         {
  52             m_writer.writeString(key);
  53             m_writer.writeString(value);
  54             m_writer.nextLine();
  55         }
  56
  57     } visitor(writer);
  58
  59     entity.forEachKeyValue(visitor);
  60 }
  61
  62 class WriteTokensWalker : public scene::Traversable::Walker {
  63     mutable Stack<bool> m_stack;
  64     TokenWriter &m_writer;
  65     bool m_ignorePatches;
  66     bool m_writeComments;
  67 public:
  68     WriteTokensWalker(TokenWriter &writer, bool ignorePatches, bool writeComments)
  69             : m_writer(writer), m_ignorePatches(ignorePatches), m_writeComments(writeComments)
  70     {
  71     }
  72
  73     bool pre(scene::Node &node) const
  74     {
  75         m_stack.push(false);
  76
  77         Entity *entity = Node_getEntity(node);
  78         if (entity != 0) {
  79             if (m_writeComments) {
  80                 m_writer.writeToken("//");
  81                 m_writer.writeToken("entity");
  82                 m_writer.writeUnsigned(g_count_entities++);
  83                 m_writer.nextLine();
  84             }
  85
  86             m_writer.writeToken("{");
  87             m_writer.nextLine();
  88             m_stack.top() = true;
  89
  90             Entity_ExportTokens(*entity, m_writer);
  91         } else {
  92             MapExporter *exporter = Node_getMapExporter(node);
  93             if (exporter != 0
  94                 && !(m_ignorePatches && Node_isPatch(node))) {
  95                 if (m_writeComments) {
  96                     m_writer.writeToken("//");
  97                     m_writer.writeToken("brush");
  98                     m_writer.writeUnsigned(g_count_brushes++);
  99                     m_writer.nextLine();
 100                 }
 101
 102                 exporter->exportTokens(m_writer);
 103             }
 104         }
 105
 106         return true;
 107     }
 108
 109     void post(scene::Node &node) const
 110     {
 111         if (m_stack.top()) {
 112             m_writer.writeToken("}");
 113             m_writer.nextLine();
 114         }
 115         m_stack.pop();
 116     }
 117 };
 118
 119 void Map_Write(scene::Node &root, GraphTraversalFunc traverse, TokenWriter &writer, bool ignorePatches, bool writeComments)
 120 {
 121     g_count_entities = 0;
 122     traverse(root, WriteTokensWalker(writer, ignorePatches, writeComments));
 123 }