worldspawn/plugins/image/tga.cpp

/*
   Copyright (C) 2001-2006, William Joseph.
   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 "tga.h"

#include "ifilesystem.h"
#include "iarchive.h"
#include "idatastream.h"

typedef unsigned char byte;

#include <stdlib.h>

#include "generic/bitfield.h"
#include "imagelib.h"
#include "bytestreamutils.h"

// represents x,y origin of tga image being decoded
class Flip00 {}; // no flip
class Flip01 {}; // vertical flip only
class Flip10 {}; // horizontal flip only
class Flip11 {}; // both

template<typename PixelDecoder>
void image_decode(PointerInputStream &istream, PixelDecoder &decode, RGBAImage &image, const Flip00 &)
{
    RGBAPixel *end = image.pixels + (image.height * image.width);
    for (RGBAPixel *row = end; row != image.pixels; row -= image.width) {
        for (RGBAPixel *pixel = row - image.width; pixel != row; ++pixel) {
            decode(istream, *pixel);
        }
    }
}

template<typename PixelDecoder>
void image_decode(PointerInputStream &istream, PixelDecoder &decode, RGBAImage &image, const Flip01 &)
{
    RGBAPixel *end = image.pixels + (image.height * image.width);
    for (RGBAPixel *row = image.pixels; row != end; row += image.width) {
        for (RGBAPixel *pixel = row; pixel != row + image.width; ++pixel) {
            decode(istream, *pixel);
        }
    }
}

template<typename PixelDecoder>
void image_decode(PointerInputStream &istream, PixelDecoder &decode, RGBAImage &image, const Flip10 &)
{
    RGBAPixel *end = image.pixels + (image.height * image.width);
    for (RGBAPixel *row = end; row != image.pixels; row -= image.width) {
        for (RGBAPixel *pixel = row; pixel != row - image.width;) {
            decode(istream, *--pixel);
        }
    }
}

template<typename PixelDecoder>
void image_decode(PointerInputStream &istream, PixelDecoder &decode, RGBAImage &image, const Flip11 &)
{
    RGBAPixel *end = image.pixels + (image.height * image.width);
    for (RGBAPixel *row = image.pixels; row != end; row += image.width) {
        for (RGBAPixel *pixel = row + image.width; pixel != row;) {
            decode(istream, *--pixel);
        }
    }
}

inline void istream_read_gray(PointerInputStream &istream, RGBAPixel &pixel)
{
    istream.read(&pixel.blue, 1);
    pixel.red = pixel.green = pixel.blue;
    pixel.alpha = 0xff;
}

inline void istream_read_rgb(PointerInputStream &istream, RGBAPixel &pixel)
{
    istream.read(&pixel.blue, 1);
    istream.read(&pixel.green, 1);
    istream.read(&pixel.red, 1);
    pixel.alpha = 0xff;
}

inline void istream_read_rgba(PointerInputStream &istream, RGBAPixel &pixel)
{
    istream.read(&pixel.blue, 1);
    istream.read(&pixel.green, 1);
    istream.read(&pixel.red, 1);
    istream.read(&pixel.alpha, 1);
}

class TargaDecodeGrayPixel {
public:
    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        istream_read_gray(istream, pixel);
    }
};

template<typename Flip>
void targa_decode_grayscale(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeGrayPixel decode;
    image_decode(istream, decode, image, flip);
}

class TargaDecodeRGBPixel {
public:
    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        istream_read_rgb(istream, pixel);
    }
};

template<typename Flip>
void targa_decode_rgb(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeRGBPixel decode;
    image_decode(istream, decode, image, flip);
}

class TargaDecodeRGBAPixel {
public:
    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        istream_read_rgba(istream, pixel);
    }
};

template<typename Flip>
void targa_decode_rgba(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeRGBAPixel decode;
    image_decode(istream, decode, image, flip);
}

typedef byte TargaPacket;
typedef byte TargaPacketSize;

inline void targa_packet_read_istream(TargaPacket &packet, PointerInputStream &istream)
{
    istream.read(&packet, 1);
}

inline bool targa_packet_is_rle(const TargaPacket &packet)
{
    return (packet & 0x80) != 0;
}

inline TargaPacketSize targa_packet_size(const TargaPacket &packet)
{
    return 1 + (packet & 0x7f);
}


class TargaDecodeGrayPixelRLE {
    TargaPacketSize m_packetSize;
    RGBAPixel m_pixel;
    TargaPacket m_packet;
public:
    TargaDecodeGrayPixelRLE() : m_packetSize(0)
    {
    }

    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        if (m_packetSize == 0) {
            targa_packet_read_istream(m_packet, istream);
            m_packetSize = targa_packet_size(m_packet);

            if (targa_packet_is_rle(m_packet)) {
                istream_read_gray(istream, m_pixel);
            }
        }

        if (targa_packet_is_rle(m_packet)) {
            pixel = m_pixel;
        } else {
            istream_read_gray(istream, pixel);
        }

        --m_packetSize;
    }
};

template<typename Flip>
void targa_decode_rle_grayscale(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeGrayPixelRLE decode;
    image_decode(istream, decode, image, flip);
}

class TargaDecodeRGBPixelRLE {
    TargaPacketSize m_packetSize;
    RGBAPixel m_pixel;
    TargaPacket m_packet;
public:
    TargaDecodeRGBPixelRLE() : m_packetSize(0)
    {
    }

    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        if (m_packetSize == 0) {
            targa_packet_read_istream(m_packet, istream);
            m_packetSize = targa_packet_size(m_packet);

            if (targa_packet_is_rle(m_packet)) {
                istream_read_rgb(istream, m_pixel);
            }
        }

        if (targa_packet_is_rle(m_packet)) {
            pixel = m_pixel;
        } else {
            istream_read_rgb(istream, pixel);
        }

        --m_packetSize;
    }
};

template<typename Flip>
void targa_decode_rle_rgb(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeRGBPixelRLE decode;
    image_decode(istream, decode, image, flip);
}

class TargaDecodeRGBAPixelRLE {
    TargaPacketSize m_packetSize;
    RGBAPixel m_pixel;
    TargaPacket m_packet;
public:
    TargaDecodeRGBAPixelRLE() : m_packetSize(0)
    {
    }

    void operator()(PointerInputStream &istream, RGBAPixel &pixel)
    {
        if (m_packetSize == 0) {
            targa_packet_read_istream(m_packet, istream);
            m_packetSize = targa_packet_size(m_packet);

            if (targa_packet_is_rle(m_packet)) {
                istream_read_rgba(istream, m_pixel);
            }
        }

        if (targa_packet_is_rle(m_packet)) {
            pixel = m_pixel;
        } else {
            istream_read_rgba(istream, pixel);
        }

        --m_packetSize;
    }
};

template<typename Flip>
void targa_decode_rle_rgba(PointerInputStream &istream, RGBAImage &image, const Flip &flip)
{
    TargaDecodeRGBAPixelRLE decode;
    image_decode(istream, decode, image, flip);
}

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;
};

inline void targa_header_read_istream(TargaHeader &targa_header, PointerInputStream &istream)
{
    targa_header.id_length = istream_read_byte(istream);
    targa_header.colormap_type = istream_read_byte(istream);
    targa_header.image_type = istream_read_byte(istream);

    targa_header.colormap_index = istream_read_int16_le(istream);
    targa_header.colormap_length = istream_read_int16_le(istream);
    targa_header.colormap_size = istream_read_byte(istream);
    targa_header.x_origin = istream_read_int16_le(istream);
    targa_header.y_origin = istream_read_int16_le(istream);
    targa_header.width = istream_read_int16_le(istream);
    targa_header.height = istream_read_int16_le(istream);
    targa_header.pixel_size = istream_read_byte(istream);
    targa_header.attributes = istream_read_byte(istream);

    if (targa_header.id_length != 0) {
        istream.seek(targa_header.id_length); // skip TARGA image comment
    }
}

template<typename Type>
class ScopeDelete {
    Type *m_value;

    ScopeDelete(const ScopeDelete &);

    ScopeDelete &operator=(const ScopeDelete &);

public:
    ScopeDelete(Type *value) : m_value(value)
    {
    }

    ~ScopeDelete()
    {
        delete m_value;
    }

    Type *get_pointer() const
    {
        return m_value;
    }
};

template<typename Flip>
Image *Targa_decodeImageData(const TargaHeader &targa_header, PointerInputStream &istream, const Flip &flip)
{
    RGBAImage *image = new RGBAImage(targa_header.width, targa_header.height);

    if (targa_header.image_type == 2 || targa_header.image_type == 3) {
        switch (targa_header.pixel_size) {
            case 8:
                targa_decode_grayscale(istream, *image, flip);
                break;
            case 24:
                targa_decode_rgb(istream, *image, flip);
                break;
            case 32:
                targa_decode_rgba(istream, *image, flip);
                break;
            default:
                globalErrorStream() << "LoadTGA: illegal pixel_size '" << targa_header.pixel_size << "'\n";
                image->release();
                return 0;
        }
    } else if (targa_header.image_type == 10 || targa_header.image_type == 11) {
        switch (targa_header.pixel_size) {
            case 8:
                targa_decode_rle_grayscale(istream, *image, flip);
                break;
            case 24:
                targa_decode_rle_rgb(istream, *image, flip);
                break;
            case 32:
                targa_decode_rle_rgba(istream, *image, flip);
                break;
            default:
                globalErrorStream() << "LoadTGA: illegal pixel_size '" << targa_header.pixel_size << "'\n";
                image->release();
                return 0;
        }
    }

    return image;
}

const unsigned int TGA_FLIP_HORIZONTAL = 0x10;
const unsigned int TGA_FLIP_VERTICAL = 0x20;

Image *LoadTGABuff(const byte *buffer)
{
    PointerInputStream istream(buffer);
    TargaHeader targa_header;

    targa_header_read_istream(targa_header, istream);

    if (targa_header.image_type != 2 && targa_header.image_type != 10 && targa_header.image_type != 3 &&
        targa_header.image_type != 11) {
        globalErrorStream() << "LoadTGA: TGA type " << targa_header.image_type << " not supported\n";
        globalErrorStream() << "LoadTGA: Only type 2 (RGB), 3 (gray), 10 (RGB), and 11 (gray) TGA images supported\n";
        return 0;
    }

    if (targa_header.colormap_type != 0) {
        globalErrorStream() << "LoadTGA: colormaps not supported\n";
        return 0;
    }

    if (((targa_header.image_type == 2 || targa_header.image_type == 10) && targa_header.pixel_size != 32 &&
         targa_header.pixel_size != 24) ||
        ((targa_header.image_type == 3 || targa_header.image_type == 11) && targa_header.pixel_size != 8)) {
        globalErrorStream() << "LoadTGA: Only 32, 24 or 8 bit images supported\n";
        return 0;
    }

    if (!bitfield_enabled(targa_header.attributes, TGA_FLIP_HORIZONTAL)
        && !bitfield_enabled(targa_header.attributes, TGA_FLIP_VERTICAL)) {
        return Targa_decodeImageData(targa_header, istream, Flip00());
    }
    if (!bitfield_enabled(targa_header.attributes, TGA_FLIP_HORIZONTAL)
        && bitfield_enabled(targa_header.attributes, TGA_FLIP_VERTICAL)) {
        return Targa_decodeImageData(targa_header, istream, Flip01());
    }
    if (bitfield_enabled(targa_header.attributes, TGA_FLIP_HORIZONTAL)
        && !bitfield_enabled(targa_header.attributes, TGA_FLIP_VERTICAL)) {
        return Targa_decodeImageData(targa_header, istream, Flip10());
    }
    if (bitfield_enabled(targa_header.attributes, TGA_FLIP_HORIZONTAL)
        && bitfield_enabled(targa_header.attributes, TGA_FLIP_VERTICAL)) {
        return Targa_decodeImageData(targa_header, istream, Flip11());
    }

    // unreachable
    return 0;
}

Image *LoadTGA(ArchiveFile &file)
{
    ScopedArchiveBuffer buffer(file);
    return LoadTGABuff(buffer.buffer);
}