/*
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
 *
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at https://www.aomedia.org/license/software-license. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at https://www.aomedia.org/license/patent-license.
 */

#ifndef AOM_DSP_PSNR_H_
#define AOM_DSP_PSNR_H_

#include "EbPictureBufferDesc.h"
#include "EbDefinitions.h"
#include "EbUtility.h"
#define MAX_PSNR 100.0

#ifdef __cplusplus
extern "C" {
#endif

typedef struct PsnrStats {
    double   psnr[4]; // total/y/u/v
    uint64_t sse[4]; // total/y/u/v
    uint32_t samples[4]; // total/y/u/v
} PsnrStats;

/*!\brief Converts SSE to PSNR
     *
     * Converts sum of squared errros (SSE) to peak signal-to-noise ratio (PNSR).
     *
     * \param[in]    samples       Number of samples
     * \param[in]    peak          Max sample value
     * \param[in]    sse           Sum of squared errors
     */
double svt_aom_sse_to_psnr(double samples, double peak, double sse);

int64_t svt_aom_get_y_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b, int32_t hstart,
                               int32_t width, int32_t vstart, int32_t height);

int64_t svt_aom_get_y_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

int64_t svt_aom_get_u_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b, int32_t hstart,
                               int32_t width, int32_t vstart, int32_t height);

int64_t svt_aom_get_u_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

int64_t svt_aom_get_v_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b, int32_t hstart,
                               int32_t width, int32_t vstart, int32_t height);

int64_t svt_aom_get_v_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

int64_t svt_aom_highbd_get_y_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b,
                                      int32_t hstart, int32_t width, int32_t vstart,
                                      int32_t height);

int64_t svt_aom_highbd_get_y_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

int64_t svt_aom_highbd_get_u_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b,
                                      int32_t hstart, int32_t width, int32_t vstart,
                                      int32_t height);

int64_t svt_aom_highbd_get_u_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

int64_t svt_aom_highbd_get_v_sse_part(const Yv12BufferConfig *a, const Yv12BufferConfig *b,
                                      int32_t hstart, int32_t width, int32_t vstart,
                                      int32_t height);

int64_t svt_aom_highbd_get_v_sse(const Yv12BufferConfig *a, const Yv12BufferConfig *b);

double aom_psnrhvs(const Yv12BufferConfig *source, const Yv12BufferConfig *dest, double *phvs_y,
                   double *phvs_u, double *phvs_v, uint32_t bd, uint32_t in_bd);

#ifdef __cplusplus
} // extern "C"
#endif
#endif // AOM_DSP_PSNR_H_