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"
28 #include "libavutil/channel_layout.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 };
#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:2821
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:718
void av_log(void *avcl, int level, const char *fmt,...) av_printf_format(3
Send the specified message to the log if the level is less than or equal to the current av_log_level...
int flags
Definition: oggdec.h:76
#define AV_CH_LAYOUT_STEREO
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1164
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
bitstream reader API header.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:123
void * priv_data
Format private data.
Definition: avformat.h:950
int ff_vorbis_stream_comment(AVFormatContext *as, AVStream *st, const uint8_t *buf, int size)
#define OGG_FLAG_EOS
Definition: oggdec.h:106
uint8_t segments[255]
Definition: oggdec.h:80
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:1852
#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
uint64_t granule
Definition: oggdec.h:70
unsigned int pstart
Definition: oggdec.h:65
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:990
void * av_malloc(size_t size) av_malloc_attrib 1(1)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:62
struct ogg_stream * streams
Definition: oggdec.h:97
int segp
Definition: oggdec.h:79
Stream structure.
Definition: avformat.h:699
enum AVMediaType codec_type
Definition: avcodec.h:1058
unsigned int pduration
Definition: oggdec.h:68
int nsegs
Definition: oggdec.h:79
enum AVCodecID codec_id
Definition: avcodec.h:1067
int sample_rate
samples per second
Definition: avcodec.h:1791
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:1165
void * private
Definition: oggdec.h:85
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
#define AV_RL32(x)
Definition: intreadwrite.h:248
int channels
number of audio channels
Definition: avcodec.h:1792
#define AV_CH_LAYOUT_MONO
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:228
void * av_mallocz(size_t size) av_malloc_attrib 1(1)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:205