1/* SPDX-License-Identifier: GPL-2.0-or-later */ 2/* 3 * Helper functions for vp9 codecs. 4 * 5 * Copyright (c) 2021 Collabora, Ltd. 6 * 7 * Author: Andrzej Pietrasiewicz <andrzej.p@collabora.com> 8 */ 9 10#ifndef _MEDIA_V4L2_VP9_H 11#define _MEDIA_V4L2_VP9_H 12 13#include <media/v4l2-ctrls.h> 14 15/** 16 * struct v4l2_vp9_frame_mv_context - motion vector-related probabilities 17 * 18 * @joint: motion vector joint probabilities. 19 * @sign: motion vector sign probabilities. 20 * @classes: motion vector class probabilities. 21 * @class0_bit: motion vector class0 bit probabilities. 22 * @bits: motion vector bits probabilities. 23 * @class0_fr: motion vector class0 fractional bit probabilities. 24 * @fr: motion vector fractional bit probabilities. 25 * @class0_hp: motion vector class0 high precision fractional bit probabilities. 26 * @hp: motion vector high precision fractional bit probabilities. 27 * 28 * A member of v4l2_vp9_frame_context. 29 */ 30struct v4l2_vp9_frame_mv_context { 31 u8 joint[3]; 32 u8 sign[2]; 33 u8 classes[2][10]; 34 u8 class0_bit[2]; 35 u8 bits[2][10]; 36 u8 class0_fr[2][2][3]; 37 u8 fr[2][3]; 38 u8 class0_hp[2]; 39 u8 hp[2]; 40}; 41 42/** 43 * struct v4l2_vp9_frame_context - frame probabilities, including motion-vector related 44 * 45 * @tx8: TX 8x8 probabilities. 46 * @tx16: TX 16x16 probabilities. 47 * @tx32: TX 32x32 probabilities. 48 * @coef: coefficient probabilities. 49 * @skip: skip probabilities. 50 * @inter_mode: inter mode probabilities. 51 * @interp_filter: interpolation filter probabilities. 52 * @is_inter: is inter-block probabilities. 53 * @comp_mode: compound prediction mode probabilities. 54 * @single_ref: single ref probabilities. 55 * @comp_ref: compound ref probabilities. 56 * @y_mode: Y prediction mode probabilities. 57 * @uv_mode: UV prediction mode probabilities. 58 * @partition: partition probabilities. 59 * @mv: motion vector probabilities. 60 * 61 * Drivers which need to keep track of frame context(s) can use this struct. 62 * The members correspond to probability tables, which are specified only implicitly in the 63 * vp9 spec. Section 10.5 "Default probability tables" contains all the types of involved 64 * tables, i.e. the actual tables are of the same kind, and when they are reset (which is 65 * mandated by the spec sometimes) they are overwritten with values from the default tables. 66 */ 67struct v4l2_vp9_frame_context { 68 u8 tx8[2][1]; 69 u8 tx16[2][2]; 70 u8 tx32[2][3]; 71 u8 coef[4][2][2][6][6][3]; 72 u8 skip[3]; 73 u8 inter_mode[7][3]; 74 u8 interp_filter[4][2]; 75 u8 is_inter[4]; 76 u8 comp_mode[5]; 77 u8 single_ref[5][2]; 78 u8 comp_ref[5]; 79 u8 y_mode[4][9]; 80 u8 uv_mode[10][9]; 81 u8 partition[16][3]; 82 83 struct v4l2_vp9_frame_mv_context mv; 84}; 85 86/** 87 * struct v4l2_vp9_frame_symbol_counts - pointers to arrays of symbol counts 88 * 89 * @partition: partition counts. 90 * @skip: skip counts. 91 * @intra_inter: is inter-block counts. 92 * @tx32p: TX32 counts. 93 * @tx16p: TX16 counts. 94 * @tx8p: TX8 counts. 95 * @y_mode: Y prediction mode counts. 96 * @uv_mode: UV prediction mode counts. 97 * @comp: compound prediction mode counts. 98 * @comp_ref: compound ref counts. 99 * @single_ref: single ref counts. 100 * @mv_mode: inter mode counts. 101 * @filter: interpolation filter counts. 102 * @mv_joint: motion vector joint counts. 103 * @sign: motion vector sign counts. 104 * @classes: motion vector class counts. 105 * @class0: motion vector class0 bit counts. 106 * @bits: motion vector bits counts. 107 * @class0_fp: motion vector class0 fractional bit counts. 108 * @fp: motion vector fractional bit counts. 109 * @class0_hp: motion vector class0 high precision fractional bit counts. 110 * @hp: motion vector high precision fractional bit counts. 111 * @coeff: coefficient counts. 112 * @eob: eob counts 113 * 114 * The fields correspond to what is specified in section 8.3 "Clear counts process" of the spec. 115 * Different pieces of hardware can report the counts in different order, so we cannot rely on 116 * simply overlaying a struct on a relevant block of memory. Instead we provide pointers to 117 * arrays or array of pointers to arrays in case of coeff, or array of pointers for eob. 118 */ 119struct v4l2_vp9_frame_symbol_counts { 120 u32 (*partition)[16][4]; 121 u32 (*skip)[3][2]; 122 u32 (*intra_inter)[4][2]; 123 u32 (*tx32p)[2][4]; 124 u32 (*tx16p)[2][4]; 125 u32 (*tx8p)[2][2]; 126 u32 (*y_mode)[4][10]; 127 u32 (*uv_mode)[10][10]; 128 u32 (*comp)[5][2]; 129 u32 (*comp_ref)[5][2]; 130 u32 (*single_ref)[5][2][2]; 131 u32 (*mv_mode)[7][4]; 132 u32 (*filter)[4][3]; 133 u32 (*mv_joint)[4]; 134 u32 (*sign)[2][2]; 135 u32 (*classes)[2][11]; 136 u32 (*class0)[2][2]; 137 u32 (*bits)[2][10][2]; 138 u32 (*class0_fp)[2][2][4]; 139 u32 (*fp)[2][4]; 140 u32 (*class0_hp)[2][2]; 141 u32 (*hp)[2][2]; 142 u32 (*coeff[4][2][2][6][6])[3]; 143 u32 *eob[4][2][2][6][6][2]; 144}; 145 146extern const u8 v4l2_vp9_kf_y_mode_prob[10][10][9]; /* Section 10.4 of the spec */ 147extern const u8 v4l2_vp9_kf_partition_probs[16][3]; /* Section 10.4 of the spec */ 148extern const u8 v4l2_vp9_kf_uv_mode_prob[10][9]; /* Section 10.4 of the spec */ 149extern const struct v4l2_vp9_frame_context v4l2_vp9_default_probs; /* Section 10.5 of the spec */ 150 151/** 152 * v4l2_vp9_fw_update_probs() - Perform forward update of vp9 probabilities 153 * 154 * @probs: current probabilities values 155 * @deltas: delta values from compressed header 156 * @dec_params: vp9 frame decoding parameters 157 * 158 * This function performs forward updates of probabilities for the vp9 boolean decoder. 159 * The frame header can contain a directive to update the probabilities (deltas), if so, then 160 * the deltas are provided in the header, too. The userspace parses those and passes the said 161 * deltas struct to the kernel. 162 */ 163void v4l2_vp9_fw_update_probs(struct v4l2_vp9_frame_context *probs, 164 const struct v4l2_ctrl_vp9_compressed_hdr *deltas, 165 const struct v4l2_ctrl_vp9_frame *dec_params); 166 167/** 168 * v4l2_vp9_reset_frame_ctx() - Reset appropriate frame context 169 * 170 * @dec_params: vp9 frame decoding parameters 171 * @frame_context: array of the 4 frame contexts 172 * 173 * This function resets appropriate frame contexts, based on what's in dec_params. 174 * 175 * Returns the frame context index after the update, which might be reset to zero if 176 * mandated by the spec. 177 */ 178u8 v4l2_vp9_reset_frame_ctx(const struct v4l2_ctrl_vp9_frame *dec_params, 179 struct v4l2_vp9_frame_context *frame_context); 180 181/** 182 * v4l2_vp9_adapt_coef_probs() - Perform backward update of vp9 coefficients probabilities 183 * 184 * @probs: current probabilities values 185 * @counts: values of symbol counts after the current frame has been decoded 186 * @use_128: flag to request that 128 is used as update factor if true, otherwise 112 is used 187 * @frame_is_intra: flag indicating that FrameIsIntra is true 188 * 189 * This function performs backward updates of coefficients probabilities for the vp9 boolean 190 * decoder. After a frame has been decoded the counts of how many times a given symbol has 191 * occurred are known and are used to update the probability of each symbol. 192 */ 193void v4l2_vp9_adapt_coef_probs(struct v4l2_vp9_frame_context *probs, 194 struct v4l2_vp9_frame_symbol_counts *counts, 195 bool use_128, 196 bool frame_is_intra); 197 198/** 199 * v4l2_vp9_adapt_noncoef_probs() - Perform backward update of vp9 non-coefficients probabilities 200 * 201 * @probs: current probabilities values 202 * @counts: values of symbol counts after the current frame has been decoded 203 * @reference_mode: specifies the type of inter prediction to be used. See 204 * &v4l2_vp9_reference_mode for more details 205 * @interpolation_filter: specifies the filter selection used for performing inter prediction. 206 * See &v4l2_vp9_interpolation_filter for more details 207 * @tx_mode: specifies the TX mode. See &v4l2_vp9_tx_mode for more details 208 * @flags: combination of V4L2_VP9_FRAME_FLAG_* flags 209 * 210 * This function performs backward updates of non-coefficients probabilities for the vp9 boolean 211 * decoder. After a frame has been decoded the counts of how many times a given symbol has 212 * occurred are known and are used to update the probability of each symbol. 213 */ 214void v4l2_vp9_adapt_noncoef_probs(struct v4l2_vp9_frame_context *probs, 215 struct v4l2_vp9_frame_symbol_counts *counts, 216 u8 reference_mode, u8 interpolation_filter, u8 tx_mode, 217 u32 flags); 218 219/** 220 * v4l2_vp9_seg_feat_enabled() - Check if a segmentation feature is enabled 221 * 222 * @feature_enabled: array of 8-bit flags (for all segments) 223 * @feature: id of the feature to check 224 * @segid: id of the segment to look up 225 * 226 * This function returns true if a given feature is active in a given segment. 227 */ 228bool 229v4l2_vp9_seg_feat_enabled(const u8 *feature_enabled, 230 unsigned int feature, 231 unsigned int segid); 232 233#endif /* _MEDIA_V4L2_VP9_H */ 234