3 * Copyright (c) 2006 Konstantin Shishkov.
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 * Based on http://wiki.multimedia.cx/index.php?title=Smacker
26 #include "libavutil/bswap.h"
29 #define SMACKER_PAL 0x01
30 #define SMACKER_FLAG_RING_FRAME 0x01
33 SMK_AUD_PACKED
= 0x80000000,
34 SMK_AUD_16BITS
= 0x20000000,
35 SMK_AUD_STEREO
= 0x10000000,
36 SMK_AUD_BINKAUD
= 0x08000000,
37 SMK_AUD_USEDCT
= 0x04000000
40 typedef struct SmackerContext
{
41 /* Smacker file header */
43 uint32_t width
, height
;
49 uint32_t mmap_size
, mclr_size
, full_size
, type_size
;
55 /* internal variables */
59 /* current frame for demuxing */
71 typedef struct SmackerFrame
{
76 /* palette used in Smacker */
77 static const uint8_t smk_pal
[64] = {
78 0x00, 0x04, 0x08, 0x0C, 0x10, 0x14, 0x18, 0x1C,
79 0x20, 0x24, 0x28, 0x2C, 0x30, 0x34, 0x38, 0x3C,
80 0x41, 0x45, 0x49, 0x4D, 0x51, 0x55, 0x59, 0x5D,
81 0x61, 0x65, 0x69, 0x6D, 0x71, 0x75, 0x79, 0x7D,
82 0x82, 0x86, 0x8A, 0x8E, 0x92, 0x96, 0x9A, 0x9E,
83 0xA2, 0xA6, 0xAA, 0xAE, 0xB2, 0xB6, 0xBA, 0xBE,
84 0xC3, 0xC7, 0xCB, 0xCF, 0xD3, 0xD7, 0xDB, 0xDF,
85 0xE3, 0xE7, 0xEB, 0xEF, 0xF3, 0xF7, 0xFB, 0xFF
89 static int smacker_probe(AVProbeData
*p
)
91 if(p
->buf
[0] == 'S' && p
->buf
[1] == 'M' && p
->buf
[2] == 'K'
92 && (p
->buf
[3] == '2' || p
->buf
[3] == '4'))
93 return AVPROBE_SCORE_MAX
;
98 static int smacker_read_header(AVFormatContext
*s
, AVFormatParameters
*ap
)
100 ByteIOContext
*pb
= s
->pb
;
101 SmackerContext
*smk
= s
->priv_data
;
102 AVStream
*st
, *ast
[7];
106 /* read and check header */
107 smk
->magic
= get_le32(pb
);
108 if (smk
->magic
!= MKTAG('S', 'M', 'K', '2') && smk
->magic
!= MKTAG('S', 'M', 'K', '4'))
110 smk
->width
= get_le32(pb
);
111 smk
->height
= get_le32(pb
);
112 smk
->frames
= get_le32(pb
);
113 smk
->pts_inc
= (int32_t)get_le32(pb
);
114 smk
->flags
= get_le32(pb
);
115 if(smk
->flags
& SMACKER_FLAG_RING_FRAME
)
117 for(i
= 0; i
< 7; i
++)
118 smk
->audio
[i
] = get_le32(pb
);
119 smk
->treesize
= get_le32(pb
);
121 if(smk
->treesize
>= UINT_MAX
/4){ // smk->treesize + 16 must not overflow (this check is probably redundant)
122 av_log(s
, AV_LOG_ERROR
, "treesize too large\n");
126 //FIXME remove extradata "rebuilding"
127 smk
->mmap_size
= get_le32(pb
);
128 smk
->mclr_size
= get_le32(pb
);
129 smk
->full_size
= get_le32(pb
);
130 smk
->type_size
= get_le32(pb
);
131 for(i
= 0; i
< 7; i
++)
132 smk
->rates
[i
] = get_le32(pb
);
133 smk
->pad
= get_le32(pb
);
135 if(smk
->frames
> 0xFFFFFF) {
136 av_log(s
, AV_LOG_ERROR
, "Too many frames: %i\n", smk
->frames
);
139 smk
->frm_size
= av_malloc(smk
->frames
* 4);
140 smk
->frm_flags
= av_malloc(smk
->frames
);
142 smk
->is_ver4
= (smk
->magic
!= MKTAG('S', 'M', 'K', '2'));
144 /* read frame info */
145 for(i
= 0; i
< smk
->frames
; i
++) {
146 smk
->frm_size
[i
] = get_le32(pb
);
148 for(i
= 0; i
< smk
->frames
; i
++) {
149 smk
->frm_flags
[i
] = get_byte(pb
);
152 /* init video codec */
153 st
= av_new_stream(s
, 0);
156 smk
->videoindex
= st
->index
;
157 st
->codec
->width
= smk
->width
;
158 st
->codec
->height
= smk
->height
;
159 st
->codec
->pix_fmt
= PIX_FMT_PAL8
;
160 st
->codec
->codec_type
= CODEC_TYPE_VIDEO
;
161 st
->codec
->codec_id
= CODEC_ID_SMACKVIDEO
;
162 st
->codec
->codec_tag
= smk
->magic
;
163 /* Smacker uses 100000 as internal timebase */
165 smk
->pts_inc
= -smk
->pts_inc
;
169 av_reduce(&tbase
, &smk
->pts_inc
, tbase
, smk
->pts_inc
, (1UL<<31)-1);
170 av_set_pts_info(st
, 33, smk
->pts_inc
, tbase
);
171 /* handle possible audio streams */
172 for(i
= 0; i
< 7; i
++) {
173 smk
->indexes
[i
] = -1;
174 if((smk
->rates
[i
] & 0xFFFFFF) && !(smk
->rates
[i
] & SMK_AUD_BINKAUD
)){
175 ast
[i
] = av_new_stream(s
, 0);
176 smk
->indexes
[i
] = ast
[i
]->index
;
177 ast
[i
]->codec
->codec_type
= CODEC_TYPE_AUDIO
;
178 ast
[i
]->codec
->codec_id
= (smk
->rates
[i
] & SMK_AUD_PACKED
) ? CODEC_ID_SMACKAUDIO
: CODEC_ID_PCM_U8
;
179 ast
[i
]->codec
->codec_tag
= MKTAG('S', 'M', 'K', 'A');
180 ast
[i
]->codec
->channels
= (smk
->rates
[i
] & SMK_AUD_STEREO
) ? 2 : 1;
181 ast
[i
]->codec
->sample_rate
= smk
->rates
[i
] & 0xFFFFFF;
182 ast
[i
]->codec
->bits_per_coded_sample
= (smk
->rates
[i
] & SMK_AUD_16BITS
) ? 16 : 8;
183 if(ast
[i
]->codec
->bits_per_coded_sample
== 16 && ast
[i
]->codec
->codec_id
== CODEC_ID_PCM_U8
)
184 ast
[i
]->codec
->codec_id
= CODEC_ID_PCM_S16LE
;
185 av_set_pts_info(ast
[i
], 64, 1, ast
[i
]->codec
->sample_rate
186 * ast
[i
]->codec
->channels
* ast
[i
]->codec
->bits_per_coded_sample
/ 8);
191 /* load trees to extradata, they will be unpacked by decoder */
192 st
->codec
->extradata
= av_malloc(smk
->treesize
+ 16);
193 st
->codec
->extradata_size
= smk
->treesize
+ 16;
194 if(!st
->codec
->extradata
){
195 av_log(s
, AV_LOG_ERROR
, "Cannot allocate %i bytes of extradata\n", smk
->treesize
+ 16);
196 av_free(smk
->frm_size
);
197 av_free(smk
->frm_flags
);
200 ret
= get_buffer(pb
, st
->codec
->extradata
+ 16, st
->codec
->extradata_size
- 16);
201 if(ret
!= st
->codec
->extradata_size
- 16){
202 av_free(smk
->frm_size
);
203 av_free(smk
->frm_flags
);
206 ((int32_t*)st
->codec
->extradata
)[0] = le2me_32(smk
->mmap_size
);
207 ((int32_t*)st
->codec
->extradata
)[1] = le2me_32(smk
->mclr_size
);
208 ((int32_t*)st
->codec
->extradata
)[2] = le2me_32(smk
->full_size
);
209 ((int32_t*)st
->codec
->extradata
)[3] = le2me_32(smk
->type_size
);
212 smk
->nextpos
= url_ftell(pb
);
218 static int smacker_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
220 SmackerContext
*smk
= s
->priv_data
;
228 if (url_feof(s
->pb
) || smk
->cur_frame
>= smk
->frames
)
231 /* if we demuxed all streams, pass another frame */
232 if(smk
->curstream
< 0) {
233 url_fseek(s
->pb
, smk
->nextpos
, 0);
234 frame_size
= smk
->frm_size
[smk
->cur_frame
] & (~3);
235 flags
= smk
->frm_flags
[smk
->cur_frame
];
236 /* handle palette change event */
237 pos
= url_ftell(s
->pb
);
238 if(flags
& SMACKER_PAL
){
239 int size
, sz
, t
, off
, j
, pos
;
240 uint8_t *pal
= smk
->pal
;
243 memcpy(oldpal
, pal
, 768);
244 size
= get_byte(s
->pb
);
249 pos
= url_ftell(s
->pb
) + size
;
252 if(t
& 0x80){ /* skip palette entries */
253 sz
+= (t
& 0x7F) + 1;
254 pal
+= ((t
& 0x7F) + 1) * 3;
255 } else if(t
& 0x40){ /* copy with offset */
256 off
= get_byte(s
->pb
) * 3;
258 while(j
-- && sz
< 256) {
259 *pal
++ = oldpal
[off
+ 0];
260 *pal
++ = oldpal
[off
+ 1];
261 *pal
++ = oldpal
[off
+ 2];
265 } else { /* new entries */
267 *pal
++ = smk_pal
[get_byte(s
->pb
) & 0x3F];
268 *pal
++ = smk_pal
[get_byte(s
->pb
) & 0x3F];
272 url_fseek(s
->pb
, pos
, 0);
277 /* if audio chunks are present, put them to stack and retrieve later */
278 for(i
= 0; i
< 7; i
++) {
281 size
= get_le32(s
->pb
) - 4;
285 smk
->bufs
[smk
->curstream
] = av_realloc(smk
->bufs
[smk
->curstream
], size
);
286 smk
->buf_sizes
[smk
->curstream
] = size
;
287 ret
= get_buffer(s
->pb
, smk
->bufs
[smk
->curstream
], size
);
290 smk
->stream_id
[smk
->curstream
] = smk
->indexes
[i
];
294 if (av_new_packet(pkt
, frame_size
+ 768))
295 return AVERROR(ENOMEM
);
296 if(smk
->frm_size
[smk
->cur_frame
] & 1)
298 pkt
->data
[0] = palchange
;
299 memcpy(pkt
->data
+ 1, smk
->pal
, 768);
300 ret
= get_buffer(s
->pb
, pkt
->data
+ 769, frame_size
);
301 if(ret
!= frame_size
)
303 pkt
->stream_index
= smk
->videoindex
;
304 pkt
->size
= ret
+ 769;
306 smk
->nextpos
= url_ftell(s
->pb
);
308 if (av_new_packet(pkt
, smk
->buf_sizes
[smk
->curstream
]))
309 return AVERROR(ENOMEM
);
310 memcpy(pkt
->data
, smk
->bufs
[smk
->curstream
], smk
->buf_sizes
[smk
->curstream
]);
311 pkt
->size
= smk
->buf_sizes
[smk
->curstream
];
312 pkt
->stream_index
= smk
->stream_id
[smk
->curstream
];
313 pkt
->pts
= smk
->aud_pts
[smk
->curstream
];
314 smk
->aud_pts
[smk
->curstream
] += AV_RL32(pkt
->data
);
321 static int smacker_read_close(AVFormatContext
*s
)
323 SmackerContext
*smk
= s
->priv_data
;
326 for(i
= 0; i
< 7; i
++)
328 av_free(smk
->bufs
[i
]);
330 av_free(smk
->frm_size
);
332 av_free(smk
->frm_flags
);
337 AVInputFormat smacker_demuxer
= {
339 NULL_IF_CONFIG_SMALL("Smacker video"),
340 sizeof(SmackerContext
),