#include <gtest/gtest.h>#include "cpu.h"#include "cpu_core.h"#include "util.h"#include "macros.h"#include "IWelsVP.h"#include "vaacalculation.h" using namespace nsWelsVP; void VAACalcSadSsd_ref (const uint8_t* pCurData, const uint8_t* pRefData, int32_t iPicWidth, int32_t iPicHeight,                        int32_t iPicStride,                        int32_t* pFrameSad, int32_t* pSad8x8, int32_t* pSum16x16, int32_t* psqsum16x16, int32_t* psqdiff16x16) {  const uint8_t* tmp_ref = pRefData;  const uint8_t* tmp_cur = pCurData;  int32_t iMbWidth = (iPicWidth >> 4);  int32_t mb_height = (iPicHeight >> 4);  int32_t mb_index = 0;  int32_t pic_stride_x8 = iPicStride << 3;  int32_t step = (iPicStride << 4) - iPicWidth;   *pFrameSad = 0;  for (int32_t i = 0; i < mb_height; i ++) {    for (int32_t j = 0; j < iMbWidth; j ++) {      int32_t k, l;      int32_t l_sad, l_sqdiff, l_sum, l_sqsum;      const uint8_t* tmp_cur_row;      const uint8_t* tmp_ref_row;       pSum16x16[mb_index] = 0;      psqsum16x16[mb_index] = 0;      psqdiff16x16[mb_index] = 0;       l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur;      tmp_ref_row = tmp_ref;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sqdiff += diff * diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 0] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;       l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + 8;      tmp_ref_row = tmp_ref + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sqdiff += diff * diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 1] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;       l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8;      tmp_ref_row = tmp_ref + pic_stride_x8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sqdiff += diff * diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 2] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;       l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8 + 8;      tmp_ref_row = tmp_ref + pic_stride_x8 + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sqdiff += diff * diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 3] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;        tmp_ref += 16;      tmp_cur += 16;      ++mb_index;    }    tmp_ref += step;    tmp_cur += step;  }}void VAACalcSadVar_ref (const uint8_t* pCurData, const uint8_t* pRefData, int32_t iPicWidth, int32_t iPicHeight,                        int32_t iPicStride,                        int32_t* pFrameSad, int32_t* pSad8x8, int32_t* pSum16x16, int32_t* psqsum16x16) {  const uint8_t* tmp_ref = pRefData;  const uint8_t* tmp_cur = pCurData;  int32_t iMbWidth = (iPicWidth >> 4);  int32_t mb_height = (iPicHeight >> 4);  int32_t mb_index = 0;  int32_t pic_stride_x8 = iPicStride << 3;  int32_t step = (iPicStride << 4) - iPicWidth;   *pFrameSad = 0;  for (int32_t i = 0; i < mb_height; i ++) {    for (int32_t j = 0; j < iMbWidth; j ++) {      int32_t k, l;      int32_t l_sad, l_sum, l_sqsum;      const uint8_t* tmp_cur_row;      const uint8_t* tmp_ref_row;       pSum16x16[mb_index] = 0;      psqsum16x16[mb_index] = 0;       l_sad =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur;      tmp_ref_row = tmp_ref;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 0] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;       l_sad =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + 8;      tmp_ref_row = tmp_ref + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 1] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;       l_sad =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8;      tmp_ref_row = tmp_ref + pic_stride_x8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 2] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;       l_sad =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8 + 8;      tmp_ref_row = tmp_ref + pic_stride_x8 + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 3] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;        tmp_ref += 16;      tmp_cur += 16;      ++mb_index;    }    tmp_ref += step;    tmp_cur += step;  }} void VAACalcSad_ref (const uint8_t* pCurData, const uint8_t* pRefData, int32_t iPicWidth, int32_t iPicHeight,                     int32_t iPicStride,                     int32_t* pFrameSad, int32_t* pSad8x8) {  const uint8_t* tmp_ref = pRefData;  const uint8_t* tmp_cur = pCurData;  int32_t iMbWidth = (iPicWidth >> 4);  int32_t mb_height = (iPicHeight >> 4);  int32_t mb_index = 0;  int32_t pic_stride_x8 = iPicStride << 3;  int32_t step = (iPicStride << 4) - iPicWidth;   *pFrameSad = 0;  for (int32_t i = 0; i < mb_height; i ++) {    for (int32_t j = 0; j < iMbWidth; j ++) {      int32_t k, l;      int32_t l_sad;      const uint8_t* tmp_cur_row;      const uint8_t* tmp_ref_row;       l_sad =  0;      tmp_cur_row = tmp_cur;      tmp_ref_row = tmp_ref;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 0] = l_sad;       l_sad =  0;      tmp_cur_row = tmp_cur + 8;      tmp_ref_row = tmp_ref + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 1] = l_sad;       l_sad =  0;      tmp_cur_row = tmp_cur + pic_stride_x8;      tmp_ref_row = tmp_ref + pic_stride_x8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 2] = l_sad;       l_sad =  0;      tmp_cur_row = tmp_cur + pic_stride_x8 + 8;      tmp_ref_row = tmp_ref + pic_stride_x8 + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = WELS_ABS (tmp_cur_row[l] - tmp_ref_row[l]);          l_sad += diff;        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 3] = l_sad;       tmp_ref += 16;      tmp_cur += 16;      ++mb_index;    }    tmp_ref += step;    tmp_cur += step;  }} void VAACalcSadSsdBgd_ref (const uint8_t* pCurData, const uint8_t* pRefData, int32_t iPicWidth, int32_t iPicHeight,                           int32_t iPicStride,                           int32_t* pFrameSad, int32_t* pSad8x8, int32_t* pSum16x16, int32_t* psqsum16x16, int32_t* psqdiff16x16, int32_t* pSd8x8,                           uint8_t* pMad8x8) {  const uint8_t* tmp_ref = pRefData;  const uint8_t* tmp_cur = pCurData;  int32_t iMbWidth = (iPicWidth >> 4);  int32_t mb_height = (iPicHeight >> 4);  int32_t mb_index = 0;  int32_t pic_stride_x8 = iPicStride << 3;  int32_t step = (iPicStride << 4) - iPicWidth;   *pFrameSad = 0;  for (int32_t i = 0; i < mb_height; i ++) {    for (int32_t j = 0; j < iMbWidth; j ++) {      int32_t k, l;      int32_t l_sad, l_sqdiff, l_sum, l_sqsum, l_sd, l_mad;      const uint8_t* tmp_cur_row;      const uint8_t* tmp_ref_row;       pSum16x16[mb_index] = 0;      psqsum16x16[mb_index] = 0;      psqdiff16x16[mb_index] = 0;       l_sd = l_mad = l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur;      tmp_ref_row = tmp_ref;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = tmp_cur_row[l] - tmp_ref_row[l];          int32_t abs_diff = WELS_ABS (diff);           l_sd += diff;          if (abs_diff > l_mad) {            l_mad = abs_diff;          }          l_sad += abs_diff;          l_sqdiff += abs_diff * abs_diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 0] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;      pSd8x8[ (mb_index << 2) + 0] = l_sd;      pMad8x8[ (mb_index << 2) + 0] = l_mad;        l_sd = l_mad = l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + 8;      tmp_ref_row = tmp_ref + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = tmp_cur_row[l] - tmp_ref_row[l];          int32_t abs_diff = WELS_ABS (diff);           l_sd += diff;          if (abs_diff > l_mad) {            l_mad = abs_diff;          }          l_sad += abs_diff;          l_sqdiff += abs_diff * abs_diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 1] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;      pSd8x8[ (mb_index << 2) + 1] = l_sd;      pMad8x8[ (mb_index << 2) + 1] = l_mad;       l_sd = l_mad = l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8;      tmp_ref_row = tmp_ref + pic_stride_x8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = tmp_cur_row[l] - tmp_ref_row[l];          int32_t abs_diff = WELS_ABS (diff);           l_sd += diff;          if (abs_diff > l_mad) {            l_mad = abs_diff;          }          l_sad += abs_diff;          l_sqdiff += abs_diff * abs_diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 2] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;      pSd8x8[ (mb_index << 2) + 2] = l_sd;      pMad8x8[ (mb_index << 2) + 2] = l_mad;       l_sd = l_mad = l_sad =  l_sqdiff =  l_sum =  l_sqsum = 0;      tmp_cur_row = tmp_cur + pic_stride_x8 + 8;      tmp_ref_row = tmp_ref + pic_stride_x8 + 8;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = tmp_cur_row[l] - tmp_ref_row[l];          int32_t abs_diff = WELS_ABS (diff);           l_sd += diff;          if (abs_diff > l_mad) {            l_mad = abs_diff;          }          l_sad += abs_diff;          l_sqdiff += abs_diff * abs_diff;          l_sum += tmp_cur_row[l];          l_sqsum += tmp_cur_row[l] * tmp_cur_row[l];        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 3] = l_sad;      pSum16x16[mb_index] += l_sum;      psqsum16x16[mb_index] += l_sqsum;      psqdiff16x16[mb_index] += l_sqdiff;      pSd8x8[ (mb_index << 2) + 3] = l_sd;      pMad8x8[ (mb_index << 2) + 3] = l_mad;       tmp_ref += 16;      tmp_cur += 16;      ++mb_index;    }    tmp_ref += step;    tmp_cur += step;  }} void VAACalcSadBgd_ref (const uint8_t* pCurData, const uint8_t* pRefData, int32_t iPicWidth, int32_t iPicHeight,                        int32_t iPicStride,                        int32_t* pFrameSad, int32_t* pSad8x8, int32_t* pSd8x8, uint8_t* pMad8x8) {  const uint8_t* tmp_ref = pRefData;  const uint8_t* tmp_cur = pCurData;  int32_t iMbWidth = (iPicWidth >> 4);  int32_t mb_height = (iPicHeight >> 4);  int32_t mb_index = 0;  int32_t pic_stride_x8 = iPicStride << 3;  int32_t step = (iPicStride << 4) - iPicWidth;   *pFrameSad = 0;  for (int32_t i = 0; i < mb_height; i ++) {    for (int32_t j = 0; j < iMbWidth; j ++) {      int32_t k, l;      int32_t l_sad, l_sd, l_mad;      const uint8_t* tmp_cur_row;      const uint8_t* tmp_ref_row;       l_mad = l_sd = l_sad =  0;      tmp_cur_row = tmp_cur;      tmp_ref_row = tmp_ref;      for (k = 0; k < 8; k ++) {        for (l = 0; l < 8; l ++) {          int32_t diff = tmp_cur_row[l] - tmp_ref_row[l];          int32_t abs_diff = WELS_ABS (diff);          l_sd += diff;          l_sad += abs_diff;          if (abs_diff > l_mad) {            l_mad = abs_diff;          }        }        tmp_cur_row += iPicStride;        tmp_ref_row += iPicStride;      }      *pFrameSad += l_sad;      pSad8x8[ (mb_index << 2) + 0] = l_sad;      pSd8x8[ (mb_index << 2) + 0] = l_sd;      pMad8x8[ (mb_index << 2) + 0] = l_mad;       l_mad = l_sd = l_sad =  0;      tmp_cur_row = tmp_cur + 8;      tmp_ref_row = tmp_ref + 8;      for (k = 0; k < 8; k ++) {

Lines 1–500 of 847. Load more