michael@0: /*
michael@0:  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
michael@0:  *
michael@0:  *  Use of this source code is governed by a BSD-style license
michael@0:  *  that can be found in the LICENSE file in the root of the source
michael@0:  *  tree. An additional intellectual property rights grant can be found
michael@0:  *  in the file PATENTS.  All contributing project authors may
michael@0:  *  be found in the AUTHORS file in the root of the source tree.
michael@0:  */
michael@0: 
michael@0: 
michael@0: #include "vpx_config.h"
michael@0: #include "vp8_rtcd.h"
michael@0: #include "blockd.h"
michael@0: 
michael@0: void vp8_intra4x4_predict_c(unsigned char *Above,
michael@0:                             unsigned char *yleft, int left_stride,
michael@0:                             int           _b_mode,
michael@0:                             unsigned char *dst, int dst_stride,
michael@0:                             unsigned char top_left)
michael@0: {
michael@0:     int i, r, c;
michael@0:     B_PREDICTION_MODE b_mode = (B_PREDICTION_MODE)_b_mode;
michael@0:     unsigned char Left[4];
michael@0:     Left[0] = yleft[0];
michael@0:     Left[1] = yleft[left_stride];
michael@0:     Left[2] = yleft[2 * left_stride];
michael@0:     Left[3] = yleft[3 * left_stride];
michael@0: 
michael@0:     switch (b_mode)
michael@0:     {
michael@0:     case B_DC_PRED:
michael@0:     {
michael@0:         int expected_dc = 0;
michael@0: 
michael@0:         for (i = 0; i < 4; i++)
michael@0:         {
michael@0:             expected_dc += Above[i];
michael@0:             expected_dc += Left[i];
michael@0:         }
michael@0: 
michael@0:         expected_dc = (expected_dc + 4) >> 3;
michael@0: 
michael@0:         for (r = 0; r < 4; r++)
michael@0:         {
michael@0:             for (c = 0; c < 4; c++)
michael@0:             {
michael@0:                 dst[c] = expected_dc;
michael@0:             }
michael@0: 
michael@0:             dst += dst_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0:     case B_TM_PRED:
michael@0:     {
michael@0:         /* prediction similar to true_motion prediction */
michael@0:         for (r = 0; r < 4; r++)
michael@0:         {
michael@0:             for (c = 0; c < 4; c++)
michael@0:             {
michael@0:                 int pred = Above[c] - top_left + Left[r];
michael@0: 
michael@0:                 if (pred < 0)
michael@0:                     pred = 0;
michael@0: 
michael@0:                 if (pred > 255)
michael@0:                     pred = 255;
michael@0: 
michael@0:                 dst[c] = pred;
michael@0:             }
michael@0: 
michael@0:             dst += dst_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0: 
michael@0:     case B_VE_PRED:
michael@0:     {
michael@0: 
michael@0:         unsigned int ap[4];
michael@0:         ap[0] = (top_left  + 2 * Above[0] + Above[1] + 2) >> 2;
michael@0:         ap[1] = (Above[0] + 2 * Above[1] + Above[2] + 2) >> 2;
michael@0:         ap[2] = (Above[1] + 2 * Above[2] + Above[3] + 2) >> 2;
michael@0:         ap[3] = (Above[2] + 2 * Above[3] + Above[4] + 2) >> 2;
michael@0: 
michael@0:         for (r = 0; r < 4; r++)
michael@0:         {
michael@0:             for (c = 0; c < 4; c++)
michael@0:             {
michael@0: 
michael@0:                 dst[c] = ap[c];
michael@0:             }
michael@0: 
michael@0:             dst += dst_stride;
michael@0:         }
michael@0: 
michael@0:     }
michael@0:     break;
michael@0: 
michael@0: 
michael@0:     case B_HE_PRED:
michael@0:     {
michael@0: 
michael@0:         unsigned int lp[4];
michael@0:         lp[0] = (top_left + 2 * Left[0] + Left[1] + 2) >> 2;
michael@0:         lp[1] = (Left[0] + 2 * Left[1] + Left[2] + 2) >> 2;
michael@0:         lp[2] = (Left[1] + 2 * Left[2] + Left[3] + 2) >> 2;
michael@0:         lp[3] = (Left[2] + 2 * Left[3] + Left[3] + 2) >> 2;
michael@0: 
michael@0:         for (r = 0; r < 4; r++)
michael@0:         {
michael@0:             for (c = 0; c < 4; c++)
michael@0:             {
michael@0:                 dst[c] = lp[r];
michael@0:             }
michael@0: 
michael@0:             dst += dst_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0:     case B_LD_PRED:
michael@0:     {
michael@0:         unsigned char *ptr = Above;
michael@0:         dst[0 * dst_stride + 0] = (ptr[0] + ptr[1] * 2 + ptr[2] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 1] =
michael@0:             dst[1 * dst_stride + 0] = (ptr[1] + ptr[2] * 2 + ptr[3] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 2] =
michael@0:             dst[1 * dst_stride + 1] =
michael@0:                 dst[2 * dst_stride + 0] = (ptr[2] + ptr[3] * 2 + ptr[4] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 3] =
michael@0:             dst[1 * dst_stride + 2] =
michael@0:                 dst[2 * dst_stride + 1] =
michael@0:                     dst[3 * dst_stride + 0] = (ptr[3] + ptr[4] * 2 + ptr[5] + 2) >> 2;
michael@0:         dst[1 * dst_stride + 3] =
michael@0:             dst[2 * dst_stride + 2] =
michael@0:                 dst[3 * dst_stride + 1] = (ptr[4] + ptr[5] * 2 + ptr[6] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 3] =
michael@0:             dst[3 * dst_stride + 2] = (ptr[5] + ptr[6] * 2 + ptr[7] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 3] = (ptr[6] + ptr[7] * 2 + ptr[7] + 2) >> 2;
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case B_RD_PRED:
michael@0:     {
michael@0: 
michael@0:         unsigned char pp[9];
michael@0: 
michael@0:         pp[0] = Left[3];
michael@0:         pp[1] = Left[2];
michael@0:         pp[2] = Left[1];
michael@0:         pp[3] = Left[0];
michael@0:         pp[4] = top_left;
michael@0:         pp[5] = Above[0];
michael@0:         pp[6] = Above[1];
michael@0:         pp[7] = Above[2];
michael@0:         pp[8] = Above[3];
michael@0: 
michael@0:         dst[3 * dst_stride + 0] = (pp[0] + pp[1] * 2 + pp[2] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 1] =
michael@0:             dst[2 * dst_stride + 0] = (pp[1] + pp[2] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 2] =
michael@0:             dst[2 * dst_stride + 1] =
michael@0:                 dst[1 * dst_stride + 0] = (pp[2] + pp[3] * 2 + pp[4] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 3] =
michael@0:             dst[2 * dst_stride + 2] =
michael@0:                 dst[1 * dst_stride + 1] =
michael@0:                     dst[0 * dst_stride + 0] = (pp[3] + pp[4] * 2 + pp[5] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 3] =
michael@0:             dst[1 * dst_stride + 2] =
michael@0:                 dst[0 * dst_stride + 1] = (pp[4] + pp[5] * 2 + pp[6] + 2) >> 2;
michael@0:         dst[1 * dst_stride + 3] =
michael@0:             dst[0 * dst_stride + 2] = (pp[5] + pp[6] * 2 + pp[7] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 3] = (pp[6] + pp[7] * 2 + pp[8] + 2) >> 2;
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case B_VR_PRED:
michael@0:     {
michael@0: 
michael@0:         unsigned char pp[9];
michael@0: 
michael@0:         pp[0] = Left[3];
michael@0:         pp[1] = Left[2];
michael@0:         pp[2] = Left[1];
michael@0:         pp[3] = Left[0];
michael@0:         pp[4] = top_left;
michael@0:         pp[5] = Above[0];
michael@0:         pp[6] = Above[1];
michael@0:         pp[7] = Above[2];
michael@0:         pp[8] = Above[3];
michael@0: 
michael@0: 
michael@0:         dst[3 * dst_stride + 0] = (pp[1] + pp[2] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 0] = (pp[2] + pp[3] * 2 + pp[4] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 1] =
michael@0:             dst[1 * dst_stride + 0] = (pp[3] + pp[4] * 2 + pp[5] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 1] =
michael@0:             dst[0 * dst_stride + 0] = (pp[4] + pp[5] + 1) >> 1;
michael@0:         dst[3 * dst_stride + 2] =
michael@0:             dst[1 * dst_stride + 1] = (pp[4] + pp[5] * 2 + pp[6] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 2] =
michael@0:             dst[0 * dst_stride + 1] = (pp[5] + pp[6] + 1) >> 1;
michael@0:         dst[3 * dst_stride + 3] =
michael@0:             dst[1 * dst_stride + 2] = (pp[5] + pp[6] * 2 + pp[7] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 3] =
michael@0:             dst[0 * dst_stride + 2] = (pp[6] + pp[7] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 3] = (pp[6] + pp[7] * 2 + pp[8] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 3] = (pp[7] + pp[8] + 1) >> 1;
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case B_VL_PRED:
michael@0:     {
michael@0: 
michael@0:         unsigned char *pp = Above;
michael@0: 
michael@0:         dst[0 * dst_stride + 0] = (pp[0] + pp[1] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 0] = (pp[0] + pp[1] * 2 + pp[2] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 0] =
michael@0:             dst[0 * dst_stride + 1] = (pp[1] + pp[2] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 1] =
michael@0:             dst[3 * dst_stride + 0] = (pp[1] + pp[2] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 1] =
michael@0:             dst[0 * dst_stride + 2] = (pp[2] + pp[3] + 1) >> 1;
michael@0:         dst[3 * dst_stride + 1] =
michael@0:             dst[1 * dst_stride + 2] = (pp[2] + pp[3] * 2 + pp[4] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 3] =
michael@0:             dst[2 * dst_stride + 2] = (pp[3] + pp[4] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 3] =
michael@0:             dst[3 * dst_stride + 2] = (pp[3] + pp[4] * 2 + pp[5] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 3] = (pp[4] + pp[5] * 2 + pp[6] + 2) >> 2;
michael@0:         dst[3 * dst_stride + 3] = (pp[5] + pp[6] * 2 + pp[7] + 2) >> 2;
michael@0:     }
michael@0:     break;
michael@0: 
michael@0:     case B_HD_PRED:
michael@0:     {
michael@0:         unsigned char pp[9];
michael@0:         pp[0] = Left[3];
michael@0:         pp[1] = Left[2];
michael@0:         pp[2] = Left[1];
michael@0:         pp[3] = Left[0];
michael@0:         pp[4] = top_left;
michael@0:         pp[5] = Above[0];
michael@0:         pp[6] = Above[1];
michael@0:         pp[7] = Above[2];
michael@0:         pp[8] = Above[3];
michael@0: 
michael@0: 
michael@0:         dst[3 * dst_stride + 0] = (pp[0] + pp[1] + 1) >> 1;
michael@0:         dst[3 * dst_stride + 1] = (pp[0] + pp[1] * 2 + pp[2] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 0] =
michael@0:             dst[3 * dst_stride + 2] = (pp[1] + pp[2] + 1) >> 1;
michael@0:         dst[2 * dst_stride + 1] =
michael@0:             dst[3 * dst_stride + 3] = (pp[1] + pp[2] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 2] =
michael@0:             dst[1 * dst_stride + 0] = (pp[2] + pp[3] + 1) >> 1;
michael@0:         dst[2 * dst_stride + 3] =
michael@0:             dst[1 * dst_stride + 1] = (pp[2] + pp[3] * 2 + pp[4] + 2) >> 2;
michael@0:         dst[1 * dst_stride + 2] =
michael@0:             dst[0 * dst_stride + 0] = (pp[3] + pp[4] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 3] =
michael@0:             dst[0 * dst_stride + 1] = (pp[3] + pp[4] * 2 + pp[5] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 2] = (pp[4] + pp[5] * 2 + pp[6] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 3] = (pp[5] + pp[6] * 2 + pp[7] + 2) >> 2;
michael@0:     }
michael@0:     break;
michael@0: 
michael@0: 
michael@0:     case B_HU_PRED:
michael@0:     {
michael@0:         unsigned char *pp = Left;
michael@0:         dst[0 * dst_stride + 0] = (pp[0] + pp[1] + 1) >> 1;
michael@0:         dst[0 * dst_stride + 1] = (pp[0] + pp[1] * 2 + pp[2] + 2) >> 2;
michael@0:         dst[0 * dst_stride + 2] =
michael@0:             dst[1 * dst_stride + 0] = (pp[1] + pp[2] + 1) >> 1;
michael@0:         dst[0 * dst_stride + 3] =
michael@0:             dst[1 * dst_stride + 1] = (pp[1] + pp[2] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[1 * dst_stride + 2] =
michael@0:             dst[2 * dst_stride + 0] = (pp[2] + pp[3] + 1) >> 1;
michael@0:         dst[1 * dst_stride + 3] =
michael@0:             dst[2 * dst_stride + 1] = (pp[2] + pp[3] * 2 + pp[3] + 2) >> 2;
michael@0:         dst[2 * dst_stride + 2] =
michael@0:             dst[2 * dst_stride + 3] =
michael@0:                 dst[3 * dst_stride + 0] =
michael@0:                     dst[3 * dst_stride + 1] =
michael@0:                         dst[3 * dst_stride + 2] =
michael@0:                             dst[3 * dst_stride + 3] = pp[3];
michael@0:     }
michael@0:     break;
michael@0: 
michael@0:     default:
michael@0:     break;
michael@0: 
michael@0:     }
michael@0: }