drivers/media/platform/verisilicon/hantro_mpeg2.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Hantro VPU codec driver
   4  *
   5  * Copyright (C) 2018 Rockchip Electronics Co., Ltd.
   6  */
   7
   8 #include "hantro.h"
   9
  10 static const u8 zigzag[64] = {
  11         0,   1,  8, 16,  9,  2,  3, 10,
  12         17, 24, 32, 25, 18, 11,  4,  5,
  13         12, 19, 26, 33, 40, 48, 41, 34,
  14         27, 20, 13,  6,  7, 14, 21, 28,
  15         35, 42, 49, 56, 57, 50, 43, 36,
  16         29, 22, 15, 23, 30, 37, 44, 51,
  17         58, 59, 52, 45, 38, 31, 39, 46,
  18         53, 60, 61, 54, 47, 55, 62, 63
  19 };
  20
  21 void hantro_mpeg2_dec_copy_qtable(u8 *qtable,
  22                                   const struct v4l2_ctrl_mpeg2_quantisation *ctrl)
  23 {
  24         int i, n;
  25
  26         if (!qtable || !ctrl)
  27                 return;
  28
  29         for (i = 0; i < ARRAY_SIZE(zigzag); i++) {
  30                 n = zigzag[i];
  31                 qtable[n + 0] = ctrl->intra_quantiser_matrix[i];
  32                 qtable[n + 64] = ctrl->non_intra_quantiser_matrix[i];
  33                 qtable[n + 128] = ctrl->chroma_intra_quantiser_matrix[i];
  34                 qtable[n + 192] = ctrl->chroma_non_intra_quantiser_matrix[i];
  35         }
  36 }
  37
  38 int hantro_mpeg2_dec_init(struct hantro_ctx *ctx)
  39 {
  40         struct hantro_dev *vpu = ctx->dev;
  41
  42         ctx->mpeg2_dec.qtable.size = ARRAY_SIZE(zigzag) * 4;
  43         ctx->mpeg2_dec.qtable.cpu =
  44                 dma_alloc_coherent(vpu->dev,
  45                                    ctx->mpeg2_dec.qtable.size,
  46                                    &ctx->mpeg2_dec.qtable.dma,
  47                                    GFP_KERNEL);
  48         if (!ctx->mpeg2_dec.qtable.cpu)
  49                 return -ENOMEM;
  50         return 0;
  51 }
  52
  53 void hantro_mpeg2_dec_exit(struct hantro_ctx *ctx)
  54 {
  55         struct hantro_dev *vpu = ctx->dev;
  56
  57         dma_free_coherent(vpu->dev,
  58                           ctx->mpeg2_dec.qtable.size,
  59                           ctx->mpeg2_dec.qtable.cpu,
  60                           ctx->mpeg2_dec.qtable.dma);
  61 }