]> git.sesse.net Git - ffmpeg/blob - libavcodec/mpeg12framerate.c
Merge commit '3152058bf1dca318898550efacf0286f4836cae6'
[ffmpeg] / libavcodec / mpeg12framerate.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg 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  * FFmpeg 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 FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include "libavutil/rational.h"
20
21 #include "mpeg12.h"
22 #include "mpeg12data.h"
23
24 const AVRational ff_mpeg12_frame_rate_tab[16] = {
25     {    0,    0},
26     {24000, 1001},
27     {   24,    1},
28     {   25,    1},
29     {30000, 1001},
30     {   30,    1},
31     {   50,    1},
32     {60000, 1001},
33     {   60,    1},
34   // Xing's 15fps: (9)
35     {   15,    1},
36   // libmpeg3's "Unofficial economy rates": (10-13)
37     {    5,    1},
38     {   10,    1},
39     {   12,    1},
40     {   15,    1},
41     {    0,    0},
42 };
43
44 void ff_mpeg12_find_best_frame_rate(AVRational frame_rate,
45                                     int *code, int *ext_n, int *ext_d,
46                                     int nonstandard)
47 {
48     int mpeg2 = ext_n && ext_d;
49     int max_code = nonstandard ? 12 : 8;
50     int c, n, d, best_c, best_n, best_d;
51     AVRational best_error = { INT_MAX, 1 };
52
53     // Default to NTSC if the inputs make no sense.
54     best_c = 4;
55     best_n = best_d = 1;
56
57     for (c = 1; c <= max_code; c++) {
58         if (av_cmp_q(frame_rate, ff_mpeg12_frame_rate_tab[c]) == 0) {
59             best_c = c;
60             goto found;
61         }
62     }
63
64     for (c = 1; c <= max_code; c++) {
65         for (n = 1; n <= (mpeg2 ? 4 : 1); n++) {
66             for (d = 1; d <= (mpeg2 ? 32 : 1); d++) {
67                 AVRational test, error;
68                 int cmp;
69
70                 test = av_mul_q(ff_mpeg12_frame_rate_tab[c],
71                                 (AVRational) { n, d });
72
73                 cmp = av_cmp_q(test, frame_rate);
74                 if (cmp == 0) {
75                     best_c = c;
76                     best_n = n;
77                     best_d = d;
78                     goto found;
79                 }
80
81                 if (cmp < 0)
82                     error = av_div_q(frame_rate, test);
83                 else
84                     error = av_div_q(test, frame_rate);
85
86                 cmp = av_cmp_q(error, best_error);
87                 if (cmp < 0 || (cmp == 0 && n == 1 && d == 1)) {
88                     best_c = c;
89                     best_n = n;
90                     best_d = d;
91                     best_error = error;
92                 }
93             }
94         }
95     }
96
97 found:
98     *code = best_c;
99     if (mpeg2) {
100         *ext_n = best_n - 1;
101         *ext_d = best_d - 1;
102     }
103 }