Merge pull request #12 from bygonexf/main

feat: add 10-bit 'sao' arm64 implementation
dujiangpku · Dec 29, 2021 · ca41581 · ca41581
2 parents f7299a0 + d0dd14b
commit ca41581
Show file tree

Hide file tree

Showing 6 changed files with 2,397 additions and 3 deletions.
diff --git a/build/android/app/src/main/jni/src/armv8/arm64.c b/build/android/app/src/main/jni/src/armv8/arm64.c
@@ -248,6 +248,9 @@ void uavs3e_funs_init_arm64()
     uavs3e_funs_handle.pel_avrg[5] = uavs3e_pel_avrg_128_arm64;
 
     */
+
+    uavs3e_funs_handle.sao = uavs3e_sao_on_lcu_arm64;
+    //todo uavs3e_funs_handle.sao_stat
 #endif
 
 }

diff --git a/build/android/app/src/main/jni/src/armv8/arm64.h b/build/android/app/src/main/jni/src/armv8/arm64.h
@@ -6,6 +6,7 @@
 #ifndef __ARM64_H__
 #define __ARM64_H__
 #include "../../inc/com_util.h"
+#include "../../inc/com_define.h"
 
 void uavs3e_if_cpy_w4_arm64(const pel *src, int i_src, pel *dst, int i_dst, int width, int height);
 void uavs3e_if_cpy_w8_arm64(const pel *src, int i_src, pel *dst, int i_dst, int width, int height);
@@ -57,8 +58,8 @@ void uavs3e_deblock_ver_luma_arm64(pel *src, int stride, int alpha, int beta, in
 void uavs3e_deblock_hor_luma_arm64(pel *src, int stride, int alpha, int beta, int flt_flag);
 void uavs3e_deblock_ver_chroma_arm64(pel *src_u, pel *src_v, int stride, int alpha_u, int beta_u, int alpha_v, int beta_v, int flt_flag);
 void uavs3e_deblock_hor_chroma_arm64(pel *src_u, pel *src_v, int stride, int alpha_u, int beta_u, int alpha_v, int beta_v, int flt_flag);
-void uavs3e_sao_on_lcu_arm64(pel *src, int i_src, pel *dst, int i_dst, com_sao_param_t *sao_params, int smb_pix_height,
-                           int smb_pix_width, int smb_available_left, int smb_available_right, int smb_available_up, int smb_available_down, int sample_bit_depth);
+void uavs3e_sao_on_lcu_arm64(pel *src, int i_src, pel *dst, int i_dst, com_sao_param_t *sao_params, int height,
+                             int width, int available_left, int available_right, int available_up, int available_down, int sample_bit_depth);
 void uavs3e_sao_on_lcu_chroma_arm64(pel *src, int i_src, pel *dst, int i_dst, com_sao_param_t *sao_params, int smb_pix_height,
                                   int smb_pix_width, int smb_available_left, int smb_available_right, int smb_available_up, int smb_available_down, int sample_bit_depth);
 void uavs3e_alf_filter_block_arm64(pel *dst, int i_dst, pel *src, int i_src, int lcu_width, int lcu_height, int *coef, int sample_bit_depth);
-Original file line number
+Diff line change
@@ Expand Up / @@ -248,6 +248,9 @@ void uavs3e_funs_init_arm64() @@
         uavs3e_funs_handle.pel_avrg[5] = uavs3e_pel_avrg_128_arm64;
         */
+        uavs3e_funs_handle.sao = uavs3e_sao_on_lcu_arm64;
+        //todo uavs3e_funs_handle.sao_stat
     #endif
     }
@@ Expand Down @@