aic: Fix slice size computation for widths multiples of 32 macroblocks
[ffmpeg.git] / libavcodec / dv_profile.c
1 /*
2  * This file is part of Libav.
3  *
4  * Libav is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * Libav is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with Libav; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include "config.h"
20
21 #include <stdint.h>
22
23 #include "libavutil/common.h"
24 #include "libavutil/log.h"
25 #include "libavutil/pixdesc.h"
26
27 #include "avcodec.h"
28 #include "dv_profile.h"
29 #include "dv_profile_internal.h"
30
31 #if CONFIG_DVPROFILE
32
33 static const uint8_t dv_audio_shuffle525[10][9] = {
34     {  0, 30, 60, 20, 50, 80, 10, 40, 70 }, /* 1st channel */
35     {  6, 36, 66, 26, 56, 86, 16, 46, 76 },
36     { 12, 42, 72,  2, 32, 62, 22, 52, 82 },
37     { 18, 48, 78,  8, 38, 68, 28, 58, 88 },
38     { 24, 54, 84, 14, 44, 74,  4, 34, 64 },
39
40     {  1, 31, 61, 21, 51, 81, 11, 41, 71 }, /* 2nd channel */
41     {  7, 37, 67, 27, 57, 87, 17, 47, 77 },
42     { 13, 43, 73,  3, 33, 63, 23, 53, 83 },
43     { 19, 49, 79,  9, 39, 69, 29, 59, 89 },
44     { 25, 55, 85, 15, 45, 75,  5, 35, 65 },
45 };
46
47 static const uint8_t dv_audio_shuffle625[12][9] = {
48     {  0, 36,  72, 26, 62,  98, 16, 52,  88 }, /* 1st channel */
49     {  6, 42,  78, 32, 68, 104, 22, 58,  94 },
50     { 12, 48,  84,  2, 38,  74, 28, 64, 100 },
51     { 18, 54,  90,  8, 44,  80, 34, 70, 106 },
52     { 24, 60,  96, 14, 50,  86,  4, 40,  76 },
53     { 30, 66, 102, 20, 56,  92, 10, 46,  82 },
54
55     {  1, 37,  73, 27, 63,  99, 17, 53,  89 }, /* 2nd channel */
56     {  7, 43,  79, 33, 69, 105, 23, 59,  95 },
57     { 13, 49,  85,  3, 39,  75, 29, 65, 101 },
58     { 19, 55,  91,  9, 45,  81, 35, 71, 107 },
59     { 25, 61,  97, 15, 51,  87,  5, 41,  77 },
60     { 31, 67, 103, 21, 57,  93, 11, 47,  83 },
61 };
62
63 /* macroblock bit budgets */
64 static const uint8_t block_sizes_dv2550[8] = {
65     112, 112, 112, 112, 80, 80, 0, 0,
66 };
67
68 static const uint8_t block_sizes_dv100[8] = {
69     80, 80, 80, 80, 80, 80, 64, 64,
70 };
71
72 static const AVDVProfile dv_profiles[] = {
73     { .dsf                = 0,
74       .video_stype        = 0x0,
75       .frame_size         = 120000, /* IEC 61834, SMPTE-314M - 525/60 (NTSC) */
76       .difseg_size        = 10,
77       .n_difchan          = 1,
78       .time_base          = { 1001, 30000 },
79       .ltc_divisor        = 30,
80       .height             = 480,
81       .width              = 720,
82       .sar                = { { 8, 9 }, { 32, 27 } },
83       .pix_fmt            = AV_PIX_FMT_YUV411P,
84       .bpm                = 6,
85       .block_sizes        = block_sizes_dv2550,
86       .audio_stride       = 90,
87       .audio_min_samples  = { 1580, 1452, 1053 }, /* for 48, 44.1 and 32kHz */
88       .audio_samples_dist = { 1600, 1602, 1602, 1602, 1602 },  /* per SMPTE-314M */
89       .audio_shuffle      = dv_audio_shuffle525, },
90     { .dsf                = 1,
91       .video_stype        = 0x0,
92       .frame_size         = 144000, /* IEC 61834 - 625/50 (PAL) */
93       .difseg_size        = 12,
94       .n_difchan          = 1,
95       .time_base          = { 1, 25 },
96       .ltc_divisor        = 25,
97       .height             = 576,
98       .width              = 720,
99       .sar                = { { 16, 15 }, { 64, 45 } },
100       .pix_fmt            = AV_PIX_FMT_YUV420P,
101       .bpm                = 6,
102       .block_sizes        = block_sizes_dv2550,
103       .audio_stride       = 108,
104       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
105       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
106       .audio_shuffle      = dv_audio_shuffle625, },
107     { .dsf                = 1,
108       .video_stype        = 0x0,
109       .frame_size         = 144000, /* SMPTE-314M - 625/50 (PAL) */
110       .difseg_size        = 12,
111       .n_difchan          = 1,
112       .time_base          = { 1, 25 },
113       .ltc_divisor        = 25,
114       .height             = 576,
115       .width              = 720,
116       .sar                = { { 16, 15 }, { 64, 45 } },
117       .pix_fmt            = AV_PIX_FMT_YUV411P,
118       .bpm                = 6,
119       .block_sizes        = block_sizes_dv2550,
120       .audio_stride       = 108,
121       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
122       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
123       .audio_shuffle      = dv_audio_shuffle625, },
124     { .dsf                = 0,
125       .video_stype        = 0x4,
126       .frame_size         = 240000, /* SMPTE-314M - 525/60 (NTSC) 50 Mbps */
127       .difseg_size        = 10,    /* also known as "DVCPRO50" */
128       .n_difchan          = 2,
129       .time_base          = { 1001, 30000 },
130       .ltc_divisor        = 30,
131       .height             = 480,
132       .width              = 720,
133       .sar                = { { 8, 9 }, { 32, 27 } },
134       .pix_fmt            = AV_PIX_FMT_YUV422P,
135       .bpm                = 6,
136       .block_sizes        = block_sizes_dv2550,
137       .audio_stride       = 90,
138       .audio_min_samples  = { 1580, 1452, 1053 }, /* for 48, 44.1 and 32kHz */
139       .audio_samples_dist = { 1600, 1602, 1602, 1602, 1602 },  /* per SMPTE-314M */
140       .audio_shuffle      = dv_audio_shuffle525, },
141     { .dsf                = 1,
142       .video_stype        = 0x4,
143       .frame_size         = 288000, /* SMPTE-314M - 625/50 (PAL) 50 Mbps */
144       .difseg_size        = 12,    /* also known as "DVCPRO50" */
145       .n_difchan          = 2,
146       .time_base          = { 1, 25 },
147       .ltc_divisor        = 25,
148       .height             = 576,
149       .width              = 720,
150       .sar                = { { 16, 15 }, { 64, 45 } },
151       .pix_fmt            = AV_PIX_FMT_YUV422P,
152       .bpm                = 6,
153       .block_sizes        = block_sizes_dv2550,
154       .audio_stride       = 108,
155       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
156       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
157       .audio_shuffle      = dv_audio_shuffle625, },
158     { .dsf                = 0,
159       .video_stype        = 0x14,
160       .frame_size         = 480000, /* SMPTE-370M - 1080i60 100 Mbps */
161       .difseg_size        = 10,    /* also known as "DVCPRO HD" */
162       .n_difchan          = 4,
163       .time_base          = { 1001, 30000 },
164       .ltc_divisor        = 30,
165       .height             = 1080,
166       .width              = 1280,
167       .sar                = { { 1, 1 }, { 3, 2 } },
168       .pix_fmt            = AV_PIX_FMT_YUV422P,
169       .bpm                = 8,
170       .block_sizes        = block_sizes_dv100,
171       .audio_stride       = 90,
172       .audio_min_samples  = { 1580, 1452, 1053 }, /* for 48, 44.1 and 32kHz */
173       .audio_samples_dist = { 1600, 1602, 1602, 1602, 1602 },  /* per SMPTE-314M */
174       .audio_shuffle      = dv_audio_shuffle525, },
175     { .dsf                = 1,
176       .video_stype        = 0x14,
177       .frame_size         = 576000, /* SMPTE-370M - 1080i50 100 Mbps */
178       .difseg_size        = 12,    /* also known as "DVCPRO HD" */
179       .n_difchan          = 4,
180       .time_base          = { 1, 25 },
181       .ltc_divisor        = 25,
182       .height             = 1080,
183       .width              = 1440,
184       .sar                = { { 1, 1 }, { 4, 3 } },
185       .pix_fmt            = AV_PIX_FMT_YUV422P,
186       .bpm                = 8,
187       .block_sizes        = block_sizes_dv100,
188       .audio_stride       = 108,
189       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
190       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
191       .audio_shuffle      = dv_audio_shuffle625, },
192     { .dsf                = 0,
193       .video_stype        = 0x18,
194       .frame_size         = 240000, /* SMPTE-370M - 720p60 100 Mbps */
195       .difseg_size        = 10,    /* also known as "DVCPRO HD" */
196       .n_difchan          = 2,
197       .time_base          = { 1001, 60000 },
198       .ltc_divisor        = 60,
199       .height             = 720,
200       .width              = 960,
201       .sar                = { { 1, 1 }, { 4, 3 } },
202       .pix_fmt            = AV_PIX_FMT_YUV422P,
203       .bpm                = 8,
204       .block_sizes        = block_sizes_dv100,
205       .audio_stride       = 90,
206       .audio_min_samples  = { 1580, 1452, 1053 }, /* for 48, 44.1 and 32kHz */
207       .audio_samples_dist = { 1600, 1602, 1602, 1602, 1602 },  /* per SMPTE-314M */
208       .audio_shuffle      = dv_audio_shuffle525, },
209     { .dsf                = 1,
210       .video_stype        = 0x18,
211       .frame_size         = 288000, /* SMPTE-370M - 720p50 100 Mbps */
212       .difseg_size        = 12,    /* also known as "DVCPRO HD" */
213       .n_difchan          = 2,
214       .time_base          = { 1, 50 },
215       .ltc_divisor        = 50,
216       .height             = 720,
217       .width              = 960,
218       .sar                = { { 1, 1 }, { 4, 3 } },
219       .pix_fmt            = AV_PIX_FMT_YUV422P,
220       .bpm                = 8,
221       .block_sizes        = block_sizes_dv100,
222       .audio_stride       = 90,
223       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
224       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
225       .audio_shuffle      = dv_audio_shuffle625, },
226     { .dsf                = 1,
227       .video_stype        = 0x1,
228       .frame_size         = 144000, /* IEC 61883-5 - 625/50 (PAL) */
229       .difseg_size        = 12,
230       .n_difchan          = 1,
231       .time_base          = { 1, 25 },
232       .ltc_divisor        = 25,
233       .height             = 576,
234       .width              = 720,
235       .sar                = { { 16, 15 }, { 64, 45 } },
236       .pix_fmt            = AV_PIX_FMT_YUV420P,
237       .bpm                = 6,
238       .block_sizes        = block_sizes_dv2550,
239       .audio_stride       = 108,
240       .audio_min_samples  = { 1896, 1742, 1264 }, /* for 48, 44.1 and 32kHz */
241       .audio_samples_dist = { 1920, 1920, 1920, 1920, 1920 },
242       .audio_shuffle      = dv_audio_shuffle625, }
243 };
244
245 void ff_dv_print_profiles(void *logctx, int loglevel)
246 {
247     int i;
248     for (i = 0; i < FF_ARRAY_ELEMS(dv_profiles); i++) {
249         const AVDVProfile *p = &dv_profiles[i];
250         av_log(logctx, loglevel,
251                "Frame size: %dx%d; pixel format: %s, framerate: %d/%d\n",
252                p->width, p->height, av_get_pix_fmt_name(p->pix_fmt),
253                p->time_base.den, p->time_base.num);
254     }
255 }
256
257 #endif /* CONFIG_DVPROFILE */
258
259 const AVDVProfile *av_dv_frame_profile(const AVDVProfile *sys,
260                                        const uint8_t *frame, unsigned buf_size)
261 {
262 #if CONFIG_DVPROFILE
263     int i, dsf, stype;
264
265     if (buf_size < 80 * 5 + 48 + 4)
266         return NULL;
267
268     dsf   = (frame[3] & 0x80) >> 7;
269     stype = frame[80 * 5 + 48 + 3] & 0x1f;
270
271     /* 576i50 25Mbps 4:1:1 is a special case */
272     if (dsf == 1 && stype == 0 && frame[4] & 0x07 /* the APT field */)
273         return &dv_profiles[2];
274
275     for (i = 0; i < FF_ARRAY_ELEMS(dv_profiles); i++)
276         if (dsf == dv_profiles[i].dsf && stype == dv_profiles[i].video_stype)
277             return &dv_profiles[i];
278
279     /* check if old sys matches and assumes corrupted input */
280     if (sys && buf_size == sys->frame_size)
281         return sys;
282 #endif
283
284     return NULL;
285 }
286
287 const AVDVProfile *av_dv_codec_profile(int width, int height,
288                                        enum AVPixelFormat pix_fmt)
289 {
290 #if CONFIG_DVPROFILE
291     int i;
292
293     for (i = 0; i < FF_ARRAY_ELEMS(dv_profiles); i++)
294         if (height  == dv_profiles[i].height  &&
295             pix_fmt == dv_profiles[i].pix_fmt &&
296             width   == dv_profiles[i].width)
297             return &dv_profiles[i];
298 #endif
299
300     return NULL;
301 }