cfl.h 3.06 KB
Newer Older
Luc Trudeau's avatar
Luc Trudeau committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
 *
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
 */

#ifndef AV1_COMMON_CFL_H_
#define AV1_COMMON_CFL_H_

15 16
#include <assert.h>

Luc Trudeau's avatar
Luc Trudeau committed
17 18
#include "av1/common/enums.h"

19 20 21 22 23 24 25 26 27 28
// Forward declaration of AV1_COMMON, in order to avoid creating a cyclic
// dependency by importing av1/common/onyxc_int.h
typedef struct AV1Common AV1_COMMON;

// Forward declaration of MACROBLOCK, in order to avoid creating a cyclic
// dependency by importing av1/common/blockd.h
typedef struct macroblockd MACROBLOCKD;

typedef struct {
  // Pixel buffer containing the luma pixels used as prediction for chroma
Luc Trudeau's avatar
Luc Trudeau committed
29
  // TODO(ltrudeau) Convert to uint16 for HBD support
30 31
  uint8_t y_pix[MAX_SB_SQUARE];

Luc Trudeau's avatar
Luc Trudeau committed
32 33 34 35 36
  // Pixel buffer containing the downsampled luma pixels used as prediction for
  // chroma
  // TODO(ltrudeau) Convert to uint16 for HBD support
  uint8_t y_down_pix[MAX_SB_SQUARE];

37 38 39
  // Height and width of the luma prediction block currently in the pixel buffer
  int y_height, y_width;

Luc Trudeau's avatar
Luc Trudeau committed
40 41 42 43
  // Height and width of the chroma prediction block currently associated with
  // this context
  int uv_height, uv_width;

44 45
  // Transform level averages of the luma reconstructed values over the entire
  // prediction unit
46 47 48
  // Fixed point y_averages is Q12.10:
  //   * Worst case division is 1/1024
  int y_averages_q10[MAX_NUM_TXB];
49
  int y_averages_stride;
Luc Trudeau's avatar
Luc Trudeau committed
50 51

  int are_parameters_computed;
52

53 54 55
  // Chroma subsampling
  int subsampling_x, subsampling_y;

56 57 58 59
  // Block level DC_PRED for each chromatic plane
  // Fixed point dc_pred is Q12.7:
  //   * Worst case division is 1/128
  int dc_pred_q7[CFL_PRED_PLANES];
Luc Trudeau's avatar
Luc Trudeau committed
60

61 62
  // The rate associated with each alpha codeword
  int costs[CFL_ALPHABET_SIZE];
63 64

  int mi_row, mi_col;
65 66
} CFL_CTX;

67 68 69 70 71
static const double cfl_alpha_mags[CFL_MAGS_SIZE] = {
  0., 0.125, -0.125, 0.25, -0.25, 0.5, -0.5
};

static const int cfl_alpha_codes[CFL_ALPHABET_SIZE][CFL_PRED_PLANES] = {
Luc Trudeau's avatar
Luc Trudeau committed
72
  // barrbrain's simple 1D quant ordered by subset 3 likelihood
73 74
  { 0, 0 }, { 1, 1 }, { 3, 0 }, { 3, 3 }, { 1, 0 }, { 3, 1 },
  { 5, 5 }, { 0, 1 }, { 5, 3 }, { 5, 0 }, { 3, 5 }, { 1, 3 },
75
  { 0, 3 }, { 5, 1 }, { 1, 5 }, { 0, 5 }
Luc Trudeau's avatar
Luc Trudeau committed
76 77
};

78 79 80 81
static INLINE int get_scaled_luma_q13(int alpha_q3, int y_pix, int avg_q10) {
  return alpha_q3 * ((y_pix << 10) - avg_q10);
}

82
void cfl_init(CFL_CTX *cfl, AV1_COMMON *cm);
83

Luc Trudeau's avatar
Luc Trudeau committed
84 85
void cfl_predict_block(MACROBLOCKD *const xd, uint8_t *dst, int dst_stride,
                       int row, int col, TX_SIZE tx_size, int plane);
Luc Trudeau's avatar
Luc Trudeau committed
86

87
void cfl_store(CFL_CTX *cfl, const uint8_t *input, int input_stride, int row,
88
               int col, TX_SIZE tx_size, BLOCK_SIZE bsize);
89

Luc Trudeau's avatar
Luc Trudeau committed
90 91
void cfl_compute_parameters(MACROBLOCKD *const xd, TX_SIZE tx_size);

Luc Trudeau's avatar
Luc Trudeau committed
92
#endif  // AV1_COMMON_CFL_H_