59 lines
2.2 KiB
C
59 lines
2.2 KiB
C
/*
|
|
* Copyright (c) 2017 Ronald S. Bultje <rsbultje@gmail.com>
|
|
* Copyright (c) 2017 Ashish Pratap Singh <ashk43712@gmail.com>
|
|
*
|
|
* 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_VMAF_MOTION_H
|
|
#define AVFILTER_VMAF_MOTION_H
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include "video.h"
|
|
|
|
typedef struct VMAFMotionDSPContext {
|
|
uint64_t (*sad)(const uint16_t *img1, const uint16_t *img2, int w, int h,
|
|
ptrdiff_t img1_stride, ptrdiff_t img2_stride);
|
|
void (*convolution_x)(const uint16_t *filter, int filt_w, const uint16_t *src,
|
|
uint16_t *dst, int w, int h, ptrdiff_t src_stride,
|
|
ptrdiff_t dst_stride);
|
|
void (*convolution_y)(const uint16_t *filter, int filt_w, const uint8_t *src,
|
|
uint16_t *dst, int w, int h, ptrdiff_t src_stride,
|
|
ptrdiff_t dst_stride);
|
|
} VMAFMotionDSPContext;
|
|
|
|
void ff_vmafmotion_init_x86(VMAFMotionDSPContext *dsp);
|
|
|
|
typedef struct VMAFMotionData {
|
|
uint16_t filter[5];
|
|
int width;
|
|
int height;
|
|
ptrdiff_t stride;
|
|
uint16_t *blur_data[2 /* cur, prev */];
|
|
uint16_t *temp_data;
|
|
double motion_sum;
|
|
uint64_t nb_frames;
|
|
VMAFMotionDSPContext vmafdsp;
|
|
} VMAFMotionData;
|
|
|
|
int ff_vmafmotion_init(VMAFMotionData *data, int w, int h, enum AVPixelFormat fmt);
|
|
double ff_vmafmotion_process(VMAFMotionData *data, AVFrame *frame);
|
|
double ff_vmafmotion_uninit(VMAFMotionData *data);
|
|
|
|
#endif /* AVFILTER_VMAF_MOTION_H */
|