154 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			154 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * QuickDraw (qdrw) codec
 | |
|  * Copyright (c) 2004 Konstantin Shishkov
 | |
|  *
 | |
|  * This file is part of Libav.
 | |
|  *
 | |
|  * Libav is free software; you can redistribute it and/or
 | |
|  * modify it under the terms of the GNU Lesser General Public
 | |
|  * License as published by the Free Software Foundation; either
 | |
|  * version 2.1 of the License, or (at your option) any later version.
 | |
|  *
 | |
|  * Libav 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
 | |
|  * Lesser General Public License for more details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU Lesser General Public
 | |
|  * License along with Libav; if not, write to the Free Software
 | |
|  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 | |
|  */
 | |
| 
 | |
| /**
 | |
|  * @file
 | |
|  * Apple QuickDraw codec.
 | |
|  */
 | |
| 
 | |
| #include "libavutil/common.h"
 | |
| #include "libavutil/intreadwrite.h"
 | |
| #include "avcodec.h"
 | |
| #include "internal.h"
 | |
| 
 | |
| static int decode_frame(AVCodecContext *avctx,
 | |
|                         void *data, int *got_frame,
 | |
|                         AVPacket *avpkt)
 | |
| {
 | |
|     const uint8_t *buf     = avpkt->data;
 | |
|     const uint8_t *buf_end = avpkt->data + avpkt->size;
 | |
|     int buf_size           = avpkt->size;
 | |
|     AVFrame * const p      = data;
 | |
|     uint8_t* outdata;
 | |
|     int colors;
 | |
|     int i, ret;
 | |
|     uint32_t *pal;
 | |
|     int r, g, b;
 | |
| 
 | |
|     if ((ret = ff_get_buffer(avctx, p, 0)) < 0) {
 | |
|         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
 | |
|         return ret;
 | |
|     }
 | |
|     p->pict_type = AV_PICTURE_TYPE_I;
 | |
|     p->key_frame = 1;
 | |
| 
 | |
|     outdata = p->data[0];
 | |
| 
 | |
|     if (buf_end - buf < 0x68 + 4)
 | |
|         return AVERROR_INVALIDDATA;
 | |
|     buf   += 0x68; /* jump to palette */
 | |
|     colors = AV_RB32(buf);
 | |
|     buf   += 4;
 | |
| 
 | |
|     if (colors < 0 || colors > 256) {
 | |
|         av_log(avctx, AV_LOG_ERROR, "Error color count - %i(0x%X)\n", colors, colors);
 | |
|         return AVERROR_INVALIDDATA;
 | |
|     }
 | |
|     if (buf_end - buf < (colors + 1) * 8)
 | |
|         return AVERROR_INVALIDDATA;
 | |
| 
 | |
|     pal = (uint32_t*)p->data[1];
 | |
|     for (i = 0; i <= colors; i++) {
 | |
|         unsigned int idx;
 | |
|         idx = AV_RB16(buf); /* color index */
 | |
|         buf += 2;
 | |
| 
 | |
|         if (idx > 255) {
 | |
|             av_log(avctx, AV_LOG_ERROR, "Palette index out of range: %u\n", idx);
 | |
|             buf += 6;
 | |
|             continue;
 | |
|         }
 | |
|         r = *buf++;
 | |
|         buf++;
 | |
|         g = *buf++;
 | |
|         buf++;
 | |
|         b = *buf++;
 | |
|         buf++;
 | |
|         pal[idx] = (r << 16) | (g << 8) | b;
 | |
|     }
 | |
|     p->palette_has_changed = 1;
 | |
| 
 | |
|     if (buf_end - buf < 18)
 | |
|         return AVERROR_INVALIDDATA;
 | |
|     buf += 18; /* skip unneeded data */
 | |
|     for (i = 0; i < avctx->height; i++) {
 | |
|         int size, left, code, pix;
 | |
|         const uint8_t *next;
 | |
|         uint8_t *out;
 | |
|         int tsize = 0;
 | |
| 
 | |
|         /* decode line */
 | |
|         out  = outdata;
 | |
|         size = AV_RB16(buf); /* size of packed line */
 | |
|         buf += 2;
 | |
|         if (buf_end - buf < size)
 | |
|             return AVERROR_INVALIDDATA;
 | |
| 
 | |
|         left = size;
 | |
|         next = buf + size;
 | |
|         while (left > 0) {
 | |
|             code = *buf++;
 | |
|             if (code & 0x80 ) { /* run */
 | |
|                 pix = *buf++;
 | |
|                 if ((out + (257 - code)) > (outdata +  p->linesize[0]))
 | |
|                     break;
 | |
|                 memset(out, pix, 257 - code);
 | |
|                 out   += 257 - code;
 | |
|                 tsize += 257 - code;
 | |
|                 left  -= 2;
 | |
|             } else { /* copy */
 | |
|                 if ((out + code) > (outdata +  p->linesize[0]))
 | |
|                     break;
 | |
|                 if (buf_end - buf < code + 1)
 | |
|                     return AVERROR_INVALIDDATA;
 | |
|                 memcpy(out, buf, code + 1);
 | |
|                 out   += code + 1;
 | |
|                 buf   += code + 1;
 | |
|                 left  -= 2 + code;
 | |
|                 tsize += code + 1;
 | |
|             }
 | |
|         }
 | |
|         buf = next;
 | |
|         outdata += p->linesize[0];
 | |
|     }
 | |
| 
 | |
|     *got_frame      = 1;
 | |
| 
 | |
|     return buf_size;
 | |
| }
 | |
| 
 | |
| static av_cold int decode_init(AVCodecContext *avctx)
 | |
| {
 | |
|     avctx->pix_fmt= AV_PIX_FMT_PAL8;
 | |
| 
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| AVCodec ff_qdraw_decoder = {
 | |
|     .name           = "qdraw",
 | |
|     .long_name      = NULL_IF_CONFIG_SMALL("Apple QuickDraw"),
 | |
|     .type           = AVMEDIA_TYPE_VIDEO,
 | |
|     .id             = AV_CODEC_ID_QDRAW,
 | |
|     .init           = decode_init,
 | |
|     .decode         = decode_frame,
 | |
|     .capabilities   = CODEC_CAP_DR1,
 | |
| };
 |