FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
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  if (os->psize < 68) {
62  av_log(s, AV_LOG_ERROR, "speex packet too small\n");
63  return AVERROR_INVALIDDATA;
64  }
65 
66  st->codec->sample_rate = AV_RL32(p + 36);
67  st->codec->channels = AV_RL32(p + 48);
68  if (st->codec->channels < 1 || st->codec->channels > 2) {
69  av_log(s, AV_LOG_ERROR, "invalid channel count. Speex must be mono or stereo.\n");
70  return AVERROR_INVALIDDATA;
71  }
74 
75  spxp->packet_size = AV_RL32(p + 56);
76  frames_per_packet = AV_RL32(p + 64);
77  if (frames_per_packet)
78  spxp->packet_size *= frames_per_packet;
79 
80  if (ff_alloc_extradata(st->codec, os->psize) < 0)
81  return AVERROR(ENOMEM);
82  memcpy(st->codec->extradata, p, st->codec->extradata_size);
83 
84  avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
85  } else
86  ff_vorbis_stream_comment(s, st, p, os->psize);
87 
88  spxp->seq++;
89  return 1;
90 }
91 
92 static int ogg_page_packets(struct ogg_stream *os)
93 {
94  int i;
95  int packets = 0;
96  for (i = 0; i < os->nsegs; i++)
97  if (os->segments[i] < 255)
98  packets++;
99  return packets;
100 }
101 
102 static int speex_packet(AVFormatContext *s, int idx)
103 {
104  struct ogg *ogg = s->priv_data;
105  struct ogg_stream *os = ogg->streams + idx;
106  struct speex_params *spxp = os->private;
107  int packet_size = spxp->packet_size;
108 
109  if (os->flags & OGG_FLAG_EOS && os->lastpts != AV_NOPTS_VALUE &&
110  os->granule > 0) {
111  /* first packet of final page. we have to calculate the final packet
112  duration here because it is the only place we know the next-to-last
113  granule position. */
114  spxp->final_packet_duration = os->granule - os->lastpts -
115  packet_size * (ogg_page_packets(os) - 1);
116  }
117 
118  if (!os->lastpts && os->granule > 0)
119  /* first packet */
120  os->lastpts = os->lastdts = os->granule - packet_size *
121  ogg_page_packets(os);
122  if (os->flags & OGG_FLAG_EOS && os->segp == os->nsegs &&
123  spxp->final_packet_duration)
124  /* final packet */
125  os->pduration = spxp->final_packet_duration;
126  else
127  os->pduration = packet_size;
128 
129  return 0;
130 }
131 
132 const struct ogg_codec ff_speex_codec = {
133  .magic = "Speex ",
134  .magicsize = 8,
135  .header = speex_header,
136  .packet = speex_packet,
137  .nb_header = 2,
138 };
const char * s
Definition: avisynth_c.h:631
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
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:92
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:4006
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:1272
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:1355
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1340
bitstream reader API header.
#define av_log(a,...)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define AVERROR(e)
Definition: error.h:43
int ff_vorbis_stream_comment(AVFormatContext *as, AVStream *st, const uint8_t *buf, int size)
#define OGG_FLAG_EOS
Definition: oggdec.h:112
uint8_t segments[255]
Definition: oggdec.h:80
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:2046
AVCodecContext * codec
Codec context associated with this stream.
Definition: avformat.h:861
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:102
int segp
Definition: oggdec.h:79
Stream structure.
Definition: avformat.h:842
enum AVMediaType codec_type
Definition: avcodec.h:1249
unsigned int pduration
Definition: oggdec.h:68
int nsegs
Definition: oggdec.h:79
enum AVCodecID codec_id
Definition: avcodec.h:1258
int sample_rate
samples per second
Definition: avcodec.h:1985
static int speex_packet(AVFormatContext *s, int idx)
static int speex_header(AVFormatContext *s, int idx)
Definition: oggparsespeex.c:41
int ff_alloc_extradata(AVCodecContext *avctx, int size)
Allocate extradata with additional FF_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0...
Definition: utils.c:2864
int extradata_size
Definition: avcodec.h:1356
void * private
Definition: oggdec.h:90
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:101
int channels
number of audio channels
Definition: avcodec.h:1986
void * priv_data
Format private data.
Definition: avformat.h:1300
#define AV_CH_LAYOUT_MONO
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:85
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:250
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:241