vp9_mvref_common.c 11.5 KB
Newer Older
Paul Wilkins's avatar
Paul Wilkins committed
1
2
3
4
5
6
7
8
9
10
/*
 *  Copyright (c) 2012 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.
 */

11
#include "vp9/common/vp9_mvref_common.h"
Paul Wilkins's avatar
Paul Wilkins committed
12
13

#define MVREF_NEIGHBOURS 8
Dmitry Kovalev's avatar
Dmitry Kovalev committed
14

15
#if CONFIG_SB8X8
16
17
18
19
20
static int b_mv_ref_search[MVREF_NEIGHBOURS][2] = {
  {0, -1}, {-1, 0}, {-1, -1}, {0, -2},
  {-2, 0}, {-1, -2}, {-2, -1}, {-2, -2}
};

21
static int mb_mv_ref_search[MVREF_NEIGHBOURS][2] = {
22
23
    {0, -1}, {-1, 0}, {-1, -1}, {0, -3},
    {-3, 0}, {-1, -3}, {-3, -1}, {-3, -3}
24
};
Dmitry Kovalev's avatar
Dmitry Kovalev committed
25

26
static int sb_mv_ref_search[MVREF_NEIGHBOURS][2] = {
27
28
    {0, -1}, {-1, 0}, {2, -1}, {-1, 2},
    {-1, -1}, {0, -3}, {-3, 0}, {-1, -3}
29
};
Dmitry Kovalev's avatar
Dmitry Kovalev committed
30

31
static int sb64_mv_ref_search[MVREF_NEIGHBOURS][2] = {
32
33
34
35
36
37
38
    {0, -1}, {-1, 0}, {2, -1}, {-1,  2},
    {4, -1}, {-1, 4}, {6, -1}, {-1, -1}
};
#else
static int mb_mv_ref_search[MVREF_NEIGHBOURS][2] = {
  {0, -1}, {-1, 0}, {-1, -1}, {0, -2},
  {-2, 0}, {-1, -2}, {-2, -1}, {-2, -2}
39
40
};

41
42
43
44
45
46
47
48
49
50
static int sb_mv_ref_search[MVREF_NEIGHBOURS][2] = {
  {0, -1}, {-1, 0}, {1, -1}, {-1, 1},
  {-1, -1}, {0, -2}, {-2, 0}, {-1, -2}
};

static int sb64_mv_ref_search[MVREF_NEIGHBOURS][2] = {
  {0, -1}, {-1, 0}, {1, -1}, {-1,  1},
  {2, -1}, {-1, 2}, {3, -1}, {-1, -1}
};
#endif
51

Paul Wilkins's avatar
Paul Wilkins committed
52
// clamp_mv_ref
Paul Wilkins's avatar
Paul Wilkins committed
53
54
#define MV_BORDER (16 << 3) // Allow 16 pels in 1/8th pel units

Dmitry Kovalev's avatar
Dmitry Kovalev committed
55
static void clamp_mv_ref(const MACROBLOCKD *xd, int_mv *mv) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
56
57
58
59
  mv->as_mv.col = clamp(mv->as_mv.col, xd->mb_to_left_edge - MV_BORDER,
                                       xd->mb_to_right_edge + MV_BORDER);
  mv->as_mv.row = clamp(mv->as_mv.row, xd->mb_to_top_edge - MV_BORDER,
                                       xd->mb_to_bottom_edge + MV_BORDER);
Paul Wilkins's avatar
Paul Wilkins committed
60
61
}

62
63
// Gets a candidate refenence motion vector from the given mode info
// structure if one exists that matches the given reference frame.
Dmitry Kovalev's avatar
Dmitry Kovalev committed
64
static int get_matching_candidate(const MODE_INFO *candidate_mi,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
65
66
                                  MV_REFERENCE_FRAME ref_frame,
                                  int_mv *c_mv) {
67
68
69
70
71
  if (ref_frame == candidate_mi->mbmi.ref_frame) {
    c_mv->as_int = candidate_mi->mbmi.mv[0].as_int;
  } else if (ref_frame == candidate_mi->mbmi.second_ref_frame) {
    c_mv->as_int = candidate_mi->mbmi.mv[1].as_int;
  } else {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
72
    return 0;
73
74
  }

Dmitry Kovalev's avatar
Dmitry Kovalev committed
75
  return 1;
76
}
Paul Wilkins's avatar
Paul Wilkins committed
77

78
79
// Gets candidate refenence motion vector(s) from the given mode info
// structure if they exists and do NOT match the given reference frame.
Dmitry Kovalev's avatar
Dmitry Kovalev committed
80
81
82
83
84
85
static void get_non_matching_candidates(const MODE_INFO *candidate_mi,
                                        MV_REFERENCE_FRAME ref_frame,
                                        MV_REFERENCE_FRAME *c_ref_frame,
                                        int_mv *c_mv,
                                        MV_REFERENCE_FRAME *c2_ref_frame,
                                        int_mv *c2_mv) {
Paul Wilkins's avatar
Paul Wilkins committed
86

87
  c_mv->as_int = 0;
Paul Wilkins's avatar
Paul Wilkins committed
88
  c2_mv->as_int = 0;
89
  *c_ref_frame = INTRA_FRAME;
Paul Wilkins's avatar
Paul Wilkins committed
90
  *c2_ref_frame = INTRA_FRAME;
Paul Wilkins's avatar
Paul Wilkins committed
91

92
93
94
95
96
97
  // If first candidate not valid neither will be.
  if (candidate_mi->mbmi.ref_frame > INTRA_FRAME) {
    // First candidate
    if (candidate_mi->mbmi.ref_frame != ref_frame) {
      *c_ref_frame = candidate_mi->mbmi.ref_frame;
      c_mv->as_int = candidate_mi->mbmi.mv[0].as_int;
Paul Wilkins's avatar
Paul Wilkins committed
98
99
    }

100
    // Second candidate
101
    if ((candidate_mi->mbmi.second_ref_frame > INTRA_FRAME) &&
Paul Wilkins's avatar
Paul Wilkins committed
102
        (candidate_mi->mbmi.second_ref_frame != ref_frame) &&
Dmitry Kovalev's avatar
Dmitry Kovalev committed
103
        (candidate_mi->mbmi.mv[1].as_int != candidate_mi->mbmi.mv[0].as_int)) {
Paul Wilkins's avatar
Paul Wilkins committed
104
      *c2_ref_frame = candidate_mi->mbmi.second_ref_frame;
105
      c2_mv->as_int = candidate_mi->mbmi.mv[1].as_int;
Paul Wilkins's avatar
Paul Wilkins committed
106
    }
Paul Wilkins's avatar
Paul Wilkins committed
107
108
109
  }
}

Paul Wilkins's avatar
Paul Wilkins committed
110
111

// Performs mv sign inversion if indicated by the reference frame combination.
Dmitry Kovalev's avatar
Dmitry Kovalev committed
112
113
114
static void scale_mv(MACROBLOCKD *xd, MV_REFERENCE_FRAME this_ref_frame,
                     MV_REFERENCE_FRAME candidate_ref_frame,
                     int_mv *candidate_mv, int *ref_sign_bias) {
Paul Wilkins's avatar
Paul Wilkins committed
115
116
117
118
119
120
121
122
123
  // int frame_distances[MAX_REF_FRAMES];
  // int last_distance = 1;
  // int gf_distance = xd->frames_since_golden;
  // int arf_distance = xd->frames_till_alt_ref_frame;

  // Sign inversion where appropriate.
  if (ref_sign_bias[candidate_ref_frame] != ref_sign_bias[this_ref_frame]) {
    candidate_mv->as_mv.row = -candidate_mv->as_mv.row;
    candidate_mv->as_mv.col = -candidate_mv->as_mv.col;
Paul Wilkins's avatar
Paul Wilkins committed
124
125
  }

Paul Wilkins's avatar
Paul Wilkins committed
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
  /*
  // Scale based on frame distance if the reference frames not the same.
  frame_distances[INTRA_FRAME] = 1;   // should never be used
  frame_distances[LAST_FRAME] = 1;
  frame_distances[GOLDEN_FRAME] =
    (xd->frames_since_golden) ? xd->frames_si nce_golden : 1;
  frame_distances[ALTREF_FRAME] =
    (xd->frames_till_alt_ref_frame) ? xd->frames_till_alt_ref_frame : 1;

  if (frame_distances[this_ref_frame] &&
      frame_distances[candidate_ref_frame]) {
    candidate_mv->as_mv.row =
      (short)(((int)(candidate_mv->as_mv.row) *
               frame_distances[this_ref_frame]) /
              frame_distances[candidate_ref_frame]);

    candidate_mv->as_mv.col =
      (short)(((int)(candidate_mv->as_mv.col) *
               frame_distances[this_ref_frame]) /
              frame_distances[candidate_ref_frame]);
  }
  */
Paul Wilkins's avatar
Paul Wilkins committed
148
149
}

Paul Wilkins's avatar
Paul Wilkins committed
150
151
// Add a candidate mv.
// Discard if it has already been seen.
Dmitry Kovalev's avatar
Dmitry Kovalev committed
152
153
154
static void add_candidate_mv(int_mv *mv_list,  int *mv_scores,
                             int *candidate_count, int_mv candidate_mv,
                             int weight) {
Paul Wilkins's avatar
Paul Wilkins committed
155
156
157
158
159
160
161
162
163
  if (*candidate_count == 0) {
    mv_list[0].as_int = candidate_mv.as_int;
    mv_scores[0] = weight;
    *candidate_count += 1;
  } else if ((*candidate_count == 1) &&
             (candidate_mv.as_int != mv_list[0].as_int)) {
    mv_list[1].as_int = candidate_mv.as_int;
    mv_scores[1] = weight;
    *candidate_count += 1;
Paul Wilkins's avatar
Paul Wilkins committed
164
165
  }
}
Paul Wilkins's avatar
Paul Wilkins committed
166

Paul Wilkins's avatar
Paul Wilkins committed
167
168
// This function searches the neighbourhood of a given MB/SB
// to try and find candidate reference vectors.
Paul Wilkins's avatar
Paul Wilkins committed
169
//
Dmitry Kovalev's avatar
Dmitry Kovalev committed
170
171
172
void vp9_find_mv_refs(VP9_COMMON *cm, MACROBLOCKD *xd, MODE_INFO *here,
                      MODE_INFO *lf_here, MV_REFERENCE_FRAME ref_frame,
                      int_mv *mv_ref_list, int *ref_sign_bias) {
Paul Wilkins's avatar
Paul Wilkins committed
173
174
  int i;
  MODE_INFO *candidate_mi;
Paul Wilkins's avatar
Paul Wilkins committed
175
  MB_MODE_INFO * mbmi = &xd->mode_info_context->mbmi;
Paul Wilkins's avatar
Paul Wilkins committed
176
  int_mv c_refmv;
Paul Wilkins's avatar
Paul Wilkins committed
177
  int_mv c2_refmv;
Paul Wilkins's avatar
Paul Wilkins committed
178
  MV_REFERENCE_FRAME c_ref_frame;
Paul Wilkins's avatar
Paul Wilkins committed
179
  MV_REFERENCE_FRAME c2_ref_frame;
Paul Wilkins's avatar
Paul Wilkins committed
180
  int candidate_scores[MAX_MV_REF_CANDIDATES];
Paul Wilkins's avatar
Paul Wilkins committed
181
  int refmv_count = 0;
182
  int split_count = 0;
183
  int (*mv_ref_search)[2];
184
  const int mi_col = get_mi_col(xd);
Paul Wilkins's avatar
Paul Wilkins committed
185
  // Blank the reference vector lists and other local structures.
Paul Wilkins's avatar
Paul Wilkins committed
186
  vpx_memset(mv_ref_list, 0, sizeof(int_mv) * MAX_MV_REF_CANDIDATES);
Paul Wilkins's avatar
Paul Wilkins committed
187
188
  vpx_memset(candidate_scores, 0, sizeof(candidate_scores));

189
190
  if (mbmi->sb_type == BLOCK_SIZE_SB64X64) {
    mv_ref_search = sb64_mv_ref_search;
191
  } else if (mbmi->sb_type >= BLOCK_SIZE_SB32X32) {
192
    mv_ref_search = sb_mv_ref_search;
193
194
195
196
197
198
#if CONFIG_SB8X8
  } else if (mbmi->sb_type >= BLOCK_SIZE_MB16X16) {
    mv_ref_search = mb_mv_ref_search;
  } else {
    mv_ref_search = b_mv_ref_search;
#else
199
200
  } else {
    mv_ref_search = mb_mv_ref_search;
201
#endif
202
  }
203
204
205

  // We first scan for candidate vectors that match the current reference frame
  // Look at nearest neigbours
Paul Wilkins's avatar
Paul Wilkins committed
206
  for (i = 0; i < 2; ++i) {
207
    const int mi_search_col = mi_col + mv_ref_search[i][0];
208

209
210
211
    if ((mi_search_col >= cm->cur_tile_mi_col_start) &&
        (mi_search_col < cm->cur_tile_mi_col_end) &&
        ((mv_ref_search[i][1] << (7 - CONFIG_SB8X8)) >= xd->mb_to_top_edge)) {
Paul Wilkins's avatar
Paul Wilkins committed
212
213
214
215

      candidate_mi = here + mv_ref_search[i][0] +
                     (mv_ref_search[i][1] * xd->mode_info_stride);

216
      if (get_matching_candidate(candidate_mi, ref_frame, &c_refmv)) {
Paul Wilkins's avatar
Paul Wilkins committed
217
218
        add_candidate_mv(mv_ref_list, candidate_scores,
                         &refmv_count, c_refmv, 16);
Paul Wilkins's avatar
Paul Wilkins committed
219
      }
220
      split_count += (candidate_mi->mbmi.mode == SPLITMV);
Paul Wilkins's avatar
Paul Wilkins committed
221
222
    }
  }
Paul Wilkins's avatar
Paul Wilkins committed
223

224
225
  // More distant neigbours
  for (i = 2; (i < MVREF_NEIGHBOURS) &&
Paul Wilkins's avatar
Paul Wilkins committed
226
              (refmv_count < MAX_MV_REF_CANDIDATES); ++i) {
227
    const int mi_search_col = mi_col + mv_ref_search[i][0];
228

229
230
231
    if ((mi_search_col >= cm->cur_tile_mi_col_start) &&
        (mi_search_col < cm->cur_tile_mi_col_end) &&
        ((mv_ref_search[i][1] << (7 - CONFIG_SB8X8)) >= xd->mb_to_top_edge)) {
Paul Wilkins's avatar
Paul Wilkins committed
232
233
234
      candidate_mi = here + mv_ref_search[i][0] +
                     (mv_ref_search[i][1] * xd->mode_info_stride);

235
      if (get_matching_candidate(candidate_mi, ref_frame, &c_refmv)) {
Paul Wilkins's avatar
Paul Wilkins committed
236
237
        add_candidate_mv(mv_ref_list, candidate_scores,
                         &refmv_count, c_refmv, 16);
238
239
240
      }
    }
  }
Paul Wilkins's avatar
Paul Wilkins committed
241

Paul Wilkins's avatar
Paul Wilkins committed
242
243
244
245
  // Look in the last frame if it exists
  if (lf_here && (refmv_count < MAX_MV_REF_CANDIDATES)) {
    candidate_mi = lf_here;
    if (get_matching_candidate(candidate_mi, ref_frame, &c_refmv)) {
Paul Wilkins's avatar
Paul Wilkins committed
246
247
      add_candidate_mv(mv_ref_list, candidate_scores,
                       &refmv_count, c_refmv, 16);
Paul Wilkins's avatar
Paul Wilkins committed
248
249
250
    }
  }

251
252
253
254
  // If we have not found enough candidates consider ones where the
  // reference frame does not match. Break out when we have
  // MAX_MV_REF_CANDIDATES candidates.
  // Look first at spatial neighbours
Paul Wilkins's avatar
Paul Wilkins committed
255
256
  for (i = 0; (i < MVREF_NEIGHBOURS) &&
              (refmv_count < MAX_MV_REF_CANDIDATES); ++i) {
257
    const int mi_search_col = mi_col + mv_ref_search[i][0];
Paul Wilkins's avatar
Paul Wilkins committed
258

259
260
261
    if ((mi_search_col >= cm->cur_tile_mi_col_start) &&
        (mi_search_col < cm->cur_tile_mi_col_end) &&
        ((mv_ref_search[i][1] << (7 - CONFIG_SB8X8)) >= xd->mb_to_top_edge)) {
Paul Wilkins's avatar
Paul Wilkins committed
262
263
264
265
266
267
268
269
270
      candidate_mi = here + mv_ref_search[i][0] +
                     (mv_ref_search[i][1] * xd->mode_info_stride);

      get_non_matching_candidates(candidate_mi, ref_frame,
                                  &c_ref_frame, &c_refmv,
                                  &c2_ref_frame, &c2_refmv);

      if (c_ref_frame != INTRA_FRAME) {
        scale_mv(xd, ref_frame, c_ref_frame, &c_refmv, ref_sign_bias);
Paul Wilkins's avatar
Paul Wilkins committed
271
272
        add_candidate_mv(mv_ref_list, candidate_scores,
                         &refmv_count, c_refmv, 1);
Paul Wilkins's avatar
Paul Wilkins committed
273
      }
274

Paul Wilkins's avatar
Paul Wilkins committed
275
276
      if (c2_ref_frame != INTRA_FRAME) {
        scale_mv(xd, ref_frame, c2_ref_frame, &c2_refmv, ref_sign_bias);
Paul Wilkins's avatar
Paul Wilkins committed
277
278
        add_candidate_mv(mv_ref_list, candidate_scores,
                         &refmv_count, c2_refmv, 1);
279
280
281
      }
    }
  }
Paul Wilkins's avatar
Paul Wilkins committed
282

283
  // Look at the last frame if it exists
Paul Wilkins's avatar
Paul Wilkins committed
284
  if (lf_here && (refmv_count < MAX_MV_REF_CANDIDATES)) {
285
286
287
288
289
290
291
    candidate_mi = lf_here;
    get_non_matching_candidates(candidate_mi, ref_frame,
                                &c_ref_frame, &c_refmv,
                                &c2_ref_frame, &c2_refmv);

    if (c_ref_frame != INTRA_FRAME) {
      scale_mv(xd, ref_frame, c_ref_frame, &c_refmv, ref_sign_bias);
Paul Wilkins's avatar
Paul Wilkins committed
292
      add_candidate_mv(mv_ref_list, candidate_scores,
Paul Wilkins's avatar
Paul Wilkins committed
293
                       &refmv_count, c_refmv, 1);
294
295
296
297
    }

    if (c2_ref_frame != INTRA_FRAME) {
      scale_mv(xd, ref_frame, c2_ref_frame, &c2_refmv, ref_sign_bias);
Paul Wilkins's avatar
Paul Wilkins committed
298
      add_candidate_mv(mv_ref_list, candidate_scores,
Paul Wilkins's avatar
Paul Wilkins committed
299
                       &refmv_count, c2_refmv, 1);
Paul Wilkins's avatar
Paul Wilkins committed
300
301
302
    }
  }

303
304
  // Define inter mode coding context.
  // 0,0 was best
Paul Wilkins's avatar
Paul Wilkins committed
305
  if (mv_ref_list[0].as_int == 0) {
306
    // 0,0 is only candidate
Paul Wilkins's avatar
Paul Wilkins committed
307
    if (refmv_count <= 1) {
Paul Wilkins's avatar
Paul Wilkins committed
308
      mbmi->mb_mode_context[ref_frame] = 0;
309
310
    // non zero candidates candidates available
    } else if (split_count == 0) {
Paul Wilkins's avatar
Paul Wilkins committed
311
312
313
314
      mbmi->mb_mode_context[ref_frame] = 1;
    } else {
      mbmi->mb_mode_context[ref_frame] = 2;
    }
315
  } else if (split_count == 0) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
316
317
    // Non zero best, No Split MV cases
    mbmi->mb_mode_context[ref_frame] = candidate_scores[0] >= 16 ? 3 : 4;
Paul Wilkins's avatar
Paul Wilkins committed
318
  } else {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
319
320
    // Non zero best, some split mv
    mbmi->mb_mode_context[ref_frame] = candidate_scores[0] >= 16 ? 5 : 6;
Paul Wilkins's avatar
Paul Wilkins committed
321
322
  }

Paul Wilkins's avatar
Paul Wilkins committed
323
  // Clamp vectors
Paul Wilkins's avatar
Paul Wilkins committed
324
  for (i = 0; i < MAX_MV_REF_CANDIDATES; ++i) {
Paul Wilkins's avatar
Paul Wilkins committed
325
    clamp_mv_ref(xd, &mv_ref_list[i]);
326
  }
Paul Wilkins's avatar
Paul Wilkins committed
327
}