From patchwork Fri Dec 22 14:35:09 2017 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: misty@brew.sh X-Patchwork-Id: 6907 Delivered-To: ffmpegpatchwork@gmail.com Received: by 10.2.79.195 with SMTP id r64csp2739512jad; Fri, 22 Dec 2017 11:50:03 -0800 (PST) X-Google-Smtp-Source: ACJfBot5WzyDh4N0k/Hbkg+rLFfTs/dbg//ZkiPtyRBCUtdBugYnaGqlz/3r9pbjCRkccKzAtYOs X-Received: by 10.223.160.25 with SMTP id k25mr10656078wrk.252.1513972203657; Fri, 22 Dec 2017 11:50:03 -0800 (PST) ARC-Seal: i=1; a=rsa-sha256; t=1513972203; cv=none; d=google.com; s=arc-20160816; b=L/T7jvbfibBXvgkz/u4Tuij5gvBOITg2oRfVOWNOjgJWbEW+kCzUl391wYxGQX1JZA eb7eGb5aZ2tduXn9dw6VlDuStw/dEZtXJR4InzsmGNKcQ2VwvO8gzCpIlupWR2FKlR+b iiN91QuFPTzM0BGY3DHoGFd7bUeSb9izLSWU9r1d16gicpThHidkqAXg/7HWa02kaZEV oSd6fGil0FhSqsT9sKbyJ38xMzXwrgck6QHndV3gvNsgE83ZhmjSjNWV4uS45B/O5zKL cofUw0LmcGf5FgOAbAqisu/A5dYavoaTIOn+73wwzlmsPU/Z0ZQYJew3o5ShhkwKSkoG 9tqA== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:dkim-signature:delivered-to:arc-authentication-results; bh=gaJZHuLlrLyxrr/4RjTH3nGSd0mKI3oE3hXbOnMEPmQ=; b=BpoBbzacvJmZzGEIkMz8pdMnTDT0GtdzzD1lP826uiSockpMCCdxxcd0mPLYKLgSw/ 8XadmjdCdipRTXLkf5YCmUZY/VcDY3DA0nxJOu4Qv+6xHcIEy/DneQuW3Vi7cNZBz9d6 QSYDuydROZpnogyVQjrMq51mGmVXbE0QdYczRQByHn7OQuUcPBW6EuVuY6VRdZ4shMud 3VWgqBHS7EtTpOyr8Iou2fqj0vw4roCH22R66V+vEdwB5PimHBjeko5TDGAb/9SNvvIA H1llF/McqhnbnKNR4kEDkjdYWlA+TB3mT9z+xkALIB8SOtgS0mVNjrzes6uVy3veHWSd WcuA== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=l4FDn4Ro; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id d23si6886162wmi.8.2017.12.22.11.50.03; Fri, 22 Dec 2017 11:50:03 -0800 (PST) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=l4FDn4Ro; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id F37D3688289; Fri, 22 Dec 2017 21:49:25 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-pl0-f48.google.com (mail-pl0-f48.google.com [209.85.160.48]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id D04BE6882CE for ; Fri, 22 Dec 2017 16:41:26 +0200 (EET) Received: by mail-pl0-f48.google.com with SMTP id s3so13002894plp.4 for ; Fri, 22 Dec 2017 06:41:37 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=sender:from:to:cc:subject:date:message-id:in-reply-to:references; bh=4dG2k7sKfmsbOIBv3yba72gGOKs3+AY3NZnNHWP4Sks=; b=l4FDn4Ro66lJ/77EIU3r9PlLRQcCO276gv7Amhl7RkjK3fCGCx//LJ5eELT5KzURJS B8FPMKM9Xz2gBrq12nYgpWEcX+w4Xhn+oqiJTZRo4NwHAp6va2lKToyNgeCz1swK8ei8 frKX0SMV0qDJ1MJv6Cog3F8A9qX4n6UPU/sm1+QdGQa4KNEtO7sssYnyf9wNYSqQXp1A +YIub95BsOiCeKHfdN9YyOj8OUGtWEhAzwCJtsaGSVdrO/hIIRb8okKGMu3icUPWBgoP xbmhjioWoQ39RqDOZncChh7UKQT79OxNlg7h20LMQre+FCZLwHp6uwXxeBp2QOtnJFYs BiOg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:sender:from:to:cc:subject:date:message-id :in-reply-to:references; bh=4dG2k7sKfmsbOIBv3yba72gGOKs3+AY3NZnNHWP4Sks=; b=lYMC4VJeap6EYyeWQnlSqjNzr8gEUpB47MgdvmM9VEsuCLTM2VOguxzDiiJJXig+bB Iw1Y6wKNb101TzVPp60FWvZwq2MHctWf89QQPPJewihc+n0QPUiu9LZN6R5/vG5afb33 TcA+koGnObFulfsJJ+8DLfZX/KTcWqCr8yayq1b0988OD7OSAEOs4kO+qjynbOt2FAMX 9lTdOIOAseT2BljfBS7bVEtr9flXWBbh0fx/yWSha8qp8Ug0Obgz+wEytWBrhcCFDFhy xvo0oUQ8Bsbqa18Up7mzFU8N+d4FZcWhnUS195VjH1Er5ULPhHyUqbPr6cWdF2dSgbda E0HA== X-Gm-Message-State: AKGB3mKjOG96ov5t6HpsC2tJHA+WP4S4XfIhvsk7hKwOHkwxnNBJPHVj dVJ7/A2rYBTtQ9KzL6AvCc2o8jWI X-Received: by 10.84.140.5 with SMTP id 5mr14442661pls.149.1513953378996; Fri, 22 Dec 2017 06:36:18 -0800 (PST) Received: from localhost.localdomain (220-253-129-240.dyn.iinet.net.au. [220.253.129.240]) by smtp.gmail.com with ESMTPSA id 73sm47522026pfr.145.2017.12.22.06.36.15 (version=TLS1_2 cipher=ECDHE-RSA-AES128-SHA bits=128/128); Fri, 22 Dec 2017 06:36:18 -0800 (PST) From: misty@brew.sh To: ffmpeg-devel@ffmpeg.org Date: Fri, 22 Dec 2017 22:35:09 +0800 Message-Id: <20171222143512.92271-3-misty@brew.sh> X-Mailer: git-send-email 2.15.1 In-Reply-To: <20171222143512.92271-1-misty@brew.sh> References: <20171222143512.92271-1-misty@brew.sh> X-Mailman-Approved-At: Fri, 22 Dec 2017 21:49:23 +0200 Subject: [FFmpeg-devel] [PATCH 2/5] mpeg: add experimental support for PSMF audio. X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Maxim Poliakovski MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" From: Maxim Poliakovski --- libavcodec/Makefile | 1 + libavcodec/allcodecs.c | 1 + libavcodec/atrac3plus_parser.c | 153 +++++++++++++++++++++++++++++++++++++++++ libavformat/mpeg.c | 27 +++++++- 4 files changed, 181 insertions(+), 1 deletion(-) create mode 100644 libavcodec/atrac3plus_parser.c diff --git a/libavcodec/Makefile b/libavcodec/Makefile index ca72138c02..e0e3f1ebac 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -977,6 +977,7 @@ OBJS-$(CONFIG_AAC_PARSER) += aac_parser.o aac_ac3_parser.o \ mpeg4audio.o OBJS-$(CONFIG_AC3_PARSER) += ac3tab.o aac_ac3_parser.o OBJS-$(CONFIG_ADX_PARSER) += adx_parser.o adx.o +OBJS-$(CONFIG_ATRAC3P_PARSER) += atrac3plus_parser.o OBJS-$(CONFIG_BMP_PARSER) += bmp_parser.o OBJS-$(CONFIG_CAVSVIDEO_PARSER) += cavs_parser.o OBJS-$(CONFIG_COOK_PARSER) += cook_parser.o diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c index ed1e7ab06e..81d5d2814a 100644 --- a/libavcodec/allcodecs.c +++ b/libavcodec/allcodecs.c @@ -623,6 +623,7 @@ static void register_all(void) REGISTER_PARSER(AAC_LATM, aac_latm); REGISTER_PARSER(AC3, ac3); REGISTER_PARSER(ADX, adx); + REGISTER_PARSER(ATRAC3P, atrac3p); REGISTER_PARSER(BMP, bmp); REGISTER_PARSER(CAVSVIDEO, cavsvideo); REGISTER_PARSER(COOK, cook); diff --git a/libavcodec/atrac3plus_parser.c b/libavcodec/atrac3plus_parser.c new file mode 100644 index 0000000000..01fcad4c45 --- /dev/null +++ b/libavcodec/atrac3plus_parser.c @@ -0,0 +1,153 @@ +/* + * Copyright (C) 2014 Maxim Poliakovski + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "parser.h" +#include "get_bits.h" +#include "libavformat/oma.h" + +typedef struct Atrac3PlusParseContext { + ParseContext pc; + uint8_t hdr[8]; + int hdr_bytes_needed, got_bytes; + int sample_rate, channel_id, frame_size; +} Atrac3PlusParseContext; + +static int parse_sound_frame_header(Atrac3PlusParseContext *c, + const uint8_t *buf) +{ + uint16_t atrac_config; + + if (AV_RB16(buf) != 0x0FD0) + return AVERROR_INVALIDDATA; + + atrac_config = AV_RB16(&buf[2]); + c->sample_rate = ff_oma_srate_tab[(atrac_config >> 13) & 7] * 100; + c->channel_id = (atrac_config >> 10) & 7; + c->frame_size = ((atrac_config & 0x3FF) * 8) + 8; + + if (!c->channel_id || !c->sample_rate || !c->frame_size) + return AVERROR_INVALIDDATA; + + return 0; +} + +static int ff_atrac3p_parse(AVCodecParserContext *s, + AVCodecContext *avctx, + const uint8_t **poutbuf, int *poutbuf_size, + const uint8_t *buf, int buf_size) +{ + Atrac3PlusParseContext *ctx = s->priv_data; + const uint8_t *hdr_buf = buf; + size_t bytes_remain; + int frame_size, hdr_bytes = 8; + int next = 0; + + if (s->flags & PARSER_FLAG_COMPLETE_FRAMES || !buf_size) { + next = buf_size; + } else { + if (buf_size >= 2) { + bytes_remain = AV_RB16(buf); + + if (bytes_remain != 0xFD0) { + next += 2; + buf += 2; + buf_size -= 2; + hdr_buf = buf; + + if (bytes_remain && !ctx->pc.index && !ctx->hdr_bytes_needed) { + av_log(avctx, AV_LOG_ERROR, + "2nd frame portion found but the 1st one is missing!\n"); + return AVERROR_INVALIDDATA; + } + + if (ctx->hdr_bytes_needed) { + if (buf_size >= ctx->hdr_bytes_needed) { + memcpy(&ctx->hdr[8 - ctx->hdr_bytes_needed], + buf, ctx->hdr_bytes_needed); + hdr_bytes = ctx->hdr_bytes_needed; + ctx->hdr_bytes_needed = 0; + hdr_buf = ctx->hdr; + } + } else if (bytes_remain) { + if (buf_size < bytes_remain) { + av_log(avctx, AV_LOG_ERROR, + "Couldn't combine frame: bytes needed=%d, bytes supplied=%d\n", + bytes_remain, buf_size); + return AVERROR_INVALIDDATA; + } + + next += bytes_remain; + ff_combine_frame(&ctx->pc, bytes_remain, &buf, &buf_size); + + *poutbuf = buf; + *poutbuf_size = buf_size; + return next; + } + } + } + + if (buf_size < hdr_bytes) { + /* looks like we got an incomplete header */ + memcpy(ctx->hdr, buf, buf_size); + ctx->hdr_bytes_needed = 8 - buf_size; + *poutbuf = NULL; + *poutbuf_size = 0; + return buf_size; + } + + if (parse_sound_frame_header(ctx, hdr_buf)) { + av_log(avctx, AV_LOG_ERROR, "Invalid sound frame header!\n"); + return AVERROR_INVALIDDATA; + } + + avctx->sample_rate = ctx->sample_rate; + avctx->block_align = ctx->frame_size; + avctx->bit_rate = ctx->sample_rate * ctx->frame_size * 8 / 2048; + avctx->channels = ff_oma_chid_to_num_channels[ctx->channel_id - 1]; + avctx->channel_layout = ff_oma_chid_to_native_layout[ctx->channel_id - 1]; + + next += hdr_bytes; + buf += hdr_bytes; + buf_size -= hdr_bytes; + frame_size = ctx->frame_size; + + if (buf_size < frame_size) + frame_size = END_NOT_FOUND; + + if (ff_combine_frame(&ctx->pc, frame_size, &buf, &buf_size) < 0) { + *poutbuf = NULL; + *poutbuf_size = 0; + return buf_size + next; + } + + next += frame_size; + } + + *poutbuf = buf; + *poutbuf_size = buf_size; + return next; +} + +AVCodecParser ff_atrac3p_parser = { + .codec_ids = { AV_CODEC_ID_ATRAC3P }, + .priv_data_size = sizeof(Atrac3PlusParseContext), + .parser_parse = ff_atrac3p_parse, + .parser_close = ff_parse_close, +}; diff --git a/libavformat/mpeg.c b/libavformat/mpeg.c index 50fe7a1a76..895c6fb231 100644 --- a/libavformat/mpeg.c +++ b/libavformat/mpeg.c @@ -128,6 +128,7 @@ typedef struct MpegDemuxContext { int sofdec; int dvd; int imkh_cctv; + int sony_psmf; // true if Play Station Movie file signature is present #if CONFIG_VOBSUB_DEMUXER AVFormatContext *sub_ctx; FFDemuxSubtitlesQueue q[32]; @@ -147,6 +148,8 @@ static int mpegps_read_header(AVFormatContext *s) avio_get_str(s->pb, 6, buffer, sizeof(buffer)); if (!memcmp("IMKH", buffer, 4)) { m->imkh_cctv = 1; + } else if (!memcmp("PSMF00", buffer, 6)) { + m->sony_psmf = 1; } else if (!memcmp("Sofdec", buffer, 6)) { m->sofdec = 1; } else @@ -441,7 +444,7 @@ redo: goto redo; } - if (startcode == PRIVATE_STREAM_1) { + if (startcode == PRIVATE_STREAM_1 && !m->sony_psmf) { startcode = avio_r8(s->pb); len--; } @@ -541,6 +544,28 @@ redo: else request_probe= 1; type = AVMEDIA_TYPE_VIDEO; + } else if (startcode == PRIVATE_STREAM_1 && m->sony_psmf) { + uint8_t stream_id; + + if (len < 2) + goto skip; + stream_id = avio_r8(s->pb); + avio_r8(s->pb); // skip padding + len -= 2; + if (!(stream_id & 0xF0)) { // seems like we got an ATRAC stream + /* check if an appropriate stream already exists */ + for (i = 0; i < s->nb_streams; i++) { + st = s->streams[i]; + if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO && + st->codec->codec_id == AV_CODEC_ID_ATRAC3P && + st->id - 0x1BD0 == (stream_id & 0xF)) + goto found; + } + + startcode = 0x1BD0 + (stream_id & 0xF); + type = AVMEDIA_TYPE_AUDIO; + codec_id = AV_CODEC_ID_ATRAC3P; + } } else if (startcode == PRIVATE_STREAM_2) { type = AVMEDIA_TYPE_DATA; codec_id = AV_CODEC_ID_DVD_NAV;