blob: bb9eabb2ba25827153f5b44bf0f0f386b9e8453b
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; |
30 | } VPKDemuxContext; |
31 | |
32 | static int vpk_probe(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 | |
40 | static int vpk_read_header(AVFormatContext *s) |
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); |
55 | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
56 | st->codecpar->codec_id = AV_CODEC_ID_ADPCM_PSX; |
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) |
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) * 28) / 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 | avio_skip(s->pb, offset - avio_tell(s->pb)); |
70 | avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); |
71 | |
72 | return 0; |
73 | } |
74 | |
75 | static int vpk_read_packet(AVFormatContext *s, AVPacket *pkt) |
76 | { |
77 | AVCodecParameters *par = s->streams[0]->codecpar; |
78 | VPKDemuxContext *vpk = s->priv_data; |
79 | int ret, i; |
80 | |
81 | vpk->current_block++; |
82 | if (vpk->current_block == vpk->block_count) { |
83 | unsigned size = vpk->last_block_size / par->channels; |
84 | unsigned skip = (par->block_align - vpk->last_block_size) / par->channels; |
85 | |
86 | ret = av_new_packet(pkt, vpk->last_block_size); |
87 | if (ret < 0) |
88 | return ret; |
89 | for (i = 0; i < par->channels; i++) { |
90 | ret = avio_read(s->pb, pkt->data + i * size, size); |
91 | avio_skip(s->pb, skip); |
92 | if (ret != size) { |
93 | av_packet_unref(pkt); |
94 | ret = AVERROR(EIO); |
95 | break; |
96 | } |
97 | } |
98 | pkt->stream_index = 0; |
99 | } else if (vpk->current_block < vpk->block_count) { |
100 | ret = av_get_packet(s->pb, pkt, par->block_align); |
101 | pkt->stream_index = 0; |
102 | } else { |
103 | return AVERROR_EOF; |
104 | } |
105 | |
106 | return ret; |
107 | } |
108 | |
109 | AVInputFormat ff_vpk_demuxer = { |
110 | .name = "vpk", |
111 | .long_name = NULL_IF_CONFIG_SMALL("Sony PS2 VPK"), |
112 | .priv_data_size = sizeof(VPKDemuxContext), |
113 | .read_probe = vpk_probe, |
114 | .read_header = vpk_read_header, |
115 | .read_packet = vpk_read_packet, |
116 | .extensions = "vpk", |
117 | }; |
118 |