FFmpeg: libavcodec/rv10.c Source File

FFmpeg

[フレーム]

libavcodec

rv10.c

Go to the documentation of this file.

1 /*

2 * RV10/RV20 decoder

5 *

6 * This file is part of FFmpeg.

7 *

8 * FFmpeg is free software; you can redistribute it and/or

9 * modify it under the terms of the GNU Lesser General Public

10 * License as published by the Free Software Foundation; either

11 * version 2.1 of the License, or (at your option) any later version.

12 *

13 * FFmpeg is distributed in the hope that it will be useful,

14 * but WITHOUT ANY WARRANTY; without even the implied warranty of

15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

16 * Lesser General Public License for more details.

17 *

18 * You should have received a copy of the GNU Lesser General Public

19 * License along with FFmpeg; if not, write to the Free Software

20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

21 */

23 /**

24 * @file

25 * RV10/RV20 decoder

26 */

28 #include <inttypes.h>

30 #include "libavutil/imgutils.h"

31 #include "libavutil/thread.h"

33 #include "avcodec.h"

34 #include "codec_internal.h"

35 #include "decode.h"

36 #include "error_resilience.h"

37 #include "h263.h"

38 #include "h263data.h"

39 #include "h263dec.h"

40 #include "mpeg_er.h"

41 #include "mpegvideo.h"

42 #include "mpegvideodec.h"

43 #include "mpeg4video.h"

44 #include "mpegvideodata.h"

45 #include "rv10dec.h"

47 #define RV_GET_MAJOR_VER(x) ((x) >> 28)

48 #define RV_GET_MINOR_VER(x) (((x) >> 20) & 0xFF)

49 #define RV_GET_MICRO_VER(x) (((x) >> 12) & 0xFF)

51 #define MAX_VLC_ENTRIES 1023 // Note: Does not include the skip entries.

52 #define DC_VLC_BITS 9

54 typedef struct RVDecContext {

55 MpegEncContext m;

56 int sub_id;

57 int orig_width, orig_height;

58 } RVDecContext;

60 /* (run, length) encoded value for the symbols table. The actual symbols

61 * are run..run - length (mod 256).

62 * The last two entries in the following table apply to luma only.

63 * The skip values are not included in this list. */

64 static const uint8_t rv_sym_run_len[][2] = {

65 { 0, 0 }, { 1, 0 }, { 255, 0 }, { 3, 1 }, { 254, 1 },

66 { 7, 3 }, { 252, 3 }, { 15, 7 }, { 248, 7 }, { 31, 15 },

67 { 240, 15 }, { 63, 31 }, { 224, 31 }, { 127, 63 }, { 192, 63 },

68 { 255, 127 }, { 128, 127 }, { 127, 255 }, { 128, 255 },

69 };

71 /* entry[i] of the following tables gives

72 * the number of VLC codes of length i + 2. */

73 static const uint16_t rv_lum_len_count[15] = {

74 1, 0, 2, 4, 8, 16, 32, 0, 64, 0, 128, 0, 256, 0, 512,

75 };

77 static const uint16_t rv_chrom_len_count[15] = {

78 1, 2, 4, 0, 8, 0, 16, 0, 32, 0, 64, 0, 128, 0, 256,

79 };

81 static VLC rv_dc_lum, rv_dc_chrom;

83 int ff_rv_decode_dc(MpegEncContext *s, int n)

84 {

85 int code;

87 if (n < 4) {

88 code = get_vlc2(&s->gb, rv_dc_lum.table, DC_VLC_BITS, 2);

89 } else {

90 code = get_vlc2(&s->gb, rv_dc_chrom.table, DC_VLC_BITS, 2);

91 if (code < 0) {

92 av_log(s->avctx, AV_LOG_ERROR, "chroma dc error\n");

93 return -1;

94 }

95 }

96 return code;

97 }

99 /* read RV 1.0 compatible frame header */

100 static int rv10_decode_picture_header(MpegEncContext *s)

101 {

102 int mb_count, pb_frame, marker, mb_xy;

103

104 marker = get_bits1(&s->gb);

105

106 if (get_bits1(&s->gb))

107 s->pict_type = AV_PICTURE_TYPE_P;

108 else

109 s->pict_type = AV_PICTURE_TYPE_I;

110

111 if (!marker)

112 av_log(s->avctx, AV_LOG_ERROR, "marker missing\n");

113

114 pb_frame = get_bits1(&s->gb);

115

116 ff_dlog(s->avctx, "pict_type=%d pb_frame=%d\n", s->pict_type, pb_frame);

117

118 if (pb_frame) {

119 avpriv_request_sample(s->avctx, "PB-frame");

120 return AVERROR_PATCHWELCOME;

121 }

122

123 s->qscale = get_bits(&s->gb, 5);

124 if (s->qscale == 0) {

125 av_log(s->avctx, AV_LOG_ERROR, "Invalid qscale value: 0\n");

126 return AVERROR_INVALIDDATA;

127 }

128

129 if (s->pict_type == AV_PICTURE_TYPE_I) {

130 if (s->rv10_version == 3) {

131 /* specific MPEG like DC coding not used */

132 s->last_dc[0] = get_bits(&s->gb, 8);

133 s->last_dc[1] = get_bits(&s->gb, 8);

134 s->last_dc[2] = get_bits(&s->gb, 8);

135 ff_dlog(s->avctx, "DC:%d %d %d\n", s->last_dc[0],

136 s->last_dc[1], s->last_dc[2]);

137 }

138 }

139 /* if multiple packets per frame are sent, the position at which

140 * to display the macroblocks is coded here */

141

142 mb_xy = s->mb_x + s->mb_y * s->mb_width;

143 if (show_bits(&s->gb, 12) == 0 || (mb_xy && mb_xy < s->mb_num)) {

144 s->mb_x = get_bits(&s->gb, 6); /* mb_x */

145 s->mb_y = get_bits(&s->gb, 6); /* mb_y */

146 mb_count = get_bits(&s->gb, 12);

147 } else {

148 s->mb_x = 0;

149 s->mb_y = 0;

150 mb_count = s->mb_width * s->mb_height;

151 }

152 skip_bits(&s->gb, 3); /* ignored */

153 s->f_code = 1;

154

155 return mb_count;

156 }

157

158 static int rv20_decode_picture_header(RVDecContext *rv, int whole_size)

159 {

160 static const enum AVPictureType pict_types[] =

161 { AV_PICTURE_TYPE_I, AV_PICTURE_TYPE_I /* hmm ... */,

162 AV_PICTURE_TYPE_P, AV_PICTURE_TYPE_B };

163 MpegEncContext *s = &rv->m;

164 int seq, mb_pos, ret;

165 int rpr_max;

166

167 s->pict_type = pict_types[get_bits(&s->gb, 2)];

168

169 if (s->low_delay && s->pict_type == AV_PICTURE_TYPE_B) {

170 av_log(s->avctx, AV_LOG_ERROR, "low delay B\n");

171 return -1;

172 }

173 if (!s->last_pic.ptr && s->pict_type == AV_PICTURE_TYPE_B) {

174 av_log(s->avctx, AV_LOG_ERROR, "early B-frame\n");

175 return AVERROR_INVALIDDATA;

176 }

177

178 if (get_bits1(&s->gb)) {

179 av_log(s->avctx, AV_LOG_ERROR, "reserved bit set\n");

180 return AVERROR_INVALIDDATA;

181 }

182

183 s->qscale = get_bits(&s->gb, 5);

184 if (s->qscale == 0) {

185 av_log(s->avctx, AV_LOG_ERROR, "Invalid qscale value: 0\n");

186 return AVERROR_INVALIDDATA;

187 }

188

189 if (RV_GET_MINOR_VER(rv->sub_id) >= 2)

190 s->loop_filter = get_bits1(&s->gb) && !s->avctx->lowres;

191

192 if (RV_GET_MINOR_VER(rv->sub_id) <= 1)

193 seq = get_bits(&s->gb, 8) << 7;

194 else

195 seq = get_bits(&s->gb, 13) << 2;

196

197 rpr_max = s->avctx->extradata[1] & 7;

198 if (rpr_max) {

199 int f, new_w, new_h;

200 int rpr_bits = av_log2(rpr_max) + 1;

201

202 f = get_bits(&s->gb, rpr_bits);

203

204 if (f) {

205 if (s->avctx->extradata_size < 8 + 2 * f) {

206 av_log(s->avctx, AV_LOG_ERROR, "Extradata too small.\n");

207 return AVERROR_INVALIDDATA;

208 }

209

210 new_w = 4 * ((uint8_t *) s->avctx->extradata)[6 + 2 * f];

211 new_h = 4 * ((uint8_t *) s->avctx->extradata)[7 + 2 * f];

212 } else {

213 new_w = rv->orig_width;

214 new_h = rv->orig_height;

215 }

216 if (new_w != s->width || new_h != s->height || !s->context_initialized) {

217 AVRational old_aspect = s->avctx->sample_aspect_ratio;

218 av_log(s->avctx, AV_LOG_DEBUG,

219 "attempting to change resolution to %dx%d\n", new_w, new_h);

220 if (av_image_check_size(new_w, new_h, 0, s->avctx) < 0)

221 return AVERROR_INVALIDDATA;

222

223 if (whole_size < (new_w + 15)/16 * ((new_h + 15)/16) / 8)

224 return AVERROR_INVALIDDATA;

225

226 ff_mpv_common_end(s);

227

228 // attempt to keep aspect during typical resolution switches

229 if (!old_aspect.num)

230 old_aspect = (AVRational){1, 1};

231 if (2 * (int64_t)new_w * s->height == (int64_t)new_h * s->width)

232 s->avctx->sample_aspect_ratio = av_mul_q(old_aspect, (AVRational){2, 1});

233 if ((int64_t)new_w * s->height == 2 * (int64_t)new_h * s->width)

234 s->avctx->sample_aspect_ratio = av_mul_q(old_aspect, (AVRational){1, 2});

235

236 ret = ff_set_dimensions(s->avctx, new_w, new_h);

237 if (ret < 0)

238 return ret;

239

240 s->width = new_w;

241 s->height = new_h;

242 if ((ret = ff_mpv_common_init(s)) < 0)

243 return ret;

244 }

245

246 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {

247 av_log(s->avctx, AV_LOG_DEBUG, "F %d/%d/%d\n", f, rpr_bits, rpr_max);

248 }

249 }

250 if (av_image_check_size(s->width, s->height, 0, s->avctx) < 0)

251 return AVERROR_INVALIDDATA;

252

253 mb_pos = ff_h263_decode_mba(s);

254

255 seq |= s->time & ~0x7FFF;

256 if (seq - s->time > 0x4000)

257 seq -= 0x8000;

258 if (seq - s->time < -0x4000)

259 seq += 0x8000;

260

261 if (seq != s->time) {

262 if (s->pict_type != AV_PICTURE_TYPE_B) {

263 s->time = seq;

264 s->pp_time = s->time - s->last_non_b_time;

265 s->last_non_b_time = s->time;

266 } else {

267 s->time = seq;

268 s->pb_time = s->pp_time - (s->last_non_b_time - s->time);

269 }

270 }

271 if (s->pict_type == AV_PICTURE_TYPE_B) {

272 if (s->pp_time <=s->pb_time || s->pp_time <= s->pp_time - s->pb_time || s->pp_time<=0) {

273 av_log(s->avctx, AV_LOG_DEBUG,

274 "messed up order, possible from seeking? skipping current B-frame\n");

275 #define ERROR_SKIP_FRAME -123

276 return ERROR_SKIP_FRAME;

277 }

278 ff_mpeg4_init_direct_mv(s);

279 }

280

281 s->no_rounding = get_bits1(&s->gb);

282

283 if (RV_GET_MINOR_VER(rv->sub_id) <= 1 && s->pict_type == AV_PICTURE_TYPE_B)

284 // binary decoder reads 3+2 bits here but they don't seem to be used

285 skip_bits(&s->gb, 5);

286

287 s->f_code = 1;

288 s->h263_aic = s->pict_type == AV_PICTURE_TYPE_I;

289 s->modified_quant = 1;

290 if (!s->avctx->lowres)

291 s->loop_filter = 1;

292

293 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {

294 av_log(s->avctx, AV_LOG_INFO,

295 "num:%5d x:%2d y:%2d type:%d qscale:%2d rnd:%d\n",

296 seq, s->mb_x, s->mb_y, s->pict_type, s->qscale,

297 s->no_rounding);

298 }

299

300 av_assert0(s->pict_type != AV_PICTURE_TYPE_B || !s->low_delay);

301

302 return s->mb_width * s->mb_height - mb_pos;

303 }

304

305 static av_cold void rv10_build_vlc(VLC *vlc, const uint16_t len_count[15],

306 const uint8_t sym_rl[][2], int sym_rl_elems)

307 {

308 uint16_t syms[MAX_VLC_ENTRIES];

309 uint8_t lens[MAX_VLC_ENTRIES];

310 unsigned nb_syms = 0, nb_lens = 0;

311

312 for (unsigned i = 0; i < sym_rl_elems; i++) {

313 unsigned cur_sym = sym_rl[i][0];

314 for (unsigned tmp = nb_syms + sym_rl[i][1]; nb_syms <= tmp; nb_syms++)

315 syms[nb_syms] = 0xFF & cur_sym--;

316 }

317

318 for (unsigned i = 0; i < 15; i++)

319 for (unsigned tmp = nb_lens + len_count[i]; nb_lens < tmp; nb_lens++)

320 lens[nb_lens] = i + 2;

321 av_assert1(nb_lens == nb_syms);

322 ff_vlc_init_from_lengths(vlc, DC_VLC_BITS, nb_lens, lens, 1,

323 syms, 2, 2, 0, VLC_INIT_STATIC_OVERLONG, NULL);

324 }

325

326 static av_cold void rv10_init_static(void)

327 {

328 static VLCElem table[1472 + 992];

329

330 rv_dc_lum.table = table;

331 rv_dc_lum.table_allocated = 1472;

332 rv10_build_vlc(&rv_dc_lum, rv_lum_len_count,

333 rv_sym_run_len, FF_ARRAY_ELEMS(rv_sym_run_len));

334 for (int i = 0; i < 1 << (DC_VLC_BITS - 7 /* Length of skip prefix */); i++) {

335 /* All codes beginning with 0x7F have the same length and value.

336 * Modifying the table directly saves us the useless subtables. */

337 rv_dc_lum.table[(0x7F << (DC_VLC_BITS - 7)) + i].sym = 255;

338 rv_dc_lum.table[(0x7F << (DC_VLC_BITS - 7)) + i].len = 18;

339 }

340 rv_dc_chrom.table = &table[1472];

341 rv_dc_chrom.table_allocated = 992;

342 rv10_build_vlc(&rv_dc_chrom, rv_chrom_len_count,

343 rv_sym_run_len, FF_ARRAY_ELEMS(rv_sym_run_len) - 2);

344 for (int i = 0; i < 1 << (DC_VLC_BITS - 9 /* Length of skip prefix */); i++) {

345 /* Same as above. */

346 rv_dc_chrom.table[(0x1FE << (DC_VLC_BITS - 9)) + i].sym = 255;

347 rv_dc_chrom.table[(0x1FE << (DC_VLC_BITS - 9)) + i].len = 18;

348 }

349 }

350

351 static av_cold int rv10_decode_init(AVCodecContext *avctx)

352 {

353 static AVOnce init_static_once = AV_ONCE_INIT;

354 RVDecContext *rv = avctx->priv_data;

355 MpegEncContext *s = &rv->m;

356 int major_ver, minor_ver, micro_ver, ret;

357

358 if (avctx->extradata_size < 8) {

359 av_log(avctx, AV_LOG_ERROR, "Extradata is too small.\n");

360 return AVERROR_INVALIDDATA;

361 }

362 if ((ret = av_image_check_size(avctx->coded_width,

363 avctx->coded_height, 0, avctx)) < 0)

364 return ret;

365

366 ret = ff_h263_decode_init(avctx);

367 if (ret < 0)

368 return ret;

369

370 rv->orig_width = avctx->coded_width;

371 rv->orig_height = avctx->coded_height;

372

373 s->h263_long_vectors = ((uint8_t *) avctx->extradata)[3] & 1;

374 rv->sub_id = AV_RB32((uint8_t *) avctx->extradata + 4);

375

376 major_ver = RV_GET_MAJOR_VER(rv->sub_id);

377 minor_ver = RV_GET_MINOR_VER(rv->sub_id);

378 micro_ver = RV_GET_MICRO_VER(rv->sub_id);

379

380 switch (major_ver) {

381 case 1:

382 s->rv10_version = micro_ver ? 3 : 1;

383 s->obmc = micro_ver == 2;

384 break;

385 case 2:

386 if (minor_ver >= 2) {

387 s->low_delay = 0;

388 avctx->has_b_frames = 1;

389 }

390 break;

391 default:

392 av_log(avctx, AV_LOG_ERROR, "unknown header %X\n", rv->sub_id);

393 avpriv_request_sample(avctx, "RV1/2 version");

394 return AVERROR_PATCHWELCOME;

395 }

396

397 if (avctx->debug & FF_DEBUG_PICT_INFO) {

398 av_log(avctx, AV_LOG_DEBUG, "ver:%X ver0:%"PRIX32"\n", rv->sub_id,

399 ((uint32_t *) avctx->extradata)[0]);

400 }

401

402 /* init static VLCs */

403 ff_thread_once(&init_static_once, rv10_init_static);

404

405 return 0;

406 }

407

408 static int rv10_decode_packet(AVCodecContext *avctx, const uint8_t *buf,

409 int buf_size, int buf_size2, int whole_size)

410 {

411 RVDecContext *rv = avctx->priv_data;

412 MpegEncContext *s = &rv->m;

413 int mb_count, mb_pos, left, start_mb_x, active_bits_size, ret;

414

415 active_bits_size = buf_size * 8;

416 init_get_bits(&s->gb, buf, FFMAX(buf_size, buf_size2) * 8);

417 if (s->codec_id == AV_CODEC_ID_RV10)

418 mb_count = rv10_decode_picture_header(s);

419 else

420 mb_count = rv20_decode_picture_header(rv, whole_size);

421 if (mb_count < 0) {

422 if (mb_count != ERROR_SKIP_FRAME)

423 av_log(s->avctx, AV_LOG_ERROR, "HEADER ERROR\n");

424 return AVERROR_INVALIDDATA;

425 }

426

427 if (s->mb_x >= s->mb_width ||

428 s->mb_y >= s->mb_height) {

429 av_log(s->avctx, AV_LOG_ERROR, "POS ERROR %d %d\n", s->mb_x, s->mb_y);

430 return AVERROR_INVALIDDATA;

431 }

432 mb_pos = s->mb_y * s->mb_width + s->mb_x;

433 left = s->mb_width * s->mb_height - mb_pos;

434 if (mb_count > left) {

435 av_log(s->avctx, AV_LOG_ERROR, "COUNT ERROR\n");

436 return AVERROR_INVALIDDATA;

437 }

438

439 if (whole_size < s->mb_width * s->mb_height / 8)

440 return AVERROR_INVALIDDATA;

441

442 if ((s->mb_x == 0 && s->mb_y == 0) || !s->cur_pic.ptr) {

443 // FIXME write parser so we always have complete frames?

444 if (s->cur_pic.ptr) {

445 ff_er_frame_end(&s->er, NULL);

446 ff_mpv_frame_end(s);

447 s->mb_x = s->mb_y = s->resync_mb_x = s->resync_mb_y = 0;

448 }

449 if ((ret = ff_mpv_frame_start(s, avctx)) < 0)

450 return ret;

451 ff_mpeg_er_frame_start(s);

452 } else {

453 if (s->cur_pic.ptr->f->pict_type != s->pict_type) {

454 av_log(s->avctx, AV_LOG_ERROR, "Slice type mismatch\n");

455 return AVERROR_INVALIDDATA;

456 }

457 }

458

459

460 ff_dlog(avctx, "qscale=%d\n", s->qscale);

461

462 /* default quantization values */

463 if (s->codec_id == AV_CODEC_ID_RV10) {

464 if (s->mb_y == 0)

465 s->first_slice_line = 1;

466 } else {

467 s->first_slice_line = 1;

468 s->resync_mb_x = s->mb_x;

469 }

470 start_mb_x = s->mb_x;

471 s->resync_mb_y = s->mb_y;

472 if (s->h263_aic) {

473 s->y_dc_scale_table =

474 s->c_dc_scale_table = ff_aic_dc_scale_table;

475 } else {

476 s->y_dc_scale_table =

477 s->c_dc_scale_table = ff_mpeg1_dc_scale_table;

478 }

479

480 if (s->modified_quant)

481 s->chroma_qscale_table = ff_h263_chroma_qscale_table;

482

483 ff_set_qscale(s, s->qscale);

484

485 s->rv10_first_dc_coded[0] = 0;

486 s->rv10_first_dc_coded[1] = 0;

487 s->rv10_first_dc_coded[2] = 0;

488 ff_init_block_index(s);

489

490 /* decode each macroblock */

491 for (s->mb_num_left = mb_count; s->mb_num_left > 0; s->mb_num_left--) {

492 int ret;

493 ff_update_block_index(s, 8, s->avctx->lowres, 1);

494 ff_tlog(avctx, "**mb x=%d y=%d\n", s->mb_x, s->mb_y);

495

496 s->mv_dir = MV_DIR_FORWARD;

497 s->mv_type = MV_TYPE_16X16;

498 ret = ff_h263_decode_mb(s, s->block);

499

500 // Repeat the slice end check from ff_h263_decode_mb with our active

501 // bitstream size

502 if (ret != SLICE_ERROR && active_bits_size >= get_bits_count(&s->gb)) {

503 int v = show_bits(&s->gb, 16);

504

505 if (get_bits_count(&s->gb) + 16 > active_bits_size)

506 v >>= get_bits_count(&s->gb) + 16 - active_bits_size;

507

508 if (!v)

509 ret = SLICE_END;

510 }

511 if (ret != SLICE_ERROR && active_bits_size < get_bits_count(&s->gb) &&

512 8 * buf_size2 >= get_bits_count(&s->gb)) {

513 active_bits_size = buf_size2 * 8;

514 av_log(avctx, AV_LOG_DEBUG, "update size from %d to %d\n",

515 8 * buf_size, active_bits_size);

516 ret = SLICE_OK;

517 }

518

519 if (ret == SLICE_ERROR || active_bits_size < get_bits_count(&s->gb)) {

520 av_log(s->avctx, AV_LOG_ERROR, "ERROR at MB %d %d\n", s->mb_x,

521 s->mb_y);

522 return AVERROR_INVALIDDATA;

523 }

524 if (s->pict_type != AV_PICTURE_TYPE_B)

525 ff_h263_update_motion_val(s);

526 ff_mpv_reconstruct_mb(s, s->block);

527 if (s->loop_filter)

528 ff_h263_loop_filter(s);

529

530 if (++s->mb_x == s->mb_width) {

531 s->mb_x = 0;

532 s->mb_y++;

533 ff_init_block_index(s);

534 }

535 if (s->mb_x == s->resync_mb_x)

536 s->first_slice_line = 0;

537 if (ret == SLICE_END)

538 break;

539 }

540

541 ff_er_add_slice(&s->er, start_mb_x, s->resync_mb_y, s->mb_x - 1, s->mb_y,

542 ER_MB_END);

543

544 return active_bits_size;

545 }

546

547 static int get_slice_offset(AVCodecContext *avctx, const uint8_t *buf, int n)

548 {

549 return AV_RL32(buf + n * 8);

550 }

551

552 static int rv10_decode_frame(AVCodecContext *avctx, AVFrame *pict,

553 int *got_frame, AVPacket *avpkt)

554 {

555 const uint8_t *buf = avpkt->data;

556 int buf_size = avpkt->size;

557 MpegEncContext *s = avctx->priv_data;

558 int i, ret;

559 int slice_count;

560 const uint8_t *slices_hdr = NULL;

561

562 ff_dlog(avctx, "*****frame %"PRId64" size=%d\n", avctx->frame_num, buf_size);

563

564 /* no supplementary picture */

565 if (buf_size == 0) {

566 return 0;

567 }

568

569 slice_count = (*buf++) + 1;

570 buf_size--;

571

572 if (!slice_count || buf_size <= 8 * slice_count) {

573 av_log(avctx, AV_LOG_ERROR, "Invalid slice count: %d.\n",

574 slice_count);

575 return AVERROR_INVALIDDATA;

576 }

577

578 slices_hdr = buf + 4;

579 buf += 8 * slice_count;

580 buf_size -= 8 * slice_count;

581

582 for (i = 0; i < slice_count; i++) {

583 unsigned offset = get_slice_offset(avctx, slices_hdr, i);

584 int size, size2;

585

586 if (offset >= buf_size)

587 return AVERROR_INVALIDDATA;

588

589 if (i + 1 == slice_count)

590 size = buf_size - offset;

591 else

592 size = get_slice_offset(avctx, slices_hdr, i + 1) - offset;

593

594 if (i + 2 >= slice_count)

595 size2 = buf_size - offset;

596 else

597 size2 = get_slice_offset(avctx, slices_hdr, i + 2) - offset;

598

599 if (size <= 0 || size2 <= 0 ||

600 offset + FFMAX(size, size2) > buf_size)

601 return AVERROR_INVALIDDATA;

602

603 if ((ret = rv10_decode_packet(avctx, buf + offset, size, size2, buf_size)) < 0)

604 return ret;

605

606 if (ret > 8 * size)

607 i++;

608 }

609

610 if (s->cur_pic.ptr && s->mb_y >= s->mb_height) {

611 ff_er_frame_end(&s->er, NULL);

612 ff_mpv_frame_end(s);

613

614 if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {

615 if ((ret = av_frame_ref(pict, s->cur_pic.ptr->f)) < 0)

616 return ret;

617 ff_print_debug_info(s, s->cur_pic.ptr, pict);

618 ff_mpv_export_qp_table(s, pict, s->cur_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG1);

619 } else if (s->last_pic.ptr) {

620 if ((ret = av_frame_ref(pict, s->last_pic.ptr->f)) < 0)

621 return ret;

622 ff_print_debug_info(s, s->last_pic.ptr, pict);

623 ff_mpv_export_qp_table(s, pict,s->last_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG1);

624 }

625

626 if (s->last_pic.ptr || s->low_delay) {

627 *got_frame = 1;

628 }

629

630 // so we can detect if frame_end was not called (find some nicer solution...)

631 ff_mpv_unref_picture(&s->cur_pic);

632 }

633

634 return avpkt->size;

635 }

636

637 const FFCodec ff_rv10_decoder = {

638 .p.name = "rv10",

639 CODEC_LONG_NAME("RealVideo 1.0"),

640 .p.type = AVMEDIA_TYPE_VIDEO,

641 .p.id = AV_CODEC_ID_RV10,

642 .priv_data_size = sizeof(RVDecContext),

643 .init = rv10_decode_init,

644 FF_CODEC_DECODE_CB(rv10_decode_frame),

645 .close = ff_mpv_decode_close,

646 .p.capabilities = AV_CODEC_CAP_DR1,

647 .p.max_lowres = 3,

648 .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,

649 };

650

651 const FFCodec ff_rv20_decoder = {

652 .p.name = "rv20",

653 CODEC_LONG_NAME("RealVideo 2.0"),

654 .p.type = AVMEDIA_TYPE_VIDEO,

655 .p.id = AV_CODEC_ID_RV20,

656 .priv_data_size = sizeof(RVDecContext),

657 .init = rv10_decode_init,

658 FF_CODEC_DECODE_CB(rv10_decode_frame),

659 .close = ff_mpv_decode_close,

660 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_DELAY,

661 .flush = ff_mpeg_flush,

662 .p.max_lowres = 3,

663 .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,

664 };

ff_mpv_common_init

av_cold int ff_mpv_common_init(MpegEncContext *s)

init common structure for both encoder and decoder.

Definition: mpegvideo.c:687

MV_TYPE_16X16

#define MV_TYPE_16X16

1 vector for the whole mb

Definition: mpegvideo.h:265

h263data.h

ff_vlc_init_from_lengths

int ff_vlc_init_from_lengths(VLC *vlc, int nb_bits, int nb_codes, const int8_t *lens, int lens_wrap, const void *symbols, int symbols_wrap, int symbols_size, int offset, int flags, void *logctx)

Build VLC decoding tables suitable for use with get_vlc2()

Definition: vlc.c:306

FF_CODEC_CAP_INIT_CLEANUP

#define FF_CODEC_CAP_INIT_CLEANUP

The codec allows calling the close function for deallocation even if the init function returned a fai...

Definition: codec_internal.h:43

RV_GET_MINOR_VER

#define RV_GET_MINOR_VER(x)

Definition: rv10.c:48

ff_h263_decode_mb

int ff_h263_decode_mb(MpegEncContext *s, int16_t block[6][64])

Definition: ituh263dec.c:788

thread.h

AVPictureType

Definition: avutil.h:277

int64_t

long long int64_t

Definition: coverity.c:34

get_bits_count

static int get_bits_count(const GetBitContext *s)

Definition: get_bits.h:266

AVFrame

This structure describes decoded (raw) audio or video data.

Definition: frame.h:389

tmp

static uint8_t tmp[11]

Definition: aes_ctr.c:28

AVPacket::data

uint8_t * data

Definition: packet.h:539

table

static const uint16_t table[]

Definition: prosumer.c:203

FFCodec

Definition: codec_internal.h:127

VLCElem::len

VLCBaseType len

Definition: vlc.h:33

ff_er_add_slice

void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy, int status)

Add a slice.

Definition: error_resilience.c:826

ff_init_block_index

void ff_init_block_index(MpegEncContext *s)

Definition: mpegvideo.c:820

mpegvideo.h

FFMAX

#define FFMAX(a, b)

Definition: macros.h:47

ff_h263_decode_mba

int ff_h263_decode_mba(MpegEncContext *s)

Definition: ituh263dec.c:143

ff_set_dimensions

int ff_set_dimensions(AVCodecContext *s, int width, int height)

Check that the provided frame dimensions are valid and set them on the codec context.

Definition: utils.c:94

init_get_bits

static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)

Initialize GetBitContext.

Definition: get_bits.h:514

RVDecContext

Definition: rv10.c:54

FF_DEBUG_PICT_INFO

#define FF_DEBUG_PICT_INFO

Definition: avcodec.h:1407

MAX_VLC_ENTRIES

#define MAX_VLC_ENTRIES

Definition: rv10.c:51

skip_bits

static void skip_bits(GetBitContext *s, int n)

Definition: get_bits.h:381

get_bits

static unsigned int get_bits(GetBitContext *s, int n)

Read 1-25 bits.

Definition: get_bits.h:335

ff_mpv_reconstruct_mb

void ff_mpv_reconstruct_mb(MpegEncContext *s, int16_t block[12][64])

Definition: mpegvideo_dec.c:910

FFCodec::p

AVCodec p

The public AVCodec.

Definition: codec_internal.h:131

VLCElem::sym

VLCBaseType sym

Definition: vlc.h:33

AVCodecContext::coded_height

int coded_height

Definition: avcodec.h:639

SLICE_END

#define SLICE_END

end marker found

Definition: mpegvideo.h:482

ff_h263_update_motion_val

void ff_h263_update_motion_val(MpegEncContext *s)

Definition: h263.c:53

AVRational::num

int num

Numerator.

Definition: rational.h:59

ff_mpv_common_end

void ff_mpv_common_end(MpegEncContext *s)

Definition: mpegvideo.c:774

ff_thread_once

static int ff_thread_once(char *control, void(*routine)(void))

Definition: thread.h:205

mpegvideodec.h

AV_LOG_ERROR

#define AV_LOG_ERROR

Something went wrong and cannot losslessly be recovered.

Definition: log.h:180

h263dec.h

FF_ARRAY_ELEMS

#define FF_ARRAY_ELEMS(a)

Definition: sinewin_tablegen.c:29

av_cold

#define av_cold

Definition: attributes.h:90

rv10_decode_picture_header

static int rv10_decode_picture_header(MpegEncContext *s)

Definition: rv10.c:100

AVCodecContext::extradata_size

int extradata_size

Definition: avcodec.h:530

AVCodecContext::has_b_frames

int has_b_frames

Size of the frame reordering buffer in the decoder.

Definition: avcodec.h:729

ff_er_frame_end

void ff_er_frame_end(ERContext *s, int *decode_error_flags)

Indicate that a frame has finished decoding and perform error concealment in case it has been enabled...

Definition: error_resilience.c:896

FF_CODEC_DECODE_CB

#define FF_CODEC_DECODE_CB(func)

Definition: codec_internal.h:311

ff_h263_chroma_qscale_table

const uint8_t ff_h263_chroma_qscale_table[32]

Definition: h263data.c:260

#define s(width, name)

Definition: cbs_vp9.c:198

RV_GET_MAJOR_VER

#define RV_GET_MAJOR_VER(x)

Definition: rv10.c:47

ff_mpeg_er_frame_start

void ff_mpeg_er_frame_start(MpegEncContext *s)

Definition: mpeg_er.c:49

ff_mpeg1_dc_scale_table

static const uint8_t *const ff_mpeg1_dc_scale_table

Definition: mpegvideodata.h:32

av_assert0

#define av_assert0(cond)

assert() equivalent, that is always enabled.

Definition: avassert.h:40

AV_LOG_DEBUG

#define AV_LOG_DEBUG

Stuff which is only useful for libav* developers.

Definition: log.h:201

rv_dc_chrom

static VLC rv_dc_chrom

Definition: rv10.c:81

decode.h

rv10_build_vlc

static av_cold void rv10_build_vlc(VLC *vlc, const uint16_t len_count[15], const uint8_t sym_rl[][2], int sym_rl_elems)

Definition: rv10.c:305

CODEC_LONG_NAME

#define CODEC_LONG_NAME(str)

Definition: codec_internal.h:296

ff_mpv_unref_picture

void ff_mpv_unref_picture(MPVWorkPicture *pic)

Definition: mpegpicture.c:98

RVDecContext::orig_height

int orig_height

Definition: rv10.c:57

AV_ONCE_INIT

#define AV_ONCE_INIT

Definition: thread.h:203

ff_mpv_export_qp_table

int ff_mpv_export_qp_table(const MpegEncContext *s, AVFrame *f, const MPVPicture *p, int qp_type)

Definition: mpegvideo_dec.c:410

NULL

#define NULL

Definition: coverity.c:32

AVERROR_PATCHWELCOME

#define AVERROR_PATCHWELCOME

Not yet implemented in FFmpeg, patches welcome.

Definition: error.h:64

RVDecContext::orig_width

int orig_width

Definition: rv10.c:57

AVRational

Rational number (pair of numerator and denominator).

Definition: rational.h:58

ERROR_SKIP_FRAME

#define ERROR_SKIP_FRAME

AV_PICTURE_TYPE_I

@ AV_PICTURE_TYPE_I

Intra.

Definition: avutil.h:279

get_bits1

static unsigned int get_bits1(GetBitContext *s)

Definition: get_bits.h:388

ff_set_qscale

void ff_set_qscale(MpegEncContext *s, int qscale)

set qscale and update qscale dependent variables.

Definition: mpegvideo.c:853

get_vlc2

static av_always_inline int get_vlc2(GetBitContext *s, const VLCElem *table, int bits, int max_depth)

Parse a vlc code.

Definition: get_bits.h:652

FF_MPV_QSCALE_TYPE_MPEG1

#define FF_MPV_QSCALE_TYPE_MPEG1

Definition: mpegvideodec.h:40

AVOnce

#define AVOnce

Definition: thread.h:202

rv10_decode_frame

static int rv10_decode_frame(AVCodecContext *avctx, AVFrame *pict, int *got_frame, AVPacket *avpkt)

Definition: rv10.c:552

rv_chrom_len_count

static const uint16_t rv_chrom_len_count[15]

Definition: rv10.c:77

ff_dlog

#define ff_dlog(a,...)

Definition: tableprint_vlc.h:28

VLC::table_allocated

int table_allocated

Definition: vlc.h:39

ff_rv20_decoder

const FFCodec ff_rv20_decoder

Definition: rv10.c:651

rv_dc_lum

static VLC rv_dc_lum

Definition: rv10.c:81

Definition: af_crystalizer.c:122

init

int(* init)(AVBSFContext *ctx)

Definition: dts2pts.c:368

AV_CODEC_CAP_DR1

#define AV_CODEC_CAP_DR1

Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.

Definition: codec.h:52

AVPacket::size

int size

Definition: packet.h:540

RV_GET_MICRO_VER

#define RV_GET_MICRO_VER(x)

Definition: rv10.c:49

av_frame_ref

int av_frame_ref(AVFrame *dst, const AVFrame *src)

Set up a new reference to the data described by the source frame.

Definition: frame.c:388

codec_internal.h

ff_update_block_index

static void ff_update_block_index(MpegEncContext *s, int bits_per_raw_sample, int lowres, int chroma_x_shift)

Definition: mpegvideo.h:609

size

int size

Definition: twinvq_data.h:10344

VLCElem

Definition: vlc.h:32

AV_RB32

uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_RB32

Definition: bytestream.h:96

ff_mpv_frame_start

int ff_mpv_frame_start(MpegEncContext *s, AVCodecContext *avctx)

generic function called after decoding the header and before a frame is decoded.

Definition: mpegvideo_dec.c:344

ff_mpeg_flush

void ff_mpeg_flush(AVCodecContext *avctx)

Definition: mpegvideo_dec.c:449

RVDecContext::m

MpegEncContext m

Definition: rv10.c:55

mpegvideodata.h

offset

it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf offset

Definition: writing_filters.txt:86

rv10_decode_packet

static int rv10_decode_packet(AVCodecContext *avctx, const uint8_t *buf, int buf_size, int buf_size2, int whole_size)

Definition: rv10.c:408

AV_LOG_INFO

#define AV_LOG_INFO

Standard information.

Definition: log.h:191

rv_sym_run_len

static const uint8_t rv_sym_run_len[][2]

Definition: rv10.c:64

ff_rv_decode_dc

int ff_rv_decode_dc(MpegEncContext *s, int n)

Definition: rv10.c:83

rv10_init_static

static av_cold void rv10_init_static(void)

Definition: rv10.c:326

ff_print_debug_info

void ff_print_debug_info(const MpegEncContext *s, const MPVPicture *p, AVFrame *pict)

Definition: mpegvideo_dec.c:403

AV_CODEC_ID_RV10

@ AV_CODEC_ID_RV10

Definition: codec_id.h:57

#define i(width, name, range_min, range_max)

Definition: cbs_h2645.c:256

code

and forward the test the status of outputs and forward it to the corresponding return FFERROR_NOT_READY If the filters stores internally one or a few frame for some it can consider them to be part of the FIFO and delay acknowledging a status change accordingly Example code

Definition: filter_design.txt:178

AVCodecContext::extradata

uint8_t * extradata

some codecs need / can use extradata like Huffman tables.

Definition: avcodec.h:529

show_bits

static unsigned int show_bits(GetBitContext *s, int n)

Show 1-25 bits.

Definition: get_bits.h:371

av_assert1

#define av_assert1(cond)

assert() equivalent, that does not lie in speed critical code.

Definition: avassert.h:56

AV_CODEC_ID_RV20

@ AV_CODEC_ID_RV20

Definition: codec_id.h:58

AVCodec::name

const char * name

Name of the codec implementation.

Definition: codec.h:194

ff_h263_decode_init

av_cold int ff_h263_decode_init(AVCodecContext *avctx)

Definition: h263dec.c:90

ff_rv10_decoder

const FFCodec ff_rv10_decoder

Definition: rv10.c:637

avcodec.h

AVCodecContext::frame_num

int64_t frame_num

Frame counter, set by libavcodec.

Definition: avcodec.h:2041

get_slice_offset

static int get_slice_offset(AVCodecContext *avctx, const uint8_t *buf, int n)

Definition: rv10.c:547

ret

Definition: filter_design.txt:187

SLICE_OK

#define SLICE_OK

Definition: mpegvideo.h:480

ff_aic_dc_scale_table

const uint8_t ff_aic_dc_scale_table[32]

Definition: h263data.c:245

rv20_decode_picture_header

static int rv20_decode_picture_header(RVDecContext *rv, int whole_size)

Definition: rv10.c:158

ff_h263_loop_filter

void ff_h263_loop_filter(MpegEncContext *s)

Definition: h263.c:94

ff_mpeg4_init_direct_mv

void ff_mpeg4_init_direct_mv(MpegEncContext *s)

Definition: mpeg4video.c:83

left

Tag MUST be and< 10hcoeff half pel interpolation filter coefficients, hcoeff[0] are the 2 middle coefficients[1] are the next outer ones and so on, resulting in a filter like:...eff[2], hcoeff[1], hcoeff[0], hcoeff[0], hcoeff[1], hcoeff[2] ... the sign of the coefficients is not explicitly stored but alternates after each coeff and coeff[0] is positive, so ...,+,-,+,-,+,+,-,+,-,+,... hcoeff[0] is not explicitly stored but found by subtracting the sum of all stored coefficients with signs from 32 hcoeff[0]=32 - hcoeff[1] - hcoeff[2] - ... a good choice for hcoeff and htaps is htaps=6 hcoeff={40,-10, 2} an alternative which requires more computations at both encoder and decoder side and may or may not be better is htaps=8 hcoeff={42,-14, 6,-2}ref_frames minimum of the number of available reference frames and max_ref_frames for example the first frame after a key frame always has ref_frames=1spatial_decomposition_type wavelet type 0 is a 9/7 symmetric compact integer wavelet 1 is a 5/3 symmetric compact integer wavelet others are reserved stored as delta from last, last is reset to 0 if always_reset||keyframeqlog quality(logarithmic quantizer scale) stored as delta from last, last is reset to 0 if always_reset||keyframemv_scale stored as delta from last, last is reset to 0 if always_reset||keyframe FIXME check that everything works fine if this changes between framesqbias dequantization bias stored as delta from last, last is reset to 0 if always_reset||keyframeblock_max_depth maximum depth of the block tree stored as delta from last, last is reset to 0 if always_reset||keyframequant_table quantization tableHighlevel bitstream structure:==============================--------------------------------------------|Header|--------------------------------------------|------------------------------------|||Block0||||split?||||yes no||||......... intra?||||:Block01 :yes no||||:Block02 :....... ..........||||:Block03 ::y DC ::ref index:||||:Block04 ::cb DC ::motion x :||||......... :cr DC ::motion y :||||....... ..........|||------------------------------------||------------------------------------|||Block1|||...|--------------------------------------------|------------ ------------ ------------|||Y subbands||Cb subbands||Cr subbands||||--- ---||--- ---||--- ---|||||LL0||HL0||||LL0||HL0||||LL0||HL0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||LH0||HH0||||LH0||HH0||||LH0||HH0|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HL1||LH1||||HL1||LH1||||HL1||LH1|||||--- ---||--- ---||--- ---||||--- ---||--- ---||--- ---|||||HH1||HL2||||HH1||HL2||||HH1||HL2|||||...||...||...|||------------ ------------ ------------|--------------------------------------------Decoding process:=================------------|||Subbands|------------||||------------|Intra DC||||LL0 subband prediction ------------|\ Dequantization ------------------- \||Reference frames|\ IDWT|------- -------|Motion \|||Frame 0||Frame 1||Compensation . OBMC v -------|------- -------|--------------. \------> Frame n output Frame Frame<----------------------------------/|...|------------------- Range Coder:============Binary Range Coder:------------------- The implemented range coder is an adapted version based upon "Range encoding: an algorithm for removing redundancy from a digitised message." by G. N. N. Martin. The symbols encoded by the Snow range coder are bits(0|1). The associated probabilities are not fix but change depending on the symbol mix seen so far. bit seen|new state ---------+----------------------------------------------- 0|256 - state_transition_table[256 - old_state];1|state_transition_table[old_state];state_transition_table={ 0, 0, 0, 0, 0, 0, 0, 0, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 190, 191, 192, 194, 194, 195, 196, 197, 198, 199, 200, 201, 202, 202, 204, 205, 206, 207, 208, 209, 209, 210, 211, 212, 213, 215, 215, 216, 217, 218, 219, 220, 220, 222, 223, 224, 225, 226, 227, 227, 229, 229, 230, 231, 232, 234, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 248, 0, 0, 0, 0, 0, 0, 0};FIXME Range Coding of integers:------------------------- FIXME Neighboring Blocks:===================left and top are set to the respective blocks unless they are outside of the image in which case they are set to the Null block top-left is set to the top left block unless it is outside of the image in which case it is set to the left block if this block has no larger parent block or it is at the left side of its parent block and the top right block is not outside of the image then the top right block is used for top-right else the top-left block is used Null block y, cb, cr are 128 level, ref, mx and my are 0 Motion Vector Prediction:=========================1. the motion vectors of all the neighboring blocks are scaled to compensate for the difference of reference frames scaled_mv=(mv *(256 *(current_reference+1)/(mv.reference+1))+128)> the median of the scaled left

Definition: snow.txt:386

AV_RL32

uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32

Definition: bytestream.h:92

rv_lum_len_count

static const uint16_t rv_lum_len_count[15]

Definition: rv10.c:73

AVCodecContext

main external API structure.

Definition: avcodec.h:451

VLC_INIT_STATIC_OVERLONG

#define VLC_INIT_STATIC_OVERLONG