Mercurial > libavformat.hg
annotate ogg.c @ 260:f3b8a6e60e71 libavformat
PowerPC warnings & optimisations patch by (Dan Christiansen <danchr at daimi dot au dot dk>)
author | michaelni |
---|---|
date | Sun, 28 Sep 2003 22:53:25 +0000 |
parents | 1e2f55eced38 |
children | a313e1080322 |
rev | line source |
---|---|
0 | 1 /* |
2 * Ogg bitstream support | |
3 * Mark Hills <mark@pogo.org.uk> | |
4 * | |
5 * Uses libogg, but requires libvorbisenc to construct correct headers | |
6 * when containing Vorbis stream -- currently the only format supported | |
7 */ | |
8 | |
9 #include <stdio.h> | |
10 #include <time.h> | |
11 | |
12 #include <ogg/ogg.h> | |
13 #include <vorbis/vorbisenc.h> | |
14 | |
15 #include "avformat.h" | |
16 #include "oggvorbis.h" | |
17 | |
18 #define DECODER_BUFFER_SIZE 4096 | |
19 | |
20 | |
21 typedef struct OggContext { | |
22 /* output */ | |
23 ogg_stream_state os ; | |
24 int header_handled ; | |
25 ogg_int64_t base_packet_no ; | |
26 ogg_int64_t base_granule_pos ; | |
27 | |
28 /* input */ | |
29 ogg_sync_state oy ; | |
30 } OggContext ; | |
31 | |
32 | |
35 | 33 static int ogg_write_header(AVFormatContext *avfcontext) |
34 { | |
35 OggContext *context = avfcontext->priv_data; | |
0 | 36 AVCodecContext *avccontext ; |
37 vorbis_info vi ; | |
38 vorbis_dsp_state vd ; | |
39 vorbis_comment vc ; | |
40 vorbis_block vb ; | |
41 ogg_packet header, header_comm, header_code ; | |
42 int n ; | |
43 | |
44 srand(time(NULL)); | |
45 ogg_stream_init(&context->os, rand()); | |
46 | |
47 for(n = 0 ; n < avfcontext->nb_streams ; n++) { | |
48 avccontext = &avfcontext->streams[n]->codec ; | |
49 | |
50 /* begin vorbis specific code */ | |
51 | |
52 vorbis_info_init(&vi) ; | |
53 | |
54 /* code copied from libavcodec/oggvorbis.c */ | |
55 | |
56 if(oggvorbis_init_encoder(&vi, avccontext) < 0) { | |
57 fprintf(stderr, "ogg_write_header: init_encoder failed") ; | |
58 return -1 ; | |
59 } | |
60 | |
61 vorbis_analysis_init(&vd, &vi) ; | |
62 vorbis_block_init(&vd, &vb) ; | |
63 | |
64 vorbis_comment_init(&vc) ; | |
65 vorbis_comment_add_tag(&vc, "encoder", "ffmpeg") ; | |
66 if(*avfcontext->title) | |
67 vorbis_comment_add_tag(&vc, "title", avfcontext->title) ; | |
68 | |
69 vorbis_analysis_headerout(&vd, &vc, &header, | |
70 &header_comm, &header_code) ; | |
71 ogg_stream_packetin(&context->os, &header) ; | |
72 ogg_stream_packetin(&context->os, &header_comm) ; | |
73 ogg_stream_packetin(&context->os, &header_code) ; | |
74 | |
68
1e2f55eced38
ogg deallocate patch by (Mark Hills <mark at pogo dot org dot uk>)
michaelni
parents:
35
diff
changeset
|
75 vorbis_block_clear(&vb) ; |
1e2f55eced38
ogg deallocate patch by (Mark Hills <mark at pogo dot org dot uk>)
michaelni
parents:
35
diff
changeset
|
76 vorbis_dsp_clear(&vd) ; |
1e2f55eced38
ogg deallocate patch by (Mark Hills <mark at pogo dot org dot uk>)
michaelni
parents:
35
diff
changeset
|
77 vorbis_info_clear(&vi) ; |
0 | 78 vorbis_comment_clear(&vc) ; |
68
1e2f55eced38
ogg deallocate patch by (Mark Hills <mark at pogo dot org dot uk>)
michaelni
parents:
35
diff
changeset
|
79 |
0 | 80 /* end of vorbis specific code */ |
81 | |
82 context->header_handled = 0 ; | |
83 context->base_packet_no = 0 ; | |
84 } | |
85 | |
86 return 0 ; | |
87 } | |
88 | |
89 | |
90 static int ogg_write_packet(AVFormatContext *avfcontext, | |
91 int stream_index, | |
92 unsigned char *buf, int size, int force_pts) | |
93 { | |
94 OggContext *context = avfcontext->priv_data ; | |
95 ogg_packet *op ; | |
96 ogg_page og ; | |
97 int l = 0 ; | |
98 | |
99 /* flush header packets so audio starts on a new page */ | |
100 | |
101 if(!context->header_handled) { | |
102 while(ogg_stream_flush(&context->os, &og)) { | |
103 put_buffer(&avfcontext->pb, og.header, og.header_len) ; | |
104 put_buffer(&avfcontext->pb, og.body, og.body_len) ; | |
105 put_flush_packet(&avfcontext->pb); | |
106 } | |
107 context->header_handled = 1 ; | |
108 } | |
109 | |
110 while(l < size) { | |
111 op = (ogg_packet*)(buf + l) ; | |
112 op->packet = buf + l + sizeof(ogg_packet) ; /* fix data pointer */ | |
113 | |
114 if(!context->base_packet_no) { /* this is the first packet */ | |
115 context->base_packet_no = op->packetno ; | |
116 context->base_granule_pos = op->granulepos ; | |
117 } | |
118 | |
119 /* correct the fields in the packet -- essential for streaming */ | |
120 | |
121 op->packetno -= context->base_packet_no ; | |
122 op->granulepos -= context->base_granule_pos ; | |
123 | |
124 ogg_stream_packetin(&context->os, op) ; | |
125 l += sizeof(ogg_packet) + op->bytes ; | |
126 | |
127 while(ogg_stream_pageout(&context->os, &og)) { | |
128 put_buffer(&avfcontext->pb, og.header, og.header_len) ; | |
129 put_buffer(&avfcontext->pb, og.body, og.body_len) ; | |
130 put_flush_packet(&avfcontext->pb); | |
131 } | |
132 } | |
133 | |
134 return 0; | |
135 } | |
136 | |
137 | |
138 static int ogg_write_trailer(AVFormatContext *avfcontext) { | |
139 OggContext *context = avfcontext->priv_data ; | |
140 ogg_page og ; | |
141 | |
142 while(ogg_stream_flush(&context->os, &og)) { | |
143 put_buffer(&avfcontext->pb, og.header, og.header_len) ; | |
144 put_buffer(&avfcontext->pb, og.body, og.body_len) ; | |
145 put_flush_packet(&avfcontext->pb); | |
146 } | |
147 | |
148 ogg_stream_clear(&context->os) ; | |
149 return 0 ; | |
150 } | |
151 | |
152 | |
153 static AVOutputFormat ogg_oformat = { | |
154 "ogg", | |
155 "Ogg Vorbis", | |
156 "audio/x-vorbis", | |
157 "ogg", | |
158 sizeof(OggContext), | |
159 CODEC_ID_VORBIS, | |
160 0, | |
161 ogg_write_header, | |
162 ogg_write_packet, | |
163 ogg_write_trailer, | |
164 } ; | |
165 | |
166 | |
167 static int next_packet(AVFormatContext *avfcontext, ogg_packet *op) { | |
168 OggContext *context = avfcontext->priv_data ; | |
169 ogg_page og ; | |
170 char *buf ; | |
171 | |
172 while(ogg_stream_packetout(&context->os, op) != 1) { | |
173 | |
174 /* while no pages are available, read in more data to the sync */ | |
175 while(ogg_sync_pageout(&context->oy, &og) != 1) { | |
176 buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ; | |
177 if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0) | |
178 return 1 ; | |
179 ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ; | |
180 } | |
181 | |
182 /* got a page. Feed it into the stream and get the packet */ | |
183 if(ogg_stream_pagein(&context->os, &og) != 0) | |
184 return 1 ; | |
185 } | |
186 | |
187 return 0 ; | |
188 } | |
189 | |
190 | |
191 static int ogg_read_header(AVFormatContext *avfcontext, AVFormatParameters *ap) | |
192 { | |
35 | 193 OggContext *context = avfcontext->priv_data; |
0 | 194 char *buf ; |
195 ogg_page og ; | |
196 AVStream *ast ; | |
197 | |
198 ogg_sync_init(&context->oy) ; | |
199 buf = ogg_sync_buffer(&context->oy, DECODER_BUFFER_SIZE) ; | |
200 | |
201 if(get_buffer(&avfcontext->pb, buf, DECODER_BUFFER_SIZE) <= 0) | |
202 return -EIO ; | |
203 | |
204 ogg_sync_wrote(&context->oy, DECODER_BUFFER_SIZE) ; | |
205 ogg_sync_pageout(&context->oy, &og) ; | |
206 ogg_stream_init(&context->os, ogg_page_serialno(&og)) ; | |
207 ogg_stream_pagein(&context->os, &og) ; | |
208 | |
209 /* currently only one vorbis stream supported */ | |
210 | |
211 ast = av_new_stream(avfcontext, 0) ; | |
212 if(!ast) | |
213 return AVERROR_NOMEM ; | |
214 | |
215 ast->codec.codec_type = CODEC_TYPE_AUDIO ; | |
216 ast->codec.codec_id = CODEC_ID_VORBIS ; | |
217 | |
218 return 0 ; | |
219 } | |
220 | |
221 | |
222 static int ogg_read_packet(AVFormatContext *avfcontext, AVPacket *pkt) { | |
223 ogg_packet op ; | |
224 | |
225 if(next_packet(avfcontext, &op)) | |
226 return -EIO ; | |
227 if(av_new_packet(pkt, sizeof(ogg_packet) + op.bytes) < 0) | |
228 return -EIO ; | |
229 pkt->stream_index = 0 ; | |
230 memcpy(pkt->data, &op, sizeof(ogg_packet)) ; | |
231 memcpy(pkt->data + sizeof(ogg_packet), op.packet, op.bytes) ; | |
232 | |
233 return sizeof(ogg_packet) + op.bytes ; | |
234 } | |
235 | |
236 | |
237 static int ogg_read_close(AVFormatContext *avfcontext) { | |
238 OggContext *context = avfcontext->priv_data ; | |
239 | |
240 ogg_stream_clear(&context->os) ; | |
241 ogg_sync_clear(&context->oy) ; | |
242 | |
243 return 0 ; | |
244 } | |
245 | |
246 | |
247 static AVInputFormat ogg_iformat = { | |
248 "ogg", | |
249 "Ogg Vorbis", | |
250 sizeof(OggContext), | |
251 NULL, | |
252 ogg_read_header, | |
253 ogg_read_packet, | |
254 ogg_read_close, | |
255 .extensions = "ogg", | |
256 } ; | |
257 | |
258 | |
259 int ogg_init(void) { | |
260 av_register_output_format(&ogg_oformat) ; | |
261 av_register_input_format(&ogg_iformat); | |
262 return 0 ; | |
263 } |