433 lines
10 KiB
C++
433 lines
10 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//
|
|
//=============================================================================//
|
|
|
|
|
|
//
|
|
// studiomdl.c: generates a studio .mdl file from a .qc script
|
|
// models/<scriptname>.mdl.
|
|
//
|
|
|
|
|
|
#pragma warning( disable : 4244 )
|
|
#pragma warning( disable : 4237 )
|
|
#pragma warning( disable : 4305 )
|
|
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/stat.h>
|
|
#include <math.h>
|
|
#include "tier1/utlbuffer.h"
|
|
#include "cmdlib.h"
|
|
#include "scriplib.h"
|
|
#include "mathlib/mathlib.h"
|
|
#include "studio.h"
|
|
#include "tier1/characterset.h"
|
|
#include "studiomdl.h"
|
|
//#include "..\..\dlls\activity.h"
|
|
|
|
bool IsEnd( char const* pLine );
|
|
int SortAndBalanceBones( int iCount, int iMaxCount, int bones[], float weights[] );
|
|
|
|
int AddToVlist( int v, int m, int n, int t, int firstref );
|
|
void DecrementReferenceVlist( int uv, int numverts );
|
|
int faceCompare( const void *elem1, const void *elem2 );
|
|
|
|
|
|
void UnifyIndices( s_source_t *psource );
|
|
|
|
struct MtlInfo_t
|
|
{
|
|
CUtlString m_MtlName;
|
|
CUtlString m_TgaName;
|
|
};
|
|
|
|
static CUtlVector<MtlInfo_t> g_MtlLib;
|
|
|
|
void ParseMtlLib( CUtlBuffer &buf )
|
|
{
|
|
int nCurrentMtl = -1;
|
|
while ( buf.IsValid() )
|
|
{
|
|
buf.GetLine( g_szLine, sizeof(g_szLine) );
|
|
|
|
if ( !Q_strnicmp( g_szLine, "newmtl ", 7 ) )
|
|
{
|
|
char mtlName[1024];
|
|
if ( sscanf( g_szLine, "newmtl %s", mtlName ) == 1 )
|
|
{
|
|
nCurrentMtl = g_MtlLib.AddToTail( );
|
|
g_MtlLib[nCurrentMtl].m_MtlName = mtlName;
|
|
g_MtlLib[nCurrentMtl].m_TgaName = "debugempty";
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if ( !Q_strnicmp( g_szLine, "map_Kd ", 7 ) )
|
|
{
|
|
if ( nCurrentMtl < 0 )
|
|
continue;
|
|
|
|
char tgaPath[MAX_PATH];
|
|
char tgaName[1024];
|
|
if ( sscanf( g_szLine, "map_Kd %s", tgaPath ) == 1 )
|
|
{
|
|
Q_FileBase( tgaPath, tgaName, sizeof(tgaName) );
|
|
g_MtlLib[nCurrentMtl].m_TgaName = tgaName;
|
|
}
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
const char *FindMtlEntry( const char *pTgaName )
|
|
{
|
|
int nCount = g_MtlLib.Count();
|
|
for ( int i = 0; i < nCount; ++i )
|
|
{
|
|
if ( !Q_stricmp( g_MtlLib[i].m_MtlName, pTgaName ) )
|
|
return g_MtlLib[i].m_TgaName;
|
|
}
|
|
return pTgaName;
|
|
}
|
|
|
|
static bool ParseVertex( CUtlBuffer& bufParse, characterset_t &breakSet, int &v, int &t, int &n )
|
|
{
|
|
char cmd[1024];
|
|
int nLen = bufParse.ParseToken( &breakSet, cmd, sizeof(cmd), false );
|
|
if ( nLen <= 0 )
|
|
return false;
|
|
|
|
v = atoi( cmd );
|
|
n = 0;
|
|
t = 0;
|
|
|
|
char c = *(char*)bufParse.PeekGet();
|
|
bool bHasTexCoord = IN_CHARACTERSET( breakSet, c ) != 0;
|
|
bool bHasNormal = false;
|
|
if ( bHasTexCoord )
|
|
{
|
|
// Snag the '/'
|
|
nLen = bufParse.ParseToken( &breakSet, cmd, sizeof(cmd), false );
|
|
Assert( nLen == 1 );
|
|
|
|
c = *(char*)bufParse.PeekGet();
|
|
if ( !IN_CHARACTERSET( breakSet, c ) )
|
|
{
|
|
nLen = bufParse.ParseToken( &breakSet, cmd, sizeof(cmd), false );
|
|
Assert( nLen > 0 );
|
|
t = atoi( cmd );
|
|
|
|
c = *(char*)bufParse.PeekGet();
|
|
bHasNormal = IN_CHARACTERSET( breakSet, c ) != 0;
|
|
}
|
|
else
|
|
{
|
|
bHasNormal = true;
|
|
bHasTexCoord = false;
|
|
}
|
|
|
|
if ( bHasNormal )
|
|
{
|
|
// Snag the '/'
|
|
nLen = bufParse.ParseToken( &breakSet, cmd, sizeof(cmd), false );
|
|
Assert( nLen == 1 );
|
|
|
|
nLen = bufParse.ParseToken( &breakSet, cmd, sizeof(cmd), false );
|
|
Assert( nLen > 0 );
|
|
n = atoi( cmd );
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
int Load_OBJ( s_source_t *psource )
|
|
{
|
|
char cmd[1024];
|
|
int i;
|
|
int material = -1;
|
|
|
|
g_MtlLib.RemoveAll();
|
|
|
|
if ( !OpenGlobalFile( psource->filename ) )
|
|
return 0;
|
|
|
|
char pFullPath[MAX_PATH];
|
|
if ( !GetGlobalFilePath( psource->filename, pFullPath, sizeof(pFullPath) ) )
|
|
return 0;
|
|
|
|
char pFullDir[MAX_PATH];
|
|
Q_ExtractFilePath( pFullPath, pFullDir, sizeof(pFullDir) );
|
|
|
|
if( !g_quiet )
|
|
{
|
|
printf( "grabbing %s\n", psource->filename );
|
|
}
|
|
|
|
g_iLinecount = 0;
|
|
|
|
psource->numbones = 1;
|
|
strcpy( psource->localBone[0].name, "default" );
|
|
psource->localBone[0].parent = -1;
|
|
Assert( psource->m_Animations.Count() == 0 );
|
|
s_sourceanim_t *pSourceAnim = FindOrAddSourceAnim( psource, "BindPose" );
|
|
pSourceAnim->numframes = 1;
|
|
pSourceAnim->startframe = 0;
|
|
pSourceAnim->endframe = 0;
|
|
pSourceAnim->rawanim[0] = (s_bone_t *)kalloc( 1, sizeof( s_bone_t ) );
|
|
pSourceAnim->rawanim[0][0].pos.Init();
|
|
pSourceAnim->rawanim[0][0].rot.Init();
|
|
Build_Reference( psource, "BindPose" );
|
|
|
|
characterset_t breakSet;
|
|
CharacterSetBuild( &breakSet, "/\\" );
|
|
|
|
while ( GetLineInput() )
|
|
{
|
|
Vector tmp;
|
|
|
|
if ( strncmp( g_szLine, "v ", 2 ) == 0 )
|
|
{
|
|
i = g_numverts++;
|
|
|
|
sscanf( g_szLine, "v %f %f %f", &g_vertex[i].x, &g_vertex[i].y, &g_vertex[i].z );
|
|
g_bone[i].numbones = 1;
|
|
g_bone[i].bone[0] = 0;
|
|
g_bone[i].weight[0] = 1.0;
|
|
continue;
|
|
}
|
|
|
|
if (strncmp( g_szLine, "vn ", 3 ) == 0)
|
|
{
|
|
i = g_numnormals++;
|
|
sscanf( g_szLine, "vn %f %f %f", &g_normal[i].x, &g_normal[i].y, &g_normal[i].z );
|
|
continue;
|
|
}
|
|
|
|
if (strncmp( g_szLine, "vt ", 3 ) == 0)
|
|
{
|
|
i = g_numtexcoords++;
|
|
sscanf( g_szLine, "vt %f %f", &g_texcoord[i].x, &g_texcoord[i].y );
|
|
g_texcoord[i].y = 1.0 - g_texcoord[i].y;
|
|
continue;
|
|
}
|
|
|
|
if ( !Q_strncmp( g_szLine, "mtllib ", 7 ) )
|
|
{
|
|
sscanf( g_szLine, "mtllib %s", &cmd[0] );
|
|
CUtlBuffer buf( 0, 0, CUtlBuffer::TEXT_BUFFER );
|
|
|
|
char pFullMtlLibPath[MAX_PATH];
|
|
Q_ComposeFileName( pFullDir, cmd, pFullMtlLibPath, sizeof(pFullMtlLibPath) );
|
|
if ( g_pFullFileSystem->ReadFile( pFullMtlLibPath, NULL, buf ) )
|
|
{
|
|
ParseMtlLib( buf );
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (strncmp( g_szLine, "usemtl ", 7 ) == 0)
|
|
{
|
|
sscanf( g_szLine, "usemtl %s", &cmd[0] );
|
|
|
|
const char *pTexture = FindMtlEntry( cmd );
|
|
int texture = LookupTexture( pTexture );
|
|
psource->texmap[texture] = texture; // hack, make it 1:1
|
|
material = UseTextureAsMaterial( texture );
|
|
continue;
|
|
}
|
|
|
|
if (strncmp( g_szLine, "f ", 2 ) == 0)
|
|
{
|
|
if ( material < 0 )
|
|
{
|
|
int texture = LookupTexture( "debugempty.tga" );
|
|
psource->texmap[texture] = texture;
|
|
material = UseTextureAsMaterial( texture );
|
|
}
|
|
|
|
int v0, n0, t0;
|
|
int v1, n1, t1;
|
|
int v2, n2, t2;
|
|
|
|
s_tmpface_t f;
|
|
|
|
// Are we specifying p only, p and t only, p and n only, or p and n and t?
|
|
char *pData = g_szLine + 2;
|
|
int nLen = Q_strlen( pData );
|
|
|
|
CUtlBuffer bufParse( pData, nLen, CUtlBuffer::TEXT_BUFFER | CUtlBuffer::READ_ONLY );
|
|
|
|
ParseVertex( bufParse, breakSet, v0, t0, n0 );
|
|
ParseVertex( bufParse, breakSet, v1, t1, n1 );
|
|
Assert( v0 <= g_numverts && t0 <= g_numtexcoords && n0 <= g_numnormals );
|
|
Assert( v1 <= g_numverts && t1 <= g_numtexcoords && n1 <= g_numnormals );
|
|
while ( bufParse.IsValid() )
|
|
{
|
|
if ( !ParseVertex( bufParse, breakSet, v2, t2, n2 ) )
|
|
break;
|
|
|
|
Assert( v2 <= g_numverts && t2 <= g_numtexcoords && n2 <= g_numnormals );
|
|
|
|
i = g_numfaces++;
|
|
f.material = material;
|
|
f.a = v0 - 1; f.na = (n0 > 0) ? n0 - 1 : 0, f.ta = (t0 > 0) ? t0 - 1 : 0;
|
|
f.b = v2 - 1; f.nb = (n2 > 0) ? n2 - 1 : 0, f.tb = (t2 > 0) ? t2 - 1 : 0;
|
|
f.c = v1 - 1; f.nc = (n1 > 0) ? n1 - 1 : 0, f.tc = (t1 > 0) ? t1 - 1 : 0;
|
|
g_face[i] = f;
|
|
|
|
v1 = v2; t1 = t2; n1 = n2;
|
|
}
|
|
continue;
|
|
}
|
|
}
|
|
|
|
UnifyIndices( psource );
|
|
|
|
BuildIndividualMeshes( psource );
|
|
|
|
fclose( g_fpInput );
|
|
|
|
return 1;
|
|
}
|
|
|
|
|
|
|
|
int AppendVTAtoOBJ( s_source_t *psource, char *filename, int frame )
|
|
{
|
|
char cmd[1024];
|
|
int i, j;
|
|
int material = 0;
|
|
|
|
Vector tmp;
|
|
matrix3x4_t m;
|
|
|
|
AngleMatrix( RadianEuler( 1.570796, 0, 0 ), m );
|
|
|
|
if ( !OpenGlobalFile( filename ) )
|
|
return 0;
|
|
|
|
if( !g_quiet )
|
|
{
|
|
printf ("grabbing %s\n", filename );
|
|
}
|
|
|
|
g_iLinecount = 0;
|
|
|
|
g_numverts = g_numnormals = g_numtexcoords = g_numfaces = 0;
|
|
|
|
while ( GetLineInput() )
|
|
{
|
|
Vector tmp;
|
|
|
|
if (strncmp( g_szLine, "v ", 2 ) == 0)
|
|
{
|
|
i = g_numverts++;
|
|
|
|
sscanf( g_szLine, "v %f %f %f", &tmp.x, &tmp.y, &tmp.z );
|
|
VectorTransform( tmp, m, g_vertex[i] );
|
|
|
|
// printf("%f %f %f\n", g_vertex[i].x, g_vertex[i].y, g_vertex[i].z );
|
|
|
|
g_bone[i].numbones = 1;
|
|
g_bone[i].bone[0] = 0;
|
|
g_bone[i].weight[0] = 1.0;
|
|
}
|
|
else if (strncmp( g_szLine, "vn ", 3 ) == 0)
|
|
{
|
|
i = g_numnormals++;
|
|
sscanf( g_szLine, "vn %f %f %f", &tmp.x, &tmp.y, &tmp.z );
|
|
VectorRotate( tmp, m, g_normal[i] );
|
|
}
|
|
else if (strncmp( g_szLine, "vt ", 3 ) == 0)
|
|
{
|
|
i = g_numtexcoords++;
|
|
sscanf( g_szLine, "vt %f %f", &g_texcoord[i].x, &g_texcoord[i].y );
|
|
}
|
|
else if (strncmp( g_szLine, "usemtl ", 7 ) == 0)
|
|
{
|
|
sscanf( g_szLine, "usemtl %s", &cmd[0] );
|
|
|
|
int texture = LookupTexture( cmd );
|
|
psource->texmap[texture] = texture; // hack, make it 1:1
|
|
material = UseTextureAsMaterial( texture );
|
|
}
|
|
else if (strncmp( g_szLine, "f ", 2 ) == 0)
|
|
{
|
|
int v0, n0, t0;
|
|
int v1, n1, t1;
|
|
int v2, n2, t2;
|
|
int v3, n3, t3;
|
|
|
|
s_tmpface_t f;
|
|
|
|
i = g_numfaces++;
|
|
|
|
j = sscanf( g_szLine, "f %d/%d/%d %d/%d/%d %d/%d/%d %d/%d/%d", &v0, &t0, &n0, &v1, &t1, &n1, &v2, &t2, &n2, &v3, &t3, &n3 );
|
|
|
|
f.material = material;
|
|
f.a = v0 - 1; f.na = n0 - 1, f.ta = 0;
|
|
f.b = v2 - 1; f.nb = n2 - 1, f.tb = 0;
|
|
f.c = v1 - 1; f.nc = n1 - 1, f.tc = 0;
|
|
|
|
Assert( v0 <= g_numverts && v1 <= g_numverts && v2 <= g_numverts );
|
|
Assert( n0 <= g_numnormals && n1 <= g_numnormals && n2 <= g_numnormals );
|
|
|
|
g_face[i] = f;
|
|
|
|
if (j == 12)
|
|
{
|
|
i = g_numfaces++;
|
|
f.a = v0 - 1; f.na = n0 - 1, f.ta = 0;
|
|
f.b = v3 - 1; f.nb = n3 - 1, f.tb = 0;
|
|
f.c = v2 - 1; f.nc = n2 - 1, f.tc = 0;
|
|
g_face[i] = f;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
UnifyIndices( psource );
|
|
|
|
s_sourceanim_t *pSourceAnim = FindOrAddSourceAnim( psource, "BindPose" );
|
|
if ( frame == 0 )
|
|
{
|
|
psource->numbones = 1;
|
|
strcpy( psource->localBone[0].name, "default" );
|
|
psource->localBone[0].parent = -1;
|
|
pSourceAnim->numframes = 1;
|
|
pSourceAnim->startframe = 0;
|
|
pSourceAnim->endframe = 0;
|
|
pSourceAnim->rawanim[0] = (s_bone_t *)kalloc( 1, sizeof( s_bone_t ) );
|
|
pSourceAnim->rawanim[0][0].pos.Init();
|
|
pSourceAnim->rawanim[0][0].rot = RadianEuler( 1.570796, 0.0, 0.0 );
|
|
Build_Reference( psource, "BindPose" );
|
|
|
|
BuildIndividualMeshes( psource );
|
|
}
|
|
|
|
// printf("%d %d : %d\n", g_numverts, g_numnormals, numvlist );
|
|
|
|
int t = frame;
|
|
int count = numvlist;
|
|
|
|
pSourceAnim->numvanims[t] = count;
|
|
pSourceAnim->vanim[t] = (s_vertanim_t *)kalloc( count, sizeof( s_vertanim_t ) );
|
|
for (i = 0; i < count; i++)
|
|
{
|
|
pSourceAnim->vanim[t][i].vertex = i;
|
|
pSourceAnim->vanim[t][i].pos = g_vertex[v_listdata[i].v];
|
|
pSourceAnim->vanim[t][i].normal = g_normal[v_listdata[i].n];
|
|
}
|
|
|
|
fclose( g_fpInput );
|
|
|
|
return 1;
|
|
}
|