/*

 * 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 <stdint.h>

#include "config.h"

#include "libavutil/attributes.h"

#include "libavutil/intreadwrite.h"

#include "pixblockdsp.h"

static void get_pixels_16_c(int16_t *restrict block, const uint8_t *pixels,

                            ptrdiff_t stride)

{

    for (int i = 0; i < 8; i++)

        AV_COPY128(block + i * 8, pixels + i * stride);

}

static void get_pixels_unaligned_16_c(int16_t *restrict block,

                                      const uint8_t *pixels, ptrdiff_t stride)

{

    AV_COPY128U(block + 0 * 8, pixels + 0 * stride);

    AV_COPY128U(block + 1 * 8, pixels + 1 * stride);

    AV_COPY128U(block + 2 * 8, pixels + 2 * stride);

    AV_COPY128U(block + 3 * 8, pixels + 3 * stride);

    AV_COPY128U(block + 4 * 8, pixels + 4 * stride);

    AV_COPY128U(block + 5 * 8, pixels + 5 * stride);

    AV_COPY128U(block + 6 * 8, pixels + 6 * stride);

    AV_COPY128U(block + 7 * 8, pixels + 7 * stride);

}

static void get_pixels_8_c(int16_t *restrict block, const uint8_t *pixels,

                           ptrdiff_t stride)

{

    int i;

    /* read the pixels */

    for (i = 0; i < 8; i++) {

        block[0] = pixels[0];

        block[1] = pixels[1];

        block[2] = pixels[2];

        block[3] = pixels[3];

        block[4] = pixels[4];

        block[5] = pixels[5];

        block[6] = pixels[6];

        block[7] = pixels[7];

        pixels  += stride;

        block   += 8;

    }

}

static void diff_pixels_c(int16_t *restrict block, const uint8_t *s1,

                          const uint8_t *s2, ptrdiff_t stride)

{

    int i;

    /* read the pixels */

    for (i = 0; i < 8; i++) {

        block[0] = s1[0] - s2[0];

        block[1] = s1[1] - s2[1];

        block[2] = s1[2] - s2[2];

        block[3] = s1[3] - s2[3];

        block[4] = s1[4] - s2[4];

        block[5] = s1[5] - s2[5];

        block[6] = s1[6] - s2[6];

        block[7] = s1[7] - s2[7];

        s1      += stride;

        s2      += stride;

        block   += 8;

    }

}

av_cold void ff_pixblockdsp_init(PixblockDSPContext *c, int bits_per_raw_sample)

{

    const unsigned high_bit_depth = bits_per_raw_sample > 8 &&

                                    bits_per_raw_sample <= 16;

    c->diff_pixels_unaligned =

    c->diff_pixels = diff_pixels_c;

    if (high_bit_depth) {

        c->get_pixels_unaligned = get_pixels_unaligned_16_c;

        c->get_pixels           = get_pixels_16_c;

    } else {

        c->get_pixels_unaligned =

        c->get_pixels           = get_pixels_8_c;

    }

#if ARCH_AARCH64

    ff_pixblockdsp_init_aarch64(c, high_bit_depth);

#elif ARCH_ARM

    ff_pixblockdsp_init_arm(c, high_bit_depth);

#elif ARCH_PPC

    ff_pixblockdsp_init_ppc(c, high_bit_depth);

#elif ARCH_RISCV

    ff_pixblockdsp_init_riscv(c, high_bit_depth);

#elif ARCH_X86 && HAVE_X86ASM

    ff_pixblockdsp_init_x86(c, high_bit_depth);

#elif ARCH_MIPS

    ff_pixblockdsp_init_mips(c, high_bit_depth);

#endif

}