/*
 * Copyright (c) 2024 Ramiro Polla
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "libavutil/aarch64/asm.S"

function ff_nv24_to_yuv420p_chroma_neon, export=1
// x0  uint8_t *dst1
// x1  int dstStride1
// x2  uint8_t *dst2
// x3  int dstStride2
// x4  const uint8_t *src
// x5  int srcStride
// w6  int w
// w7  int h

        add             x9, x4, w5, sxtw            // x9 = src + srcStride
        lsl             w5, w5, #1                  // srcStride *= 2
        sub             w5, w5, w6, lsl #2          // srcPadding = (2 * srcStride) - (4 * w)
        sub             w1, w1, w6                  // dstPadding1 = dstStride1 - w
        sub             w3, w3, w6                  // dstPadding2 = dstStride2 - w

1:
        mov             w10, w6                     // w10 = w

2:
        ld2             {v0.16b, v1.16b}, [x4], #32 // v0 = U1, v1 = V1
        ld2             {v2.16b, v3.16b}, [x9], #32 // v2 = U2, v3 = V2

        uaddlp          v0.8h, v0.16b               // pairwise add U1 into v0
        uaddlp          v1.8h, v1.16b               // pairwise add V1 into v1
        uadalp          v0.8h, v2.16b               // pairwise add U2, accumulate into v0
        uadalp          v1.8h, v3.16b               // pairwise add V2, accumulate into v1

        shrn            v0.8b, v0.8h, #2            // divide by 4
        shrn            v1.8b, v1.8h, #2            // divide by 4

        st1             {v0.8b}, [x0], #8           // store U into dst1
        st1             {v1.8b}, [x2], #8           // store V into dst2

        subs            w10, w10, #8
        b.gt            2b

        // next row
        add             x4, x4, w5, sxtw            // src1 += srcPadding
        add             x9, x9, w5, sxtw            // src2 += srcPadding
        add             x0, x0, w1, sxtw            // dst1 += dstPadding1
        add             x2, x2, w3, sxtw            // dst2 += dstPadding2

        subs            w7, w7, #2
        b.gt            1b

        ret
endfunc