move common stuff from avienc.c and wav.c to new file riff.c
[ffmpeg.git] / libavformat / gxf.c
1 /*
2  * GXF demuxer.
3  * Copyright (c) 2006 Reimar Doeffinger.
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18  */
19 #include "avformat.h"
20
21 typedef enum {
22     PKT_MAP = 0xbc,
23     PKT_MEDIA = 0xbf,
24     PKT_EOS = 0xfb,
25     PKT_FLT = 0xfc,
26     PKT_UMF = 0xfd
27 } pkt_type_t;
28
29 /**
30  * \brief parses a packet header, extracting type and length
31  * \param pb ByteIOContext to read header from
32  * \param type detected packet type is stored here
33  * \param length detected packet length, excluding header is stored here
34  * \return 0 if header not found or contains invalid data, 1 otherwise
35  */
36 static int parse_packet_header(ByteIOContext *pb, pkt_type_t *type, int *length) {
37     if (get_be32(pb))
38         return 0;
39     if (get_byte(pb) != 1)
40         return 0;
41     *type = get_byte(pb);
42     *length = get_be32(pb);
43     if ((*length >> 24) || *length < 16)
44         return 0;
45     *length -= 16;
46     if (get_be32(pb))
47         return 0;
48     if (get_byte(pb) != 0xe1)
49         return 0;
50     if (get_byte(pb) != 0xe2)
51         return 0;
52     return 1;
53 }
54
55 /**
56  * \brief check if file starts with a PKT_MAP header
57  */
58 static int gxf_probe(AVProbeData *p) {
59     static const uint8_t startcode[] = {0, 0, 0, 0, 1, 0xbc}; // start with map packet
60     static const uint8_t endcode[] = {0, 0, 0, 0, 0xe1, 0xe2};
61     if (p->buf_size < 16)
62         return 0;
63     if (!memcmp(p->buf, startcode, sizeof(startcode)) &&
64         !memcmp(&p->buf[16 - sizeof(endcode)], endcode, sizeof(endcode)))
65         return AVPROBE_SCORE_MAX;
66     return 0;
67 }
68
69 /**
70  * \brief gets the stream index for the track with the specified id, creates new
71  *        stream if not found
72  * \param stream id of stream to find / add
73  * \param format stream format identifier
74  */
75 static int get_sindex(AVFormatContext *s, int id, int format) {
76     int i;
77     AVStream *st = NULL;
78     for (i = 0; i < s->nb_streams; i++) {
79         if (s->streams[i]->id == id)
80             return i;
81     }
82     st = av_new_stream(s, id);
83     switch (format) {
84         case 3:
85         case 4:
86             st->codec->codec_type = CODEC_TYPE_VIDEO;
87             st->codec->codec_id = CODEC_ID_MJPEG;
88             st->codec->codec_tag = MKTAG('M', 'J', 'P', 'G');
89             break;
90         case 13:
91         case 15:
92             st->codec->codec_type = CODEC_TYPE_VIDEO;
93             st->codec->codec_id = CODEC_ID_DVVIDEO;
94             st->codec->codec_tag = MKTAG('d', 'v', 'c', ' ');
95             break;
96         case 14:
97         case 16:
98             st->codec->codec_type = CODEC_TYPE_VIDEO;
99             st->codec->codec_id = CODEC_ID_DVVIDEO;
100             st->codec->codec_tag = MKTAG('d', 'v', 'c', 'p');
101             break;
102         case 11:
103         case 12:
104         case 20:
105             st->codec->codec_type = CODEC_TYPE_VIDEO;
106             st->codec->codec_id = CODEC_ID_MPEG2VIDEO;
107             st->codec->codec_tag = MKTAG('M', 'P', 'G', '2');
108             break;
109         case 22:
110         case 23:
111             st->codec->codec_type = CODEC_TYPE_VIDEO;
112             st->codec->codec_id = CODEC_ID_MPEG1VIDEO;
113             st->codec->codec_tag = MKTAG('M', 'P', 'G', '1');
114             break;
115         case 9:
116             st->codec->codec_type = CODEC_TYPE_AUDIO;
117             st->codec->codec_id = CODEC_ID_PCM_S24LE;
118             st->codec->codec_tag = 0x1;
119             st->codec->channels = 1;
120             st->codec->sample_rate = 48000;
121             st->codec->bit_rate = 3 * 1 * 48000 * 8;
122             st->codec->block_align = 3 * 1;
123             st->codec->bits_per_sample = 24;
124             break;
125         case 10:
126             st->codec->codec_type = CODEC_TYPE_AUDIO;
127             st->codec->codec_id = CODEC_ID_PCM_S16LE;
128             st->codec->codec_tag = 0x1;
129             st->codec->channels = 1;
130             st->codec->sample_rate = 48000;
131             st->codec->bit_rate = 2 * 1 * 48000 * 8;
132             st->codec->block_align = 2 * 1;
133             st->codec->bits_per_sample = 16;
134             break;
135         case 17:
136             st->codec->codec_type = CODEC_TYPE_AUDIO;
137             st->codec->codec_id = CODEC_ID_AC3;
138             st->codec->codec_tag = 0x2000;
139             st->codec->channels = 2;
140             st->codec->sample_rate = 48000;
141             break;
142         default:
143             st->codec->codec_type = CODEC_TYPE_UNKNOWN;
144             st->codec->codec_id = CODEC_ID_NONE;
145             break;
146     }
147     return s->nb_streams - 1;
148 }
149
150 static int gxf_header(AVFormatContext *s, AVFormatParameters *ap) {
151     ByteIOContext *pb = &s->pb;
152     pkt_type_t pkt_type;
153     int map_len;
154     int len;
155     if (!parse_packet_header(pb, &pkt_type, &map_len) || pkt_type != PKT_MAP) {
156         av_log(s, AV_LOG_ERROR, "GXF: map packet not found\n");
157         return 0;
158     }
159     map_len -= 2;
160     if (get_byte(pb) != 0x0e0 || get_byte(pb) != 0xff) {
161         av_log(s, AV_LOG_ERROR, "GXF: unknown version or invalid map preamble\n");
162         return 0;
163     }
164     map_len -= 2;
165     len = get_be16(pb); // length of material data section
166     if (len > map_len) {
167         av_log(s, AV_LOG_ERROR, "GXF: material data longer than map data\n");
168         return 0;
169     }
170     map_len -= len;
171     url_fskip(pb, len);
172     map_len -= 2;
173     len = get_be16(pb); // length of track description
174     if (len > map_len) {
175         av_log(s, AV_LOG_ERROR, "GXF: track description longer than map data\n");
176         return 0;
177     }
178     map_len -= len;
179     while (len > 0) {
180         int track_type, track_id, track_len;
181         len -= 4;
182         track_type = get_byte(pb);
183         track_id = get_byte(pb);
184         track_len = get_be16(pb);
185         len -= track_len;
186         url_fskip(pb, track_len);
187         if (!(track_type & 0x80)) {
188            av_log(s, AV_LOG_ERROR, "GXF: invalid track type %x\n", track_type);
189            continue;
190         }
191         track_type &= 0x7f;
192         if ((track_id & 0xc0) != 0xc0) {
193            av_log(s, AV_LOG_ERROR, "GXF: invalid track id %x\n", track_id);
194            continue;
195         }
196         track_id &= 0x3f;
197         get_sindex(s, track_id, track_type);
198     }
199     if (len < 0)
200         av_log(s, AV_LOG_ERROR, "GXF: invalid track description length specified\n");
201     if (map_len)
202         url_fskip(pb, map_len);
203     return 0;
204 }
205
206 static int gxf_packet(AVFormatContext *s, AVPacket *pkt) {
207     ByteIOContext *pb = &s->pb;
208     pkt_type_t pkt_type;
209     int pkt_len;
210     while (!url_feof(pb)) {
211         int track_type, track_id, ret;
212         if (!parse_packet_header(pb, &pkt_type, &pkt_len)) {
213             if (!url_feof(pb))
214                 av_log(s, AV_LOG_ERROR, "GXF: sync lost\n");
215             return -1;
216         }
217         if (pkt_type != PKT_MEDIA) {
218             url_fskip(pb, pkt_len);
219             continue;
220         }
221         if (pkt_len < 16) {
222             av_log(s, AV_LOG_ERROR, "GXF: invalid media packet length\n");
223             continue;
224         }
225         pkt_len -= 16;
226         track_type = get_byte(pb);
227         track_id = get_byte(pb);
228         get_be32(pb); // field number
229         get_be32(pb); // field information
230         get_be32(pb); // "timeline" field number
231         get_byte(pb); // flags
232         get_byte(pb); // reserved
233         // NOTE: there is also data length information in the
234         // field information, it might be better to take this int account
235         // as well.
236         ret = av_get_packet(pb, pkt, pkt_len);
237         pkt->stream_index = get_sindex(s, track_id, track_type);
238         return ret;
239     }
240     return AVERROR_IO;
241 }
242
243 AVInputFormat gxf_demuxer = {
244     "gxf",
245     "GXF format",
246     0,
247     gxf_probe,
248     gxf_header,
249     gxf_packet,
250     NULL,
251     NULL,
252 };