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_stream_comment(s, st, 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:62
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:54
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:2800
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:922
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:1158
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:990
bitstream reader API header.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:123
int ff_vorbis_stream_comment(AVFormatContext *as, AVStream *st, 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:168
uint8_t segments[255]
Definition: oggdec.h:80
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:1846
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:718
#define FF_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
Definition: avcodec.h:531
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
Stream structure.
Definition: avformat.h:699
enum AVMediaType codec_type
Definition: avcodec.h:1052
unsigned int pduration
Definition: oggdec.h:68
int nsegs
Definition: oggdec.h:79
enum AVCodecID codec_id
Definition: avcodec.h:1061
int sample_rate
samples per second
Definition: avcodec.h:1785
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:1159
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:1786
void * priv_data
Format private data.
Definition: avformat.h:950
#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:205
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:228