OpenRaider/src/utils/tga.cpp

/*!
 * \file src/utils/tga.cpp
 * \brief TGA image reader/writer
 *
 * \todo type should pass more info (2 bits for RGBA|RGB|GREY; val for depth)
 *
 * \author Mongoose
 * \author xythobuz
 */

#include <stdio.h>
#include <string.h>
#include <stdarg.h>
#include <assert.h>

#include "utils/tga.h"

int tgaCheck(FILE *f) {
    char buffer[10];

    assert(f != NULL);

    /* Read the header */
    fseek(f, 0, SEEK_SET);
    fread(buffer, 8, 1, f);

    // buffer[1] = 0 - Means not color mapped (1 would mean mapped)
    if (!(buffer[1] == 0 && (buffer[2] == TGA_TYPE__COLOR ||
                    //buffer[2] == TGA_TYPE__GREYSCALE ||
                    buffer[2] == TGA_TYPE__COLOR_RLE))) {
        printf("tgaCheck> Inavlid or unknown TGA format.\n");
        return -2;
    }
    return 0;
}

int tgaLoad(FILE *f, unsigned char **image, unsigned int *width, unsigned int *height, char *type) {
    tga_t header;
    char comment[256];
    unsigned char pixel[4];
    unsigned char *swap_row = NULL;
    unsigned char tmp, packet;
    bool must_flip = 0;
    unsigned int size;
    unsigned int i, j;

    assert(f != NULL);
    assert(image != NULL);
    assert(width != NULL);
    assert(height != NULL);
    assert(type != NULL);

    fseek(f, 0, SEEK_SET);

    // Read TGA header
    fread(&header.comment_lenght, 1, 1, f);
    fread(&header.colormap_type, 1, 1, f);
    fread(&header.image_type, 1, 1, f);
    fread(&header.colormap_index, 2, 1, f);
    fread(&header.colormap_lenght, 2, 1, f);
    fread(&header.colormap_bbp, 1, 1, f);

    fread(&tmp, 1, 1, f);
    header.origin_x = tmp;
    fread(&tmp, 1, 1, f);
    header.origin_x += tmp * 256;

    fread(&tmp, 1, 1, f);
    header.origin_y = tmp;
    fread(&tmp, 1, 1, f);
    header.origin_y += tmp * 256;

    fread(&tmp, 1, 1, f);
    header.width = tmp;
    fread(&tmp, 1, 1, f);
    header.width += tmp * 256;

    fread(&tmp, 1, 1, f);
    header.height = tmp;
    fread(&tmp, 1, 1, f);
    header.height += tmp * 256;

    fread(&header.bpp, 1, 1, f);
    fread(&header.desc_flags, 1, 1, f);

    *width = header.width;
    *height = header.height;

    switch (header.bpp) {
        case 32:
            *type = 2; //32;
            break;
        case 24:
            *type = 1; //24;
            break;
        case 8:
        default:
            *type = 0; //8;
            break;
    }

#ifdef DEBUG_TGA
    printf("TGA [%ix%i@%ibpp, %it, %ix, %iy, %uf]\n",
            header.width, header.height, header.bpp, header.image_type,
            header.origin_x, header.origin_y,
            header.desc_flags);
#endif

    // Comments can be 0 - 255
    if (header.comment_lenght) {
        fread(&comment, 1, header.comment_lenght, f);
        for (i = 0; i < 255; ++i) {
            if (!(comment[i] > 32 && comment[i] < 127))
                comment[i] = 183; // print a dot for invalid text
        }
        comment[255] = 0;
        printf("Comment: '%s'\n", comment);
    }

    *image = NULL;
    size = header.width * header.height;

    if (!size || (!(header.colormap_type == 0 && (header.image_type == 2 || header.image_type == 10)))) {
        fprintf(stderr, "tgaLoad> Unknown image format.\n");
        return -2;
    }

    // Mongoose: Added 'screen origin bit' support back here
    if (!(header.desc_flags & 32)) {
        must_flip = true;
    }

    switch (header.bpp) {
        case 32:
            size *= 4;
            *image = new unsigned char [size];
            switch (header.image_type) {
                case TGA_TYPE__COLOR_RLE:
                    for (i = 0; i < size;) {
                        fread(&packet, 1, 1, f);
                        if (packet & 0x80) { // Run Length
                            packet = (packet &0x7F) + 1;
                            fread(&pixel, 4, 1, f);
                            for (j = 0; j < packet; j++) {
                                (*image)[i++] = pixel[2];
                                (*image)[i++] = pixel[1];
                                (*image)[i++] = pixel[0];
                                (*image)[i++] = pixel[3];
                            }
                        } else { // RAW
                            packet = (packet &0x7F) + 1;
                            for (j = 0; j < packet; j++) {
                                fread(&pixel, 4, 1, f);
                                (*image)[i++] = pixel[2];
                                (*image)[i++] = pixel[1];
                                (*image)[i++] = pixel[0];
                                (*image)[i++] = pixel[3];
                            }
                        }
                    }
                    break;
                case TGA_TYPE__COLOR:
                    if (fread((*image), size, 1, f) < 1) {
                        fprintf(stderr, "tgaLoad> Image fread failed.\n");
                        delete [] *image;
                        return -4;
                    }
                    for (i = 0; i < size; i += 4) {
                        tmp = (*image)[i];
                        (*image)[i] = (*image)[i + 2];
                        (*image)[i + 2] = tmp;
                    }
                    break;
                default:
                    ;
            }
            if (must_flip) {
                swap_row = new unsigned char [header.width * 4];
                for (i = 0, j = header.height-1; (int)i < header.height/2; i++, j--) {
                    memcpy(swap_row, &(*image)[i*header.width*4], header.width*4);
                    memcpy(&(*image)[i*header.width*4], &(*image)[j*header.width*4],
                            header.width*4);
                    memcpy(&(*image)[j*header.width*4], swap_row, header.width*4);
                }
                delete [] swap_row;
            }
            break;
        case 24:
            size *= 3;
            *image = new unsigned char [size];
            switch (header.image_type) {
                case TGA_TYPE__COLOR_RLE:
                    for (i = 0; i < size;) {
                        fread(&packet, 1, 1, f);
                        if (packet & 0x80) { // Run Length
                            packet = (packet &0x7F) + 1;
                            fread(&pixel, 3, 1, f);
                            for (j = 0; j < packet; j++) {
                                (*image)[i++] = pixel[2];
                                (*image)[i++] = pixel[1];
                                (*image)[i++] = pixel[0];
                            }
                        } else { // RAW
                            packet = (packet &0x7F) + 1;
                            for (j = 0; j < packet; j++) {
                                fread(&pixel, 3, 1, f);
                                (*image)[i++] = pixel[2];
                                (*image)[i++] = pixel[1];
                                (*image)[i++] = pixel[0];
                            }
                        }
                    }
                    break;
                case TGA_TYPE__COLOR:
                    if (fread((*image), size, 1, f) < 1) {
                        fprintf(stderr, "tgaLoad> Image fread failed.\n");
                        delete [] *image;
                        return -4;
                    }
                    for (i = 0; i < size; i += 3) {
                        tmp = (*image)[i];
                        (*image)[i] = (*image)[i + 2];
                        (*image)[i + 2] = tmp;
                    }
                    break;
                default:
                    ;
            }
            if (must_flip) {
                swap_row = new unsigned char [header.width * 3];
                for (i = 0, j = header.height - 1; (int)i < header.height / 2; i++, j--) {
                    memcpy(swap_row, &(*image)[i*header.width*3], header.width*3);
                    memcpy(&(*image)[i*header.width*3], &(*image)[j*header.width*3],
                            header.width*3);
                    memcpy(&(*image)[j*header.width*3], swap_row, header.width*3);
                }
                delete [] swap_row;
            }
            break;
        case 8:
            printf("tgaLoad> 8bpp Not implemented\n");
            break;
        default:
            ;
    }

#ifdef DEBUG_TGA
    char c;
    printf("TGA Comment: ");
    while (fread(&c, 1, 1, f) == 1) {
        printf("%c", c);
    }
    printf("\n");
#endif

    return 0;
}

int tgaSave(FILE *f, unsigned char *image, unsigned int width, unsigned int height, char type) {
    tga_t header;
    unsigned int size;
    char comment[64];
    //unsigned int i;
    //unsigned char tmp;

    assert(f != NULL);
    assert(image != NULL);
    assert(width > 0);
    assert(height > 0);

    strncpy(comment, "OpenRaider TGA", 63);
    comment[63] = 0;

    header.comment_lenght = (unsigned char)strlen(comment);
    header.colormap_type = 0;

    // No colormaps
    header.colormap_index = 0;
    header.colormap_lenght = 0;
    header.colormap_bbp = 0;

    header.origin_x = header.origin_y = 0;
    header.width = (unsigned short)width;
    header.height = (unsigned short)height;

    header.desc_flags = 0;

    switch (type) {
        case 4:
            header.image_type = TGA_TYPE__COLOR;
            header.desc_flags = 32;
            header.bpp = 32;
            break;
        case 2:
            header.bpp = 32;
            break;
        case 1:
            header.image_type = TGA_TYPE__GREYSCALE;
            header.bpp = 8;
            break;
        default:
            header.image_type = TGA_TYPE__COLOR;
            header.bpp = 24;
    }

    // Write TGA header
    fwrite(&header.comment_lenght, 1, 1, f);
    fwrite(&header.colormap_type, 1, 1, f);
    fwrite(&header.image_type, 1, 1, f);
    fwrite(&header.colormap_index, 2, 1, f);
    fwrite(&header.colormap_lenght, 2, 1, f);
    fwrite(&header.colormap_bbp, 1, 1, f);
    fwrite(&header.origin_x, 2, 1, f);
    fwrite(&header.origin_y, 2, 1, f);
    fwrite(&header.width, 2, 1, f);
    fwrite(&header.height, 2, 1, f);
    fwrite(&header.bpp, 1, 1, f);
    fwrite(&header.desc_flags, 1, 1, f);

    // Write comment
    fwrite(&comment, 1, header.comment_lenght, f);

    switch (header.bpp) {
        case 32:
            size = header.width * header.height * 4;

            //for (i = 0; i < size; i += 4)
            //{
            //  tmp = image[i];
            //  image[i] = image[i + 2];
            //  image[i + 2] = tmp;
            //}
            break;
        case 24:
            size = header.width * header.height * 3;

            //for (i = 0; i < size; i += 3)
            //{
            //  tmp = image[i];
            //  image[i] = image[i + 2];
            //  image[i + 2] = tmp;
            //}
            break;
        case 8:
        default:
            size = header.width * header.height;
            break;
    }

    // Write image data
    if (fwrite(image, size, 1, f) < 1) {
        perror("tgaSave> Disk write failed.\n");
        return -2;
    }
    return 0;
}

int tgaSaveFilename(unsigned char *image, unsigned int width, unsigned int height, char type, const char *s, ...) {
    char buffer[1024];
    FILE *f;
    int v;
    va_list args;

    assert(image != NULL);
    assert(width > 0);
    assert(height > 0);
    assert(s != NULL);
    assert(s[0] != '\0');

    va_start(args, s);
    vsnprintf(buffer, 1023, s, args);
    va_end(args);
    f = fopen(buffer, "wb");
    if (!f) {
        perror(buffer);
        return -1;
    }
    v = tgaSave(f, image, width, height, type);
    fclose(f);
    return v;
}