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