worldspawn/plugins/iqmmodel/iqm.cpp
2020-11-17 12:16:16 +01:00

313 lines
8.8 KiB
C++

/*
Copyright (C) 2001-2006, William Joseph.
Copyright (C) 2010-2014 COR Entertainment, LLC.
All Rights Reserved.
This file is part of GtkRadiant.
GtkRadiant is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
GtkRadiant is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with GtkRadiant; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "iqm.h"
#include "ifilesystem.h"
#include "imodel.h"
#include "imagelib.h"
#include "bytestreamutils.h"
#include "../md3model/model.h"
typedef unsigned char byte;
/*
========================================================================
.IQM triangle model file format
========================================================================
*/
typedef struct {
float s;
float t;
} iqmSt_t;
void istream_read_iqmSt( PointerInputStream &inputStream, iqmSt_t &st ){
st.s = istream_read_float32_le( inputStream );
st.t = istream_read_float32_le( inputStream );
}
typedef struct {
unsigned int indices[3];
} iqmTriangle_t;
void istream_read_iqmTriangle( PointerInputStream &inputStream, iqmTriangle_t &triangle ){
triangle.indices[0] = istream_read_int32_le( inputStream );
triangle.indices[1] = istream_read_int32_le( inputStream );
triangle.indices[2] = istream_read_int32_le( inputStream );
}
typedef struct {
float v[3];
} iqmPos_t;
void istream_read_iqmPos( PointerInputStream &inputStream, iqmPos_t &iqmPos ){
iqmPos.v[0] = istream_read_float32_le( inputStream );
iqmPos.v[1] = istream_read_float32_le( inputStream );
iqmPos.v[2] = istream_read_float32_le( inputStream );
}
const int IQM_POSITION = 0;
const int IQM_TEXCOORD = 1;
const int IQM_NORMAL = 2;
const int IQM_TANGENT = 3;
const int IQM_BLENDINDEXES = 4;
const int IQM_BLENDWEIGHTS = 5;
const int IQM_COLOR = 6;
const int IQM_CUSTOM = 0x10;
const int IQM_BYTE = 0;
const int IQM_UBYTE = 1;
const int IQM_SHORT = 2;
const int IQM_USHORT = 3;
const int IQM_INT = 4;
const int IQM_UINT = 5;
const int IQM_HALF = 6;
const int IQM_FLOAT = 7;
const int IQM_DOUBLE = 8;
// animflags
const int IQM_LOOP = 1;
typedef struct iqmHeader_s {
byte id[16];
unsigned int version;
unsigned int filesize;
unsigned int flags;
unsigned int num_text, ofs_text;
unsigned int num_meshes, ofs_meshes;
unsigned int num_vertexarrays, num_vertexes, ofs_vertexarrays;
unsigned int num_triangles, ofs_triangles, ofs_neighbors;
unsigned int num_joints, ofs_joints;
unsigned int num_poses, ofs_poses;
unsigned int num_anims, ofs_anims;
unsigned int num_frames, num_framechannels, ofs_frames, ofs_bounds;
unsigned int num_comment, ofs_comment;
unsigned int num_extensions, ofs_extensions;
} iqmHeader_t;
void istream_read_iqmHeader( PointerInputStream &inputStream, iqmHeader_t &header ){
inputStream.read( header.id, 16 );
#define READINT( x ) header.x = istream_read_int32_le( inputStream );
READINT( version )
READINT( filesize )
READINT( flags )
READINT( num_text )
READINT( ofs_text )
READINT( num_meshes )
READINT( ofs_meshes )
READINT( num_vertexarrays )
READINT( num_vertexes )
READINT( ofs_vertexarrays )
READINT( num_triangles )
READINT( ofs_triangles )
READINT( ofs_neighbors )
READINT( num_joints )
READINT( ofs_joints )
READINT( num_frames )
READINT( num_framechannels )
READINT( ofs_frames )
READINT( ofs_bounds )
READINT( num_comment )
READINT( ofs_comment )
READINT( num_extensions )
READINT( ofs_extensions )
#undef READINT
}
typedef struct iqmmesh_s {
unsigned int name;
unsigned int material;
unsigned int first_vertex;
unsigned int num_vertexes;
unsigned int first_triangle;
unsigned int num_triangles;
} iqmmesh_t;
void istream_read_iqmMesh( PointerInputStream &inputStream, iqmmesh_t &iqmmesh ){
#define READUINT( x ) iqmmesh.x = istream_read_uint32_le( inputStream );
READUINT( name )
READUINT( material )
READUINT( first_vertex )
READUINT( num_vertexes )
READUINT( first_triangle )
READUINT( num_triangles )
#undef READUINT
}
typedef struct iqmvertexarray_s {
unsigned int type;
unsigned int flags;
unsigned int format;
unsigned int size;
unsigned int offset;
} iqmvertexarray_t;
void istream_read_iqmVertexarray( PointerInputStream &inputStream, iqmvertexarray_t &vertexarray ){
#define READINT( x ) vertexarray.x = istream_read_int32_le( inputStream );
READINT( type )
READINT( flags )
READINT( format )
READINT( size )
READINT( offset )
#undef READINT
}
ArbitraryMeshVertex IQMVertex_construct( const iqmPos_t *pos, const iqmPos_t *norm, const iqmSt_t *st ){
return ArbitraryMeshVertex(
Vertex3f( pos->v[0], pos->v[1], pos->v[2] ),
Normal3f( norm->v[0], norm->v[1], norm->v[2] ),
TexCoord2f( st->s, st->t )
);
}
void IQMSurface_read( Model &model, const byte *buffer, ArchiveFile &file ){
iqmHeader_t header;
{
PointerInputStream inputStream( buffer );
istream_read_iqmHeader( inputStream, header );
}
int ofs_position = -1, ofs_st = -1, ofs_normal = -1;
PointerInputStream vaStream( buffer + header.ofs_vertexarrays );
for ( unsigned int i = 0; i < header.num_vertexarrays; i++ ) {
iqmvertexarray_t va;
istream_read_iqmVertexarray( vaStream, va );
switch ( va.type ) {
case IQM_POSITION:
if ( va.format == IQM_FLOAT && va.size == 3 ) {
ofs_position = va.offset;
}
break;
case IQM_TEXCOORD:
if ( va.format == IQM_FLOAT && va.size == 2 ) {
ofs_st = va.offset;
}
break;
case IQM_NORMAL:
if ( va.format == IQM_FLOAT && va.size == 3 ) {
ofs_normal = va.offset;
}
break;
}
}
PointerInputStream posStream( buffer + ofs_position );
Array<iqmPos_t> iqmPos( header.num_vertexes );
for ( Array<iqmPos_t>::iterator i = iqmPos.begin(); i != iqmPos.end(); ++i ) {
istream_read_iqmPos( posStream, *i );
}
PointerInputStream normStream( buffer + ofs_normal );
Array<iqmPos_t> iqmNorm( header.num_vertexes );
for ( Array<iqmPos_t>::iterator i = iqmNorm.begin(); i != iqmNorm.end(); ++i ) {
istream_read_iqmPos( normStream, *i );
}
Array<iqmSt_t> iqmSt( header.num_vertexes );
PointerInputStream stStream( buffer + ofs_st );
for ( Array<iqmSt_t>::iterator i = iqmSt.begin(); i != iqmSt.end(); ++i ) {
istream_read_iqmSt( stStream, *i );
}
PointerInputStream iqmMesh( buffer + header.ofs_meshes );
for ( unsigned int m = 0; m < header.num_meshes; m++ ) {
Surface &surface = model.newSurface();
iqmmesh_t iqmmesh;
istream_read_iqmMesh( iqmMesh, iqmmesh );
bool material_found = false;
// if not malformed data neither missing string
if ( iqmmesh.material <= header.num_text && iqmmesh.material > 0 ) {
char *material;
material = (char*) buffer + header.ofs_text + iqmmesh.material;
if ( material[0] != '\0' ) {
surface.setShader( material );
material_found = true;
}
}
if ( !material_found ) {
// empty string will trigger "textures/shader/notex" on display
surface.setShader( "" );
}
UniqueVertexBuffer<ArbitraryMeshVertex> inserter( surface.vertices() );
inserter.reserve( iqmmesh.num_vertexes );
surface.indices().reserve( iqmmesh.num_vertexes );
unsigned int triangle_offset = header.ofs_triangles + iqmmesh.first_triangle * sizeof( iqmTriangle_t );
PointerInputStream triangleStream( buffer + triangle_offset );
for ( unsigned int i = 0; i < iqmmesh.num_triangles; ++i ) {
iqmTriangle_t triangle;
istream_read_iqmTriangle( triangleStream, triangle );
for ( int j = 0; j < 3; j++ ) {
surface.indices().insert( inserter.insert( IQMVertex_construct(
&iqmPos[triangle.indices[j]],
&iqmNorm[triangle.indices[j]],
&iqmSt[triangle.indices[j]] ) ) );
}
}
surface.updateAABB();
}
}
void IQMModel_read( Model &model, const byte *buffer, ArchiveFile &file ){
IQMSurface_read( model, buffer, file );
model.updateAABB();
}
scene::Node &IQMModel_new( const byte *buffer, ArchiveFile &file ){
ModelNode *modelNode = new ModelNode();
IQMModel_read( modelNode->model(), buffer, file );
return modelNode->node();
}
scene::Node &IQMModel_default(){
ModelNode *modelNode = new ModelNode();
Model_constructNull( modelNode->model() );
return modelNode->node();
}
scene::Node &IQMModel_fromBuffer( unsigned char *buffer, ArchiveFile &file ){
if ( memcmp( buffer, "INTERQUAKEMODEL", 16 ) ) {
globalErrorStream() << "IQM read error: incorrect ident\n";
return IQMModel_default();
}
else {
return IQMModel_new( buffer, file );
}
}
scene::Node &loadIQMModel( ArchiveFile &file ){
ScopedArchiveBuffer buffer( file );
return IQMModel_fromBuffer( buffer.buffer, file );
}