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 "vpx_mem/vpx_mem.h"
michael@0: #include "blockd.h"
michael@0: 
michael@0: void vp8_build_intra_predictors_mby_s_c(MACROBLOCKD *x,
michael@0:                                           unsigned char * yabove_row,
michael@0:                                           unsigned char * yleft,
michael@0:                                           int left_stride,
michael@0:                                           unsigned char * ypred_ptr,
michael@0:                                           int y_stride)
michael@0: {
michael@0:     unsigned char yleft_col[16];
michael@0:     unsigned char ytop_left = yabove_row[-1];
michael@0:     int r, c, i;
michael@0: 
michael@0:     for (i = 0; i < 16; i++)
michael@0:     {
michael@0:         yleft_col[i] = yleft[i* left_stride];
michael@0:     }
michael@0: 
michael@0:     /* for Y */
michael@0:     switch (x->mode_info_context->mbmi.mode)
michael@0:     {
michael@0:     case DC_PRED:
michael@0:     {
michael@0:         int expected_dc;
michael@0:         int shift;
michael@0:         int average = 0;
michael@0: 
michael@0: 
michael@0:         if (x->up_available || x->left_available)
michael@0:         {
michael@0:             if (x->up_available)
michael@0:             {
michael@0:                 for (i = 0; i < 16; i++)
michael@0:                 {
michael@0:                     average += yabove_row[i];
michael@0:                 }
michael@0:             }
michael@0: 
michael@0:             if (x->left_available)
michael@0:             {
michael@0: 
michael@0:                 for (i = 0; i < 16; i++)
michael@0:                 {
michael@0:                     average += yleft_col[i];
michael@0:                 }
michael@0: 
michael@0:             }
michael@0: 
michael@0: 
michael@0: 
michael@0:             shift = 3 + x->up_available + x->left_available;
michael@0:             expected_dc = (average + (1 << (shift - 1))) >> shift;
michael@0:         }
michael@0:         else
michael@0:         {
michael@0:             expected_dc = 128;
michael@0:         }
michael@0: 
michael@0:         /*vpx_memset(ypred_ptr, expected_dc, 256);*/
michael@0:         for (r = 0; r < 16; r++)
michael@0:         {
michael@0:             vpx_memset(ypred_ptr, expected_dc, 16);
michael@0:             ypred_ptr += y_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0:     case V_PRED:
michael@0:     {
michael@0: 
michael@0:         for (r = 0; r < 16; r++)
michael@0:         {
michael@0: 
michael@0:             ((int *)ypred_ptr)[0] = ((int *)yabove_row)[0];
michael@0:             ((int *)ypred_ptr)[1] = ((int *)yabove_row)[1];
michael@0:             ((int *)ypred_ptr)[2] = ((int *)yabove_row)[2];
michael@0:             ((int *)ypred_ptr)[3] = ((int *)yabove_row)[3];
michael@0:             ypred_ptr += y_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0:     case H_PRED:
michael@0:     {
michael@0: 
michael@0:         for (r = 0; r < 16; r++)
michael@0:         {
michael@0: 
michael@0:             vpx_memset(ypred_ptr, yleft_col[r], 16);
michael@0:             ypred_ptr += y_stride;
michael@0:         }
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case TM_PRED:
michael@0:     {
michael@0: 
michael@0:         for (r = 0; r < 16; r++)
michael@0:         {
michael@0:             for (c = 0; c < 16; c++)
michael@0:             {
michael@0:                 int pred =  yleft_col[r] + yabove_row[ c] - ytop_left;
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:                 ypred_ptr[c] = pred;
michael@0:             }
michael@0: 
michael@0:             ypred_ptr += y_stride;
michael@0:         }
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case B_PRED:
michael@0:     case NEARESTMV:
michael@0:     case NEARMV:
michael@0:     case ZEROMV:
michael@0:     case NEWMV:
michael@0:     case SPLITMV:
michael@0:     case MB_MODE_COUNT:
michael@0:         break;
michael@0:     }
michael@0: }
michael@0: 
michael@0: void vp8_build_intra_predictors_mbuv_s_c(MACROBLOCKD *x,
michael@0:                                          unsigned char * uabove_row,
michael@0:                                          unsigned char * vabove_row,
michael@0:                                          unsigned char * uleft,
michael@0:                                          unsigned char * vleft,
michael@0:                                          int left_stride,
michael@0:                                          unsigned char * upred_ptr,
michael@0:                                          unsigned char * vpred_ptr,
michael@0:                                          int pred_stride)
michael@0: {
michael@0:     unsigned char uleft_col[8];
michael@0:     unsigned char utop_left = uabove_row[-1];
michael@0:     unsigned char vleft_col[8];
michael@0:     unsigned char vtop_left = vabove_row[-1];
michael@0: 
michael@0:     int i, j;
michael@0: 
michael@0:     for (i = 0; i < 8; i++)
michael@0:     {
michael@0:         uleft_col[i] = uleft [i* left_stride];
michael@0:         vleft_col[i] = vleft [i* left_stride];
michael@0:     }
michael@0: 
michael@0:     switch (x->mode_info_context->mbmi.uv_mode)
michael@0:     {
michael@0:     case DC_PRED:
michael@0:     {
michael@0:         int expected_udc;
michael@0:         int expected_vdc;
michael@0:         int shift;
michael@0:         int Uaverage = 0;
michael@0:         int Vaverage = 0;
michael@0: 
michael@0:         if (x->up_available)
michael@0:         {
michael@0:             for (i = 0; i < 8; i++)
michael@0:             {
michael@0:                 Uaverage += uabove_row[i];
michael@0:                 Vaverage += vabove_row[i];
michael@0:             }
michael@0:         }
michael@0: 
michael@0:         if (x->left_available)
michael@0:         {
michael@0:             for (i = 0; i < 8; i++)
michael@0:             {
michael@0:                 Uaverage += uleft_col[i];
michael@0:                 Vaverage += vleft_col[i];
michael@0:             }
michael@0:         }
michael@0: 
michael@0:         if (!x->up_available && !x->left_available)
michael@0:         {
michael@0:             expected_udc = 128;
michael@0:             expected_vdc = 128;
michael@0:         }
michael@0:         else
michael@0:         {
michael@0:             shift = 2 + x->up_available + x->left_available;
michael@0:             expected_udc = (Uaverage + (1 << (shift - 1))) >> shift;
michael@0:             expected_vdc = (Vaverage + (1 << (shift - 1))) >> shift;
michael@0:         }
michael@0: 
michael@0: 
michael@0:         /*vpx_memset(upred_ptr,expected_udc,64);*/
michael@0:         /*vpx_memset(vpred_ptr,expected_vdc,64);*/
michael@0:         for (i = 0; i < 8; i++)
michael@0:         {
michael@0:             vpx_memset(upred_ptr, expected_udc, 8);
michael@0:             vpx_memset(vpred_ptr, expected_vdc, 8);
michael@0:             upred_ptr += pred_stride;
michael@0:             vpred_ptr += pred_stride;
michael@0:         }
michael@0:     }
michael@0:     break;
michael@0:     case V_PRED:
michael@0:     {
michael@0:         for (i = 0; i < 8; i++)
michael@0:         {
michael@0:             vpx_memcpy(upred_ptr, uabove_row, 8);
michael@0:             vpx_memcpy(vpred_ptr, vabove_row, 8);
michael@0:             upred_ptr += pred_stride;
michael@0:             vpred_ptr += pred_stride;
michael@0:         }
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case H_PRED:
michael@0:     {
michael@0:         for (i = 0; i < 8; i++)
michael@0:         {
michael@0:             vpx_memset(upred_ptr, uleft_col[i], 8);
michael@0:             vpx_memset(vpred_ptr, vleft_col[i], 8);
michael@0:             upred_ptr += pred_stride;
michael@0:             vpred_ptr += pred_stride;
michael@0:         }
michael@0:     }
michael@0: 
michael@0:     break;
michael@0:     case TM_PRED:
michael@0:     {
michael@0:         for (i = 0; i < 8; i++)
michael@0:         {
michael@0:             for (j = 0; j < 8; j++)
michael@0:             {
michael@0:                 int predu = uleft_col[i] + uabove_row[j] - utop_left;
michael@0:                 int predv = vleft_col[i] + vabove_row[j] - vtop_left;
michael@0: 
michael@0:                 if (predu < 0)
michael@0:                     predu = 0;
michael@0: 
michael@0:                 if (predu > 255)
michael@0:                     predu = 255;
michael@0: 
michael@0:                 if (predv < 0)
michael@0:                     predv = 0;
michael@0: 
michael@0:                 if (predv > 255)
michael@0:                     predv = 255;
michael@0: 
michael@0:                 upred_ptr[j] = predu;
michael@0:                 vpred_ptr[j] = predv;
michael@0:             }
michael@0: 
michael@0:             upred_ptr += pred_stride;
michael@0:             vpred_ptr += pred_stride;
michael@0:         }
michael@0: 
michael@0:     }
michael@0:     break;
michael@0:     case B_PRED:
michael@0:     case NEARESTMV:
michael@0:     case NEARMV:
michael@0:     case ZEROMV:
michael@0:     case NEWMV:
michael@0:     case SPLITMV:
michael@0:     case MB_MODE_COUNT:
michael@0:         break;
michael@0:     }
michael@0: }