FFmpeg
oggparsedirac.c
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2008 David Conrad
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #include "libavutil/imgutils.h"
22 #include "libavutil/intreadwrite.h"
23 #include "libavcodec/dirac.h"
24 #include "avformat.h"
25 #include "internal.h"
26 #include "oggdec.h"
27 
28 static int dirac_header(AVFormatContext *s, int idx)
29 {
30  struct ogg *ogg = s->priv_data;
31  struct ogg_stream *os = ogg->streams + idx;
32  AVStream *st = s->streams[idx];
33  AVDiracSeqHeader *dsh;
34  int ret;
35 
36  // already parsed the header
38  return 0;
39 
40  ret = av_dirac_parse_sequence_header(&dsh, os->buf + os->pstart + 13, (os->psize - 13), s);
41  if (ret < 0)
42  return ret;
43 
46  st->codecpar->width = dsh->width;
47  st->codecpar->height = dsh->height;
48  st->codecpar->format = dsh->pix_fmt;
49  st->codecpar->color_range = dsh->color_range;
50  st->codecpar->color_trc = dsh->color_trc;
52  st->codecpar->color_space = dsh->colorspace;
53  st->codecpar->profile = dsh->profile;
54  st->codecpar->level = dsh->level;
57 
58  // Dirac in Ogg always stores timestamps as though the video were interlaced
59  avpriv_set_pts_info(st, 64, dsh->framerate.den, 2 * dsh->framerate.num);
60 
61  av_freep(&dsh);
62  return 1;
63 }
64 
65 // various undocumented things: granule is signed (only for Dirac!)
66 static uint64_t dirac_gptopts(AVFormatContext *s, int idx, uint64_t granule,
67  int64_t *dts_out)
68 {
69  int64_t gp = granule;
70  struct ogg *ogg = s->priv_data;
71  struct ogg_stream *os = ogg->streams + idx;
72 
73  unsigned dist = ((gp >> 14) & 0xff00) | (gp & 0xff);
74  int64_t dts = (gp >> 31);
75  int64_t pts = dts + ((gp >> 9) & 0x1fff);
76 
77  if (!dist)
78  os->pflags |= AV_PKT_FLAG_KEY;
79 
80  if (dts_out)
81  *dts_out = dts;
82 
83  return pts;
84 }
85 
86 static int old_dirac_header(AVFormatContext *s, int idx)
87 {
88  struct ogg *ogg = s->priv_data;
89  struct ogg_stream *os = ogg->streams + idx;
90  AVStream *st = s->streams[idx];
91  uint8_t *buf = os->buf + os->pstart;
92 
93  if (buf[0] != 'K')
94  return 0;
95 
98  avpriv_set_pts_info(st, 64, AV_RB32(buf+12), AV_RB32(buf+8));
99  return 1;
100 }
101 
102 static uint64_t old_dirac_gptopts(AVFormatContext *s, int idx, uint64_t gp,
103  int64_t *dts)
104 {
105  struct ogg *ogg = s->priv_data;
106  struct ogg_stream *os = ogg->streams + idx;
107  uint64_t iframe = gp >> 30;
108  uint64_t pframe = gp & 0x3fffffff;
109 
110  if (!pframe)
111  os->pflags |= AV_PKT_FLAG_KEY;
112 
113  return iframe + pframe;
114 }
115 
116 const struct ogg_codec ff_dirac_codec = {
117  .magic = "BBCD\0",
118  .magicsize = 5,
119  .header = dirac_header,
120  .gptopts = dirac_gptopts,
121  .granule_is_start = 1,
122  .nb_header = 1,
123 };
124 
126  .magic = "KW-DIRAC",
127  .magicsize = 8,
128  .header = old_dirac_header,
129  .gptopts = old_dirac_gptopts,
130  .granule_is_start = 1,
131  .nb_header = 1,
132 };
av_dirac_parse_sequence_header
int av_dirac_parse_sequence_header(AVDiracSeqHeader **pdsh, const uint8_t *buf, size_t buf_size, void *log_ctx)
Parse a Dirac sequence header.
Definition: dirac.c:403
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:51
AVCodecParameters::color_space
enum AVColorSpace color_space
Definition: codec_par.h:169
old_dirac_gptopts
static uint64_t old_dirac_gptopts(AVFormatContext *s, int idx, uint64_t gp, int64_t *dts)
Definition: oggparsedirac.c:102
AV_CODEC_ID_DIRAC
@ AV_CODEC_ID_DIRAC
Definition: codec_id.h:168
int64_t
long long int64_t
Definition: coverity.c:34
AVDiracSeqHeader::color_trc
enum AVColorTransferCharacteristic color_trc
Definition: dirac.h:113
ogg_stream::granule
uint64_t granule
Definition: oggdec.h:69
dirac_gptopts
static uint64_t dirac_gptopts(AVFormatContext *s, int idx, uint64_t granule, int64_t *dts_out)
Definition: oggparsedirac.c:66
ogg_stream::buf
uint8_t * buf
Definition: oggdec.h:61
AVDiracSeqHeader::color_range
enum AVColorRange color_range
Definition: dirac.h:111
ogg
Definition: oggdec.h:101
AV_PKT_FLAG_KEY
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:577
AVDiracSeqHeader::sample_aspect_ratio
AVRational sample_aspect_ratio
Definition: dirac.h:108
AVCodecParameters::color_primaries
enum AVColorPrimaries color_primaries
Definition: codec_par.h:167
AVDiracSeqHeader::level
int level
Definition: dirac.h:105
avpriv_set_pts_info
void avpriv_set_pts_info(AVStream *st, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: avformat.c:853
pts
static int64_t pts
Definition: transcode_aac.c:643
AVRational::num
int num
Numerator.
Definition: rational.h:59
dirac.h
AVCodecParameters::color_trc
enum AVColorTransferCharacteristic color_trc
Definition: codec_par.h:168
ogg_stream::pstart
unsigned int pstart
Definition: oggdec.h:64
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:198
AVCodecParameters::width
int width
Video only.
Definition: codec_par.h:134
AVDiracSeqHeader::profile
int profile
Definition: dirac.h:104
old_dirac_header
static int old_dirac_header(AVFormatContext *s, int idx)
Definition: oggparsedirac.c:86
AVDiracSeqHeader::framerate
AVRational framerate
Definition: dirac.h:107
AVFormatContext
Format I/O context.
Definition: avformat.h:1255
internal.h
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:766
AVDiracSeqHeader::colorspace
enum AVColorSpace colorspace
Definition: dirac.h:114
ff_dirac_codec
const struct ogg_codec ff_dirac_codec
Definition: oggparsedirac.c:116
ogg::streams
struct ogg_stream * streams
Definition: oggdec.h:102
AVCodecParameters::level
int level
Definition: codec_par.h:129
gp
#define gp
Definition: regdef.h:62
AV_RB32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_RB32
Definition: bytestream.h:96
AVCodecParameters::profile
int profile
Codec-specific bitstream restrictions that the stream conforms to.
Definition: codec_par.h:128
AVStream::sample_aspect_ratio
AVRational sample_aspect_ratio
sample aspect ratio (0 if unknown)
Definition: avformat.h:821
ogg_stream::pflags
unsigned int pflags
Definition: oggdec.h:66
ogg_stream
Definition: oggdec.h:60
AVCodecParameters::height
int height
Definition: codec_par.h:135
AVCodecParameters::color_range
enum AVColorRange color_range
Video only.
Definition: codec_par.h:166
AVDiracSeqHeader
Definition: dirac.h:85
dirac_header
static int dirac_header(AVFormatContext *s, int idx)
Definition: oggparsedirac.c:28
ret
ret
Definition: filter_design.txt:187
AVStream
Stream structure.
Definition: avformat.h:743
avformat.h
AVDiracSeqHeader::pix_fmt
enum AVPixelFormat pix_fmt
Definition: dirac.h:110
ff_old_dirac_codec
const struct ogg_codec ff_old_dirac_codec
Definition: oggparsedirac.c:125
oggdec.h
AVRational::den
int den
Denominator.
Definition: rational.h:60
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVCodecParameters::format
int format
Definition: codec_par.h:92
AVDiracSeqHeader::width
unsigned width
Definition: dirac.h:86
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:55
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
ogg_stream::psize
unsigned int psize
Definition: oggdec.h:65
ogg_codec
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård.
Definition: oggdec.h:30
imgutils.h
AVDiracSeqHeader::height
unsigned height
Definition: dirac.h:87
ogg_codec::magic
const int8_t * magic
Definition: oggdec.h:31
av_image_check_sar
int av_image_check_sar(unsigned int w, unsigned int h, AVRational sar)
Check if the given sample aspect ratio of an image is valid.
Definition: imgutils.c:323
AVDiracSeqHeader::color_primaries
enum AVColorPrimaries color_primaries
Definition: dirac.h:112