tTh/libtthimage
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
4f41ae399b
libtthimage/Lib/tga.c
tTh 449705c0ee check error on header read
2024-07-16 18:58:11 +02:00

614 lines
17 KiB
C
Raw Blame History

/*
tga.c
=====
7 Oct 2001: les problemes de boutisme sont regles.
*/
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include "../tthimage.h"
/*::------------------------------------------------------------------::*/
#define LIBTGA_VERSION "0.2.42"
typedef struct {
uint8_t text_size;
char is_color_map; /* 0=NO, 1=YES */
char type;
uint16_t map_start, map_length;
char map_bits;
uint16_t x_start, y_start;
uint16_t width, height;
char bits; /* bits per pixel (8, 24) */
char o_flags; /* orientation flags ? */
} Tga_file_header;
/*::------------------------------------------------------------------::*/
int Image_TGA_write_header(FILE *fp, Tga_file_header *head)
{
fwrite (&head->text_size, 1, 1, fp);
fwrite (&head->is_color_map, 1, 1, fp);
fwrite (&head->type, 1, 1, fp);
Image_basicIO_write_I_short(fp, head->map_start);
Image_basicIO_write_I_short(fp, head->map_length);
fwrite (&head->map_bits, 1, 1, fp);
Image_basicIO_write_I_short(fp, head->x_start);
Image_basicIO_write_I_short(fp, head->y_start);
Image_basicIO_write_I_short(fp, head->width);
Image_basicIO_write_I_short(fp, head->height);
fwrite (&head->bits, 1, 1, fp);
fwrite (&head->o_flags, 1, 1, fp);
return 0;
}
/*::------------------------------------------------------------------::*/
static void fatal_error(char *txt)
{
fprintf(stderr, "TGA: (pid:%d) Fatal error: %s\n", getpid(), txt);
#if FORCE_ABORT
abort();
#endif
exit(10);
}
/*::------------------------------------------------------------------::*/
static int tga_pack_line(void)
{
/*
* ___TODO___ compressed targa file !
*
* Novembre 2002: il serait temps de s'y mettre...
*
*/
return -1;
}
/*::------------------------------------------------------------------::*/
/*
* il manque un grand nombre d'options
*
* - la compression ---> URGENT !!!
* - les images a palettes
*/
int Image_TGA_save(char *nom, Image_Desc *img, int compress)
{
int ltxt, line, foo;
Tga_file_header head;
FILE *fp;
uint8_t *Rptr, *Gptr, *Bptr, *Aptr;
uint8_t *buffer, *ptr;
if (nom == NULL) fatal_error("save image: name ptr is NULL");
if (img == NULL) fatal_error("save image: descriptor is NULL");
/* 5 Dec 2001: add a check on img */
if (img->magic != MAGIC_OF_IMAGE) {
fprintf(stderr, "TGASAVE: need Dead Beef!\n");
return NOT_AN_IMAGE_DESC;
}
/*
* mais c,a marche pas a tous les coups. il faudrait pouvoir tester si
* le pointeur 'img' est coherent, mais je sais pas comment faire...
*/
if (compress != 0) {
fprintf(stderr, "%s: compress %d not implemented\n",
__func__, compress);
}
/*
10 Aout 2000: si le nom est "-", alors on ecrit le fichier
sur STDOUT ce qui permet(tra) de faire des CGI :)
*/
if (strcmp("-", nom)) {
if ( (fp=fopen(nom, "wb")) == NULL )
{
perror(nom);
return FILE_CREATE_ERR;
/* fatal_error("save TGA image: error fopen"); */
}
}
else {
fp = stdout;
}
ltxt = strlen(img->comment);
#if DEBUG_LEVEL > 1
fprintf(stderr, "text size = %d\n", ltxt);
if (ltxt)
fprintf(stderr, " [%s]\n", img->comment);
#endif
/* first, we populate the file header... */
head.text_size = ltxt;
head.x_start = head.y_start = 0;
head.width = img->width;
head.height = img->height;
head.o_flags = 0; /* XXX a preciser ... */
#if DEBUG_LEVEL > 1
fprintf(stderr, "TGASAVE: img@ %p nb planes: %d\n", img, img->nb_planes);
#endif
switch (img->nb_planes) {
case 1: /* images en niveau de gris */
#if DEBUG_LEVEL
fprintf(stderr, "%s saving gray level image\n", __func__);
#endif
head.type = 3;
head.bits = 8;
break;
case 3:
head.type = 2; /* uncompressed RGB image */
head.bits = 24;
head.is_color_map = 0; /* No */
head.map_start = 0; /* Color */
head.map_length = 0; /* Map */
head.map_bits = 0; /* ? (Aout 1999) */
break;
case 4:
#if DEBUG_LEVEL
fputs("*!* saving image with alpha channel\n", stderr);
#endif
head.type = 2; /* uncompressed RGB image with alpha */
head.bits = 32;
head.is_color_map = 0; /* No */
head.map_start = 0; /* Color */
head.map_length = 0; /* Map */
head.map_bits = 0;
head.o_flags = 0x08; /* fixed by Betatech */
break;
default:
fprintf(stderr, "img->nb_planes = %d\n", img->nb_planes);
fatal_error("Tga_save_image: unknow type");
break;
}
/*
* 2 octobre 2001 - prise en main du BOUTISME
*
fwrite(&head, sizeof(head), 1, fp);
*
* 5 oct 01: il faudrait faire une fonction genre 'ecrire_header_tga'
*/
fwrite (&head.text_size, 1, 1, fp);
fwrite (&head.is_color_map, 1, 1, fp);
fwrite (&head.type, 1, 1, fp);
Image_basicIO_write_I_short(fp, head.map_start);
Image_basicIO_write_I_short(fp, head.map_length);
fwrite (&head.map_bits, 1, 1, fp);
Image_basicIO_write_I_short(fp, head.x_start);
Image_basicIO_write_I_short(fp, head.y_start);
Image_basicIO_write_I_short(fp, head.width);
Image_basicIO_write_I_short(fp, head.height);
fwrite (&head.bits, 1, 1, fp);
fwrite (&head.o_flags, 1, 1, fp);
/* if we have a comment, write it to the file */
if (ltxt) {
fwrite(img->comment, 1, ltxt, fp);
}
if ((buffer=calloc(img->width, 4)) == NULL) {
fatal_error("no memory buffer for save operation");
}
/* XXX why ?
memset(buffer, 0x55, sizeof(uint8_t)*4*img->width);
*/
/* Ahem, now, write all the picture datas to the file... */
for (line=(img->height-1); line>=0; line--) {
switch (img->nb_planes) {
case 1:
fwrite(img->Rpix[line], 1, img->width, fp);
break;
case 3:
Rptr = img->Rpix[line];
Gptr = img->Gpix[line];
Bptr = img->Bpix[line];
ptr = buffer;
for (foo=0; foo<img->width; foo++) {
*ptr++ = Bptr[foo];
*ptr++ = Gptr[foo];
*ptr++ = Rptr[foo];
}
fwrite(buffer, 3, img->width, fp);
break;
case 4:
Rptr = img->Rpix[line];
Gptr = img->Gpix[line];
Bptr = img->Bpix[line];
Aptr = img->Apix[line];
ptr = buffer;
for (foo=0; foo<img->width; foo++) {
*ptr++ = Bptr[foo];
*ptr++ = Gptr[foo];
*ptr++ = Rptr[foo];
*ptr++ = Aptr[foo];
}
fwrite(buffer, 4, img->width, fp);
break;
default:
fatal_error("TGA save image: unknow type...");
break;
}
}
free(buffer); /* 1 Fev 00 */
img->modified = 0;
if (fp != stdout) fclose(fp);
return OLL_KORRECT;
}
/*::------------------------------------------------------------------::*/
/* new 31 Jan 2002
* save one of the component as a gray TGA file
*
* 1 mars 2002: in fact, it was a palettized 8 bits file
*
*/
int Image_TGA_save_component(char *nom, Image_Desc *img, char channel, int comp)
{
int ltxt, line, foo;
FILE *fp;
uint8_t colormap[256*3];
Tga_file_header head;
if (img->magic != MAGIC_OF_IMAGE) {
fprintf(stderr, "TGASAVE: need Dead Beef!\n");
return NOT_AN_IMAGE_DESC;
}
if (comp != 0) {
fprintf(stderr, "%s : param 'comp' %d invalid\n", __func__, comp);
return BAD_PARAMETER;
}
if ( (fp=fopen(nom, "wb")) == NULL ) {
fatal_error("save image: err fopen");
}
ltxt = strlen(img->comment);
head.text_size = ltxt;
head.x_start = head.y_start = 0;
head.width = img->width;
head.height = img->height;
head.is_color_map = 1;
head.o_flags = 0; /* XXX a preciser ... */
head.type = 3; /* Grayscale Image */
head.map_start = 0;
head.map_length = 256;
head.map_bits = 24;
head.bits = 8;
fwrite (&head.text_size, 1, 1, fp);
fwrite (&head.is_color_map, 1, 1, fp);
fwrite (&head.type, 1, 1, fp);
Image_basicIO_write_I_short(fp, head.map_start);
Image_basicIO_write_I_short(fp, head.map_length);
fwrite (&head.map_bits, 1, 1, fp);
Image_basicIO_write_I_short(fp, head.x_start);
Image_basicIO_write_I_short(fp, head.y_start);
Image_basicIO_write_I_short(fp, head.width);
Image_basicIO_write_I_short(fp, head.height);
fwrite (&head.bits, 1, 1, fp);
fwrite (&head.o_flags, 1, 1, fp);
/* if we have a comment, write it to the file */
if (ltxt) fwrite(img->comment, 1, ltxt, fp);
/* now, populate the colormap */
for (foo=0; foo<256; foo++) {
colormap[foo*3+0] = foo;
colormap[foo*3+1] = foo;
colormap[foo*3+2] = foo;
}
/* and write the colormap */
fwrite (colormap, 256, 3, fp);
for (line=(img->height-1); line>=0; line--) {
switch (channel) {
case 'r': case 'R':
fwrite(img->Rpix[line], 1, img->width, fp);
break;
case 'g': case 'G':
fwrite(img->Gpix[line], 1, img->width, fp);
break;
case 'b': case 'B':
fwrite(img->Bpix[line], 1, img->width, fp);
break;
}
}
fclose(fp);
fprintf(stderr, "AIE! this func (tga save component as gray) is bogus\n");
return FUNC_NOT_FINISH;
}
/*::------------------------------------------------------------------::*/
/* new 16 fevrier 2009 - ave St Exupery */
static char * tgatype2txt(int type)
{
switch (type)
{
case 2: return "Uncomp RGB";
case 3: return "Uncomp Gray";
case 4: return "Uncomp RGB + alpha";
default: return "wtf?";
}
return "Gni ?";
}
/*::------------------------------------------------------------------::*/
/* new 17 mars 2007 */
int
Image_TGA_print_header(Tga_file_header *ph)
{
if (NULL == ph) {
fprintf(stderr, "null head ptr in '%s'\n", __func__);
return 1;
}
printf("+-- TGA v %s header @ %p ---\n", LIBTGA_VERSION, ph);
printf("| comment size %d\n", ph->text_size);
printf("| is color map ? %d\n", ph->is_color_map);
printf("| type %d %s\n",
ph->type, tgatype2txt(ph->type));
printf("| map start %d\n", ph->map_start);
printf("| map length %d\n", ph->map_length);
printf("| map bits %d\n", ph->map_bits);
printf("| x start %d\n", ph->x_start);
printf("| y start %d\n", ph->y_start);
printf("| width %d\n", ph->width);
printf("| height %d\n", ph->height);
printf("| nbr of bits %d\n", ph->bits);
printf("| o_flags $%02X\n", ph->o_flags);
printf("+---------------------------\n");
return 0;
}
/*::------------------------------------------------------------------::*/
/* new 13 mars 2007 */
int Image_TGA_read_header(FILE *fp, void *pvoid)
{
Tga_file_header *phead;
int foo;
phead = (Tga_file_header *)pvoid;
foo = fread(&phead->text_size, 1, 1, fp);
if (foo != 1) {
fprintf(stderr, "TGA READ HEAD: err read first byte of header\n");
return VERY_STRANGE;
}
fread(&phead->is_color_map, 1, 1, fp);
fread(&phead->type, 1, 1, fp);
foo = Image_basicIO_read_I_short(fp, &phead->map_start);
/* fprintf(stderr, "foo=%d\n", foo); */
Image_basicIO_read_I_short(fp, &phead->map_length);
fread(&phead->map_bits, 1, 1, fp);
Image_basicIO_read_I_short(fp, &phead->x_start);
Image_basicIO_read_I_short(fp, &phead->y_start);
phead->width = phead->height = -1;
Image_basicIO_read_I_short(fp, &phead->width);
Image_basicIO_read_I_short(fp, &phead->height);
if ( (phead->width<1) || (phead->width<1) ) {
fprintf(stderr, "HEAD TGA: dimension < 1: %d %d\n",
phead->width, phead->height);
return VERY_STRANGE;
}
fread(&phead->bits, 1, 1, fp);
if ( fread(&phead->o_flags, 1, 1, fp) != 1) {
fprintf(stderr, "HEAD TGA: end of file before time...\n");
return VERY_STRANGE;
}
#if DEBUG_LEVEL > 1
Image_TGA_print_header(phead);
#endif
return OLL_KORRECT;
}
/*::------------------------------------------------------------------::*/
/* new 1er mai 2007 (la france qui bosse tout le temps) */
int Image_TGA_show_header(char *filename, int flag)
{
FILE *fp;
Tga_file_header header;
int foo;
#if DEBUG_LEVEL
fprintf(stderr, "%s ( %p 0x%04x )\n", __func__, filename, flag);
#endif
if ( NULL == (fp=fopen(filename, "rb")) ) {
fprintf(stderr, "can't open %s\n", filename);
return FILE_NOT_FOUND;
}
foo = Image_TGA_read_header(fp, &header);
if (foo) {
fprintf(stderr, "lecture header -> %d\n", foo);
}
foo = Image_TGA_print_header(&header);
if (foo)
{
;
}
return FUNC_NOT_FINISH;
}
/*::------------------------------------------------------------------::*/
/* new 13 mars 2007 */
int Image_TGA_get_dims(char *path, int *pwidth, int *pheight)
{
FILE *fp;
Tga_file_header head;
int foo;
*pwidth = *pheight = 0;
if ( (fp=fopen(path, "rb")) == NULL ) {
fprintf(stderr, "%s: file '%s' is not readable\n", __func__, path);
return FILE_NOT_FOUND;
}
foo = Image_TGA_read_header(fp, &head);
fclose(fp);
if (foo) {
fprintf(stderr, "%s: lecture header '%s' -> %d\n",
__func__, path, foo);
}
if (OLL_KORRECT != foo) {
*pwidth = *pheight = 0;
return foo;
}
/* verification de coerence du header lu */
if (head.is_color_map!=0 && head.is_color_map!=1)
{
fprintf(stderr, "suspicious 'is_color_map' 0x%02x\n",
head.is_color_map);
}
*pwidth = (int)head.width;
*pheight = (int)head.height;
#if DEBUG_LEVEL
fprintf(stderr, "%s --> %d x %d\n", path, (int)head.width, (int)head.height);
#endif
return OLL_KORRECT;
}
/*::------------------------------------------------------------------::*/
/*
* pour le moment, cette fonction ne charge
* que les images RGB. il manque le GRIS !!!
* ! il manque aussi la decompression !
*/
Image_Desc * Image_TGA_alloc_load(char *nom)
{
Tga_file_header head;
FILE *fp;
Image_Desc *image = NULL;
uint8_t *buffer;
int ligne, ligne2, col, foo, alpha;
int imagetype;
if ( (fp=fopen(nom, "rb")) == NULL )
{
fprintf(stderr, "%s : file '%s' not readable\n", __func__, nom);
return NULL;
}
memset(&head, 0, sizeof(Tga_file_header));
foo = Image_TGA_read_header(fp, &head);
if (foo) {
fprintf(stderr, "%s: read error %d on header\n", __func__, foo);
fclose(fp);
return NULL;
}
#if DEBUG_LEVEL > 1
printf("TGA ALLOC LOAD: read header -> %d\n", foo);
printf("comment size %d\n", head.text_size);
printf("is color map ? %d\n", head.is_color_map);
printf("type %d\n", head.type);
printf("map start %d\n", head.map_start);
printf("map length %d\n", head.map_length);
printf("map bits %d\n", head.map_bits);
printf("x start %d\n", head.x_start);
printf("y start %d\n", head.y_start);
printf("width %d\n", head.width);
printf("height %d\n", head.height);
printf("nbr of bits %d\n", head.bits);
#endif
switch (head.type)
{
case 2:
#if DEBUG_LEVEL > 1
fprintf(stderr, "load file %s, type 2, %d bits\n", nom, head.map_bits);
#endif
if (head.bits == 24) {
alpha = 0; imagetype = IMAGE_RGB;
}
else {
alpha = 1; imagetype = IMAGE_RGBA;
}
if ( (image=Image_alloc(head.width, head.height, 3+alpha))==NULL)
fatal_error("no memory for picture in 'alloc_load'\n");
if ( (buffer=(uint8_t *)malloc(4*head.width))==NULL)
fatal_error("no memory for buffer in 'alloc_load'\n");
#if DEBUG_LEVEL > 1
fprintf(stderr, " ... mem allocated\n");
if (head.o_flags & 0x20)
fprintf(stderr, "flags= 0x%02x Upside-Down ?\n", head.o_flags);
if (head.o_flags & 0x10)
fprintf(stderr, "flags= 0x%02x Left-Right ?\n", head.o_flags);
#endif
if (head.text_size)
{
#if DEBUG_LEVEL > 1
fprintf(stderr,"%s: skipping %d bytes of comment\n",
nom, head.text_size);
#endif
fseek(fp, (long)head.text_size, SEEK_CUR);
}
for (ligne=0; ligne<head.height; ligne++)
{
foo=fread(buffer, 3+alpha, head.width, fp);
if (!(head.o_flags & 0x20)) ligne2 = head.height - ligne - 1;
else ligne2 = ligne;
for (col=0; col<head.width; col++)
{
(image->Bpix[ligne2])[col] = buffer[ col*(3+alpha) ];
(image->Gpix[ligne2])[col] = buffer[ col*(3+alpha) + 1 ];
(image->Rpix[ligne2])[col] = buffer[ col*(3+alpha) + 2 ];
if (alpha)
(image->Apix[ligne2])[col] = buffer[ col*4 + 3 ];
}
}
#if DEBUG_LEVEL > 1
fprintf(stderr, " ... image loaded\n\n");
#endif
break;
default:
fprintf(stderr, "%s: file '%s' type %d not supported\n\n",
__func__, nom, head.type);
fclose (fp);
return NULL;
}
free(buffer); /* added 17 May 2001 :) */
fclose(fp);
/*-------- image is in memory... return @ of descriptor to caller. */
return image;
}
/*::------------------------------------------------------------------::*/
/* c'est la fin... drink a beer :) */
Reference in New Issue View Git Blame Copy Permalink