dumptga.cpp

Go to the documentation of this file.

/* xoreos - A reimplementation of BioWare's Aurora engine
 *
 * xoreos is the legal property of its developers, whose names
 * can be found in the AUTHORS file distributed with this source
 * distribution.
 *
 * xoreos 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 3
 * of the License, or (at your option) any later version.
 *
 * xoreos 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 xoreos. If not, see <http://www.gnu.org/licenses/>.
 */

#include <cstdio>

#include "src/common/scopedptr.h"
#include "src/common/error.h"
#include "src/common/ustring.h"
#include "src/common/writefile.h"

#include "src/graphics/images/decoder.h"

namespace Graphics {

static void writePixel(Common::WriteStream &file, const byte *&data, PixelFormat format) {
    if (format == kPixelFormatRGB) {
        file.writeByte(data[2]);
        file.writeByte(data[1]);
        file.writeByte(data[0]);
        file.writeByte(0xFF);
        data += 3;
    } else if (format == kPixelFormatBGR) {
        file.writeByte(data[0]);
        file.writeByte(data[1]);
        file.writeByte(data[2]);
        file.writeByte(0xFF);
        data += 3;
    } else if (format == kPixelFormatRGBA) {
        file.writeByte(data[2]);
        file.writeByte(data[1]);
        file.writeByte(data[0]);
        file.writeByte(data[3]);
        data += 4;
    } else if (format == kPixelFormatBGRA) {
        file.writeByte(data[0]);
        file.writeByte(data[1]);
        file.writeByte(data[2]);
        file.writeByte(data[3]);
        data += 4;
    } else
        throw Common::Exception("Unsupported pixel format: %d", (int) format);

}

static Common::WriteStream *openTGA(const Common::UString &fileName, int width, int height) {
    Common::WriteFile *file = new Common::WriteFile(fileName);

    file->writeByte(0);     // ID Length
    file->writeByte(0);     // Palette size
    file->writeByte(2);     // Unmapped RGB
    file->writeUint32LE(0); // Color map
    file->writeByte(0);     // Color map
    file->writeUint16LE(0); // X
    file->writeUint16LE(0); // Y

    file->writeUint16LE(width);
    file->writeUint16LE(height);

    file->writeByte(32); // Pixel depths

    file->writeByte(0);

    return file;
}

static void writeMipMap(Common::WriteStream &stream, const ImageDecoder::MipMap &mipMap, PixelFormat format) {
    const byte *data = mipMap.data.get();

    uint32 count = mipMap.width * mipMap.height;
    while (count-- > 0)
        writePixel(stream, data, format);
}

void dumpTGA(const Common::UString &fileName, const ImageDecoder *image) {
    if (!image || (image->getLayerCount() < 1) || (image->getMipMapCount() < 1))
        throw Common::Exception("No image");

    int32 width  = image->getMipMap(0, 0).width;
    int32 height = 0;

    for (size_t i = 0; i < image->getLayerCount(); i++) {
        const ImageDecoder::MipMap &mipMap = image->getMipMap(0, i);

        if (mipMap.width != width)
            throw Common::Exception("dumpTGA(): Unsupported image with variable layer width");

        height += mipMap.height;
    }

    Common::ScopedPtr<Common::WriteStream> file(openTGA(fileName, width, height));

    for (size_t i = 0; i < image->getLayerCount(); i++)
        writeMipMap(*file, image->getMipMap(0, i), image->getFormat());

    file->flush();
}

} // End of namespace Graphics