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 };
void * av_malloc(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:61
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård.
Definition: oggdec.h:31
static int ogg_page_packets(struct ogg_stream *os)
Definition: oggparsespeex.c:88
const struct ogg_codec ff_speex_codec
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:3283
int flags
Definition: oggdec.h:76
#define AV_CH_LAYOUT_STEREO
int64_t lastpts
Definition: oggdec.h:72
Format I/O context.
Definition: avformat.h:828
unsigned int psize
Definition: oggdec.h:66
uint8_t
int final_packet_duration
Definition: oggparsespeex.c:37
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1454
AVStream ** streams
Definition: avformat.h:876
bitstream reader API header.
int ff_vorbis_comment(AVFormatContext *ms, AVDictionary **m, const uint8_t *buf, int size)
#define OGG_FLAG_EOS
Definition: oggdec.h:106
void av_log(void *avcl, int level, const char *fmt,...)
Definition: log.c:146
uint8_t segments[255]
Definition: oggdec.h:80
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:2165
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:641
audio channel layout utility functions
uint64_t granule
Definition: oggdec.h:70
unsigned int pstart
Definition: oggdec.h:65
struct ogg_stream * streams
Definition: oggdec.h:97
int segp
Definition: oggdec.h:79
#define AV_RL32
Definition: intreadwrite.h:146
AVDictionary * metadata
Definition: avformat.h:699
Stream structure.
Definition: avformat.h:622
enum AVMediaType codec_type
Definition: avcodec.h:1347
unsigned int pduration
Definition: oggdec.h:68
int nsegs
Definition: oggdec.h:79
enum AVCodecID codec_id
Definition: avcodec.h:1350
int sample_rate
samples per second
Definition: avcodec.h:2104
static int speex_packet(AVFormatContext *s, int idx)
Definition: oggparsespeex.c:98
static int speex_header(AVFormatContext *s, int idx)
Definition: oggparsespeex.c:41
int extradata_size
Definition: avcodec.h:1455
void * private
Definition: oggdec.h:85
byte swapping routines
int64_t lastdts
Definition: oggdec.h:73
const int8_t * magic
Definition: oggdec.h:32
uint8_t * buf
Definition: oggdec.h:62
Main libavformat public API header.
Definition: oggdec.h:96
int channels
number of audio channels
Definition: avcodec.h:2105
void * priv_data
Format private data.
Definition: avformat.h:848
#define AV_CH_LAYOUT_MONO
void * av_mallocz(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:158