1 /**************************************************************************
2 *
3 * Copyright 2013 Advanced Micro Devices, Inc.
4 *
5 * SPDX-License-Identifier: MIT
6 *
7 **************************************************************************/
8
9 #include "pipe/p_video_codec.h"
10 #include "radeon_vce.h"
11 #include "radeon_video.h"
12 #include "si_pipe.h"
13 #include "util/u_memory.h"
14 #include "util/u_video.h"
15 #include "vl/vl_video_buffer.h"
16
17 #include <stdio.h>
18
rate_control(struct rvce_encoder * enc)19 static void rate_control(struct rvce_encoder *enc)
20 {
21 RVCE_BEGIN(0x04000005); // rate control
22 RVCE_CS(enc->pic.rate_ctrl[0].rate_ctrl_method); // encRateControlMethod
23 RVCE_CS(enc->pic.rate_ctrl[0].target_bitrate); // encRateControlTargetBitRate
24 RVCE_CS(enc->pic.rate_ctrl[0].peak_bitrate); // encRateControlPeakBitRate
25 RVCE_CS(enc->pic.rate_ctrl[0].frame_rate_num); // encRateControlFrameRateNum
26 RVCE_CS(0x00000000); // encGOPSize
27 RVCE_CS(enc->pic.quant_i_frames); // encQP_I
28 RVCE_CS(enc->pic.quant_p_frames); // encQP_P
29 RVCE_CS(enc->pic.quant_b_frames); // encQP_B
30 RVCE_CS(enc->pic.rate_ctrl[0].vbv_buffer_size); // encVBVBufferSize
31 RVCE_CS(enc->pic.rate_ctrl[0].frame_rate_den); // encRateControlFrameRateDen
32 RVCE_CS(0x00000000); // encVBVBufferLevel
33 RVCE_CS(0x00000000); // encMaxAUSize
34 RVCE_CS(0x00000000); // encQPInitialMode
35 RVCE_CS(enc->pic.rate_ctrl[0].target_bits_picture); // encTargetBitsPerPicture
36 RVCE_CS(enc->pic.rate_ctrl[0].peak_bits_picture_integer); // encPeakBitsPerPictureInteger
37 RVCE_CS(enc->pic.rate_ctrl[0].peak_bits_picture_fraction); // encPeakBitsPerPictureFractional
38 RVCE_CS(0x00000000); // encMinQP
39 RVCE_CS(0x00000033); // encMaxQP
40 RVCE_CS(0x00000000); // encSkipFrameEnable
41 RVCE_CS(0x00000000); // encFillerDataEnable
42 RVCE_CS(0x00000000); // encEnforceHRD
43 RVCE_CS(0x00000000); // encBPicsDeltaQP
44 RVCE_CS(0x00000000); // encReferenceBPicsDeltaQP
45 RVCE_CS(0x00000000); // encRateControlReInitDisable
46 RVCE_CS(0x00000000); // encLCVBRInitQPFlag
47 RVCE_CS(0x00000000); // encLCVBRSATDBasedNonlinearBitBudgetFlag
48 RVCE_END();
49 }
50
encode(struct rvce_encoder * enc)51 static void encode(struct rvce_encoder *enc)
52 {
53 signed luma_offset, chroma_offset, bs_offset;
54 unsigned dep, bs_idx = enc->bs_idx++;
55 int i;
56
57 if (enc->dual_inst) {
58 if (bs_idx == 0)
59 dep = 1;
60 else if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_IDR)
61 dep = 0;
62 else
63 dep = 2;
64 } else
65 dep = 0;
66
67 enc->task_info(enc, 0x00000003, dep, 0, bs_idx);
68
69 RVCE_BEGIN(0x05000001); // context buffer
70 RVCE_READWRITE(enc->cpb.res->buf, enc->cpb.res->domains, 0); // encodeContextAddressHi/Lo
71 RVCE_END();
72
73 bs_offset = -(signed)(bs_idx * enc->bs_size);
74
75 RVCE_BEGIN(0x05000004); // video bitstream buffer
76 RVCE_WRITE(enc->bs_handle, RADEON_DOMAIN_GTT, bs_offset); // videoBitstreamRingAddressHi/Lo
77 RVCE_CS(enc->bs_size); // videoBitstreamRingSize
78 RVCE_END();
79
80 if (enc->dual_pipe) {
81 unsigned aux_offset =
82 enc->cpb.res->buf->size - RVCE_MAX_AUX_BUFFER_NUM * RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE * 2;
83 RVCE_BEGIN(0x05000002); // auxiliary buffer
84 for (i = 0; i < 8; ++i) {
85 RVCE_CS(aux_offset);
86 aux_offset += RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE;
87 }
88 for (i = 0; i < 8; ++i)
89 RVCE_CS(RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE);
90 RVCE_END();
91 }
92
93 RVCE_BEGIN(0x03000001); // encode
94 RVCE_CS(enc->pic.frame_num ? 0x0 : 0x11); // insertHeaders
95 RVCE_CS(0x00000000); // pictureStructure
96 RVCE_CS(enc->bs_size); // allowedMaxBitstreamSize
97 RVCE_CS(0x00000000); // forceRefreshMap
98 RVCE_CS(0x00000000); // insertAUD
99 RVCE_CS(0x00000000); // endOfSequence
100 RVCE_CS(0x00000000); // endOfStream
101 RVCE_READ(enc->handle, RADEON_DOMAIN_VRAM,
102 (uint64_t)enc->luma->u.legacy.level[0].offset_256B * 256); // inputPictureLumaAddressHi/Lo
103 RVCE_READ(enc->handle, RADEON_DOMAIN_VRAM,
104 (uint64_t)enc->chroma->u.legacy.level[0].offset_256B * 256); // inputPictureChromaAddressHi/Lo
105 RVCE_CS(align(enc->luma->u.legacy.level[0].nblk_y, 16)); // encInputFrameYPitch
106 RVCE_CS(enc->luma->u.legacy.level[0].nblk_x * enc->luma->bpe); // encInputPicLumaPitch
107 RVCE_CS(enc->chroma->u.legacy.level[0].nblk_x * enc->chroma->bpe); // encInputPicChromaPitch
108 if (enc->dual_pipe)
109 RVCE_CS(0x00000000); // encInputPic(Addr|Array)Mode,encDisable(TwoPipeMode|MBOffloading)
110 else
111 RVCE_CS(0x00010000); // encInputPic(Addr|Array)Mode,encDisable(TwoPipeMode|MBOffloading)
112 RVCE_CS(0x00000000); // encInputPicTileConfig
113 RVCE_CS(enc->pic.picture_type); // encPicType
114 RVCE_CS(enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_IDR);// encIdrFlag
115 RVCE_CS(0x00000000); // encIdrPicId
116 RVCE_CS(0x00000000); // encMGSKeyPic
117 RVCE_CS(!enc->pic.not_referenced); // encReferenceFlag
118 RVCE_CS(0x00000000); // encTemporalLayerIndex
119 RVCE_CS(0x00000000); // num_ref_idx_active_override_flag
120 RVCE_CS(0x00000000); // num_ref_idx_l0_active_minus1
121 RVCE_CS(0x00000000); // num_ref_idx_l1_active_minus1
122
123 i = enc->pic.frame_num - enc->pic.ref_idx_l0_list[0];
124 if (i > 1 && enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_P) {
125 RVCE_CS(0x00000001); // encRefListModificationOp
126 RVCE_CS(i - 1); // encRefListModificationNum
127 } else {
128 RVCE_CS(0x00000000); // encRefListModificationOp
129 RVCE_CS(0x00000000); // encRefListModificationNum
130 }
131
132 for (i = 0; i < 3; ++i) {
133 RVCE_CS(0x00000000); // encRefListModificationOp
134 RVCE_CS(0x00000000); // encRefListModificationNum
135 }
136 for (i = 0; i < 4; ++i) {
137 RVCE_CS(0x00000000); // encDecodedPictureMarkingOp
138 RVCE_CS(0x00000000); // encDecodedPictureMarkingNum
139 RVCE_CS(0x00000000); // encDecodedPictureMarkingIdx
140 RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingOp
141 RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingNum
142 }
143
144 // encReferencePictureL0[0]
145 RVCE_CS(0x00000000); // pictureStructure
146 if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_P ||
147 enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_B) {
148 struct rvce_cpb_slot *l0 = si_l0_slot(enc);
149 si_vce_frame_offset(enc, l0, &luma_offset, &chroma_offset);
150 RVCE_CS(l0->picture_type); // encPicType
151 RVCE_CS(l0->frame_num); // frameNumber
152 RVCE_CS(l0->pic_order_cnt); // pictureOrderCount
153 RVCE_CS(luma_offset); // lumaOffset
154 RVCE_CS(chroma_offset); // chromaOffset
155 } else {
156 RVCE_CS(0x00000000); // encPicType
157 RVCE_CS(0x00000000); // frameNumber
158 RVCE_CS(0x00000000); // pictureOrderCount
159 RVCE_CS(0xffffffff); // lumaOffset
160 RVCE_CS(0xffffffff); // chromaOffset
161 }
162
163 // encReferencePictureL0[1]
164 RVCE_CS(0x00000000); // pictureStructure
165 RVCE_CS(0x00000000); // encPicType
166 RVCE_CS(0x00000000); // frameNumber
167 RVCE_CS(0x00000000); // pictureOrderCount
168 RVCE_CS(0xffffffff); // lumaOffset
169 RVCE_CS(0xffffffff); // chromaOffset
170
171 // encReferencePictureL1[0]
172 RVCE_CS(0x00000000); // pictureStructure
173 if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_B) {
174 struct rvce_cpb_slot *l1 = si_l1_slot(enc);
175 si_vce_frame_offset(enc, l1, &luma_offset, &chroma_offset);
176 RVCE_CS(l1->picture_type); // encPicType
177 RVCE_CS(l1->frame_num); // frameNumber
178 RVCE_CS(l1->pic_order_cnt); // pictureOrderCount
179 RVCE_CS(luma_offset); // lumaOffset
180 RVCE_CS(chroma_offset); // chromaOffset
181 } else {
182 RVCE_CS(0x00000000); // encPicType
183 RVCE_CS(0x00000000); // frameNumber
184 RVCE_CS(0x00000000); // pictureOrderCount
185 RVCE_CS(0xffffffff); // lumaOffset
186 RVCE_CS(0xffffffff); // chromaOffset
187 }
188
189 si_vce_frame_offset(enc, si_current_slot(enc), &luma_offset, &chroma_offset);
190 RVCE_CS(luma_offset); // encReconstructedLumaOffset
191 RVCE_CS(chroma_offset); // encReconstructedChromaOffset
192 RVCE_CS(0x00000000); // encColocBufferOffset
193 RVCE_CS(0x00000000); // encReconstructedRefBasePictureLumaOffset
194 RVCE_CS(0x00000000); // encReconstructedRefBasePictureChromaOffset
195 RVCE_CS(0x00000000); // encReferenceRefBasePictureLumaOffset
196 RVCE_CS(0x00000000); // encReferenceRefBasePictureChromaOffset
197 RVCE_CS(0x00000000); // pictureCount
198 RVCE_CS(enc->pic.frame_num); // frameNumber
199 RVCE_CS(enc->pic.pic_order_cnt); // pictureOrderCount
200 RVCE_CS(0x00000000); // numIPicRemainInRCGOP
201 RVCE_CS(0x00000000); // numPPicRemainInRCGOP
202 RVCE_CS(0x00000000); // numBPicRemainInRCGOP
203 RVCE_CS(0x00000000); // numIRPicRemainInRCGOP
204 RVCE_CS(0x00000000); // enableIntraRefresh
205 RVCE_END();
206 }
207
si_vce_50_get_param(struct rvce_encoder * enc,struct pipe_h264_enc_picture_desc * pic)208 void si_vce_50_get_param(struct rvce_encoder *enc, struct pipe_h264_enc_picture_desc *pic)
209 {
210 }
211
si_vce_50_init(struct rvce_encoder * enc)212 void si_vce_50_init(struct rvce_encoder *enc)
213 {
214 si_vce_40_2_2_init(enc);
215
216 /* only the two below are different */
217 enc->rate_control = rate_control;
218 enc->encode = encode;
219 enc->si_get_pic_param = si_vce_50_get_param;
220 }
221