src/third_party/libvpx/source/libvpx/vp9/common/vp9_reconinter.h

   1 /*
   2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #ifndef VP9_COMMON_VP9_RECONINTER_H_
  12 #define VP9_COMMON_VP9_RECONINTER_H_
  13
  14 #include "vpx/vpx_integer.h"
  15 #include "vp9/common/vp9_onyxc_int.h"
  16
  17 #ifdef __cplusplus
  18 extern "C" {
  19 #endif
  20
  21 void vp9_build_inter_predictors_sby(MACROBLOCKD *xd, int mi_row, int mi_col,
  22                                     BLOCK_SIZE bsize);
  23
  24 void vp9_build_inter_predictors_sbuv(MACROBLOCKD *xd, int mi_row, int mi_col,
  25                                      BLOCK_SIZE bsize);
  26
  27 void vp9_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col,
  28                                    BLOCK_SIZE bsize);
  29
  30 void vp9_dec_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col,
  31                                        BLOCK_SIZE bsize);
  32
  33 void vp9_build_inter_predictor(const uint8_t *src, int src_stride,
  34                                uint8_t *dst, int dst_stride,
  35                                const MV *mv_q3,
  36                                const struct scale_factors *sf,
  37                                int w, int h, int do_avg,
  38                                const InterpKernel *kernel,
  39                                enum mv_precision precision,
  40                                int x, int y);
  41
  42 static INLINE int scaled_buffer_offset(int x_offset, int y_offset, int stride,
  43                                        const struct scale_factors *sf) {
  44   const int x = sf ? sf->scale_value_x(x_offset, sf) : x_offset;
  45   const int y = sf ? sf->scale_value_y(y_offset, sf) : y_offset;
  46   return y * stride + x;
  47 }
  48
  49 static INLINE void setup_pred_plane(struct buf_2d *dst,
  50                                     uint8_t *src, int stride,
  51                                     int mi_row, int mi_col,
  52                                     const struct scale_factors *scale,
  53                                     int subsampling_x, int subsampling_y) {
  54   const int x = (MI_SIZE * mi_col) >> subsampling_x;
  55   const int y = (MI_SIZE * mi_row) >> subsampling_y;
  56   dst->buf = src + scaled_buffer_offset(x, y, stride, scale);
  57   dst->stride = stride;
  58 }
  59
  60 // TODO(jkoleszar): audit all uses of this that don't set mb_row, mb_col
  61 static void setup_dst_planes(MACROBLOCKD *xd,
  62                              const YV12_BUFFER_CONFIG *src,
  63                              int mi_row, int mi_col) {
  64   uint8_t *const buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
  65                                src->alpha_buffer};
  66   const int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
  67                           src->alpha_stride};
  68   int i;
  69
  70   for (i = 0; i < MAX_MB_PLANE; ++i) {
  71     struct macroblockd_plane *const pd = &xd->plane[i];
  72     setup_pred_plane(&pd->dst, buffers[i], strides[i], mi_row, mi_col, NULL,
  73                      pd->subsampling_x, pd->subsampling_y);
  74   }
  75 }
  76
  77 static void setup_pre_planes(MACROBLOCKD *xd, int idx,
  78                              const YV12_BUFFER_CONFIG *src,
  79                              int mi_row, int mi_col,
  80                              const struct scale_factors *sf) {
  81   if (src != NULL) {
  82     int i;
  83     uint8_t *const buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
  84                                  src->alpha_buffer};
  85     const int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
  86                             src->alpha_stride};
  87
  88     for (i = 0; i < MAX_MB_PLANE; ++i) {
  89       struct macroblockd_plane *const pd = &xd->plane[i];
  90       setup_pred_plane(&pd->pre[idx], buffers[i], strides[i], mi_row, mi_col,
  91                        sf, pd->subsampling_x, pd->subsampling_y);
  92     }
  93   }
  94 }
  95
  96 #ifdef __cplusplus
  97 }  // extern "C"
  98 #endif
  99
 100 #endif  // VP9_COMMON_VP9_RECONINTER_H_