segment: Use the public av_write_header/av_write_trailer functions
[ffmpeg.git] / libavformat / oggdec.h
1 /**
2     Copyright (C) 2005  Michael Ahlberg, Måns Rullgård
3
4     Permission is hereby granted, free of charge, to any person
5     obtaining a copy of this software and associated documentation
6     files (the "Software"), to deal in the Software without
7     restriction, including without limitation the rights to use, copy,
8     modify, merge, publish, distribute, sublicense, and/or sell copies
9     of the Software, and to permit persons to whom the Software is
10     furnished to do so, subject to the following conditions:
11
12     The above copyright notice and this permission notice shall be
13     included in all copies or substantial portions of the Software.
14
15     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
19     HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
20     WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21     OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22     DEALINGS IN THE SOFTWARE.
23 **/
24
25 #ifndef AVFORMAT_OGGDEC_H
26 #define AVFORMAT_OGGDEC_H
27
28 #include "avformat.h"
29 #include "metadata.h"
30
31 struct ogg_codec {
32     const int8_t *magic;
33     uint8_t magicsize;
34     const int8_t *name;
35     /**
36      * Attempt to process a packet as a header
37      * @return 1 if the packet was a valid header,
38      *         0 if the packet was not a header (was a data packet)
39      *         -1 if an error occurred or for unsupported stream
40      */
41     int (*header)(AVFormatContext *, int);
42     int (*packet)(AVFormatContext *, int);
43     /**
44      * Translate a granule into a timestamp.
45      * Will set dts if non-null and known.
46      * @return pts
47      */
48     uint64_t (*gptopts)(AVFormatContext *, int, uint64_t, int64_t *dts);
49     /**
50      * 1 if granule is the start time of the associated packet.
51      * 0 if granule is the end time of the associated packet.
52      */
53     int granule_is_start;
54     /**
55      * Number of expected headers
56      */
57     int nb_header;
58 };
59
60 struct ogg_stream {
61     uint8_t *buf;
62     unsigned int bufsize;
63     unsigned int bufpos;
64     unsigned int pstart;
65     unsigned int psize;
66     unsigned int pflags;
67     unsigned int pduration;
68     uint32_t serial;
69     uint64_t granule;
70     uint64_t start_granule;
71     int64_t lastpts;
72     int64_t lastdts;
73     int64_t sync_pos;   ///< file offset of the first page needed to reconstruct the current packet
74     int64_t page_pos;   ///< file offset of the current page
75     int flags;
76     const struct ogg_codec *codec;
77     int header;
78     int nsegs, segp;
79     uint8_t segments[255];
80     int incomplete; ///< whether we're expecting a continuation in the next page
81     int page_end;   ///< current packet is the last one completed in the page
82     int keyframe_seek;
83     int nb_header; ///< set to the number of parsed headers
84     void *private;
85 };
86
87 struct ogg_state {
88     uint64_t pos;
89     int curidx;
90     struct ogg_state *next;
91     int nstreams;
92     struct ogg_stream streams[1];
93 };
94
95 struct ogg {
96     struct ogg_stream *streams;
97     int nstreams;
98     int headers;
99     int curidx;
100     struct ogg_state *state;
101 };
102
103 #define OGG_FLAG_CONT 1
104 #define OGG_FLAG_BOS  2
105 #define OGG_FLAG_EOS  4
106
107 #define OGG_NOGRANULE_VALUE -1ull
108
109 extern const struct ogg_codec ff_celt_codec;
110 extern const struct ogg_codec ff_dirac_codec;
111 extern const struct ogg_codec ff_flac_codec;
112 extern const struct ogg_codec ff_ogm_audio_codec;
113 extern const struct ogg_codec ff_ogm_old_codec;
114 extern const struct ogg_codec ff_ogm_text_codec;
115 extern const struct ogg_codec ff_ogm_video_codec;
116 extern const struct ogg_codec ff_old_dirac_codec;
117 extern const struct ogg_codec ff_old_flac_codec;
118 extern const struct ogg_codec ff_skeleton_codec;
119 extern const struct ogg_codec ff_speex_codec;
120 extern const struct ogg_codec ff_theora_codec;
121 extern const struct ogg_codec ff_vorbis_codec;
122
123 int ff_vorbis_comment(AVFormatContext *ms, AVDictionary **m, const uint8_t *buf, int size);
124
125 static inline int
126 ogg_find_stream (struct ogg * ogg, int serial)
127 {
128     int i;
129
130     for (i = 0; i < ogg->nstreams; i++)
131         if (ogg->streams[i].serial == serial)
132             return i;
133
134     return -1;
135 }
136
137 static inline uint64_t
138 ogg_gptopts (AVFormatContext * s, int i, uint64_t gp, int64_t *dts)
139 {
140     struct ogg *ogg = s->priv_data;
141     struct ogg_stream *os = ogg->streams + i;
142     uint64_t pts = AV_NOPTS_VALUE;
143
144     if(os->codec && os->codec->gptopts){
145         pts = os->codec->gptopts(s, i, gp, dts);
146     } else {
147         pts = gp;
148         if (dts)
149             *dts = pts;
150     }
151
152     return pts;
153 }
154
155 #endif /* AVFORMAT_OGGDEC_H */