Mercurial > libavcodec.hg
comparison mpeg12enc.c @ 5208:430f60820b21 libavcodec
split mpeg12 encoder out of mpeg12.c
author | aurel |
---|---|
date | Thu, 05 Jul 2007 09:08:28 +0000 |
parents | |
children | b52fd9b223bb |
comparison
equal
deleted
inserted
replaced
5207:429a90b631a5 | 5208:430f60820b21 |
---|---|
1 /* | |
2 * MPEG1/2 encoder | |
3 * Copyright (c) 2000,2001 Fabrice Bellard. | |
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at> | |
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 */ | |
22 | |
23 /** | |
24 * @file mpeg12enc.c | |
25 * MPEG1/2 encoder | |
26 */ | |
27 | |
28 #include "avcodec.h" | |
29 #include "dsputil.h" | |
30 #include "mpegvideo.h" | |
31 | |
32 #include "mpeg12.h" | |
33 #include "mpeg12data.h" | |
34 #include "bytestream.h" | |
35 | |
36 | |
37 static const uint8_t inv_non_linear_qscale[13] = { | |
38 0, 2, 4, 6, 8, | |
39 9,10,11,12,13,14,15,16, | |
40 }; | |
41 | |
42 static const uint8_t svcd_scan_offset_placeholder[14] = { | |
43 0x10, 0x0E, | |
44 0x00, 0x80, 0x81, | |
45 0x00, 0x80, 0x81, | |
46 0xff, 0xff, 0xff, | |
47 0xff, 0xff, 0xff, | |
48 }; | |
49 | |
50 static void mpeg1_encode_block(MpegEncContext *s, | |
51 DCTELEM *block, | |
52 int component); | |
53 static void mpeg1_encode_motion(MpegEncContext *s, int val, int f_or_b_code); // RAL: f_code parameter added | |
54 | |
55 static uint8_t mv_penalty[MAX_FCODE+1][MAX_MV*2+1]; | |
56 static uint8_t fcode_tab[MAX_MV*2+1]; | |
57 | |
58 static uint8_t uni_mpeg1_ac_vlc_len [64*64*2]; | |
59 static uint8_t uni_mpeg2_ac_vlc_len [64*64*2]; | |
60 | |
61 /* simple include everything table for dc, first byte is bits number next 3 are code*/ | |
62 static uint32_t mpeg1_lum_dc_uni[512]; | |
63 static uint32_t mpeg1_chr_dc_uni[512]; | |
64 | |
65 static uint8_t mpeg1_index_run[2][64]; | |
66 static int8_t mpeg1_max_level[2][64]; | |
67 | |
68 static void init_uni_ac_vlc(RLTable *rl, uint8_t *uni_ac_vlc_len){ | |
69 int i; | |
70 | |
71 for(i=0; i<128; i++){ | |
72 int level= i-64; | |
73 int run; | |
74 for(run=0; run<64; run++){ | |
75 int len, bits, code; | |
76 | |
77 int alevel= FFABS(level); | |
78 int sign= (level>>31)&1; | |
79 | |
80 if (alevel > rl->max_level[0][run]) | |
81 code= 111; /*rl->n*/ | |
82 else | |
83 code= rl->index_run[0][run] + alevel - 1; | |
84 | |
85 if (code < 111 /* rl->n */) { | |
86 /* store the vlc & sign at once */ | |
87 len= rl->table_vlc[code][1]+1; | |
88 bits= (rl->table_vlc[code][0]<<1) + sign; | |
89 } else { | |
90 len= rl->table_vlc[111/*rl->n*/][1]+6; | |
91 bits= rl->table_vlc[111/*rl->n*/][0]<<6; | |
92 | |
93 bits|= run; | |
94 if (alevel < 128) { | |
95 bits<<=8; len+=8; | |
96 bits|= level & 0xff; | |
97 } else { | |
98 bits<<=16; len+=16; | |
99 bits|= level & 0xff; | |
100 if (level < 0) { | |
101 bits|= 0x8001 + level + 255; | |
102 } else { | |
103 bits|= level & 0xffff; | |
104 } | |
105 } | |
106 } | |
107 | |
108 uni_ac_vlc_len [UNI_AC_ENC_INDEX(run, i)]= len; | |
109 } | |
110 } | |
111 } | |
112 | |
113 | |
114 static int find_frame_rate_index(MpegEncContext *s){ | |
115 int i; | |
116 int64_t dmin= INT64_MAX; | |
117 int64_t d; | |
118 | |
119 for(i=1;i<14;i++) { | |
120 int64_t n0= 1001LL/ff_frame_rate_tab[i].den*ff_frame_rate_tab[i].num*s->avctx->time_base.num; | |
121 int64_t n1= 1001LL*s->avctx->time_base.den; | |
122 if(s->avctx->strict_std_compliance > FF_COMPLIANCE_INOFFICIAL && i>=9) break; | |
123 | |
124 d = FFABS(n0 - n1); | |
125 if(d < dmin){ | |
126 dmin=d; | |
127 s->frame_rate_index= i; | |
128 } | |
129 } | |
130 if(dmin) | |
131 return -1; | |
132 else | |
133 return 0; | |
134 } | |
135 | |
136 static int encode_init(AVCodecContext *avctx) | |
137 { | |
138 MpegEncContext *s = avctx->priv_data; | |
139 | |
140 if(MPV_encode_init(avctx) < 0) | |
141 return -1; | |
142 | |
143 if(find_frame_rate_index(s) < 0){ | |
144 if(s->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL){ | |
145 av_log(avctx, AV_LOG_ERROR, "MPEG1/2 does not support %d/%d fps\n", avctx->time_base.den, avctx->time_base.num); | |
146 return -1; | |
147 }else{ | |
148 av_log(avctx, AV_LOG_INFO, "MPEG1/2 does not support %d/%d fps, there may be AV sync issues\n", avctx->time_base.den, avctx->time_base.num); | |
149 } | |
150 } | |
151 | |
152 if(avctx->profile == FF_PROFILE_UNKNOWN){ | |
153 if(avctx->level != FF_LEVEL_UNKNOWN){ | |
154 av_log(avctx, AV_LOG_ERROR, "Set profile and level\n"); | |
155 return -1; | |
156 } | |
157 avctx->profile = s->chroma_format == CHROMA_420 ? 4 : 0; /* Main or 4:2:2 */ | |
158 } | |
159 | |
160 if(avctx->level == FF_LEVEL_UNKNOWN){ | |
161 if(avctx->profile == 0){ /* 4:2:2 */ | |
162 if(avctx->width <= 720 && avctx->height <= 608) avctx->level = 5; /* Main */ | |
163 else avctx->level = 2; /* High */ | |
164 }else{ | |
165 if(avctx->profile != 1 && s->chroma_format != CHROMA_420){ | |
166 av_log(avctx, AV_LOG_ERROR, "Only High(1) and 4:2:2(0) profiles support 4:2:2 color sampling\n"); | |
167 return -1; | |
168 } | |
169 if(avctx->width <= 720 && avctx->height <= 576) avctx->level = 8; /* Main */ | |
170 else if(avctx->width <= 1440) avctx->level = 6; /* High 1440 */ | |
171 else avctx->level = 4; /* High */ | |
172 } | |
173 } | |
174 | |
175 if((avctx->flags2 & CODEC_FLAG2_DROP_FRAME_TIMECODE) && s->frame_rate_index != 4){ | |
176 av_log(avctx, AV_LOG_ERROR, "Drop frame time code only allowed with 1001/30000 fps\n"); | |
177 return -1; | |
178 } | |
179 | |
180 return 0; | |
181 } | |
182 | |
183 static void put_header(MpegEncContext *s, int header) | |
184 { | |
185 align_put_bits(&s->pb); | |
186 put_bits(&s->pb, 16, header>>16); | |
187 put_bits(&s->pb, 16, header&0xFFFF); | |
188 } | |
189 | |
190 /* put sequence header if needed */ | |
191 static void mpeg1_encode_sequence_header(MpegEncContext *s) | |
192 { | |
193 unsigned int vbv_buffer_size; | |
194 unsigned int fps, v; | |
195 int i; | |
196 uint64_t time_code; | |
197 float best_aspect_error= 1E10; | |
198 float aspect_ratio= av_q2d(s->avctx->sample_aspect_ratio); | |
199 int constraint_parameter_flag; | |
200 | |
201 if(aspect_ratio==0.0) aspect_ratio= 1.0; //pixel aspect 1:1 (VGA) | |
202 | |
203 if (s->current_picture.key_frame) { | |
204 AVRational framerate= ff_frame_rate_tab[s->frame_rate_index]; | |
205 | |
206 /* mpeg1 header repeated every gop */ | |
207 put_header(s, SEQ_START_CODE); | |
208 | |
209 put_bits(&s->pb, 12, s->width); | |
210 put_bits(&s->pb, 12, s->height); | |
211 | |
212 for(i=1; i<15; i++){ | |
213 float error= aspect_ratio; | |
214 if(s->codec_id == CODEC_ID_MPEG1VIDEO || i <=1) | |
215 error-= 1.0/mpeg1_aspect[i]; | |
216 else | |
217 error-= av_q2d(mpeg2_aspect[i])*s->height/s->width; | |
218 | |
219 error= FFABS(error); | |
220 | |
221 if(error < best_aspect_error){ | |
222 best_aspect_error= error; | |
223 s->aspect_ratio_info= i; | |
224 } | |
225 } | |
226 | |
227 put_bits(&s->pb, 4, s->aspect_ratio_info); | |
228 put_bits(&s->pb, 4, s->frame_rate_index); | |
229 | |
230 if(s->avctx->rc_max_rate){ | |
231 v = (s->avctx->rc_max_rate + 399) / 400; | |
232 if (v > 0x3ffff && s->codec_id == CODEC_ID_MPEG1VIDEO) | |
233 v = 0x3ffff; | |
234 }else{ | |
235 v= 0x3FFFF; | |
236 } | |
237 | |
238 if(s->avctx->rc_buffer_size) | |
239 vbv_buffer_size = s->avctx->rc_buffer_size; | |
240 else | |
241 /* VBV calculation: Scaled so that a VCD has the proper VBV size of 40 kilobytes */ | |
242 vbv_buffer_size = (( 20 * s->bit_rate) / (1151929 / 2)) * 8 * 1024; | |
243 vbv_buffer_size= (vbv_buffer_size + 16383) / 16384; | |
244 | |
245 put_bits(&s->pb, 18, v & 0x3FFFF); | |
246 put_bits(&s->pb, 1, 1); /* marker */ | |
247 put_bits(&s->pb, 10, vbv_buffer_size & 0x3FF); | |
248 | |
249 constraint_parameter_flag= | |
250 s->width <= 768 && s->height <= 576 && | |
251 s->mb_width * s->mb_height <= 396 && | |
252 s->mb_width * s->mb_height * framerate.num <= framerate.den*396*25 && | |
253 framerate.num <= framerate.den*30 && | |
254 s->avctx->me_range && s->avctx->me_range < 128 && | |
255 vbv_buffer_size <= 20 && | |
256 v <= 1856000/400 && | |
257 s->codec_id == CODEC_ID_MPEG1VIDEO; | |
258 | |
259 put_bits(&s->pb, 1, constraint_parameter_flag); | |
260 | |
261 ff_write_quant_matrix(&s->pb, s->avctx->intra_matrix); | |
262 ff_write_quant_matrix(&s->pb, s->avctx->inter_matrix); | |
263 | |
264 if(s->codec_id == CODEC_ID_MPEG2VIDEO){ | |
265 put_header(s, EXT_START_CODE); | |
266 put_bits(&s->pb, 4, 1); //seq ext | |
267 | |
268 put_bits(&s->pb, 1, s->avctx->profile == 0); //escx 1 for 4:2:2 profile */ | |
269 | |
270 put_bits(&s->pb, 3, s->avctx->profile); //profile | |
271 put_bits(&s->pb, 4, s->avctx->level); //level | |
272 | |
273 put_bits(&s->pb, 1, s->progressive_sequence); | |
274 put_bits(&s->pb, 2, s->chroma_format); | |
275 put_bits(&s->pb, 2, 0); //horizontal size ext | |
276 put_bits(&s->pb, 2, 0); //vertical size ext | |
277 put_bits(&s->pb, 12, v>>18); //bitrate ext | |
278 put_bits(&s->pb, 1, 1); //marker | |
279 put_bits(&s->pb, 8, vbv_buffer_size >>10); //vbv buffer ext | |
280 put_bits(&s->pb, 1, s->low_delay); | |
281 put_bits(&s->pb, 2, 0); // frame_rate_ext_n | |
282 put_bits(&s->pb, 5, 0); // frame_rate_ext_d | |
283 } | |
284 | |
285 put_header(s, GOP_START_CODE); | |
286 put_bits(&s->pb, 1, !!(s->avctx->flags2 & CODEC_FLAG2_DROP_FRAME_TIMECODE)); /* drop frame flag */ | |
287 /* time code : we must convert from the real frame rate to a | |
288 fake mpeg frame rate in case of low frame rate */ | |
289 fps = (framerate.num + framerate.den/2)/ framerate.den; | |
290 time_code = s->current_picture_ptr->coded_picture_number + s->avctx->timecode_frame_start; | |
291 | |
292 s->gop_picture_number = s->current_picture_ptr->coded_picture_number; | |
293 if (s->avctx->flags2 & CODEC_FLAG2_DROP_FRAME_TIMECODE) { | |
294 /* only works for NTSC 29.97 */ | |
295 int d = time_code / 17982; | |
296 int m = time_code % 17982; | |
297 //if (m < 2) m += 2; /* not needed since -2,-1 / 1798 in C returns 0 */ | |
298 time_code += 18 * d + 2 * ((m - 2) / 1798); | |
299 } | |
300 put_bits(&s->pb, 5, (uint32_t)((time_code / (fps * 3600)) % 24)); | |
301 put_bits(&s->pb, 6, (uint32_t)((time_code / (fps * 60)) % 60)); | |
302 put_bits(&s->pb, 1, 1); | |
303 put_bits(&s->pb, 6, (uint32_t)((time_code / fps) % 60)); | |
304 put_bits(&s->pb, 6, (uint32_t)((time_code % fps))); | |
305 put_bits(&s->pb, 1, !!(s->flags & CODEC_FLAG_CLOSED_GOP)); | |
306 put_bits(&s->pb, 1, 0); /* broken link */ | |
307 } | |
308 } | |
309 | |
310 static inline void encode_mb_skip_run(MpegEncContext *s, int run){ | |
311 while (run >= 33) { | |
312 put_bits(&s->pb, 11, 0x008); | |
313 run -= 33; | |
314 } | |
315 put_bits(&s->pb, mbAddrIncrTable[run][1], | |
316 mbAddrIncrTable[run][0]); | |
317 } | |
318 | |
319 static av_always_inline void put_qscale(MpegEncContext *s) | |
320 { | |
321 if(s->q_scale_type){ | |
322 assert(s->qscale>=1 && s->qscale <=12); | |
323 put_bits(&s->pb, 5, inv_non_linear_qscale[s->qscale]); | |
324 }else{ | |
325 put_bits(&s->pb, 5, s->qscale); | |
326 } | |
327 } | |
328 | |
329 void ff_mpeg1_encode_slice_header(MpegEncContext *s){ | |
330 put_header(s, SLICE_MIN_START_CODE + s->mb_y); | |
331 put_qscale(s); | |
332 put_bits(&s->pb, 1, 0); /* slice extra information */ | |
333 } | |
334 | |
335 void mpeg1_encode_picture_header(MpegEncContext *s, int picture_number) | |
336 { | |
337 mpeg1_encode_sequence_header(s); | |
338 | |
339 /* mpeg1 picture header */ | |
340 put_header(s, PICTURE_START_CODE); | |
341 /* temporal reference */ | |
342 | |
343 // RAL: s->picture_number instead of s->fake_picture_number | |
344 put_bits(&s->pb, 10, (s->picture_number - | |
345 s->gop_picture_number) & 0x3ff); | |
346 put_bits(&s->pb, 3, s->pict_type); | |
347 | |
348 s->vbv_delay_ptr= s->pb.buf + put_bits_count(&s->pb)/8; | |
349 put_bits(&s->pb, 16, 0xFFFF); /* vbv_delay */ | |
350 | |
351 // RAL: Forward f_code also needed for B frames | |
352 if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) { | |
353 put_bits(&s->pb, 1, 0); /* half pel coordinates */ | |
354 if(s->codec_id == CODEC_ID_MPEG1VIDEO) | |
355 put_bits(&s->pb, 3, s->f_code); /* forward_f_code */ | |
356 else | |
357 put_bits(&s->pb, 3, 7); /* forward_f_code */ | |
358 } | |
359 | |
360 // RAL: Backward f_code necessary for B frames | |
361 if (s->pict_type == B_TYPE) { | |
362 put_bits(&s->pb, 1, 0); /* half pel coordinates */ | |
363 if(s->codec_id == CODEC_ID_MPEG1VIDEO) | |
364 put_bits(&s->pb, 3, s->b_code); /* backward_f_code */ | |
365 else | |
366 put_bits(&s->pb, 3, 7); /* backward_f_code */ | |
367 } | |
368 | |
369 put_bits(&s->pb, 1, 0); /* extra bit picture */ | |
370 | |
371 s->frame_pred_frame_dct = 1; | |
372 if(s->codec_id == CODEC_ID_MPEG2VIDEO){ | |
373 put_header(s, EXT_START_CODE); | |
374 put_bits(&s->pb, 4, 8); //pic ext | |
375 if (s->pict_type == P_TYPE || s->pict_type == B_TYPE) { | |
376 put_bits(&s->pb, 4, s->f_code); | |
377 put_bits(&s->pb, 4, s->f_code); | |
378 }else{ | |
379 put_bits(&s->pb, 8, 255); | |
380 } | |
381 if (s->pict_type == B_TYPE) { | |
382 put_bits(&s->pb, 4, s->b_code); | |
383 put_bits(&s->pb, 4, s->b_code); | |
384 }else{ | |
385 put_bits(&s->pb, 8, 255); | |
386 } | |
387 put_bits(&s->pb, 2, s->intra_dc_precision); | |
388 | |
389 assert(s->picture_structure == PICT_FRAME); | |
390 put_bits(&s->pb, 2, s->picture_structure); | |
391 if (s->progressive_sequence) { | |
392 put_bits(&s->pb, 1, 0); /* no repeat */ | |
393 } else { | |
394 put_bits(&s->pb, 1, s->current_picture_ptr->top_field_first); | |
395 } | |
396 /* XXX: optimize the generation of this flag with entropy | |
397 measures */ | |
398 s->frame_pred_frame_dct = s->progressive_sequence; | |
399 | |
400 put_bits(&s->pb, 1, s->frame_pred_frame_dct); | |
401 put_bits(&s->pb, 1, s->concealment_motion_vectors); | |
402 put_bits(&s->pb, 1, s->q_scale_type); | |
403 put_bits(&s->pb, 1, s->intra_vlc_format); | |
404 put_bits(&s->pb, 1, s->alternate_scan); | |
405 put_bits(&s->pb, 1, s->repeat_first_field); | |
406 s->progressive_frame = s->progressive_sequence; | |
407 put_bits(&s->pb, 1, s->chroma_format == CHROMA_420 ? s->progressive_frame : 0); /* chroma_420_type */ | |
408 put_bits(&s->pb, 1, s->progressive_frame); | |
409 put_bits(&s->pb, 1, 0); //composite_display_flag | |
410 } | |
411 if(s->flags & CODEC_FLAG_SVCD_SCAN_OFFSET){ | |
412 int i; | |
413 | |
414 put_header(s, USER_START_CODE); | |
415 for(i=0; i<sizeof(svcd_scan_offset_placeholder); i++){ | |
416 put_bits(&s->pb, 8, svcd_scan_offset_placeholder[i]); | |
417 } | |
418 } | |
419 | |
420 s->mb_y=0; | |
421 ff_mpeg1_encode_slice_header(s); | |
422 } | |
423 | |
424 static inline void put_mb_modes(MpegEncContext *s, int n, int bits, | |
425 int has_mv, int field_motion) | |
426 { | |
427 put_bits(&s->pb, n, bits); | |
428 if (!s->frame_pred_frame_dct) { | |
429 if (has_mv) | |
430 put_bits(&s->pb, 2, 2 - field_motion); /* motion_type: frame/field */ | |
431 put_bits(&s->pb, 1, s->interlaced_dct); | |
432 } | |
433 } | |
434 | |
435 static av_always_inline void mpeg1_encode_mb_internal(MpegEncContext *s, | |
436 DCTELEM block[6][64], | |
437 int motion_x, int motion_y, | |
438 int mb_block_count) | |
439 { | |
440 int i, cbp; | |
441 const int mb_x = s->mb_x; | |
442 const int mb_y = s->mb_y; | |
443 const int first_mb= mb_x == s->resync_mb_x && mb_y == s->resync_mb_y; | |
444 | |
445 /* compute cbp */ | |
446 cbp = 0; | |
447 for(i=0;i<mb_block_count;i++) { | |
448 if (s->block_last_index[i] >= 0) | |
449 cbp |= 1 << (mb_block_count - 1 - i); | |
450 } | |
451 | |
452 if (cbp == 0 && !first_mb && s->mv_type == MV_TYPE_16X16 && | |
453 (mb_x != s->mb_width - 1 || (mb_y != s->mb_height - 1 && s->codec_id == CODEC_ID_MPEG1VIDEO)) && | |
454 ((s->pict_type == P_TYPE && (motion_x | motion_y) == 0) || | |
455 (s->pict_type == B_TYPE && s->mv_dir == s->last_mv_dir && (((s->mv_dir & MV_DIR_FORWARD) ? ((s->mv[0][0][0] - s->last_mv[0][0][0])|(s->mv[0][0][1] - s->last_mv[0][0][1])) : 0) | | |
456 ((s->mv_dir & MV_DIR_BACKWARD) ? ((s->mv[1][0][0] - s->last_mv[1][0][0])|(s->mv[1][0][1] - s->last_mv[1][0][1])) : 0)) == 0))) { | |
457 s->mb_skip_run++; | |
458 s->qscale -= s->dquant; | |
459 s->skip_count++; | |
460 s->misc_bits++; | |
461 s->last_bits++; | |
462 if(s->pict_type == P_TYPE){ | |
463 s->last_mv[0][1][0]= s->last_mv[0][0][0]= | |
464 s->last_mv[0][1][1]= s->last_mv[0][0][1]= 0; | |
465 } | |
466 } else { | |
467 if(first_mb){ | |
468 assert(s->mb_skip_run == 0); | |
469 encode_mb_skip_run(s, s->mb_x); | |
470 }else{ | |
471 encode_mb_skip_run(s, s->mb_skip_run); | |
472 } | |
473 | |
474 if (s->pict_type == I_TYPE) { | |
475 if(s->dquant && cbp){ | |
476 put_mb_modes(s, 2, 1, 0, 0); /* macroblock_type : macroblock_quant = 1 */ | |
477 put_qscale(s); | |
478 }else{ | |
479 put_mb_modes(s, 1, 1, 0, 0); /* macroblock_type : macroblock_quant = 0 */ | |
480 s->qscale -= s->dquant; | |
481 } | |
482 s->misc_bits+= get_bits_diff(s); | |
483 s->i_count++; | |
484 } else if (s->mb_intra) { | |
485 if(s->dquant && cbp){ | |
486 put_mb_modes(s, 6, 0x01, 0, 0); | |
487 put_qscale(s); | |
488 }else{ | |
489 put_mb_modes(s, 5, 0x03, 0, 0); | |
490 s->qscale -= s->dquant; | |
491 } | |
492 s->misc_bits+= get_bits_diff(s); | |
493 s->i_count++; | |
494 memset(s->last_mv, 0, sizeof(s->last_mv)); | |
495 } else if (s->pict_type == P_TYPE) { | |
496 if(s->mv_type == MV_TYPE_16X16){ | |
497 if (cbp != 0) { | |
498 if ((motion_x|motion_y) == 0) { | |
499 if(s->dquant){ | |
500 put_mb_modes(s, 5, 1, 0, 0); /* macroblock_pattern & quant */ | |
501 put_qscale(s); | |
502 }else{ | |
503 put_mb_modes(s, 2, 1, 0, 0); /* macroblock_pattern only */ | |
504 } | |
505 s->misc_bits+= get_bits_diff(s); | |
506 } else { | |
507 if(s->dquant){ | |
508 put_mb_modes(s, 5, 2, 1, 0); /* motion + cbp */ | |
509 put_qscale(s); | |
510 }else{ | |
511 put_mb_modes(s, 1, 1, 1, 0); /* motion + cbp */ | |
512 } | |
513 s->misc_bits+= get_bits_diff(s); | |
514 mpeg1_encode_motion(s, motion_x - s->last_mv[0][0][0], s->f_code); // RAL: f_code parameter added | |
515 mpeg1_encode_motion(s, motion_y - s->last_mv[0][0][1], s->f_code); // RAL: f_code parameter added | |
516 s->mv_bits+= get_bits_diff(s); | |
517 } | |
518 } else { | |
519 put_bits(&s->pb, 3, 1); /* motion only */ | |
520 if (!s->frame_pred_frame_dct) | |
521 put_bits(&s->pb, 2, 2); /* motion_type: frame */ | |
522 s->misc_bits+= get_bits_diff(s); | |
523 mpeg1_encode_motion(s, motion_x - s->last_mv[0][0][0], s->f_code); // RAL: f_code parameter added | |
524 mpeg1_encode_motion(s, motion_y - s->last_mv[0][0][1], s->f_code); // RAL: f_code parameter added | |
525 s->qscale -= s->dquant; | |
526 s->mv_bits+= get_bits_diff(s); | |
527 } | |
528 s->last_mv[0][1][0]= s->last_mv[0][0][0]= motion_x; | |
529 s->last_mv[0][1][1]= s->last_mv[0][0][1]= motion_y; | |
530 }else{ | |
531 assert(!s->frame_pred_frame_dct && s->mv_type == MV_TYPE_FIELD); | |
532 | |
533 if (cbp) { | |
534 if(s->dquant){ | |
535 put_mb_modes(s, 5, 2, 1, 1); /* motion + cbp */ | |
536 put_qscale(s); | |
537 }else{ | |
538 put_mb_modes(s, 1, 1, 1, 1); /* motion + cbp */ | |
539 } | |
540 } else { | |
541 put_bits(&s->pb, 3, 1); /* motion only */ | |
542 put_bits(&s->pb, 2, 1); /* motion_type: field */ | |
543 s->qscale -= s->dquant; | |
544 } | |
545 s->misc_bits+= get_bits_diff(s); | |
546 for(i=0; i<2; i++){ | |
547 put_bits(&s->pb, 1, s->field_select[0][i]); | |
548 mpeg1_encode_motion(s, s->mv[0][i][0] - s->last_mv[0][i][0] , s->f_code); | |
549 mpeg1_encode_motion(s, s->mv[0][i][1] - (s->last_mv[0][i][1]>>1), s->f_code); | |
550 s->last_mv[0][i][0]= s->mv[0][i][0]; | |
551 s->last_mv[0][i][1]= 2*s->mv[0][i][1]; | |
552 } | |
553 s->mv_bits+= get_bits_diff(s); | |
554 } | |
555 if(cbp) { | |
556 if (s->chroma_y_shift) { | |
557 put_bits(&s->pb, mbPatTable[cbp][1], mbPatTable[cbp][0]); | |
558 } else { | |
559 put_bits(&s->pb, mbPatTable[cbp>>2][1], mbPatTable[cbp>>2][0]); | |
560 put_bits(&s->pb, 2, cbp & 3); | |
561 } | |
562 } | |
563 s->f_count++; | |
564 } else{ | |
565 static const int mb_type_len[4]={0,3,4,2}; //bak,for,bi | |
566 | |
567 if(s->mv_type == MV_TYPE_16X16){ | |
568 if (cbp){ // With coded bloc pattern | |
569 if (s->dquant) { | |
570 if(s->mv_dir == MV_DIR_FORWARD) | |
571 put_mb_modes(s, 6, 3, 1, 0); | |
572 else | |
573 put_mb_modes(s, mb_type_len[s->mv_dir]+3, 2, 1, 0); | |
574 put_qscale(s); | |
575 } else { | |
576 put_mb_modes(s, mb_type_len[s->mv_dir], 3, 1, 0); | |
577 } | |
578 }else{ // No coded bloc pattern | |
579 put_bits(&s->pb, mb_type_len[s->mv_dir], 2); | |
580 if (!s->frame_pred_frame_dct) | |
581 put_bits(&s->pb, 2, 2); /* motion_type: frame */ | |
582 s->qscale -= s->dquant; | |
583 } | |
584 s->misc_bits += get_bits_diff(s); | |
585 if (s->mv_dir&MV_DIR_FORWARD){ | |
586 mpeg1_encode_motion(s, s->mv[0][0][0] - s->last_mv[0][0][0], s->f_code); | |
587 mpeg1_encode_motion(s, s->mv[0][0][1] - s->last_mv[0][0][1], s->f_code); | |
588 s->last_mv[0][0][0]=s->last_mv[0][1][0]= s->mv[0][0][0]; | |
589 s->last_mv[0][0][1]=s->last_mv[0][1][1]= s->mv[0][0][1]; | |
590 s->f_count++; | |
591 } | |
592 if (s->mv_dir&MV_DIR_BACKWARD){ | |
593 mpeg1_encode_motion(s, s->mv[1][0][0] - s->last_mv[1][0][0], s->b_code); | |
594 mpeg1_encode_motion(s, s->mv[1][0][1] - s->last_mv[1][0][1], s->b_code); | |
595 s->last_mv[1][0][0]=s->last_mv[1][1][0]= s->mv[1][0][0]; | |
596 s->last_mv[1][0][1]=s->last_mv[1][1][1]= s->mv[1][0][1]; | |
597 s->b_count++; | |
598 } | |
599 }else{ | |
600 assert(s->mv_type == MV_TYPE_FIELD); | |
601 assert(!s->frame_pred_frame_dct); | |
602 if (cbp){ // With coded bloc pattern | |
603 if (s->dquant) { | |
604 if(s->mv_dir == MV_DIR_FORWARD) | |
605 put_mb_modes(s, 6, 3, 1, 1); | |
606 else | |
607 put_mb_modes(s, mb_type_len[s->mv_dir]+3, 2, 1, 1); | |
608 put_qscale(s); | |
609 } else { | |
610 put_mb_modes(s, mb_type_len[s->mv_dir], 3, 1, 1); | |
611 } | |
612 }else{ // No coded bloc pattern | |
613 put_bits(&s->pb, mb_type_len[s->mv_dir], 2); | |
614 put_bits(&s->pb, 2, 1); /* motion_type: field */ | |
615 s->qscale -= s->dquant; | |
616 } | |
617 s->misc_bits += get_bits_diff(s); | |
618 if (s->mv_dir&MV_DIR_FORWARD){ | |
619 for(i=0; i<2; i++){ | |
620 put_bits(&s->pb, 1, s->field_select[0][i]); | |
621 mpeg1_encode_motion(s, s->mv[0][i][0] - s->last_mv[0][i][0] , s->f_code); | |
622 mpeg1_encode_motion(s, s->mv[0][i][1] - (s->last_mv[0][i][1]>>1), s->f_code); | |
623 s->last_mv[0][i][0]= s->mv[0][i][0]; | |
624 s->last_mv[0][i][1]= 2*s->mv[0][i][1]; | |
625 } | |
626 s->f_count++; | |
627 } | |
628 if (s->mv_dir&MV_DIR_BACKWARD){ | |
629 for(i=0; i<2; i++){ | |
630 put_bits(&s->pb, 1, s->field_select[1][i]); | |
631 mpeg1_encode_motion(s, s->mv[1][i][0] - s->last_mv[1][i][0] , s->b_code); | |
632 mpeg1_encode_motion(s, s->mv[1][i][1] - (s->last_mv[1][i][1]>>1), s->b_code); | |
633 s->last_mv[1][i][0]= s->mv[1][i][0]; | |
634 s->last_mv[1][i][1]= 2*s->mv[1][i][1]; | |
635 } | |
636 s->b_count++; | |
637 } | |
638 } | |
639 s->mv_bits += get_bits_diff(s); | |
640 if(cbp) { | |
641 if (s->chroma_y_shift) { | |
642 put_bits(&s->pb, mbPatTable[cbp][1], mbPatTable[cbp][0]); | |
643 } else { | |
644 put_bits(&s->pb, mbPatTable[cbp>>2][1], mbPatTable[cbp>>2][0]); | |
645 put_bits(&s->pb, 2, cbp & 3); | |
646 } | |
647 } | |
648 } | |
649 for(i=0;i<mb_block_count;i++) { | |
650 if (cbp & (1 << (mb_block_count - 1 - i))) { | |
651 mpeg1_encode_block(s, block[i], i); | |
652 } | |
653 } | |
654 s->mb_skip_run = 0; | |
655 if(s->mb_intra) | |
656 s->i_tex_bits+= get_bits_diff(s); | |
657 else | |
658 s->p_tex_bits+= get_bits_diff(s); | |
659 } | |
660 } | |
661 | |
662 void mpeg1_encode_mb(MpegEncContext *s, DCTELEM block[6][64], int motion_x, int motion_y) | |
663 { | |
664 if (s->chroma_format == CHROMA_420) mpeg1_encode_mb_internal(s, block, motion_x, motion_y, 6); | |
665 else mpeg1_encode_mb_internal(s, block, motion_x, motion_y, 8); | |
666 } | |
667 | |
668 // RAL: Parameter added: f_or_b_code | |
669 static void mpeg1_encode_motion(MpegEncContext *s, int val, int f_or_b_code) | |
670 { | |
671 int code, bit_size, l, bits, range, sign; | |
672 | |
673 if (val == 0) { | |
674 /* zero vector */ | |
675 code = 0; | |
676 put_bits(&s->pb, | |
677 mbMotionVectorTable[0][1], | |
678 mbMotionVectorTable[0][0]); | |
679 } else { | |
680 bit_size = f_or_b_code - 1; | |
681 range = 1 << bit_size; | |
682 /* modulo encoding */ | |
683 l= INT_BIT - 5 - bit_size; | |
684 val= (val<<l)>>l; | |
685 | |
686 if (val >= 0) { | |
687 val--; | |
688 code = (val >> bit_size) + 1; | |
689 bits = val & (range - 1); | |
690 sign = 0; | |
691 } else { | |
692 val = -val; | |
693 val--; | |
694 code = (val >> bit_size) + 1; | |
695 bits = val & (range - 1); | |
696 sign = 1; | |
697 } | |
698 | |
699 assert(code > 0 && code <= 16); | |
700 | |
701 put_bits(&s->pb, | |
702 mbMotionVectorTable[code][1], | |
703 mbMotionVectorTable[code][0]); | |
704 | |
705 put_bits(&s->pb, 1, sign); | |
706 if (bit_size > 0) { | |
707 put_bits(&s->pb, bit_size, bits); | |
708 } | |
709 } | |
710 } | |
711 | |
712 void ff_mpeg1_encode_init(MpegEncContext *s) | |
713 { | |
714 static int done=0; | |
715 | |
716 common_init(s); | |
717 | |
718 if(!done){ | |
719 int f_code; | |
720 int mv; | |
721 int i; | |
722 | |
723 done=1; | |
724 init_rl(&rl_mpeg1, static_rl_table_store[0]); | |
725 init_rl(&rl_mpeg2, static_rl_table_store[1]); | |
726 | |
727 for(i=0; i<64; i++) | |
728 { | |
729 mpeg1_max_level[0][i]= rl_mpeg1.max_level[0][i]; | |
730 mpeg1_index_run[0][i]= rl_mpeg1.index_run[0][i]; | |
731 } | |
732 | |
733 init_uni_ac_vlc(&rl_mpeg1, uni_mpeg1_ac_vlc_len); | |
734 if(s->intra_vlc_format) | |
735 init_uni_ac_vlc(&rl_mpeg2, uni_mpeg2_ac_vlc_len); | |
736 | |
737 /* build unified dc encoding tables */ | |
738 for(i=-255; i<256; i++) | |
739 { | |
740 int adiff, index; | |
741 int bits, code; | |
742 int diff=i; | |
743 | |
744 adiff = FFABS(diff); | |
745 if(diff<0) diff--; | |
746 index = av_log2(2*adiff); | |
747 | |
748 bits= vlc_dc_lum_bits[index] + index; | |
749 code= (vlc_dc_lum_code[index]<<index) + (diff & ((1 << index) - 1)); | |
750 mpeg1_lum_dc_uni[i+255]= bits + (code<<8); | |
751 | |
752 bits= vlc_dc_chroma_bits[index] + index; | |
753 code= (vlc_dc_chroma_code[index]<<index) + (diff & ((1 << index) - 1)); | |
754 mpeg1_chr_dc_uni[i+255]= bits + (code<<8); | |
755 } | |
756 | |
757 for(f_code=1; f_code<=MAX_FCODE; f_code++){ | |
758 for(mv=-MAX_MV; mv<=MAX_MV; mv++){ | |
759 int len; | |
760 | |
761 if(mv==0) len= mbMotionVectorTable[0][1]; | |
762 else{ | |
763 int val, bit_size, range, code; | |
764 | |
765 bit_size = f_code - 1; | |
766 range = 1 << bit_size; | |
767 | |
768 val=mv; | |
769 if (val < 0) | |
770 val = -val; | |
771 val--; | |
772 code = (val >> bit_size) + 1; | |
773 if(code<17){ | |
774 len= mbMotionVectorTable[code][1] + 1 + bit_size; | |
775 }else{ | |
776 len= mbMotionVectorTable[16][1] + 2 + bit_size; | |
777 } | |
778 } | |
779 | |
780 mv_penalty[f_code][mv+MAX_MV]= len; | |
781 } | |
782 } | |
783 | |
784 | |
785 for(f_code=MAX_FCODE; f_code>0; f_code--){ | |
786 for(mv=-(8<<f_code); mv<(8<<f_code); mv++){ | |
787 fcode_tab[mv+MAX_MV]= f_code; | |
788 } | |
789 } | |
790 } | |
791 s->me.mv_penalty= mv_penalty; | |
792 s->fcode_tab= fcode_tab; | |
793 if(s->codec_id == CODEC_ID_MPEG1VIDEO){ | |
794 s->min_qcoeff=-255; | |
795 s->max_qcoeff= 255; | |
796 }else{ | |
797 s->min_qcoeff=-2047; | |
798 s->max_qcoeff= 2047; | |
799 } | |
800 if (s->intra_vlc_format) { | |
801 s->intra_ac_vlc_length= | |
802 s->intra_ac_vlc_last_length= uni_mpeg2_ac_vlc_len; | |
803 } else { | |
804 s->intra_ac_vlc_length= | |
805 s->intra_ac_vlc_last_length= uni_mpeg1_ac_vlc_len; | |
806 } | |
807 s->inter_ac_vlc_length= | |
808 s->inter_ac_vlc_last_length= uni_mpeg1_ac_vlc_len; | |
809 } | |
810 | |
811 static inline void encode_dc(MpegEncContext *s, int diff, int component) | |
812 { | |
813 if(((unsigned) (diff+255)) >= 511){ | |
814 int index; | |
815 | |
816 if(diff<0){ | |
817 index= av_log2_16bit(-2*diff); | |
818 diff--; | |
819 }else{ | |
820 index= av_log2_16bit(2*diff); | |
821 } | |
822 if (component == 0) { | |
823 put_bits( | |
824 &s->pb, | |
825 vlc_dc_lum_bits[index] + index, | |
826 (vlc_dc_lum_code[index]<<index) + (diff & ((1 << index) - 1))); | |
827 }else{ | |
828 put_bits( | |
829 &s->pb, | |
830 vlc_dc_chroma_bits[index] + index, | |
831 (vlc_dc_chroma_code[index]<<index) + (diff & ((1 << index) - 1))); | |
832 } | |
833 }else{ | |
834 if (component == 0) { | |
835 put_bits( | |
836 &s->pb, | |
837 mpeg1_lum_dc_uni[diff+255]&0xFF, | |
838 mpeg1_lum_dc_uni[diff+255]>>8); | |
839 } else { | |
840 put_bits( | |
841 &s->pb, | |
842 mpeg1_chr_dc_uni[diff+255]&0xFF, | |
843 mpeg1_chr_dc_uni[diff+255]>>8); | |
844 } | |
845 } | |
846 } | |
847 | |
848 static void mpeg1_encode_block(MpegEncContext *s, | |
849 DCTELEM *block, | |
850 int n) | |
851 { | |
852 int alevel, level, last_non_zero, dc, diff, i, j, run, last_index, sign; | |
853 int code, component; | |
854 const uint16_t (*table_vlc)[2] = rl_mpeg1.table_vlc; | |
855 | |
856 last_index = s->block_last_index[n]; | |
857 | |
858 /* DC coef */ | |
859 if (s->mb_intra) { | |
860 component = (n <= 3 ? 0 : (n&1) + 1); | |
861 dc = block[0]; /* overflow is impossible */ | |
862 diff = dc - s->last_dc[component]; | |
863 encode_dc(s, diff, component); | |
864 s->last_dc[component] = dc; | |
865 i = 1; | |
866 if (s->intra_vlc_format) | |
867 table_vlc = rl_mpeg2.table_vlc; | |
868 } else { | |
869 /* encode the first coefficient : needs to be done here because | |
870 it is handled slightly differently */ | |
871 level = block[0]; | |
872 if (abs(level) == 1) { | |
873 code = ((uint32_t)level >> 31); /* the sign bit */ | |
874 put_bits(&s->pb, 2, code | 0x02); | |
875 i = 1; | |
876 } else { | |
877 i = 0; | |
878 last_non_zero = -1; | |
879 goto next_coef; | |
880 } | |
881 } | |
882 | |
883 /* now quantify & encode AC coefs */ | |
884 last_non_zero = i - 1; | |
885 | |
886 for(;i<=last_index;i++) { | |
887 j = s->intra_scantable.permutated[i]; | |
888 level = block[j]; | |
889 next_coef: | |
890 #if 0 | |
891 if (level != 0) | |
892 dprintf(s->avctx, "level[%d]=%d\n", i, level); | |
893 #endif | |
894 /* encode using VLC */ | |
895 if (level != 0) { | |
896 run = i - last_non_zero - 1; | |
897 | |
898 alevel= level; | |
899 MASK_ABS(sign, alevel) | |
900 sign&=1; | |
901 | |
902 if (alevel <= mpeg1_max_level[0][run]){ | |
903 code= mpeg1_index_run[0][run] + alevel - 1; | |
904 /* store the vlc & sign at once */ | |
905 put_bits(&s->pb, table_vlc[code][1]+1, (table_vlc[code][0]<<1) + sign); | |
906 } else { | |
907 /* escape seems to be pretty rare <5% so I do not optimize it */ | |
908 put_bits(&s->pb, table_vlc[111][1], table_vlc[111][0]); | |
909 /* escape: only clip in this case */ | |
910 put_bits(&s->pb, 6, run); | |
911 if(s->codec_id == CODEC_ID_MPEG1VIDEO){ | |
912 if (alevel < 128) { | |
913 put_bits(&s->pb, 8, level & 0xff); | |
914 } else { | |
915 if (level < 0) { | |
916 put_bits(&s->pb, 16, 0x8001 + level + 255); | |
917 } else { | |
918 put_bits(&s->pb, 16, level & 0xffff); | |
919 } | |
920 } | |
921 }else{ | |
922 put_bits(&s->pb, 12, level & 0xfff); | |
923 } | |
924 } | |
925 last_non_zero = i; | |
926 } | |
927 } | |
928 /* end of block */ | |
929 put_bits(&s->pb, table_vlc[112][1], table_vlc[112][0]); | |
930 } | |
931 | |
932 AVCodec mpeg1video_encoder = { | |
933 "mpeg1video", | |
934 CODEC_TYPE_VIDEO, | |
935 CODEC_ID_MPEG1VIDEO, | |
936 sizeof(MpegEncContext), | |
937 encode_init, | |
938 MPV_encode_picture, | |
939 MPV_encode_end, | |
940 .supported_framerates= ff_frame_rate_tab+1, | |
941 .pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV420P, -1}, | |
942 .capabilities= CODEC_CAP_DELAY, | |
943 }; | |
944 | |
945 AVCodec mpeg2video_encoder = { | |
946 "mpeg2video", | |
947 CODEC_TYPE_VIDEO, | |
948 CODEC_ID_MPEG2VIDEO, | |
949 sizeof(MpegEncContext), | |
950 encode_init, | |
951 MPV_encode_picture, | |
952 MPV_encode_end, | |
953 .supported_framerates= ff_frame_rate_tab+1, | |
954 .pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, -1}, | |
955 .capabilities= CODEC_CAP_DELAY, | |
956 }; |