diff options
Diffstat (limited to 'image/codecs/qtrle.cpp')
| -rw-r--r-- | image/codecs/qtrle.cpp | 431 | 
1 files changed, 431 insertions, 0 deletions
| diff --git a/image/codecs/qtrle.cpp b/image/codecs/qtrle.cpp new file mode 100644 index 0000000000..94744efa5a --- /dev/null +++ b/image/codecs/qtrle.cpp @@ -0,0 +1,431 @@ +/* ScummVM - Graphic Adventure Engine + * + * ScummVM is the legal property of its developers, whose names + * are too numerous to list here. Please refer to the COPYRIGHT + * file distributed with this source distribution. + * + * This program 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. + * + * This program 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 this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + */ + +// QuickTime RLE Decoder +// Based off ffmpeg's QuickTime RLE decoder (written by Mike Melanson) + +#include "image/codecs/qtrle.h" + +#include "common/debug.h" +#include "common/scummsys.h" +#include "common/stream.h" +#include "common/system.h" +#include "common/textconsole.h" +#include "graphics/colormasks.h" +#include "graphics/surface.h" + +namespace Image { + +QTRLEDecoder::QTRLEDecoder(uint16 width, uint16 height, byte bitsPerPixel) : Codec() { +	_bitsPerPixel = bitsPerPixel; + +	// We need to ensure the width is a multiple of 4 +	uint16 wMod = width % 4; +	if (wMod != 0) +		width += 4 - wMod; + +	_surface = new Graphics::Surface(); +	_surface->create(width, height, getPixelFormat()); +} + +#define CHECK_STREAM_PTR(n) \ +  if ((stream.pos() + n) > stream.size()) { \ +    warning("QTRLE Problem: stream out of bounds (%d > %d)", stream.pos() + n, stream.size()); \ +    return; \ +  } + +#define CHECK_PIXEL_PTR(n) \ +  if ((int32)pixelPtr + n > _surface->w * _surface->h) { \ +    warning("QTRLE Problem: pixel ptr = %d, pixel limit = %d", pixelPtr + n, _surface->w * _surface->h); \ +    return; \ +  } \ + +void QTRLEDecoder::decode1(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange) { +	uint32 pixelPtr = 0; +	byte *rgb = (byte *)_surface->getPixels(); + +	while (linesToChange) { +		CHECK_STREAM_PTR(2); +		byte skip = stream.readByte(); +		int8 rleCode = stream.readSByte(); + +		if (rleCode == 0) +			break; + +		if (skip & 0x80) { +			linesToChange--; +			rowPtr += _surface->w; +			pixelPtr = rowPtr + 2 * (skip & 0x7f); +		} else +			pixelPtr += 2 * skip; + +		if (rleCode < 0) { +			// decode the run length code +			rleCode = -rleCode; +			// get the next 2 bytes from the stream, treat them as groups of 8 pixels, and output them rleCode times */ +			CHECK_STREAM_PTR(2); +			byte pi0 = stream.readByte(); +			byte pi1 = stream.readByte(); +			CHECK_PIXEL_PTR(rleCode * 2); + +			while (rleCode--) { +				rgb[pixelPtr++] = pi0; +				rgb[pixelPtr++] = pi1; +			} +		} else { +			// copy the same pixel directly to output 2 times +			rleCode *= 2; +			CHECK_STREAM_PTR(rleCode); +			CHECK_PIXEL_PTR(rleCode); + +			while (rleCode--) +				rgb[pixelPtr++] = stream.readByte(); +		} +	} +} + +void QTRLEDecoder::decode2_4(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange, byte bpp) { +	uint32 pixelPtr = 0; +	byte *rgb = (byte *)_surface->getPixels(); +	byte numPixels = (bpp == 4) ? 8 : 16; + +	while (linesToChange--) { +		CHECK_STREAM_PTR(2); +		pixelPtr = rowPtr + (numPixels * (stream.readByte() - 1)); + +		for (int8 rleCode = stream.readSByte(); rleCode != -1; rleCode = stream.readSByte()) { +			if (rleCode == 0) { +				// there's another skip code in the stream +				CHECK_STREAM_PTR(1); +				pixelPtr += (numPixels * (stream.readByte() - 1)); +			} else if (rleCode < 0) { +				// decode the run length code +				rleCode = -rleCode; + +				// get the next 4 bytes from the stream, treat them as palette indices, and output them rleCode times */ +				CHECK_STREAM_PTR(4); + +				byte pi[16]; // 16 palette indices + +				for (int8 i = numPixels - 1; i >= 0; i--) { +					pi[numPixels - 1 - i] = (stream.readByte() >> ((i * bpp) & 0x07)) & ((1 << bpp) - 1); + +					if ((i & ((numPixels >> 2) - 1)) == 0) +						stream.readByte(); +				} + +				CHECK_PIXEL_PTR(rleCode * numPixels); + +				while (rleCode--) +					for (byte i = 0; i < numPixels; i++) +						rgb[pixelPtr++] = pi[i]; +			} else { +				// copy the same pixel directly to output 4 times +				rleCode *= 4; +				CHECK_STREAM_PTR(rleCode); +				CHECK_PIXEL_PTR(rleCode * (numPixels >> 2)); + +				while (rleCode--) { +					byte temp = stream.readByte(); +					if (bpp == 4) { +						rgb[pixelPtr++] = (temp >> 4) & 0x0f; +						rgb[pixelPtr++] = temp & 0x0f; +					} else { +						rgb[pixelPtr++] = (temp >> 6) & 0x03; +						rgb[pixelPtr++] = (temp >> 4) & 0x03; +						rgb[pixelPtr++] = (temp >> 2) & 0x03; +						rgb[pixelPtr++] = temp & 0x03; +					} +				} +			} +		} + +		rowPtr += _surface->w; +	} +} + +void QTRLEDecoder::decode8(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange) { +	uint32 pixelPtr = 0; +	byte *rgb = (byte *)_surface->getPixels(); + +	while (linesToChange--) { +		CHECK_STREAM_PTR(2); +		pixelPtr = rowPtr + 4 * (stream.readByte() - 1); + +		for (int8 rleCode = stream.readSByte(); rleCode != -1; rleCode = stream.readSByte()) { +			if (rleCode == 0) { +				// there's another skip code in the stream +				CHECK_STREAM_PTR(1); +				pixelPtr += 4 * (stream.readByte() - 1); +			} else if (rleCode < 0) { +				// decode the run length code +				rleCode = -rleCode; + +				// get the next 4 bytes from the stream, treat them as palette indices, and output them rleCode times +				CHECK_STREAM_PTR(4); + +				byte pi[4];  // 4 palette indexes + +				for (byte i = 0; i < 4; i++) +					pi[i] = stream.readByte(); + +				CHECK_PIXEL_PTR(rleCode * 4); + +				while (rleCode--) +					for (byte i = 0; i < 4; i++) +						rgb[pixelPtr++] = pi[i]; +			} else { +				// copy the same pixel directly to output 4 times +				rleCode *= 4; +				CHECK_STREAM_PTR(rleCode); +				CHECK_PIXEL_PTR(rleCode); + +				while (rleCode--) +					rgb[pixelPtr++] = stream.readByte(); +			} +		} + +		rowPtr += _surface->w; +	} +} + +void QTRLEDecoder::decode16(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange) { +	uint32 pixelPtr = 0; +	uint16 *rgb = (uint16 *)_surface->getPixels(); + +	while (linesToChange--) { +		CHECK_STREAM_PTR(2); +		pixelPtr = rowPtr + stream.readByte() - 1; + +		for (int8 rleCode = stream.readSByte(); rleCode != -1; rleCode = stream.readSByte()) { +			if (rleCode == 0) { +				// there's another skip code in the stream +				CHECK_STREAM_PTR(1); +				pixelPtr += stream.readByte() - 1; +			} else if (rleCode < 0) { +				// decode the run length code +				rleCode = -rleCode; +				CHECK_STREAM_PTR(2); + +				uint16 rgb16 = stream.readUint16BE(); + +				CHECK_PIXEL_PTR(rleCode); + +				while (rleCode--) +					rgb[pixelPtr++] = rgb16; +			} else { +				CHECK_STREAM_PTR(rleCode * 2); +				CHECK_PIXEL_PTR(rleCode); + +				// copy pixels directly to output +				while (rleCode--) +					rgb[pixelPtr++] = stream.readUint16BE(); +			} +		} + +		rowPtr += _surface->w; +	} +} + +void QTRLEDecoder::decode24(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange) { +	uint32 pixelPtr = 0; +	uint32 *rgb = (uint32 *)_surface->getPixels(); + +	while (linesToChange--) { +		CHECK_STREAM_PTR(2); +		pixelPtr = rowPtr + stream.readByte() - 1; + +		for (int8 rleCode = stream.readSByte(); rleCode != -1; rleCode = stream.readSByte()) { +			if (rleCode == 0) { +				// there's another skip code in the stream +				CHECK_STREAM_PTR(1); +				pixelPtr += stream.readByte() - 1; +			} else if (rleCode < 0) { +				// decode the run length code +				rleCode = -rleCode; + +				CHECK_STREAM_PTR(3); + +				byte r = stream.readByte(); +				byte g = stream.readByte(); +				byte b = stream.readByte(); +				uint32 color = _surface->format.RGBToColor(r, g, b); + +				CHECK_PIXEL_PTR(rleCode); + +				while (rleCode--) +					rgb[pixelPtr++] = color; +			} else { +				CHECK_STREAM_PTR(rleCode * 3); +				CHECK_PIXEL_PTR(rleCode); + +				// copy pixels directly to output +				while (rleCode--) { +					byte r = stream.readByte(); +					byte g = stream.readByte(); +					byte b = stream.readByte(); +					rgb[pixelPtr++] = _surface->format.RGBToColor(r, g, b); +				} +			} +		} + +		rowPtr += _surface->w; +	} +} + +void QTRLEDecoder::decode32(Common::SeekableReadStream &stream, uint32 rowPtr, uint32 linesToChange) { +	uint32 pixelPtr = 0; +	uint32 *rgb = (uint32 *)_surface->getPixels(); + +	while (linesToChange--) { +		CHECK_STREAM_PTR(2); +		pixelPtr = rowPtr + stream.readByte() - 1; + +		for (int8 rleCode = stream.readSByte(); rleCode != -1; rleCode = stream.readSByte()) { +			if (rleCode == 0) { +				// there's another skip code in the stream +				CHECK_STREAM_PTR(1); +				pixelPtr += stream.readByte() - 1; +			} else if (rleCode < 0) { +				// decode the run length code +				rleCode = -rleCode; + +				CHECK_STREAM_PTR(4); + +				byte a = stream.readByte(); +				byte r = stream.readByte(); +				byte g = stream.readByte(); +				byte b = stream.readByte(); +				uint32 color = _surface->format.ARGBToColor(a, r, g, b); + +				CHECK_PIXEL_PTR(rleCode); + +				while (rleCode--) +					rgb[pixelPtr++] = color; +			} else { +				CHECK_STREAM_PTR(rleCode * 4); +				CHECK_PIXEL_PTR(rleCode); + +				// copy pixels directly to output +				while (rleCode--) { +					byte a = stream.readByte(); +					byte r = stream.readByte(); +					byte g = stream.readByte(); +					byte b = stream.readByte(); +					rgb[pixelPtr++] = _surface->format.ARGBToColor(a, r, g, b); +				} +			} +		} + +		rowPtr += _surface->w; +	} +} + +const Graphics::Surface *QTRLEDecoder::decodeFrame(Common::SeekableReadStream &stream) { +	uint16 startLine = 0; +	uint16 height = _surface->h; + +	// check if this frame is even supposed to change +	if (stream.size() < 8) +		return _surface; + +	// start after the chunk size +	stream.readUint32BE(); + +	// fetch the header +	uint16 header = stream.readUint16BE(); + +	// if a header is present, fetch additional decoding parameters +	if (header & 8) { +		if (stream.size() < 14) +			return _surface; + +		startLine = stream.readUint16BE(); +		stream.readUint16BE(); // Unknown +		height = stream.readUint16BE(); +		stream.readUint16BE(); // Unknown +	} + +	uint32 rowPtr = _surface->w * startLine; + +	switch (_bitsPerPixel) { +	case 1: +	case 33: +		decode1(stream, rowPtr, height); +		break; +	case 2: +	case 34: +		decode2_4(stream, rowPtr, height, 2); +		break; +	case 4: +	case 36: +		decode2_4(stream, rowPtr, height, 4); +		break; +	case 8: +	case 40: +		decode8(stream, rowPtr, height); +		break; +	case 16: +		decode16(stream, rowPtr, height); +		break; +	case 24: +		decode24(stream, rowPtr, height); +		break; +	case 32: +		decode32(stream, rowPtr, height); +		break; +	default: +		error("Unsupported QTRLE bits per pixel %d", _bitsPerPixel); +	} + +	return _surface; +} + +QTRLEDecoder::~QTRLEDecoder() { +	_surface->free(); +	delete _surface; +} + +Graphics::PixelFormat QTRLEDecoder::getPixelFormat() const { +	switch (_bitsPerPixel) { +	case 1: +	case 33: +	case 2: +	case 34: +	case 4: +	case 36: +	case 8: +	case 40: +		return Graphics::PixelFormat::createFormatCLUT8(); +	case 16: +		return Graphics::PixelFormat(2, 5, 5, 5, 0, 10, 5, 0, 0); +	case 24: +	case 32: +		return Graphics::PixelFormat(4, 8, 8, 8, 8, 16, 8, 0, 24); +	default: +		error("Unsupported QTRLE bits per pixel %d", _bitsPerPixel); +	} + +	return Graphics::PixelFormat(); +} + +} // End of namespace Image | 
