/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */


#include "vpx_ports/config.h"
#include "recon.h"
#include "reconintra.h"
#include "vpx_mem/vpx_mem.h"

/* For skip_recon_mb(), add vp8_build_intra_predictors_mby_s(MACROBLOCKD *x) and
 * vp8_build_intra_predictors_mbuv_s(MACROBLOCKD *x).
 */
void vp8_recon_intra_mbuv(const vp8_recon_rtcd_vtable_t *rtcd, MACROBLOCKD *x)
{
    int i;

    for (i = 16; i < 24; i += 2)
    {
        BLOCKD *b = &x->block[i];
        RECON_INVOKE(rtcd, recon2)(b->predictor, b->diff, *(b->base_dst) + b->dst, b->dst_stride);
    }
}

void vp8_build_intra_predictors_mby_internal(MACROBLOCKD *x, unsigned char *ypred_ptr, int y_stride, int mode)
{

    unsigned char *yabove_row = x->dst.y_buffer - x->dst.y_stride;
    unsigned char yleft_col[16];
    unsigned char ytop_left = yabove_row[-1];
    int r, c, i;

    for (i = 0; i < 16; i++)
    {
        yleft_col[i] = x->dst.y_buffer [i* x->dst.y_stride -1];
    }

    /* for Y */
    switch (mode)
    {
    case DC_PRED:
    {
        int expected_dc;
        int i;
        int shift;
        int average = 0;


        if (x->up_available || x->left_available)
        {
            if (x->up_available)
            {
                for (i = 0; i < 16; i++)
                {
                    average += yabove_row[i];
                }
            }

            if (x->left_available)
            {

                for (i = 0; i < 16; i++)
                {
                    average += yleft_col[i];
                }
            }
            shift = 3 + x->up_available + x->left_available;
            expected_dc = (average + (1 << (shift - 1))) >> shift;
        }
        else
        {
            expected_dc = 128;
        }

        for (r = 0; r < 16; r++)
        {
            vpx_memset(ypred_ptr, expected_dc, 16);
            ypred_ptr += y_stride; /*16;*/
        }
    }
    break;
    case V_PRED:
    {

        for (r = 0; r < 16; r++)
        {

            ((int *)ypred_ptr)[0] = ((int *)yabove_row)[0];
            ((int *)ypred_ptr)[1] = ((int *)yabove_row)[1];
            ((int *)ypred_ptr)[2] = ((int *)yabove_row)[2];
            ((int *)ypred_ptr)[3] = ((int *)yabove_row)[3];
            ypred_ptr += y_stride;
        }
    }
    break;
    case H_PRED:
    {

        for (r = 0; r < 16; r++)
        {

            vpx_memset(ypred_ptr, yleft_col[r], 16);
            ypred_ptr += y_stride;
        }

    }
    break;
    case TM_PRED:
    {

        for (r = 0; r < 16; r++)
        {
            for (c = 0; c < 16; c++)
            {
                int pred =  yleft_col[r] + yabove_row[ c] - ytop_left;

                if (pred < 0)
                    pred = 0;

                if (pred > 255)
                    pred = 255;

                ypred_ptr[c] = pred;
            }

            ypred_ptr += y_stride;
        }

    }
    break;
#if CONIFG_I8X8
    case I8X8_PRED:
#endif
    case B_PRED:
    case NEARESTMV:
    case NEARMV:
    case ZEROMV:
    case NEWMV:
    case SPLITMV:
    case MB_MODE_COUNT:
        break;
    }
}

void vp8_build_intra_predictors_mby(MACROBLOCKD *x)
{
    vp8_build_intra_predictors_mby_internal(x, x->predictor, 16,
                                            x->mode_info_context->mbmi.mode);
}

void vp8_build_intra_predictors_mby_s(MACROBLOCKD *x)
{
    vp8_build_intra_predictors_mby_internal(x, x->dst.y_buffer, x->dst.y_stride,
                                            x->mode_info_context->mbmi.mode);
}

#if CONFIG_COMP_INTRA_PRED
void vp8_build_comp_intra_predictors_mby(MACROBLOCKD *x)
{
    unsigned char predictor[2][256];
    int i;

    vp8_build_intra_predictors_mby_internal(x, predictor[0], 16,
                                            x->mode_info_context->mbmi.mode);
    vp8_build_intra_predictors_mby_internal(x, predictor[1], 16,
                                            x->mode_info_context->mbmi.second_mode);

    for (i = 0; i < 256; i++)
    {
        x->predictor[i] = (predictor[0][i] + predictor[1][i] + 1) >> 1;
    }
}
#endif

void vp8_build_intra_predictors_mbuv_internal(MACROBLOCKD *x,
                                              unsigned char *upred_ptr,
                                              unsigned char *vpred_ptr,
                                              int uv_stride,
                                              int mode)
{
    unsigned char *uabove_row = x->dst.u_buffer - x->dst.uv_stride;
    unsigned char uleft_col[16];
    unsigned char utop_left = uabove_row[-1];
    unsigned char *vabove_row = x->dst.v_buffer - x->dst.uv_stride;
    unsigned char vleft_col[20];
    unsigned char vtop_left = vabove_row[-1];

    int i, j;

    for (i = 0; i < 8; i++)
    {
        uleft_col[i] = x->dst.u_buffer [i* x->dst.uv_stride -1];
        vleft_col[i] = x->dst.v_buffer [i* x->dst.uv_stride -1];
    }

    switch (mode)
    {
    case DC_PRED:
    {
        int expected_udc;
        int expected_vdc;
        int i;
        int shift;
        int Uaverage = 0;
        int Vaverage = 0;

        if (x->up_available)
        {
            for (i = 0; i < 8; i++)
            {
                Uaverage += uabove_row[i];
                Vaverage += vabove_row[i];
            }
        }

        if (x->left_available)
        {
            for (i = 0; i < 8; i++)
            {
                Uaverage += uleft_col[i];
                Vaverage += vleft_col[i];
            }
        }

        if (!x->up_available && !x->left_available)
        {
            expected_udc = 128;
            expected_vdc = 128;
        }
        else
        {
            shift = 2 + x->up_available + x->left_available;
            expected_udc = (Uaverage + (1 << (shift - 1))) >> shift;
            expected_vdc = (Vaverage + (1 << (shift - 1))) >> shift;
        }


        /*vpx_memset(upred_ptr,expected_udc,64);*/
        /*vpx_memset(vpred_ptr,expected_vdc,64);*/
        for (i = 0; i < 8; i++)
        {
            vpx_memset(upred_ptr, expected_udc, 8);
            vpx_memset(vpred_ptr, expected_vdc, 8);
            upred_ptr += uv_stride; /*8;*/
            vpred_ptr += uv_stride; /*8;*/
        }
    }
    break;
    case V_PRED:
    {
        int i;

        for (i = 0; i < 8; i++)
        {
            vpx_memcpy(upred_ptr, uabove_row, 8);
            vpx_memcpy(vpred_ptr, vabove_row, 8);
            upred_ptr += uv_stride; /*8;*/
            vpred_ptr += uv_stride; /*8;*/
        }

    }
    break;
    case H_PRED:
    {
        int i;

        for (i = 0; i < 8; i++)
        {
            vpx_memset(upred_ptr, uleft_col[i], 8);
            vpx_memset(vpred_ptr, vleft_col[i], 8);
            upred_ptr += uv_stride; /*8;*/
            vpred_ptr += uv_stride; /*8;*/
        }
    }

    break;
    case TM_PRED:
    {
        int i;

        for (i = 0; i < 8; i++)
        {
            for (j = 0; j < 8; j++)
            {
                int predu = uleft_col[i] + uabove_row[j] - utop_left;
                int predv = vleft_col[i] + vabove_row[j] - vtop_left;

                if (predu < 0)
                    predu = 0;

                if (predu > 255)
                    predu = 255;

                if (predv < 0)
                    predv = 0;

                if (predv > 255)
                    predv = 255;

                upred_ptr[j] = predu;
                vpred_ptr[j] = predv;
            }

            upred_ptr += uv_stride; /*8;*/
            vpred_ptr += uv_stride; /*8;*/
        }

    }
    break;
    case B_PRED:
    case NEARESTMV:
    case NEARMV:
    case ZEROMV:
    case NEWMV:
    case SPLITMV:
    case MB_MODE_COUNT:
        break;
    }
}

void vp8_build_intra_predictors_mbuv(MACROBLOCKD *x)
{
    vp8_build_intra_predictors_mbuv_internal(x,
                                             &x->predictor[256],
                                             &x->predictor[320],
                                             8,
                                             x->mode_info_context->mbmi.uv_mode);
}

void vp8_build_intra_predictors_mbuv_s(MACROBLOCKD *x)
{
    vp8_build_intra_predictors_mbuv_internal(x,
                                             x->dst.u_buffer,
                                             x->dst.v_buffer,
                                             x->dst.uv_stride,
                                             x->mode_info_context->mbmi.uv_mode);
}

#if CONFIG_COMP_INTRA_PRED
void vp8_build_comp_intra_predictors_mbuv(MACROBLOCKD *x)
{
    unsigned char predictor[2][2][64];
    int i;

    vp8_build_intra_predictors_mbuv_internal(x, predictor[0][0], predictor[1][0], 8,
                                             x->mode_info_context->mbmi.uv_mode);
    vp8_build_intra_predictors_mbuv_internal(x, predictor[0][1], predictor[1][1], 8,
                                             x->mode_info_context->mbmi.second_uv_mode);
    for (i = 0; i < 64; i++)
    {
        x->predictor[256 + i] = (predictor[0][0][i] + predictor[0][1][i] + 1) >> 1;
        x->predictor[256 + 64 + i] = (predictor[1][0][i] + predictor[1][1][i] + 1) >> 1;
    }
}
#endif

void vp8_intra8x8_predict(BLOCKD *x,
                          int mode,
                          unsigned char *predictor)
{

    unsigned char *yabove_row = *(x->base_dst) + x->dst - x->dst_stride;
    unsigned char yleft_col[8];
    unsigned char ytop_left = yabove_row[-1];
    int r, c, i;

    for (i = 0; i < 8; i++)
    {
        yleft_col[i] = (*(x->base_dst))[x->dst - 1 + i * x->dst_stride];
    }
    switch (mode)
    {
    case DC_PRED:
        {
            int expected_dc = 0;

            for (i = 0; i < 8; i++)
            {
                expected_dc += yabove_row[i];
                expected_dc += yleft_col[i];
            }
            expected_dc = (expected_dc + 8) >> 4;

            for (r = 0; r < 8; r++)
            {
                for (c = 0; c < 8; c++)
                {
                    predictor[c] = expected_dc;
                }
                predictor += 16;
            }
        }
        break;
    case V_PRED:
        {
            for (r = 0; r < 8; r++)
            {
                for (c = 0; c < 8; c++)
                {

                    predictor[c] = yabove_row[c];
                }
                predictor += 16;
            }

        }
        break;
    case H_PRED:
        {

            for (r = 0; r < 8; r++)
            {
                for (c = 0; c < 8; c++)
                {
                    predictor[c] = yleft_col[r];
                }
                predictor += 16;
            }
        }
        break;
    case TM_PRED:
        {
            /* prediction similar to true_motion prediction */
            for (r = 0; r < 8; r++)
            {
                for (c = 0; c < 8; c++)
                {
                    int pred = yabove_row[c] - ytop_left + yleft_col[r];
                    if (pred < 0)
                        pred = 0;

                    if (pred > 255)
                        pred = 255;
                    predictor[c] = pred;
                }

                predictor += 16;
            }
        }
        break;
    }
}

#if CONFIG_COMP_INTRA_PRED
void vp8_comp_intra8x8_predict(BLOCKD *x,
                               int mode, int second_mode,
                               unsigned char *out_predictor)
{
    
    unsigned char predictor[2][8*16];
    int i, j;

    vp8_intra8x8_predict(x, mode, predictor[0]);
    vp8_intra8x8_predict(x, second_mode, predictor[1]);

    for (i = 0; i < 8*16; i += 16)
    {
        for (j = i; j < i + 8; j++)
        {
            out_predictor[j] = (predictor[0][j] + predictor[1][j] + 1) >> 1;
        }
    }
}
#endif

void vp8_intra_uv4x4_predict(BLOCKD *x,
                             int mode,
                             unsigned char *predictor)
{

    unsigned char *above_row = *(x->base_dst) + x->dst - x->dst_stride;
    unsigned char left_col[4];
    unsigned char top_left = above_row[-1];
    int r, c, i;

    for (i = 0; i < 4; i++)
    {
        left_col[i] = (*(x->base_dst))[x->dst - 1 + i * x->dst_stride];
    }
    switch (mode)
    {
    case DC_PRED:
        {
            int expected_dc = 0;

            for (i = 0; i < 4; i++)
            {
                expected_dc += above_row[i];
                expected_dc += left_col[i];
            }
            expected_dc = (expected_dc + 4) >> 3;

            for (r = 0; r < 4; r++)
            {
                for (c = 0; c < 4; c++)
                {
                    predictor[c] = expected_dc;
                }
                predictor += 8;
            }
        }
        break;
    case V_PRED:
        {
            for (r = 0; r < 4; r++)
            {
                for (c = 0; c < 4; c++)
                {

                    predictor[c] = above_row[c];
                }
                predictor += 8;
            }

        }
        break;
    case H_PRED:
        {

            for (r = 0; r < 4; r++)
            {
                for (c = 0; c < 4; c++)
                {
                    predictor[c] = left_col[r];
                }
                predictor += 8;
            }
        }
        break;
    case TM_PRED:
        {
            /* prediction similar to true_motion prediction */
            for (r = 0; r < 4; r++)
            {
                for (c = 0; c < 4; c++)
                {
                    int pred = above_row[c] - top_left + left_col[r];
                    if (pred < 0)
                        pred = 0;

                    if (pred > 255)
                        pred = 255;
                    predictor[c] = pred;
                }

                predictor += 8;
            }
        }
        break;
    }
}

#if CONFIG_COMP_INTRA_PRED
void vp8_comp_intra_uv4x4_predict(BLOCKD *x,
                                  int mode, int mode2,
                                  unsigned char *out_predictor)
{
    unsigned char predictor[2][8*4];
    int i, j;

    vp8_intra_uv4x4_predict(x, mode, predictor[0]);
    vp8_intra_uv4x4_predict(x, mode2, predictor[1]);

    for (i = 0; i < 4*8; i += 8)
    {
        for (j = i; j < i + 4; j++)
        {
            out_predictor[j] = (predictor[0][j] + predictor[1][j] + 1) >> 1;
        }
    }
}
#endif

/* TODO: try different ways of use Y-UV mode correlation
 Current code assumes that a uv 4x4 block use same mode
 as corresponding Y 8x8 area
 */