avformat/matroskaenc: implement AVOutputFormat.query_codec for webm
[ffmpeg.git] / libavformat / s337m.c
1 /*
2  * Copyright (C) 2017 foo86
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include "libavutil/intreadwrite.h"
22 #include "avformat.h"
23 #include "spdif.h"
24
25 #define MARKER_16LE         0x72F81F4E
26 #define MARKER_20LE         0x20876FF0E154
27 #define MARKER_24LE         0x72F8961F4EA5
28
29 #define IS_16LE_MARKER(state)   ((state & 0xFFFFFFFF) == MARKER_16LE)
30 #define IS_20LE_MARKER(state)   ((state & 0xF0FFFFF0FFFF) == MARKER_20LE)
31 #define IS_24LE_MARKER(state)   ((state & 0xFFFFFFFFFFFF) == MARKER_24LE)
32 #define IS_LE_MARKER(state)     (IS_16LE_MARKER(state) || IS_20LE_MARKER(state) || IS_24LE_MARKER(state))
33
34 static int s337m_get_offset_and_codec(AVFormatContext *s,
35                                       uint64_t state,
36                                       int data_type, int data_size,
37                                       int *offset, enum AVCodecID *codec)
38 {
39     int word_bits;
40
41     if (IS_16LE_MARKER(state)) {
42         word_bits = 16;
43     } else if (IS_20LE_MARKER(state)) {
44         data_type >>= 8;
45         data_size >>= 4;
46         word_bits = 20;
47     } else {
48         data_type >>= 8;
49         word_bits = 24;
50     }
51
52     if ((data_type & 0x1F) != 0x1C) {
53         if (s)
54             avpriv_report_missing_feature(s, "Data type %#x in SMPTE 337M", data_type & 0x1F);
55         return AVERROR_PATCHWELCOME;
56     }
57
58     if (codec)
59         *codec = AV_CODEC_ID_DOLBY_E;
60
61     switch (data_size / word_bits) {
62     case 3648:
63         *offset = 1920;
64         break;
65     case 3644:
66         *offset = 2002;
67         break;
68     case 3640:
69         *offset = 2000;
70         break;
71     case 3040:
72         *offset = 1601;
73         break;
74     default:
75         if (s)
76             avpriv_report_missing_feature(s, "Dolby E data size %d in SMPTE 337M", data_size);
77         return AVERROR_PATCHWELCOME;
78     }
79
80     *offset -= 4;
81     *offset *= (word_bits + 7 >> 3) * 2;
82     return 0;
83 }
84
85 static int s337m_probe(AVProbeData *p)
86 {
87     uint64_t state = 0;
88     int markers[3] = { 0 };
89     int i, pos, sum, max, data_type, data_size, offset;
90     uint8_t *buf;
91
92     for (pos = 0; pos < p->buf_size; pos++) {
93         state = (state << 8) | p->buf[pos];
94         if (!IS_LE_MARKER(state))
95             continue;
96
97         buf = p->buf + pos + 1;
98         if (IS_16LE_MARKER(state)) {
99             data_type = AV_RL16(buf    );
100             data_size = AV_RL16(buf + 2);
101         } else {
102             data_type = AV_RL24(buf    );
103             data_size = AV_RL24(buf + 3);
104         }
105
106         if (s337m_get_offset_and_codec(NULL, state, data_type, data_size, &offset, NULL))
107             continue;
108
109         i = IS_16LE_MARKER(state) ? 0 : IS_20LE_MARKER(state) ? 1 : 2;
110         markers[i]++;
111
112         pos  += IS_16LE_MARKER(state) ? 4 : 6;
113         pos  += offset;
114         state = 0;
115     }
116
117     sum = max = 0;
118     for (i = 0; i < FF_ARRAY_ELEMS(markers); i++) {
119         sum += markers[i];
120         if (markers[max] < markers[i])
121             max = i;
122     }
123
124     if (markers[max] > 3 && markers[max] * 4 > sum * 3)
125         return AVPROBE_SCORE_EXTENSION + 1;
126
127     return 0;
128 }
129
130 static int s337m_read_header(AVFormatContext *s)
131 {
132     s->ctx_flags |= AVFMTCTX_NOHEADER;
133     return 0;
134 }
135
136 static void bswap_buf24(uint8_t *data, int size)
137 {
138     int i;
139
140     for (i = 0; i < size / 3; i++, data += 3)
141         FFSWAP(uint8_t, data[0], data[2]);
142 }
143
144 static int s337m_read_packet(AVFormatContext *s, AVPacket *pkt)
145 {
146     AVIOContext *pb = s->pb;
147     uint64_t state = 0;
148     int ret, data_type, data_size, offset;
149     enum AVCodecID codec;
150     int64_t pos;
151
152     while (!IS_LE_MARKER(state)) {
153         state = (state << 8) | avio_r8(pb);
154         if (avio_feof(pb))
155             return AVERROR_EOF;
156     }
157
158     if (IS_16LE_MARKER(state)) {
159         data_type = avio_rl16(pb);
160         data_size = avio_rl16(pb);
161     } else {
162         data_type = avio_rl24(pb);
163         data_size = avio_rl24(pb);
164     }
165
166     pos = avio_tell(pb);
167
168     if ((ret = s337m_get_offset_and_codec(s, state, data_type, data_size, &offset, &codec)) < 0)
169         return ret;
170
171     if ((ret = av_new_packet(pkt, offset)) < 0)
172         return ret;
173
174     pkt->pos = pos;
175
176     if (avio_read(pb, pkt->data, pkt->size) < pkt->size) {
177         av_packet_unref(pkt);
178         return AVERROR_EOF;
179     }
180
181     if (IS_16LE_MARKER(state))
182         ff_spdif_bswap_buf16((uint16_t *)pkt->data, (uint16_t *)pkt->data, pkt->size >> 1);
183     else
184         bswap_buf24(pkt->data, pkt->size);
185
186     if (!s->nb_streams) {
187         AVStream *st = avformat_new_stream(s, NULL);
188         if (!st) {
189             av_packet_unref(pkt);
190             return AVERROR(ENOMEM);
191         }
192         st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
193         st->codecpar->codec_id   = codec;
194     }
195
196     return 0;
197 }
198
199 AVInputFormat ff_s337m_demuxer = {
200     .name           = "s337m",
201     .long_name      = NULL_IF_CONFIG_SMALL("SMPTE 337M"),
202     .read_probe     = s337m_probe,
203     .read_header    = s337m_read_header,
204     .read_packet    = s337m_read_packet,
205     .flags          = AVFMT_GENERIC_INDEX,
206 };