drivers/staging/media/hantro/hantro_vp9.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /*
   3  * Hantro VP9 codec driver
   4  *
   5  * Copyright (C) 2021 Collabora Ltd.
   6  */
   7
   8 struct hantro_g2_mv_probs {
   9         u8 joint[3];
  10         u8 sign[2];
  11         u8 class0_bit[2][1];
  12         u8 fr[2][3];
  13         u8 class0_hp[2];
  14         u8 hp[2];
  15         u8 classes[2][10];
  16         u8 class0_fr[2][2][3];
  17         u8 bits[2][10];
  18 };
  19
  20 struct hantro_g2_probs {
  21         u8 inter_mode[7][4];
  22         u8 is_inter[4];
  23         u8 uv_mode[10][8];
  24         u8 tx8[2][1];
  25         u8 tx16[2][2];
  26         u8 tx32[2][3];
  27         u8 y_mode_tail[4][1];
  28         u8 y_mode[4][8];
  29         u8 partition[2][16][4]; /* [keyframe][][], [inter][][] */
  30         u8 uv_mode_tail[10][1];
  31         u8 interp_filter[4][2];
  32         u8 comp_mode[5];
  33         u8 skip[3];
  34
  35         u8 pad1[1];
  36
  37         struct hantro_g2_mv_probs mv;
  38
  39         u8 single_ref[5][2];
  40         u8 comp_ref[5];
  41
  42         u8 pad2[17];
  43
  44         u8 coef[4][2][2][6][6][4];
  45 };
  46
  47 struct hantro_g2_all_probs {
  48         u8 kf_y_mode_prob[10][10][8];
  49
  50         u8 kf_y_mode_prob_tail[10][10][1];
  51         u8 ref_pred_probs[3];
  52         u8 mb_segment_tree_probs[7];
  53         u8 segment_pred_probs[3];
  54         u8 ref_scores[4];
  55         u8 prob_comppred[2];
  56
  57         u8 pad1[9];
  58
  59         u8 kf_uv_mode_prob[10][8];
  60         u8 kf_uv_mode_prob_tail[10][1];
  61
  62         u8 pad2[6];
  63
  64         struct hantro_g2_probs probs;
  65 };
  66
  67 struct mv_counts {
  68         u32 joints[4];
  69         u32 sign[2][2];
  70         u32 classes[2][11];
  71         u32 class0[2][2];
  72         u32 bits[2][10][2];
  73         u32 class0_fp[2][2][4];
  74         u32 fp[2][4];
  75         u32 class0_hp[2][2];
  76         u32 hp[2][2];
  77 };
  78
  79 struct symbol_counts {
  80         u32 inter_mode_counts[7][3][2];
  81         u32 sb_ymode_counts[4][10];
  82         u32 uv_mode_counts[10][10];
  83         u32 partition_counts[16][4];
  84         u32 switchable_interp_counts[4][3];
  85         u32 intra_inter_count[4][2];
  86         u32 comp_inter_count[5][2];
  87         u32 single_ref_count[5][2][2];
  88         u32 comp_ref_count[5][2];
  89         u32 tx32x32_count[2][4];
  90         u32 tx16x16_count[2][3];
  91         u32 tx8x8_count[2][2];
  92         u32 mbskip_count[3][2];
  93
  94         struct mv_counts mv_counts;
  95
  96         u32 count_coeffs[2][2][6][6][4];
  97         u32 count_coeffs8x8[2][2][6][6][4];
  98         u32 count_coeffs16x16[2][2][6][6][4];
  99         u32 count_coeffs32x32[2][2][6][6][4];
 100
 101         u32 count_eobs[4][2][2][6][6];
 102 };