//****************************************************************************** //* //* Copyright (C) 2015 The Android Open Source Project //* //* Licensed under the Apache License, Version 2.0 (the "License"); //* you may not use this file except in compliance with the License. //* You may obtain a copy of the License at: //* //* http://www.apache.org/licenses/LICENSE-2.0 //* //* Unless required by applicable law or agreed to in writing, software //* distributed under the License is distributed on an "AS IS" BASIS, //* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. //* See the License for the specific language governing permissions and //* limitations under the License. //* //***************************************************************************** //* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore //*/ ///** //****************************************************************************** //* @file //* ih264_inter_pred_chroma_av8.s //* //* @brief //* Contains function definitions for inter prediction interpolation. //* //* @author //* Ittaim //* //* @par List of Functions: //* //* - ih264_inter_pred_chroma_av8() //* //* @remarks //* None //* //******************************************************************************* //*/ ///* All the functions here are replicated from ih264_inter_pred_filters.c // ///** ///** ///** // ///** //******************************************************************************* //* //* @brief //* Interprediction chroma filter //* //* @par Description: //* Applies filtering to chroma samples as mentioned in //* sec 8.4.2.2.2 titled "chroma sample interpolation process" //* //* @param[in] pu1_src //* UWORD8 pointer to the source containing alternate U and V samples //* //* @param[out] pu1_dst //* UWORD8 pointer to the destination //* //* @param[in] src_strd //* integer source stride //* //* @param[in] dst_strd //* integer destination stride //* //* @param[in]uc_dx //* dx value where the sample is to be produced(refer sec 8.4.2.2.2 ) //* //* @param[in] uc_dy //* dy value where the sample is to be produced(refer sec 8.4.2.2.2 ) //* //* @param[in] ht //* integer height of the array //* //* @param[in] wd //* integer width of the array //* //* @returns //* //* @remarks //* None //* //******************************************************************************* //*/ //void ih264_inter_pred_chroma(UWORD8 *pu1_src, // UWORD8 *pu1_dst, // WORD32 src_strd, // WORD32 dst_strd, // WORD32 u1_dx, // WORD32 u1_dy, // WORD32 ht, // WORD32 wd) //**************Variables Vs Registers***************************************** // x0 => *pu1_src // x1 => *pu1_dst // w2 => src_strd // w3 => dst_strd // w4 => u1_dx // w5 => u1_dy // w6 => height // w7 => width // .text .p2align 2 .include "ih264_neon_macros.s" .global ih264_inter_pred_chroma_av8 ih264_inter_pred_chroma_av8: // STMFD sp!, {x4-x12, x14} //store register values to stack push_v_regs stp x19, x20, [sp, #-16]! sxtw x2, w2 sxtw x3, w3 sxtw x4, w4 sxtw x5, w5 sxtw x6, w6 sxtw x7, w7 sub x20, x4, #8 //8-u1_dx neg x8, x20 sub x20, x5, #8 //8-u1_dy neg x9, x20 mul x10, x8, x9 // mul x11, x4, x9 // dup v28.8b, w10 dup v29.8b, w11 mul x10, x8, x5 // mul x11, x4, x5 // dup v30.8b, w10 dup v31.8b, w11 subs x12, x7, #2 //if wd=4 branch to loop_4 beq loop_2 subs x12, x7, #4 //if wd=8 branch to loop_8 beq loop_4 loop_8: ld1 {v0.8b, v1.8b, v2.8b}, [x0], x2 //// Load row0 ; ext v3.8b, v0.8b , v1.8b , #2 ld1 {v5.8b, v6.8b, v7.8b}, [x0], x2 //// Load row1; umull v20.8h, v0.8b, v28.8b ext v8.8b, v5.8b , v6.8b , #2 umlal v20.8h, v3.8b, v29.8b ext v9.8b, v6.8b , v7.8b , #2 umlal v20.8h, v5.8b, v30.8b ext v4.8b, v1.8b , v2.8b , #2 umlal v20.8h, v8.8b, v31.8b sqrshrun v26.8b, v20.8h, #6 umull v22.8h, v1.8b, v28.8b ld1 {v10.8b, v11.8b, v12.8b}, [x0], x2 //// Load row2 ; umlal v22.8h, v4.8b, v29.8b ext v13.8b, v10.8b , v11.8b , #2 umlal v22.8h, v6.8b, v30.8b ext v14.8b, v11.8b , v12.8b , #2 umlal v22.8h, v9.8b, v31.8b sqrshrun v27.8b, v22.8h, #6 umull v24.8h, v5.8b, v28.8b st1 { v26.8b, v27.8b}, [x1], x3 ////Store dest row umlal v24.8h, v8.8b, v29.8b ld1 {v0.8b, v1.8b, v2.8b}, [x0], x2 //// Load row3 ; umlal v24.8h, v10.8b, v30.8b ext v3.8b, v0.8b , v1.8b , #2 umlal v24.8h, v13.8b, v31.8b ext v4.8b, v1.8b , v2.8b , #2 umull v16.8h, v6.8b, v28.8b sqrshrun v18.8b, v24.8h, #6 umlal v16.8h, v9.8b, v29.8b umlal v16.8h, v11.8b, v30.8b umlal v16.8h, v14.8b, v31.8b sqrshrun v19.8b, v16.8h, #6 st1 {v18.8b, v19.8b}, [x1], x3 // store row 1 umull v20.8h, v10.8b, v28.8b umlal v20.8h, v13.8b, v29.8b umlal v20.8h, v0.8b, v30.8b umlal v20.8h, v3.8b, v31.8b sqrshrun v26.8b, v20.8h, #6 umull v24.8h, v11.8b, v28.8b ld1 {v5.8b, v6.8b, v7.8b}, [x0], x2 //// Load row4; umlal v24.8h, v14.8b, v29.8b ext v8.8b, v5.8b , v6.8b , #2 umlal v24.8h, v1.8b, v30.8b ext v9.8b, v6.8b , v7.8b , #2 umlal v24.8h, v4.8b, v31.8b umull v20.8h, v0.8b, v28.8b sqrshrun v27.8b, v24.8h, #6 umlal v20.8h, v3.8b, v29.8b st1 { v26.8b, v27.8b}, [x1], x3 ////Store dest row2 umlal v20.8h, v5.8b, v30.8b umlal v20.8h, v8.8b, v31.8b umull v22.8h, v1.8b, v28.8b umlal v22.8h, v4.8b, v29.8b umlal v22.8h, v6.8b, v30.8b sqrshrun v26.8b, v20.8h, #6 umlal v22.8h, v9.8b, v31.8b subs x12, x6, #4 sqrshrun v27.8b, v22.8h, #6 st1 { v26.8b, v27.8b}, [x1], x3 ////Store dest row3 beq end_func //If ht=4 ld1 {v10.8b, v11.8b, v12.8b}, [x0], x2 //// Load row5 ext v13.8b, v10.8b , v11.8b , #2 umull v24.8h, v5.8b, v28.8b ext v14.8b, v11.8b , v12.8b , #2 ld1 {v0.8b, v1.8b, v2.8b}, [x0], x2 //// Load row6; umlal v24.8h, v8.8b, v29.8b umlal v24.8h, v10.8b, v30.8b umlal v24.8h, v13.8b, v31.8b ext v3.8b, v0.8b , v1.8b , #2 umull v16.8h, v6.8b, v28.8b sqrshrun v18.8b, v24.8h, #6 umlal v16.8h, v9.8b, v29.8b umlal v16.8h, v11.8b, v30.8b umlal v16.8h, v14.8b, v31.8b ext v4.8b, v1.8b , v2.8b , #2 sqrshrun v19.8b, v16.8h, #6 st1 { v18.8b, v19.8b}, [x1], x3 // store row 4 umull v20.8h, v10.8b, v28.8b umlal v20.8h, v13.8b, v29.8b umlal v20.8h, v0.8b, v30.8b umlal v20.8h, v3.8b, v31.8b ld1 {v5.8b, v6.8b, v7.8b}, [x0], x2 //// Load row7; sqrshrun v26.8b, v20.8h, #6 umull v24.8h, v11.8b, v28.8b umlal v24.8h, v14.8b, v29.8b ext v8.8b, v5.8b , v6.8b , #2 umlal v24.8h, v1.8b, v30.8b umlal v24.8h, v4.8b, v31.8b ext v9.8b, v6.8b , v7.8b , #2 sqrshrun v27.8b, v24.8h, #6 st1 {v26.8b, v27.8b}, [x1], x3 ////Store dest row5 umull v20.8h, v0.8b, v28.8b umlal v20.8h, v3.8b, v29.8b umlal v20.8h, v5.8b, v30.8b umlal v20.8h, v8.8b, v31.8b ld1 {v10.8b, v11.8b, v12.8b}, [x0], x2 //// Load row8 ; sqrshrun v26.8b, v20.8h, #6 umull v22.8h, v1.8b, v28.8b umlal v22.8h, v4.8b, v29.8b umlal v22.8h, v6.8b, v30.8b ext v13.8b, v10.8b , v11.8b , #2 umlal v22.8h, v9.8b, v31.8b ext v14.8b, v11.8b , v12.8b , #2 sqrshrun v27.8b, v22.8h, #6 st1 { v26.8b, v27.8b}, [x1], x3 ////Store dest row6 umull v24.8h, v5.8b, v28.8b umlal v24.8h, v8.8b, v29.8b umlal v24.8h, v10.8b, v30.8b umlal v24.8h, v13.8b, v31.8b umull v16.8h, v6.8b, v28.8b sqrshrun v18.8b, v24.8h, #6 umlal v16.8h, v9.8b, v29.8b umlal v16.8h, v11.8b, v30.8b umlal v16.8h, v14.8b, v31.8b sqrshrun v19.8b, v16.8h, #6 st1 { v18.8b, v19.8b}, [x1], x3 // store row 7 b end_func loop_4: ld1 {v0.8b, v1.8b}, [x0], x2 //// Load row0 ; ext v2.8b, v0.8b , v1.8b , #2 ld1 {v3.8b, v4.8b}, [x0], x2 //// Load row1; ext v5.8b, v3.8b , v4.8b , #2 umull v20.8h, v0.8b, v28.8b umlal v20.8h, v2.8b, v29.8b umlal v20.8h, v3.8b, v30.8b umlal v20.8h, v5.8b, v31.8b ld1 {v6.8b, v7.8b}, [x0], x2 //// Load row2 sqrshrun v26.8b, v20.8h, #6 ext v8.8b, v6.8b , v7.8b , #2 st1 {v26.8b}, [x1], x3 ////Store dest row0 umull v22.8h, v3.8b, v28.8b umlal v22.8h, v5.8b, v29.8b umlal v22.8h, v6.8b, v30.8b umlal v22.8h, v8.8b, v31.8b subs x12, x6, #2 sqrshrun v27.8b, v22.8h, #6 st1 {v27.8b}, [x1], x3 ////Store dest row1 beq end_func //If ht=2 ld1 {v9.8b, v10.8b}, [x0], x2 //// Load row3; ext v11.8b, v9.8b , v10.8b , #2 umull v24.8h, v6.8b, v28.8b umlal v24.8h, v8.8b, v29.8b umlal v24.8h, v9.8b, v30.8b umlal v24.8h, v11.8b, v31.8b ld1 {v0.8b, v1.8b}, [x0], x2 //// Load row4 ; sqrshrun v16.8b, v24.8h, #6 ext v2.8b, v0.8b , v1.8b , #2 st1 {v16.8b}, [x1], x3 ////Store dest row2 umull v18.8h, v9.8b, v28.8b umlal v18.8h, v11.8b, v29.8b umlal v18.8h, v0.8b, v30.8b umlal v18.8h, v2.8b, v31.8b subs x12, x6, #4 sqrshrun v17.8b, v18.8h, #6 st1 {v17.8b}, [x1], x3 ////Store dest row3 beq end_func //If ht=4 ld1 {v3.8b, v4.8b}, [x0], x2 //// Load row5; ext v5.8b, v3.8b , v4.8b , #2 umull v20.8h, v0.8b, v28.8b umlal v20.8h, v2.8b, v29.8b umlal v20.8h, v3.8b, v30.8b umlal v20.8h, v5.8b, v31.8b ld1 {v6.8b, v7.8b}, [x0], x2 //// Load row6 ; sqrshrun v26.8b, v20.8h, #6 ext v8.8b, v6.8b , v7.8b , #2 st1 {v26.8b}, [x1], x3 ////Store dest row4 umull v22.8h, v3.8b, v28.8b umlal v22.8h, v5.8b, v29.8b umlal v22.8h, v6.8b, v30.8b umlal v22.8h, v8.8b, v31.8b ld1 {v9.8b, v10.8b}, [x0], x2 //// Load row7; sqrshrun v27.8b, v22.8h, #6 ext v11.8b, v9.8b , v10.8b , #2 st1 {v27.8b}, [x1], x3 ////Store dest row5 umull v24.8h, v6.8b, v28.8b umlal v24.8h, v8.8b, v29.8b umlal v24.8h, v9.8b, v30.8b umlal v24.8h, v11.8b, v31.8b ld1 {v0.8b, v1.8b}, [x0], x2 //// Load row8; sqrshrun v16.8b, v24.8h, #6 ext v2.8b, v0.8b , v1.8b , #2 st1 {v16.8b}, [x1], x3 ////Store dest row6 umull v18.8h, v9.8b, v28.8b umlal v18.8h, v11.8b, v29.8b umlal v18.8h, v0.8b, v30.8b umlal v18.8h, v2.8b, v31.8b sqrshrun v17.8b, v18.8h, #6 st1 {v17.8b}, [x1], x3 ////Store dest row7 b end_func loop_2: ld1 {v0.8b}, [x0], x2 //// Load row0 ; ext v2.8b, v0.8b , v0.8b , #2 ld1 {v3.8b}, [x0], x2 //// Load row1; ext v5.8b, v3.8b , v3.8b , #2 umull v20.8h, v0.8b, v28.8b umlal v20.8h, v2.8b, v29.8b umlal v20.8h, v3.8b, v30.8b umlal v20.8h, v5.8b, v31.8b ld1 {v6.8b}, [x0], x2 //// Load row2 sqrshrun v26.8b, v20.8h, #6 ext v8.8b, v6.8b , v6.8b , #2 st1 {v26.s}[0], [x1], x3 ////Store dest row0 umull v22.8h, v3.8b, v28.8b umlal v22.8h, v5.8b, v29.8b umlal v22.8h, v6.8b, v30.8b umlal v22.8h, v8.8b, v31.8b subs x12, x6, #2 sqrshrun v27.8b, v22.8h, #6 st1 {v27.s}[0], [x1], x3 ////Store dest row1 beq end_func //If ht=2 ld1 {v9.8b}, [x0], x2 //// Load row3; ext v11.8b, v9.8b , v9.8b , #2 umull v24.8h, v6.8b, v28.8b umlal v24.8h, v8.8b, v29.8b umlal v24.8h, v9.8b, v30.8b umlal v24.8h, v11.8b, v31.8b ld1 {v0.8b}, [x0], x2 //// Load row4 ; sqrshrun v16.8b, v24.8h, #6 ext v2.8b, v0.8b , v0.8b , #2 st1 {v16.s}[0], [x1], x3 ////Store dest row2 umull v18.8h, v9.8b, v28.8b umlal v18.8h, v11.8b, v29.8b umlal v18.8h, v0.8b, v30.8b umlal v18.8h, v2.8b, v31.8b sqrshrun v17.8b, v18.8h, #6 st1 {v17.s}[0], [x1], x3 ////Store dest row3 end_func: // LDMFD sp!,{x4-x12,PC} //Restoring registers from stack ldp x19, x20, [sp], #16 pop_v_regs ret