- Added sunplug (Mapcoordinator-plugin for ET) by Topsun
[xonotic/netradiant.git] / contrib / brushexport / plugin.cpp
1 /*\r
2 Copyright (C) 2006, Thomas Nitschke.\r
3 All Rights Reserved.\r
4 \r
5 This file is part of GtkRadiant.\r
6 \r
7 GtkRadiant is free software; you can redistribute it and/or modify\r
8 it under the terms of the GNU General Public License as published by\r
9 the Free Software Foundation; either version 2 of the License, or\r
10 (at your option) any later version.\r
11 \r
12 GtkRadiant is distributed in the hope that it will be useful,\r
13 but WITHOUT ANY WARRANTY; without even the implied warranty of\r
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
15 GNU General Public License for more details.\r
16 \r
17 You should have received a copy of the GNU General Public License\r
18 along with GtkRadiant; if not, write to the Free Software\r
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA\r
20 */\r
21 #include "plugin.h"\r
22 \r
23 #include "iplugin.h"\r
24 #include "qerplugin.h"\r
25 \r
26 #include <gtk/gtk.h>\r
27 \r
28 #include "debugging/debugging.h"\r
29 #include "string/string.h"\r
30 #include "modulesystem/singletonmodule.h"\r
31 #include "stream/textfilestream.h"\r
32 #include "stream/stringstream.h"\r
33 #include "gtkutil/messagebox.h"\r
34 #include "gtkutil/filechooser.h"\r
35 \r
36 #include "ibrush.h"\r
37 #include "iscenegraph.h"\r
38 #include "iselection.h"\r
39 #include "ifilesystem.h"\r
40 #include "ifiletypes.h"\r
41 \r
42 #include "../../radiant/brush.h"\r
43 \r
44 namespace BrushExport\r
45\r
46   GtkWindow* g_mainwnd;\r
47   \r
48   class CExportFormatWavefront : public BrushVisitor\r
49   {\r
50     TextFileOutputStream& m_file;\r
51 \r
52     StringOutputStream vertexbuffer;\r
53     StringOutputStream texcoordbuffer;\r
54     StringOutputStream facebuffer;\r
55     \r
56     size_t vertices;\r
57     size_t exported;\r
58     \r
59   public:\r
60     \r
61     CExportFormatWavefront(TextFileOutputStream& file)\r
62         : m_file(file)\r
63     {\r
64       exported = 0;\r
65       vertices = 0;\r
66     }\r
67     \r
68     virtual ~CExportFormatWavefront(void) {}\r
69     \r
70     void visit(scene::Instance& instance)\r
71     {\r
72       BrushInstance* bptr = InstanceTypeCast<BrushInstance>::cast(instance);\r
73       if(bptr)\r
74       {\r
75         Brush& brush(bptr->getBrush());\r
76         \r
77         m_file << "\ng " << brush.name() << exported << "\n";\r
78         \r
79         brush.forEachFace(*this);\r
80     \r
81         m_file << vertexbuffer.c_str() << "\n";\r
82         m_file << texcoordbuffer.c_str();\r
83         m_file << facebuffer.c_str() << "\n";\r
84         \r
85         vertexbuffer.clear();\r
86         texcoordbuffer.clear();\r
87         facebuffer.clear();\r
88         ++exported;\r
89       }\r
90     }\r
91    \r
92     void visit(Face& face) const\r
93     {\r
94       // cast the stupid const away\r
95       const_cast<CExportFormatWavefront*>(this)->visit(face);\r
96     }\r
97     \r
98     void visit(Face& face)\r
99     {\r
100       size_t v_start = vertices;\r
101       const Winding& w(face.getWinding());\r
102       for(size_t i = 0; i < w.numpoints; ++i)\r
103       {\r
104         vertexbuffer << "v " << w[i].vertex.x() << " " << w[i].vertex.y() << " " << w[i].vertex.z() << "\n";\r
105         texcoordbuffer << "vt " << w[i].texcoord.x() << " " << w[i].texcoord.y() << "\n";\r
106         ++vertices;\r
107       }\r
108       \r
109       facebuffer << "\nf";\r
110       for(size_t i = v_start; i < vertices; ++i)\r
111         facebuffer << " " << i+1 << "/" << i+1;\r
112     }\r
113   };\r
114   \r
115   /**\r
116     Exporterclass which will pass every visit-call\r
117     to a special formatexporter.\r
118   */\r
119   template<class TExporterFormat>\r
120   class CExporter : public SelectionSystem::Visitor\r
121   {\r
122   public:\r
123     CExporter(TextFileOutputStream& file)\r
124       : m_exporter(file)\r
125     {}\r
126     \r
127     virtual ~CExporter(void) {}\r
128     \r
129     void visit(scene::Instance& instance) const\r
130     {\r
131       m_exporter.visit(instance);\r
132     }\r
133     \r
134   private:\r
135     mutable TExporterFormat m_exporter;\r
136   };\r
137   \r
138   template<class T>\r
139   void export_selected(TextFileOutputStream& file)\r
140   {\r
141     CExporter<T> exporter(file);\r
142     GlobalSelectionSystem().foreachSelected(exporter);\r
143   }\r
144 \r
145   const char* init(void* hApp, void* pMainWidget)\r
146   {\r
147     g_mainwnd = (GtkWindow*)pMainWidget;\r
148     ASSERT_NOTNULL(g_mainwnd);\r
149     return "";\r
150   }\r
151   const char* getName()\r
152   {\r
153     return "Brush export Plugin";\r
154   }\r
155   const char* getCommandList()\r
156   {\r
157     return "Export selected as Wavefront Object;About";\r
158   }\r
159   const char* getCommandTitleList()\r
160   {\r
161     return "";\r
162   }\r
163   \r
164   void dispatch(const char* command, float* vMin, float* vMax, bool bSingleBrush)\r
165   {\r
166     if(string_equal(command, "About"))\r
167     {\r
168       GlobalRadiant().m_pfnMessageBox(GTK_WIDGET(g_mainwnd), "Brushexport plugin v 1.0 by namespace (www.codecreator.net)\n"\r
169                                         "Enjoy!\n\nSend feedback to spam@codecreator.net", "About me...",\r
170                                         eMB_OK,\r
171                                         eMB_ICONDEFAULT);\r
172     }\r
173     else if(string_equal(command, "Export selected as Wavefront Object"))\r
174     {\r
175       if(const char* path = GlobalRadiant().m_pfnFileDialog(GTK_WIDGET(g_mainwnd), false, "Save as Obj", 0, 0))\r
176       {\r
177         TextFileOutputStream file(path); \r
178         if(file.failed())\r
179         {\r
180           GlobalRadiant().m_pfnMessageBox(GTK_WIDGET(g_mainwnd), "Unable to write to file", "Error",\r
181                                         eMB_OK,\r
182                                         eMB_ICONERROR);\r
183         }\r
184         else\r
185         {\r
186           export_selected<CExportFormatWavefront>(file);\r
187         }\r
188       }\r
189     }\r
190   }\r
191   \r
192 } // namespace\r
193 \r
194 class BrushExportDependencies :\r
195   public GlobalRadiantModuleRef,\r
196   public GlobalFiletypesModuleRef,\r
197   public GlobalBrushModuleRef,\r
198   public GlobalFileSystemModuleRef,\r
199   public GlobalSceneGraphModuleRef,\r
200   public GlobalSelectionModuleRef\r
201 {\r
202 public:\r
203   BrushExportDependencies(void)\r
204     : GlobalBrushModuleRef(GlobalRadiant().getRequiredGameDescriptionKeyValue("brushtypes"))\r
205   {}\r
206 };\r
207 \r
208 class BrushExportModule : public TypeSystemRef\r
209 {\r
210   _QERPluginTable m_plugin;\r
211 public:\r
212   typedef _QERPluginTable Type;\r
213   STRING_CONSTANT(Name, "brushexport");\r
214 \r
215   BrushExportModule()\r
216   {\r
217     m_plugin.m_pfnQERPlug_Init = &BrushExport::init;\r
218     m_plugin.m_pfnQERPlug_GetName = &BrushExport::getName;\r
219     m_plugin.m_pfnQERPlug_GetCommandList = &BrushExport::getCommandList;\r
220     m_plugin.m_pfnQERPlug_GetCommandTitleList = &BrushExport::getCommandTitleList;\r
221     m_plugin.m_pfnQERPlug_Dispatch = &BrushExport::dispatch;\r
222   }\r
223   _QERPluginTable* getTable()\r
224   {\r
225     return &m_plugin;\r
226   }\r
227 };\r
228 \r
229 typedef SingletonModule<BrushExportModule, BrushExportDependencies> SingletonBrushExportModule;\r
230 SingletonBrushExportModule g_BrushExportModule;\r
231 \r
232 extern "C" void RADIANT_DLLEXPORT Radiant_RegisterModules(ModuleServer& server)\r
233 {\r
234   initialiseModule(server);\r
235   g_BrushExportModule.selfRegister();\r
236 }\r