mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2026-01-12 00:06:51 +08:00
91ae6d10ab
Implement NEON optimization for compute_weights_line. Also update the function signature to use ptrdiff_t for stack arguments (max_meaningful_diff, startx, endx). This is done to unify the stack layout between Apple platforms (which pack 32-bit stack arguments tightly) and the generic AAPCS64 ABI (which requires 8-byte stack slots for 32-bit arguments). Using ptrdiff_t ensures 8-byte slots are used on all AArch64 platforms, avoiding ABI mismatches with the assembly implementation. The x86 AVX2 prototype is updated to match the new signature. Performance benchmark (AArch64) in MacOS M4: ./tests/checkasm/checkasm --test=vf_nlmeans --bench compute_weights_line_c: 151.1 ( 1.00x) compute_weights_line_neon: 62.6 ( 2.42x) Reviewed-by: Martin Storsjö <martin@martin.st> Signed-off-by: Jun Zhao <barryjzhao@tencent.com>
46 lines
1.9 KiB
C
46 lines
1.9 KiB
C
/*
|
|
* 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
|
|
*/
|
|
|
|
#ifndef AVFILTER_NLMEANS_H
|
|
#define AVFILTER_NLMEANS_H
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
typedef struct NLMeansDSPContext {
|
|
void (*compute_safe_ssd_integral_image)(uint32_t *dst, ptrdiff_t dst_linesize_32,
|
|
const uint8_t *s1, ptrdiff_t linesize1,
|
|
const uint8_t *s2, ptrdiff_t linesize2,
|
|
int w, int h);
|
|
void (*compute_weights_line)(const uint32_t *const iia,
|
|
const uint32_t *const iib,
|
|
const uint32_t *const iid,
|
|
const uint32_t *const iie,
|
|
const uint8_t *const src,
|
|
float *total_weight,
|
|
float *sum,
|
|
const float *const weight_lut,
|
|
ptrdiff_t max_meaningful_diff,
|
|
ptrdiff_t startx, ptrdiff_t endx);
|
|
} NLMeansDSPContext;
|
|
|
|
void ff_nlmeans_init_aarch64(NLMeansDSPContext *dsp);
|
|
void ff_nlmeans_init_x86(NLMeansDSPContext *dsp);
|
|
|
|
#endif /* AVFILTER_NLMEANS_H */
|