forked from KolibriOS/kolibrios
495 lines
12 KiB
C
495 lines
12 KiB
C
/* readtex.c */
|
|
|
|
/*
|
|
* Read an SGI .rgb image file and generate a mipmap texture set.
|
|
* Much of this code was borrowed from SGI's tk OpenGL toolkit.
|
|
*/
|
|
|
|
|
|
|
|
#include <GL/gl.h>
|
|
#include <GL/glu.h>
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "readtex.h"
|
|
|
|
|
|
#ifndef SEEK_SET
|
|
# define SEEK_SET 0
|
|
#endif
|
|
|
|
|
|
/*
|
|
** RGB Image Structure
|
|
*/
|
|
|
|
typedef struct _TK_RGBImageRec {
|
|
GLint sizeX, sizeY;
|
|
GLint components;
|
|
unsigned char *data;
|
|
} TK_RGBImageRec;
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
typedef struct _rawImageRec {
|
|
unsigned short imagic;
|
|
unsigned short type;
|
|
unsigned short dim;
|
|
unsigned short sizeX, sizeY, sizeZ;
|
|
unsigned long min, max;
|
|
unsigned long wasteBytes;
|
|
char name[80];
|
|
unsigned long colorMap;
|
|
FILE *file;
|
|
unsigned char *tmp, *tmpR, *tmpG, *tmpB, *tmpA;
|
|
unsigned long rleEnd;
|
|
GLuint *rowStart;
|
|
GLint *rowSize;
|
|
} rawImageRec;
|
|
|
|
/******************************************************************************/
|
|
|
|
static void ConvertShort(unsigned short *array, long length)
|
|
{
|
|
unsigned long b1, b2;
|
|
unsigned char *ptr;
|
|
|
|
ptr = (unsigned char *)array;
|
|
while (length--) {
|
|
b1 = *ptr++;
|
|
b2 = *ptr++;
|
|
*array++ = (unsigned short) ((b1 << 8) | (b2));
|
|
}
|
|
}
|
|
|
|
static void ConvertLong(GLuint *array, long length)
|
|
{
|
|
unsigned long b1, b2, b3, b4;
|
|
unsigned char *ptr;
|
|
|
|
ptr = (unsigned char *)array;
|
|
while (length--) {
|
|
b1 = *ptr++;
|
|
b2 = *ptr++;
|
|
b3 = *ptr++;
|
|
b4 = *ptr++;
|
|
*array++ = (b1 << 24) | (b2 << 16) | (b3 << 8) | (b4);
|
|
}
|
|
}
|
|
|
|
static rawImageRec *RawImageOpen(const char *fileName)
|
|
{
|
|
union {
|
|
int testWord;
|
|
char testByte[4];
|
|
} endianTest;
|
|
rawImageRec *raw;
|
|
GLenum swapFlag;
|
|
int x;
|
|
size_t result;
|
|
|
|
endianTest.testWord = 1;
|
|
if (endianTest.testByte[0] == 1) {
|
|
swapFlag = GL_TRUE;
|
|
} else {
|
|
swapFlag = GL_FALSE;
|
|
}
|
|
|
|
raw = (rawImageRec *)calloc(1, sizeof(rawImageRec));
|
|
if (raw == NULL) {
|
|
fprintf(stderr, "Out of memory!\n");
|
|
return NULL;
|
|
}
|
|
raw->file = fopen(fileName, "rb");
|
|
if (raw->file == NULL) {
|
|
const char *baseName = strrchr(fileName, '/');
|
|
if(baseName)
|
|
raw->file = fopen(baseName + 1, "rb");
|
|
if(raw->file == NULL) {
|
|
free(raw);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
result = fread(raw, 1, 12, raw->file);
|
|
assert(result == 12);
|
|
|
|
if (swapFlag) {
|
|
ConvertShort(&raw->imagic, 1);
|
|
ConvertShort(&raw->type, 1);
|
|
ConvertShort(&raw->dim, 1);
|
|
ConvertShort(&raw->sizeX, 1);
|
|
ConvertShort(&raw->sizeY, 1);
|
|
ConvertShort(&raw->sizeZ, 1);
|
|
}
|
|
|
|
raw->tmp = (unsigned char *)malloc(raw->sizeX*256);
|
|
raw->tmpR = (unsigned char *)malloc(raw->sizeX*256);
|
|
raw->tmpG = (unsigned char *)malloc(raw->sizeX*256);
|
|
raw->tmpB = (unsigned char *)malloc(raw->sizeX*256);
|
|
if (raw->sizeZ==4) {
|
|
raw->tmpA = (unsigned char *)malloc(raw->sizeX*256);
|
|
}
|
|
if (raw->tmp == NULL || raw->tmpR == NULL || raw->tmpG == NULL ||
|
|
raw->tmpB == NULL) {
|
|
fprintf(stderr, "Out of memory!\n");
|
|
free(raw->tmp);
|
|
free(raw->tmpR);
|
|
free(raw->tmpG);
|
|
free(raw->tmpB);
|
|
free(raw->tmpA);
|
|
free(raw);
|
|
return NULL;
|
|
}
|
|
|
|
if ((raw->type & 0xFF00) == 0x0100) {
|
|
x = raw->sizeY * raw->sizeZ * sizeof(GLuint);
|
|
raw->rowStart = (GLuint *)malloc(x);
|
|
raw->rowSize = (GLint *)malloc(x);
|
|
if (raw->rowStart == NULL || raw->rowSize == NULL) {
|
|
fprintf(stderr, "Out of memory!\n");
|
|
free(raw->tmp);
|
|
free(raw->tmpR);
|
|
free(raw->tmpG);
|
|
free(raw->tmpB);
|
|
free(raw->tmpA);
|
|
free(raw->rowStart);
|
|
free(raw->rowSize);
|
|
free(raw);
|
|
return NULL;
|
|
}
|
|
raw->rleEnd = 512 + (2 * x);
|
|
fseek(raw->file, 512, SEEK_SET);
|
|
result = fread(raw->rowStart, 1, x, raw->file);
|
|
assert(result == x);
|
|
result = fread(raw->rowSize, 1, x, raw->file);
|
|
assert(result == x);
|
|
if (swapFlag) {
|
|
ConvertLong(raw->rowStart, (long) (x/sizeof(GLuint)));
|
|
ConvertLong((GLuint *)raw->rowSize, (long) (x/sizeof(GLint)));
|
|
}
|
|
}
|
|
return raw;
|
|
}
|
|
|
|
static void RawImageClose(rawImageRec *raw)
|
|
{
|
|
fclose(raw->file);
|
|
free(raw->tmp);
|
|
free(raw->tmpR);
|
|
free(raw->tmpG);
|
|
free(raw->tmpB);
|
|
if (raw->rowStart)
|
|
free(raw->rowStart);
|
|
if (raw->rowSize)
|
|
free(raw->rowSize);
|
|
if (raw->sizeZ>3) {
|
|
free(raw->tmpA);
|
|
}
|
|
free(raw);
|
|
}
|
|
|
|
static void RawImageGetRow(rawImageRec *raw, unsigned char *buf, int y, int z)
|
|
{
|
|
unsigned char *iPtr, *oPtr, pixel;
|
|
int count, done = 0;
|
|
size_t result;
|
|
|
|
if ((raw->type & 0xFF00) == 0x0100) {
|
|
fseek(raw->file, (long) raw->rowStart[y+z*raw->sizeY], SEEK_SET);
|
|
result = fread(raw->tmp, 1, (unsigned int)raw->rowSize[y+z*raw->sizeY],
|
|
raw->file);
|
|
assert(result == (unsigned int)raw->rowSize[y+z*raw->sizeY]);
|
|
|
|
iPtr = raw->tmp;
|
|
oPtr = buf;
|
|
while (!done) {
|
|
pixel = *iPtr++;
|
|
count = (int)(pixel & 0x7F);
|
|
if (!count) {
|
|
done = 1;
|
|
return;
|
|
}
|
|
if (pixel & 0x80) {
|
|
while (count--) {
|
|
*oPtr++ = *iPtr++;
|
|
}
|
|
} else {
|
|
pixel = *iPtr++;
|
|
while (count--) {
|
|
*oPtr++ = pixel;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
fseek(raw->file, 512+(y*raw->sizeX)+(z*raw->sizeX*raw->sizeY),
|
|
SEEK_SET);
|
|
result = fread(buf, 1, raw->sizeX, raw->file);
|
|
assert(result == raw->sizeX);
|
|
}
|
|
}
|
|
|
|
|
|
static void RawImageGetData(rawImageRec *raw, TK_RGBImageRec *final)
|
|
{
|
|
unsigned char *ptr;
|
|
int i, j;
|
|
|
|
final->data = (unsigned char *)malloc((raw->sizeX+1)*(raw->sizeY+1)*4);
|
|
if (final->data == NULL) {
|
|
fprintf(stderr, "Out of memory!\n");
|
|
return;
|
|
}
|
|
|
|
ptr = final->data;
|
|
for (i = 0; i < (int)(raw->sizeY); i++) {
|
|
RawImageGetRow(raw, raw->tmpR, i, 0);
|
|
RawImageGetRow(raw, raw->tmpG, i, 1);
|
|
RawImageGetRow(raw, raw->tmpB, i, 2);
|
|
if (raw->sizeZ>3) {
|
|
RawImageGetRow(raw, raw->tmpA, i, 3);
|
|
}
|
|
for (j = 0; j < (int)(raw->sizeX); j++) {
|
|
*ptr++ = *(raw->tmpR + j);
|
|
*ptr++ = *(raw->tmpG + j);
|
|
*ptr++ = *(raw->tmpB + j);
|
|
if (raw->sizeZ>3) {
|
|
*ptr++ = *(raw->tmpA + j);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
static TK_RGBImageRec *tkRGBImageLoad(const char *fileName)
|
|
{
|
|
rawImageRec *raw;
|
|
TK_RGBImageRec *final;
|
|
|
|
raw = RawImageOpen(fileName);
|
|
if (!raw) {
|
|
fprintf(stderr, "File not found\n");
|
|
return NULL;
|
|
}
|
|
final = (TK_RGBImageRec *)malloc(sizeof(TK_RGBImageRec));
|
|
if (final == NULL) {
|
|
fprintf(stderr, "Out of memory!\n");
|
|
RawImageClose(raw);
|
|
return NULL;
|
|
}
|
|
final->sizeX = raw->sizeX;
|
|
final->sizeY = raw->sizeY;
|
|
final->components = raw->sizeZ;
|
|
RawImageGetData(raw, final);
|
|
RawImageClose(raw);
|
|
return final;
|
|
}
|
|
|
|
|
|
static void FreeImage( TK_RGBImageRec *image )
|
|
{
|
|
free(image->data);
|
|
free(image);
|
|
}
|
|
|
|
|
|
/*
|
|
* Load an SGI .rgb file and generate a set of 2-D mipmaps from it.
|
|
* Input: imageFile - name of .rgb to read
|
|
* intFormat - internal texture format to use, or number of components
|
|
* Return: GL_TRUE if success, GL_FALSE if error.
|
|
*/
|
|
GLboolean LoadRGBMipmaps( const char *imageFile, GLint intFormat )
|
|
{
|
|
GLint w, h;
|
|
return LoadRGBMipmaps2( imageFile, GL_TEXTURE_2D, intFormat, &w, &h );
|
|
}
|
|
|
|
|
|
|
|
GLboolean LoadRGBMipmaps2( const char *imageFile, GLenum target,
|
|
GLint intFormat, GLint *width, GLint *height )
|
|
{
|
|
GLint error;
|
|
GLenum format;
|
|
TK_RGBImageRec *image;
|
|
|
|
image = tkRGBImageLoad( imageFile );
|
|
if (!image) {
|
|
return GL_FALSE;
|
|
}
|
|
|
|
if (image->components==3) {
|
|
format = GL_RGB;
|
|
}
|
|
else if (image->components==4) {
|
|
format = GL_RGBA;
|
|
}
|
|
else {
|
|
/* not implemented */
|
|
fprintf(stderr,
|
|
"Error in LoadRGBMipmaps %d-component images not implemented\n",
|
|
image->components );
|
|
FreeImage(image);
|
|
return GL_FALSE;
|
|
}
|
|
|
|
error = gluBuild2DMipmaps( target,
|
|
intFormat,
|
|
image->sizeX, image->sizeY,
|
|
format,
|
|
GL_UNSIGNED_BYTE,
|
|
image->data );
|
|
|
|
*width = image->sizeX;
|
|
*height = image->sizeY;
|
|
|
|
FreeImage(image);
|
|
|
|
return error ? GL_FALSE : GL_TRUE;
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
* Load an SGI .rgb file and return a pointer to the image data.
|
|
* Input: imageFile - name of .rgb to read
|
|
* Output: width - width of image
|
|
* height - height of image
|
|
* format - format of image (GL_RGB or GL_RGBA)
|
|
* Return: pointer to image data or NULL if error
|
|
*/
|
|
GLubyte *LoadRGBImage( const char *imageFile, GLint *width, GLint *height,
|
|
GLenum *format )
|
|
{
|
|
TK_RGBImageRec *image;
|
|
GLint bytes;
|
|
GLubyte *buffer;
|
|
|
|
image = tkRGBImageLoad( imageFile );
|
|
if (!image) {
|
|
return NULL;
|
|
}
|
|
|
|
if (image->components==3) {
|
|
*format = GL_RGB;
|
|
}
|
|
else if (image->components==4) {
|
|
*format = GL_RGBA;
|
|
}
|
|
else {
|
|
/* not implemented */
|
|
fprintf(stderr,
|
|
"Error in LoadRGBImage %d-component images not implemented\n",
|
|
image->components );
|
|
FreeImage(image);
|
|
return NULL;
|
|
}
|
|
|
|
*width = image->sizeX;
|
|
*height = image->sizeY;
|
|
|
|
bytes = image->sizeX * image->sizeY * image->components;
|
|
buffer = (GLubyte *) malloc(bytes);
|
|
if (!buffer) {
|
|
FreeImage(image);
|
|
return NULL;
|
|
}
|
|
|
|
memcpy( (void *) buffer, (void *) image->data, bytes );
|
|
|
|
FreeImage(image);
|
|
|
|
return buffer;
|
|
}
|
|
|
|
#define CLAMP( X, MIN, MAX ) ( (X)<(MIN) ? (MIN) : ((X)>(MAX) ? (MAX) : (X)) )
|
|
|
|
|
|
static void ConvertRGBtoYUV(GLint w, GLint h, GLint texel_bytes,
|
|
const GLubyte *src,
|
|
GLushort *dest)
|
|
{
|
|
GLint i, j;
|
|
|
|
for (i = 0; i < h; i++) {
|
|
for (j = 0; j < w; j++) {
|
|
const GLfloat r = (src[0]) / 255.0;
|
|
const GLfloat g = (src[1]) / 255.0;
|
|
const GLfloat b = (src[2]) / 255.0;
|
|
GLfloat y, cr, cb;
|
|
GLint iy, icr, icb;
|
|
|
|
y = r * 65.481 + g * 128.553 + b * 24.966 + 16;
|
|
cb = r * -37.797 + g * -74.203 + b * 112.0 + 128;
|
|
cr = r * 112.0 + g * -93.786 + b * -18.214 + 128;
|
|
/*printf("%f %f %f -> %f %f %f\n", r, g, b, y, cb, cr);*/
|
|
iy = (GLint) CLAMP(y, 0, 254);
|
|
icb = (GLint) CLAMP(cb, 0, 254);
|
|
icr = (GLint) CLAMP(cr, 0, 254);
|
|
|
|
if (j & 1) {
|
|
/* odd */
|
|
*dest = (iy << 8) | icr;
|
|
}
|
|
else {
|
|
/* even */
|
|
*dest = (iy << 8) | icb;
|
|
}
|
|
dest++;
|
|
src += texel_bytes;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
* Load an SGI .rgb file and return a pointer to the image data, converted
|
|
* to 422 yuv.
|
|
*
|
|
* Input: imageFile - name of .rgb to read
|
|
* Output: width - width of image
|
|
* height - height of image
|
|
* Return: pointer to image data or NULL if error
|
|
*/
|
|
GLushort *LoadYUVImage( const char *imageFile, GLint *width, GLint *height )
|
|
{
|
|
TK_RGBImageRec *image;
|
|
GLushort *buffer;
|
|
|
|
image = tkRGBImageLoad( imageFile );
|
|
if (!image) {
|
|
return NULL;
|
|
}
|
|
|
|
if (image->components != 3 && image->components !=4 ) {
|
|
/* not implemented */
|
|
fprintf(stderr,
|
|
"Error in LoadYUVImage %d-component images not implemented\n",
|
|
image->components );
|
|
FreeImage(image);
|
|
return NULL;
|
|
}
|
|
|
|
*width = image->sizeX;
|
|
*height = image->sizeY;
|
|
|
|
buffer = (GLushort *) malloc( image->sizeX * image->sizeY * 2 );
|
|
|
|
if (buffer)
|
|
ConvertRGBtoYUV( image->sizeX,
|
|
image->sizeY,
|
|
image->components,
|
|
image->data,
|
|
buffer );
|
|
|
|
|
|
FreeImage(image);
|
|
return buffer;
|
|
}
|
|
|