FFmpeg
vpk.c
Go to the documentation of this file.
1 /*
2  * VPK demuxer
3  * Copyright (c) 2015 Paul B Mahol
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 "avformat.h"
24 #include "internal.h"
25 
26 typedef struct VPKDemuxContext {
27  unsigned block_count;
28  unsigned current_block;
29  unsigned last_block_size;
31 
32 static int vpk_probe(const AVProbeData *p)
33 {
34  if (AV_RL32(p->buf) != MKBETAG('V','P','K',' '))
35  return 0;
36 
37  return AVPROBE_SCORE_MAX / 3 * 2;
38 }
39 
41 {
42  VPKDemuxContext *vpk = s->priv_data;
43  unsigned offset;
44  unsigned samples_per_block;
45  AVStream *st;
46 
47  vpk->current_block = 0;
48  st = avformat_new_stream(s, NULL);
49  if (!st)
50  return AVERROR(ENOMEM);
51 
52  avio_skip(s->pb, 4);
53  st->duration = avio_rl32(s->pb) * 28 / 16;
54  offset = avio_rl32(s->pb);
57  st->codecpar->block_align = avio_rl32(s->pb);
58  st->codecpar->sample_rate = avio_rl32(s->pb);
59  if (st->codecpar->sample_rate <= 0 || st->codecpar->block_align <= 0)
60  return AVERROR_INVALIDDATA;
61  st->codecpar->channels = avio_rl32(s->pb);
62  if (st->codecpar->channels <= 0)
63  return AVERROR_INVALIDDATA;
64  samples_per_block = ((st->codecpar->block_align / st->codecpar->channels) * 28LL) / 16;
65  if (samples_per_block <= 0)
66  return AVERROR_INVALIDDATA;
67  vpk->block_count = (st->duration + (samples_per_block - 1)) / samples_per_block;
68  vpk->last_block_size = (st->duration % samples_per_block) * 16 * st->codecpar->channels / 28;
69 
70  if (offset < avio_tell(s->pb))
71  return AVERROR_INVALIDDATA;
72  avio_skip(s->pb, offset - avio_tell(s->pb));
73  avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
74 
75  return 0;
76 }
77 
79 {
80  AVCodecParameters *par = s->streams[0]->codecpar;
81  VPKDemuxContext *vpk = s->priv_data;
82  int ret, i;
83 
84  vpk->current_block++;
85  if (vpk->current_block == vpk->block_count) {
86  unsigned size = vpk->last_block_size / par->channels;
87  unsigned skip = (par->block_align - vpk->last_block_size) / par->channels;
88 
89  ret = av_new_packet(pkt, vpk->last_block_size);
90  if (ret < 0)
91  return ret;
92  for (i = 0; i < par->channels; i++) {
93  ret = avio_read(s->pb, pkt->data + i * size, size);
94  avio_skip(s->pb, skip);
95  if (ret != size) {
96  av_packet_unref(pkt);
97  ret = AVERROR(EIO);
98  break;
99  }
100  }
101  pkt->stream_index = 0;
102  } else if (vpk->current_block < vpk->block_count) {
103  ret = av_get_packet(s->pb, pkt, par->block_align);
104  pkt->stream_index = 0;
105  } else {
106  return AVERROR_EOF;
107  }
108 
109  return ret;
110 }
111 
113  .name = "vpk",
114  .long_name = NULL_IF_CONFIG_SMALL("Sony PS2 VPK"),
115  .priv_data_size = sizeof(VPKDemuxContext),
119  .extensions = "vpk",
120 };
#define NULL
Definition: coverity.c:32
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
unsigned block_count
Definition: vpk.c:27
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:4896
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3960
static int vpk_read_header(AVFormatContext *s)
Definition: vpk.c:40
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:334
static AVPacket pkt
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3952
Format I/O context.
Definition: avformat.h:1358
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf offset
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4469
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1426
uint8_t * data
Definition: avcodec.h:1480
#define AVERROR_EOF
End of file.
Definition: error.h:55
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:310
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
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:650
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:86
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:259
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:772
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
enum AVMediaType codec_type
General type of the encoded data.
Definition: avcodec.h:3956
static int vpk_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: vpk.c:78
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:448
int block_align
Audio only.
Definition: avcodec.h:4077
#define s(width, name)
Definition: cbs_vp9.c:257
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:530
if(ret)
Stream structure.
Definition: avformat.h:881
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
AVIOContext * pb
I/O context.
Definition: avformat.h:1400
AVInputFormat ff_vpk_demuxer
Definition: vpk.c:112
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:599
unsigned last_block_size
Definition: vpk.c:29
static int vpk_probe(const AVProbeData *p)
Definition: vpk.c:32
This structure contains the data a format has to probe a file.
Definition: avformat.h:446
static int read_probe(const AVProbeData *pd)
Definition: jvdec.c:55
int64_t duration
Decoding: duration of the stream, in stream time base.
Definition: avformat.h:930
int sample_rate
Audio only.
Definition: avcodec.h:4070
#define AVPROBE_SCORE_MAX
maximum score
Definition: avformat.h:458
Main libavformat public API header.
#define MKBETAG(a, b, c, d)
Definition: common.h:367
unsigned current_block
Definition: vpk.c:28
void * priv_data
Format private data.
Definition: avformat.h:1386
int channels
Audio only.
Definition: avcodec.h:4066
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:654
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1028
int stream_index
Definition: avcodec.h:1482
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
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
This structure stores compressed data.
Definition: avcodec.h:1457