Mercurial > libavcodec.hg
annotate dv.c @ 3167:312b86bc4202 libavcodec
DVCPRO50 support.
Patch by Daniel Maas dmaas at maasdigital dot com
author | romansh |
---|---|
date | Mon, 06 Mar 2006 08:54:33 +0000 |
parents | 28a087ad41d1 |
children | 1deaaa111fbc |
rev | line source |
---|---|
723 | 1 /* |
2 * DV decoder | |
3 * Copyright (c) 2002 Fabrice Bellard. | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
4 * Copyright (c) 2004 Roman Shaposhnik. |
723 | 5 * |
2967 | 6 * DV encoder |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
7 * Copyright (c) 2003 Roman Shaposhnik. |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
8 * |
3167 | 9 * 50 Mbps (DVCPRO50) support |
10 * Copyright (c) 2006 Daniel Maas <dmaas@maasdigital.com> | |
11 * | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
12 * Many thanks to Dan Dennedy <dan@dennedy.org> for providing wealth |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
13 * of DV technical info. |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
14 * |
723 | 15 * This library is free software; you can redistribute it and/or |
16 * modify it under the terms of the GNU Lesser General Public | |
17 * License as published by the Free Software Foundation; either | |
18 * version 2 of the License, or (at your option) any later version. | |
19 * | |
20 * This library is distributed in the hope that it will be useful, | |
21 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
22 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
23 * Lesser General Public License for more details. | |
24 * | |
25 * You should have received a copy of the GNU Lesser General Public | |
26 * License along with this library; if not, write to the Free Software | |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
3016
diff
changeset
|
27 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
723 | 28 */ |
1106 | 29 |
30 /** | |
31 * @file dv.c | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
32 * DV codec. |
1106 | 33 */ |
723 | 34 #include "avcodec.h" |
35 #include "dsputil.h" | |
36 #include "mpegvideo.h" | |
37 #include "simple_idct.h" | |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
38 #include "dvdata.h" |
723 | 39 |
2847 | 40 //#undef NDEBUG |
41 //#include <assert.h> | |
42 | |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
43 typedef struct DVVideoContext { |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
44 const DVprofile* sys; |
925 | 45 AVFrame picture; |
2847 | 46 AVCodecContext *avctx; |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
47 uint8_t *buf; |
2967 | 48 |
1064 | 49 uint8_t dv_zigzag[2][64]; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
50 uint8_t dv_idct_shift[2][2][22][64]; |
2967 | 51 |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
52 void (*get_pixels)(DCTELEM *block, const uint8_t *pixels, int line_size); |
1567 | 53 void (*fdct[2])(DCTELEM *block); |
54 void (*idct_put[2])(uint8_t *dest, int line_size, DCTELEM *block); | |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
55 } DVVideoContext; |
2991
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
56 |
3167 | 57 /* MultiThreading - dv_anchor applies to entire DV codec, not just the avcontext */ |
58 /* one element is needed for each video segment in a DV frame */ | |
59 /* at most there are 2 DIF channels * 12 DIF sequences * 27 video segments (PAL 50Mbps) */ | |
60 #define DV_ANCHOR_SIZE (2*12*27) | |
61 | |
62 static void* dv_anchor[DV_ANCHOR_SIZE]; | |
723 | 63 |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
64 #define TEX_VLC_BITS 9 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
65 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
66 #ifdef DV_CODEC_TINY_TARGET |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
67 #define DV_VLC_MAP_RUN_SIZE 15 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
68 #define DV_VLC_MAP_LEV_SIZE 23 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
69 #else |
2967 | 70 #define DV_VLC_MAP_RUN_SIZE 64 |
2847 | 71 #define DV_VLC_MAP_LEV_SIZE 512 //FIXME sign was removed so this should be /2 but needs check |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
72 #endif |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
73 |
723 | 74 /* XXX: also include quantization */ |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
75 static RL_VLC_ELEM *dv_rl_vlc; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
76 /* VLC encoding lookup table */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
77 static struct dv_vlc_pair { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
78 uint32_t vlc; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
79 uint8_t size; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
80 } (*dv_vlc_map)[DV_VLC_MAP_LEV_SIZE] = NULL; |
723 | 81 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
82 static void dv_build_unquantize_tables(DVVideoContext *s, uint8_t* perm) |
723 | 83 { |
725 | 84 int i, q, j; |
723 | 85 |
86 /* NOTE: max left shift is 6 */ | |
725 | 87 for(q = 0; q < 22; q++) { |
1567 | 88 /* 88DCT */ |
725 | 89 for(i = 1; i < 64; i++) { |
90 /* 88 table */ | |
1567 | 91 j = perm[i]; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
92 s->dv_idct_shift[0][0][q][j] = |
725 | 93 dv_quant_shifts[q][dv_88_areas[i]] + 1; |
2979 | 94 s->dv_idct_shift[1][0][q][j] = s->dv_idct_shift[0][0][q][j] + 1; |
725 | 95 } |
2967 | 96 |
1567 | 97 /* 248DCT */ |
725 | 98 for(i = 1; i < 64; i++) { |
99 /* 248 table */ | |
2967 | 100 s->dv_idct_shift[0][1][q][i] = |
1567 | 101 dv_quant_shifts[q][dv_248_areas[i]] + 1; |
2979 | 102 s->dv_idct_shift[1][1][q][i] = s->dv_idct_shift[0][1][q][i] + 1; |
723 | 103 } |
104 } | |
105 } | |
106 | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
107 static int dvvideo_init(AVCodecContext *avctx) |
723 | 108 { |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
109 DVVideoContext *s = avctx->priv_data; |
1567 | 110 DSPContext dsp; |
1070
6da5ae9ee199
more #ifdef CONFIG_ENCODERS patch by (Wolfgang Hesseler <qv at multimediaware dot com>) with modifications by me (s/WOLFGANG/CONFIG_ENCODERS/ and some other fixes)
michaelni
parents:
1064
diff
changeset
|
111 static int done=0; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
112 int i, j; |
723 | 113 |
114 if (!done) { | |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
115 VLC dv_vlc; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
116 uint16_t new_dv_vlc_bits[NB_DV_VLC*2]; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
117 uint8_t new_dv_vlc_len[NB_DV_VLC*2]; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
118 uint8_t new_dv_vlc_run[NB_DV_VLC*2]; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
119 int16_t new_dv_vlc_level[NB_DV_VLC*2]; |
723 | 120 |
121 done = 1; | |
122 | |
2845
d9f4b93e81c5
Fix for memleak in dv.c patch by (Burkhard Plaum; plaum, ipf uni-stuttgart de)
michael
parents:
2822
diff
changeset
|
123 dv_vlc_map = av_mallocz_static(DV_VLC_MAP_LEV_SIZE*DV_VLC_MAP_RUN_SIZE*sizeof(struct dv_vlc_pair)); |
2979 | 124 if (!dv_vlc_map) |
125 return -ENOMEM; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
126 |
2979 | 127 /* dv_anchor lets each thread know its Id */ |
3167 | 128 for (i=0; i<DV_ANCHOR_SIZE; i++) |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
129 dv_anchor[i] = (void*)(size_t)i; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
130 |
2979 | 131 /* it's faster to include sign bit in a generic VLC parsing scheme */ |
132 for (i=0, j=0; i<NB_DV_VLC; i++, j++) { | |
133 new_dv_vlc_bits[j] = dv_vlc_bits[i]; | |
134 new_dv_vlc_len[j] = dv_vlc_len[i]; | |
135 new_dv_vlc_run[j] = dv_vlc_run[i]; | |
136 new_dv_vlc_level[j] = dv_vlc_level[i]; | |
2967 | 137 |
2979 | 138 if (dv_vlc_level[i]) { |
139 new_dv_vlc_bits[j] <<= 1; | |
140 new_dv_vlc_len[j]++; | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
141 |
2979 | 142 j++; |
143 new_dv_vlc_bits[j] = (dv_vlc_bits[i] << 1) | 1; | |
144 new_dv_vlc_len[j] = dv_vlc_len[i] + 1; | |
145 new_dv_vlc_run[j] = dv_vlc_run[i]; | |
146 new_dv_vlc_level[j] = -dv_vlc_level[i]; | |
147 } | |
148 } | |
2967 | 149 |
723 | 150 /* NOTE: as a trick, we use the fact the no codes are unused |
151 to accelerate the parsing of partial codes */ | |
2967 | 152 init_vlc(&dv_vlc, TEX_VLC_BITS, j, |
2370
26560d4fdb1f
Memory leak fix patch by (Burkhard Plaum <plaum >at< ipf.uni-stuttgart )dot( de>)
michael
parents:
2247
diff
changeset
|
153 new_dv_vlc_len, 1, 1, new_dv_vlc_bits, 2, 2, 0); |
723 | 154 |
2991
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
155 dv_rl_vlc = av_mallocz_static(dv_vlc.table_size * sizeof(RL_VLC_ELEM)); |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
156 if (!dv_rl_vlc) |
2979 | 157 return -ENOMEM; |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
158 |
723 | 159 for(i = 0; i < dv_vlc.table_size; i++){ |
160 int code= dv_vlc.table[i][0]; | |
161 int len = dv_vlc.table[i][1]; | |
162 int level, run; | |
2967 | 163 |
723 | 164 if(len<0){ //more bits needed |
165 run= 0; | |
166 level= code; | |
167 } else { | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
168 run= new_dv_vlc_run[code] + 1; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
169 level= new_dv_vlc_level[code]; |
723 | 170 } |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
171 dv_rl_vlc[i].len = len; |
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
172 dv_rl_vlc[i].level = level; |
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
173 dv_rl_vlc[i].run = run; |
723 | 174 } |
2979 | 175 free_vlc(&dv_vlc); |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
176 |
2979 | 177 for (i = 0; i < NB_DV_VLC - 1; i++) { |
2027
3a78447c3b53
oops, forgot to commit that change from the warning fixes by (Michael Roitzsch <mroi at users dot sourceforge dot net>)
michael
parents:
1994
diff
changeset
|
178 if (dv_vlc_run[i] >= DV_VLC_MAP_RUN_SIZE) |
2979 | 179 continue; |
2027
3a78447c3b53
oops, forgot to commit that change from the warning fixes by (Michael Roitzsch <mroi at users dot sourceforge dot net>)
michael
parents:
1994
diff
changeset
|
180 #ifdef DV_CODEC_TINY_TARGET |
3a78447c3b53
oops, forgot to commit that change from the warning fixes by (Michael Roitzsch <mroi at users dot sourceforge dot net>)
michael
parents:
1994
diff
changeset
|
181 if (dv_vlc_level[i] >= DV_VLC_MAP_LEV_SIZE) |
2979 | 182 continue; |
2027
3a78447c3b53
oops, forgot to commit that change from the warning fixes by (Michael Roitzsch <mroi at users dot sourceforge dot net>)
michael
parents:
1994
diff
changeset
|
183 #endif |
2967 | 184 |
2979 | 185 if (dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].size != 0) |
186 continue; | |
2967 | 187 |
2979 | 188 dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].vlc = dv_vlc_bits[i] << |
189 (!!dv_vlc_level[i]); | |
190 dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].size = dv_vlc_len[i] + | |
191 (!!dv_vlc_level[i]); | |
192 } | |
193 for (i = 0; i < DV_VLC_MAP_RUN_SIZE; i++) { | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
194 #ifdef DV_CODEC_TINY_TARGET |
2979 | 195 for (j = 1; j < DV_VLC_MAP_LEV_SIZE; j++) { |
196 if (dv_vlc_map[i][j].size == 0) { | |
197 dv_vlc_map[i][j].vlc = dv_vlc_map[0][j].vlc | | |
198 (dv_vlc_map[i-1][0].vlc << (dv_vlc_map[0][j].size)); | |
199 dv_vlc_map[i][j].size = dv_vlc_map[i-1][0].size + | |
200 dv_vlc_map[0][j].size; | |
201 } | |
202 } | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
203 #else |
2979 | 204 for (j = 1; j < DV_VLC_MAP_LEV_SIZE/2; j++) { |
205 if (dv_vlc_map[i][j].size == 0) { | |
206 dv_vlc_map[i][j].vlc = dv_vlc_map[0][j].vlc | | |
207 (dv_vlc_map[i-1][0].vlc << (dv_vlc_map[0][j].size)); | |
208 dv_vlc_map[i][j].size = dv_vlc_map[i-1][0].size + | |
209 dv_vlc_map[0][j].size; | |
210 } | |
211 dv_vlc_map[i][((uint16_t)(-j))&0x1ff].vlc = | |
212 dv_vlc_map[i][j].vlc | 1; | |
213 dv_vlc_map[i][((uint16_t)(-j))&0x1ff].size = | |
214 dv_vlc_map[i][j].size; | |
215 } | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
216 #endif |
2979 | 217 } |
723 | 218 } |
725 | 219 |
1567 | 220 /* Generic DSP setup */ |
221 dsputil_init(&dsp, avctx); | |
222 s->get_pixels = dsp.get_pixels; | |
725 | 223 |
1567 | 224 /* 88DCT setup */ |
225 s->fdct[0] = dsp.fdct; | |
226 s->idct_put[0] = dsp.idct_put; | |
227 for (i=0; i<64; i++) | |
228 s->dv_zigzag[0][i] = dsp.idct_permutation[ff_zigzag_direct[i]]; | |
725 | 229 |
1567 | 230 /* 248DCT setup */ |
231 s->fdct[1] = dsp.fdct248; | |
232 s->idct_put[1] = simple_idct248_put; // FIXME: need to add it to DSP | |
2849 | 233 if(avctx->lowres){ |
234 for (i=0; i<64; i++){ | |
235 int j= ff_zigzag248_direct[i]; | |
236 s->dv_zigzag[1][i] = dsp.idct_permutation[(j&7) + (j&8)*4 + (j&48)/2]; | |
237 } | |
238 }else | |
239 memcpy(s->dv_zigzag[1], ff_zigzag248_direct, 64); | |
725 | 240 |
723 | 241 /* XXX: do it only for constant case */ |
1567 | 242 dv_build_unquantize_tables(s, dsp.idct_permutation); |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
243 |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
244 /* FIXME: I really don't think this should be here */ |
3167 | 245 s->sys = dv_codec_profile(avctx); |
246 if(!s->sys) { | |
247 av_log(avctx, AV_LOG_ERROR, "Cannot determine type of DV output stream\n"); | |
248 return -EINVAL; | |
249 } | |
250 avctx->pix_fmt = s->sys->pix_fmt; | |
251 avctx->bit_rate = av_rescale(s->sys->frame_size * 8, | |
252 s->sys->frame_rate, | |
253 s->sys->frame_rate_base); | |
1543
7542cb99b950
* providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents:
1540
diff
changeset
|
254 avctx->coded_frame = &s->picture; |
2847 | 255 s->avctx= avctx; |
2967 | 256 |
723 | 257 return 0; |
258 } | |
259 | |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
260 // #define VLC_DEBUG |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
261 // #define printf(...) av_log(NULL, AV_LOG_ERROR, __VA_ARGS__) |
723 | 262 |
725 | 263 typedef struct BlockInfo { |
1064 | 264 const uint8_t *shift_table; |
265 const uint8_t *scan_table; | |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
266 const int *iweight_table; |
1064 | 267 uint8_t pos; /* position in block */ |
268 uint8_t dct_mode; | |
269 uint8_t partial_bit_count; | |
270 uint16_t partial_bit_buffer; | |
725 | 271 int shift_offset; |
272 } BlockInfo; | |
723 | 273 |
274 /* block size in bits */ | |
1064 | 275 static const uint16_t block_sizes[6] = { |
723 | 276 112, 112, 112, 112, 80, 80 |
277 }; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
278 /* bit budget for AC only in 5 MBs */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
279 static const int vs_total_ac_bits = (100 * 4 + 68*2) * 5; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
280 /* see dv_88_areas and dv_248_areas for details */ |
2967 | 281 static const int mb_area_start[5] = { 1, 6, 21, 43, 64 }; |
723 | 282 |
283 #ifndef ALT_BITSTREAM_READER | |
1256
b8e1c17b8d7d
some libmpeg2 style bitstream reader fixes (no dv doesnt yet work with it)
michaelni
parents:
1228
diff
changeset
|
284 #warning only works with ALT_BITSTREAM_READER |
2614
5e24800ab329
various fixes related to the non alt_bitstream_reader
michael
parents:
2453
diff
changeset
|
285 static int re_index; //Hack to make it compile |
723 | 286 #endif |
287 | |
1895
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
288 static inline int get_bits_left(GetBitContext *s) |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
289 { |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
290 return s->size_in_bits - get_bits_count(s); |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
291 } |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
292 |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
293 static inline int get_bits_size(GetBitContext *s) |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
294 { |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
295 return s->size_in_bits; |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
296 } |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
297 |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
298 static inline int put_bits_left(PutBitContext* s) |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
299 { |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
300 return (s->buf_end - s->buf) * 8 - put_bits_count(s); |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
301 } |
e5687117cc7f
* removing casualties of battle of the wits and English language
romansh
parents:
1887
diff
changeset
|
302 |
723 | 303 /* decode ac coefs */ |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
304 static void dv_decode_ac(GetBitContext *gb, BlockInfo *mb, DCTELEM *block) |
723 | 305 { |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
306 int last_index = get_bits_size(gb); |
1064 | 307 const uint8_t *scan_table = mb->scan_table; |
308 const uint8_t *shift_table = mb->shift_table; | |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
309 const int *iweight_table = mb->iweight_table; |
725 | 310 int pos = mb->pos; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
311 int partial_bit_count = mb->partial_bit_count; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
312 int level, pos1, run, vlc_len, index; |
2967 | 313 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
314 OPEN_READER(re, gb); |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
315 UPDATE_CACHE(re, gb); |
2967 | 316 |
723 | 317 /* if we must parse a partial vlc, we do it here */ |
318 if (partial_bit_count > 0) { | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
319 re_cache = ((unsigned)re_cache >> partial_bit_count) | |
2979 | 320 (mb->partial_bit_buffer << (sizeof(re_cache)*8 - partial_bit_count)); |
321 re_index -= partial_bit_count; | |
322 mb->partial_bit_count = 0; | |
723 | 323 } |
324 | |
325 /* get the AC coefficients until last_index is reached */ | |
326 for(;;) { | |
327 #ifdef VLC_DEBUG | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
328 printf("%2d: bits=%04x index=%d\n", pos, SHOW_UBITS(re, gb, 16), re_index); |
723 | 329 #endif |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
330 /* our own optimized GET_RL_VLC */ |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
331 index = NEG_USR32(re_cache, TEX_VLC_BITS); |
2979 | 332 vlc_len = dv_rl_vlc[index].len; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
333 if (vlc_len < 0) { |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
334 index = NEG_USR32((unsigned)re_cache << TEX_VLC_BITS, -vlc_len) + dv_rl_vlc[index].level; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
335 vlc_len = TEX_VLC_BITS - vlc_len; |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
336 } |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
337 level = dv_rl_vlc[index].level; |
2979 | 338 run = dv_rl_vlc[index].run; |
2967 | 339 |
2979 | 340 /* gotta check if we're still within gb boundaries */ |
341 if (re_index + vlc_len > last_index) { | |
342 /* should be < 16 bits otherwise a codeword could have been parsed */ | |
343 mb->partial_bit_count = last_index - re_index; | |
344 mb->partial_bit_buffer = NEG_USR32(re_cache, mb->partial_bit_count); | |
345 re_index = last_index; | |
346 break; | |
347 } | |
348 re_index += vlc_len; | |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
349 |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
350 #ifdef VLC_DEBUG |
2979 | 351 printf("run=%d level=%d\n", run, level); |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
352 #endif |
2979 | 353 pos += run; |
354 if (pos >= 64) | |
355 break; | |
2967 | 356 |
2847 | 357 pos1 = scan_table[pos]; |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
358 level <<= shift_table[pos1]; |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
359 |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
360 /* unweigh, round, and shift down */ |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
361 level = (level*iweight_table[pos] + (1 << (dv_iweight_bits-1))) >> dv_iweight_bits; |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
362 |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
363 block[pos1] = level; |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
364 |
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
365 UPDATE_CACHE(re, gb); |
723 | 366 } |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
367 CLOSE_READER(re, gb); |
725 | 368 mb->pos = pos; |
723 | 369 } |
370 | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
371 static inline void bit_copy(PutBitContext *pb, GetBitContext *gb) |
723 | 372 { |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
373 int bits_left = get_bits_left(gb); |
2847 | 374 while (bits_left >= MIN_CACHE_BITS) { |
375 put_bits(pb, MIN_CACHE_BITS, get_bits(gb, MIN_CACHE_BITS)); | |
376 bits_left -= MIN_CACHE_BITS; | |
723 | 377 } |
378 if (bits_left > 0) { | |
379 put_bits(pb, bits_left, get_bits(gb, bits_left)); | |
380 } | |
381 } | |
382 | |
383 /* mb_x and mb_y are in units of 8 pixels */ | |
2967 | 384 static inline void dv_decode_video_segment(DVVideoContext *s, |
385 uint8_t *buf_ptr1, | |
1064 | 386 const uint16_t *mb_pos_ptr) |
723 | 387 { |
388 int quant, dc, dct_mode, class1, j; | |
389 int mb_index, mb_x, mb_y, v, last_index; | |
390 DCTELEM *block, *block1; | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
391 int c_offset; |
1064 | 392 uint8_t *y_ptr; |
393 void (*idct_put)(uint8_t *dest, int line_size, DCTELEM *block); | |
394 uint8_t *buf_ptr; | |
723 | 395 PutBitContext pb, vs_pb; |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
396 GetBitContext gb; |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
397 BlockInfo mb_data[5 * 6], *mb, *mb1; |
3089 | 398 DECLARE_ALIGNED_8(DCTELEM, sblock[5*6][64]); |
399 DECLARE_ALIGNED_8(uint8_t, mb_bit_buffer[80 + 4]); /* allow some slack */ | |
400 DECLARE_ALIGNED_8(uint8_t, vs_bit_buffer[5 * 80 + 4]); /* allow some slack */ | |
2849 | 401 const int log2_blocksize= 3-s->avctx->lowres; |
2967 | 402 |
2847 | 403 assert((((int)mb_bit_buffer)&7)==0); |
404 assert((((int)vs_bit_buffer)&7)==0); | |
2967 | 405 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
406 memset(sblock, 0, sizeof(sblock)); |
723 | 407 |
408 /* pass 1 : read DC and AC coefficients in blocks */ | |
409 buf_ptr = buf_ptr1; | |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
410 block1 = &sblock[0][0]; |
725 | 411 mb1 = mb_data; |
1522
79dddc5cd990
removed the obsolete and unused parameters of init_put_bits
alex
parents:
1507
diff
changeset
|
412 init_put_bits(&vs_pb, vs_bit_buffer, 5 * 80); |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
413 for(mb_index = 0; mb_index < 5; mb_index++, mb1 += 6, block1 += 6 * 64) { |
723 | 414 /* skip header */ |
415 quant = buf_ptr[3] & 0x0f; | |
416 buf_ptr += 4; | |
1522
79dddc5cd990
removed the obsolete and unused parameters of init_put_bits
alex
parents:
1507
diff
changeset
|
417 init_put_bits(&pb, mb_bit_buffer, 80); |
725 | 418 mb = mb1; |
723 | 419 block = block1; |
420 for(j = 0;j < 6; j++) { | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
421 last_index = block_sizes[j]; |
2979 | 422 init_get_bits(&gb, buf_ptr, last_index); |
2967 | 423 |
723 | 424 /* get the dc */ |
2247 | 425 dc = get_sbits(&gb, 9); |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
426 dct_mode = get_bits1(&gb); |
725 | 427 mb->dct_mode = dct_mode; |
428 mb->scan_table = s->dv_zigzag[dct_mode]; | |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
429 mb->iweight_table = dct_mode ? dv_iweight_248 : dv_iweight_88; |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
430 class1 = get_bits(&gb, 2); |
1905
2761950695cc
* some significant clean-up of the dv_decode_ac (it looks real
romansh
parents:
1904
diff
changeset
|
431 mb->shift_table = s->dv_idct_shift[class1 == 3][dct_mode] |
723 | 432 [quant + dv_quant_offset[class1]]; |
433 dc = dc << 2; | |
434 /* convert to unsigned because 128 is not added in the | |
435 standard IDCT */ | |
436 dc += 1024; | |
437 block[0] = dc; | |
438 buf_ptr += last_index >> 3; | |
725 | 439 mb->pos = 0; |
440 mb->partial_bit_count = 0; | |
723 | 441 |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
442 #ifdef VLC_DEBUG |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
443 printf("MB block: %d, %d ", mb_index, j); |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
444 #endif |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
445 dv_decode_ac(&gb, mb, block); |
723 | 446 |
447 /* write the remaining bits in a new buffer only if the | |
448 block is finished */ | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
449 if (mb->pos >= 64) |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
450 bit_copy(&pb, &gb); |
2967 | 451 |
723 | 452 block += 64; |
725 | 453 mb++; |
723 | 454 } |
2967 | 455 |
723 | 456 /* pass 2 : we can do it just after */ |
457 #ifdef VLC_DEBUG | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
458 printf("***pass 2 size=%d MB#=%d\n", put_bits_count(&pb), mb_index); |
723 | 459 #endif |
460 block = block1; | |
725 | 461 mb = mb1; |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
462 init_get_bits(&gb, mb_bit_buffer, put_bits_count(&pb)); |
2979 | 463 flush_put_bits(&pb); |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
464 for(j = 0;j < 6; j++, block += 64, mb++) { |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
465 if (mb->pos < 64 && get_bits_left(&gb) > 0) { |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
466 dv_decode_ac(&gb, mb, block); |
723 | 467 /* if still not finished, no need to parse other blocks */ |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
468 if (mb->pos < 64) |
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
469 break; |
723 | 470 } |
471 } | |
472 /* all blocks are finished, so the extra bytes can be used at | |
473 the video segment level */ | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
474 if (j >= 6) |
2979 | 475 bit_copy(&vs_pb, &gb); |
723 | 476 } |
477 | |
478 /* we need a pass other the whole video segment */ | |
479 #ifdef VLC_DEBUG | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
480 printf("***pass 3 size=%d\n", put_bits_count(&vs_pb)); |
723 | 481 #endif |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
482 block = &sblock[0][0]; |
723 | 483 mb = mb_data; |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
484 init_get_bits(&gb, vs_bit_buffer, put_bits_count(&vs_pb)); |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
485 flush_put_bits(&vs_pb); |
723 | 486 for(mb_index = 0; mb_index < 5; mb_index++) { |
487 for(j = 0;j < 6; j++) { | |
1886
fbcf02596520
* DV decoder simplifications. Now it looks to be 6% faster. At least
romansh
parents:
1875
diff
changeset
|
488 if (mb->pos < 64) { |
723 | 489 #ifdef VLC_DEBUG |
490 printf("start %d:%d\n", mb_index, j); | |
491 #endif | |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
492 dv_decode_ac(&gb, mb, block); |
723 | 493 } |
2979 | 494 if (mb->pos >= 64 && mb->pos < 127) |
495 av_log(NULL, AV_LOG_ERROR, "AC EOB marker is absent pos=%d\n", mb->pos); | |
723 | 496 block += 64; |
725 | 497 mb++; |
723 | 498 } |
499 } | |
2967 | 500 |
723 | 501 /* compute idct and place blocks */ |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
502 block = &sblock[0][0]; |
723 | 503 mb = mb_data; |
504 for(mb_index = 0; mb_index < 5; mb_index++) { | |
505 v = *mb_pos_ptr++; | |
506 mb_x = v & 0xff; | |
507 mb_y = v >> 8; | |
3167 | 508 if (s->sys->pix_fmt == PIX_FMT_YUV422P) { |
509 y_ptr = s->picture.data[0] + ((mb_y * s->picture.linesize[0] + (mb_x>>1))<<log2_blocksize); | |
2849 | 510 c_offset = ((mb_y * s->picture.linesize[1] + (mb_x >> 2))<<log2_blocksize); |
3167 | 511 } else { /* 4:1:1 or 4:2:0 */ |
512 y_ptr = s->picture.data[0] + ((mb_y * s->picture.linesize[0] + mb_x)<<log2_blocksize); | |
513 if (s->sys->pix_fmt == PIX_FMT_YUV411P) | |
514 c_offset = ((mb_y * s->picture.linesize[1] + (mb_x >> 2))<<log2_blocksize); | |
515 else /* 4:2:0 */ | |
516 c_offset = (((mb_y >> 1) * s->picture.linesize[1] + (mb_x >> 1))<<log2_blocksize); | |
517 } | |
723 | 518 for(j = 0;j < 6; j++) { |
2849 | 519 idct_put = s->idct_put[mb->dct_mode && log2_blocksize==3]; |
3167 | 520 if (s->sys->pix_fmt == PIX_FMT_YUV422P) { /* 4:2:2 */ |
521 if (j == 0 || j == 2) { | |
522 /* Y0 Y1 */ | |
523 idct_put(y_ptr + ((j >> 1)<<log2_blocksize), | |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
524 s->picture.linesize[0], block); |
3167 | 525 } else if(j > 3) { |
526 /* Cr Cb */ | |
527 idct_put(s->picture.data[6 - j] + c_offset, | |
528 s->picture.linesize[6 - j], block); | |
723 | 529 } |
3167 | 530 /* note: j=1 and j=3 are "dummy" blocks in 4:2:2 */ |
531 } else { /* 4:1:1 or 4:2:0 */ | |
532 if (j < 4) { | |
533 if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x < (704 / 8)) { | |
534 /* NOTE: at end of line, the macroblock is handled as 420 */ | |
535 idct_put(y_ptr + (j<<log2_blocksize), s->picture.linesize[0], block); | |
536 } else { | |
537 idct_put(y_ptr + (((j & 1) + (j >> 1) * s->picture.linesize[0])<<log2_blocksize), | |
538 s->picture.linesize[0], block); | |
737 | 539 } |
540 } else { | |
3167 | 541 if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x >= (704 / 8)) { |
542 uint64_t aligned_pixels[64/8]; | |
543 uint8_t *pixels= (uint8_t*)aligned_pixels; | |
544 uint8_t *c_ptr, *c_ptr1, *ptr, *ptr1; | |
545 int x, y, linesize; | |
546 /* NOTE: at end of line, the macroblock is handled as 420 */ | |
547 idct_put(pixels, 8, block); | |
548 linesize = s->picture.linesize[6 - j]; | |
549 c_ptr = s->picture.data[6 - j] + c_offset; | |
550 ptr = pixels; | |
551 for(y = 0;y < (1<<log2_blocksize); y++) { | |
552 ptr1= ptr + (1<<(log2_blocksize-1)); | |
553 c_ptr1 = c_ptr + (linesize<<log2_blocksize); | |
554 for(x=0; x < (1<<(log2_blocksize-1)); x++){ | |
555 c_ptr[x]= ptr[x]; c_ptr1[x]= ptr1[x]; | |
556 } | |
557 c_ptr += linesize; | |
558 ptr += 8; | |
559 } | |
560 } else { | |
561 /* don't ask me why they inverted Cb and Cr ! */ | |
562 idct_put(s->picture.data[6 - j] + c_offset, | |
563 s->picture.linesize[6 - j], block); | |
564 } | |
737 | 565 } |
723 | 566 } |
567 block += 64; | |
725 | 568 mb++; |
723 | 569 } |
570 } | |
571 } | |
572 | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
573 #ifdef DV_CODEC_TINY_TARGET |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
574 /* Converts run and level (where level != 0) pair into vlc, returning bit size */ |
2847 | 575 static always_inline int dv_rl2vlc(int run, int level, int sign, uint32_t* vlc) |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
576 { |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
577 int size; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
578 if (run < DV_VLC_MAP_RUN_SIZE && level < DV_VLC_MAP_LEV_SIZE) { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
579 *vlc = dv_vlc_map[run][level].vlc | sign; |
2979 | 580 size = dv_vlc_map[run][level].size; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
581 } |
2967 | 582 else { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
583 if (level < DV_VLC_MAP_LEV_SIZE) { |
2979 | 584 *vlc = dv_vlc_map[0][level].vlc | sign; |
585 size = dv_vlc_map[0][level].size; | |
586 } else { | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
587 *vlc = 0xfe00 | (level << 1) | sign; |
2979 | 588 size = 16; |
589 } | |
590 if (run) { | |
591 *vlc |= ((run < 16) ? dv_vlc_map[run-1][0].vlc : | |
592 (0x1f80 | (run - 1))) << size; | |
593 size += (run < 16) ? dv_vlc_map[run-1][0].size : 13; | |
594 } | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
595 } |
2967 | 596 |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
597 return size; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
598 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
599 |
2847 | 600 static always_inline int dv_rl2vlc_size(int run, int level) |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
601 { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
602 int size; |
2967 | 603 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
604 if (run < DV_VLC_MAP_RUN_SIZE && level < DV_VLC_MAP_LEV_SIZE) { |
2979 | 605 size = dv_vlc_map[run][level].size; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
606 } |
2967 | 607 else { |
2979 | 608 size = (level < DV_VLC_MAP_LEV_SIZE) ? dv_vlc_map[0][level].size : 16; |
609 if (run) { | |
610 size += (run < 16) ? dv_vlc_map[run-1][0].size : 13; | |
611 } | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
612 } |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
613 return size; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
614 } |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
615 #else |
2847 | 616 static always_inline int dv_rl2vlc(int run, int l, int sign, uint32_t* vlc) |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
617 { |
2847 | 618 *vlc = dv_vlc_map[run][l].vlc | sign; |
619 return dv_vlc_map[run][l].size; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
620 } |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
621 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
622 static always_inline int dv_rl2vlc_size(int run, int l) |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
623 { |
2847 | 624 return dv_vlc_map[run][l].size; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
625 } |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
626 #endif |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
627 |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
628 typedef struct EncBlockInfo { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
629 int area_q[4]; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
630 int bit_size[4]; |
2847 | 631 int prev[5]; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
632 int cur_ac; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
633 int cno; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
634 int dct_mode; |
2847 | 635 DCTELEM mb[64]; |
636 uint8_t next[64]; | |
637 uint8_t sign[64]; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
638 uint8_t partial_bit_count; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
639 uint32_t partial_bit_buffer; /* we can't use uint16_t here */ |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
640 } EncBlockInfo; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
641 |
2967 | 642 static always_inline PutBitContext* dv_encode_ac(EncBlockInfo* bi, PutBitContext* pb_pool, |
2847 | 643 PutBitContext* pb_end) |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
644 { |
2847 | 645 int prev; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
646 int bits_left; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
647 PutBitContext* pb = pb_pool; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
648 int size = bi->partial_bit_count; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
649 uint32_t vlc = bi->partial_bit_buffer; |
2847 | 650 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
651 bi->partial_bit_count = bi->partial_bit_buffer = 0; |
2847 | 652 for(;;){ |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
653 /* Find suitable storage space */ |
1875
45a1592dadca
* moving some of the commonly used bit reading/writing functions
romansh
parents:
1726
diff
changeset
|
654 for (; size > (bits_left = put_bits_left(pb)); pb++) { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
655 if (bits_left) { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
656 size -= bits_left; |
2979 | 657 put_bits(pb, bits_left, vlc >> size); |
658 vlc = vlc & ((1<<size)-1); | |
659 } | |
660 if (pb + 1 >= pb_end) { | |
661 bi->partial_bit_count = size; | |
662 bi->partial_bit_buffer = vlc; | |
663 return pb; | |
664 } | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
665 } |
2967 | 666 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
667 /* Store VLC */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
668 put_bits(pb, size, vlc); |
2967 | 669 |
2847 | 670 if(bi->cur_ac>=64) |
671 break; | |
2967 | 672 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
673 /* Construct the next VLC */ |
2847 | 674 prev= bi->cur_ac; |
675 bi->cur_ac = bi->next[prev]; | |
676 if(bi->cur_ac < 64){ | |
677 size = dv_rl2vlc(bi->cur_ac - prev - 1, bi->mb[bi->cur_ac], bi->sign[bi->cur_ac], &vlc); | |
678 } else { | |
679 size = 4; vlc = 6; /* End Of Block stamp */ | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
680 } |
2847 | 681 } |
682 return pb; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
683 } |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
684 |
2967 | 685 static always_inline void dv_set_class_number(DCTELEM* blk, EncBlockInfo* bi, |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
686 const uint8_t* zigzag_scan, const int *weight, int bias) |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
687 { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
688 int i, area; |
3152
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
689 /* We offer two different methods for class number assignment: the |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
690 method suggested in SMPTE 314M Table 22, and an improved |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
691 method. The SMPTE method is very conservative; it assigns class |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
692 3 (i.e. severe quantization) to any block where the largest AC |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
693 component is greater than 36. ffmpeg's DV encoder tracks AC bit |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
694 consumption precisely, so there is no need to bias most blocks |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
695 towards strongly lossy compression. Instead, we assign class 2 |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
696 to most blocks, and use class 3 only when strictly necessary |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
697 (for blocks whose largest AC component exceeds 255). */ |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
698 |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
699 #if 0 /* SMPTE spec method */ |
2847 | 700 static const int classes[] = {12, 24, 36, 0xffff}; |
3152
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
701 #else /* improved ffmpeg method */ |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
702 static const int classes[] = {-1, -1, 255, 0xffff}; |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
703 #endif |
28a087ad41d1
Assign class 2 to most macroblocks by default, instead of a more conservative
romansh
parents:
3150
diff
changeset
|
704 int max=classes[0]; |
2847 | 705 int prev=0; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
706 |
2967 | 707 bi->mb[0] = blk[0]; |
708 | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
709 for (area = 0; area < 4; area++) { |
2847 | 710 bi->prev[area] = prev; |
711 bi->bit_size[area] = 1; // 4 areas 4 bits for EOB :) | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
712 for (i=mb_area_start[area]; i<mb_area_start[area+1]; i++) { |
2847 | 713 int level = blk[zigzag_scan[i]]; |
2967 | 714 |
2847 | 715 if (level+15 > 30U) { |
716 bi->sign[i] = (level>>31)&1; | |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
717 /* weigh it and and shift down into range, adding for rounding */ |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
718 /* the extra division by a factor of 2^4 reverses the 8x expansion of the DCT |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
719 AND the 2x doubling of the weights */ |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
720 level = (ABS(level) * weight[i] + (1<<(dv_weight_bits+3))) >> (dv_weight_bits+4); |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
721 bi->mb[i] = level; |
2847 | 722 if(level>max) max= level; |
723 bi->bit_size[area] += dv_rl2vlc_size(i - prev - 1, level); | |
724 bi->next[prev]= i; | |
725 prev= i; | |
726 } | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
727 } |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
728 } |
2847 | 729 bi->next[prev]= i; |
730 for(bi->cno = 0; max > classes[bi->cno]; bi->cno++); | |
731 | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
732 bi->cno += bias; |
2967 | 733 |
2847 | 734 if (bi->cno >= 3) { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
735 bi->cno = 3; |
2847 | 736 prev=0; |
737 i= bi->next[prev]; | |
738 for (area = 0; area < 4; area++) { | |
739 bi->prev[area] = prev; | |
740 bi->bit_size[area] = 1; // 4 areas 4 bits for EOB :) | |
741 for (; i<mb_area_start[area+1]; i= bi->next[i]) { | |
742 bi->mb[i] >>=1; | |
2967 | 743 |
2847 | 744 if (bi->mb[i]) { |
745 bi->bit_size[area] += dv_rl2vlc_size(i - prev - 1, bi->mb[i]); | |
746 bi->next[prev]= i; | |
747 prev= i; | |
748 } | |
749 } | |
750 } | |
751 bi->next[prev]= i; | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
752 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
753 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
754 |
2847 | 755 //FIXME replace this by dsputil |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
756 #define SC(x, y) ((s[x] - s[y]) ^ ((s[x] - s[y]) >> 7)) |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
757 static always_inline int dv_guess_dct_mode(DCTELEM *blk) { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
758 DCTELEM *s; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
759 int score88 = 0; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
760 int score248 = 0; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
761 int i; |
2967 | 762 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
763 /* Compute 8-8 score (small values give a better chance for 8-8 DCT) */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
764 s = blk; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
765 for(i=0; i<7; i++) { |
2967 | 766 score88 += SC(0, 8) + SC(1, 9) + SC(2, 10) + SC(3, 11) + |
2979 | 767 SC(4, 12) + SC(5,13) + SC(6, 14) + SC(7, 15); |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
768 s += 8; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
769 } |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
770 /* Compute 2-4-8 score (small values give a better chance for 2-4-8 DCT) */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
771 s = blk; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
772 for(i=0; i<6; i++) { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
773 score248 += SC(0, 16) + SC(1,17) + SC(2, 18) + SC(3, 19) + |
2979 | 774 SC(4, 20) + SC(5,21) + SC(6, 22) + SC(7, 23); |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
775 s += 8; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
776 } |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
777 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
778 return (score88 - score248 > -10); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
779 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
780 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
781 static inline void dv_guess_qnos(EncBlockInfo* blks, int* qnos) |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
782 { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
783 int size[5]; |
3140 | 784 int i, j, k, a, prev, a2; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
785 EncBlockInfo* b; |
2847 | 786 |
3146
dcae1bde37ac
size[0-3] are not initialized (and can get random negative trash
michael
parents:
3143
diff
changeset
|
787 size[0] = size[1] = size[2] = size[3] = size[4] = 1<<24; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
788 do { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
789 b = blks; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
790 for (i=0; i<5; i++) { |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
791 if (!qnos[i]) |
2979 | 792 continue; |
2967 | 793 |
2979 | 794 qnos[i]--; |
795 size[i] = 0; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
796 for (j=0; j<6; j++, b++) { |
2979 | 797 for (a=0; a<4; a++) { |
798 if (b->area_q[a] != dv_quant_shifts[qnos[i] + dv_quant_offset[b->cno]][a]) { | |
799 b->bit_size[a] = 1; // 4 areas 4 bits for EOB :) | |
800 b->area_q[a]++; | |
2847 | 801 prev= b->prev[a]; |
3147 | 802 assert(b->next[prev] >= mb_area_start[a+1] || b->mb[prev]); |
2847 | 803 for (k= b->next[prev] ; k<mb_area_start[a+1]; k= b->next[k]) { |
2979 | 804 b->mb[k] >>= 1; |
805 if (b->mb[k]) { | |
2847 | 806 b->bit_size[a] += dv_rl2vlc_size(k - prev - 1, b->mb[k]); |
2979 | 807 prev= k; |
2847 | 808 } else { |
3140 | 809 if(b->next[k] >= mb_area_start[a+1] && b->next[k]<64){ |
3146
dcae1bde37ac
size[0-3] are not initialized (and can get random negative trash
michael
parents:
3143
diff
changeset
|
810 for(a2=a+1; b->next[k] >= mb_area_start[a2+1]; a2++) |
dcae1bde37ac
size[0-3] are not initialized (and can get random negative trash
michael
parents:
3143
diff
changeset
|
811 b->prev[a2] = prev; |
3140 | 812 assert(a2<4); |
813 assert(b->mb[b->next[k]]); | |
814 b->bit_size[a2] += dv_rl2vlc_size(b->next[k] - prev - 1, b->mb[b->next[k]]) | |
815 -dv_rl2vlc_size(b->next[k] - k - 1, b->mb[b->next[k]]); | |
3147 | 816 assert(b->prev[a2]==k && (a2+1 >= 4 || b->prev[a2+1]!=k)); |
817 b->prev[a2] = prev; | |
3140 | 818 } |
2847 | 819 b->next[prev] = b->next[k]; |
820 } | |
2979 | 821 } |
2847 | 822 b->prev[a+1]= prev; |
2979 | 823 } |
824 size[i] += b->bit_size[a]; | |
825 } | |
826 } | |
3142 | 827 if(vs_total_ac_bits >= size[0] + size[1] + size[2] + size[3] + size[4]) |
828 return; | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
829 } |
3140 | 830 } while (qnos[0]|qnos[1]|qnos[2]|qnos[3]|qnos[4]); |
831 | |
832 | |
833 for(a=2; a==2 || vs_total_ac_bits < size[0]; a+=a){ | |
834 b = blks; | |
3143 | 835 size[0] = 5*6*4; //EOB |
3140 | 836 for (j=0; j<6*5; j++, b++) { |
837 prev= b->prev[0]; | |
838 for (k= b->next[prev]; k<64; k= b->next[k]) { | |
839 if(b->mb[k] < a && b->mb[k] > -a){ | |
840 b->next[prev] = b->next[k]; | |
841 }else{ | |
842 size[0] += dv_rl2vlc_size(k - prev - 1, b->mb[k]); | |
843 prev= k; | |
844 } | |
845 } | |
846 } | |
847 } | |
1567 | 848 } |
849 | |
2967 | 850 static inline void dv_encode_video_segment(DVVideoContext *s, |
851 uint8_t *dif, | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
852 const uint16_t *mb_pos_ptr) |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
853 { |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
854 int mb_index, i, j, v; |
2967 | 855 int mb_x, mb_y, c_offset, linesize; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
856 uint8_t* y_ptr; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
857 uint8_t* data; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
858 uint8_t* ptr; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
859 int do_edge_wrap; |
3089 | 860 DECLARE_ALIGNED_8(DCTELEM, block[64]); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
861 EncBlockInfo enc_blks[5*6]; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
862 PutBitContext pbs[5*6]; |
2967 | 863 PutBitContext* pb; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
864 EncBlockInfo* enc_blk; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
865 int vs_bit_size = 0; |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
866 int qnos[5]; |
2967 | 867 |
2847 | 868 assert((((int)block) & 7) == 0); |
2967 | 869 |
1567 | 870 enc_blk = &enc_blks[0]; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
871 pb = &pbs[0]; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
872 for(mb_index = 0; mb_index < 5; mb_index++) { |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
873 v = *mb_pos_ptr++; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
874 mb_x = v & 0xff; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
875 mb_y = v >> 8; |
3167 | 876 if (s->sys->pix_fmt == PIX_FMT_YUV422P) { |
877 y_ptr = s->picture.data[0] + (mb_y * s->picture.linesize[0] * 8) + (mb_x * 4); | |
878 } else { /* 4:1:1 */ | |
879 y_ptr = s->picture.data[0] + (mb_y * s->picture.linesize[0] * 8) + (mb_x * 8); | |
880 } | |
881 if (s->sys->pix_fmt == PIX_FMT_YUV420P) { | |
882 c_offset = (((mb_y >> 1) * s->picture.linesize[1] * 8) + ((mb_x >> 1) * 8)); | |
883 } else { /* 4:2:2 or 4:1:1 */ | |
884 c_offset = ((mb_y * s->picture.linesize[1] * 8) + ((mb_x >> 2) * 8)); | |
885 } | |
2979 | 886 do_edge_wrap = 0; |
887 qnos[mb_index] = 15; /* No quantization */ | |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
888 ptr = dif + mb_index*80 + 4; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
889 for(j = 0;j < 6; j++) { |
3167 | 890 int dummy = 0; |
891 if (s->sys->pix_fmt == PIX_FMT_YUV422P) { /* 4:2:2 */ | |
892 if (j == 0 || j == 2) { | |
893 /* Y0 Y1 */ | |
894 data = y_ptr + ((j>>1) * 8); | |
895 linesize = s->picture.linesize[0]; | |
896 } else if (j > 3) { | |
897 /* Cr Cb */ | |
898 data = s->picture.data[6 - j] + c_offset; | |
899 linesize = s->picture.linesize[6 - j]; | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
900 } else { |
3167 | 901 /* j=1 and j=3 are "dummy" blocks, used for AC data only */ |
902 data = 0; | |
903 linesize = 0; | |
904 dummy = 1; | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
905 } |
3167 | 906 } else { /* 4:1:1 or 4:2:0 */ |
907 if (j < 4) { /* Four Y blocks */ | |
908 /* NOTE: at end of line, the macroblock is handled as 420 */ | |
909 if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x < (704 / 8)) { | |
910 data = y_ptr + (j * 8); | |
911 } else { | |
912 data = y_ptr + ((j & 1) * 8) + ((j >> 1) * 8 * s->picture.linesize[0]); | |
913 } | |
914 linesize = s->picture.linesize[0]; | |
915 } else { /* Cr and Cb blocks */ | |
916 /* don't ask Fabrice why they inverted Cb and Cr ! */ | |
917 data = s->picture.data[6 - j] + c_offset; | |
918 linesize = s->picture.linesize[6 - j]; | |
919 if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x >= (704 / 8)) | |
920 do_edge_wrap = 1; | |
921 } | |
2979 | 922 } |
2967 | 923 |
2979 | 924 /* Everything is set up -- now just copy data -> DCT block */ |
925 if (do_edge_wrap) { /* Edge wrap copy: 4x16 -> 8x8 */ | |
926 uint8_t* d; | |
927 DCTELEM *b = block; | |
928 for (i=0;i<8;i++) { | |
929 d = data + 8 * linesize; | |
930 b[0] = data[0]; b[1] = data[1]; b[2] = data[2]; b[3] = data[3]; | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
931 b[4] = d[0]; b[5] = d[1]; b[6] = d[2]; b[7] = d[3]; |
2979 | 932 data += linesize; |
933 b += 8; | |
934 } | |
935 } else { /* Simple copy: 8x8 -> 8x8 */ | |
3167 | 936 if (!dummy) |
937 s->get_pixels(block, data, linesize); | |
2979 | 938 } |
2967 | 939 |
2847 | 940 if(s->avctx->flags & CODEC_FLAG_INTERLACED_DCT) |
941 enc_blk->dct_mode = dv_guess_dct_mode(block); | |
942 else | |
943 enc_blk->dct_mode = 0; | |
2979 | 944 enc_blk->area_q[0] = enc_blk->area_q[1] = enc_blk->area_q[2] = enc_blk->area_q[3] = 0; |
945 enc_blk->partial_bit_count = 0; | |
946 enc_blk->partial_bit_buffer = 0; | |
947 enc_blk->cur_ac = 0; | |
2967 | 948 |
3167 | 949 if (dummy) { |
950 /* We rely on the fact that encoding all zeros leads to an immediate EOB, | |
951 which is precisely what the spec calls for in the "dummy" blocks. */ | |
952 memset(block, 0, sizeof(block)); | |
953 } else { | |
954 s->fdct[enc_blk->dct_mode](block); | |
955 } | |
2967 | 956 |
2979 | 957 dv_set_class_number(block, enc_blk, |
3149
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
958 enc_blk->dct_mode ? ff_zigzag248_direct : ff_zigzag_direct, |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
959 enc_blk->dct_mode ? dv_weight_248 : dv_weight_88, |
2679184e8be0
AC coefficient weighing (per SMPTE 314M) for the DV encoder and decoder. The
romansh
parents:
3148
diff
changeset
|
960 j/4); |
2967 | 961 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
962 init_put_bits(pb, ptr, block_sizes[j]/8); |
2979 | 963 put_bits(pb, 9, (uint16_t)(((enc_blk->mb[0] >> 3) - 1024 + 2) >> 2)); |
964 put_bits(pb, 1, enc_blk->dct_mode); | |
965 put_bits(pb, 2, enc_blk->cno); | |
2967 | 966 |
2979 | 967 vs_bit_size += enc_blk->bit_size[0] + enc_blk->bit_size[1] + |
968 enc_blk->bit_size[2] + enc_blk->bit_size[3]; | |
969 ++enc_blk; | |
970 ++pb; | |
971 ptr += block_sizes[j]/8; | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
972 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
973 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
974 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
975 if (vs_total_ac_bits < vs_bit_size) |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
976 dv_guess_qnos(&enc_blks[0], &qnos[0]); |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
977 |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
978 for (i=0; i<5; i++) { |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
979 dif[i*80 + 3] = qnos[i]; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
980 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
981 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
982 /* First pass over individual cells only */ |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
983 for (j=0; j<5*6; j++) |
2847 | 984 dv_encode_ac(&enc_blks[j], &pbs[j], &pbs[j+1]); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
985 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
986 /* Second pass over each MB space */ |
2847 | 987 for (j=0; j<5*6; j+=6) { |
988 pb= &pbs[j]; | |
989 for (i=0; i<6; i++) { | |
990 if (enc_blks[i+j].partial_bit_count) | |
991 pb=dv_encode_ac(&enc_blks[i+j], pb, &pbs[j+6]); | |
992 } | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
993 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
994 |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
995 /* Third and final pass over the whole vides segment space */ |
2847 | 996 pb= &pbs[0]; |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
997 for (j=0; j<5*6; j++) { |
2847 | 998 if (enc_blks[j].partial_bit_count) |
999 pb=dv_encode_ac(&enc_blks[j], pb, &pbs[6*5]); | |
3139
e58fb7ffbb4f
print a big warning if we mess up and run out of space ...
michael
parents:
3089
diff
changeset
|
1000 if (enc_blks[j].partial_bit_count) |
e58fb7ffbb4f
print a big warning if we mess up and run out of space ...
michael
parents:
3089
diff
changeset
|
1001 av_log(NULL, AV_LOG_ERROR, "ac bitstream overflow\n"); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1002 } |
1631
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
1003 |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
1004 for (j=0; j<5*6; j++) |
59f2fa833449
* 3x encoding speedup. Finally we seem to be on par with libdv
romansh
parents:
1598
diff
changeset
|
1005 flush_put_bits(&pbs[j]); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1006 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1007 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1008 static int dv_decode_mt(AVCodecContext *avctx, void* sl) |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1009 { |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1010 DVVideoContext *s = avctx->priv_data; |
1904 | 1011 int slice = (size_t)sl; |
3167 | 1012 |
1013 /* which DIF channel is this? */ | |
1014 int chan = slice / (s->sys->difseg_size * 27); | |
1015 | |
1016 /* slice within the DIF channel */ | |
1017 int chan_slice = slice % (s->sys->difseg_size * 27); | |
1018 | |
1019 /* byte offset of this channel's data */ | |
1020 int chan_offset = chan * s->sys->difseg_size * 150 * 80; | |
1021 | |
1022 dv_decode_video_segment(s, &s->buf[((chan_slice/27)*6+(chan_slice/3)+chan_slice*5+7)*80 + chan_offset], | |
2979 | 1023 &s->sys->video_place[slice*5]); |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1024 return 0; |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1025 } |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1026 |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1027 static int dv_encode_mt(AVCodecContext *avctx, void* sl) |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1028 { |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1029 DVVideoContext *s = avctx->priv_data; |
1904 | 1030 int slice = (size_t)sl; |
3167 | 1031 |
1032 /* which DIF channel is this? */ | |
1033 int chan = slice / (s->sys->difseg_size * 27); | |
1034 | |
1035 /* slice within the DIF channel */ | |
1036 int chan_slice = slice % (s->sys->difseg_size * 27); | |
1037 | |
1038 /* byte offset of this channel's data */ | |
1039 int chan_offset = chan * s->sys->difseg_size * 150 * 80; | |
1040 | |
1041 dv_encode_video_segment(s, &s->buf[((chan_slice/27)*6+(chan_slice/3)+chan_slice*5+7)*80 + chan_offset], | |
2979 | 1042 &s->sys->video_place[slice*5]); |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1043 return 0; |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1044 } |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1045 |
723 | 1046 /* NOTE: exactly one frame must be given (120000 bytes for NTSC, |
3167 | 1047 144000 bytes for PAL - or twice those for 50Mbps) */ |
2967 | 1048 static int dvvideo_decode_frame(AVCodecContext *avctx, |
723 | 1049 void *data, int *data_size, |
1064 | 1050 uint8_t *buf, int buf_size) |
723 | 1051 { |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1052 DVVideoContext *s = avctx->priv_data; |
2967 | 1053 |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1054 s->sys = dv_frame_profile(buf); |
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1055 if (!s->sys || buf_size < s->sys->frame_size) |
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1056 return -1; /* NOTE: we only accept several full frames */ |
723 | 1057 |
1228 | 1058 if(s->picture.data[0]) |
1059 avctx->release_buffer(avctx, &s->picture); | |
2967 | 1060 |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1061 s->picture.reference = 0; |
2822
fdedaa2e6da4
DV video encoder/decoder doesn't set keyframe and picture type properties patch by (Edward Hervey:bilboed,gmail com)
michael
parents:
2661
diff
changeset
|
1062 s->picture.key_frame = 1; |
fdedaa2e6da4
DV video encoder/decoder doesn't set keyframe and picture type properties patch by (Edward Hervey:bilboed,gmail com)
michael
parents:
2661
diff
changeset
|
1063 s->picture.pict_type = FF_I_TYPE; |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1064 avctx->pix_fmt = s->sys->pix_fmt; |
2849 | 1065 avcodec_set_dimensions(avctx, s->sys->width, s->sys->height); |
903 | 1066 if(avctx->get_buffer(avctx, &s->picture) < 0) { |
1598
932d306bf1dc
av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents:
1567
diff
changeset
|
1067 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); |
903 | 1068 return -1; |
835 | 1069 } |
1543
7542cb99b950
* providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents:
1540
diff
changeset
|
1070 s->picture.interlaced_frame = 1; |
1547 | 1071 s->picture.top_field_first = 0; |
835 | 1072 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1073 s->buf = buf; |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
1074 avctx->execute(avctx, dv_decode_mt, (void**)&dv_anchor[0], NULL, |
3167 | 1075 s->sys->n_difchan * s->sys->difseg_size * 27); |
2967 | 1076 |
734
2d6b3e3d6c6f
10l - MMX/FPU state was not restored, causing nonsense fpu behaviour in caller (mplayer)
arpi_esp
parents:
733
diff
changeset
|
1077 emms_c(); |
2d6b3e3d6c6f
10l - MMX/FPU state was not restored, causing nonsense fpu behaviour in caller (mplayer)
arpi_esp
parents:
733
diff
changeset
|
1078 |
723 | 1079 /* return image */ |
925 | 1080 *data_size = sizeof(AVFrame); |
1081 *(AVFrame*)data= s->picture; | |
2967 | 1082 |
1489
337d13aee605
* DV handling was streamlined for both muxing/demuxing and
romansh
parents:
1416
diff
changeset
|
1083 return s->sys->frame_size; |
723 | 1084 } |
1085 | |
2967 | 1086 static int dvvideo_encode_frame(AVCodecContext *c, uint8_t *buf, int buf_size, |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1087 void *data) |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1088 { |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1089 DVVideoContext *s = c->priv_data; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1090 |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1091 s->sys = dv_codec_profile(c); |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1092 if (!s->sys) |
2979 | 1093 return -1; |
2422 | 1094 if(buf_size < s->sys->frame_size) |
1095 return -1; | |
1096 | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1097 c->pix_fmt = s->sys->pix_fmt; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1098 s->picture = *((AVFrame *)data); |
2822
fdedaa2e6da4
DV video encoder/decoder doesn't set keyframe and picture type properties patch by (Edward Hervey:bilboed,gmail com)
michael
parents:
2661
diff
changeset
|
1099 s->picture.key_frame = 1; |
fdedaa2e6da4
DV video encoder/decoder doesn't set keyframe and picture type properties patch by (Edward Hervey:bilboed,gmail com)
michael
parents:
2661
diff
changeset
|
1100 s->picture.pict_type = FF_I_TYPE; |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1101 |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1102 s->buf = buf; |
3016
97e6c0f82182
Moving dv_anchor back to the global scope. This creates a tiny memory
romansh
parents:
2991
diff
changeset
|
1103 c->execute(c, dv_encode_mt, (void**)&dv_anchor[0], NULL, |
3167 | 1104 s->sys->n_difchan * s->sys->difseg_size * 27); |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1105 |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1106 emms_c(); |
3167 | 1107 |
1108 /* Fill in just enough of the header for dv_frame_profile() to | |
1109 return the correct result, so that the frame can be decoded | |
1110 correctly. The rest of the metadata is filled in by the dvvideo | |
1111 avformat. (this should probably change so that encode_frame() | |
1112 fills in ALL of the metadata - e.g. for Quicktime-wrapped DV | |
1113 streams) */ | |
1114 | |
1115 /* NTSC/PAL format */ | |
1116 buf[3] = s->sys->dsf ? 0x80 : 0x00; | |
1117 | |
1118 /* 25Mbps or 50Mbps */ | |
1119 buf[80*5 + 48 + 3] = (s->sys->pix_fmt == PIX_FMT_YUV422P) ? 0x4 : 0x0; | |
1120 | |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1121 return s->sys->frame_size; |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1122 } |
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1123 |
2991
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1124 static int dvvideo_close(AVCodecContext *c) |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1125 { |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1126 |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1127 return 0; |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1128 } |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1129 |
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1130 |
2661
b2846918585c
a few #ifdef CONFIG_X_ENCODER, patch by (Roine Gustafsson <roine users.sourceforge net]
michael
parents:
2614
diff
changeset
|
1131 #ifdef CONFIG_DVVIDEO_ENCODER |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1132 AVCodec dvvideo_encoder = { |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1133 "dvvideo", |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1134 CODEC_TYPE_VIDEO, |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1135 CODEC_ID_DVVIDEO, |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1136 sizeof(DVVideoContext), |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1137 dvvideo_init, |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1138 dvvideo_encode_frame, |
2991
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1139 dvvideo_close, |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1140 NULL, |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1141 CODEC_CAP_DR1, |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1142 NULL |
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1143 }; |
2661
b2846918585c
a few #ifdef CONFIG_X_ENCODER, patch by (Roine Gustafsson <roine users.sourceforge net]
michael
parents:
2614
diff
changeset
|
1144 #endif // CONFIG_DVVIDEO_ENCODER |
723 | 1145 |
1146 AVCodec dvvideo_decoder = { | |
1147 "dvvideo", | |
1148 CODEC_TYPE_VIDEO, | |
1149 CODEC_ID_DVVIDEO, | |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1150 sizeof(DVVideoContext), |
1493
ad7e62df9962
* preAlpha DV encoding support -- there's still a truckload
romansh
parents:
1489
diff
changeset
|
1151 dvvideo_init, |
1887
85fe2f4633ec
* DV decoding/encoding now supports MultiThreading for up to 324 CPUs ;-)
romansh
parents:
1886
diff
changeset
|
1152 NULL, |
2991
d377b0788381
Fix DV memory leak and allow multiple instances patch by (Burkhard Plaum?)
michael
parents:
2979
diff
changeset
|
1153 dvvideo_close, |
723 | 1154 dvvideo_decode_frame, |
835 | 1155 CODEC_CAP_DR1, |
723 | 1156 NULL |
1157 }; |