path: root/libavformat/voc_packet.c (plain)
blob: 1e2e19e1c36b97b34b032e74f7c76348d21bacbe

1	/*
2	* Copyright (c) 2006 Aurelien Jacobs <aurel@gnuage.org>
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/intreadwrite.h"
22	#include "avformat.h"
23	#include "internal.h"
24	#include "voc.h"
25
26	int
27	ff_voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
28	{
29	VocDecContext *voc = s->priv_data;
30	AVCodecParameters *par = st->codecpar;
31	AVIOContext *pb = s->pb;
32	VocType type;
33	int size, tmp_codec=-1;
34	int sample_rate = 0;
35	int channels = 1;
36	int64_t duration;
37	int ret;
38
39	av_add_index_entry(st,
40	avio_tell(pb),
41	voc->pts,
42	voc->remaining_size,
43	0,
44	AVINDEX_KEYFRAME);
45
46	while (!voc->remaining_size) {
47	type = avio_r8(pb);
48	if (type == VOC_TYPE_EOF)
49	return AVERROR_EOF;
50	voc->remaining_size = avio_rl24(pb);
51	if (!voc->remaining_size) {
52	if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL))
53	return AVERROR(EIO);
54	voc->remaining_size = avio_size(pb) - avio_tell(pb);
55	}
56	max_size -= 4;
57
58	switch (type) {
59	case VOC_TYPE_VOICE_DATA:
60	if (!par->sample_rate) {
61	par->sample_rate = 1000000 / (256 - avio_r8(pb));
62	if (sample_rate)
63	par->sample_rate = sample_rate;
64	avpriv_set_pts_info(st, 64, 1, par->sample_rate);
65	par->channels = channels;
66	par->bits_per_coded_sample = av_get_bits_per_sample(par->codec_id);
67	} else
68	avio_skip(pb, 1);
69	tmp_codec = avio_r8(pb);
70	voc->remaining_size -= 2;
71	max_size -= 2;
72	channels = 1;
73	break;
74
75	case VOC_TYPE_VOICE_DATA_CONT:
76	break;
77
78	case VOC_TYPE_EXTENDED:
79	sample_rate = avio_rl16(pb);
80	avio_r8(pb);
81	channels = avio_r8(pb) + 1;
82	sample_rate = 256000000 / (channels * (65536 - sample_rate));
83	voc->remaining_size = 0;
84	max_size -= 4;
85	break;
86
87	case VOC_TYPE_NEW_VOICE_DATA:
88	if (!par->sample_rate) {
89	par->sample_rate = avio_rl32(pb);
90	avpriv_set_pts_info(st, 64, 1, par->sample_rate);
91	par->bits_per_coded_sample = avio_r8(pb);
92	par->channels = avio_r8(pb);
93	} else
94	avio_skip(pb, 6);
95	tmp_codec = avio_rl16(pb);
96	avio_skip(pb, 4);
97	voc->remaining_size -= 12;
98	max_size -= 12;
99	break;
100
101	default:
102	avio_skip(pb, voc->remaining_size);
103	max_size -= voc->remaining_size;
104	voc->remaining_size = 0;
105	break;
106	}
107	}
108
109	if (par->sample_rate <= 0) {
110	av_log(s, AV_LOG_ERROR, "Invalid sample rate %d\n", par->sample_rate);
111	return AVERROR_INVALIDDATA;
112	}
113
114	if (tmp_codec >= 0) {
115	tmp_codec = ff_codec_get_id(ff_voc_codec_tags, tmp_codec);
116	if (par->codec_id == AV_CODEC_ID_NONE)
117	par->codec_id = tmp_codec;
118	else if (par->codec_id != tmp_codec)
119	av_log(s, AV_LOG_WARNING, "Ignoring mid-stream change in audio codec\n");
120	if (par->codec_id == AV_CODEC_ID_NONE) {
121	if (s->audio_codec_id == AV_CODEC_ID_NONE) {
122	av_log(s, AV_LOG_ERROR, "unknown codec tag\n");
123	return AVERROR(EINVAL);
124	}
125	av_log(s, AV_LOG_WARNING, "unknown codec tag\n");
126	}
127	}
128
129	par->bit_rate = (int64_t)par->sample_rate * par->channels * par->bits_per_coded_sample;
130
131	if (max_size <= 0)
132	max_size = 2048;
133	size = FFMIN(voc->remaining_size, max_size);
134	voc->remaining_size -= size;
135
136	ret = av_get_packet(pb, pkt, size);
137	pkt->dts = pkt->pts = voc->pts;
138
139	duration = av_get_audio_frame_duration2(st->codecpar, size);
140	if (duration > 0 && voc->pts != AV_NOPTS_VALUE)
141	voc->pts += duration;
142	else
143	voc->pts = AV_NOPTS_VALUE;
144
145	return ret;
146	}
147