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