1cabdff1aSopenharmony_ci/* 2cabdff1aSopenharmony_ci * SDR2 demuxer 3cabdff1aSopenharmony_ci * Copyright (c) 2014 Paul B Mahol 4cabdff1aSopenharmony_ci * 5cabdff1aSopenharmony_ci * This file is part of FFmpeg. 6cabdff1aSopenharmony_ci * 7cabdff1aSopenharmony_ci * FFmpeg is free software; you can redistribute it and/or 8cabdff1aSopenharmony_ci * modify it under the terms of the GNU Lesser General Public 9cabdff1aSopenharmony_ci * License as published by the Free Software Foundation; either 10cabdff1aSopenharmony_ci * version 2.1 of the License, or (at your option) any later version. 11cabdff1aSopenharmony_ci * 12cabdff1aSopenharmony_ci * FFmpeg is distributed in the hope that it will be useful, 13cabdff1aSopenharmony_ci * but WITHOUT ANY WARRANTY; without even the implied warranty of 14cabdff1aSopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15cabdff1aSopenharmony_ci * Lesser General Public License for more details. 16cabdff1aSopenharmony_ci * 17cabdff1aSopenharmony_ci * You should have received a copy of the GNU Lesser General Public 18cabdff1aSopenharmony_ci * License along with FFmpeg; if not, write to the Free Software 19cabdff1aSopenharmony_ci * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20cabdff1aSopenharmony_ci */ 21cabdff1aSopenharmony_ci 22cabdff1aSopenharmony_ci#include "libavutil/intreadwrite.h" 23cabdff1aSopenharmony_ci#include "avformat.h" 24cabdff1aSopenharmony_ci#include "internal.h" 25cabdff1aSopenharmony_ci 26cabdff1aSopenharmony_cistatic int sdr2_probe(const AVProbeData *p) 27cabdff1aSopenharmony_ci{ 28cabdff1aSopenharmony_ci if (AV_RL32(p->buf) != MKTAG('S', 'R', 'A', 1)) 29cabdff1aSopenharmony_ci return 0; 30cabdff1aSopenharmony_ci 31cabdff1aSopenharmony_ci return AVPROBE_SCORE_EXTENSION; 32cabdff1aSopenharmony_ci} 33cabdff1aSopenharmony_ci 34cabdff1aSopenharmony_ci#define FIRST 0xA8 35cabdff1aSopenharmony_ci 36cabdff1aSopenharmony_cistatic int sdr2_read_header(AVFormatContext *s) 37cabdff1aSopenharmony_ci{ 38cabdff1aSopenharmony_ci AVStream *st, *ast; 39cabdff1aSopenharmony_ci 40cabdff1aSopenharmony_ci ast = avformat_new_stream(s, 0); 41cabdff1aSopenharmony_ci if (!ast) 42cabdff1aSopenharmony_ci return AVERROR(ENOMEM); 43cabdff1aSopenharmony_ci 44cabdff1aSopenharmony_ci st = avformat_new_stream(s, 0); 45cabdff1aSopenharmony_ci if (!st) 46cabdff1aSopenharmony_ci return AVERROR(ENOMEM); 47cabdff1aSopenharmony_ci 48cabdff1aSopenharmony_ci avio_skip(s->pb, 20); 49cabdff1aSopenharmony_ci avpriv_set_pts_info(st, 64, 1, avio_rl32(s->pb)); 50cabdff1aSopenharmony_ci st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; 51cabdff1aSopenharmony_ci st->codecpar->width = avio_rl32(s->pb); 52cabdff1aSopenharmony_ci st->codecpar->height = avio_rl32(s->pb); 53cabdff1aSopenharmony_ci st->codecpar->codec_id = AV_CODEC_ID_H264; 54cabdff1aSopenharmony_ci ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL; 55cabdff1aSopenharmony_ci 56cabdff1aSopenharmony_ci ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; 57cabdff1aSopenharmony_ci ast->codecpar->ch_layout.nb_channels = 1; 58cabdff1aSopenharmony_ci ast->codecpar->sample_rate = 8000; 59cabdff1aSopenharmony_ci ast->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE; 60cabdff1aSopenharmony_ci avpriv_set_pts_info(ast, 64, 1, 8000); 61cabdff1aSopenharmony_ci 62cabdff1aSopenharmony_ci avio_seek(s->pb, FIRST, SEEK_SET); 63cabdff1aSopenharmony_ci 64cabdff1aSopenharmony_ci return 0; 65cabdff1aSopenharmony_ci} 66cabdff1aSopenharmony_ci 67cabdff1aSopenharmony_cistatic const uint8_t header[24] = { 68cabdff1aSopenharmony_ci 0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e, 69cabdff1aSopenharmony_ci 0xa6, 0x80, 0xb0, 0x7e, 0x40, 0x00, 0x00, 0x00, 70cabdff1aSopenharmony_ci 0x01, 0x68, 0xce, 0x38, 0x80, 0x00, 0x00, 0x00 71cabdff1aSopenharmony_ci}; 72cabdff1aSopenharmony_ci 73cabdff1aSopenharmony_cistatic int sdr2_read_packet(AVFormatContext *s, AVPacket *pkt) 74cabdff1aSopenharmony_ci{ 75cabdff1aSopenharmony_ci int64_t pos; 76cabdff1aSopenharmony_ci unsigned next; 77cabdff1aSopenharmony_ci int flags, ret = 0, is_video; 78cabdff1aSopenharmony_ci 79cabdff1aSopenharmony_ci pos = avio_tell(s->pb); 80cabdff1aSopenharmony_ci 81cabdff1aSopenharmony_ci flags = avio_rl32(s->pb); 82cabdff1aSopenharmony_ci avio_skip(s->pb, 4); 83cabdff1aSopenharmony_ci 84cabdff1aSopenharmony_ci next = avio_rl32(s->pb); 85cabdff1aSopenharmony_ci if (next <= 52) 86cabdff1aSopenharmony_ci return AVERROR_INVALIDDATA; 87cabdff1aSopenharmony_ci 88cabdff1aSopenharmony_ci avio_skip(s->pb, 6); 89cabdff1aSopenharmony_ci is_video = avio_rl32(s->pb); 90cabdff1aSopenharmony_ci avio_skip(s->pb, 30); 91cabdff1aSopenharmony_ci 92cabdff1aSopenharmony_ci if (pos == FIRST) { 93cabdff1aSopenharmony_ci if ((ret = av_new_packet(pkt, next - 52 + 24)) < 0) 94cabdff1aSopenharmony_ci return ret; 95cabdff1aSopenharmony_ci memcpy(pkt->data, header, 24); 96cabdff1aSopenharmony_ci ret = avio_read(s->pb, pkt->data + 24, next - 52); 97cabdff1aSopenharmony_ci if (ret < 0) { 98cabdff1aSopenharmony_ci return ret; 99cabdff1aSopenharmony_ci } 100cabdff1aSopenharmony_ci av_shrink_packet(pkt, ret + 24); 101cabdff1aSopenharmony_ci } else { 102cabdff1aSopenharmony_ci ret = av_get_packet(s->pb, pkt, next - 52); 103cabdff1aSopenharmony_ci } 104cabdff1aSopenharmony_ci pkt->stream_index = !!is_video; 105cabdff1aSopenharmony_ci pkt->pos = pos; 106cabdff1aSopenharmony_ci if (flags & (1 << 12)) 107cabdff1aSopenharmony_ci pkt->flags |= AV_PKT_FLAG_KEY; 108cabdff1aSopenharmony_ci 109cabdff1aSopenharmony_ci return ret; 110cabdff1aSopenharmony_ci} 111cabdff1aSopenharmony_ci 112cabdff1aSopenharmony_ciconst AVInputFormat ff_sdr2_demuxer = { 113cabdff1aSopenharmony_ci .name = "sdr2", 114cabdff1aSopenharmony_ci .long_name = NULL_IF_CONFIG_SMALL("SDR2"), 115cabdff1aSopenharmony_ci .read_probe = sdr2_probe, 116cabdff1aSopenharmony_ci .read_header = sdr2_read_header, 117cabdff1aSopenharmony_ci .read_packet = sdr2_read_packet, 118cabdff1aSopenharmony_ci .extensions = "sdr2", 119cabdff1aSopenharmony_ci .flags = AVFMT_GENERIC_INDEX, 120cabdff1aSopenharmony_ci}; 121