1a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora// Copyright 2011 Google Inc. All Rights Reserved. 27c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 37c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// This code is licensed under the same terms as WebM: 47c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Software License Agreement: http://www.webmproject.org/license/software/ 57c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Additional IP Rights Grant: http://www.webmproject.org/license/additional/ 67c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// ----------------------------------------------------------------------------- 77c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 87c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Selecting filter level 97c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Author: somnath@google.com (Somnath Banerjee) 117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 12a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora#include "./vp8enci.h" 137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraextern "C" { 167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#endif 177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// NOTE: clip1, tables and InitTables are repeated entries of dsp.c 197c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic uint8_t abs0[255 + 255 + 1]; // abs(i) 207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic uint8_t abs1[255 + 255 + 1]; // abs(i)>>1 217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic int8_t sclip1[1020 + 1020 + 1]; // clips [-1020, 1020] to [-128, 127] 227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic int8_t sclip2[112 + 112 + 1]; // clips [-112, 112] to [-16, 15] 237c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic uint8_t clip1[255 + 510 + 1]; // clips [-255,510] to [0,255] 247c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 257c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic int tables_ok = 0; 267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void InitTables(void) { 287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (!tables_ok) { 297c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i; 307c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = -255; i <= 255; ++i) { 317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora abs0[255 + i] = (i < 0) ? -i : i; 327c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora abs1[255 + i] = abs0[255 + i] >> 1; 337c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = -1020; i <= 1020; ++i) { 357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora sclip1[1020 + i] = (i < -128) ? -128 : (i > 127) ? 127 : i; 367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = -112; i <= 112; ++i) { 387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora sclip2[112 + i] = (i < -16) ? -16 : (i > 15) ? 15 : i; 397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = -255; i <= 255 + 255; ++i) { 417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora clip1[255 + i] = (i < 0) ? 0 : (i > 255) ? 255 : i; 427c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 437c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora tables_ok = 1; 447c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 457c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 47a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 487c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Edge filtering functions 497c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 507c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 4 pixels in, 2 pixels out 51a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void do_filter2(uint8_t* p, int step) { 527c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step]; 537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a = 3 * (q0 - p0) + sclip1[1020 + p1 - q1]; 547c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a1 = sclip2[112 + ((a + 4) >> 3)]; 557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a2 = sclip2[112 + ((a + 3) >> 3)]; 567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[-step] = clip1[255 + p0 + a2]; 577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[ 0] = clip1[255 + q0 - a1]; 587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// 4 pixels in, 4 pixels out 61a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void do_filter4(uint8_t* p, int step) { 627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step]; 637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a = 3 * (q0 - p0); 647c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a1 = sclip2[112 + ((a + 4) >> 3)]; 657c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a2 = sclip2[112 + ((a + 3) >> 3)]; 667c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int a3 = (a1 + 1) >> 1; 677c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[-2*step] = clip1[255 + p1 + a3]; 687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[- step] = clip1[255 + p0 + a2]; 697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[ 0] = clip1[255 + q0 - a1]; 707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p[ step] = clip1[255 + q1 - a3]; 717c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 727c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 737c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// high edge-variance 74a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE int hev(const uint8_t* p, int step, int thresh) { 757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step]; 767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return (abs0[255 + p1 - p0] > thresh) || (abs0[255 + q1 - q0] > thresh); 777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 787c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 79a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE int needs_filter(const uint8_t* p, int step, int thresh) { 807c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int p1 = p[-2*step], p0 = p[-step], q0 = p[0], q1 = p[step]; 817c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return (2 * abs0[255 + p0 - q0] + abs1[255 + p1 - q1]) <= thresh; 827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 84a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE int needs_filter2(const uint8_t* p, 85a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int step, int t, int it) { 867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int p3 = p[-4*step], p2 = p[-3*step], p1 = p[-2*step], p0 = p[-step]; 877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int q0 = p[0], q1 = p[step], q2 = p[2*step], q3 = p[3*step]; 887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if ((2 * abs0[255 + p0 - q0] + abs1[255 + p1 - q1]) > t) 897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return 0; 907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return abs0[255 + p3 - p2] <= it && abs0[255 + p2 - p1] <= it && 917c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora abs0[255 + p1 - p0] <= it && abs0[255 + q3 - q2] <= it && 927c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora abs0[255 + q2 - q1] <= it && abs0[255 + q1 - q0] <= it; 937c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 947c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 95a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Simple In-loop filtering (Paragraph 15.2) 977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void SimpleVFilter16(uint8_t* p, int stride, int thresh) { 997c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i; 1007c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = 0; i < 16; ++i) { 1017c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (needs_filter(p + i, stride, thresh)) { 1027c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora do_filter2(p + i, stride); 1037c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1047c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void SimpleHFilter16(uint8_t* p, int stride, int thresh) { 1087c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i; 1097c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = 0; i < 16; ++i) { 1107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (needs_filter(p + i * stride, 1, thresh)) { 1117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora do_filter2(p + i * stride, 1); 1127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void SimpleVFilter16i(uint8_t* p, int stride, int thresh) { 1177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int k; 1187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (k = 3; k > 0; --k) { 1197c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p += 4 * stride; 1207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora SimpleVFilter16(p, stride, thresh); 1217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1237c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1247c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void SimpleHFilter16i(uint8_t* p, int stride, int thresh) { 1257c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int k; 1267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (k = 3; k > 0; --k) { 1277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p += 4; 1287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora SimpleHFilter16(p, stride, thresh); 1297c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1307c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 132a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 1337c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Complex In-loop filtering (Paragraph 15.3) 1347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 135a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic WEBP_INLINE void FilterLoop24(uint8_t* p, 136a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int hstride, int vstride, int size, 137a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int thresh, int ithresh, int hev_thresh) { 1387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora while (size-- > 0) { 1397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (needs_filter2(p, hstride, thresh, ithresh)) { 1407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (hev(p, hstride, hev_thresh)) { 1417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora do_filter2(p, hstride); 1427c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } else { 1437c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora do_filter4(p, hstride); 1447c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1457c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p += vstride; 1477c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1487c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1497c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1507c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// on three inner edges 1517c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void VFilter16i(uint8_t* p, int stride, 1527c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int thresh, int ithresh, int hev_thresh) { 1537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int k; 1547c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (k = 3; k > 0; --k) { 1557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p += 4 * stride; 1567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(p, stride, 1, 16, thresh, ithresh, hev_thresh); 1577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void HFilter16i(uint8_t* p, int stride, 1617c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int thresh, int ithresh, int hev_thresh) { 1627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int k; 1637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (k = 3; k > 0; --k) { 1647c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora p += 4; 1657c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(p, 1, stride, 16, thresh, ithresh, hev_thresh); 1667c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 1677c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void VFilter8i(uint8_t* u, uint8_t* v, int stride, 1707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int thresh, int ithresh, int hev_thresh) { 1717c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(u + 4 * stride, stride, 1, 8, thresh, ithresh, hev_thresh); 1727c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(v + 4 * stride, stride, 1, 8, thresh, ithresh, hev_thresh); 1737c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1747c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void HFilter8i(uint8_t* u, uint8_t* v, int stride, 1767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int thresh, int ithresh, int hev_thresh) { 1777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(u + 4, 1, stride, 8, thresh, ithresh, hev_thresh); 1787c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora FilterLoop24(v + 4, 1, stride, 8, thresh, ithresh, hev_thresh); 1797c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 1807c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 181a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 1827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncVFilter16i)(uint8_t*, int, int, int, int) = VFilter16i; 1847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncHFilter16i)(uint8_t*, int, int, int, int) = HFilter16i; 1857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncVFilter8i)(uint8_t*, uint8_t*, int, int, int, int) = VFilter8i; 1867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncHFilter8i)(uint8_t*, uint8_t*, int, int, int, int) = HFilter8i; 1877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncSimpleVFilter16i)(uint8_t*, int, int) = SimpleVFilter16i; 1897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid (*VP8EncSimpleHFilter16i)(uint8_t*, int, int) = SimpleHFilter16i; 1907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 191a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 1927c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Paragraph 15.4: compute the inner-edge filtering strength 1937c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 1947c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic int GetILevel(int sharpness, int level) { 1957c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (sharpness > 0) { 1967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (sharpness > 4) { 1977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora level >>= 2; 1987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } else { 1997c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora level >>= 1; 2007c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2017c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (level > 9 - sharpness) { 2027c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora level = 9 - sharpness; 2037c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2047c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (level < 1) level = 1; 2067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return level; 2077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 2087c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2097c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic void DoFilter(const VP8EncIterator* const it, int level) { 2107c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const VP8Encoder* const enc = it->enc_; 2117c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int ilevel = GetILevel(enc->config_->filter_sharpness, level); 2127c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int limit = 2 * level + ilevel; 2137c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2147c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t* const y_dst = it->yuv_out2_ + Y_OFF; 2157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t* const u_dst = it->yuv_out2_ + U_OFF; 2167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora uint8_t* const v_dst = it->yuv_out2_ + V_OFF; 2177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // copy current block to yuv_out2_ 2197c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora memcpy(y_dst, it->yuv_out_, YUV_SIZE * sizeof(uint8_t)); 2207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (enc->filter_hdr_.simple_ == 1) { // simple 2227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncSimpleHFilter16i(y_dst, BPS, limit); 2237c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncSimpleVFilter16i(y_dst, BPS, limit); 2247c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } else { // complex 2257c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int hev_thresh = (level >= 40) ? 2 : (level >= 15) ? 1 : 0; 2267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncHFilter16i(y_dst, BPS, limit, ilevel, hev_thresh); 2277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncHFilter8i(u_dst, v_dst, BPS, limit, ilevel, hev_thresh); 2287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncVFilter16i(y_dst, BPS, limit, ilevel, hev_thresh); 2297c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8EncVFilter8i(u_dst, v_dst, BPS, limit, ilevel, hev_thresh); 2307c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 2327c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 233a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 2347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// SSIM metric 2357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 2367c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroraenum { KERNEL = 3 }; 237a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic const double kMinValue = 1.e-10; // minimal threshold 238a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 239a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8SSIMAddStats(const DistoStats* const src, DistoStats* const dst) { 240a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->w += src->w; 241a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->xm += src->xm; 242a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->ym += src->ym; 243a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->xxm += src->xxm; 244a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->xym += src->xym; 245a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora dst->yym += src->yym; 246a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora} 247a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 248a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arorastatic void VP8SSIMAccumulate(const uint8_t* src1, int stride1, 249a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const uint8_t* src2, int stride2, 250a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int xo, int yo, int W, int H, 251a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora DistoStats* const stats) { 2527c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int ymin = (yo - KERNEL < 0) ? 0 : yo - KERNEL; 2537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int ymax = (yo + KERNEL > H - 1) ? H - 1 : yo + KERNEL; 2547c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int xmin = (xo - KERNEL < 0) ? 0 : xo - KERNEL; 2557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int xmax = (xo + KERNEL > W - 1) ? W - 1 : xo + KERNEL; 2567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int x, y; 2577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora src1 += ymin * stride1; 2587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora src2 += ymin * stride2; 2597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (y = ymin; y <= ymax; ++y, src1 += stride1, src2 += stride2) { 2607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (x = xmin; x <= xmax; ++x) { 2617c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int s1 = src1[x]; 2627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int s2 = src2[x]; 2637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->w += 1; 2647c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->xm += s1; 2657c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->ym += s2; 2667c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->xxm += s1 * s1; 2677c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->xym += s1 * s2; 2687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora stats->yym += s2 * s2; 2697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 2717c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 2727c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 273a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroradouble VP8SSIMGet(const DistoStats* const stats) { 2747c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const double xmxm = stats->xm * stats->xm; 2757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const double ymym = stats->ym * stats->ym; 2767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const double xmym = stats->xm * stats->ym; 2777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const double w2 = stats->w * stats->w; 2787c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double sxx = stats->xxm * stats->w - xmxm; 2797c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double syy = stats->yym * stats->w - ymym; 2807c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double sxy = stats->xym * stats->w - xmym; 2817c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double C1, C2; 2827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double fnum; 2837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double fden; 2847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // small errors are possible, due to rounding. Clamp to zero. 2857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (sxx < 0.) sxx = 0.; 2867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (syy < 0.) syy = 0.; 2877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora C1 = 6.5025 * w2; 2887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora C2 = 58.5225 * w2; 2897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora fnum = (2 * xmym + C1) * (2 * sxy + C2); 2907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora fden = (xmxm + ymym + C1) * (sxx + syy + C2); 291a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora return (fden != 0.) ? fnum / fden : kMinValue; 292a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora} 293a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 294a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroradouble VP8SSIMGetSquaredError(const DistoStats* const s) { 295a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora if (s->w > 0.) { 296a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const double iw2 = 1. / (s->w * s->w); 297a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const double sxx = s->xxm * s->w - s->xm * s->xm; 298a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const double syy = s->yym * s->w - s->ym * s->ym; 299a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const double sxy = s->xym * s->w - s->xm * s->ym; 300a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const double SSE = iw2 * (sxx + syy - 2. * sxy); 301a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora if (SSE > kMinValue) return SSE; 302a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora } 303a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora return kMinValue; 304a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora} 305a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora 306a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Aroravoid VP8SSIMAccumulatePlane(const uint8_t* src1, int stride1, 307a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora const uint8_t* src2, int stride2, 308a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int W, int H, DistoStats* const stats) { 309a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora int x, y; 310a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora for (y = 0; y < H; ++y) { 311a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora for (x = 0; x < W; ++x) { 312a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora VP8SSIMAccumulate(src1, stride1, src2, stride2, x, y, W, H, stats); 313a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora } 314a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora } 3157c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 3167c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3177c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arorastatic double GetMBSSIM(const uint8_t* yuv1, const uint8_t* yuv2) { 3187c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int x, y; 319a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora DistoStats s = { .0, .0, .0, .0, .0, .0 }; 3207c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3217c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // compute SSIM in a 10 x 10 window 3227c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (x = 3; x < 13; x++) { 3237c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (y = 3; y < 13; y++) { 324a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora VP8SSIMAccumulate(yuv1 + Y_OFF, BPS, yuv2 + Y_OFF, BPS, x, y, 16, 16, &s); 3257c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3267c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3277c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (x = 1; x < 7; x++) { 3287c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (y = 1; y < 7; y++) { 329a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora VP8SSIMAccumulate(yuv1 + U_OFF, BPS, yuv2 + U_OFF, BPS, x, y, 8, 8, &s); 330a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora VP8SSIMAccumulate(yuv1 + V_OFF, BPS, yuv2 + V_OFF, BPS, x, y, 8, 8, &s); 3317c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3327c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 333a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora return VP8SSIMGet(&s); 3347c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 3357c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 336a2415724fb3466168b2af5b08bd94ba732c0e753Vikas Arora//------------------------------------------------------------------------------ 3377c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// Exposed APIs: Encoder should call the following 3 functions to adjust 3387c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora// loop filter strength 3397c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3407c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8InitFilter(VP8EncIterator* const it) { 3417c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int s, i; 3427c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (!it->lf_stats_) return; 3437c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3447c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora InitTables(); 3457c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (s = 0; s < NUM_MB_SEGMENTS; s++) { 3467c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = 0; i < MAX_LF_LEVELS; i++) { 3477c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora (*it->lf_stats_)[s][i] = 0; 3487c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3497c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3507c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 3517c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3527c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8StoreFilterStats(VP8EncIterator* const it) { 3537c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int d; 3547c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int s = it->mb_->segment_; 3557c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int level0 = it->enc_->dqm_[s].fstrength_; // TODO: ref_lf_delta[] 3567c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3577c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // explore +/-quant range of values around level0 3587c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int delta_min = -it->enc_->dqm_[s].quant_; 3597c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int delta_max = it->enc_->dqm_[s].quant_; 3607c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int step_size = (delta_max - delta_min >= 4) ? 4 : 1; 3617c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3627c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (!it->lf_stats_) return; 3637c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3647c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // NOTE: Currently we are applying filter only across the sublock edges 3657c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // There are two reasons for that. 3667c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // 1. Applying filter on macro block edges will change the pixels in 3677c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // the left and top macro blocks. That will be hard to restore 3687c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // 2. Macro Blocks on the bottom and right are not yet compressed. So we 3697c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // cannot apply filter on the right and bottom macro block edges. 3707c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (it->mb_->type_ == 1 && it->mb_->skip_) return; 3717c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3727c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // Always try filter level zero 3737c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora (*it->lf_stats_)[s][0] += GetMBSSIM(it->yuv_in_, it->yuv_out_); 3747c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3757c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (d = delta_min; d <= delta_max; d += step_size) { 3767c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const int level = level0 + d; 3777c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (level <= 0 || level >= MAX_LF_LEVELS) { 3787c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora continue; 3797c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3807c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora DoFilter(it, level); 3817c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora (*it->lf_stats_)[s][level] += GetMBSSIM(it->yuv_in_, it->yuv_out2_); 3827c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3837c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 3847c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3857c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Aroravoid VP8AdjustFilterStrength(VP8EncIterator* const it) { 3867c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int s; 3877c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora VP8Encoder* const enc = it->enc_; 3887c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 3897c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (!it->lf_stats_) { 3907c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora return; 3917c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 3927c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (s = 0; s < NUM_MB_SEGMENTS; s++) { 3937c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora int i, best_level = 0; 3947c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora // Improvement over filter level 0 should be at least 1e-5 (relatively) 3957c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora double best_v = 1.00001 * (*it->lf_stats_)[s][0]; 3967c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora for (i = 1; i < MAX_LF_LEVELS; i++) { 3977c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora const double v = (*it->lf_stats_)[s][i]; 3987c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora if (v > best_v) { 3997c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora best_v = v; 4007c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora best_level = i; 4017c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 4027c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 4037c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora enc->dqm_[s].fstrength_ = best_level; 4047c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora } 4057c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} 4067c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora 4077c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#if defined(__cplusplus) || defined(c_plusplus) 4087c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora} // extern "C" 4097c970a0a679089e416c5887cf7fcece15a70bfa4Vikas Arora#endif 410