/*

 * software YUV to RGB converter

 *

 * Copyright (C) 2001-2007 Michael Niedermayer

 * Copyright (C) 2009-2010 Konstantin Shishkov

 *

 * MMX/MMXEXT template stuff (needed for fast movntq support),

 * 1,4,8bpp support and context / deglobalize stuff

 * by Michael Niedermayer (michaelni@gmx.at)

 *

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

#include <stdlib.h>

#include <inttypes.h>

#include "config.h"

#include "libswscale/rgb2rgb.h"

#include "libswscale/swscale.h"

#include "libswscale/swscale_internal.h"

#include "libavutil/attributes.h"

#include "libavutil/x86/asm.h"

#include "libavutil/x86/cpu.h"

#include "libavutil/cpu.h"

#if HAVE_X86ASM

#define YUV2RGB_LOOP(depth)                                          \

    h_size = (c->opts.dst_w + 7) & ~7;                               \

    if (h_size * depth > FFABS(dstStride[0]))                        \

        h_size -= 8;                                                 \

                                                                     \

    vshift = c->opts.src_format != AV_PIX_FMT_YUV422P;               \

                                                                     \

    for (y = 0; y < srcSliceH; y++) {                                \

        uint8_t *image    = dst[0] + (y + srcSliceY) * dstStride[0]; \

        const uint8_t *py = src[0] +               y * srcStride[0]; \

        const uint8_t *pu = src[1] +   (y >> vshift) * srcStride[1]; \

        const uint8_t *pv = src[2] +   (y >> vshift) * srcStride[2]; \

        x86_reg index = -h_size / 2;                                 \

extern void ff_yuv_420_rgb24_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuv_420_bgr24_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuv_420_rgb15_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuv_420_rgb16_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuv_420_rgb32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuv_420_bgr32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                   const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                   const uint8_t *py_2index);

extern void ff_yuva_420_rgb32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                    const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                    const uint8_t *py_2index, const uint8_t *pa_2index);

extern void ff_yuva_420_bgr32_ssse3(x86_reg index, uint8_t *image, const uint8_t *pu_index,

                                    const uint8_t *pv_index, const uint64_t *pointer_c_dither,

                                    const uint8_t *py_2index, const uint8_t *pa_2index);

#if ARCH_X86_64

extern void ff_yuv_420_gbrp24_ssse3(x86_reg index, uint8_t *image, uint8_t *dst_b, uint8_t *dst_r,

                                    const uint8_t *pu_index, const uint8_t *pv_index,

                                    const uint64_t *pointer_c_dither,

                                    const uint8_t *py_2index);

#endif

static inline int yuv420_rgb15_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(2)

        c->blueDither  = ff_dither8[y       & 1];

        c->greenDither = ff_dither8[y       & 1];

        c->redDither   = ff_dither8[(y + 1) & 1];

        ff_yuv_420_rgb15_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

static inline int yuv420_rgb16_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(2)

        c->blueDither  = ff_dither8[y       & 1];

        c->greenDither = ff_dither4[y       & 1];

        c->redDither   = ff_dither8[(y + 1) & 1];

        ff_yuv_420_rgb16_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

static inline int yuv420_rgb32_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(4)

        ff_yuv_420_rgb32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

static inline int yuv420_bgr32_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(4)

        ff_yuv_420_bgr32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

static inline int yuva420_rgb32_ssse3(SwsInternal *c, const uint8_t *const src[],

                                      const int srcStride[],

                                      int srcSliceY, int srcSliceH,

                                      uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(4)

        const uint8_t *pa = src[3] + y * srcStride[3];

        ff_yuva_420_rgb32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index, pa - 2 * index);

    }

    return srcSliceH;

}

static inline int yuva420_bgr32_ssse3(SwsInternal *c, const uint8_t *const src[],

                                      const int srcStride[],

                                      int srcSliceY, int srcSliceH,

                                      uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(4)

        const uint8_t *pa = src[3] + y * srcStride[3];

        ff_yuva_420_bgr32_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index, pa - 2 * index);

    }

    return srcSliceH;

}

static inline int yuv420_rgb24_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(3)

        ff_yuv_420_rgb24_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

static inline int yuv420_bgr24_ssse3(SwsInternal *c, const uint8_t *const src[],

                                     const int srcStride[],

                                     int srcSliceY, int srcSliceH,

                                     uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    YUV2RGB_LOOP(3)

        ff_yuv_420_bgr24_ssse3(index, image, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

#if ARCH_X86_64

static inline int yuv420_gbrp_ssse3(SwsInternal *c, const uint8_t *const src[],

                                    const int srcStride[],

                                    int srcSliceY, int srcSliceH,

                                    uint8_t *const dst[], const int dstStride[])

{

    int y, h_size, vshift;

    h_size = (c->opts.dst_w + 7) & ~7;

    if (h_size * 3 > FFABS(dstStride[0]))

        h_size -= 8;

    vshift = c->opts.src_format != AV_PIX_FMT_YUV422P;

    for (y = 0; y < srcSliceH; y++) {

        uint8_t *dst_g    = dst[0] + (y + srcSliceY) * dstStride[0];

        uint8_t *dst_b    = dst[1] + (y + srcSliceY) * dstStride[1];

        uint8_t *dst_r    = dst[2] + (y + srcSliceY) * dstStride[2];

        const uint8_t *py = src[0] +               y * srcStride[0];

        const uint8_t *pu = src[1] +   (y >> vshift) * srcStride[1];

        const uint8_t *pv = src[2] +   (y >> vshift) * srcStride[2];

        x86_reg index = -h_size / 2;

        ff_yuv_420_gbrp24_ssse3(index, dst_g, dst_b, dst_r, pu - index, pv - index, &(c->redDither), py - 2 * index);

    }

    return srcSliceH;

}

#endif

#endif /* HAVE_X86ASM */

av_cold SwsFunc ff_yuv2rgb_init_x86(SwsInternal *c)

{

#if HAVE_X86ASM

    int cpu_flags = av_get_cpu_flags();

    if (EXTERNAL_SSSE3(cpu_flags)) {

        switch (c->opts.dst_format) {

        case AV_PIX_FMT_RGB32:

            if (c->opts.src_format == AV_PIX_FMT_YUVA420P) {

#if CONFIG_SWSCALE_ALPHA

                return yuva420_rgb32_ssse3;

#endif

                break;

            } else

                return yuv420_rgb32_ssse3;

        case AV_PIX_FMT_BGR32:

            if (c->opts.src_format == AV_PIX_FMT_YUVA420P) {

#if CONFIG_SWSCALE_ALPHA

                return yuva420_bgr32_ssse3;

#endif

                break;

            } else

                return yuv420_bgr32_ssse3;

        case AV_PIX_FMT_RGB24:

            return yuv420_rgb24_ssse3;

        case AV_PIX_FMT_BGR24:

            return yuv420_bgr24_ssse3;

        case AV_PIX_FMT_RGB565:

            return yuv420_rgb16_ssse3;

        case AV_PIX_FMT_RGB555:

            return yuv420_rgb15_ssse3;

#if ARCH_X86_64

        case AV_PIX_FMT_GBRP:

            return yuv420_gbrp_ssse3;

#endif

        }

    }

#endif /* HAVE_X86ASM */

    return NULL;

}