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(dev: vpu->dev, |
45 | size: ctx->mpeg2_dec.qtable.size, |
46 | dma_handle: &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(dev: vpu->dev, |
58 | size: ctx->mpeg2_dec.qtable.size, |
59 | cpu_addr: ctx->mpeg2_dec.qtable.cpu, |
60 | dma_handle: ctx->mpeg2_dec.qtable.dma); |
61 | } |
62 | |