From patchwork Thu Jun 15 15:08:33 2017 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: James Darnley X-Patchwork-Id: 3988 Delivered-To: ffmpegpatchwork@gmail.com Received: by 10.103.22.4 with SMTP id 4csp847987vsw; Thu, 15 Jun 2017 08:15:22 -0700 (PDT) X-Received: by 10.223.130.85 with SMTP id 79mr1743730wrb.111.1497539722523; Thu, 15 Jun 2017 08:15:22 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1497539722; cv=none; d=google.com; s=arc-20160816; b=QvT8IY5ciIhlDflRW/t0InQnn4z9SV6hKEpA5YMR7GR6bX6LLkpl6qg1VTeCstQFNS gyATpRy9ndUvFY5K3A8SkZlvyyuFC8AupK2dql/EFnZVfb1zU6gMaZFeD34oFNQZaE1h AJaeDGsAMowLuLxbvw9wuzBrCtkEjfGMLDu0CBe5RZRGxjlQOqcjkWFOzVPFLI+YXnoS qXakDAx1NMH6oUJH2F4ej9/zi6eQKAz5YA5yph+LJBZWZ+Sw4+Z6/c0KgDvVPjfzDIFX Jv42SNq9XG1OQOVPBl5YLaHP3V3K1Beb3AwIF1I93qjdxt85y5W8fAR21KwHK+tUjN7M Nhqw== 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:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:message-id:date:to:from:dkim-signature :delivered-to:arc-authentication-results; bh=1qSmEzk/Dt6K3QjLPJn5+J0+H01r86fpGD0bojBP0HY=; b=ECIwRHJocbZpb4eh8DTyE11jqL2kNN8EeApg+wqsnHo/YjrSG8aOhxL/fcl7vQjQNV 2LckfERu9a3LZidaSgrnTZepZwpreXRg/VqX7mU4yWN1Iy8quBESnzqo76SAhXUDcFCc x/K0RSiDtRf2Bu6qwCf6Cq0uRLs0Cw0ttSm0aeQz5M8nA+fLK1YgJPxF4MFjkmOsjFVW 9het4BKcqZDnsAgPEbFX/IIN/D0ojft7vvuDLS2AEbk7P3ErrCUr6gZqlZdGVuHBi5MX S3Oxv/FFDBnRrpfXsyx+lEe2UA9Rdu6M7SA+ga1zgT4Qjct4SHWSK51SjQ53y3xwNxes kXYw== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@ob-encoder-com.20150623.gappssmtp.com header.b=egphChY3; 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 g26si409485wrg.365.2017.06.15.08.15.21; Thu, 15 Jun 2017 08:15:22 -0700 (PDT) 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=@ob-encoder-com.20150623.gappssmtp.com header.b=egphChY3; 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 71DB668A34E; Thu, 15 Jun 2017 18:15:16 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wr0-f194.google.com (mail-wr0-f194.google.com [209.85.128.194]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 6FFF468A340 for ; Thu, 15 Jun 2017 18:15:09 +0300 (EEST) Received: by mail-wr0-f194.google.com with SMTP id z45so4080461wrb.2 for ; Thu, 15 Jun 2017 08:15:12 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=ob-encoder-com.20150623.gappssmtp.com; s=20150623; h=sender:from:to:subject:date:message-id; bh=GgxKhG3vej7e53lQmoj2HYYkbULw0iAFQaaYg3eYA48=; b=egphChY3ouXT9Qok8WLuyW8cOM4S/M3aMMwoPI492OpRsfUoJQ7Q06lRl5EHUDMPNl LSd34VEqauyK1q6ISKLdi3KDSpI/uV6MsCLisdAEhSA7nUveUaoNIpAZ91s0dVBkK73x FA5/qzGdSeVFtD2Z7n66wJmdAL3euFc7Sc3e3FugjGMfDTHZB8yarcxD5dlfiWPSIfPm tdB/ZXNGafNqTr0S1QoB/Y981irwTYP1yzz+Nwo0P8jtCQx0k/uJRLCNsh16qG9vpdJQ /N9R5h7w5v7SY5e68FWZXC2XKTDVziDZHrOOv5Mlqt2JS48iMdW5sSVwy0/+2ZjlhcUH Ja+Q== 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:subject:date:message-id; bh=GgxKhG3vej7e53lQmoj2HYYkbULw0iAFQaaYg3eYA48=; b=PMuaZ9G3jqGG/mFNxQIIWc5QYhv8BtqgbwWudIRd6xACaGZtnxVizkJCmrQmeDr3yS 4VAast9+7d6+blMjAsf4CPu3/+XuaoweGp8ncEMRSJYiIZ2D8e4D4L1LG4glTAz7z3rB XE/jP9/aKLIMjiJS5nV3SoPVm7p2fL7P0z63vBCj/PBW29tzduHBTa8uguga1lVpu11H EDQkLf39xVgv/XqNzwFQRsBoPqApSI01O7uHX6SZRRFDWiDS4kQOpMgRFoP7GeIO4/ZJ 1QXw0biBT3qNRwU2fWgyfDYUmzf9i3jV57SGbTzpwSXBAeWsbZ9WJSX24hiylrvINwhy KXOA== X-Gm-Message-State: AKS2vOw4W5ul8jznIkglAbyaQ7HRewuObln4zPTH0L87tFnzqTZhaZpI OML8lTuf/FIqukkNU3I= X-Received: by 10.223.135.185 with SMTP id b54mr3361182wrb.170.1497539711970; Thu, 15 Jun 2017 08:15:11 -0700 (PDT) Received: from Ifrit.systemlords.lan (d51a44418.access.telenet.be. [81.164.68.24]) by smtp.gmail.com with ESMTPSA id t15sm445744wmd.13.2017.06.15.08.15.11 for (version=TLS1_2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128/128); Thu, 15 Jun 2017 08:15:11 -0700 (PDT) From: James Darnley To: FFmpeg development discussions and patches Date: Thu, 15 Jun 2017 17:08:33 +0200 Message-Id: <20170615150833.894-1-jdarnley@obe.tv> X-Mailer: git-send-email 2.13.1 Subject: [FFmpeg-devel] [PATCH] avcodec/x86: add an 8-bit simple IDCT function based on the x86-64 high depth functions 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 MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Includes add/put functions Rounding contributed by Ronald S. Bultje --- I must be stupid. I dropped the stack space change somewhere. libavcodec/tests/x86/dct.c | 2 + libavcodec/x86/idctdsp_init.c | 23 ++++++++++ libavcodec/x86/simple_idct.h | 9 ++++ libavcodec/x86/simple_idct10.asm | 94 ++++++++++++++++++++++++++++++++++++++++ 4 files changed, 128 insertions(+) diff --git a/libavcodec/tests/x86/dct.c b/libavcodec/tests/x86/dct.c index 34f5b8767b..317d973f9f 100644 --- a/libavcodec/tests/x86/dct.c +++ b/libavcodec/tests/x86/dct.c @@ -88,10 +88,12 @@ static const struct algo idct_tab_arch[] = { #if HAVE_YASM #if ARCH_X86_64 #if HAVE_SSE2_EXTERNAL + { "SIMPLE8-SSE2", ff_simple_idct8_sse2, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_SSE2}, { "SIMPLE10-SSE2", ff_simple_idct10_sse2, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_SSE2}, { "SIMPLE12-SSE2", ff_simple_idct12_sse2, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_SSE2, 1 }, #endif #if HAVE_AVX_EXTERNAL + { "SIMPLE8-AVX", ff_simple_idct8_avx, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_AVX}, { "SIMPLE10-AVX", ff_simple_idct10_avx, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_AVX}, { "SIMPLE12-AVX", ff_simple_idct12_avx, FF_IDCT_PERM_TRANSPOSE, AV_CPU_FLAG_AVX, 1 }, #endif diff --git a/libavcodec/x86/idctdsp_init.c b/libavcodec/x86/idctdsp_init.c index f1c915aa00..9da60d1a1e 100644 --- a/libavcodec/x86/idctdsp_init.c +++ b/libavcodec/x86/idctdsp_init.c @@ -94,9 +94,32 @@ av_cold void ff_idctdsp_init_x86(IDCTDSPContext *c, AVCodecContext *avctx, c->idct_add = ff_simple_idct_add_sse2; c->perm_type = FF_IDCT_PERM_SIMPLE; } + + if (ARCH_X86_64 && + !high_bit_depth && + avctx->lowres == 0 && + (avctx->idct_algo == FF_IDCT_AUTO || + avctx->idct_algo == FF_IDCT_SIMPLEAUTO || + avctx->idct_algo == FF_IDCT_SIMPLEMMX)) { + c->idct = ff_simple_idct8_sse2; + c->idct_put = ff_simple_idct8_put_sse2; + c->idct_add = ff_simple_idct8_add_sse2; + c->perm_type = FF_IDCT_PERM_TRANSPOSE; + } } if (ARCH_X86_64 && avctx->lowres == 0) { + if (EXTERNAL_AVX(cpu_flags) && + !high_bit_depth && + (avctx->idct_algo == FF_IDCT_AUTO || + avctx->idct_algo == FF_IDCT_SIMPLEAUTO || + avctx->idct_algo == FF_IDCT_SIMPLEMMX)) { + c->idct = ff_simple_idct8_avx; + c->idct_put = ff_simple_idct8_put_avx; + c->idct_add = ff_simple_idct8_add_avx; + c->perm_type = FF_IDCT_PERM_TRANSPOSE; + } + if (avctx->bits_per_raw_sample == 10 && (avctx->idct_algo == FF_IDCT_AUTO || avctx->idct_algo == FF_IDCT_SIMPLEAUTO || diff --git a/libavcodec/x86/simple_idct.h b/libavcodec/x86/simple_idct.h index d17ef6a462..9b64cfe9bc 100644 --- a/libavcodec/x86/simple_idct.h +++ b/libavcodec/x86/simple_idct.h @@ -29,6 +29,15 @@ void ff_simple_idct_put_mmx(uint8_t *dest, ptrdiff_t line_size, int16_t *block); void ff_simple_idct_add_sse2(uint8_t *dest, ptrdiff_t line_size, int16_t *block); void ff_simple_idct_put_sse2(uint8_t *dest, ptrdiff_t line_size, int16_t *block); +void ff_simple_idct8_sse2(int16_t *block); +void ff_simple_idct8_avx(int16_t *block); + +void ff_simple_idct8_put_sse2(uint8_t *dest, ptrdiff_t line_size, int16_t *block); +void ff_simple_idct8_put_avx(uint8_t *dest, ptrdiff_t line_size, int16_t *block); + +void ff_simple_idct8_add_sse2(uint8_t *dest, ptrdiff_t line_size, int16_t *block); +void ff_simple_idct8_add_avx(uint8_t *dest, ptrdiff_t line_size, int16_t *block); + void ff_simple_idct10_sse2(int16_t *block); void ff_simple_idct10_avx(int16_t *block); diff --git a/libavcodec/x86/simple_idct10.asm b/libavcodec/x86/simple_idct10.asm index b492303a57..10c5cd4340 100644 --- a/libavcodec/x86/simple_idct10.asm +++ b/libavcodec/x86/simple_idct10.asm @@ -31,11 +31,14 @@ SECTION_RODATA cextern pw_2 cextern pw_16 +cextern pw_32 cextern pw_1023 cextern pw_4095 +pd_round_11: times 4 dd 1<<(11-1) pd_round_12: times 4 dd 1<<(12-1) pd_round_15: times 4 dd 1<<(15-1) pd_round_19: times 4 dd 1<<(19-1) +pd_round_20: times 4 dd 1<<(20-1) %macro CONST_DEC 3 const %1 @@ -77,8 +80,99 @@ CONST_DEC w3_min_w7_lo, W3sh2_lo, -W7sh2 SECTION .text +%macro STORE_HI_LO 12 + movq %1, %9 + movq %3, %10 + movq %5, %11 + movq %7, %12 + movhps %2, %9 + movhps %4, %10 + movhps %6, %11 + movhps %8, %12 +%endmacro + +%macro LOAD_ZXBW_8 16 + pmovzxbw %1, %9 + pmovzxbw %2, %10 + pmovzxbw %3, %11 + pmovzxbw %4, %12 + pmovzxbw %5, %13 + pmovzxbw %6, %14 + pmovzxbw %7, %15 + pmovzxbw %8, %16 +%endmacro + +%macro LOAD_ZXBW_4 9 + movh %1, %5 + movh %2, %6 + movh %3, %7 + movh %4, %8 + punpcklbw %1, %9 + punpcklbw %2, %9 + punpcklbw %3, %9 + punpcklbw %4, %9 +%endmacro + +%define PASS4ROWS(base, stride, stride3) \ + [base], [base + stride], [base + 2*stride], [base + stride3] + %macro idct_fn 0 +define_constants _lo + +cglobal simple_idct8, 1, 1, 16, 32, block + IDCT_FN "", 11, pw_32, 20, "store" +RET + +; TODO: optimise by not writing the final data to the block. +cglobal simple_idct8_put, 3, 4, 16, 32, pixels, lsize, block + IDCT_FN "", 11, pw_32, 20 + lea r3, [3*lsizeq] + lea r2, [pixelsq + r3] + packuswb m8, m0 + packuswb m1, m2 + packuswb m4, m11 + packuswb m9, m10 + STORE_HI_LO PASS8ROWS(pixelsq, r2, lsizeq, r3), m8, m1, m4, m9 +RET + +; TODO: optimise by not writing the final data to the block. +cglobal simple_idct8_add, 3, 4, 16, 32, pixels, lsize, block + IDCT_FN "", 11, pw_32, 20 + lea r2, [3*lsizeq] + %if cpuflag(sse4) + lea r3, [pixelsq + r2] + LOAD_ZXBW_8 m3, m5, m6, m7, m12, m13, m14, m15, PASS8ROWS(pixelsq, r3, lsizeq, r2) + paddsw m8, m3 + paddsw m0, m5 + paddsw m1, m6 + paddsw m2, m7 + paddsw m4, m12 + paddsw m11, m13 + paddsw m9, m14 + paddsw m10, m15 + %else + pxor m12, m12 + LOAD_ZXBW_4 m3, m5, m6, m7, PASS4ROWS(pixelsq, lsizeq, r2), m12 + paddsw m8, m3 + paddsw m0, m5 + paddsw m1, m6 + paddsw m2, m7 + lea r3, [pixelsq + 4*lsizeq] + LOAD_ZXBW_4 m3, m5, m6, m7, PASS4ROWS(r3, lsizeq, r2), m12 + paddsw m4, m3 + paddsw m11, m5 + paddsw m9, m6 + paddsw m10, m7 + lea r3, [pixelsq + r2] + %endif + packuswb m8, m0 + packuswb m1, m2 + packuswb m4, m11 + packuswb m9, m10 + STORE_HI_LO PASS8ROWS(pixelsq, r3, lsizeq, r2), m8, m1, m4, m9 +RET + define_constants _hi cglobal simple_idct10, 1, 1, 16, block