4 * Copyright (c) 2019 Swaraj Hota
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
25 #include "avio_internal.h"
27 typedef struct IFVContext {
28 uint32_t next_video_index;
29 uint32_t next_audio_index;
30 uint32_t total_vframes;
31 uint32_t total_aframes;
37 int video_stream_index;
38 int audio_stream_index;
41 static int ifv_probe(const AVProbeData *p)
43 static const uint8_t ifv_magic[] = {0x11, 0xd2, 0xd3, 0xab, 0xba, 0xa9,
44 0xcf, 0x11, 0x8e, 0xe6, 0x00, 0xc0, 0x0c, 0x20, 0x53, 0x65, 0x44};
46 if (!memcmp(p->buf, ifv_magic, sizeof(ifv_magic)))
47 return AVPROBE_SCORE_MAX;
52 static int read_index(AVFormatContext *s,
53 enum AVMediaType frame_type,
56 IFVContext *ifv = s->priv_data;
58 int64_t pos, size, timestamp;
59 uint32_t end_index, i;
62 if (frame_type == AVMEDIA_TYPE_VIDEO) {
63 end_index = ifv->total_vframes;
64 st = s->streams[ifv->video_stream_index];
66 end_index = ifv->total_aframes;
67 st = s->streams[ifv->audio_stream_index];
70 for (i = start_index; i < end_index; i++) {
71 pos = avio_rl32(s->pb);
72 size = avio_rl32(s->pb);
75 timestamp = avio_rl32(s->pb);
77 ret = av_add_index_entry(st, pos, timestamp, size, 0, 0);
81 avio_skip(s->pb, frame_type == AVMEDIA_TYPE_VIDEO ? 8: 4);
87 static int parse_header(AVFormatContext *s)
89 IFVContext *ifv = s->priv_data;
93 avio_skip(s->pb, 0x5c);
94 ifv->width = avio_rl16(s->pb);
95 ifv->height = avio_rl16(s->pb);
97 avio_skip(s->pb, 0x8);
98 vid_magic = avio_rl32(s->pb);
100 if (vid_magic != MKTAG('H','2','6','4'))
101 avpriv_request_sample(s, "Unknown video codec %x", vid_magic);
103 avio_skip(s->pb, 0x2c);
104 ifv->sample_rate = avio_rl32(s->pb);
105 aud_magic = avio_rl32(s->pb);
107 if (aud_magic == MKTAG('G','R','A','W')) {
108 ifv->is_audio_present = 1;
109 } else if (aud_magic == MKTAG('P','C','M','U')) {
110 ifv->is_audio_present = 0;
112 avpriv_request_sample(s, "Unknown audio codec %x", aud_magic);
115 avio_skip(s->pb, 0x44);
116 ifv->total_vframes = avio_rl32(s->pb);
117 ifv->total_aframes = avio_rl32(s->pb);
122 static int ifv_read_header(AVFormatContext *s)
124 IFVContext *ifv = s->priv_data;
128 ret = parse_header(s);
132 st = avformat_new_stream(s, NULL);
134 return AVERROR(ENOMEM);
136 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
137 st->codecpar->codec_id = AV_CODEC_ID_H264;
138 st->codecpar->width = ifv->width;
139 st->codecpar->height = ifv->height;
141 ifv->video_stream_index = st->index;
143 avpriv_set_pts_info(st, 32, 1, 1000);
145 if (ifv->is_audio_present) {
146 st = avformat_new_stream(s, NULL);
148 return AVERROR(ENOMEM);
150 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
151 st->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE;
152 st->codecpar->channels = 1;
153 st->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
154 st->codecpar->sample_rate = ifv->sample_rate;
155 ifv->audio_stream_index = st->index;
157 avpriv_set_pts_info(st, 32, 1, 1000);
161 avio_seek(s->pb, 0xf8, SEEK_SET);
163 ret = read_index(s, AVMEDIA_TYPE_VIDEO, 0);
167 if (ifv->is_audio_present) {
169 avio_seek(s->pb, 0x14918, SEEK_SET);
171 ret = read_index(s, AVMEDIA_TYPE_AUDIO, 0);
176 ifv->next_video_index = 0;
177 ifv->next_audio_index = 0;
182 static int ifv_read_packet(AVFormatContext *s, AVPacket *pkt)
184 IFVContext *ifv = s->priv_data;
186 AVIndexEntry *ev, *ea, *e_next;
189 ev = ea = e_next = NULL;
191 if (ifv->next_video_index < ifv->total_vframes) {
192 st = s->streams[ifv->video_stream_index];
193 if (ifv->next_video_index < st->nb_index_entries)
194 e_next = ev = &st->index_entries[ifv->next_video_index];
197 if (ifv->is_audio_present &&
198 ifv->next_audio_index < ifv->total_aframes) {
199 st = s->streams[ifv->audio_stream_index];
200 if (ifv->next_audio_index < st->nb_index_entries) {
201 ea = &st->index_entries[ifv->next_audio_index];
202 if (!ev || ea->timestamp < ev->timestamp)
208 if (ifv->is_audio_present && !ea) {
209 /*read new video and audio indexes*/
211 ifv->next_video_index = ifv->total_vframes;
212 ifv->next_audio_index = ifv->total_aframes;
214 avio_skip(s->pb, 0x1c);
215 ifv->total_vframes += avio_rl32(s->pb);
216 ifv->total_aframes += avio_rl32(s->pb);
217 avio_skip(s->pb, 0xc);
219 if (avio_feof(s->pb))
222 ret = read_index(s, AVMEDIA_TYPE_VIDEO, ifv->next_video_index);
226 ret = read_index(s, AVMEDIA_TYPE_AUDIO, ifv->next_audio_index);
232 } else if (!ifv->is_audio_present) {
233 /*read new video index*/
235 ifv->next_video_index = ifv->total_vframes;
237 avio_skip(s->pb, 0x1c);
238 ifv->total_vframes += avio_rl32(s->pb);
239 avio_skip(s->pb, 0x10);
241 if (avio_feof(s->pb))
244 ret = read_index(s, AVMEDIA_TYPE_VIDEO, ifv->next_video_index);
252 if (!e_next) return AVERROR_EOF;
254 avio_seek(s->pb, e_next->pos, SEEK_SET);
255 ret = av_get_packet(s->pb, pkt, e_next->size);
260 ifv->next_video_index++;
261 pkt->stream_index = ifv->video_stream_index;
263 ifv->next_audio_index++;
264 pkt->stream_index = ifv->audio_stream_index;
267 pkt->pts = e_next->timestamp;
268 pkt->pos = e_next->pos;
273 static int ifv_read_seek(AVFormatContext *s, int stream_index, int64_t ts, int flags)
275 IFVContext *ifv = s->priv_data;
277 for (unsigned i = 0; i < s->nb_streams; i++) {
278 int index = av_index_search_timestamp(s->streams[i], ts, AVSEEK_FLAG_ANY);
280 ifv->next_video_index = ifv->total_vframes - 1;
281 ifv->next_audio_index = ifv->total_aframes - 1;
285 if (i == ifv->video_stream_index) {
286 ifv->next_video_index = index;
288 ifv->next_audio_index = index;
295 AVInputFormat ff_ifv_demuxer = {
297 .long_name = NULL_IF_CONFIG_SMALL("IFV CCTV DVR"),
298 .priv_data_size = sizeof(IFVContext),
300 .read_probe = ifv_probe,
301 .read_header = ifv_read_header,
302 .read_packet = ifv_read_packet,
303 .read_seek = ifv_read_seek,