* rebuilt frame (not the reference), and since there is no coordinate system
* they contain exactly as many pixel as the keyframe.
*
- * Supports: BGR24
+ * Supports: BGRA, BGR24, RGB555
*/
#include <stdint.h>
typedef struct ScreenpressoContext {
AVFrame *current;
- /* zlib interation */
+ /* zlib interaction */
uint8_t *inflated_buf;
uLongf inflated_size;
} ScreenpressoContext;
if (!ctx->current)
return AVERROR(ENOMEM);
- avctx->pix_fmt = AV_PIX_FMT_BGR24;
-
- /* Allocate maximum size possible, a full frame */
- ctx->inflated_size = avctx->width * avctx->height * 3;
+ /* Allocate maximum size possible, a full RGBA frame */
+ ctx->inflated_size = avctx->width * avctx->height * 4;
ctx->inflated_buf = av_malloc(ctx->inflated_size);
if (!ctx->inflated_buf)
return AVERROR(ENOMEM);
ScreenpressoContext *ctx = avctx->priv_data;
AVFrame *frame = data;
uLongf length = ctx->inflated_size;
- int keyframe;
+ int keyframe, component_size, src_linesize;
int ret;
/* Size check */
return AVERROR_INVALIDDATA;
}
- /* Basic sanity check, but not really harmful */
- if ((avpkt->data[0] != 0x73 && avpkt->data[0] != 0x72) ||
- avpkt->data[1] != 8) { // bpp probably
- av_log(avctx, AV_LOG_WARNING, "Unknown header 0x%02X%02X\n",
- avpkt->data[0], avpkt->data[1]);
+ /* Compression level (4 bits) and keyframe information (1 bit) */
+ av_log(avctx, AV_LOG_DEBUG, "Compression level %d\n", avpkt->data[0] >> 4);
+ keyframe = avpkt->data[0] & 1;
+
+ /* Pixel size */
+ component_size = ((avpkt->data[1] >> 2) & 0x03) + 1;
+ switch (component_size) {
+ case 2:
+ avctx->pix_fmt = AV_PIX_FMT_RGB555LE;
+ break;
+ case 3:
+ avctx->pix_fmt = AV_PIX_FMT_BGR24;
+ break;
+ case 4:
+ avctx->pix_fmt = AV_PIX_FMT_BGRA;
+ break;
+ default:
+ av_log(avctx, AV_LOG_ERROR, "Invalid bits per pixel value (%d)\n",
+ component_size);
+ return AVERROR_INVALIDDATA;
}
- keyframe = (avpkt->data[0] == 0x73);
/* Inflate the frame after the 2 byte header */
ret = uncompress(ctx->inflated_buf, &length,
if (ret < 0)
return ret;
+ /* Codec has aligned strides */
+ src_linesize = FFALIGN(avctx->width * component_size, 4);
+
/* When a keyframe is found, copy it (flipped) */
if (keyframe)
av_image_copy_plane(ctx->current->data[0] +
ctx->current->linesize[0] * (avctx->height - 1),
-1 * ctx->current->linesize[0],
- ctx->inflated_buf, avctx->width * 3,
- avctx->width * 3, avctx->height);
+ ctx->inflated_buf, src_linesize,
+ avctx->width * component_size, avctx->height);
/* Otherwise sum the delta on top of the current frame */
else
sum_delta_flipped(ctx->current->data[0], ctx->current->linesize[0],
- ctx->inflated_buf, avctx->width * 3,
- avctx->width * 3, avctx->height);
+ ctx->inflated_buf, src_linesize,
+ avctx->width * component_size, avctx->height);
/* Frame is ready to be output */
ret = av_frame_ref(frame, ctx->current);