开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > Audio > 查看源码
img_dist_snr.c
资源名称：chapter15.rar　[点击查看]
上传用户：hjq518
上传日期：2021-12-09
资源大小：5084k
文件大小：2k
源码类别：

Audio

开发平台：

Visual C++

img_dist_snr.c：源码内容
  2. /*!
  3.  *************************************************************************************
  4.  * file img_dist_snr.c
  5.  *
  6.  * brief
  7.  *    Compute signal to noise ratio (SNR) between the encoded image and the reference image
  8.  *
  9.  * author
  10.  *    Main contributors (see contributors.h for copyright, address and affiliation details)
  11.  *     - Woo-Shik Kim                    <wooshik.kim@usc.edu>
  12.  *     - Alexis Michael Tourapis         <alexismt@ieee.org>
  13.  *************************************************************************************
  14.  */
  15. #include "contributors.h"
  17. #include <math.h>
  19. #include "global.h"
  20. #include "img_distortion.h"
  21. #include "enc_statistics.h"
  23. /*!
  24.  ************************************************************************
  25.  * brief
  26.  *    Find SNR for all three components
  27.  ************************************************************************
  28.  */
  29. void find_snr(ImageStructure *imgREF, ImageStructure *imgSRC, DistMetric *metricSSE, DistMetric *metricPSNR)
  30. {
  31.   FrameFormat *format = &imgREF->format;
  32.   // Luma.
  33.   metricSSE ->value[0] = (float) compute_SSE(imgREF->data[0], imgSRC->data[0], 0, 0, format->height, format->width);
  34.   metricPSNR->value[0] = psnr(format->max_value_sq[0], format->size_cmp[0], metricSSE->value[0]);
  36.   // Chroma.
  37.   if (format->yuv_format != YUV400)
  38.   {   
  39.     metricSSE ->value[1] = (float) compute_SSE(imgREF->data[1], imgSRC->data[1], 0, 0, format->height_cr, format->width_cr);
  40.     metricPSNR->value[1] = psnr(format->max_value_sq[1], format->size_cmp[1], metricSSE->value[1]);
  41.     metricSSE ->value[2] = (float) compute_SSE(imgREF->data[2], imgSRC->data[2], 0, 0, format->height_cr, format->width_cr);
  42.     metricPSNR->value[2] = psnr(format->max_value_sq[2], format->size_cmp[2], metricSSE->value[2]);
  43.   }
  44.    
  45.   {
  46.     accumulate_average(metricSSE,  dist->frame_ctr);
  47.     accumulate_average(metricPSNR, dist->frame_ctr);
  49.     accumulate_avslice(metricSSE,  img->type, stats->frame_ctr[img->type]);
  50.     accumulate_avslice(metricPSNR, img->type, stats->frame_ctr[img->type]);
  51.   }
  52. }