vera/worldspawn
4
0
Fork 0
Code Issues 16 Pull requests Projects Releases Packages Wiki Activity
worldspawn/tools/common/imagelib.c

1534 lines
35 KiB
C
Raw Blame History

/*
Copyright (C) 1999-2007 id Software, Inc. and contributors.
For a list of contributors, see the accompanying CONTRIBUTORS file.
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
*/
// imagelib.c
#include "inout.h"
#include "cmdlib.h"
#include "etclib.h"
#include "imagelib.h"
#include "vfs.h"
int fgetLittleShort( FILE *f ){
byte b1, b2;
b1 = fgetc( f );
b2 = fgetc( f );
return (short)( b1 + b2 * 256 );
}
int fgetLittleLong( FILE *f ){
byte b1, b2, b3, b4;
b1 = fgetc( f );
b2 = fgetc( f );
b3 = fgetc( f );
b4 = fgetc( f );
return b1 + ( b2 << 8 ) + ( b3 << 16 ) + ( b4 << 24 );
}
int bufLittleShort( byte *buf, int len, int *pos ){
byte b1, b2;
if ( ( len - *pos ) < 2 ) {
Error( "Unexpected buffer end" );
}
b1 = buf[*pos]; *pos += 1;
b2 = buf[*pos]; *pos += 1;
return (short)( b1 + b2 * 256 );
}
int bufLittleLong( byte *buf, int len, int *pos ){
byte b1, b2, b3, b4;
if ( ( len - *pos ) < 4 ) {
Error( "Unexpected buffer end" );
}
b1 = buf[*pos]; *pos += 1;
b2 = buf[*pos]; *pos += 1;
b3 = buf[*pos]; *pos += 1;
b4 = buf[*pos]; *pos += 1;
return b1 + ( b2 << 8 ) + ( b3 << 16 ) + ( b4 << 24 );
}
/*
============================================================================
LBM STUFF
============================================================================
*/
typedef unsigned char UBYTE;
//conflicts with windows typedef short WORD;
typedef unsigned short UWORD;
typedef long LONG;
typedef enum
{
ms_none,
ms_mask,
ms_transcolor,
ms_lasso
} mask_t;
typedef enum
{
cm_none,
cm_rle1
} compress_t;
typedef struct
{
UWORD w,h;
short x,y;
UBYTE nPlanes;
UBYTE masking;
UBYTE compression;
UBYTE pad1;
UWORD transparentColor;
UBYTE xAspect,yAspect;
short pageWidth,pageHeight;
} bmhd_t;
extern bmhd_t bmhd; // will be in native byte order
#define FORMID ( 'F' + ( 'O' << 8 ) + ( (int)'R' << 16 ) + ( (int)'M' << 24 ) )
#define ILBMID ( 'I' + ( 'L' << 8 ) + ( (int)'B' << 16 ) + ( (int)'M' << 24 ) )
#define PBMID ( 'P' + ( 'B' << 8 ) + ( (int)'M' << 16 ) + ( (int)' ' << 24 ) )
#define BMHDID ( 'B' + ( 'M' << 8 ) + ( (int)'H' << 16 ) + ( (int)'D' << 24 ) )
#define BODYID ( 'B' + ( 'O' << 8 ) + ( (int)'D' << 16 ) + ( (int)'Y' << 24 ) )
#define CMAPID ( 'C' + ( 'M' << 8 ) + ( (int)'A' << 16 ) + ( (int)'P' << 24 ) )
bmhd_t bmhd;
int Align( int l ){
if ( l & 1 ) {
return l + 1;
}
return l;
}
/*
================
LBMRLEdecompress
Source must be evenly aligned!
================
*/
byte *LBMRLEDecompress( byte *source,byte *unpacked, int bpwidth ){
int count;
byte b,rept;
count = 0;
do
{
rept = *source++;
if ( rept > 0x80 ) {
rept = ( rept ^ 0xff ) + 2;
b = *source++;
memset( unpacked,b,rept );
unpacked += rept;
}
else if ( rept < 0x80 ) {
rept++;
memcpy( unpacked,source,rept );
unpacked += rept;
source += rept;
}
else{
rept = 0; // rept of 0x80 is NOP
}
count += rept;
} while ( count < bpwidth );
if ( count > bpwidth ) {
Error( "Decompression exceeded width!\n" );
}
return source;
}
/*
=================
LoadLBM
=================
*/
void LoadLBM( const char *filename, byte **picture, byte **palette ){
byte *LBMbuffer, *picbuffer, *cmapbuffer;
int y;
byte *LBM_P, *LBMEND_P;
byte *pic_p;
byte *body_p;
int formtype,formlength;
int chunktype,chunklength;
// qiet compiler warnings
picbuffer = NULL;
cmapbuffer = NULL;
//
// load the LBM
//
LoadFile( filename, (void **)&LBMbuffer );
//
// parse the LBM header
//
LBM_P = LBMbuffer;
if ( *(int *)LBMbuffer != LittleLong( FORMID ) ) {
Error( "No FORM ID at start of file!\n" );
}
LBM_P += 4;
formlength = BigLong( *(int *)LBM_P );
LBM_P += 4;
LBMEND_P = LBM_P + Align( formlength );
formtype = LittleLong( *(int *)LBM_P );
if ( formtype != ILBMID && formtype != PBMID ) {
Error( "Unrecognized form type: %c%c%c%c\n", formtype & 0xff
,( formtype >> 8 ) & 0xff,( formtype >> 16 ) & 0xff,( formtype >> 24 ) & 0xff );
}
LBM_P += 4;
//
// parse chunks
//
while ( LBM_P < LBMEND_P )
{
chunktype = LBM_P[0] + ( LBM_P[1] << 8 ) + ( LBM_P[2] << 16 ) + ( LBM_P[3] << 24 );
LBM_P += 4;
chunklength = LBM_P[3] + ( LBM_P[2] << 8 ) + ( LBM_P[1] << 16 ) + ( LBM_P[0] << 24 );
LBM_P += 4;
switch ( chunktype )
{
case BMHDID:
memcpy( &bmhd,LBM_P,sizeof( bmhd ) );
bmhd.w = BigShort( bmhd.w );
bmhd.h = BigShort( bmhd.h );
bmhd.x = BigShort( bmhd.x );
bmhd.y = BigShort( bmhd.y );
bmhd.pageWidth = BigShort( bmhd.pageWidth );
bmhd.pageHeight = BigShort( bmhd.pageHeight );
break;
case CMAPID:
cmapbuffer = safe_malloc( 768 );
memset( cmapbuffer, 0, 768 );
memcpy( cmapbuffer, LBM_P, chunklength );
break;
case BODYID:
body_p = LBM_P;
pic_p = picbuffer = safe_malloc( bmhd.w * bmhd.h );
if ( formtype == PBMID ) {
//
// unpack PBM
//
for ( y = 0; y < bmhd.h; y++, pic_p += bmhd.w )
{
if ( bmhd.compression == cm_rle1 ) {
body_p = LBMRLEDecompress( (byte *)body_p
, pic_p, bmhd.w );
}
else if ( bmhd.compression == cm_none ) {
memcpy( pic_p,body_p,bmhd.w );
body_p += Align( bmhd.w );
}
}
}
else
{
//
// unpack ILBM
//
Error( "%s is an interlaced LBM, not packed", filename );
}
break;
}
LBM_P += Align( chunklength );
}
free( LBMbuffer );
*picture = picbuffer;
if ( palette ) {
*palette = cmapbuffer;
}
}
/*
============================================================================
WRITE LBM
============================================================================
*/
/*
==============
WriteLBMfile
==============
*/
void WriteLBMfile( const char *filename, byte *data,
int width, int height, byte *palette ){
byte *lbm, *lbmptr;
int *formlength, *bmhdlength, *cmaplength, *bodylength;
int length;
bmhd_t basebmhd;
lbm = lbmptr = safe_malloc( width * height + 1000 );
//
// start FORM
//
*lbmptr++ = 'F';
*lbmptr++ = 'O';
*lbmptr++ = 'R';
*lbmptr++ = 'M';
formlength = (int*)lbmptr;
lbmptr += 4; // leave space for length
*lbmptr++ = 'P';
*lbmptr++ = 'B';
*lbmptr++ = 'M';
*lbmptr++ = ' ';
//
// write BMHD
//
*lbmptr++ = 'B';
*lbmptr++ = 'M';
*lbmptr++ = 'H';
*lbmptr++ = 'D';
bmhdlength = (int *)lbmptr;
lbmptr += 4; // leave space for length
memset( &basebmhd,0,sizeof( basebmhd ) );
basebmhd.w = BigShort( (short)width );
basebmhd.h = BigShort( (short)height );
basebmhd.nPlanes = BigShort( 8 );
basebmhd.xAspect = BigShort( 5 );
basebmhd.yAspect = BigShort( 6 );
basebmhd.pageWidth = BigShort( (short)width );
basebmhd.pageHeight = BigShort( (short)height );
memcpy( lbmptr,&basebmhd,sizeof( basebmhd ) );
lbmptr += sizeof( basebmhd );
length = lbmptr - (byte *)bmhdlength - 4;
*bmhdlength = BigLong( length );
if ( length & 1 ) {
*lbmptr++ = 0; // pad chunk to even offset
}
//
// write CMAP
//
*lbmptr++ = 'C';
*lbmptr++ = 'M';
*lbmptr++ = 'A';
*lbmptr++ = 'P';
cmaplength = (int *)lbmptr;
lbmptr += 4; // leave space for length
memcpy( lbmptr,palette,768 );
lbmptr += 768;
length = lbmptr - (byte *)cmaplength - 4;
*cmaplength = BigLong( length );
if ( length & 1 ) {
*lbmptr++ = 0; // pad chunk to even offset
}
//
// write BODY
//
*lbmptr++ = 'B';
*lbmptr++ = 'O';
*lbmptr++ = 'D';
*lbmptr++ = 'Y';
bodylength = (int *)lbmptr;
lbmptr += 4; // leave space for length
memcpy( lbmptr,data,width * height );
lbmptr += width * height;
length = lbmptr - (byte *)bodylength - 4;
*bodylength = BigLong( length );
if ( length & 1 ) {
*lbmptr++ = 0; // pad chunk to even offset
}
//
// done
//
length = lbmptr - (byte *)formlength - 4;
*formlength = BigLong( length );
if ( length & 1 ) {
*lbmptr++ = 0; // pad chunk to even offset
}
//
// write output file
//
SaveFile( filename, lbm, lbmptr - lbm );
free( lbm );
}
/*
============================================================================
LOAD PCX
============================================================================
*/
typedef struct
{
char manufacturer;
char version;
char encoding;
char bits_per_pixel;
unsigned short xmin,ymin,xmax,ymax;
unsigned short hres,vres;
unsigned char palette[48];
char reserved;
char color_planes;
unsigned short bytes_per_line;
unsigned short palette_type;
char filler[58];
unsigned char data; // unbounded
} pcx_t;
/*
==============
LoadPCX
==============
*/
/* RR2DO2 */
#define DECODEPCX( b, d, r ) d = *b++; if ( ( d & 0xC0 ) == 0xC0 ) {r = d & 0x3F; d = *b++; }else{r = 1; }
void LoadPCX( const char *filename, byte **pic, byte **palette, int *width, int *height ){
byte *raw;
pcx_t *pcx;
int x, y, lsize;
int len;
int dataByte, runLength;
byte *out, *pix;
/* load the file */
len = vfsLoadFile( filename, (void **)&raw, 0 );
if ( len == -1 ) {
Error( "LoadPCX: Couldn't read %s", filename );
}
/* parse the PCX file */
pcx = (pcx_t *)raw;
raw = &pcx->data;
pcx->xmin = LittleShort( pcx->xmin );
pcx->ymin = LittleShort( pcx->ymin );
pcx->xmax = LittleShort( pcx->xmax );
pcx->ymax = LittleShort( pcx->ymax );
pcx->hres = LittleShort( pcx->hres );
pcx->vres = LittleShort( pcx->vres );
pcx->bytes_per_line = LittleShort( pcx->bytes_per_line );
pcx->palette_type = LittleShort( pcx->palette_type );
if ( pcx->manufacturer != 0x0a
|| pcx->version != 5
|| pcx->encoding != 1
|| pcx->bits_per_pixel != 8
|| pcx->xmax >= 640
|| pcx->ymax >= 480 ) {
Error( "Bad pcx file %s", filename );
}
if ( palette ) {
*palette = safe_malloc( 768 );
memcpy( *palette, (byte *)pcx + len - 768, 768 );
}
if ( width ) {
*width = pcx->xmax + 1;
}
if ( height ) {
*height = pcx->ymax + 1;
}
if ( !pic ) {
return;
}
out = safe_malloc( ( pcx->ymax + 1 ) * ( pcx->xmax + 1 ) );
if ( !out ) {
Error( "LoadPCX: couldn't allocate" );
}
*pic = out;
pix = out;
/* RR2DO2: pcx fix */
lsize = pcx->color_planes * pcx->bytes_per_line;
/* go scanline by scanline */
for ( y = 0; y <= pcx->ymax; y++, pix += pcx->xmax + 1 )
{
/* do a scanline */
runLength = 0;
for ( x = 0; x <= pcx->xmax; )
{
/* RR2DO2 */
DECODEPCX( raw, dataByte, runLength );
while ( runLength-- > 0 )
pix[ x++ ] = dataByte;
}
/* RR2DO2: discard any other data */
while ( x < lsize )
{
DECODEPCX( raw, dataByte, runLength );
x++;
}
while ( runLength-- > 0 )
x++;
}
/* validity check */
if ( raw - (byte *) pcx > len ) {
Error( "PCX file %s was malformed", filename );
}
free( pcx );
}
/*
==============
WritePCXfile
==============
*/
void WritePCXfile( const char *filename, byte *data,
int width, int height, byte *palette ){
int i, j, length;
pcx_t *pcx;
byte *pack;
pcx = safe_malloc( width * height * 2 + 1000 );
memset( pcx, 0, sizeof( *pcx ) );
pcx->manufacturer = 0x0a; // PCX id
pcx->version = 5; // 256 color
pcx->encoding = 1; // uncompressed
pcx->bits_per_pixel = 8; // 256 color
pcx->xmin = 0;
pcx->ymin = 0;
pcx->xmax = LittleShort( (short)( width - 1 ) );
pcx->ymax = LittleShort( (short)( height - 1 ) );
pcx->hres = LittleShort( (short)width );
pcx->vres = LittleShort( (short)height );
pcx->color_planes = 1; // chunky image
pcx->bytes_per_line = LittleShort( (short)width );
pcx->palette_type = LittleShort( 1 ); // not a grey scale
// pack the image
pack = &pcx->data;
for ( i = 0; i < height; i++ )
{
for ( j = 0; j < width; j++ )
{
if ( ( *data & 0xc0 ) != 0xc0 ) {
*pack++ = *data++;
}
else
{
*pack++ = 0xc1;
*pack++ = *data++;
}
}
}
// write the palette
*pack++ = 0x0c; // palette ID byte
for ( i = 0; i < 768; i++ )
*pack++ = *palette++;
// write output file
length = pack - (byte *)pcx;
SaveFile( filename, pcx, length );
free( pcx );
}
/*
============================================================================
LOAD BMP
============================================================================
*/
/*
// we can't just use these structures, because
// compiler structure alignment will not be portable
// on this unaligned stuff
typedef struct tagBITMAPFILEHEADER { // bmfh
WORD bfType; // BM
DWORD bfSize;
WORD bfReserved1;
WORD bfReserved2;
DWORD bfOffBits;
} BITMAPFILEHEADER;
typedef struct tagBITMAPINFOHEADER{ // bmih
DWORD biSize;
LONG biWidth;
LONG biHeight;
WORD biPlanes;
WORD biBitCount
DWORD biCompression;
DWORD biSizeImage;
LONG biXPelsPerMeter;
LONG biYPelsPerMeter;
DWORD biClrUsed;
DWORD biClrImportant;
} BITMAPINFOHEADER;
typedef struct tagBITMAPINFO { // bmi
BITMAPINFOHEADER bmiHeader;
RGBQUAD bmiColors[1];
} BITMAPINFO;
typedef struct tagBITMAPCOREHEADER { // bmch
DWORD bcSize;
WORD bcWidth;
WORD bcHeight;
WORD bcPlanes;
WORD bcBitCount;
} BITMAPCOREHEADER;
typedef struct _BITMAPCOREINFO { // bmci
BITMAPCOREHEADER bmciHeader;
RGBTRIPLE bmciColors[1];
} BITMAPCOREINFO;
*/
/*
==============
LoadBMP
==============
*/
void LoadBMP( const char *filename, byte **pic, byte **palette, int *width, int *height ){
byte *out;
int i;
int bfOffBits;
int structSize;
int bcWidth;
int bcHeight;
int bcPlanes;
int bcBitCount;
byte bcPalette[1024];
qboolean flipped;
byte *in;
int len, pos = 0;
len = vfsLoadFile( filename, (void **)&in, 0 );
if ( len == -1 ) {
Error( "Couldn't read %s", filename );
}
i = bufLittleShort( in, len, &pos );
if ( i != 'B' + ( 'M' << 8 ) ) {
Error( "%s is not a bmp file", filename );
}
/* bfSize = */ bufLittleLong( in, len, &pos );
bufLittleShort( in, len, &pos );
bufLittleShort( in, len, &pos );
bfOffBits = bufLittleLong( in, len, &pos );
// the size will tell us if it is a
// bitmapinfo or a bitmapcore
structSize = bufLittleLong( in, len, &pos );
if ( structSize == 40 ) {
// bitmapinfo
bcWidth = bufLittleLong( in, len, &pos );
bcHeight = bufLittleLong( in, len, &pos );
bcPlanes = bufLittleShort( in, len, &pos );
bcBitCount = bufLittleShort( in, len, &pos );
pos += 24;
if ( palette ) {
memcpy( bcPalette, in + pos, 1024 );
pos += 1024;
*palette = safe_malloc( 768 );
for ( i = 0; i < 256; i++ )
{
( *palette )[i * 3 + 0] = bcPalette[i * 4 + 2];
( *palette )[i * 3 + 1] = bcPalette[i * 4 + 1];
( *palette )[i * 3 + 2] = bcPalette[i * 4 + 0];
}
}
}
else if ( structSize == 12 ) {
// bitmapcore
bcWidth = bufLittleShort( in, len, &pos );
bcHeight = bufLittleShort( in, len, &pos );
bcPlanes = bufLittleShort( in, len, &pos );
bcBitCount = bufLittleShort( in, len, &pos );
if ( palette ) {
memcpy( bcPalette, in + pos, 768 );
pos += 768;
*palette = safe_malloc( 768 );
for ( i = 0; i < 256; i++ ) {
( *palette )[i * 3 + 0] = bcPalette[i * 3 + 2];
( *palette )[i * 3 + 1] = bcPalette[i * 3 + 1];
( *palette )[i * 3 + 2] = bcPalette[i * 3 + 0];
}
}
}
else {
Error( "%s had strange struct size", filename );
}
if ( bcPlanes != 1 ) {
Error( "%s was not a single plane image", filename );
}
if ( bcBitCount != 8 ) {
Error( "%s was not an 8 bit image", filename );
}
if ( bcHeight < 0 ) {
bcHeight = -bcHeight;
flipped = qtrue;
}
else {
flipped = qfalse;
}
if ( width ) {
*width = bcWidth;
}
if ( height ) {
*height = bcHeight;
}
if ( !pic ) {
free( in );
return;
}
out = safe_malloc( bcWidth * bcHeight );
*pic = out;
pos = bfOffBits;
if ( flipped ) {
for ( i = 0; i < bcHeight; i++ ) {
memcpy( out + bcWidth * ( bcHeight - 1 - i ), in + pos, bcWidth );
pos += bcWidth;
}
}
else {
memcpy( out, in + pos, bcWidth * bcHeight );
pos += bcWidth * bcHeight;
}
free( in );
}
/*
============================================================================
LOAD IMAGE
============================================================================
*/
/*
==============
Load256Image
Will load either an lbm or pcx, depending on extension.
Any of the return pointers can be NULL if you don't want them.
==============
*/
void Load256Image( const char *name, byte **pixels, byte **palette, int *width, int *height ){
char ext[128];
ExtractFileExtension( name, ext );
if ( !Q_stricmp( ext, "lbm" ) ) {
LoadLBM( name, pixels, palette );
if ( width ) {
*width = bmhd.w;
}
if ( height ) {
*height = bmhd.h;
}
}
else if ( !Q_stricmp( ext, "pcx" ) ) {
LoadPCX( name, pixels, palette, width, height );
}
else if ( !Q_stricmp( ext, "bmp" ) ) {
LoadBMP( name, pixels, palette, width, height );
}
else{
Error( "%s doesn't have a known image extension", name );
}
}
/*
==============
Save256Image
Will save either an lbm or pcx, depending on extension.
==============
*/
void Save256Image( const char *name, byte *pixels, byte *palette,
int width, int height ){
char ext[128];
ExtractFileExtension( name, ext );
if ( !Q_stricmp( ext, "lbm" ) ) {
WriteLBMfile( name, pixels, width, height, palette );
}
else if ( !Q_stricmp( ext, "pcx" ) ) {
WritePCXfile( name, pixels, width, height, palette );
}
else{
Error( "%s doesn't have a known image extension", name );
}
}
/*
============================================================================
TARGA IMAGE
============================================================================
*/
typedef struct _TargaHeader {
unsigned char id_length, colormap_type, image_type;
unsigned short colormap_index, colormap_length;
unsigned char colormap_size;
unsigned short x_origin, y_origin, width, height;
unsigned char pixel_size, attributes;
} TargaHeader;
void TargaError( TargaHeader *t, const char *message ){
Sys_Printf( "%s\n:TargaHeader:\nuint8 id_length = %i;\nuint8 colormap_type = %i;\nuint8 image_type = %i;\nuint16 colormap_index = %i;\nuint16 colormap_length = %i;\nuint8 colormap_size = %i;\nuint16 x_origin = %i;\nuint16 y_origin = %i;\nuint16 width = %i;\nuint16 height = %i;\nuint8 pixel_size = %i;\nuint8 attributes = %i;\n", message, t->id_length, t->colormap_type, t->image_type, t->colormap_index, t->colormap_length, t->colormap_size, t->x_origin, t->y_origin, t->width, t->height, t->pixel_size, t->attributes );
}
/*
=============
LoadTGABuffer
=============
*/
void LoadTGABuffer( const byte *f, const byte *enddata, byte **pic, int *width, int *height ){
int x, y, row_inc, compressed, readpixelcount, red, green, blue, alpha, runlen, pindex, alphabits, image_width, image_height;
byte *pixbuf, *image_rgba;
const byte *fin;
unsigned char *p;
TargaHeader targa_header;
unsigned char palette[256 * 4];
*pic = NULL;
// abort if it is too small to parse
if ( enddata - f < 19 ) {
return;
}
targa_header.id_length = f[0];
targa_header.colormap_type = f[1];
targa_header.image_type = f[2];
targa_header.colormap_index = f[3] + f[4] * 256;
targa_header.colormap_length = f[5] + f[6] * 256;
targa_header.colormap_size = f[7];
targa_header.x_origin = f[8] + f[9] * 256;
targa_header.y_origin = f[10] + f[11] * 256;
targa_header.width = image_width = f[12] + f[13] * 256;
targa_header.height = image_height = f[14] + f[15] * 256;
targa_header.pixel_size = f[16];
targa_header.attributes = f[17];
// advance to end of header
fin = f + 18;
// skip TARGA image comment (usually 0 bytes)
fin += targa_header.id_length;
// read/skip the colormap if present (note: according to the TARGA spec it
// can be present even on truecolor or greyscale images, just not used by
// the image data)
if ( targa_header.colormap_type ) {
if ( targa_header.colormap_length > 256 ) {
TargaError( &targa_header, "LoadTGA: only up to 256 colormap_length supported\n" );
return;
}
if ( targa_header.colormap_index ) {
TargaError( &targa_header, "LoadTGA: colormap_index not supported\n" );
return;
}
if ( targa_header.colormap_size == 24 ) {
for ( x = 0; x < targa_header.colormap_length; x++ )
{
palette[x * 4 + 2] = *fin++;
palette[x * 4 + 1] = *fin++;
palette[x * 4 + 0] = *fin++;
palette[x * 4 + 3] = 255;
}
}
else if ( targa_header.colormap_size == 32 ) {
for ( x = 0; x < targa_header.colormap_length; x++ )
{
palette[x * 4 + 2] = *fin++;
palette[x * 4 + 1] = *fin++;
palette[x * 4 + 0] = *fin++;
palette[x * 4 + 3] = *fin++;
}
}
else
{
TargaError( &targa_header, "LoadTGA: Only 32 and 24 bit colormap_size supported\n" );
return;
}
}
// check our pixel_size restrictions according to image_type
if ( targa_header.image_type == 2 || targa_header.image_type == 10 ) {
if ( targa_header.pixel_size != 24 && targa_header.pixel_size != 32 ) {
TargaError( &targa_header, "LoadTGA: only 24bit and 32bit pixel sizes supported for type 2 and type 10 images\n" );
return;
}
}
else if ( targa_header.image_type == 1 || targa_header.image_type == 9 ) {
if ( targa_header.pixel_size != 8 ) {
TargaError( &targa_header, "LoadTGA: only 8bit pixel size for type 1, 3, 9, and 11 images supported\n" );
return;
}
}
else if ( targa_header.image_type == 3 || targa_header.image_type == 11 ) {
if ( targa_header.pixel_size != 8 ) {
TargaError( &targa_header, "LoadTGA: only 8bit pixel size for type 1, 3, 9, and 11 images supported\n" );
return;
}
}
else
{
TargaError( &targa_header, "LoadTGA: Only type 1, 2, 3, 9, 10, and 11 targa RGB images supported" );
return;
}
if ( targa_header.attributes & 0x10 ) {
TargaError( &targa_header, "LoadTGA: origin must be in top left or bottom left, top right and bottom right are not supported\n" );
return;
}
// number of attribute bits per pixel, we only support 0 or 8
alphabits = targa_header.attributes & 0x0F;
if ( alphabits != 8 && alphabits != 0 ) {
TargaError( &targa_header, "LoadTGA: only 0 or 8 attribute (alpha) bits supported\n" );
return;
}
image_rgba = safe_malloc( image_width * image_height * 4 );
if ( !image_rgba ) {
Sys_Printf( "LoadTGA: not enough memory for %i by %i image\n", image_width, image_height );
return;
}
// If bit 5 of attributes isn't set, the image has been stored from bottom to top
if ( ( targa_header.attributes & 0x20 ) == 0 ) {
pixbuf = image_rgba + ( image_height - 1 ) * image_width * 4;
row_inc = -image_width * 4 * 2;
}
else
{
pixbuf = image_rgba;
row_inc = 0;
}
compressed = targa_header.image_type == 9 || targa_header.image_type == 10 || targa_header.image_type == 11;
x = 0;
y = 0;
red = green = blue = alpha = 255;
while ( y < image_height )
{
// decoder is mostly the same whether it's compressed or not
readpixelcount = 1000000;
runlen = 1000000;
if ( compressed && fin < enddata ) {
runlen = *fin++;
// high bit indicates this is an RLE compressed run
if ( runlen & 0x80 ) {
readpixelcount = 1;
}
runlen = 1 + ( runlen & 0x7f );
}
while ( ( runlen-- ) && y < image_height )
{
if ( readpixelcount > 0 ) {
readpixelcount--;
red = green = blue = alpha = 255;
if ( fin < enddata ) {
switch ( targa_header.image_type )
{
case 1:
case 9:
// colormapped
pindex = *fin++;
if ( pindex >= targa_header.colormap_length ) {
pindex = 0; // error
}
p = palette + pindex * 4;
red = p[0];
green = p[1];
blue = p[2];
alpha = p[3];
break;
case 2:
case 10:
// BGR or BGRA
blue = *fin++;
if ( fin < enddata ) {
green = *fin++;
}
if ( fin < enddata ) {
red = *fin++;
}
if ( targa_header.pixel_size == 32 && fin < enddata ) {
alpha = *fin++;
}
break;
case 3:
case 11:
// greyscale
red = green = blue = *fin++;
break;
}
if ( !alphabits ) {
alpha = 255;
}
}
}
*pixbuf++ = red;
*pixbuf++ = green;
*pixbuf++ = blue;
*pixbuf++ = alpha;
x++;
if ( x == image_width ) {
// end of line, advance to next
x = 0;
y++;
pixbuf += row_inc;
}
}
}
*pic = image_rgba;
if ( width ) {
*width = image_width;
}
if ( height ) {
*height = image_height;
}
}
/*
=============
LoadTGA
=============
*/
void LoadTGA( const char *name, byte **pixels, int *width, int *height ){
byte *buffer;
int nLen;
//
// load the file
//
nLen = vfsLoadFile( name, (void **)&buffer, 0 );
if ( nLen == -1 ) {
Error( "Couldn't read %s", name );
}
LoadTGABuffer( buffer, buffer + nLen, pixels, width, height );
}
/*
================
WriteTGA
================
*/
void WriteTGA( const char *filename, byte *data, int width, int height ) {
byte *buffer;
int i;
int c;
FILE *f;
buffer = safe_malloc( width * height * 4 + 18 );
memset( buffer, 0, 18 );
buffer[2] = 2; // uncompressed type
buffer[12] = width & 255;
buffer[13] = width >> 8;
buffer[14] = height & 255;
buffer[15] = height >> 8;
buffer[16] = 32; // pixel size
// swap rgb to bgr
c = 18 + width * height * 4;
for ( i = 18; i < c; i += 4 )
{
buffer[i] = data[i - 18 + 2]; // blue
buffer[i + 1] = data[i - 18 + 1]; // green
buffer[i + 2] = data[i - 18 + 0]; // red
buffer[i + 3] = data[i - 18 + 3]; // alpha
}
f = fopen( filename, "wb" );
fwrite( buffer, 1, c, f );
fclose( f );
free( buffer );
}
void WriteTGAGray( const char *filename, byte *data, int width, int height ) {
byte buffer[18];
FILE *f;
memset( buffer, 0, 18 );
buffer[2] = 3; // uncompressed type
buffer[12] = width & 255;
buffer[13] = width >> 8;
buffer[14] = height & 255;
buffer[15] = height >> 8;
buffer[16] = 8; // pixel size
f = fopen( filename, "wb" );
fwrite( buffer, 1, 18, f );
fwrite( data, 1, width * height, f );
fclose( f );
}
/*
============================================================================
LOAD32BITIMAGE
============================================================================
*/
/*
==============
Load32BitImage
Any of the return pointers can be NULL if you don't want them.
==============
*/
void Load32BitImage( const char *name, unsigned **pixels, int *width, int *height ){
char ext[128];
byte *palette;
byte *pixels8;
byte *pixels32;
int size;
int i;
int v;
ExtractFileExtension( name, ext );
if ( !Q_stricmp( ext, "tga" ) ) {
LoadTGA( name, (byte **)pixels, width, height );
}
else {
Load256Image( name, &pixels8, &palette, width, height );
if ( !pixels ) {
return;
}
size = *width * *height;
pixels32 = safe_malloc( size * 4 );
*pixels = (unsigned *)pixels32;
for ( i = 0; i < size; i++ ) {
v = pixels8[i];
pixels32[i * 4 + 0] = palette[ v * 3 + 0 ];
pixels32[i * 4 + 1] = palette[ v * 3 + 1 ];
pixels32[i * 4 + 2] = palette[ v * 3 + 2 ];
pixels32[i * 4 + 3] = 0xff;
}
}
}
/*
============================================================================
KHRONOS TEXTURE
============================================================================
*/
#define KTX_UINT32_LE( buf ) ( ( unsigned int )( (buf)[0] | ( (buf)[1] << 8 ) | ( (buf)[2] << 16 ) | ( (buf)[3] << 24 ) ) )
#define KTX_UINT32_BE( buf ) ( ( unsigned int )( (buf)[3] | ( (buf)[2] << 8 ) | ( (buf)[1] << 16 ) | ( (buf)[0] << 24 ) ) )
#define KTX_TYPE_UNSIGNED_BYTE 0x1401
#define KTX_TYPE_UNSIGNED_SHORT_4_4_4_4 0x8033
#define KTX_TYPE_UNSIGNED_SHORT_5_5_5_1 0x8034
#define KTX_TYPE_UNSIGNED_SHORT_5_6_5 0x8363
#define KTX_FORMAT_ALPHA 0x1906
#define KTX_FORMAT_RGB 0x1907
#define KTX_FORMAT_RGBA 0x1908
#define KTX_FORMAT_LUMINANCE 0x1909
#define KTX_FORMAT_LUMINANCE_ALPHA 0x190A
#define KTX_FORMAT_BGR 0x80E0
#define KTX_FORMAT_BGRA 0x80E1
#define KTX_FORMAT_ETC1_RGB8 0x8D64
static void KTX_DecodeA8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = out[1] = out[2] = 0;
out[3] = in[0];
}
static void KTX_DecodeRGB8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = in[0];
out[1] = in[1];
out[2] = in[2];
out[3] = 255;
}
static void KTX_DecodeRGBA8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = in[0];
out[1] = in[1];
out[2] = in[2];
out[3] = in[3];
}
static void KTX_DecodeL8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = out[1] = out[2] = in[0];
out[3] = 255;
}
static void KTX_DecodeLA8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = out[1] = out[2] = in[0];
out[3] = in[1];
}
static void KTX_DecodeBGR8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = in[2];
out[1] = in[1];
out[2] = in[0];
out[3] = 255;
}
static void KTX_DecodeBGRA8( const byte *in, qboolean bigEndian, byte *out ){
out[0] = in[2];
out[1] = in[1];
out[2] = in[0];
out[3] = in[3];
}
static void KTX_DecodeRGBA4( const byte *in, qboolean bigEndian, byte *out ){
unsigned short rgba;
int r, g, b, a;
if ( bigEndian ) {
rgba = ( in[0] << 8 ) | in[1];
}
else {
rgba = ( in[1] << 8 ) | in[0];
}
r = ( rgba >> 12 ) & 0xf;
g = ( rgba >> 8 ) & 0xf;
b = ( rgba >> 4 ) & 0xf;
a = rgba & 0xf;
out[0] = ( r << 4 ) | r;
out[1] = ( g << 4 ) | g;
out[2] = ( b << 4 ) | b;
out[3] = ( a << 4 ) | a;
}
static void KTX_DecodeRGBA5( const byte *in, qboolean bigEndian, byte *out ){
unsigned short rgba;
int r, g, b;
if ( bigEndian ) {
rgba = ( in[0] << 8 ) | in[1];
}
else {
rgba = ( in[1] << 8 ) | in[0];
}
r = ( rgba >> 11 ) & 0x1f;
g = ( rgba >> 6 ) & 0x1f;
b = ( rgba >> 1 ) & 0x1f;
out[0] = ( r << 3 ) | ( r >> 2 );
out[1] = ( g << 3 ) | ( g >> 2 );
out[2] = ( b << 3 ) | ( b >> 2 );
out[3] = ( rgba & 1 ) * 255;
}
static void KTX_DecodeRGB5( const byte *in, qboolean bigEndian, byte *out ){
unsigned short rgba;
int r, g, b;
if ( bigEndian ) {
rgba = ( in[0] << 8 ) | in[1];
}
else {
rgba = ( in[1] << 8 ) | in[0];
}
r = ( rgba >> 11 ) & 0x1f;
g = ( rgba >> 5 ) & 0x3f;
b = rgba & 0x1f;
out[0] = ( r << 3 ) | ( r >> 2 );
out[1] = ( g << 2 ) | ( g >> 4 );
out[2] = ( b << 3 ) | ( b >> 2 );
out[3] = 255;
}
typedef struct
{
unsigned int type;
unsigned int format;
unsigned int pixelSize;
void ( *decode )( const byte *in, qboolean bigEndian, byte *out );
} KTX_UncompressedFormat_t;
static const KTX_UncompressedFormat_t KTX_UncompressedFormats[] =
{
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_ALPHA, 1, KTX_DecodeA8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_RGB, 3, KTX_DecodeRGB8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_RGBA, 4, KTX_DecodeRGBA8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_LUMINANCE, 1, KTX_DecodeL8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_LUMINANCE_ALPHA, 2, KTX_DecodeLA8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_BGR, 3, KTX_DecodeBGR8 },
{ KTX_TYPE_UNSIGNED_BYTE, KTX_FORMAT_BGRA, 4, KTX_DecodeBGRA8 },
{ KTX_TYPE_UNSIGNED_SHORT_4_4_4_4, KTX_FORMAT_RGBA, 2, KTX_DecodeRGBA4 },
{ KTX_TYPE_UNSIGNED_SHORT_5_5_5_1, KTX_FORMAT_RGBA, 2, KTX_DecodeRGBA5 },
{ KTX_TYPE_UNSIGNED_SHORT_5_6_5, KTX_FORMAT_RGB, 2, KTX_DecodeRGB5 },
{ 0, 0, 0, NULL }
};
static qboolean KTX_DecodeETC1( const byte* in, size_t inSize, unsigned int width, unsigned int height, byte* out ){
unsigned int y, stride = width * 4;
byte rgba[64];
if ( inSize < ( ( ( ( width + 3 ) & ~3 ) * ( ( height + 3 ) & ~3 ) ) >> 1 ) ) {
return qfalse;
}
for ( y = 0; y < height; y += 4, out += stride * 4 )
{
byte *p;
unsigned int x, blockrows;
blockrows = height - y;
if ( blockrows > 4 ) {
blockrows = 4;
}
p = out;
for ( x = 0; x < width; x += 4, p += 16 )
{
unsigned int blockrowsize, blockrow;
ETC_DecodeETC1Block( in, rgba, qtrue );
in += 8;
blockrowsize = width - x;
if ( blockrowsize > 4 ) {
blockrowsize = 4;
}
blockrowsize *= 4;
for ( blockrow = 0; blockrow < blockrows; blockrow++ )
{
memcpy( p + blockrow * stride, rgba + blockrow * 16, blockrowsize );
}
}
}
return qtrue;
}
#define KTX_HEADER_UINT32( buf ) ( bigEndian ? KTX_UINT32_BE( buf ) : KTX_UINT32_LE( buf ) )
void LoadKTXBufferFirstImage( const byte *buffer, size_t bufSize, byte **pic, int *picWidth, int *picHeight ){
unsigned int type, format, width, height, imageOffset;
byte *pixels;
if ( bufSize < 64 ) {
Error( "LoadKTX: Image doesn't have a header" );
}
if ( memcmp( buffer, "\xABKTX 11\xBB\r\n\x1A\n", 12 ) ) {
Error( "LoadKTX: Image has the wrong identifier" );
}
qboolean bigEndian = (qboolean)( buffer[4] == 4 );
type = KTX_HEADER_UINT32( buffer + 16 );
if ( type ) {
format = KTX_HEADER_UINT32( buffer + 32 );
}
else {
format = KTX_HEADER_UINT32( buffer + 28 );
}
width = KTX_HEADER_UINT32( buffer + 36 );
height = KTX_HEADER_UINT32( buffer + 40 );
if ( !width ) {
Error( "LoadKTX: Image has zero width" );
}
if ( !height ) {
height = 1;
}
if ( picWidth ) {
*picWidth = width;
}
if ( picHeight ) {
*picHeight = height;
}
imageOffset = 64 + KTX_HEADER_UINT32( buffer + 60 ) + 4;
if ( bufSize < imageOffset ) {
Error( "LoadKTX: No image in the file" );
}
buffer += imageOffset;
bufSize -= imageOffset;
pixels = safe_malloc( width * height * 4 );
*pic = pixels;
if ( type ) {
const KTX_UncompressedFormat_t *ktxFormat = KTX_UncompressedFormats;
unsigned int pixelSize;
unsigned int inRowLength, inPadding;
unsigned int y;
while ( ktxFormat->type )
{
if ( ktxFormat->type == type && ktxFormat->format == format ) {
break;
}
ktxFormat++;
}
if ( !ktxFormat->type ) {
Error( "LoadKTX: Image has an unsupported pixel type 0x%X or format 0x%X", type, format );
}
pixelSize = ktxFormat->pixelSize;
inRowLength = width * pixelSize;
inPadding = ( ( inRowLength + 3 ) & ~3 ) - inRowLength;
if ( bufSize < height * ( inRowLength + inPadding ) ) {
Error( "LoadKTX: Image is truncated" );
}
for ( y = 0; y < height; y++ )
{
unsigned int x;
for ( x = 0; x < width; x++, buffer += pixelSize, pixels += 4 )
{
ktxFormat->decode( buffer, bigEndian, pixels );
}
buffer += inPadding;
}
}
else {
qboolean decoded = qfalse;
switch ( format )
{
case KTX_FORMAT_ETC1_RGB8:
decoded = KTX_DecodeETC1( buffer, bufSize, width, height, pixels );
break;
default:
Error( "LoadKTX: Image has an unsupported compressed format format 0x%X", format );
break;
}
if ( !decoded ) {
Error( "LoadKTX: Image is truncated" );
}
}
}
Reference in a new issue View git blame Copy permalink