Merge remote-tracking branch 'origin/divVerent/weird-shift-a'
[divverent/netradiant.git] / tools / quake3 / q3map2 / convert_obj.c
1 /* -------------------------------------------------------------------------------
2
3 Copyright (C) 1999-2007 id Software, Inc. and contributors.
4 For a list of contributors, see the accompanying CONTRIBUTORS file.
5
6 This file is part of GtkRadiant.
7
8 GtkRadiant is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 2 of the License, or
11 (at your option) any later version.
12
13 GtkRadiant is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with GtkRadiant; if not, write to the Free Software
20 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
21
22 ----------------------------------------------------------------------------------
23
24 This code has been altered significantly from its original form, to support
25 several games based on the Quake III Arena engine, in the form of "Q3Map2."
26
27 ------------------------------------------------------------------------------- */
28
29
30
31 /* marker */
32 #define CONVERT_ASE_C
33
34
35
36 /* dependencies */
37 #include "q3map2.h"
38
39
40
41 /*
42 ConvertSurface()
43 converts a bsp drawsurface to an obj chunk
44 */
45
46 int firstLightmap = 0;
47 int lastLightmap = -1;
48 static void ConvertLightmapToMTL( FILE *f, const char *base, int lightmapNum );
49
50 int objVertexCount = 0;
51 int objLastShaderNum = -1;
52 static void ConvertSurfaceToOBJ( FILE *f, bspModel_t *model, int modelNum, bspDrawSurface_t *ds, int surfaceNum, vec3_t origin )
53 {
54         int                             i, v, face, a, b, c;
55         bspDrawVert_t   *dv;
56         
57         /* ignore patches for now */
58         if( ds->surfaceType != MST_PLANAR && ds->surfaceType != MST_TRIANGLE_SOUP )
59                 return;
60
61         fprintf(f, "g mat%dmodel%dsurf%d\r\n", ds->shaderNum, modelNum, surfaceNum);
62         switch( ds->surfaceType )
63         {
64                 case MST_PLANAR:
65                         fprintf( f, "# SURFACETYPE MST_PLANAR\r\n" );
66                         break;
67                 case MST_TRIANGLE_SOUP:
68                         fprintf( f, "# SURFACETYPE MST_TRIANGLE_SOUP\r\n" );
69                         break;
70         }
71
72         /* export shader */
73         if(lightmapsAsTexcoord)
74         {
75                 if(objLastShaderNum != ds->lightmapNum[0])
76                 {
77                         fprintf(f, "usemtl lm_%04d\r\n", ds->lightmapNum[0] + deluxemap);
78                         objLastShaderNum = ds->lightmapNum[0] + deluxemap;
79                 }
80                 if(ds->lightmapNum[0] + deluxemap < firstLightmap)
81                 {
82                         Sys_Printf( "WARNING: lightmap %d out of range (exporting anyway)\n", ds->lightmapNum[0] + deluxemap );
83                         firstLightmap = ds->lightmapNum[0] + deluxemap;
84                 }
85                 if(ds->lightmapNum[0] > lastLightmap)
86                 {
87                         Sys_Printf( "WARNING: lightmap %d out of range (exporting anyway)\n", ds->lightmapNum[0] + deluxemap );
88                         lastLightmap = ds->lightmapNum[0] + deluxemap;
89                 }
90         }
91         else
92         {
93                 if(objLastShaderNum != ds->shaderNum)
94                 {
95                         fprintf(f, "usemtl %s\r\n", bspShaders[ds->shaderNum].shader);
96                         objLastShaderNum = ds->shaderNum;
97                 }
98         }
99         
100         /* export vertex */
101         for( i = 0; i < ds->numVerts; i++ )
102         {
103                 v = i + ds->firstVert;
104                 dv = &bspDrawVerts[ v ];
105                 fprintf(f, "# vertex %d\r\n", i + objVertexCount + 1);
106                 fprintf(f, "v %f %f %f\r\n", dv->xyz[ 0 ], dv->xyz[ 1 ], dv->xyz[ 2 ]);
107                 fprintf(f, "vn %f %f %f\r\n", dv->normal[ 0 ], dv->normal[ 1 ], dv->normal[ 2 ]);
108                 if(lightmapsAsTexcoord)
109                         fprintf(f, "vt %f %f\r\n", dv->lightmap[0][0], 1.0 - dv->lightmap[0][1]);
110                 else
111                         fprintf(f, "vt %f %f\r\n", dv->st[ 0 ], 1.0 - dv->st[ 1 ]);
112         }
113
114         /* export faces */
115         for( i = 0; i < ds->numIndexes; i += 3 )
116         {
117                 face = (i / 3);
118                 a = bspDrawIndexes[ i + ds->firstIndex ];
119                 c = bspDrawIndexes[ i + ds->firstIndex + 1 ];
120                 b = bspDrawIndexes[ i + ds->firstIndex + 2 ];
121                 fprintf(f, "f %d/%d/%d %d/%d/%d %d/%d/%d\r\n",
122                         a + objVertexCount + 1, a + objVertexCount + 1, a + objVertexCount + 1, 
123                         b + objVertexCount + 1, b + objVertexCount + 1, b + objVertexCount + 1, 
124                         c + objVertexCount + 1, c + objVertexCount + 1, c + objVertexCount + 1
125                 );
126         }
127
128         objVertexCount += ds->numVerts;
129 }
130
131
132
133 /*
134 ConvertModel()
135 exports a bsp model to an ase chunk
136 */
137
138 static void ConvertModelToOBJ( FILE *f, bspModel_t *model, int modelNum, vec3_t origin )
139 {
140         int                                     i, s;
141         bspDrawSurface_t        *ds;
142         
143         
144         /* go through each drawsurf in the model */
145         for( i = 0; i < model->numBSPSurfaces; i++ )
146         {
147                 s = i + model->firstBSPSurface;
148                 ds = &bspDrawSurfaces[ s ];
149                 ConvertSurfaceToOBJ( f, model, modelNum, ds, s, origin );
150         }
151 }
152
153
154
155 /*
156 ConvertShader()
157 exports a bsp shader to an ase chunk
158 */
159
160 static void ConvertShaderToMTL( FILE *f, bspShader_t *shader, int shaderNum )
161 {
162         shaderInfo_t    *si;
163         char                    *c, filename[ 1024 ];
164         
165         
166         /* get shader */
167         si = ShaderInfoForShader( shader->shader );
168         if( si == NULL )
169         {
170                 Sys_Printf( "WARNING: NULL shader in BSP\n" );
171                 return;
172         }
173         
174         /* set bitmap filename */
175         if( si->shaderImage->filename[ 0 ] != '*' )
176                 strcpy( filename, si->shaderImage->filename );
177         else
178                 sprintf( filename, "%s.tga", si->shader );
179
180         /* blender hates this, so let's not do it
181         for( c = filename; *c != '\0'; c++ )
182                 if( *c == '/' )
183                         *c = '\\';
184         */
185         
186         /* print shader info */
187         fprintf( f, "newmtl %s\r\n", shader->shader );
188         fprintf( f, "Kd %f %f %f\r\n", si->color[ 0 ], si->color[ 1 ], si->color[ 2 ] );
189         if(shadersAsBitmap)
190                 fprintf( f, "map_Kd %s\r\n", shader->shader );
191         else
192         /* blender hates this, so let's not do it
193                 fprintf( f, "map_Kd ..\\%s\r\n", filename );
194         */
195                 fprintf( f, "map_Kd ../%s\r\n", filename );
196 }
197
198 static void ConvertLightmapToMTL( FILE *f, const char *base, int lightmapNum )
199 {
200         /* print shader info */
201         fprintf( f, "newmtl lm_%04d\r\n", lightmapNum );
202         if(lightmapNum >= 0)
203         /* blender hates this, so let's not do it
204                 fprintf( f, "map_Kd %s\\lm_%04d.tga\r\n", base, lightmapNum );
205         */
206                 fprintf( f, "map_Kd %s/lm_%04d.tga\r\n", base, lightmapNum );
207 }
208
209
210
211 /*
212 ConvertBSPToASE()
213 exports an 3d studio ase file from the bsp
214 */
215
216 int ConvertBSPToOBJ( char *bspName )
217 {
218         int                             i, modelNum;
219         FILE                    *f, *fmtl;
220         bspShader_t             *shader;
221         bspModel_t              *model;
222         entity_t                *e;
223         vec3_t                  origin;
224         const char              *key;
225         char                    name[ 1024 ], base[ 1024 ], mtlname[ 1024 ], dirname[ 1024 ];
226         
227         
228         /* note it */
229         Sys_Printf( "--- Convert BSP to OBJ ---\n" );
230
231         /* create the ase filename from the bsp name */
232         strcpy( dirname, bspName );
233         StripExtension( dirname );
234         strcpy( name, bspName );
235         StripExtension( name );
236         strcat( name, ".obj" );
237         Sys_Printf( "writing %s\n", name );
238         strcpy( mtlname, bspName );
239         StripExtension( mtlname );
240         strcat( mtlname, ".mtl" );
241         Sys_Printf( "writing %s\n", mtlname );
242         
243         ExtractFileBase( bspName, base );
244         
245         /* open it */
246         f = fopen( name, "wb" );
247         if( f == NULL )
248                 Error( "Open failed on %s\n", name );
249         fmtl = fopen( mtlname, "wb" );
250         if( fmtl == NULL )
251                 Error( "Open failed on %s\n", mtlname );
252         
253         /* print header */
254         fprintf( f, "o %s\r\n", base );
255         fprintf( f, "# Generated by Q3Map2 (ydnar) -convert -format obj\r\n" );
256         fprintf( f, "mtllib %s.mtl\r\n", base );
257
258         fprintf( fmtl, "# Generated by Q3Map2 (ydnar) -convert -format obj\r\n" );
259         if(lightmapsAsTexcoord)
260         {
261                 int lightmapCount;
262                 for( lightmapCount = 0; lightmapCount < numBSPLightmaps; lightmapCount++ )
263                         ;
264                 for( ; ; lightmapCount++ )
265                 {
266                         char buf[1024];
267                         FILE *tmp;
268                         snprintf(buf, sizeof(buf), "%s/lm_%04d.tga", dirname, lightmapCount);
269                         buf[sizeof(buf) - 1] = 0;
270                         tmp = fopen(buf, "rb");
271                         if(!tmp)
272                                 break;
273                         fclose(tmp);
274                 }
275                 lastLightmap = lightmapCount - 1;
276         }
277         else
278         {
279                 for( i = 0; i < numBSPShaders; i++ )
280                 {
281                         shader = &bspShaders[ i ];
282                         ConvertShaderToMTL( fmtl, shader, i );
283                 }
284         }
285         
286         /* walk entity list */
287         for( i = 0; i < numEntities; i++ )
288         {
289                 /* get entity and model */
290                 e = &entities[ i ];
291                 if( i == 0 )
292                         modelNum = 0;
293                 else
294                 {
295                         key = ValueForKey( e, "model" );
296                         if( key[ 0 ] != '*' )
297                                 continue;
298                         modelNum = atoi( key + 1 );
299                 }
300                 model = &bspModels[ modelNum ];
301                 
302                 /* get entity origin */
303                 key = ValueForKey( e, "origin" );
304                 if( key[ 0 ] == '\0' )
305                         VectorClear( origin );
306                 else
307                         GetVectorForKey( e, "origin", origin );
308                 
309                 /* convert model */
310                 ConvertModelToOBJ( f, model, modelNum, origin );
311         }
312         
313         if(lightmapsAsTexcoord)
314         {
315                 for( i = firstLightmap; i <= lastLightmap; i++ )
316                         ConvertLightmapToMTL( fmtl, base, i );
317         }
318
319         /* close the file and return */
320         fclose( f );
321         fclose( fmtl );
322         
323         /* return to sender */
324         return 0;
325 }
326
327
328