drivers/staging/media/rkvdec/rkvdec-h264.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Rockchip Video Decoder H264 backend
   4  *
   5  * Copyright (C) 2019 Collabora, Ltd.
   6  *      Boris Brezillon <boris.brezillon@collabora.com>
   7  *
   8  * Copyright (C) 2016 Rockchip Electronics Co., Ltd.
   9  *      Jeffy Chen <jeffy.chen@rock-chips.com>
  10  */
  11
  12 #include <media/v4l2-h264.h>
  13 #include <media/v4l2-mem2mem.h>
  14
  15 #include "rkvdec.h"
  16 #include "rkvdec-regs.h"
  17
  18 /* Size with u32 units. */
  19 #define RKV_CABAC_INIT_BUFFER_SIZE      (3680 + 128)
  20 #define RKV_RPS_SIZE                    ((128 + 128) / 4)
  21 #define RKV_ERROR_INFO_SIZE             (256 * 144 * 4)
  22
  23 #define RKVDEC_NUM_REFLIST              3
  24
  25 struct rkvdec_h264_scaling_list {
  26         u8 scaling_list_4x4[6][16];
  27         u8 scaling_list_8x8[6][64];
  28         u8 padding[128];
  29 };
  30
  31 struct rkvdec_sps_pps_packet {
  32         u32 info[8];
  33 };
  34
  35 struct rkvdec_ps_field {
  36         u16 offset;
  37         u8 len;
  38 };
  39
  40 #define PS_FIELD(_offset, _len) \
  41         ((struct rkvdec_ps_field){ _offset, _len })
  42
  43 #define SEQ_PARAMETER_SET_ID                            PS_FIELD(0, 4)
  44 #define PROFILE_IDC                                     PS_FIELD(4, 8)
  45 #define CONSTRAINT_SET3_FLAG                            PS_FIELD(12, 1)
  46 #define CHROMA_FORMAT_IDC                               PS_FIELD(13, 2)
  47 #define BIT_DEPTH_LUMA                                  PS_FIELD(15, 3)
  48 #define BIT_DEPTH_CHROMA                                PS_FIELD(18, 3)
  49 #define QPPRIME_Y_ZERO_TRANSFORM_BYPASS_FLAG            PS_FIELD(21, 1)
  50 #define LOG2_MAX_FRAME_NUM_MINUS4                       PS_FIELD(22, 4)
  51 #define MAX_NUM_REF_FRAMES                              PS_FIELD(26, 5)
  52 #define PIC_ORDER_CNT_TYPE                              PS_FIELD(31, 2)
  53 #define LOG2_MAX_PIC_ORDER_CNT_LSB_MINUS4               PS_FIELD(33, 4)
  54 #define DELTA_PIC_ORDER_ALWAYS_ZERO_FLAG                PS_FIELD(37, 1)
  55 #define PIC_WIDTH_IN_MBS                                PS_FIELD(38, 9)
  56 #define PIC_HEIGHT_IN_MBS                               PS_FIELD(47, 9)
  57 #define FRAME_MBS_ONLY_FLAG                             PS_FIELD(56, 1)
  58 #define MB_ADAPTIVE_FRAME_FIELD_FLAG                    PS_FIELD(57, 1)
  59 #define DIRECT_8X8_INFERENCE_FLAG                       PS_FIELD(58, 1)
  60 #define MVC_EXTENSION_ENABLE                            PS_FIELD(59, 1)
  61 #define NUM_VIEWS                                       PS_FIELD(60, 2)
  62 #define VIEW_ID(i)                                      PS_FIELD(62 + ((i) * 10), 10)
  63 #define NUM_ANCHOR_REFS_L(i)                            PS_FIELD(82 + ((i) * 11), 1)
  64 #define ANCHOR_REF_L(i)                         PS_FIELD(83 + ((i) * 11), 10)
  65 #define NUM_NON_ANCHOR_REFS_L(i)                        PS_FIELD(104 + ((i) * 11), 1)
  66 #define NON_ANCHOR_REFS_L(i)                            PS_FIELD(105 + ((i) * 11), 10)
  67 #define PIC_PARAMETER_SET_ID                            PS_FIELD(128, 8)
  68 #define PPS_SEQ_PARAMETER_SET_ID                        PS_FIELD(136, 5)
  69 #define ENTROPY_CODING_MODE_FLAG                        PS_FIELD(141, 1)
  70 #define BOTTOM_FIELD_PIC_ORDER_IN_FRAME_PRESENT_FLAG    PS_FIELD(142, 1)
  71 #define NUM_REF_IDX_L_DEFAULT_ACTIVE_MINUS1(i)          PS_FIELD(143 + ((i) * 5), 5)
  72 #define WEIGHTED_PRED_FLAG                              PS_FIELD(153, 1)
  73 #define WEIGHTED_BIPRED_IDC                             PS_FIELD(154, 2)
  74 #define PIC_INIT_QP_MINUS26                             PS_FIELD(156, 7)
  75 #define PIC_INIT_QS_MINUS26                             PS_FIELD(163, 6)
  76 #define CHROMA_QP_INDEX_OFFSET                          PS_FIELD(169, 5)
  77 #define DEBLOCKING_FILTER_CONTROL_PRESENT_FLAG          PS_FIELD(174, 1)
  78 #define CONSTRAINED_INTRA_PRED_FLAG                     PS_FIELD(175, 1)
  79 #define REDUNDANT_PIC_CNT_PRESENT                       PS_FIELD(176, 1)
  80 #define TRANSFORM_8X8_MODE_FLAG                 PS_FIELD(177, 1)
  81 #define SECOND_CHROMA_QP_INDEX_OFFSET                   PS_FIELD(178, 5)
  82 #define SCALING_LIST_ENABLE_FLAG                        PS_FIELD(183, 1)
  83 #define SCALING_LIST_ADDRESS                            PS_FIELD(184, 32)
  84 #define IS_LONG_TERM(i)                         PS_FIELD(216 + (i), 1)
  85
  86 #define DPB_OFFS(i, j)                                  (288 + ((j) * 32 * 7) + ((i) * 7))
  87 #define DPB_INFO(i, j)                                  PS_FIELD(DPB_OFFS(i, j), 5)
  88 #define BOTTOM_FLAG(i, j)                               PS_FIELD(DPB_OFFS(i, j) + 5, 1)
  89 #define VIEW_INDEX_OFF(i, j)                            PS_FIELD(DPB_OFFS(i, j) + 6, 1)
  90
  91 /* Data structure describing auxiliary buffer format. */
  92 struct rkvdec_h264_priv_tbl {
  93         s8 cabac_table[4][464][2];
  94         struct rkvdec_h264_scaling_list scaling_list;
  95         u32 rps[RKV_RPS_SIZE];
  96         struct rkvdec_sps_pps_packet param_set[256];
  97         u8 err_info[RKV_ERROR_INFO_SIZE];
  98 };
  99
 100 #define RKVDEC_H264_DPB_SIZE 16
 101
 102 struct rkvdec_h264_reflists {
 103         u8 p[RKVDEC_H264_DPB_SIZE];
 104         u8 b0[RKVDEC_H264_DPB_SIZE];
 105         u8 b1[RKVDEC_H264_DPB_SIZE];
 106         u8 num_valid;
 107 };
 108
 109 struct rkvdec_h264_run {
 110         struct rkvdec_run base;
 111         const struct v4l2_ctrl_h264_decode_params *decode_params;
 112         const struct v4l2_ctrl_h264_sps *sps;
 113         const struct v4l2_ctrl_h264_pps *pps;
 114         const struct v4l2_ctrl_h264_scaling_matrix *scaling_matrix;
 115         int ref_buf_idx[V4L2_H264_NUM_DPB_ENTRIES];
 116 };
 117
 118 struct rkvdec_h264_ctx {
 119         struct rkvdec_aux_buf priv_tbl;
 120         struct rkvdec_h264_reflists reflists;
 121 };
 122
 123 #define CABAC_ENTRY(ctxidx, idc0_m, idc0_n, idc1_m, idc1_n,             \
 124             idc2_m, idc2_n, intra_m, intra_n)                   \
 125         [0][(ctxidx)] = {idc0_m, idc0_n},                       \
 126         [1][(ctxidx)] = {idc1_m, idc1_n},                       \
 127         [2][(ctxidx)] = {idc2_m, idc2_n},                       \
 128         [3][(ctxidx)] = {intra_m, intra_n}
 129
 130 /*
 131  * Constant CABAC table.
 132  * Built from the tables described in section '9.3.1.1 Initialisation process
 133  * for context variables' of the H264 spec.
 134  */
 135 static const s8 rkvdec_h264_cabac_table[4][464][2] = {
 136         /* Table 9-12 – Values of variables m and n for ctxIdx from 0 to 10 */
 137         CABAC_ENTRY(0, 20, -15, 20, -15, 20, -15, 20, -15),
 138         CABAC_ENTRY(1, 2, 54, 2, 54, 2, 54, 2, 54),
 139         CABAC_ENTRY(2, 3, 74, 3, 74, 3, 74, 3, 74),
 140         CABAC_ENTRY(3, 20, -15, 20, -15, 20, -15, 20, -15),
 141         CABAC_ENTRY(4, 2, 54, 2, 54, 2, 54, 2, 54),
 142         CABAC_ENTRY(5, 3, 74, 3, 74, 3, 74, 3, 74),
 143         CABAC_ENTRY(6, -28, 127, -28, 127, -28, 127, -28, 127),
 144         CABAC_ENTRY(7, -23, 104, -23, 104, -23, 104, -23, 104),
 145         CABAC_ENTRY(8, -6, 53, -6, 53, -6, 53, -6, 53),
 146         CABAC_ENTRY(9, -1, 54, -1, 54, -1, 54, -1, 54),
 147         CABAC_ENTRY(10, 7, 51, 7, 51, 7, 51, 7, 51),
 148
 149         /* Table 9-13 – Values of variables m and n for ctxIdx from 11 to 23 */
 150         CABAC_ENTRY(11, 23, 33, 22, 25, 29, 16, 0, 0),
 151         CABAC_ENTRY(12, 23, 2, 34, 0, 25, 0, 0, 0),
 152         CABAC_ENTRY(13, 21, 0, 16, 0, 14, 0, 0, 0),
 153         CABAC_ENTRY(14, 1, 9, -2, 9, -10, 51, 0, 0),
 154         CABAC_ENTRY(15, 0, 49, 4, 41, -3, 62, 0, 0),
 155         CABAC_ENTRY(16, -37, 118, -29, 118, -27, 99, 0, 0),
 156         CABAC_ENTRY(17, 5, 57, 2, 65, 26, 16, 0, 0),
 157         CABAC_ENTRY(18, -13, 78, -6, 71, -4, 85, 0, 0),
 158         CABAC_ENTRY(19, -11, 65, -13, 79, -24, 102, 0, 0),
 159         CABAC_ENTRY(20, 1, 62, 5, 52, 5, 57, 0, 0),
 160         CABAC_ENTRY(21, 12, 49, 9, 50, 6, 57, 0, 0),
 161         CABAC_ENTRY(22, -4, 73, -3, 70, -17, 73, 0, 0),
 162         CABAC_ENTRY(23, 17, 50, 10, 54, 14, 57, 0, 0),
 163
 164         /* Table 9-14 – Values of variables m and n for ctxIdx from 24 to 39 */
 165         CABAC_ENTRY(24, 18, 64, 26, 34, 20, 40, 0, 0),
 166         CABAC_ENTRY(25, 9, 43, 19, 22, 20, 10, 0, 0),
 167         CABAC_ENTRY(26, 29, 0, 40, 0, 29, 0, 0, 0),
 168         CABAC_ENTRY(27, 26, 67, 57, 2, 54, 0, 0, 0),
 169         CABAC_ENTRY(28, 16, 90, 41, 36, 37, 42, 0, 0),
 170         CABAC_ENTRY(29, 9, 104, 26, 69, 12, 97, 0, 0),
 171         CABAC_ENTRY(30, -46, 127, -45, 127, -32, 127, 0, 0),
 172         CABAC_ENTRY(31, -20, 104, -15, 101, -22, 117, 0, 0),
 173         CABAC_ENTRY(32, 1, 67, -4, 76, -2, 74, 0, 0),
 174         CABAC_ENTRY(33, -13, 78, -6, 71, -4, 85, 0, 0),
 175         CABAC_ENTRY(34, -11, 65, -13, 79, -24, 102, 0, 0),
 176         CABAC_ENTRY(35, 1, 62, 5, 52, 5, 57, 0, 0),
 177         CABAC_ENTRY(36, -6, 86, 6, 69, -6, 93, 0, 0),
 178         CABAC_ENTRY(37, -17, 95, -13, 90, -14, 88, 0, 0),
 179         CABAC_ENTRY(38, -6, 61, 0, 52, -6, 44, 0, 0),
 180         CABAC_ENTRY(39, 9, 45, 8, 43, 4, 55, 0, 0),
 181
 182         /* Table 9-15 – Values of variables m and n for ctxIdx from 40 to 53 */
 183         CABAC_ENTRY(40, -3, 69, -2, 69, -11, 89, 0, 0),
 184         CABAC_ENTRY(41, -6, 81, -5, 82, -15, 103, 0, 0),
 185         CABAC_ENTRY(42, -11, 96, -10, 96, -21, 116, 0, 0),
 186         CABAC_ENTRY(43, 6, 55, 2, 59, 19, 57, 0, 0),
 187         CABAC_ENTRY(44, 7, 67, 2, 75, 20, 58, 0, 0),
 188         CABAC_ENTRY(45, -5, 86, -3, 87, 4, 84, 0, 0),
 189         CABAC_ENTRY(46, 2, 88, -3, 100, 6, 96, 0, 0),
 190         CABAC_ENTRY(47, 0, 58, 1, 56, 1, 63, 0, 0),
 191         CABAC_ENTRY(48, -3, 76, -3, 74, -5, 85, 0, 0),
 192         CABAC_ENTRY(49, -10, 94, -6, 85, -13, 106, 0, 0),
 193         CABAC_ENTRY(50, 5, 54, 0, 59, 5, 63, 0, 0),
 194         CABAC_ENTRY(51, 4, 69, -3, 81, 6, 75, 0, 0),
 195         CABAC_ENTRY(52, -3, 81, -7, 86, -3, 90, 0, 0),
 196         CABAC_ENTRY(53, 0, 88, -5, 95, -1, 101, 0, 0),
 197
 198         /* Table 9-16 – Values of variables m and n for ctxIdx from 54 to 59 */
 199         CABAC_ENTRY(54, -7, 67, -1, 66, 3, 55, 0, 0),
 200         CABAC_ENTRY(55, -5, 74, -1, 77, -4, 79, 0, 0),
 201         CABAC_ENTRY(56, -4, 74, 1, 70, -2, 75, 0, 0),
 202         CABAC_ENTRY(57, -5, 80, -2, 86, -12, 97, 0, 0),
 203         CABAC_ENTRY(58, -7, 72, -5, 72, -7, 50, 0, 0),
 204         CABAC_ENTRY(59, 1, 58, 0, 61, 1, 60, 0, 0),
 205
 206         /* Table 9-17 – Values of variables m and n for ctxIdx from 60 to 69 */
 207         CABAC_ENTRY(60, 0, 41, 0, 41, 0, 41, 0, 41),
 208         CABAC_ENTRY(61, 0, 63, 0, 63, 0, 63, 0, 63),
 209         CABAC_ENTRY(62, 0, 63, 0, 63, 0, 63, 0, 63),
 210         CABAC_ENTRY(63, 0, 63, 0, 63, 0, 63, 0, 63),
 211         CABAC_ENTRY(64, -9, 83, -9, 83, -9, 83, -9, 83),
 212         CABAC_ENTRY(65, 4, 86, 4, 86, 4, 86, 4, 86),
 213         CABAC_ENTRY(66, 0, 97, 0, 97, 0, 97, 0, 97),
 214         CABAC_ENTRY(67, -7, 72, -7, 72, -7, 72, -7, 72),
 215         CABAC_ENTRY(68, 13, 41, 13, 41, 13, 41, 13, 41),
 216         CABAC_ENTRY(69, 3, 62, 3, 62, 3, 62, 3, 62),
 217
 218         /* Table 9-18 – Values of variables m and n for ctxIdx from 70 to 104 */
 219         CABAC_ENTRY(70, 0, 45, 13, 15, 7, 34, 0, 11),
 220         CABAC_ENTRY(71, -4, 78, 7, 51, -9, 88, 1, 55),
 221         CABAC_ENTRY(72, -3, 96, 2, 80, -20, 127, 0, 69),
 222         CABAC_ENTRY(73, -27, 126, -39, 127, -36, 127, -17, 127),
 223         CABAC_ENTRY(74, -28, 98, -18, 91, -17, 91, -13, 102),
 224         CABAC_ENTRY(75, -25, 101, -17, 96, -14, 95, 0, 82),
 225         CABAC_ENTRY(76, -23, 67, -26, 81, -25, 84, -7, 74),
 226         CABAC_ENTRY(77, -28, 82, -35, 98, -25, 86, -21, 107),
 227         CABAC_ENTRY(78, -20, 94, -24, 102, -12, 89, -27, 127),
 228         CABAC_ENTRY(79, -16, 83, -23, 97, -17, 91, -31, 127),
 229         CABAC_ENTRY(80, -22, 110, -27, 119, -31, 127, -24, 127),
 230         CABAC_ENTRY(81, -21, 91, -24, 99, -14, 76, -18, 95),
 231         CABAC_ENTRY(82, -18, 102, -21, 110, -18, 103, -27, 127),
 232         CABAC_ENTRY(83, -13, 93, -18, 102, -13, 90, -21, 114),
 233         CABAC_ENTRY(84, -29, 127, -36, 127, -37, 127, -30, 127),
 234         CABAC_ENTRY(85, -7, 92, 0, 80, 11, 80, -17, 123),
 235         CABAC_ENTRY(86, -5, 89, -5, 89, 5, 76, -12, 115),
 236         CABAC_ENTRY(87, -7, 96, -7, 94, 2, 84, -16, 122),
 237         CABAC_ENTRY(88, -13, 108, -4, 92, 5, 78, -11, 115),
 238         CABAC_ENTRY(89, -3, 46, 0, 39, -6, 55, -12, 63),
 239         CABAC_ENTRY(90, -1, 65, 0, 65, 4, 61, -2, 68),
 240         CABAC_ENTRY(91, -1, 57, -15, 84, -14, 83, -15, 84),
 241         CABAC_ENTRY(92, -9, 93, -35, 127, -37, 127, -13, 104),
 242         CABAC_ENTRY(93, -3, 74, -2, 73, -5, 79, -3, 70),
 243         CABAC_ENTRY(94, -9, 92, -12, 104, -11, 104, -8, 93),
 244         CABAC_ENTRY(95, -8, 87, -9, 91, -11, 91, -10, 90),
 245         CABAC_ENTRY(96, -23, 126, -31, 127, -30, 127, -30, 127),
 246         CABAC_ENTRY(97, 5, 54, 3, 55, 0, 65, -1, 74),
 247         CABAC_ENTRY(98, 6, 60, 7, 56, -2, 79, -6, 97),
 248         CABAC_ENTRY(99, 6, 59, 7, 55, 0, 72, -7, 91),
 249         CABAC_ENTRY(100, 6, 69, 8, 61, -4, 92, -20, 127),
 250         CABAC_ENTRY(101, -1, 48, -3, 53, -6, 56, -4, 56),
 251         CABAC_ENTRY(102, 0, 68, 0, 68, 3, 68, -5, 82),
 252         CABAC_ENTRY(103, -4, 69, -7, 74, -8, 71, -7, 76),
 253         CABAC_ENTRY(104, -8, 88, -9, 88, -13, 98, -22, 125),
 254
 255         /* Table 9-19 – Values of variables m and n for ctxIdx from 105 to 165 */
 256         CABAC_ENTRY(105, -2, 85, -13, 103, -4, 86, -7, 93),
 257         CABAC_ENTRY(106, -6, 78, -13, 91, -12, 88, -11, 87),
 258         CABAC_ENTRY(107, -1, 75, -9, 89, -5, 82, -3, 77),
 259         CABAC_ENTRY(108, -7, 77, -14, 92, -3, 72, -5, 71),
 260         CABAC_ENTRY(109, 2, 54, -8, 76, -4, 67, -4, 63),
 261         CABAC_ENTRY(110, 5, 50, -12, 87, -8, 72, -4, 68),
 262         CABAC_ENTRY(111, -3, 68, -23, 110, -16, 89, -12, 84),
 263         CABAC_ENTRY(112, 1, 50, -24, 105, -9, 69, -7, 62),
 264         CABAC_ENTRY(113, 6, 42, -10, 78, -1, 59, -7, 65),
 265         CABAC_ENTRY(114, -4, 81, -20, 112, 5, 66, 8, 61),
 266         CABAC_ENTRY(115, 1, 63, -17, 99, 4, 57, 5, 56),
 267         CABAC_ENTRY(116, -4, 70, -78, 127, -4, 71, -2, 66),
 268         CABAC_ENTRY(117, 0, 67, -70, 127, -2, 71, 1, 64),
 269         CABAC_ENTRY(118, 2, 57, -50, 127, 2, 58, 0, 61),
 270         CABAC_ENTRY(119, -2, 76, -46, 127, -1, 74, -2, 78),
 271         CABAC_ENTRY(120, 11, 35, -4, 66, -4, 44, 1, 50),
 272         CABAC_ENTRY(121, 4, 64, -5, 78, -1, 69, 7, 52),
 273         CABAC_ENTRY(122, 1, 61, -4, 71, 0, 62, 10, 35),
 274         CABAC_ENTRY(123, 11, 35, -8, 72, -7, 51, 0, 44),
 275         CABAC_ENTRY(124, 18, 25, 2, 59, -4, 47, 11, 38),
 276         CABAC_ENTRY(125, 12, 24, -1, 55, -6, 42, 1, 45),
 277         CABAC_ENTRY(126, 13, 29, -7, 70, -3, 41, 0, 46),
 278         CABAC_ENTRY(127, 13, 36, -6, 75, -6, 53, 5, 44),
 279         CABAC_ENTRY(128, -10, 93, -8, 89, 8, 76, 31, 17),
 280         CABAC_ENTRY(129, -7, 73, -34, 119, -9, 78, 1, 51),
 281         CABAC_ENTRY(130, -2, 73, -3, 75, -11, 83, 7, 50),
 282         CABAC_ENTRY(131, 13, 46, 32, 20, 9, 52, 28, 19),
 283         CABAC_ENTRY(132, 9, 49, 30, 22, 0, 67, 16, 33),
 284         CABAC_ENTRY(133, -7, 100, -44, 127, -5, 90, 14, 62),
 285         CABAC_ENTRY(134, 9, 53, 0, 54, 1, 67, -13, 108),
 286         CABAC_ENTRY(135, 2, 53, -5, 61, -15, 72, -15, 100),
 287         CABAC_ENTRY(136, 5, 53, 0, 58, -5, 75, -13, 101),
 288         CABAC_ENTRY(137, -2, 61, -1, 60, -8, 80, -13, 91),
 289         CABAC_ENTRY(138, 0, 56, -3, 61, -21, 83, -12, 94),
 290         CABAC_ENTRY(139, 0, 56, -8, 67, -21, 64, -10, 88),
 291         CABAC_ENTRY(140, -13, 63, -25, 84, -13, 31, -16, 84),
 292         CABAC_ENTRY(141, -5, 60, -14, 74, -25, 64, -10, 86),
 293         CABAC_ENTRY(142, -1, 62, -5, 65, -29, 94, -7, 83),
 294         CABAC_ENTRY(143, 4, 57, 5, 52, 9, 75, -13, 87),
 295         CABAC_ENTRY(144, -6, 69, 2, 57, 17, 63, -19, 94),
 296         CABAC_ENTRY(145, 4, 57, 0, 61, -8, 74, 1, 70),
 297         CABAC_ENTRY(146, 14, 39, -9, 69, -5, 35, 0, 72),
 298         CABAC_ENTRY(147, 4, 51, -11, 70, -2, 27, -5, 74),
 299         CABAC_ENTRY(148, 13, 68, 18, 55, 13, 91, 18, 59),
 300         CABAC_ENTRY(149, 3, 64, -4, 71, 3, 65, -8, 102),
 301         CABAC_ENTRY(150, 1, 61, 0, 58, -7, 69, -15, 100),
 302         CABAC_ENTRY(151, 9, 63, 7, 61, 8, 77, 0, 95),
 303         CABAC_ENTRY(152, 7, 50, 9, 41, -10, 66, -4, 75),
 304         CABAC_ENTRY(153, 16, 39, 18, 25, 3, 62, 2, 72),
 305         CABAC_ENTRY(154, 5, 44, 9, 32, -3, 68, -11, 75),
 306         CABAC_ENTRY(155, 4, 52, 5, 43, -20, 81, -3, 71),
 307         CABAC_ENTRY(156, 11, 48, 9, 47, 0, 30, 15, 46),
 308         CABAC_ENTRY(157, -5, 60, 0, 44, 1, 7, -13, 69),
 309         CABAC_ENTRY(158, -1, 59, 0, 51, -3, 23, 0, 62),
 310         CABAC_ENTRY(159, 0, 59, 2, 46, -21, 74, 0, 65),
 311         CABAC_ENTRY(160, 22, 33, 19, 38, 16, 66, 21, 37),
 312         CABAC_ENTRY(161, 5, 44, -4, 66, -23, 124, -15, 72),
 313         CABAC_ENTRY(162, 14, 43, 15, 38, 17, 37, 9, 57),
 314         CABAC_ENTRY(163, -1, 78, 12, 42, 44, -18, 16, 54),
 315         CABAC_ENTRY(164, 0, 60, 9, 34, 50, -34, 0, 62),
 316         CABAC_ENTRY(165, 9, 69, 0, 89, -22, 127, 12, 72),
 317
 318         /* Table 9-20 – Values of variables m and n for ctxIdx from 166 to 226 */
 319         CABAC_ENTRY(166, 11, 28, 4, 45, 4, 39, 24, 0),
 320         CABAC_ENTRY(167, 2, 40, 10, 28, 0, 42, 15, 9),
 321         CABAC_ENTRY(168, 3, 44, 10, 31, 7, 34, 8, 25),
 322         CABAC_ENTRY(169, 0, 49, 33, -11, 11, 29, 13, 18),
 323         CABAC_ENTRY(170, 0, 46, 52, -43, 8, 31, 15, 9),
 324         CABAC_ENTRY(171, 2, 44, 18, 15, 6, 37, 13, 19),
 325         CABAC_ENTRY(172, 2, 51, 28, 0, 7, 42, 10, 37),
 326         CABAC_ENTRY(173, 0, 47, 35, -22, 3, 40, 12, 18),
 327         CABAC_ENTRY(174, 4, 39, 38, -25, 8, 33, 6, 29),
 328         CABAC_ENTRY(175, 2, 62, 34, 0, 13, 43, 20, 33),
 329         CABAC_ENTRY(176, 6, 46, 39, -18, 13, 36, 15, 30),
 330         CABAC_ENTRY(177, 0, 54, 32, -12, 4, 47, 4, 45),
 331         CABAC_ENTRY(178, 3, 54, 102, -94, 3, 55, 1, 58),
 332         CABAC_ENTRY(179, 2, 58, 0, 0, 2, 58, 0, 62),
 333         CABAC_ENTRY(180, 4, 63, 56, -15, 6, 60, 7, 61),
 334         CABAC_ENTRY(181, 6, 51, 33, -4, 8, 44, 12, 38),
 335         CABAC_ENTRY(182, 6, 57, 29, 10, 11, 44, 11, 45),
 336         CABAC_ENTRY(183, 7, 53, 37, -5, 14, 42, 15, 39),
 337         CABAC_ENTRY(184, 6, 52, 51, -29, 7, 48, 11, 42),
 338         CABAC_ENTRY(185, 6, 55, 39, -9, 4, 56, 13, 44),
 339         CABAC_ENTRY(186, 11, 45, 52, -34, 4, 52, 16, 45),
 340         CABAC_ENTRY(187, 14, 36, 69, -58, 13, 37, 12, 41),
 341         CABAC_ENTRY(188, 8, 53, 67, -63, 9, 49, 10, 49),
 342         CABAC_ENTRY(189, -1, 82, 44, -5, 19, 58, 30, 34),
 343         CABAC_ENTRY(190, 7, 55, 32, 7, 10, 48, 18, 42),
 344         CABAC_ENTRY(191, -3, 78, 55, -29, 12, 45, 10, 55),
 345         CABAC_ENTRY(192, 15, 46, 32, 1, 0, 69, 17, 51),
 346         CABAC_ENTRY(193, 22, 31, 0, 0, 20, 33, 17, 46),
 347         CABAC_ENTRY(194, -1, 84, 27, 36, 8, 63, 0, 89),
 348         CABAC_ENTRY(195, 25, 7, 33, -25, 35, -18, 26, -19),
 349         CABAC_ENTRY(196, 30, -7, 34, -30, 33, -25, 22, -17),
 350         CABAC_ENTRY(197, 28, 3, 36, -28, 28, -3, 26, -17),
 351         CABAC_ENTRY(198, 28, 4, 38, -28, 24, 10, 30, -25),
 352         CABAC_ENTRY(199, 32, 0, 38, -27, 27, 0, 28, -20),
 353         CABAC_ENTRY(200, 34, -1, 34, -18, 34, -14, 33, -23),
 354         CABAC_ENTRY(201, 30, 6, 35, -16, 52, -44, 37, -27),
 355         CABAC_ENTRY(202, 30, 6, 34, -14, 39, -24, 33, -23),
 356         CABAC_ENTRY(203, 32, 9, 32, -8, 19, 17, 40, -28),
 357         CABAC_ENTRY(204, 31, 19, 37, -6, 31, 25, 38, -17),
 358         CABAC_ENTRY(205, 26, 27, 35, 0, 36, 29, 33, -11),
 359         CABAC_ENTRY(206, 26, 30, 30, 10, 24, 33, 40, -15),
 360         CABAC_ENTRY(207, 37, 20, 28, 18, 34, 15, 41, -6),
 361         CABAC_ENTRY(208, 28, 34, 26, 25, 30, 20, 38, 1),
 362         CABAC_ENTRY(209, 17, 70, 29, 41, 22, 73, 41, 17),
 363         CABAC_ENTRY(210, 1, 67, 0, 75, 20, 34, 30, -6),
 364         CABAC_ENTRY(211, 5, 59, 2, 72, 19, 31, 27, 3),
 365         CABAC_ENTRY(212, 9, 67, 8, 77, 27, 44, 26, 22),
 366         CABAC_ENTRY(213, 16, 30, 14, 35, 19, 16, 37, -16),
 367         CABAC_ENTRY(214, 18, 32, 18, 31, 15, 36, 35, -4),
 368         CABAC_ENTRY(215, 18, 35, 17, 35, 15, 36, 38, -8),
 369         CABAC_ENTRY(216, 22, 29, 21, 30, 21, 28, 38, -3),
 370         CABAC_ENTRY(217, 24, 31, 17, 45, 25, 21, 37, 3),
 371         CABAC_ENTRY(218, 23, 38, 20, 42, 30, 20, 38, 5),
 372         CABAC_ENTRY(219, 18, 43, 18, 45, 31, 12, 42, 0),
 373         CABAC_ENTRY(220, 20, 41, 27, 26, 27, 16, 35, 16),
 374         CABAC_ENTRY(221, 11, 63, 16, 54, 24, 42, 39, 22),
 375         CABAC_ENTRY(222, 9, 59, 7, 66, 0, 93, 14, 48),
 376         CABAC_ENTRY(223, 9, 64, 16, 56, 14, 56, 27, 37),
 377         CABAC_ENTRY(224, -1, 94, 11, 73, 15, 57, 21, 60),
 378         CABAC_ENTRY(225, -2, 89, 10, 67, 26, 38, 12, 68),
 379         CABAC_ENTRY(226, -9, 108, -10, 116, -24, 127, 2, 97),
 380
 381         /* Table 9-21 – Values of variables m and n for ctxIdx from 227 to 275 */
 382         CABAC_ENTRY(227, -6, 76, -23, 112, -24, 115, -3, 71),
 383         CABAC_ENTRY(228, -2, 44, -15, 71, -22, 82, -6, 42),
 384         CABAC_ENTRY(229, 0, 45, -7, 61, -9, 62, -5, 50),
 385         CABAC_ENTRY(230, 0, 52, 0, 53, 0, 53, -3, 54),
 386         CABAC_ENTRY(231, -3, 64, -5, 66, 0, 59, -2, 62),
 387         CABAC_ENTRY(232, -2, 59, -11, 77, -14, 85, 0, 58),
 388         CABAC_ENTRY(233, -4, 70, -9, 80, -13, 89, 1, 63),
 389         CABAC_ENTRY(234, -4, 75, -9, 84, -13, 94, -2, 72),
 390         CABAC_ENTRY(235, -8, 82, -10, 87, -11, 92, -1, 74),
 391         CABAC_ENTRY(236, -17, 102, -34, 127, -29, 127, -9, 91),
 392         CABAC_ENTRY(237, -9, 77, -21, 101, -21, 100, -5, 67),
 393         CABAC_ENTRY(238, 3, 24, -3, 39, -14, 57, -5, 27),
 394         CABAC_ENTRY(239, 0, 42, -5, 53, -12, 67, -3, 39),
 395         CABAC_ENTRY(240, 0, 48, -7, 61, -11, 71, -2, 44),
 396         CABAC_ENTRY(241, 0, 55, -11, 75, -10, 77, 0, 46),
 397         CABAC_ENTRY(242, -6, 59, -15, 77, -21, 85, -16, 64),
 398         CABAC_ENTRY(243, -7, 71, -17, 91, -16, 88, -8, 68),
 399         CABAC_ENTRY(244, -12, 83, -25, 107, -23, 104, -10, 78),
 400         CABAC_ENTRY(245, -11, 87, -25, 111, -15, 98, -6, 77),
 401         CABAC_ENTRY(246, -30, 119, -28, 122, -37, 127, -10, 86),
 402         CABAC_ENTRY(247, 1, 58, -11, 76, -10, 82, -12, 92),
 403         CABAC_ENTRY(248, -3, 29, -10, 44, -8, 48, -15, 55),
 404         CABAC_ENTRY(249, -1, 36, -10, 52, -8, 61, -10, 60),
 405         CABAC_ENTRY(250, 1, 38, -10, 57, -8, 66, -6, 62),
 406         CABAC_ENTRY(251, 2, 43, -9, 58, -7, 70, -4, 65),
 407         CABAC_ENTRY(252, -6, 55, -16, 72, -14, 75, -12, 73),
 408         CABAC_ENTRY(253, 0, 58, -7, 69, -10, 79, -8, 76),
 409         CABAC_ENTRY(254, 0, 64, -4, 69, -9, 83, -7, 80),
 410         CABAC_ENTRY(255, -3, 74, -5, 74, -12, 92, -9, 88),
 411         CABAC_ENTRY(256, -10, 90, -9, 86, -18, 108, -17, 110),
 412         CABAC_ENTRY(257, 0, 70, 2, 66, -4, 79, -11, 97),
 413         CABAC_ENTRY(258, -4, 29, -9, 34, -22, 69, -20, 84),
 414         CABAC_ENTRY(259, 5, 31, 1, 32, -16, 75, -11, 79),
 415         CABAC_ENTRY(260, 7, 42, 11, 31, -2, 58, -6, 73),
 416         CABAC_ENTRY(261, 1, 59, 5, 52, 1, 58, -4, 74),
 417         CABAC_ENTRY(262, -2, 58, -2, 55, -13, 78, -13, 86),
 418         CABAC_ENTRY(263, -3, 72, -2, 67, -9, 83, -13, 96),
 419         CABAC_ENTRY(264, -3, 81, 0, 73, -4, 81, -11, 97),
 420         CABAC_ENTRY(265, -11, 97, -8, 89, -13, 99, -19, 117),
 421         CABAC_ENTRY(266, 0, 58, 3, 52, -13, 81, -8, 78),
 422         CABAC_ENTRY(267, 8, 5, 7, 4, -6, 38, -5, 33),
 423         CABAC_ENTRY(268, 10, 14, 10, 8, -13, 62, -4, 48),
 424         CABAC_ENTRY(269, 14, 18, 17, 8, -6, 58, -2, 53),
 425         CABAC_ENTRY(270, 13, 27, 16, 19, -2, 59, -3, 62),
 426         CABAC_ENTRY(271, 2, 40, 3, 37, -16, 73, -13, 71),
 427         CABAC_ENTRY(272, 0, 58, -1, 61, -10, 76, -10, 79),
 428         CABAC_ENTRY(273, -3, 70, -5, 73, -13, 86, -12, 86),
 429         CABAC_ENTRY(274, -6, 79, -1, 70, -9, 83, -13, 90),
 430         CABAC_ENTRY(275, -8, 85, -4, 78, -10, 87, -14, 97),
 431
 432         /* Table 9-22 – Values of variables m and n for ctxIdx from 277 to 337 */
 433         CABAC_ENTRY(277, -13, 106, -21, 126, -22, 127, -6, 93),
 434         CABAC_ENTRY(278, -16, 106, -23, 124, -25, 127, -6, 84),
 435         CABAC_ENTRY(279, -10, 87, -20, 110, -25, 120, -8, 79),
 436         CABAC_ENTRY(280, -21, 114, -26, 126, -27, 127, 0, 66),
 437         CABAC_ENTRY(281, -18, 110, -25, 124, -19, 114, -1, 71),
 438         CABAC_ENTRY(282, -14, 98, -17, 105, -23, 117, 0, 62),
 439         CABAC_ENTRY(283, -22, 110, -27, 121, -25, 118, -2, 60),
 440         CABAC_ENTRY(284, -21, 106, -27, 117, -26, 117, -2, 59),
 441         CABAC_ENTRY(285, -18, 103, -17, 102, -24, 113, -5, 75),
 442         CABAC_ENTRY(286, -21, 107, -26, 117, -28, 118, -3, 62),
 443         CABAC_ENTRY(287, -23, 108, -27, 116, -31, 120, -4, 58),
 444         CABAC_ENTRY(288, -26, 112, -33, 122, -37, 124, -9, 66),
 445         CABAC_ENTRY(289, -10, 96, -10, 95, -10, 94, -1, 79),
 446         CABAC_ENTRY(290, -12, 95, -14, 100, -15, 102, 0, 71),
 447         CABAC_ENTRY(291, -5, 91, -8, 95, -10, 99, 3, 68),
 448         CABAC_ENTRY(292, -9, 93, -17, 111, -13, 106, 10, 44),
 449         CABAC_ENTRY(293, -22, 94, -28, 114, -50, 127, -7, 62),
 450         CABAC_ENTRY(294, -5, 86, -6, 89, -5, 92, 15, 36),
 451         CABAC_ENTRY(295, 9, 67, -2, 80, 17, 57, 14, 40),
 452         CABAC_ENTRY(296, -4, 80, -4, 82, -5, 86, 16, 27),
 453         CABAC_ENTRY(297, -10, 85, -9, 85, -13, 94, 12, 29),
 454         CABAC_ENTRY(298, -1, 70, -8, 81, -12, 91, 1, 44),
 455         CABAC_ENTRY(299, 7, 60, -1, 72, -2, 77, 20, 36),
 456         CABAC_ENTRY(300, 9, 58, 5, 64, 0, 71, 18, 32),
 457         CABAC_ENTRY(301, 5, 61, 1, 67, -1, 73, 5, 42),
 458         CABAC_ENTRY(302, 12, 50, 9, 56, 4, 64, 1, 48),
 459         CABAC_ENTRY(303, 15, 50, 0, 69, -7, 81, 10, 62),
 460         CABAC_ENTRY(304, 18, 49, 1, 69, 5, 64, 17, 46),
 461         CABAC_ENTRY(305, 17, 54, 7, 69, 15, 57, 9, 64),
 462         CABAC_ENTRY(306, 10, 41, -7, 69, 1, 67, -12, 104),
 463         CABAC_ENTRY(307, 7, 46, -6, 67, 0, 68, -11, 97),
 464         CABAC_ENTRY(308, -1, 51, -16, 77, -10, 67, -16, 96),
 465         CABAC_ENTRY(309, 7, 49, -2, 64, 1, 68, -7, 88),
 466         CABAC_ENTRY(310, 8, 52, 2, 61, 0, 77, -8, 85),
 467         CABAC_ENTRY(311, 9, 41, -6, 67, 2, 64, -7, 85),
 468         CABAC_ENTRY(312, 6, 47, -3, 64, 0, 68, -9, 85),
 469         CABAC_ENTRY(313, 2, 55, 2, 57, -5, 78, -13, 88),
 470         CABAC_ENTRY(314, 13, 41, -3, 65, 7, 55, 4, 66),
 471         CABAC_ENTRY(315, 10, 44, -3, 66, 5, 59, -3, 77),
 472         CABAC_ENTRY(316, 6, 50, 0, 62, 2, 65, -3, 76),
 473         CABAC_ENTRY(317, 5, 53, 9, 51, 14, 54, -6, 76),
 474         CABAC_ENTRY(318, 13, 49, -1, 66, 15, 44, 10, 58),
 475         CABAC_ENTRY(319, 4, 63, -2, 71, 5, 60, -1, 76),
 476         CABAC_ENTRY(320, 6, 64, -2, 75, 2, 70, -1, 83),
 477         CABAC_ENTRY(321, -2, 69, -1, 70, -2, 76, -7, 99),
 478         CABAC_ENTRY(322, -2, 59, -9, 72, -18, 86, -14, 95),
 479         CABAC_ENTRY(323, 6, 70, 14, 60, 12, 70, 2, 95),
 480         CABAC_ENTRY(324, 10, 44, 16, 37, 5, 64, 0, 76),
 481         CABAC_ENTRY(325, 9, 31, 0, 47, -12, 70, -5, 74),
 482         CABAC_ENTRY(326, 12, 43, 18, 35, 11, 55, 0, 70),
 483         CABAC_ENTRY(327, 3, 53, 11, 37, 5, 56, -11, 75),
 484         CABAC_ENTRY(328, 14, 34, 12, 41, 0, 69, 1, 68),
 485         CABAC_ENTRY(329, 10, 38, 10, 41, 2, 65, 0, 65),
 486         CABAC_ENTRY(330, -3, 52, 2, 48, -6, 74, -14, 73),
 487         CABAC_ENTRY(331, 13, 40, 12, 41, 5, 54, 3, 62),
 488         CABAC_ENTRY(332, 17, 32, 13, 41, 7, 54, 4, 62),
 489         CABAC_ENTRY(333, 7, 44, 0, 59, -6, 76, -1, 68),
 490         CABAC_ENTRY(334, 7, 38, 3, 50, -11, 82, -13, 75),
 491         CABAC_ENTRY(335, 13, 50, 19, 40, -2, 77, 11, 55),
 492         CABAC_ENTRY(336, 10, 57, 3, 66, -2, 77, 5, 64),
 493         CABAC_ENTRY(337, 26, 43, 18, 50, 25, 42, 12, 70),
 494
 495         /* Table 9-23 – Values of variables m and n for ctxIdx from 338 to 398 */
 496         CABAC_ENTRY(338, 14, 11, 19, -6, 17, -13, 15, 6),
 497         CABAC_ENTRY(339, 11, 14, 18, -6, 16, -9, 6, 19),
 498         CABAC_ENTRY(340, 9, 11, 14, 0, 17, -12, 7, 16),
 499         CABAC_ENTRY(341, 18, 11, 26, -12, 27, -21, 12, 14),
 500         CABAC_ENTRY(342, 21, 9, 31, -16, 37, -30, 18, 13),
 501         CABAC_ENTRY(343, 23, -2, 33, -25, 41, -40, 13, 11),
 502         CABAC_ENTRY(344, 32, -15, 33, -22, 42, -41, 13, 15),
 503         CABAC_ENTRY(345, 32, -15, 37, -28, 48, -47, 15, 16),
 504         CABAC_ENTRY(346, 34, -21, 39, -30, 39, -32, 12, 23),
 505         CABAC_ENTRY(347, 39, -23, 42, -30, 46, -40, 13, 23),
 506         CABAC_ENTRY(348, 42, -33, 47, -42, 52, -51, 15, 20),
 507         CABAC_ENTRY(349, 41, -31, 45, -36, 46, -41, 14, 26),
 508         CABAC_ENTRY(350, 46, -28, 49, -34, 52, -39, 14, 44),
 509         CABAC_ENTRY(351, 38, -12, 41, -17, 43, -19, 17, 40),
 510         CABAC_ENTRY(352, 21, 29, 32, 9, 32, 11, 17, 47),
 511         CABAC_ENTRY(353, 45, -24, 69, -71, 61, -55, 24, 17),
 512         CABAC_ENTRY(354, 53, -45, 63, -63, 56, -46, 21, 21),
 513         CABAC_ENTRY(355, 48, -26, 66, -64, 62, -50, 25, 22),
 514         CABAC_ENTRY(356, 65, -43, 77, -74, 81, -67, 31, 27),
 515         CABAC_ENTRY(357, 43, -19, 54, -39, 45, -20, 22, 29),
 516         CABAC_ENTRY(358, 39, -10, 52, -35, 35, -2, 19, 35),
 517         CABAC_ENTRY(359, 30, 9, 41, -10, 28, 15, 14, 50),
 518         CABAC_ENTRY(360, 18, 26, 36, 0, 34, 1, 10, 57),
 519         CABAC_ENTRY(361, 20, 27, 40, -1, 39, 1, 7, 63),
 520         CABAC_ENTRY(362, 0, 57, 30, 14, 30, 17, -2, 77),
 521         CABAC_ENTRY(363, -14, 82, 28, 26, 20, 38, -4, 82),
 522         CABAC_ENTRY(364, -5, 75, 23, 37, 18, 45, -3, 94),
 523         CABAC_ENTRY(365, -19, 97, 12, 55, 15, 54, 9, 69),
 524         CABAC_ENTRY(366, -35, 125, 11, 65, 0, 79, -12, 109),
 525         CABAC_ENTRY(367, 27, 0, 37, -33, 36, -16, 36, -35),
 526         CABAC_ENTRY(368, 28, 0, 39, -36, 37, -14, 36, -34),
 527         CABAC_ENTRY(369, 31, -4, 40, -37, 37, -17, 32, -26),
 528         CABAC_ENTRY(370, 27, 6, 38, -30, 32, 1, 37, -30),
 529         CABAC_ENTRY(371, 34, 8, 46, -33, 34, 15, 44, -32),
 530         CABAC_ENTRY(372, 30, 10, 42, -30, 29, 15, 34, -18),
 531         CABAC_ENTRY(373, 24, 22, 40, -24, 24, 25, 34, -15),
 532         CABAC_ENTRY(374, 33, 19, 49, -29, 34, 22, 40, -15),
 533         CABAC_ENTRY(375, 22, 32, 38, -12, 31, 16, 33, -7),
 534         CABAC_ENTRY(376, 26, 31, 40, -10, 35, 18, 35, -5),
 535         CABAC_ENTRY(377, 21, 41, 38, -3, 31, 28, 33, 0),
 536         CABAC_ENTRY(378, 26, 44, 46, -5, 33, 41, 38, 2),
 537         CABAC_ENTRY(379, 23, 47, 31, 20, 36, 28, 33, 13),
 538         CABAC_ENTRY(380, 16, 65, 29, 30, 27, 47, 23, 35),
 539         CABAC_ENTRY(381, 14, 71, 25, 44, 21, 62, 13, 58),
 540         CABAC_ENTRY(382, 8, 60, 12, 48, 18, 31, 29, -3),
 541         CABAC_ENTRY(383, 6, 63, 11, 49, 19, 26, 26, 0),
 542         CABAC_ENTRY(384, 17, 65, 26, 45, 36, 24, 22, 30),
 543         CABAC_ENTRY(385, 21, 24, 22, 22, 24, 23, 31, -7),
 544         CABAC_ENTRY(386, 23, 20, 23, 22, 27, 16, 35, -15),
 545         CABAC_ENTRY(387, 26, 23, 27, 21, 24, 30, 34, -3),
 546         CABAC_ENTRY(388, 27, 32, 33, 20, 31, 29, 34, 3),
 547         CABAC_ENTRY(389, 28, 23, 26, 28, 22, 41, 36, -1),
 548         CABAC_ENTRY(390, 28, 24, 30, 24, 22, 42, 34, 5),
 549         CABAC_ENTRY(391, 23, 40, 27, 34, 16, 60, 32, 11),
 550         CABAC_ENTRY(392, 24, 32, 18, 42, 15, 52, 35, 5),
 551         CABAC_ENTRY(393, 28, 29, 25, 39, 14, 60, 34, 12),
 552         CABAC_ENTRY(394, 23, 42, 18, 50, 3, 78, 39, 11),
 553         CABAC_ENTRY(395, 19, 57, 12, 70, -16, 123, 30, 29),
 554         CABAC_ENTRY(396, 22, 53, 21, 54, 21, 53, 34, 26),
 555         CABAC_ENTRY(397, 22, 61, 14, 71, 22, 56, 29, 39),
 556         CABAC_ENTRY(398, 11, 86, 11, 83, 25, 61, 19, 66),
 557
 558         /* Values of variables m and n for ctxIdx from 399 to 463 (not documented) */
 559         CABAC_ENTRY(399, 12, 40, 25, 32, 21, 33, 31, 21),
 560         CABAC_ENTRY(400, 11, 51, 21, 49, 19, 50, 31, 31),
 561         CABAC_ENTRY(401, 14, 59, 21, 54, 17, 61, 25, 50),
 562         CABAC_ENTRY(402, -4, 79, -5, 85, -3, 78, -17, 120),
 563         CABAC_ENTRY(403, -7, 71, -6, 81, -8, 74, -20, 112),
 564         CABAC_ENTRY(404, -5, 69, -10, 77, -9, 72, -18, 114),
 565         CABAC_ENTRY(405, -9, 70, -7, 81, -10, 72, -11, 85),
 566         CABAC_ENTRY(406, -8, 66, -17, 80, -18, 75, -15, 92),
 567         CABAC_ENTRY(407, -10, 68, -18, 73, -12, 71, -14, 89),
 568         CABAC_ENTRY(408, -19, 73, -4, 74, -11, 63, -26, 71),
 569         CABAC_ENTRY(409, -12, 69, -10, 83, -5, 70, -15, 81),
 570         CABAC_ENTRY(410, -16, 70, -9, 71, -17, 75, -14, 80),
 571         CABAC_ENTRY(411, -15, 67, -9, 67, -14, 72, 0, 68),
 572         CABAC_ENTRY(412, -20, 62, -1, 61, -16, 67, -14, 70),
 573         CABAC_ENTRY(413, -19, 70, -8, 66, -8, 53, -24, 56),
 574         CABAC_ENTRY(414, -16, 66, -14, 66, -14, 59, -23, 68),
 575         CABAC_ENTRY(415, -22, 65, 0, 59, -9, 52, -24, 50),
 576         CABAC_ENTRY(416, -20, 63, 2, 59, -11, 68, -11, 74),
 577         CABAC_ENTRY(417, 9, -2, 17, -10, 9, -2, 23, -13),
 578         CABAC_ENTRY(418, 26, -9, 32, -13, 30, -10, 26, -13),
 579         CABAC_ENTRY(419, 33, -9, 42, -9, 31, -4, 40, -15),
 580         CABAC_ENTRY(420, 39, -7, 49, -5, 33, -1, 49, -14),
 581         CABAC_ENTRY(421, 41, -2, 53, 0, 33, 7, 44, 3),
 582         CABAC_ENTRY(422, 45, 3, 64, 3, 31, 12, 45, 6),
 583         CABAC_ENTRY(423, 49, 9, 68, 10, 37, 23, 44, 34),
 584         CABAC_ENTRY(424, 45, 27, 66, 27, 31, 38, 33, 54),
 585         CABAC_ENTRY(425, 36, 59, 47, 57, 20, 64, 19, 82),
 586         CABAC_ENTRY(426, -6, 66, -5, 71, -9, 71, -3, 75),
 587         CABAC_ENTRY(427, -7, 35, 0, 24, -7, 37, -1, 23),
 588         CABAC_ENTRY(428, -7, 42, -1, 36, -8, 44, 1, 34),
 589         CABAC_ENTRY(429, -8, 45, -2, 42, -11, 49, 1, 43),
 590         CABAC_ENTRY(430, -5, 48, -2, 52, -10, 56, 0, 54),
 591         CABAC_ENTRY(431, -12, 56, -9, 57, -12, 59, -2, 55),
 592         CABAC_ENTRY(432, -6, 60, -6, 63, -8, 63, 0, 61),
 593         CABAC_ENTRY(433, -5, 62, -4, 65, -9, 67, 1, 64),
 594         CABAC_ENTRY(434, -8, 66, -4, 67, -6, 68, 0, 68),
 595         CABAC_ENTRY(435, -8, 76, -7, 82, -10, 79, -9, 92),
 596         CABAC_ENTRY(436, -5, 85, -3, 81, -3, 78, -14, 106),
 597         CABAC_ENTRY(437, -6, 81, -3, 76, -8, 74, -13, 97),
 598         CABAC_ENTRY(438, -10, 77, -7, 72, -9, 72, -15, 90),
 599         CABAC_ENTRY(439, -7, 81, -6, 78, -10, 72, -12, 90),
 600         CABAC_ENTRY(440, -17, 80, -12, 72, -18, 75, -18, 88),
 601         CABAC_ENTRY(441, -18, 73, -14, 68, -12, 71, -10, 73),
 602         CABAC_ENTRY(442, -4, 74, -3, 70, -11, 63, -9, 79),
 603         CABAC_ENTRY(443, -10, 83, -6, 76, -5, 70, -14, 86),
 604         CABAC_ENTRY(444, -9, 71, -5, 66, -17, 75, -10, 73),
 605         CABAC_ENTRY(445, -9, 67, -5, 62, -14, 72, -10, 70),
 606         CABAC_ENTRY(446, -1, 61, 0, 57, -16, 67, -10, 69),
 607         CABAC_ENTRY(447, -8, 66, -4, 61, -8, 53, -5, 66),
 608         CABAC_ENTRY(448, -14, 66, -9, 60, -14, 59, -9, 64),
 609         CABAC_ENTRY(449, 0, 59, 1, 54, -9, 52, -5, 58),
 610         CABAC_ENTRY(450, 2, 59, 2, 58, -11, 68, 2, 59),
 611         CABAC_ENTRY(451, 21, -13, 17, -10, 9, -2, 21, -10),
 612         CABAC_ENTRY(452, 33, -14, 32, -13, 30, -10, 24, -11),
 613         CABAC_ENTRY(453, 39, -7, 42, -9, 31, -4, 28, -8),
 614         CABAC_ENTRY(454, 46, -2, 49, -5, 33, -1, 28, -1),
 615         CABAC_ENTRY(455, 51, 2, 53, 0, 33, 7, 29, 3),
 616         CABAC_ENTRY(456, 60, 6, 64, 3, 31, 12, 29, 9),
 617         CABAC_ENTRY(457, 61, 17, 68, 10, 37, 23, 35, 20),
 618         CABAC_ENTRY(458, 55, 34, 66, 27, 31, 38, 29, 36),
 619         CABAC_ENTRY(459, 42, 62, 47, 57, 20, 64, 14, 67),
 620 };
 621
 622 static void set_ps_field(u32 *buf, struct rkvdec_ps_field field, u32 value)
 623 {
 624         u8 bit = field.offset % 32, word = field.offset / 32;
 625         u64 mask = GENMASK_ULL(bit + field.len - 1, bit);
 626         u64 val = ((u64)value << bit) & mask;
 627
 628         buf[word] &= ~mask;
 629         buf[word] |= val;
 630         if (bit + field.len > 32) {
 631                 buf[word + 1] &= ~(mask >> 32);
 632                 buf[word + 1] |= val >> 32;
 633         }
 634 }
 635
 636 static void assemble_hw_pps(struct rkvdec_ctx *ctx,
 637                             struct rkvdec_h264_run *run)
 638 {
 639         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
 640         const struct v4l2_ctrl_h264_sps *sps = run->sps;
 641         const struct v4l2_ctrl_h264_pps *pps = run->pps;
 642         const struct v4l2_ctrl_h264_decode_params *dec_params = run->decode_params;
 643         const struct v4l2_h264_dpb_entry *dpb = dec_params->dpb;
 644         struct rkvdec_h264_priv_tbl *priv_tbl = h264_ctx->priv_tbl.cpu;
 645         struct rkvdec_sps_pps_packet *hw_ps;
 646         dma_addr_t scaling_list_address;
 647         u32 scaling_distance;
 648         u32 i;
 649
 650         /*
 651          * HW read the SPS/PPS information from PPS packet index by PPS id.
 652          * offset from the base can be calculated by PPS_id * 32 (size per PPS
 653          * packet unit). so the driver copy SPS/PPS information to the exact PPS
 654          * packet unit for HW accessing.
 655          */
 656         hw_ps = &priv_tbl->param_set[pps->pic_parameter_set_id];
 657         memset(hw_ps, 0, sizeof(*hw_ps));
 658
 659 #define WRITE_PPS(value, field) set_ps_field(hw_ps->info, field, value)
 660         /* write sps */
 661         WRITE_PPS(0xf, SEQ_PARAMETER_SET_ID);
 662         WRITE_PPS(0xff, PROFILE_IDC);
 663         WRITE_PPS(1, CONSTRAINT_SET3_FLAG);
 664         WRITE_PPS(sps->chroma_format_idc, CHROMA_FORMAT_IDC);
 665         WRITE_PPS(sps->bit_depth_luma_minus8, BIT_DEPTH_LUMA);
 666         WRITE_PPS(sps->bit_depth_chroma_minus8, BIT_DEPTH_CHROMA);
 667         WRITE_PPS(0, QPPRIME_Y_ZERO_TRANSFORM_BYPASS_FLAG);
 668         WRITE_PPS(sps->log2_max_frame_num_minus4, LOG2_MAX_FRAME_NUM_MINUS4);
 669         WRITE_PPS(sps->max_num_ref_frames, MAX_NUM_REF_FRAMES);
 670         WRITE_PPS(sps->pic_order_cnt_type, PIC_ORDER_CNT_TYPE);
 671         WRITE_PPS(sps->log2_max_pic_order_cnt_lsb_minus4,
 672                   LOG2_MAX_PIC_ORDER_CNT_LSB_MINUS4);
 673         WRITE_PPS(!!(sps->flags & V4L2_H264_SPS_FLAG_DELTA_PIC_ORDER_ALWAYS_ZERO),
 674                   DELTA_PIC_ORDER_ALWAYS_ZERO_FLAG);
 675         WRITE_PPS(DIV_ROUND_UP(ctx->coded_fmt.fmt.pix_mp.width, 16), PIC_WIDTH_IN_MBS);
 676         WRITE_PPS(DIV_ROUND_UP(ctx->coded_fmt.fmt.pix_mp.height, 16), PIC_HEIGHT_IN_MBS);
 677         WRITE_PPS(!!(sps->flags & V4L2_H264_SPS_FLAG_FRAME_MBS_ONLY),
 678                   FRAME_MBS_ONLY_FLAG);
 679         WRITE_PPS(!!(sps->flags & V4L2_H264_SPS_FLAG_MB_ADAPTIVE_FRAME_FIELD),
 680                   MB_ADAPTIVE_FRAME_FIELD_FLAG);
 681         WRITE_PPS(!!(sps->flags & V4L2_H264_SPS_FLAG_DIRECT_8X8_INFERENCE),
 682                   DIRECT_8X8_INFERENCE_FLAG);
 683
 684         /* write pps */
 685         WRITE_PPS(0xff, PIC_PARAMETER_SET_ID);
 686         WRITE_PPS(0x1f, PPS_SEQ_PARAMETER_SET_ID);
 687         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_ENTROPY_CODING_MODE),
 688                   ENTROPY_CODING_MODE_FLAG);
 689         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_BOTTOM_FIELD_PIC_ORDER_IN_FRAME_PRESENT),
 690                   BOTTOM_FIELD_PIC_ORDER_IN_FRAME_PRESENT_FLAG);
 691         WRITE_PPS(pps->num_ref_idx_l0_default_active_minus1,
 692                   NUM_REF_IDX_L_DEFAULT_ACTIVE_MINUS1(0));
 693         WRITE_PPS(pps->num_ref_idx_l1_default_active_minus1,
 694                   NUM_REF_IDX_L_DEFAULT_ACTIVE_MINUS1(1));
 695         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_WEIGHTED_PRED),
 696                   WEIGHTED_PRED_FLAG);
 697         WRITE_PPS(pps->weighted_bipred_idc, WEIGHTED_BIPRED_IDC);
 698         WRITE_PPS(pps->pic_init_qp_minus26, PIC_INIT_QP_MINUS26);
 699         WRITE_PPS(pps->pic_init_qs_minus26, PIC_INIT_QS_MINUS26);
 700         WRITE_PPS(pps->chroma_qp_index_offset, CHROMA_QP_INDEX_OFFSET);
 701         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_DEBLOCKING_FILTER_CONTROL_PRESENT),
 702                   DEBLOCKING_FILTER_CONTROL_PRESENT_FLAG);
 703         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_CONSTRAINED_INTRA_PRED),
 704                   CONSTRAINED_INTRA_PRED_FLAG);
 705         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_REDUNDANT_PIC_CNT_PRESENT),
 706                   REDUNDANT_PIC_CNT_PRESENT);
 707         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_TRANSFORM_8X8_MODE),
 708                   TRANSFORM_8X8_MODE_FLAG);
 709         WRITE_PPS(pps->second_chroma_qp_index_offset,
 710                   SECOND_CHROMA_QP_INDEX_OFFSET);
 711
 712         WRITE_PPS(!!(pps->flags & V4L2_H264_PPS_FLAG_SCALING_MATRIX_PRESENT),
 713                   SCALING_LIST_ENABLE_FLAG);
 714         /* To be on the safe side, program the scaling matrix address */
 715         scaling_distance = offsetof(struct rkvdec_h264_priv_tbl, scaling_list);
 716         scaling_list_address = h264_ctx->priv_tbl.dma + scaling_distance;
 717         WRITE_PPS(scaling_list_address, SCALING_LIST_ADDRESS);
 718
 719         for (i = 0; i < ARRAY_SIZE(dec_params->dpb); i++) {
 720                 u32 is_longterm = 0;
 721
 722                 if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_LONG_TERM)
 723                         is_longterm = 1;
 724
 725                 WRITE_PPS(is_longterm, IS_LONG_TERM(i));
 726         }
 727 }
 728
 729 static void lookup_ref_buf_idx(struct rkvdec_ctx *ctx,
 730                                struct rkvdec_h264_run *run)
 731 {
 732         const struct v4l2_ctrl_h264_decode_params *dec_params = run->decode_params;
 733         u32 i;
 734
 735         for (i = 0; i < ARRAY_SIZE(dec_params->dpb); i++) {
 736                 struct v4l2_m2m_ctx *m2m_ctx = ctx->fh.m2m_ctx;
 737                 const struct v4l2_h264_dpb_entry *dpb = run->decode_params->dpb;
 738                 struct vb2_queue *cap_q = &m2m_ctx->cap_q_ctx.q;
 739                 int buf_idx = -1;
 740
 741                 if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_ACTIVE)
 742                         buf_idx = vb2_find_timestamp(cap_q,
 743                                                      dpb[i].reference_ts, 0);
 744
 745                 run->ref_buf_idx[i] = buf_idx;
 746         }
 747 }
 748
 749 static void assemble_hw_rps(struct rkvdec_ctx *ctx,
 750                             struct rkvdec_h264_run *run)
 751 {
 752         const struct v4l2_ctrl_h264_decode_params *dec_params = run->decode_params;
 753         const struct v4l2_h264_dpb_entry *dpb = dec_params->dpb;
 754         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
 755         const struct v4l2_ctrl_h264_sps *sps = run->sps;
 756         struct rkvdec_h264_priv_tbl *priv_tbl = h264_ctx->priv_tbl.cpu;
 757         u32 max_frame_num = 1 << (sps->log2_max_frame_num_minus4 + 4);
 758
 759         u32 *hw_rps = priv_tbl->rps;
 760         u32 i, j;
 761         u16 *p = (u16 *)hw_rps;
 762
 763         memset(hw_rps, 0, sizeof(priv_tbl->rps));
 764
 765         /*
 766          * Assign an invalid pic_num if DPB entry at that position is inactive.
 767          * If we assign 0 in that position hardware will treat that as a real
 768          * reference picture with pic_num 0, triggering output picture
 769          * corruption.
 770          */
 771         for (i = 0; i < ARRAY_SIZE(dec_params->dpb); i++) {
 772                 if (!(dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_ACTIVE))
 773                         continue;
 774
 775                 if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_LONG_TERM ||
 776                     dpb[i].frame_num < dec_params->frame_num) {
 777                         p[i] = dpb[i].frame_num;
 778                         continue;
 779                 }
 780
 781                 p[i] = dpb[i].frame_num - max_frame_num;
 782         }
 783
 784         for (j = 0; j < RKVDEC_NUM_REFLIST; j++) {
 785                 for (i = 0; i < h264_ctx->reflists.num_valid; i++) {
 786                         bool dpb_valid = run->ref_buf_idx[i] >= 0;
 787                         u8 idx = 0;
 788
 789                         switch (j) {
 790                         case 0:
 791                                 idx = h264_ctx->reflists.p[i];
 792                                 break;
 793                         case 1:
 794                                 idx = h264_ctx->reflists.b0[i];
 795                                 break;
 796                         case 2:
 797                                 idx = h264_ctx->reflists.b1[i];
 798                                 break;
 799                         }
 800
 801                         if (idx >= ARRAY_SIZE(dec_params->dpb))
 802                                 continue;
 803
 804                         set_ps_field(hw_rps, DPB_INFO(i, j),
 805                                      idx | dpb_valid << 4);
 806                 }
 807         }
 808 }
 809
 810 static void assemble_hw_scaling_list(struct rkvdec_ctx *ctx,
 811                                      struct rkvdec_h264_run *run)
 812 {
 813         const struct v4l2_ctrl_h264_scaling_matrix *scaling = run->scaling_matrix;
 814         const struct v4l2_ctrl_h264_pps *pps = run->pps;
 815         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
 816         struct rkvdec_h264_priv_tbl *tbl = h264_ctx->priv_tbl.cpu;
 817
 818         if (!(pps->flags & V4L2_H264_PPS_FLAG_SCALING_MATRIX_PRESENT))
 819                 return;
 820
 821         BUILD_BUG_ON(sizeof(tbl->scaling_list.scaling_list_4x4) !=
 822                      sizeof(scaling->scaling_list_4x4));
 823         BUILD_BUG_ON(sizeof(tbl->scaling_list.scaling_list_8x8) !=
 824                      sizeof(scaling->scaling_list_8x8));
 825
 826         memcpy(tbl->scaling_list.scaling_list_4x4,
 827                scaling->scaling_list_4x4,
 828                sizeof(scaling->scaling_list_4x4));
 829
 830         memcpy(tbl->scaling_list.scaling_list_8x8,
 831                scaling->scaling_list_8x8,
 832                sizeof(scaling->scaling_list_8x8));
 833 }
 834
 835 /*
 836  * dpb poc related registers table
 837  */
 838 static const u32 poc_reg_tbl_top_field[16] = {
 839         RKVDEC_REG_H264_POC_REFER0(0),
 840         RKVDEC_REG_H264_POC_REFER0(2),
 841         RKVDEC_REG_H264_POC_REFER0(4),
 842         RKVDEC_REG_H264_POC_REFER0(6),
 843         RKVDEC_REG_H264_POC_REFER0(8),
 844         RKVDEC_REG_H264_POC_REFER0(10),
 845         RKVDEC_REG_H264_POC_REFER0(12),
 846         RKVDEC_REG_H264_POC_REFER0(14),
 847         RKVDEC_REG_H264_POC_REFER1(1),
 848         RKVDEC_REG_H264_POC_REFER1(3),
 849         RKVDEC_REG_H264_POC_REFER1(5),
 850         RKVDEC_REG_H264_POC_REFER1(7),
 851         RKVDEC_REG_H264_POC_REFER1(9),
 852         RKVDEC_REG_H264_POC_REFER1(11),
 853         RKVDEC_REG_H264_POC_REFER1(13),
 854         RKVDEC_REG_H264_POC_REFER2(0)
 855 };
 856
 857 static const u32 poc_reg_tbl_bottom_field[16] = {
 858         RKVDEC_REG_H264_POC_REFER0(1),
 859         RKVDEC_REG_H264_POC_REFER0(3),
 860         RKVDEC_REG_H264_POC_REFER0(5),
 861         RKVDEC_REG_H264_POC_REFER0(7),
 862         RKVDEC_REG_H264_POC_REFER0(9),
 863         RKVDEC_REG_H264_POC_REFER0(11),
 864         RKVDEC_REG_H264_POC_REFER0(13),
 865         RKVDEC_REG_H264_POC_REFER1(0),
 866         RKVDEC_REG_H264_POC_REFER1(2),
 867         RKVDEC_REG_H264_POC_REFER1(4),
 868         RKVDEC_REG_H264_POC_REFER1(6),
 869         RKVDEC_REG_H264_POC_REFER1(8),
 870         RKVDEC_REG_H264_POC_REFER1(10),
 871         RKVDEC_REG_H264_POC_REFER1(12),
 872         RKVDEC_REG_H264_POC_REFER1(14),
 873         RKVDEC_REG_H264_POC_REFER2(1)
 874 };
 875
 876 static struct vb2_buffer *
 877 get_ref_buf(struct rkvdec_ctx *ctx, struct rkvdec_h264_run *run,
 878             unsigned int dpb_idx)
 879 {
 880         struct v4l2_m2m_ctx *m2m_ctx = ctx->fh.m2m_ctx;
 881         struct vb2_queue *cap_q = &m2m_ctx->cap_q_ctx.q;
 882         int buf_idx = run->ref_buf_idx[dpb_idx];
 883
 884         /*
 885          * If a DPB entry is unused or invalid, address of current destination
 886          * buffer is returned.
 887          */
 888         if (buf_idx < 0)
 889                 return &run->base.bufs.dst->vb2_buf;
 890
 891         return vb2_get_buffer(cap_q, buf_idx);
 892 }
 893
 894 static void config_registers(struct rkvdec_ctx *ctx,
 895                              struct rkvdec_h264_run *run)
 896 {
 897         struct rkvdec_dev *rkvdec = ctx->dev;
 898         const struct v4l2_ctrl_h264_decode_params *dec_params = run->decode_params;
 899         const struct v4l2_ctrl_h264_sps *sps = run->sps;
 900         const struct v4l2_h264_dpb_entry *dpb = dec_params->dpb;
 901         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
 902         dma_addr_t priv_start_addr = h264_ctx->priv_tbl.dma;
 903         const struct v4l2_pix_format_mplane *dst_fmt;
 904         struct vb2_v4l2_buffer *src_buf = run->base.bufs.src;
 905         struct vb2_v4l2_buffer *dst_buf = run->base.bufs.dst;
 906         const struct v4l2_format *f;
 907         dma_addr_t rlc_addr;
 908         dma_addr_t refer_addr;
 909         u32 rlc_len;
 910         u32 hor_virstride = 0;
 911         u32 ver_virstride = 0;
 912         u32 y_virstride = 0;
 913         u32 yuv_virstride = 0;
 914         u32 offset;
 915         dma_addr_t dst_addr;
 916         u32 reg, i;
 917
 918         reg = RKVDEC_MODE(RKVDEC_MODE_H264);
 919         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_SYSCTRL);
 920
 921         f = &ctx->decoded_fmt;
 922         dst_fmt = &f->fmt.pix_mp;
 923         hor_virstride = (sps->bit_depth_luma_minus8 + 8) * dst_fmt->width / 8;
 924         ver_virstride = round_up(dst_fmt->height, 16);
 925         y_virstride = hor_virstride * ver_virstride;
 926
 927         if (sps->chroma_format_idc == 0)
 928                 yuv_virstride = y_virstride;
 929         else if (sps->chroma_format_idc == 1)
 930                 yuv_virstride += y_virstride + y_virstride / 2;
 931         else if (sps->chroma_format_idc == 2)
 932                 yuv_virstride += 2 * y_virstride;
 933
 934         reg = RKVDEC_Y_HOR_VIRSTRIDE(hor_virstride / 16) |
 935               RKVDEC_UV_HOR_VIRSTRIDE(hor_virstride / 16) |
 936               RKVDEC_SLICE_NUM_HIGHBIT |
 937               RKVDEC_SLICE_NUM_LOWBITS(0x7ff);
 938         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_PICPAR);
 939
 940         /* config rlc base address */
 941         rlc_addr = vb2_dma_contig_plane_dma_addr(&src_buf->vb2_buf, 0);
 942         writel_relaxed(rlc_addr, rkvdec->regs + RKVDEC_REG_STRM_RLC_BASE);
 943         writel_relaxed(rlc_addr, rkvdec->regs + RKVDEC_REG_RLCWRITE_BASE);
 944
 945         rlc_len = vb2_get_plane_payload(&src_buf->vb2_buf, 0);
 946         reg = RKVDEC_STRM_LEN(rlc_len);
 947         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_STRM_LEN);
 948
 949         /* config cabac table */
 950         offset = offsetof(struct rkvdec_h264_priv_tbl, cabac_table);
 951         writel_relaxed(priv_start_addr + offset,
 952                        rkvdec->regs + RKVDEC_REG_CABACTBL_PROB_BASE);
 953
 954         /* config output base address */
 955         dst_addr = vb2_dma_contig_plane_dma_addr(&dst_buf->vb2_buf, 0);
 956         writel_relaxed(dst_addr, rkvdec->regs + RKVDEC_REG_DECOUT_BASE);
 957
 958         reg = RKVDEC_Y_VIRSTRIDE(y_virstride / 16);
 959         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_Y_VIRSTRIDE);
 960
 961         reg = RKVDEC_YUV_VIRSTRIDE(yuv_virstride / 16);
 962         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_YUV_VIRSTRIDE);
 963
 964         /* config ref pic address & poc */
 965         for (i = 0; i < ARRAY_SIZE(dec_params->dpb); i++) {
 966                 struct vb2_buffer *vb_buf = get_ref_buf(ctx, run, i);
 967
 968                 refer_addr = vb2_dma_contig_plane_dma_addr(vb_buf, 0);
 969
 970                 if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_ACTIVE)
 971                         refer_addr |= RKVDEC_COLMV_USED_FLAG_REF;
 972                 if (dpb[i].flags & V4L2_H264_DPB_ENTRY_FLAG_FIELD)
 973                         refer_addr |= RKVDEC_FIELD_REF;
 974
 975                 if (dpb[i].fields & V4L2_H264_TOP_FIELD_REF)
 976                         refer_addr |= RKVDEC_TOPFIELD_USED_REF;
 977                 if (dpb[i].fields & V4L2_H264_BOTTOM_FIELD_REF)
 978                         refer_addr |= RKVDEC_BOTFIELD_USED_REF;
 979
 980                 writel_relaxed(dpb[i].top_field_order_cnt,
 981                                rkvdec->regs +  poc_reg_tbl_top_field[i]);
 982                 writel_relaxed(dpb[i].bottom_field_order_cnt,
 983                                rkvdec->regs + poc_reg_tbl_bottom_field[i]);
 984
 985                 if (i < V4L2_H264_NUM_DPB_ENTRIES - 1)
 986                         writel_relaxed(refer_addr,
 987                                        rkvdec->regs + RKVDEC_REG_H264_BASE_REFER(i));
 988                 else
 989                         writel_relaxed(refer_addr,
 990                                        rkvdec->regs + RKVDEC_REG_H264_BASE_REFER15);
 991         }
 992
 993         /*
 994          * Since support frame mode only
 995          * top_field_order_cnt is the same as bottom_field_order_cnt
 996          */
 997         reg = RKVDEC_CUR_POC(dec_params->top_field_order_cnt);
 998         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_CUR_POC0);
 999
1000         reg = RKVDEC_CUR_POC(dec_params->bottom_field_order_cnt);
1001         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_CUR_POC1);
1002
1003         /* config hw pps address */
1004         offset = offsetof(struct rkvdec_h264_priv_tbl, param_set);
1005         writel_relaxed(priv_start_addr + offset,
1006                        rkvdec->regs + RKVDEC_REG_PPS_BASE);
1007
1008         /* config hw rps address */
1009         offset = offsetof(struct rkvdec_h264_priv_tbl, rps);
1010         writel_relaxed(priv_start_addr + offset,
1011                        rkvdec->regs + RKVDEC_REG_RPS_BASE);
1012
1013         reg = RKVDEC_AXI_DDR_RDATA(0);
1014         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_AXI_DDR_RDATA);
1015
1016         reg = RKVDEC_AXI_DDR_WDATA(0);
1017         writel_relaxed(reg, rkvdec->regs + RKVDEC_REG_AXI_DDR_WDATA);
1018
1019         offset = offsetof(struct rkvdec_h264_priv_tbl, err_info);
1020         writel_relaxed(priv_start_addr + offset,
1021                        rkvdec->regs + RKVDEC_REG_H264_ERRINFO_BASE);
1022 }
1023
1024 #define RKVDEC_H264_MAX_DEPTH_IN_BYTES          2
1025
1026 static int rkvdec_h264_adjust_fmt(struct rkvdec_ctx *ctx,
1027                                   struct v4l2_format *f)
1028 {
1029         struct v4l2_pix_format_mplane *fmt = &f->fmt.pix_mp;
1030
1031         fmt->num_planes = 1;
1032         if (!fmt->plane_fmt[0].sizeimage)
1033                 fmt->plane_fmt[0].sizeimage = fmt->width * fmt->height *
1034                                               RKVDEC_H264_MAX_DEPTH_IN_BYTES;
1035         return 0;
1036 }
1037
1038 static int rkvdec_h264_start(struct rkvdec_ctx *ctx)
1039 {
1040         struct rkvdec_dev *rkvdec = ctx->dev;
1041         struct rkvdec_h264_priv_tbl *priv_tbl;
1042         struct rkvdec_h264_ctx *h264_ctx;
1043         int ret;
1044
1045         h264_ctx = kzalloc(sizeof(*h264_ctx), GFP_KERNEL);
1046         if (!h264_ctx)
1047                 return -ENOMEM;
1048
1049         priv_tbl = dma_alloc_coherent(rkvdec->dev, sizeof(*priv_tbl),
1050                                       &h264_ctx->priv_tbl.dma, GFP_KERNEL);
1051         if (!priv_tbl) {
1052                 ret = -ENOMEM;
1053                 goto err_free_ctx;
1054         }
1055
1056         h264_ctx->priv_tbl.size = sizeof(*priv_tbl);
1057         h264_ctx->priv_tbl.cpu = priv_tbl;
1058         memcpy(priv_tbl->cabac_table, rkvdec_h264_cabac_table,
1059                sizeof(rkvdec_h264_cabac_table));
1060
1061         ctx->priv = h264_ctx;
1062         return 0;
1063
1064 err_free_ctx:
1065         kfree(h264_ctx);
1066         return ret;
1067 }
1068
1069 static void rkvdec_h264_stop(struct rkvdec_ctx *ctx)
1070 {
1071         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
1072         struct rkvdec_dev *rkvdec = ctx->dev;
1073
1074         dma_free_coherent(rkvdec->dev, h264_ctx->priv_tbl.size,
1075                           h264_ctx->priv_tbl.cpu, h264_ctx->priv_tbl.dma);
1076         kfree(h264_ctx);
1077 }
1078
1079 static void rkvdec_h264_run_preamble(struct rkvdec_ctx *ctx,
1080                                      struct rkvdec_h264_run *run)
1081 {
1082         struct v4l2_ctrl *ctrl;
1083
1084         ctrl = v4l2_ctrl_find(&ctx->ctrl_hdl,
1085                               V4L2_CID_MPEG_VIDEO_H264_DECODE_PARAMS);
1086         run->decode_params = ctrl ? ctrl->p_cur.p : NULL;
1087         ctrl = v4l2_ctrl_find(&ctx->ctrl_hdl,
1088                               V4L2_CID_MPEG_VIDEO_H264_SPS);
1089         run->sps = ctrl ? ctrl->p_cur.p : NULL;
1090         ctrl = v4l2_ctrl_find(&ctx->ctrl_hdl,
1091                               V4L2_CID_MPEG_VIDEO_H264_PPS);
1092         run->pps = ctrl ? ctrl->p_cur.p : NULL;
1093         ctrl = v4l2_ctrl_find(&ctx->ctrl_hdl,
1094                               V4L2_CID_MPEG_VIDEO_H264_SCALING_MATRIX);
1095         run->scaling_matrix = ctrl ? ctrl->p_cur.p : NULL;
1096
1097         rkvdec_run_preamble(ctx, &run->base);
1098 }
1099
1100 static int rkvdec_h264_run(struct rkvdec_ctx *ctx)
1101 {
1102         struct v4l2_h264_reflist_builder reflist_builder;
1103         struct rkvdec_dev *rkvdec = ctx->dev;
1104         struct rkvdec_h264_ctx *h264_ctx = ctx->priv;
1105         struct rkvdec_h264_run run;
1106
1107         rkvdec_h264_run_preamble(ctx, &run);
1108
1109         /* Build the P/B{0,1} ref lists. */
1110         v4l2_h264_init_reflist_builder(&reflist_builder, run.decode_params,
1111                                        run.sps, run.decode_params->dpb);
1112         h264_ctx->reflists.num_valid = reflist_builder.num_valid;
1113         v4l2_h264_build_p_ref_list(&reflist_builder, h264_ctx->reflists.p);
1114         v4l2_h264_build_b_ref_lists(&reflist_builder, h264_ctx->reflists.b0,
1115                                     h264_ctx->reflists.b1);
1116
1117         assemble_hw_scaling_list(ctx, &run);
1118         assemble_hw_pps(ctx, &run);
1119         lookup_ref_buf_idx(ctx, &run);
1120         assemble_hw_rps(ctx, &run);
1121         config_registers(ctx, &run);
1122
1123         rkvdec_run_postamble(ctx, &run.base);
1124
1125         schedule_delayed_work(&rkvdec->watchdog_work, msecs_to_jiffies(2000));
1126
1127         writel(0, rkvdec->regs + RKVDEC_REG_STRMD_ERR_EN);
1128         writel(0, rkvdec->regs + RKVDEC_REG_H264_ERR_E);
1129         writel(1, rkvdec->regs + RKVDEC_REG_PREF_LUMA_CACHE_COMMAND);
1130         writel(1, rkvdec->regs + RKVDEC_REG_PREF_CHR_CACHE_COMMAND);
1131
1132         /* Start decoding! */
1133         writel(RKVDEC_INTERRUPT_DEC_E | RKVDEC_CONFIG_DEC_CLK_GATE_E |
1134                RKVDEC_TIMEOUT_E | RKVDEC_BUF_EMPTY_E,
1135                rkvdec->regs + RKVDEC_REG_INTERRUPT);
1136
1137         return 0;
1138 }
1139
1140 const struct rkvdec_coded_fmt_ops rkvdec_h264_fmt_ops = {
1141         .adjust_fmt = rkvdec_h264_adjust_fmt,
1142         .start = rkvdec_h264_start,
1143         .stop = rkvdec_h264_stop,
1144         .run = rkvdec_h264_run,
1145 };