tahoma2d/toonz/sources/image/tga/tiio_tga.cpp
Toshihiro Shimizu 890dddabbd first commit
2016-03-19 02:57:51 +09:00

790 lines
19 KiB
C++

#if _MSC_VER >= 1400
#define _CRT_SECURE_NO_DEPRECATE 1
#pragma warning(disable : 4996)
#endif
//#include "texception.h"
//#include "tfilepath.h"
#include "tiio_tga.h"
#include "tpixel.h"
//#include <iostream.h>
inline unsigned short fgetUSHORT(FILE *chan)
{
int b0 = fgetc(chan);
int b1 = fgetc(chan);
return b1 << 8 | b0;
}
/*
inline unsigned short fgetULONG(FILE *chan)
{
int a = fgetc(chan);
int b = fgetc(chan);
int c = fgetc(chan);
int d = fgetc(chan);
return a<<24|b<<16|c<<8|d;
}
*/
inline void fputUSHORT(unsigned short x, FILE *chan)
{
int b1 = x >> 8;
int b0 = x & 0xFF;
fputc(b0, chan);
fputc(b1, chan);
}
/*
inline void fputULONG(unsigned long x, FILE *chan)
{
int a = x>>24;
int b = (x>>16)&0xff;
int c = (x>>8)&0xff;
int d = x&0xFF;
fputc(a, chan);
fputc(b, chan);
fputc(c, chan);
fputc(d, chan);
}
*/
typedef unsigned char UCHAR;
typedef unsigned short USHORT;
struct TgaHeader {
UCHAR IdentificationField;
UCHAR ColorMapType;
UCHAR ImageTypeCode;
USHORT ColorMapOrigin;
USHORT ColorMapLength;
UCHAR ColorMapSize;
USHORT Ximage;
USHORT Yimage;
USHORT Width;
USHORT Height;
UCHAR ImagePixelSize;
UCHAR ImageDescriptor;
};
//============================================================
void readTgaHeader(TgaHeader &header, FILE *chan)
{
header.IdentificationField = fgetc(chan);
header.ColorMapType = fgetc(chan);
header.ImageTypeCode = fgetc(chan);
header.ColorMapOrigin = fgetUSHORT(chan);
header.ColorMapLength = fgetUSHORT(chan);
header.ColorMapSize = fgetc(chan);
header.Ximage = fgetUSHORT(chan);
header.Yimage = fgetUSHORT(chan);
header.Width = fgetUSHORT(chan);
header.Height = fgetUSHORT(chan);
header.ImagePixelSize = fgetc(chan);
header.ImageDescriptor = fgetc(chan);
}
//------------------------------------------------------------
void writeTgaHeader(TgaHeader &header, FILE *chan)
{
fputc(header.IdentificationField, chan);
fputc(header.ColorMapType, chan);
fputc(header.ImageTypeCode, chan);
fputUSHORT(header.ColorMapOrigin, chan);
fputUSHORT(header.ColorMapLength, chan);
fputc(header.ColorMapSize, chan);
fputUSHORT(header.Ximage, chan);
fputUSHORT(header.Yimage, chan);
fputUSHORT(header.Width, chan);
fputUSHORT(header.Height, chan);
fputc(header.ImagePixelSize, chan);
fputc(header.ImageDescriptor, chan);
}
//============================================================
class TgaReader : public Tiio::Reader
{
FILE *m_chan;
TgaHeader m_header;
TPixel32 *m_palette;
typedef void (TgaReader::*ReadLineProc)(char *buffer, int x0, int x1, int shrink);
ReadLineProc m_readLineProc;
typedef int (TgaReader::*SkipLineProc)(int count);
SkipLineProc m_skipLineProc;
public:
TgaReader()
: m_chan(0), m_readLineProc(&TgaReader::readNoLine), m_skipLineProc(&TgaReader::skipLines0), m_palette(0)
{
}
~TgaReader();
void open(FILE *file);
int skipLines0(int count)
{
int lineSize = m_info.m_lx;
fseek(m_chan, count * lineSize, SEEK_CUR);
return count;
}
int skipNoLines(int count) { return 0; };
void skipBytes(int count)
{
//fseek(m_chan, count, SEEK_CUR); //inefficiente se count è piccolo
for (int i = 0; i < count; i++)
getc(m_chan);
}
int skipLines32(int count)
{
skipLines0(4 * count);
return count;
}
int skipLines24(int count)
{
skipLines0(3 * count);
return count;
}
int skipLines16(int count)
{
skipLines0(2 * count);
return count;
}
inline TPixel32 fgetPixel32(FILE *chan)
{
TPixel32 pix;
pix.b = fgetc(chan);
pix.g = fgetc(chan);
pix.r = fgetc(chan);
pix.m = fgetc(chan);
return pix;
}
inline TPixel32 fgetPixel24(FILE *chan)
{
TPixel32 pix;
pix.b = fgetc(chan);
pix.g = fgetc(chan);
pix.r = fgetc(chan);
pix.m = 255;
return pix;
}
inline TPixel32 fgetPixel16(FILE *chan)
{
TPixel32 pix;
unsigned short s = fgetUSHORT(chan);
int b = s & 0x1f;
int g = (s >> 5) & 0x1f;
int r = (s >> 10) & 0x1f;
pix.r = r << 3 | r >> 2;
pix.g = g << 3 | g >> 2;
pix.b = b << 3 | b >> 2;
pix.m = 255;
return pix;
}
Tiio::RowOrder getRowOrder() const
{
return ((m_header.ImageDescriptor >> 5) & 1) == 0
? Tiio::BOTTOM2TOP
: Tiio::TOP2BOTTOM;
}
void readLine(char *buffer, int x0, int x1, int shrink)
{
(this->*m_readLineProc)(buffer, x0, x1, shrink);
}
int skipLines(int count)
{
return (this->*m_skipLineProc)(count);
}
void readPalette();
void readNoLine(char *buffer, int x0, int x1, int shrink) {}
void readLineRGB32(char *buffer, int x0, int x1, int shrink);
void readLineRGB32rle(char *buffer, int x0, int x1, int shrink);
void readLineRGB24(char *buffer, int x0, int x1, int shrink);
void readLineRGB24rle(char *buffer, int x0, int x1, int shrink);
void readLineRGB16(char *buffer, int x0, int x1, int shrink);
void readLineRGB16rle(char *buffer, int x0, int x1, int shrink);
void readLineGR8(char *buffer, int x0, int x1, int shrink);
void readLineGR8rle(char *buffer, int x0, int x1, int shrink);
void readLineCmapped(char *buffer, int x0, int x1, int shrink);
void readLineCmappedRle(char *buffer, int x0, int x1, int shrink);
};
//------------------------------------------------------------
TgaReader::~TgaReader()
{
delete m_palette;
}
//------------------------------------------------------------
void TgaReader::open(FILE *file)
{
{
try {
m_chan = file;
} catch (...) {
perror("uffa");
return;
}
readTgaHeader(m_header, m_chan);
m_info.m_lx = m_header.Width;
m_info.m_ly = m_header.Height;
if (m_header.IdentificationField > 0) {
char tmp[256];
fread(tmp, 1, m_header.IdentificationField, m_chan);
}
if (m_header.ColorMapType > 0) {
readPalette();
}
Tiio::TgaWriterProperties *prop = new Tiio::TgaWriterProperties();
m_info.m_properties = prop;
prop->m_compressed.setValue(false);
switch (m_header.ImageTypeCode) {
case 0:
//cout << "NO IMAGE" << endl;
break;
case 1:
//cout << "Uncompressed Cmapped; pixelsize=" << m_header.ImagePixelSize << endl;
m_readLineProc = &TgaReader::readLineCmapped;
m_info.m_samplePerPixel = 4;
break;
case 2:
//cout << "Uncompressed RGB; pixelsize=" << m_header.ImagePixelSize << endl;
m_info.m_samplePerPixel = 4;
if (m_header.ImagePixelSize == 32) {
m_readLineProc = &TgaReader::readLineRGB32;
m_skipLineProc = &TgaReader::skipLines32;
} else if (m_header.ImagePixelSize == 24) {
m_info.m_samplePerPixel = 3;
m_readLineProc = &TgaReader::readLineRGB24;
m_skipLineProc = &TgaReader::skipLines24;
} else if (m_header.ImagePixelSize == 16) {
m_readLineProc = &TgaReader::readLineRGB16;
m_skipLineProc = &TgaReader::skipLines16;
}
break;
case 3:
m_info.m_samplePerPixel = 1;
m_readLineProc = &TgaReader::readLineGR8;
break;
case 9:
//cout << "Compressed Cmapped; pixelsize=" << m_header.ImagePixelSize << endl;
prop->m_compressed.setValue(true);
m_readLineProc = &TgaReader::readLineCmappedRle;
m_info.m_samplePerPixel = 4;
break;
case 10:
//cout << "Compressed RGB; pixelsize=" << m_header.ImagePixelSize << endl;
prop->m_compressed.setValue(true);
m_info.m_samplePerPixel = 4;
if (m_header.ImagePixelSize == 32) {
m_readLineProc = &TgaReader::readLineRGB32rle;
m_skipLineProc = &TgaReader::skipNoLines;
} else if (m_header.ImagePixelSize == 24) {
m_info.m_samplePerPixel = 3;
m_readLineProc = &TgaReader::readLineRGB24rle;
m_skipLineProc = &TgaReader::skipNoLines;
} else if (m_header.ImagePixelSize == 16) {
m_readLineProc = &TgaReader::readLineRGB16rle;
m_skipLineProc = &TgaReader::skipNoLines;
}
break;
case 11:
m_info.m_samplePerPixel = 1;
m_readLineProc = &TgaReader::readLineGR8rle;
break;
default:;
//cout << "***Unknown***" << endl;
}
}
}
//------------------------------------------------------------
void TgaReader::readPalette()
{
int colorCount = m_header.ColorMapLength;
int bitPerColor = m_header.ColorMapSize;
// cout << "Reading palette: " << colorCount << " " << bitPerColor << "-entries" << endl;
m_palette = new TPixel32[colorCount];
int i;
switch (bitPerColor) {
case 16:
for (i = 0; i < colorCount; i++)
m_palette[i] = fgetPixel16(m_chan);
break;
case 24:
for (i = 0; i < colorCount; i++)
m_palette[i] = fgetPixel24(m_chan);
break;
case 32:
for (i = 0; i < colorCount; i++)
m_palette[i] = fgetPixel32(m_chan);
break;
break;
}
}
//------------------------------------------------------------
void TgaReader::readLineRGB32(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *pix = (TPixel32 *)buffer;
pix += x0;
if (x0 > 0)
skipBytes(4 * x0);
TPixel32 *endPix = pix + x1 - x0 + 1;
while (pix < endPix) {
*pix++ = fgetPixel32(m_chan);
if (pix < endPix && shrink > 1) {
pix += (shrink - 1);
skipBytes(4 * (shrink - 1));
}
}
if (m_info.m_lx - x1 - 1 > 0)
skipBytes(4 * (m_info.m_lx - x1 - 1));
}
//------------------------------------------------------------
void TgaReader::readLineRGB24(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *pix = (TPixel32 *)buffer;
pix += x0;
if (x0 > 0)
skipBytes(3 * x0);
TPixel32 *endPix = pix + x1 - x0 + 1;
while (pix < endPix) {
*pix++ = fgetPixel24(m_chan);
if (pix < endPix && shrink > 1) {
pix += (shrink - 1);
skipBytes(3 * (shrink - 1));
}
}
if (m_info.m_lx - x1 - 1 > 0)
skipBytes(3 * (m_info.m_lx - x1 - 1));
}
//------------------------------------------------------------
void TgaReader::readLineRGB16(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *pix = (TPixel32 *)buffer;
pix += x0;
if (x0 > 0)
skipBytes(2 * x0);
TPixel32 *endPix = pix + x1 - x0 + 1;
while (pix < endPix) {
*pix++ = fgetPixel16(m_chan);
if (pix < endPix && shrink > 1) {
pix += (shrink - 1);
skipBytes(2 * (shrink - 1));
}
}
if (m_info.m_lx - x1 - 1 > 0)
skipBytes(2 * (m_info.m_lx - x1 - 1));
}
//------------------------------------------------------------
void TgaReader::readLineCmapped(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *pix = (TPixel32 *)buffer;
pix += x0;
if (x0 > 0)
skipBytes(x0);
TPixel32 *endPix = pix + x1 - x0 + 1;
while (pix < endPix) {
int index = fgetc(m_chan);
*pix++ = m_palette[index];
if (pix < endPix && shrink > 1) {
pix += (shrink - 1);
skipBytes(shrink - 1);
}
}
if (m_info.m_lx - x1 - 1 > 0)
skipBytes(m_info.m_lx - x1 - 1);
}
//------------------------------------------------------------
void TgaReader::readLineRGB32rle(char *buffer, int x0, int x1, int shrink)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
UCHAR h = fgetc(m_chan);
int count = (h & 0x7F) + 1;
if (h & 0x80) {
TPixel32 pix = fgetPixel32(m_chan);
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = pix;
} else
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = fgetPixel32(m_chan);
}
}
//------------------------------------------------------------
void TgaReader::readLineRGB24rle(char *buffer, int x0, int x1, int shrink)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
UCHAR h = fgetc(m_chan);
int count = (h & 0x7F) + 1;
if (h & 0x80) {
TPixel32 pix = fgetPixel24(m_chan);
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = pix;
} else
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = fgetPixel24(m_chan);
}
}
//------------------------------------------------------------
void TgaReader::readLineRGB16rle(char *buffer, int x0, int x1, int shrink)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
UCHAR h = fgetc(m_chan);
int count = (h & 0x7F) + 1;
if (h & 0x80) {
TPixel32 pix = fgetPixel16(m_chan);
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = pix;
} else
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = fgetPixel16(m_chan);
}
}
//------------------------------------------------------------
void TgaReader::readLineGR8(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *row = (TPixel32 *)buffer;
int w = x1 - x0 + 1;
while (w--) {
row->r = row->g = row->b = fgetc(m_chan);
row->m = 0xff;
++row;
}
}
//------------------------------------------------------------
void TgaReader::readLineGR8rle(char *buffer, int x0, int x1, int shrink)
{
TPixel32 *row = (TPixel32 *)buffer;
UCHAR count, byte;
int nx, x, i;
int width = x1 - x0 + 1;
for (nx = 0; nx < width;) {
count = fgetc(m_chan);
if ((count >> 7) & 0x01) {
count -= 127;
byte = fgetc(m_chan);
for (x = 0; x < count; x++) {
row->r = row->g = row->b = byte;
row->m = 0xff;
++row;
nx++;
}
} else {
for (i = 0; i < count + 1; i++) {
byte = fgetc(m_chan);
row->r = row->g = row->b = byte;
row->m = 0xff;
++row;
nx++;
}
}
}
if (nx != width)
fprintf(stderr, "tga read error: decode failed\n");
}
//------------------------------------------------------------
void TgaReader::readLineCmappedRle(char *buffer, int x0, int x1, int shrink)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
UCHAR h = fgetc(m_chan);
int count = (h & 0x7F) + 1;
if (h & 0x80) {
int index = fgetc(m_chan);
TPixel32 pix = m_palette[index];
for (int i = 0; i < count && x < m_info.m_lx; i++)
row[x++] = pix;
} else {
for (int i = 0; i < count && x < m_info.m_lx; i++) {
int index = fgetc(m_chan);
row[x++] = m_palette[index];
}
}
}
}
//============================================================
Tiio::TgaWriterProperties::TgaWriterProperties()
: m_pixelSize("Bits Per Pixel"), m_compressed("Compression", true)
{
m_pixelSize.addValue(L"16 bits");
m_pixelSize.addValue(L"24 bits");
m_pixelSize.addValue(L"32 bits");
m_pixelSize.setValue(L"24 bits");
bind(m_pixelSize);
bind(m_compressed);
}
//============================================================
class TgaWriter : public Tiio::Writer
{
FILE *m_chan;
TgaHeader m_header;
typedef void (TgaWriter::*WriteLineProc)(char *buffer);
WriteLineProc m_writeLineProc;
public:
TgaWriter()
: m_chan(0), m_writeLineProc(&TgaWriter::writeNoLine)
{
}
void open(FILE *file, const TImageInfo &info)
{
m_info = info;
try {
m_chan = file;
} catch (...) {
perror("uffa");
return;
}
if (!m_properties)
m_properties = new Tiio::TgaWriterProperties();
bool compressed = ((TBoolProperty *)(m_properties->getProperty("Compression")))->getValue();
memset(&m_header, 0, sizeof(m_header));
m_header.ImageTypeCode = compressed ? 10 : 2;
m_header.Width = m_info.m_lx;
m_header.Height = m_info.m_ly;
wstring pixelSizeW = ((TEnumProperty *)(m_properties->getProperty("Bits Per Pixel")))->getValue().substr(0, 2);
if (pixelSizeW == L"16") {
m_header.ImagePixelSize = 16;
m_writeLineProc = compressed
? &TgaWriter::writeLine16rle
: &TgaWriter::writeLine16;
} else if (pixelSizeW == L"24") {
m_header.ImagePixelSize = 24;
m_writeLineProc = compressed
? &TgaWriter::writeLine24rle
: &TgaWriter::writeLine24;
} else {
m_header.ImagePixelSize = 32;
m_writeLineProc = compressed
? &TgaWriter::writeLine32rle
: &TgaWriter::writeLine32;
}
writeTgaHeader(m_header, m_chan);
}
void flush()
{
fflush(m_chan);
}
~TgaWriter()
{
delete m_properties;
}
inline void fputPixel32(const TPixel32 &pix, FILE *chan)
{
fputc(pix.b, m_chan);
fputc(pix.g, m_chan);
fputc(pix.r, m_chan);
fputc(pix.m, m_chan);
}
inline void fputPixel24(const TPixel32 &pix, FILE *chan)
{
fputc(pix.b, m_chan);
fputc(pix.g, m_chan);
fputc(pix.r, m_chan);
}
inline void fputPixel16(const TPixel32 &pix, FILE *chan)
{
unsigned short word;
word = (pix.r >> 3) << 10 | (pix.g >> 3) << 5 | (pix.b >> 3);
fputc(word & 0xFF, m_chan);
fputc((word >> 8) & 0xFF, m_chan);
}
void writeNoLine(char *buffer) {}
void writeLine16(char *buffer);
void writeLine24(char *buffer);
void writeLine32(char *buffer);
void writeLine16rle(char *buffer);
void writeLine24rle(char *buffer);
void writeLine32rle(char *buffer);
void writeLine(char *buffer)
{
(this->*m_writeLineProc)(buffer);
}
};
//------------------------------------------------------------
void TgaWriter::writeLine32rle(char *buffer)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
if (x + 1 < m_info.m_lx && row[x] == row[x + 1]) {
int count = 2;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] == row[x + count])
count++;
fputc((count - 1) | 0x80, m_chan);
fputPixel32(row[x], m_chan);
x += count;
} else {
int count = 1;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] != row[x + count])
count++;
fputc(count - 1, m_chan);
for (int i = 0; i < count; i++)
fputPixel32(row[x++], m_chan);
}
}
}
//------------------------------------------------------------
void TgaWriter::writeLine24rle(char *buffer)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
if (x + 1 < m_info.m_lx && row[x] == row[x + 1]) {
int count = 2;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] == row[x + count])
count++;
fputc((count - 1) | 0x80, m_chan);
fputPixel24(row[x], m_chan);
x += count;
} else {
int count = 1;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] != row[x + count])
count++;
fputc(count - 1, m_chan);
for (int i = 0; i < count; i++)
fputPixel24(row[x++], m_chan);
}
}
}
//------------------------------------------------------------
void TgaWriter::writeLine16rle(char *buffer)
{
int x = 0;
TPixel32 *row = (TPixel32 *)buffer;
while (x < m_info.m_lx) {
if (x + 1 < m_info.m_lx && row[x] == row[x + 1]) {
int count = 2;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] == row[x + count])
count++;
fputc((count - 1) | 0x80, m_chan);
fputPixel16(row[x], m_chan);
x += count;
} else {
int count = 1;
int max = tmin(128, m_info.m_lx - x);
while (count < max && row[x + count - 1] != row[x + count])
count++;
fputc(count - 1, m_chan);
for (int i = 0; i < count; i++)
fputPixel16(row[x++], m_chan);
}
}
}
//------------------------------------------------------------
void TgaWriter::writeLine32(char *buffer)
{
TPixel32 *pix = (TPixel32 *)buffer;
TPixel32 *endPix = pix + m_info.m_lx;
while (pix < endPix)
fputPixel32(*pix++, m_chan);
}
//------------------------------------------------------------
void TgaWriter::writeLine24(char *buffer)
{
TPixel32 *pix = (TPixel32 *)buffer;
TPixel32 *endPix = pix + m_info.m_lx;
while (pix < endPix)
fputPixel24(*pix++, m_chan);
}
//------------------------------------------------------------
void TgaWriter::writeLine16(char *buffer)
{
TPixel32 *pix = (TPixel32 *)buffer;
TPixel32 *endPix = pix + m_info.m_lx;
while (pix < endPix)
fputPixel16(*pix++, m_chan);
}
//============================================================
Tiio::Reader *Tiio::makeTgaReader()
{
return new TgaReader();
}
Tiio::Writer *Tiio::makeTgaWriter()
{
return new TgaWriter();
}