vp9_pred_common.h 1.58 KB
Newer Older
1
/*
Paul Wilkins's avatar
Paul Wilkins committed
2
 *  Copyright (c) 2012 The WebM project authors. All Rights Reserved.
3
4
5
6
7
8
9
10
 *
 *  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
12
#ifndef VP9_COMMON_VP9_PRED_COMMON_H_
#define VP9_COMMON_VP9_PRED_COMMON_H_
13

Dmitry Kovalev's avatar
Dmitry Kovalev committed
14
15
#include "vp9/common/vp9_blockd.h"
#include "vp9/common/vp9_onyxc_int.h"
16
17

// Predicted items
John Koleszar's avatar
John Koleszar committed
18
typedef enum {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
19
  PRED_SEG_ID = 0,  // Segment identifier
Ronald S. Bultje's avatar
Ronald S. Bultje committed
20
21
22
23
24
25
26
  PRED_MBSKIP = 1,
  PRED_SWITCHABLE_INTERP = 2,
  PRED_INTRA_INTER = 3,
  PRED_COMP_INTER_INTER = 4,
  PRED_SINGLE_REF_P1 = 5,
  PRED_SINGLE_REF_P2 = 6,
  PRED_COMP_REF_P = 7,
27
  PRED_TX_SIZE = 8
28
29
} PRED_ID;

30
unsigned char vp9_get_pred_context(const VP9_COMMON *cm, const MACROBLOCKD *xd,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
31
32
                                   PRED_ID pred_id);

33
vp9_prob vp9_get_pred_prob(const VP9_COMMON *cm, const MACROBLOCKD *xd,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
34
                           PRED_ID pred_id);
35

36
const vp9_prob *vp9_get_pred_probs(const VP9_COMMON *cm, const MACROBLOCKD *xd,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
37
                                   PRED_ID pred_id);
38

39
unsigned char vp9_get_pred_flag(const MACROBLOCKD *xd, PRED_ID pred_id);
40

Dmitry Kovalev's avatar
Dmitry Kovalev committed
41
void vp9_set_pred_flag(MACROBLOCKD *xd, BLOCK_SIZE_TYPE bsize, PRED_ID pred_id,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
42
                       unsigned char pred_flag);
43

44
45
int vp9_get_segment_id(VP9_COMMON *cm, const uint8_t *segment_ids,
                       BLOCK_SIZE_TYPE bsize, int mi_row, int mi_col);
46

47
#endif  // VP9_COMMON_VP9_PRED_COMMON_H_