#include "gfx_util.h"

#include <strings.h>
#include <stdio.h>

extern "C" {
#include <libavutil/pixdesc.h>
}

#include "CpuCapabilities.h"
#include "gfx_conv_c.h"
#include "gfx_conv_mmx.h"


// ref docs
// http://www.joemaller.com/fcp/fxscript_yuv_color.shtml


#if DEBUG
  #define TRACE(a...) printf(a)
#else
  #define TRACE(a...)
#endif


//! This function will try to find the best colorspaces for both the ff-codec
// and the Media Kit sides.
gfx_convert_func
resolve_colorspace(color_space colorSpace, AVPixelFormat pixelFormat, int width,
        int height)
{
        CPUCapabilities cpu;

        switch (colorSpace) {
                case B_RGB32:
                        // Planar Formats
                        if (pixelFormat == AV_PIX_FMT_YUV410P) {
                                TRACE("resolve_colorspace: gfx_conv_yuv410p_rgb32_c\n");
                                return gfx_conv_yuv410p_rgb32_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV411P) {
                                TRACE("resolve_colorspace: gfx_conv_yuv411p_rgb32_c\n");
                                return gfx_conv_yuv411p_rgb32_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV420P
                                || pixelFormat == AV_PIX_FMT_YUVJ420P) {
#if 0
                                if (cpu.HasSSSE3() && width % 8 == 0 && height % 2 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv420p_rgba32_ssse3\n");
                                        return gfx_conv_yuv420p_rgba32_ssse3;
                                } else if (cpu.HasSSE2() && width % 8 == 0 && height % 2 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv420p_rgba32_sse2\n");
                                        return gfx_conv_yuv420p_rgba32_sse2;
                                } else if (cpu.HasSSE1() && width % 4 == 0
                                        && height % 2 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv420p_rgba32_sse\n");
                                        return gfx_conv_yuv420p_rgba32_sse;
                                }
#endif
                                TRACE("resolve_colorspace: gfx_conv_YCbCr420p_RGB32_c\n");
                                return gfx_conv_YCbCr420p_RGB32_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV422P
                                || pixelFormat == AV_PIX_FMT_YUVJ422P) {
#if 0
                                if (cpu.HasSSSE3() && width % 8 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv422p_RGB32_ssse3\n");
                                        return gfx_conv_yuv422p_rgba32_ssse3;
                                } else if (cpu.HasSSE2() && width % 8 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv422p_RGB32_sse2\n");
                                        return gfx_conv_yuv422p_rgba32_sse2;
                                } else if (cpu.HasSSE1() && width % 4 == 0) {
                                        TRACE("resolve_colorspace: gfx_conv_yuv422p_RGB32_sse\n");
                                        return gfx_conv_yuv422p_rgba32_sse;
                                }
#endif
                                TRACE("resolve_colorspace: gfx_conv_YCbCr422p_RGB32_c\n");
                                return gfx_conv_YCbCr422_RGB32_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_GBRP) {
                                return gfx_conv_GBRP_RGB32_c;
                        }

                        // Packed Formats
                        if (pixelFormat == AV_PIX_FMT_YUYV422) {
#if 0
                                if (cpu.HasSSSE3() && width % 8 == 0) {
                                        return gfx_conv_yuv422_rgba32_ssse3;
                                } else if (cpu.HasSSE2() && width % 8 == 0) {
                                        return gfx_conv_yuv422_rgba32_sse2;
                                } else if (cpu.HasSSE1() && width % 4 == 0
                                        && height % 2 == 0) {
                                        return gfx_conv_yuv422_rgba32_sse;
                                }
#endif
                                return gfx_conv_YCbCr422_RGB32_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV420P10LE)
                                return gfx_conv_yuv420p10le_rgb32_c;

                        TRACE("resolve_colorspace: %s => B_RGB32: NULL\n",
                                pixfmt_to_string(pixelFormat));
                        return NULL;

                case B_RGB24_BIG:
                        TRACE("resolve_colorspace: %s => B_RGB24_BIG: NULL\n",
                                pixfmt_to_string(pixelFormat));
                        return NULL;

                case B_RGB24:
                        TRACE("resolve_colorspace: %s => B_RGB24: NULL\n",
                                pixfmt_to_string(pixelFormat));
                        return NULL;

                case B_YCbCr422:
                        if (pixelFormat == AV_PIX_FMT_YUV410P) {
                                TRACE("resolve_colorspace: gfx_conv_yuv410p_ycbcr422_c\n");
                                return gfx_conv_yuv410p_ycbcr422_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV411P) {
                                TRACE("resolve_colorspace: gfx_conv_yuv411p_ycbcr422_c\n");
                                return gfx_conv_yuv411p_ycbcr422_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUV420P
                                || pixelFormat == AV_PIX_FMT_YUVJ420P) {
                                TRACE("resolve_colorspace: gfx_conv_yuv420p_ycbcr422_c\n");
                                return gfx_conv_yuv420p_ycbcr422_c;
                        }

                        if (pixelFormat == AV_PIX_FMT_YUYV422) {
                                TRACE("resolve_colorspace: PIX_FMT_YUV422 => B_YCbCr422: "
                                        "gfx_conv_null\n");
                                return gfx_conv_null;
                        }

                        TRACE("resolve_colorspace: %s => B_YCbCr422: NULL\n",
                                pixfmt_to_string(pixelFormat));
                        return NULL;

                default:
                        TRACE("resolve_colorspace: default: NULL!!!\n");
                        return NULL;
        }
}


const char*
pixfmt_to_string(int pixFormat)
{
        const char* name = av_get_pix_fmt_name((enum AVPixelFormat)pixFormat);
        if (name == NULL)
                return "(unknown)";
        return name;
}


color_space
pixfmt_to_colorspace(int pixFormat)
{
        switch (pixFormat) {
                default:
                        TRACE("No BE API colorspace definition for pixel format "
                                "\"%s\".\n", pixfmt_to_string(pixFormat));
                        // Supposed to fall through.
                case AV_PIX_FMT_NONE:
                        return B_NO_COLOR_SPACE;

                // NOTE: See pixfmt_to_colorspace() for what these are.
                case AV_PIX_FMT_YUV420P:
                        return B_YUV420;
                case AV_PIX_FMT_RGB24:
                        return B_RGB24_BIG;
                case AV_PIX_FMT_BGR24:
                        return B_RGB24;
                case AV_PIX_FMT_YUV422P:
                        return B_YUV422;
                case AV_PIX_FMT_YUV444P:
                        return B_YUV444;
                case AV_PIX_FMT_RGB32:
                        return B_RGBA32_BIG;
                case AV_PIX_FMT_YUV410P:
                        return B_YUV9;
                case AV_PIX_FMT_YUV411P:
                        return B_YUV12;
                case AV_PIX_FMT_RGB565:
                        return B_RGB16_BIG;
                case AV_PIX_FMT_RGB555:
                        return B_RGB15_BIG;
                case AV_PIX_FMT_GRAY8:
                        return B_GRAY8;
                case AV_PIX_FMT_MONOBLACK:
                        return B_GRAY1;
                case AV_PIX_FMT_PAL8:
                        return B_CMAP8;
                case AV_PIX_FMT_BGR32:
                        return B_RGB32;
                case AV_PIX_FMT_BGR565:
                        return B_RGB16;
                case AV_PIX_FMT_BGR555:
                        return B_RGB15;
                // TODO: more YCbCr color spaces! These are not the same as YUV!
                case AV_PIX_FMT_YUYV422:
                        return B_YCbCr422;

        }
}


AVPixelFormat
colorspace_to_pixfmt(color_space format)
{
        switch(format) {
                default:
                case B_NO_COLOR_SPACE:
                        return AV_PIX_FMT_NONE;

                // NOTE: See pixfmt_to_colorspace() for what these are.
                case B_YUV420:
                        return AV_PIX_FMT_YUV420P;
                case B_YUV422:
                        return AV_PIX_FMT_YUV422P;
                case B_RGB24_BIG:
                        return AV_PIX_FMT_RGB24;
                case B_RGB24:
                        return AV_PIX_FMT_BGR24;
                case B_YUV444:
                        return AV_PIX_FMT_YUV444P;
                case B_RGBA32_BIG:
                case B_RGB32_BIG:
                        return AV_PIX_FMT_BGR32;
                case B_YUV9:
                        return AV_PIX_FMT_YUV410P;
                case B_YUV12:
                        return AV_PIX_FMT_YUV411P;
                case B_RGB16_BIG:
                        return AV_PIX_FMT_RGB565;
                case B_RGB15_BIG:
                        return AV_PIX_FMT_RGB555;
                case B_GRAY8:
                        return AV_PIX_FMT_GRAY8;
                case B_GRAY1:
                        return AV_PIX_FMT_MONOBLACK;
                case B_CMAP8:
                        return AV_PIX_FMT_PAL8;
                case B_RGBA32:
                case B_RGB32:
                        return AV_PIX_FMT_RGB32;
                case B_RGB16:
                        return AV_PIX_FMT_BGR565;
                case B_RGB15:
                        return AV_PIX_FMT_BGR555;
                // TODO: more YCbCr color spaces! These are not the same as YUV!
                case B_YCbCr422:
                        return AV_PIX_FMT_YUYV422;
        }
}


#define BEGIN_TAG "\033[31m"
#define END_TAG "\033[0m"

void
dump_ffframe_audio(AVFrame* frame, const char* name)
{
        printf(BEGIN_TAG "AVFrame(%s) [ pkt_dts:%-10" PRId64 " #samples:%-5d %s"
                " ]\n" END_TAG,
                name,
                frame->pkt_dts,
                frame->nb_samples,
                av_get_sample_fmt_name(static_cast<AVSampleFormat>(frame->format)));
}


void
dump_ffframe_video(AVFrame* frame, const char* name)
{
        const char* picttypes[] = {"no pict type", "intra", "predicted",
                "bidir pre", "s(gmc)-vop", "switching intra", "switching predicted", "BI"};
#if LIBAVCODEC_VERSION_MAJOR >= 60
        printf(BEGIN_TAG "AVFrame(%s) [ pkt_dts:%-10" PRId64 " %s%s%s%s%s%s ]\n" END_TAG, name,
                frame->pkt_dts,
                (frame->flags & AV_FRAME_FLAG_CORRUPT) ? "corrupt, " : "",
                (frame->flags & AV_FRAME_FLAG_KEY) ? "keyframe, " : "",
                (frame->flags & AV_FRAME_FLAG_DISCARD) ? "discard, " : "",
                (frame->flags & AV_FRAME_FLAG_INTERLACED) ? "interlaced, " : "",
                (frame->flags & AV_FRAME_FLAG_TOP_FIELD_FIRST) ? "top field first, " : "",
                picttypes[frame->pict_type]);
#else
        printf(BEGIN_TAG "AVFrame(%s) [ pkt_dts:%-10" PRId64 " %s%s ]\n" END_TAG, name, frame->pkt_dts,
                frame->key_frame ? "keyframe, " : "", picttypes[frame->pict_type]);
#endif
}