CoCalc -- radeon_vce

GitHub Repository: PojavLauncherTeam/mesa
Path: blob/21.2-virgl/src/gallium/drivers/radeon/radeon_vce_50.c
⁴⁵⁷⁰ views
1
/**************************************************************************
2
 *
3
 * Copyright 2013 Advanced Micro Devices, Inc.
4
 * All Rights Reserved.
5
 *
6
 * Permission is hereby granted, free of charge, to any person obtaining a
7
 * copy of this software and associated documentation files (the
8
 * "Software"), to deal in the Software without restriction, including
9
 * without limitation the rights to use, copy, modify, merge, publish,
10
 * distribute, sub license, and/or sell copies of the Software, and to
11
 * permit persons to whom the Software is furnished to do so, subject to
12
 * the following conditions:
13
 *
14
 * The above copyright notice and this permission notice (including the
15
 * next paragraph) shall be included in all copies or substantial portions
16
 * of the Software.
17
 *
18
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21
 * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
22
 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25
 *
26
 **************************************************************************/
27

28
#include "pipe/p_video_codec.h"
29
#include "radeon_vce.h"
30
#include "radeon_video.h"
31
#include "si_pipe.h"
32
#include "util/u_memory.h"
33
#include "util/u_video.h"
34
#include "vl/vl_video_buffer.h"
35

36
#include <stdio.h>
37

38
static void rate_control(struct rvce_encoder *enc)
39
{
40
   RVCE_BEGIN(0x04000005);                                 // rate control
41
   RVCE_CS(enc->pic.rate_ctrl.rate_ctrl_method);           // encRateControlMethod
42
   RVCE_CS(enc->pic.rate_ctrl.target_bitrate);             // encRateControlTargetBitRate
43
   RVCE_CS(enc->pic.rate_ctrl.peak_bitrate);               // encRateControlPeakBitRate
44
   RVCE_CS(enc->pic.rate_ctrl.frame_rate_num);             // encRateControlFrameRateNum
45
   RVCE_CS(0x00000000);                                    // encGOPSize
46
   RVCE_CS(enc->pic.quant_i_frames);                       // encQP_I
47
   RVCE_CS(enc->pic.quant_p_frames);                       // encQP_P
48
   RVCE_CS(enc->pic.quant_b_frames);                       // encQP_B
49
   RVCE_CS(enc->pic.rate_ctrl.vbv_buffer_size);            // encVBVBufferSize
50
   RVCE_CS(enc->pic.rate_ctrl.frame_rate_den);             // encRateControlFrameRateDen
51
   RVCE_CS(0x00000000);                                    // encVBVBufferLevel
52
   RVCE_CS(0x00000000);                                    // encMaxAUSize
53
   RVCE_CS(0x00000000);                                    // encQPInitialMode
54
   RVCE_CS(enc->pic.rate_ctrl.target_bits_picture);        // encTargetBitsPerPicture
55
   RVCE_CS(enc->pic.rate_ctrl.peak_bits_picture_integer);  // encPeakBitsPerPictureInteger
56
   RVCE_CS(enc->pic.rate_ctrl.peak_bits_picture_fraction); // encPeakBitsPerPictureFractional
57
   RVCE_CS(0x00000000);                                    // encMinQP
58
   RVCE_CS(0x00000033);                                    // encMaxQP
59
   RVCE_CS(0x00000000);                                    // encSkipFrameEnable
60
   RVCE_CS(0x00000000);                                    // encFillerDataEnable
61
   RVCE_CS(0x00000000);                                    // encEnforceHRD
62
   RVCE_CS(0x00000000);                                    // encBPicsDeltaQP
63
   RVCE_CS(0x00000000);                                    // encReferenceBPicsDeltaQP
64
   RVCE_CS(0x00000000);                                    // encRateControlReInitDisable
65
   RVCE_CS(0x00000000);                                    // encLCVBRInitQPFlag
66
   RVCE_CS(0x00000000); // encLCVBRSATDBasedNonlinearBitBudgetFlag
67
   RVCE_END();
68
}
69

70
static void encode(struct rvce_encoder *enc)
71
{
72
   signed luma_offset, chroma_offset, bs_offset;
73
   unsigned dep, bs_idx = enc->bs_idx++;
74
   int i;
75

76
   if (enc->dual_inst) {
77
      if (bs_idx == 0)
78
         dep = 1;
79
      else if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_IDR)
80
         dep = 0;
81
      else
82
         dep = 2;
83
   } else
84
      dep = 0;
85

86
   enc->task_info(enc, 0x00000003, dep, 0, bs_idx);
87

88
   RVCE_BEGIN(0x05000001);                                      // context buffer
89
   RVCE_READWRITE(enc->cpb.res->buf, enc->cpb.res->domains, 0); // encodeContextAddressHi/Lo
90
   RVCE_END();
91

92
   bs_offset = -(signed)(bs_idx * enc->bs_size);
93

94
   RVCE_BEGIN(0x05000004);                                   // video bitstream buffer
95
   RVCE_WRITE(enc->bs_handle, RADEON_DOMAIN_GTT, bs_offset); // videoBitstreamRingAddressHi/Lo
96
   RVCE_CS(enc->bs_size);                                    // videoBitstreamRingSize
97
   RVCE_END();
98

99
   if (enc->dual_pipe) {
100
      unsigned aux_offset =
101
         enc->cpb.res->buf->size - RVCE_MAX_AUX_BUFFER_NUM * RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE * 2;
102
      RVCE_BEGIN(0x05000002); // auxiliary buffer
103
      for (i = 0; i < 8; ++i) {
104
         RVCE_CS(aux_offset);
105
         aux_offset += RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE;
106
      }
107
      for (i = 0; i < 8; ++i)
108
         RVCE_CS(RVCE_MAX_BITSTREAM_OUTPUT_ROW_SIZE);
109
      RVCE_END();
110
   }
111

112
   RVCE_BEGIN(0x03000001);                   // encode
113
   RVCE_CS(enc->pic.frame_num ? 0x0 : 0x11); // insertHeaders
114
   RVCE_CS(0x00000000);                      // pictureStructure
115
   RVCE_CS(enc->bs_size);                    // allowedMaxBitstreamSize
116
   RVCE_CS(0x00000000);                      // forceRefreshMap
117
   RVCE_CS(0x00000000);                      // insertAUD
118
   RVCE_CS(0x00000000);                      // endOfSequence
119
   RVCE_CS(0x00000000);                      // endOfStream
120
   RVCE_READ(enc->handle, RADEON_DOMAIN_VRAM,
121
             (uint64_t)enc->luma->u.legacy.level[0].offset_256B * 256); // inputPictureLumaAddressHi/Lo
122
   RVCE_READ(enc->handle, RADEON_DOMAIN_VRAM,
123
             (uint64_t)enc->chroma->u.legacy.level[0].offset_256B * 256);              // inputPictureChromaAddressHi/Lo
124
   RVCE_CS(align(enc->luma->u.legacy.level[0].nblk_y, 16));       // encInputFrameYPitch
125
   RVCE_CS(enc->luma->u.legacy.level[0].nblk_x * enc->luma->bpe); // encInputPicLumaPitch
126
   RVCE_CS(enc->chroma->u.legacy.level[0].nblk_x * enc->chroma->bpe); // encInputPicChromaPitch
127
   if (enc->dual_pipe)
128
      RVCE_CS(0x00000000); // encInputPic(Addr|Array)Mode,encDisable(TwoPipeMode|MBOffloading)
129
   else
130
      RVCE_CS(0x00010000); // encInputPic(Addr|Array)Mode,encDisable(TwoPipeMode|MBOffloading)
131
   RVCE_CS(0x00000000);    // encInputPicTileConfig
132
   RVCE_CS(enc->pic.picture_type);                                   // encPicType
133
   RVCE_CS(enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_IDR);// encIdrFlag
134
   RVCE_CS(0x00000000);                                              // encIdrPicId
135
   RVCE_CS(0x00000000);                                              // encMGSKeyPic
136
   RVCE_CS(!enc->pic.not_referenced);                                // encReferenceFlag
137
   RVCE_CS(0x00000000);                                              // encTemporalLayerIndex
138
   RVCE_CS(0x00000000); // num_ref_idx_active_override_flag
139
   RVCE_CS(0x00000000); // num_ref_idx_l0_active_minus1
140
   RVCE_CS(0x00000000); // num_ref_idx_l1_active_minus1
141

142
   i = enc->pic.frame_num - enc->pic.ref_idx_l0;
143
   if (i > 1 && enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_P) {
144
      RVCE_CS(0x00000001); // encRefListModificationOp
145
      RVCE_CS(i - 1);      // encRefListModificationNum
146
   } else {
147
      RVCE_CS(0x00000000); // encRefListModificationOp
148
      RVCE_CS(0x00000000); // encRefListModificationNum
149
   }
150

151
   for (i = 0; i < 3; ++i) {
152
      RVCE_CS(0x00000000); // encRefListModificationOp
153
      RVCE_CS(0x00000000); // encRefListModificationNum
154
   }
155
   for (i = 0; i < 4; ++i) {
156
      RVCE_CS(0x00000000); // encDecodedPictureMarkingOp
157
      RVCE_CS(0x00000000); // encDecodedPictureMarkingNum
158
      RVCE_CS(0x00000000); // encDecodedPictureMarkingIdx
159
      RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingOp
160
      RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingNum
161
   }
162

163
   // encReferencePictureL0[0]
164
   RVCE_CS(0x00000000); // pictureStructure
165
   if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_P ||
166
       enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_B) {
167
      struct rvce_cpb_slot *l0 = si_l0_slot(enc);
168
      si_vce_frame_offset(enc, l0, &luma_offset, &chroma_offset);
169
      RVCE_CS(l0->picture_type);  // encPicType
170
      RVCE_CS(l0->frame_num);     // frameNumber
171
      RVCE_CS(l0->pic_order_cnt); // pictureOrderCount
172
      RVCE_CS(luma_offset);       // lumaOffset
173
      RVCE_CS(chroma_offset);     // chromaOffset
174
   } else {
175
      RVCE_CS(0x00000000); // encPicType
176
      RVCE_CS(0x00000000); // frameNumber
177
      RVCE_CS(0x00000000); // pictureOrderCount
178
      RVCE_CS(0xffffffff); // lumaOffset
179
      RVCE_CS(0xffffffff); // chromaOffset
180
   }
181

182
   // encReferencePictureL0[1]
183
   RVCE_CS(0x00000000); // pictureStructure
184
   RVCE_CS(0x00000000); // encPicType
185
   RVCE_CS(0x00000000); // frameNumber
186
   RVCE_CS(0x00000000); // pictureOrderCount
187
   RVCE_CS(0xffffffff); // lumaOffset
188
   RVCE_CS(0xffffffff); // chromaOffset
189

190
   // encReferencePictureL1[0]
191
   RVCE_CS(0x00000000); // pictureStructure
192
   if (enc->pic.picture_type == PIPE_H2645_ENC_PICTURE_TYPE_B) {
193
      struct rvce_cpb_slot *l1 = si_l1_slot(enc);
194
      si_vce_frame_offset(enc, l1, &luma_offset, &chroma_offset);
195
      RVCE_CS(l1->picture_type);  // encPicType
196
      RVCE_CS(l1->frame_num);     // frameNumber
197
      RVCE_CS(l1->pic_order_cnt); // pictureOrderCount
198
      RVCE_CS(luma_offset);       // lumaOffset
199
      RVCE_CS(chroma_offset);     // chromaOffset
200
   } else {
201
      RVCE_CS(0x00000000); // encPicType
202
      RVCE_CS(0x00000000); // frameNumber
203
      RVCE_CS(0x00000000); // pictureOrderCount
204
      RVCE_CS(0xffffffff); // lumaOffset
205
      RVCE_CS(0xffffffff); // chromaOffset
206
   }
207

208
   si_vce_frame_offset(enc, si_current_slot(enc), &luma_offset, &chroma_offset);
209
   RVCE_CS(luma_offset);            // encReconstructedLumaOffset
210
   RVCE_CS(chroma_offset);          // encReconstructedChromaOffset
211
   RVCE_CS(0x00000000);             // encColocBufferOffset
212
   RVCE_CS(0x00000000);             // encReconstructedRefBasePictureLumaOffset
213
   RVCE_CS(0x00000000);             // encReconstructedRefBasePictureChromaOffset
214
   RVCE_CS(0x00000000);             // encReferenceRefBasePictureLumaOffset
215
   RVCE_CS(0x00000000);             // encReferenceRefBasePictureChromaOffset
216
   RVCE_CS(0x00000000);             // pictureCount
217
   RVCE_CS(enc->pic.frame_num);     // frameNumber
218
   RVCE_CS(enc->pic.pic_order_cnt); // pictureOrderCount
219
   RVCE_CS(0x00000000);             // numIPicRemainInRCGOP
220
   RVCE_CS(0x00000000);             // numPPicRemainInRCGOP
221
   RVCE_CS(0x00000000);             // numBPicRemainInRCGOP
222
   RVCE_CS(0x00000000);             // numIRPicRemainInRCGOP
223
   RVCE_CS(0x00000000);             // enableIntraRefresh
224
   RVCE_END();
225
}
226

227
void si_vce_50_get_param(struct rvce_encoder *enc, struct pipe_h264_enc_picture_desc *pic)
228
{
229
}
230

231
void si_vce_50_init(struct rvce_encoder *enc)
232
{
233
   si_vce_40_2_2_init(enc);
234

235
   /* only the two below are different */
236
   enc->rate_control = rate_control;
237
   enc->encode = encode;
238
   enc->si_get_pic_param = si_vce_50_get_param;
239
}
240

241
Product

Resources

Company
