FFmpeg
r3d.c
Go to the documentation of this file.
1 /*
2  * R3D REDCODE demuxer
3  * Copyright (c) 2008 Baptiste Coudurier <baptiste dot coudurier at gmail dot com>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/intreadwrite.h"
23 #include "libavutil/dict.h"
24 #include "libavutil/mathematics.h"
25 #include "avformat.h"
26 #include "internal.h"
27 
28 typedef struct R3DContext {
30  unsigned rdvo_offset;
31 
33 } R3DContext;
34 
35 typedef struct Atom {
36  unsigned size;
37  uint32_t tag;
38  uint64_t offset;
39 } Atom;
40 
41 static int read_atom(AVFormatContext *s, Atom *atom)
42 {
43  atom->offset = avio_tell(s->pb);
44  atom->size = avio_rb32(s->pb);
45  if (atom->size < 8)
46  return -1;
47  atom->tag = avio_rl32(s->pb);
48  av_log(s, AV_LOG_TRACE, "atom %u %.4s offset %#"PRIx64"\n",
49  atom->size, (char*)&atom->tag, atom->offset);
50  return atom->size;
51 }
52 
54 {
56  R3DContext *r3d = s->priv_data;
57  char filename[258];
58  int tmp;
59  int av_unused tmp2;
61 
62  if (!st)
63  return AVERROR(ENOMEM);
66 
67  tmp = avio_r8(s->pb); // major version
68  tmp2 = avio_r8(s->pb); // minor version
69  av_log(s, AV_LOG_TRACE, "version %d.%d\n", tmp, tmp2);
70 
71  tmp = avio_rb16(s->pb); // unknown
72  av_log(s, AV_LOG_TRACE, "unknown1 %d\n", tmp);
73 
74  tmp = avio_rb32(s->pb);
75  avpriv_set_pts_info(st, 32, 1, tmp);
76 
77  tmp = avio_rb32(s->pb); // filenum
78  av_log(s, AV_LOG_TRACE, "filenum %d\n", tmp);
79 
80  avio_skip(s->pb, 32); // unknown
81 
82  st->codecpar->width = avio_rb32(s->pb);
83  st->codecpar->height = avio_rb32(s->pb);
84 
85  tmp = avio_rb16(s->pb); // unknown
86  av_log(s, AV_LOG_TRACE, "unknown2 %d\n", tmp);
87 
88  framerate.num = avio_rb16(s->pb);
89  framerate.den = avio_rb16(s->pb);
90  if (framerate.num > 0 && framerate.den > 0) {
91 #if FF_API_R_FRAME_RATE
92  st->r_frame_rate =
93 #endif
95  }
96 
97  r3d->audio_channels = avio_r8(s->pb); // audio channels
98  av_log(s, AV_LOG_TRACE, "audio channels %d\n", tmp);
99 
100  avio_read(s->pb, filename, 257);
101  filename[sizeof(filename)-1] = 0;
102  av_dict_set(&st->metadata, "filename", filename, 0);
103 
104  av_log(s, AV_LOG_TRACE, "filename %s\n", filename);
105  av_log(s, AV_LOG_TRACE, "resolution %dx%d\n", st->codecpar->width, st->codecpar->height);
106  av_log(s, AV_LOG_TRACE, "timescale %d\n", st->time_base.den);
107  av_log(s, AV_LOG_TRACE, "frame rate %d/%d\n",
108  framerate.num, framerate.den);
109 
110  return 0;
111 }
112 
113 static int r3d_read_rdvo(AVFormatContext *s, Atom *atom)
114 {
115  R3DContext *r3d = s->priv_data;
116  AVStream *st = s->streams[0];
117  int i;
118 
119  r3d->video_offsets_count = (atom->size - 8) / 4;
120 
121  for (i = 0; i < r3d->video_offsets_count; i++) {
122  unsigned video_offset = avio_rb32(s->pb);
123  if (!video_offset) {
124  r3d->video_offsets_count = i;
125  break;
126  }
127  av_log(s, AV_LOG_TRACE, "video offset %d: %#x\n", i, video_offset);
128  }
129 
130  if (st->avg_frame_rate.num)
133  st->time_base);
134  av_log(s, AV_LOG_TRACE, "duration %"PRId64"\n", st->duration);
135 
136  return 0;
137 }
138 
140 {
141  R3DContext *r3d = s->priv_data;
142  int av_unused tmp;
143 
144  r3d->rdvo_offset = avio_rb32(s->pb);
145  avio_rb32(s->pb); // rdvs offset
146  avio_rb32(s->pb); // rdao offset
147  avio_rb32(s->pb); // rdas offset
148 
149  tmp = avio_rb32(s->pb);
150  av_log(s, AV_LOG_TRACE, "num video chunks %d\n", tmp);
151 
152  tmp = avio_rb32(s->pb);
153  av_log(s, AV_LOG_TRACE, "num audio chunks %d\n", tmp);
154 
155  avio_skip(s->pb, 6*4);
156 }
157 
159 {
160  R3DContext *r3d = s->priv_data;
161  Atom atom;
162  int ret;
163 
164  if (read_atom(s, &atom) < 0) {
165  av_log(s, AV_LOG_ERROR, "error reading atom\n");
166  return -1;
167  }
168  if (atom.tag == MKTAG('R','E','D','1')) {
169  if ((ret = r3d_read_red1(s)) < 0) {
170  av_log(s, AV_LOG_ERROR, "error parsing 'red1' atom\n");
171  return ret;
172  }
173  } else {
174  av_log(s, AV_LOG_ERROR, "could not find 'red1' atom\n");
175  return -1;
176  }
177 
178  /* we cannot create the audio stream now because we do not know the
179  * sample rate */
180  if (r3d->audio_channels)
182 
183  s->internal->data_offset = avio_tell(s->pb);
184  av_log(s, AV_LOG_TRACE, "data offset %#"PRIx64"\n", s->internal->data_offset);
185  if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL))
186  return 0;
187  // find REOB/REOF/REOS to load index
188  avio_seek(s->pb, avio_size(s->pb)-48-8, SEEK_SET);
189  if (read_atom(s, &atom) < 0)
190  av_log(s, AV_LOG_ERROR, "error reading end atom\n");
191 
192  if (atom.tag != MKTAG('R','E','O','B') &&
193  atom.tag != MKTAG('R','E','O','F') &&
194  atom.tag != MKTAG('R','E','O','S'))
195  goto out;
196 
197  r3d_read_reos(s);
198 
199  if (r3d->rdvo_offset) {
200  avio_seek(s->pb, r3d->rdvo_offset, SEEK_SET);
201  if (read_atom(s, &atom) < 0)
202  av_log(s, AV_LOG_ERROR, "error reading 'rdvo' atom\n");
203  if (atom.tag == MKTAG('R','D','V','O')) {
204  if (r3d_read_rdvo(s, &atom) < 0)
205  av_log(s, AV_LOG_ERROR, "error parsing 'rdvo' atom\n");
206  }
207  }
208 
209  out:
210  avio_seek(s->pb, s->internal->data_offset, SEEK_SET);
211  return 0;
212 }
213 
215 {
216  AVStream *st = s->streams[0];
217  int tmp;
218  int av_unused tmp2;
219  int64_t pos = avio_tell(s->pb);
220  unsigned dts;
221  int ret;
222 
223  dts = avio_rb32(s->pb);
224 
225  tmp = avio_rb32(s->pb);
226  av_log(s, AV_LOG_TRACE, "frame num %d\n", tmp);
227 
228  tmp = avio_r8(s->pb); // major version
229  tmp2 = avio_r8(s->pb); // minor version
230  av_log(s, AV_LOG_TRACE, "version %d.%d\n", tmp, tmp2);
231 
232  tmp = avio_rb16(s->pb); // unknown
233  av_log(s, AV_LOG_TRACE, "unknown %d\n", tmp);
234 
235  if (tmp > 4) {
236  tmp = avio_rb16(s->pb); // unknown
237  av_log(s, AV_LOG_TRACE, "unknown %d\n", tmp);
238 
239  tmp = avio_rb16(s->pb); // unknown
240  av_log(s, AV_LOG_TRACE, "unknown %d\n", tmp);
241 
242  tmp = avio_rb32(s->pb);
243  av_log(s, AV_LOG_TRACE, "width %d\n", tmp);
244  tmp = avio_rb32(s->pb);
245  av_log(s, AV_LOG_TRACE, "height %d\n", tmp);
246 
247  tmp = avio_rb32(s->pb);
248  av_log(s, AV_LOG_TRACE, "metadata len %d\n", tmp);
249  }
250  tmp = atom->size - 8 - (avio_tell(s->pb) - pos);
251  if (tmp < 0)
252  return -1;
253  ret = av_get_packet(s->pb, pkt, tmp);
254  if (ret < 0) {
255  av_log(s, AV_LOG_ERROR, "error reading video packet\n");
256  return -1;
257  }
258 
259  pkt->stream_index = 0;
260  pkt->dts = dts;
261  if (st->avg_frame_rate.num)
262  pkt->duration = (uint64_t)st->time_base.den*
264  av_log(s, AV_LOG_TRACE, "pkt dts %"PRId64" duration %"PRId64"\n", pkt->dts, pkt->duration);
265 
266  return 0;
267 }
268 
270 {
271  R3DContext *r3d = s->priv_data;
272  AVStream *st;
273  int av_unused tmp, tmp2;
274  int samples, size;
275  int64_t pos = avio_tell(s->pb);
276  unsigned dts;
277  int ret;
278 
279  if (s->nb_streams < 2) {
280  st = avformat_new_stream(s, NULL);
281  if (!st)
282  return AVERROR(ENOMEM);
285  st->codecpar->channels = r3d->audio_channels;
286  avpriv_set_pts_info(st, 32, 1, s->streams[0]->time_base.den);
287  } else {
288  st = s->streams[1];
289  }
290 
291  dts = avio_rb32(s->pb);
292 
293  st->codecpar->sample_rate = avio_rb32(s->pb);
294  if (st->codecpar->sample_rate <= 0) {
295  av_log(s, AV_LOG_ERROR, "Bad sample rate\n");
296  return AVERROR_INVALIDDATA;
297  }
298 
299  samples = avio_rb32(s->pb);
300 
301  tmp = avio_rb32(s->pb);
302  av_log(s, AV_LOG_TRACE, "packet num %d\n", tmp);
303 
304  tmp = avio_rb16(s->pb); // unknown
305  av_log(s, AV_LOG_TRACE, "unknown %d\n", tmp);
306 
307  tmp = avio_r8(s->pb); // major version
308  tmp2 = avio_r8(s->pb); // minor version
309  av_log(s, AV_LOG_TRACE, "version %d.%d\n", tmp, tmp2);
310 
311  tmp = avio_rb32(s->pb); // unknown
312  av_log(s, AV_LOG_TRACE, "unknown %d\n", tmp);
313 
314  size = atom->size - 8 - (avio_tell(s->pb) - pos);
315  if (size < 0)
316  return -1;
317  ret = av_get_packet(s->pb, pkt, size);
318  if (ret < 0) {
319  av_log(s, AV_LOG_ERROR, "error reading audio packet\n");
320  return ret;
321  }
322 
323  pkt->stream_index = 1;
324  pkt->dts = dts;
325  if (st->codecpar->sample_rate)
326  pkt->duration = av_rescale(samples, st->time_base.den, st->codecpar->sample_rate);
327  av_log(s, AV_LOG_TRACE, "pkt dts %"PRId64" duration %"PRId64" samples %d sample rate %d\n",
328  pkt->dts, pkt->duration, samples, st->codecpar->sample_rate);
329 
330  return 0;
331 }
332 
334 {
335  R3DContext *r3d = s->priv_data;
336  Atom atom;
337  int err = 0;
338 
339  while (!err) {
340  if (read_atom(s, &atom) < 0) {
341  err = -1;
342  break;
343  }
344  switch (atom.tag) {
345  case MKTAG('R','E','D','V'):
346  if (s->streams[0]->discard == AVDISCARD_ALL)
347  goto skip;
348  if (!(err = r3d_read_redv(s, pkt, &atom)))
349  return 0;
350  break;
351  case MKTAG('R','E','D','A'):
352  if (!r3d->audio_channels)
353  return -1;
354  if (s->nb_streams >= 2 && s->streams[1]->discard == AVDISCARD_ALL)
355  goto skip;
356  if (!(err = r3d_read_reda(s, pkt, &atom)))
357  return 0;
358  break;
359  default:
360  skip:
361  avio_skip(s->pb, atom.size-8);
362  }
363  }
364  return err;
365 }
366 
367 static int r3d_probe(const AVProbeData *p)
368 {
369  if (AV_RL32(p->buf + 4) == MKTAG('R','E','D','1'))
370  return AVPROBE_SCORE_MAX;
371  return 0;
372 }
373 
374 static int r3d_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
375 {
376  AVStream *st = s->streams[0]; // video stream
377  R3DContext *r3d = s->priv_data;
378  int frame_num;
379 
380  if (!st->avg_frame_rate.num)
381  return -1;
382 
383  frame_num = av_rescale_q(sample_time, st->time_base,
384  av_inv_q(st->avg_frame_rate));
385  av_log(s, AV_LOG_TRACE, "seek frame num %d timestamp %"PRId64"\n",
386  frame_num, sample_time);
387 
388  if (frame_num < r3d->video_offsets_count) {
389  if (avio_seek(s->pb, r3d->video_offsets_count, SEEK_SET) < 0)
390  return -1;
391  } else {
392  av_log(s, AV_LOG_ERROR, "could not seek to frame %d\n", frame_num);
393  return -1;
394  }
395 
396  return 0;
397 }
398 
400  .name = "r3d",
401  .long_name = NULL_IF_CONFIG_SMALL("REDCODE R3D"),
402  .priv_data_size = sizeof(R3DContext),
406  .read_seek = r3d_seek,
407 };
#define NULL
Definition: coverity.c:32
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
uint64_t offset
Definition: r3d.c:38
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:346
uint32_t tag
Definition: r3d.c:37
unsigned video_offsets_count
Definition: r3d.c:29
static int read_atom(AVFormatContext *s, Atom *atom)
Definition: r3d.c:41
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:4938
int64_t data_offset
offset of the first packet
Definition: internal.h:80
static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)
Definition: libcdio.c:153
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
int num
Numerator.
Definition: rational.h:59
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:253
unsigned size
Definition: r3d.c:36
AVFormatInternal * internal
An opaque field for libavformat internal usage.
Definition: avformat.h:1804
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:341
discard all
Definition: avcodec.h:236
static AVPacket pkt
unsigned int avio_rb16(AVIOContext *s)
Definition: aviobuf.c:775
int ctx_flags
Flags signalling stream properties.
Definition: avformat.h:1400
int framerate
Definition: h264_levels.c:65
Format I/O context.
Definition: avformat.h:1351
Public dictionary API.
int width
Video only.
Definition: codec_par.h:126
#define AVFMTCTX_NOHEADER
signal that no header is present (streams are added dynamically)
Definition: avformat.h:1295
static int r3d_probe(const AVProbeData *p)
Definition: r3d.c:367
#define AV_LOG_TRACE
Extremely verbose debugging, useful for libav* development.
Definition: log.h:220
unsigned int avio_rb32(AVIOContext *s)
Definition: aviobuf.c:790
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:381
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4519
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1419
static int r3d_read_reda(AVFormatContext *s, AVPacket *pkt, Atom *atom)
Definition: r3d.c:269
int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
Allocate and read the payload of a packet and initialize its fields with default values.
Definition: utils.c:304
ptrdiff_t size
Definition: opengl_enc.c:100
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
#define av_log(a,...)
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:637
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
Definition: mathematics.c:142
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:759
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:153
unsigned int pos
Definition: spdifenc.c:410
static void r3d_read_reos(AVFormatContext *s)
Definition: r3d.c:139
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:56
Definition: r3d.c:35
int audio_channels
Definition: r3d.c:32
AVRational avg_frame_rate
Average framerate.
Definition: avformat.h:949
Definition: r3d.c:28
unsigned rdvo_offset
Definition: r3d.c:30
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:628
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:443
unsigned int nb_streams
Number of elements in AVFormatContext.streams.
Definition: avformat.h:1407
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:260
AVInputFormat ff_r3d_demuxer
Definition: r3d.c:399
int64_t av_rescale(int64_t a, int64_t b, int64_t c)
Rescale a 64-bit integer with rounding to nearest.
Definition: mathematics.c:129
#define s(width, name)
Definition: cbs_vp9.c:257
AVDictionary * metadata
Definition: avformat.h:940
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:527
Stream structure.
Definition: avformat.h:876
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:40
AVIOContext * pb
I/O context.
Definition: avformat.h:1393
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:70
Rational number (pair of numerator and denominator).
Definition: rational.h:58
static int r3d_read_rdvo(AVFormatContext *s, Atom *atom)
Definition: r3d.c:113
This structure contains the data a format has to probe a file.
Definition: avformat.h:441
#define flags(name, subs,...)
Definition: cbs_av1.c:560
static int read_probe(const AVProbeData *pd)
Definition: jvdec.c:55
static int r3d_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
Definition: r3d.c:374
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:925
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
Definition: rational.h:159
int sample_rate
Audio only.
Definition: codec_par.h:170
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:453
Main libavformat public API header.
static int r3d_read_header(AVFormatContext *s)
Definition: r3d.c:158
static int r3d_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: r3d.c:333
int den
Denominator.
Definition: rational.h:60
void * priv_data
Format private data.
Definition: avformat.h:1379
static int r3d_read_redv(AVFormatContext *s, AVPacket *pkt, Atom *atom)
Definition: r3d.c:214
int channels
Audio only.
Definition: codec_par.h:166
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed...
Definition: packet.h:362
FILE * out
Definition: movenc.c:54
static int r3d_read_red1(AVFormatContext *s)
Definition: r3d.c:53
Filter the word “frame” indicates either a video frame or a group of audio samples
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:650
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1023
int stream_index
Definition: packet.h:365
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
Definition: avformat.h:905
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
#define MKTAG(a, b, c, d)
Definition: common.h:406
enum AVDiscard discard
Selects which packets can be discarded at will and do not need to be demuxed.
Definition: avformat.h:931
AVRational r_frame_rate
Real base framerate of the stream.
Definition: avformat.h:1000
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
This structure stores compressed data.
Definition: packet.h:340
int i
Definition: input.c:407
#define av_unused
Definition: attributes.h:131
static uint8_t tmp[11]
Definition: aes_ctr.c:26