Libav
oggparsespeex.c
Go to the documentation of this file.
1 /*
2  Copyright (C) 2008 Reimar Döffinger
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 #include <stdlib.h>
26 #include "libavutil/bswap.h"
27 #include "libavutil/avstring.h"
29 #include "libavcodec/get_bits.h"
30 #include "libavcodec/bytestream.h"
31 #include "avformat.h"
32 #include "internal.h"
33 #include "oggdec.h"
34 
35 struct speex_params {
38  int seq;
39 };
40 
41 static int speex_header(AVFormatContext *s, int idx) {
42  struct ogg *ogg = s->priv_data;
43  struct ogg_stream *os = ogg->streams + idx;
44  struct speex_params *spxp = os->private;
45  AVStream *st = s->streams[idx];
46  uint8_t *p = os->buf + os->pstart;
47 
48  if (!spxp) {
49  spxp = av_mallocz(sizeof(*spxp));
50  os->private = spxp;
51  }
52 
53  if (spxp->seq > 1)
54  return 0;
55 
56  if (spxp->seq == 0) {
57  int frames_per_packet;
60 
61  st->codec->sample_rate = AV_RL32(p + 36);
62  st->codec->channels = AV_RL32(p + 48);
63  if (st->codec->channels < 1 || st->codec->channels > 2) {
64  av_log(s, AV_LOG_ERROR, "invalid channel count. Speex must be mono or stereo.\n");
65  return AVERROR_INVALIDDATA;
66  }
69 
70  spxp->packet_size = AV_RL32(p + 56);
71  frames_per_packet = AV_RL32(p + 64);
72  if (frames_per_packet)
73  spxp->packet_size *= frames_per_packet;
74 
75  st->codec->extradata_size = os->psize;
78  memcpy(st->codec->extradata, p, st->codec->extradata_size);
79 
80  avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
81  } else
82  ff_vorbis_comment(s, &st->metadata, p, os->psize);
83 
84  spxp->seq++;
85  return 1;
86 }
87 
88 static int ogg_page_packets(struct ogg_stream *os)
89 {
90  int i;
91  int packets = 0;
92  for (i = 0; i < os->nsegs; i++)
93  if (os->segments[i] < 255)
94  packets++;
95  return packets;
96 }
97 
98 static int speex_packet(AVFormatContext *s, int idx)
99 {
100  struct ogg *ogg = s->priv_data;
101  struct ogg_stream *os = ogg->streams + idx;
102  struct speex_params *spxp = os->private;
103  int packet_size = spxp->packet_size;
104 
105  if (os->flags & OGG_FLAG_EOS && os->lastpts != AV_NOPTS_VALUE &&
106  os->granule > 0) {
107  /* first packet of final page. we have to calculate the final packet
108  duration here because it is the only place we know the next-to-last
109  granule position. */
110  spxp->final_packet_duration = os->granule - os->lastpts -
111  packet_size * (ogg_page_packets(os) - 1);
112  }
113 
114  if (!os->lastpts && os->granule > 0)
115  /* first packet */
116  os->lastpts = os->lastdts = os->granule - packet_size *
117  ogg_page_packets(os);
118  if (os->flags & OGG_FLAG_EOS && os->segp == os->nsegs &&
119  spxp->final_packet_duration)
120  /* final packet */
121  os->pduration = spxp->final_packet_duration;
122  else
123  os->pduration = packet_size;
124 
125  return 0;
126 }
127 
128 const struct ogg_codec ff_speex_codec = {
129  .magic = "Speex ",
130  .magicsize = 8,
131  .header = speex_header,
132  .packet = speex_packet,
133  .nb_header = 2,
134 };