2 * Copyright © 2010-2011 Intel Corporation
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the
6 * "Software"), to deal in the Software without restriction, including
7 * without limitation the rights to use, copy, modify, merge, publish,
8 * distribute, sub license, and/or sell copies of the Software, and to
9 * permit persons to whom the Software is furnished to do so, subject to
10 * the following conditions:
12 * The above copyright notice and this permission notice (including the
13 * next paragraph) shall be included in all copies or substantial portions
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
19 * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
20 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
21 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 * Zhou Chang <chang.zhou@intel.com>
34 #include "intel_batchbuffer.h"
35 #include "intel_driver.h"
37 #include "i965_defines.h"
38 #include "i965_drv_video.h"
39 #include "i965_encoder.h"
43 #define SURFACE_STATE_PADDED_SIZE_0_GEN7 ALIGN(sizeof(struct gen7_surface_state), 32)
44 #define SURFACE_STATE_PADDED_SIZE_1_GEN7 ALIGN(sizeof(struct gen7_surface_state2), 32)
45 #define SURFACE_STATE_PADDED_SIZE_GEN7 MAX(SURFACE_STATE_PADDED_SIZE_0_GEN7, SURFACE_STATE_PADDED_SIZE_1_GEN7)
47 #define SURFACE_STATE_PADDED_SIZE_0_GEN6 ALIGN(sizeof(struct i965_surface_state), 32)
48 #define SURFACE_STATE_PADDED_SIZE_1_GEN6 ALIGN(sizeof(struct i965_surface_state2), 32)
49 #define SURFACE_STATE_PADDED_SIZE_GEN6 MAX(SURFACE_STATE_PADDED_SIZE_0_GEN6, SURFACE_STATE_PADDED_SIZE_1_GEN6)
51 #define SURFACE_STATE_PADDED_SIZE MAX(SURFACE_STATE_PADDED_SIZE_GEN6, SURFACE_STATE_PADDED_SIZE_GEN7)
52 #define SURFACE_STATE_OFFSET(index) (SURFACE_STATE_PADDED_SIZE * index)
53 #define BINDING_TABLE_OFFSET(index) (SURFACE_STATE_OFFSET(MAX_MEDIA_SURFACES_GEN6) + sizeof(unsigned int) * index)
55 #define CURBE_ALLOCATION_SIZE 37 /* in 256-bit */
56 #define CURBE_TOTAL_DATA_LENGTH (4 * 32) /* in byte, it should be less than or equal to CURBE_ALLOCATION_SIZE * 32 */
57 #define CURBE_URB_ENTRY_LENGTH 4 /* in 256-bit, it should be less than or equal to CURBE_TOTAL_DATA_LENGTH / 32 */
59 enum VIDEO_CODING_TYPE{
65 enum AVC_VME_KERNEL_TYPE{
66 AVC_VME_INTRA_SHADER = 0,
72 enum MPEG2_VME_KERNEL_TYPE{
73 MPEG2_VME_INTER_SHADER = 0,
74 MPEG2_VME_BATCHBUFFER,
78 static const uint32_t gen6_vme_intra_frame[][4] = {
79 #include "shaders/vme/intra_frame.g6b"
82 static const uint32_t gen6_vme_inter_frame[][4] = {
83 #include "shaders/vme/inter_frame.g6b"
86 static const uint32_t gen6_vme_batchbuffer[][4] = {
87 #include "shaders/vme/batchbuffer.g6b"
90 static struct i965_kernel gen6_vme_kernels[] = {
92 "AVC VME Intra Frame",
93 AVC_VME_INTRA_SHADER, /*index*/
95 sizeof(gen6_vme_intra_frame),
99 "AVC VME inter Frame",
100 AVC_VME_INTER_SHADER,
101 gen6_vme_inter_frame,
102 sizeof(gen6_vme_inter_frame),
106 "AVC VME BATCHBUFFER",
108 gen6_vme_batchbuffer,
109 sizeof(gen6_vme_batchbuffer),
114 static const uint32_t gen7_vme_intra_frame[][4] = {
115 #include "shaders/vme/intra_frame.g7b"
118 static const uint32_t gen7_vme_inter_frame[][4] = {
119 #include "shaders/vme/inter_frame.g7b"
122 static const uint32_t gen7_vme_batchbuffer[][4] = {
123 #include "shaders/vme/batchbuffer.g7b"
126 static struct i965_kernel gen7_vme_kernels[] = {
128 "AVC VME Intra Frame",
129 AVC_VME_INTRA_SHADER, /*index*/
130 gen7_vme_intra_frame,
131 sizeof(gen7_vme_intra_frame),
135 "AVC VME inter Frame",
136 AVC_VME_INTER_SHADER,
137 gen7_vme_inter_frame,
138 sizeof(gen7_vme_inter_frame),
142 "AVC VME BATCHBUFFER",
144 gen7_vme_batchbuffer,
145 sizeof(gen7_vme_batchbuffer),
150 static const uint32_t gen7_vme_mpeg2_inter_frame[][4] = {
151 #include "shaders/vme/mpeg2_inter_frame.g7b"
154 static const uint32_t gen7_vme_mpeg2_batchbuffer[][4] = {
155 #include "shaders/vme/batchbuffer.g7b"
158 static struct i965_kernel gen7_vme_mpeg2_kernels[] = {
160 "MPEG2 VME inter Frame",
161 MPEG2_VME_INTER_SHADER,
162 gen7_vme_mpeg2_inter_frame,
163 sizeof(gen7_vme_mpeg2_inter_frame),
167 "MPEG2 VME BATCHBUFFER",
168 MPEG2_VME_BATCHBUFFER,
169 gen7_vme_mpeg2_batchbuffer,
170 sizeof(gen7_vme_mpeg2_batchbuffer),
175 /* only used for VME source surface state */
177 gen6_vme_source_surface_state(VADriverContextP ctx,
179 struct object_surface *obj_surface,
180 struct intel_encoder_context *encoder_context)
182 struct gen6_vme_context *vme_context = encoder_context->vme_context;
184 vme_context->vme_surface2_setup(ctx,
185 &vme_context->gpe_context,
187 BINDING_TABLE_OFFSET(index),
188 SURFACE_STATE_OFFSET(index));
192 gen6_vme_media_source_surface_state(VADriverContextP ctx,
194 struct object_surface *obj_surface,
195 struct intel_encoder_context *encoder_context)
197 struct gen6_vme_context *vme_context = encoder_context->vme_context;
199 vme_context->vme_media_rw_surface_setup(ctx,
200 &vme_context->gpe_context,
202 BINDING_TABLE_OFFSET(index),
203 SURFACE_STATE_OFFSET(index));
207 gen6_vme_output_buffer_setup(VADriverContextP ctx,
208 struct encode_state *encode_state,
210 struct intel_encoder_context *encoder_context)
213 struct i965_driver_data *i965 = i965_driver_data(ctx);
214 struct gen6_vme_context *vme_context = encoder_context->vme_context;
215 VAEncSequenceParameterBufferH264 *pSequenceParameter = (VAEncSequenceParameterBufferH264 *)encode_state->seq_param_ext->buffer;
216 VAEncSliceParameterBufferH264 *pSliceParameter = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[0]->buffer;
217 int is_intra = pSliceParameter->slice_type == SLICE_TYPE_I;
218 int width_in_mbs = pSequenceParameter->picture_width_in_mbs;
219 int height_in_mbs = pSequenceParameter->picture_height_in_mbs;
221 vme_context->vme_output.num_blocks = width_in_mbs * height_in_mbs;
222 vme_context->vme_output.pitch = 16; /* in bytes, always 16 */
225 vme_context->vme_output.size_block = INTRA_VME_OUTPUT_IN_BYTES;
227 vme_context->vme_output.size_block = INTER_VME_OUTPUT_IN_BYTES;
229 vme_context->vme_output.bo = dri_bo_alloc(i965->intel.bufmgr,
231 vme_context->vme_output.num_blocks * vme_context->vme_output.size_block,
233 assert(vme_context->vme_output.bo);
234 vme_context->vme_buffer_suface_setup(ctx,
235 &vme_context->gpe_context,
236 &vme_context->vme_output,
237 BINDING_TABLE_OFFSET(index),
238 SURFACE_STATE_OFFSET(index));
242 gen6_vme_output_vme_batchbuffer_setup(VADriverContextP ctx,
243 struct encode_state *encode_state,
245 struct intel_encoder_context *encoder_context)
248 struct i965_driver_data *i965 = i965_driver_data(ctx);
249 struct gen6_vme_context *vme_context = encoder_context->vme_context;
250 VAEncSequenceParameterBufferH264 *pSequenceParameter = (VAEncSequenceParameterBufferH264 *)encode_state->seq_param_ext->buffer;
251 int width_in_mbs = pSequenceParameter->picture_width_in_mbs;
252 int height_in_mbs = pSequenceParameter->picture_height_in_mbs;
254 vme_context->vme_batchbuffer.num_blocks = width_in_mbs * height_in_mbs + 1;
255 vme_context->vme_batchbuffer.size_block = 32; /* 2 OWORDs */
256 vme_context->vme_batchbuffer.pitch = 16;
257 vme_context->vme_batchbuffer.bo = dri_bo_alloc(i965->intel.bufmgr,
259 vme_context->vme_batchbuffer.num_blocks * vme_context->vme_batchbuffer.size_block,
261 vme_context->vme_buffer_suface_setup(ctx,
262 &vme_context->gpe_context,
263 &vme_context->vme_batchbuffer,
264 BINDING_TABLE_OFFSET(index),
265 SURFACE_STATE_OFFSET(index));
269 gen6_vme_surface_setup(VADriverContextP ctx,
270 struct encode_state *encode_state,
272 struct intel_encoder_context *encoder_context)
274 struct i965_driver_data *i965 = i965_driver_data(ctx);
275 struct object_surface *obj_surface;
276 VAEncPictureParameterBufferH264 *pPicParameter = (VAEncPictureParameterBufferH264 *)encode_state->pic_param_ext->buffer;
278 /*Setup surfaces state*/
279 /* current picture for encoding */
280 obj_surface = encode_state->input_yuv_object;
281 gen6_vme_source_surface_state(ctx, 0, obj_surface, encoder_context);
282 gen6_vme_media_source_surface_state(ctx, 4, obj_surface, encoder_context);
286 obj_surface = SURFACE(pPicParameter->ReferenceFrames[0].picture_id);
288 if ( obj_surface->bo != NULL)
289 gen6_vme_source_surface_state(ctx, 1, obj_surface, encoder_context);
292 obj_surface = SURFACE(pPicParameter->ReferenceFrames[1].picture_id);
294 if ( obj_surface->bo != NULL )
295 gen6_vme_source_surface_state(ctx, 2, obj_surface, encoder_context);
299 gen6_vme_output_buffer_setup(ctx, encode_state, 3, encoder_context);
300 gen6_vme_output_vme_batchbuffer_setup(ctx, encode_state, 5, encoder_context);
302 return VA_STATUS_SUCCESS;
305 static VAStatus gen6_vme_interface_setup(VADriverContextP ctx,
306 struct encode_state *encode_state,
307 struct intel_encoder_context *encoder_context)
309 struct gen6_vme_context *vme_context = encoder_context->vme_context;
310 struct gen6_interface_descriptor_data *desc;
314 bo = vme_context->gpe_context.idrt.bo;
319 for (i = 0; i < vme_context->vme_kernel_sum; i++) {
320 struct i965_kernel *kernel;
321 kernel = &vme_context->gpe_context.kernels[i];
322 assert(sizeof(*desc) == 32);
323 /*Setup the descritor table*/
324 memset(desc, 0, sizeof(*desc));
325 desc->desc0.kernel_start_pointer = (kernel->bo->offset >> 6);
326 desc->desc2.sampler_count = 1; /* FIXME: */
327 desc->desc2.sampler_state_pointer = (vme_context->vme_state.bo->offset >> 5);
328 desc->desc3.binding_table_entry_count = 1; /* FIXME: */
329 desc->desc3.binding_table_pointer = (BINDING_TABLE_OFFSET(0) >> 5);
330 desc->desc4.constant_urb_entry_read_offset = 0;
331 desc->desc4.constant_urb_entry_read_length = CURBE_URB_ENTRY_LENGTH;
334 dri_bo_emit_reloc(bo,
335 I915_GEM_DOMAIN_INSTRUCTION, 0,
337 i * sizeof(*desc) + offsetof(struct gen6_interface_descriptor_data, desc0),
339 /*Sampler State(VME state pointer)*/
340 dri_bo_emit_reloc(bo,
341 I915_GEM_DOMAIN_INSTRUCTION, 0,
343 i * sizeof(*desc) + offsetof(struct gen6_interface_descriptor_data, desc2),
344 vme_context->vme_state.bo);
349 return VA_STATUS_SUCCESS;
352 static VAStatus gen6_vme_constant_setup(VADriverContextP ctx,
353 struct encode_state *encode_state,
354 struct intel_encoder_context *encoder_context)
356 struct gen6_vme_context *vme_context = encoder_context->vme_context;
357 // unsigned char *constant_buffer;
358 unsigned int *vme_state_message;
360 if (vme_context->h264_level >= 30) {
362 if (vme_context->h264_level >= 31)
366 dri_bo_map(vme_context->gpe_context.curbe.bo, 1);
367 assert(vme_context->gpe_context.curbe.bo->virtual);
368 // constant_buffer = vme_context->curbe.bo->virtual;
369 vme_state_message = (unsigned int *)vme_context->gpe_context.curbe.bo->virtual;
370 vme_state_message[31] = mv_num;
372 /*TODO copy buffer into CURB*/
374 dri_bo_unmap( vme_context->gpe_context.curbe.bo);
376 return VA_STATUS_SUCCESS;
379 static const unsigned int intra_mb_mode_cost_table[] = {
380 0x31110001, // for qp0
381 0x09110001, // for qp1
382 0x15030001, // for qp2
383 0x0b030001, // for qp3
384 0x0d030011, // for qp4
385 0x17210011, // for qp5
386 0x41210011, // for qp6
387 0x19210011, // for qp7
388 0x25050003, // for qp8
389 0x1b130003, // for qp9
390 0x1d130003, // for qp10
391 0x27070021, // for qp11
392 0x51310021, // for qp12
393 0x29090021, // for qp13
394 0x35150005, // for qp14
395 0x2b0b0013, // for qp15
396 0x2d0d0013, // for qp16
397 0x37170007, // for qp17
398 0x61410031, // for qp18
399 0x39190009, // for qp19
400 0x45250015, // for qp20
401 0x3b1b000b, // for qp21
402 0x3d1d000d, // for qp22
403 0x47270017, // for qp23
404 0x71510041, // for qp24 ! center for qp=0..30
405 0x49290019, // for qp25
406 0x55350025, // for qp26
407 0x4b2b001b, // for qp27
408 0x4d2d001d, // for qp28
409 0x57370027, // for qp29
410 0x81610051, // for qp30
411 0x57270017, // for qp31
412 0x81510041, // for qp32 ! center for qp=31..51
413 0x59290019, // for qp33
414 0x65350025, // for qp34
415 0x5b2b001b, // for qp35
416 0x5d2d001d, // for qp36
417 0x67370027, // for qp37
418 0x91610051, // for qp38
419 0x69390029, // for qp39
420 0x75450035, // for qp40
421 0x6b3b002b, // for qp41
422 0x6d3d002d, // for qp42
423 0x77470037, // for qp43
424 0xa1710061, // for qp44
425 0x79490039, // for qp45
426 0x85550045, // for qp46
427 0x7b4b003b, // for qp47
428 0x7d4d003d, // for qp48
429 0x87570047, // for qp49
430 0xb1810071, // for qp50
431 0x89590049 // for qp51
434 static void gen6_vme_state_setup_fixup(VADriverContextP ctx,
435 struct encode_state *encode_state,
436 struct intel_encoder_context *encoder_context,
437 unsigned int *vme_state_message)
439 struct gen6_mfc_context *mfc_context = encoder_context->mfc_context;
440 VAEncPictureParameterBufferH264 *pic_param = (VAEncPictureParameterBufferH264 *)encode_state->pic_param_ext->buffer;
441 VAEncSliceParameterBufferH264 *slice_param = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[0]->buffer;
443 if (slice_param->slice_type != SLICE_TYPE_I &&
444 slice_param->slice_type != SLICE_TYPE_SI)
447 if (encoder_context->rate_control_mode == VA_RC_CQP)
448 vme_state_message[16] = intra_mb_mode_cost_table[pic_param->pic_init_qp + slice_param->slice_qp_delta];
450 vme_state_message[16] = intra_mb_mode_cost_table[mfc_context->bit_rate_control_context[SLICE_TYPE_I].QpPrimeY];
453 static VAStatus gen6_vme_vme_state_setup(VADriverContextP ctx,
454 struct encode_state *encode_state,
456 struct intel_encoder_context *encoder_context)
458 struct gen6_vme_context *vme_context = encoder_context->vme_context;
459 unsigned int *vme_state_message;
462 //building VME state message
463 dri_bo_map(vme_context->vme_state.bo, 1);
464 assert(vme_context->vme_state.bo->virtual);
465 vme_state_message = (unsigned int *)vme_context->vme_state.bo->virtual;
467 vme_state_message[0] = 0x01010101;
468 vme_state_message[1] = 0x10010101;
469 vme_state_message[2] = 0x0F0F0F0F;
470 vme_state_message[3] = 0x100F0F0F;
471 vme_state_message[4] = 0x01010101;
472 vme_state_message[5] = 0x10010101;
473 vme_state_message[6] = 0x0F0F0F0F;
474 vme_state_message[7] = 0x100F0F0F;
475 vme_state_message[8] = 0x01010101;
476 vme_state_message[9] = 0x10010101;
477 vme_state_message[10] = 0x0F0F0F0F;
478 vme_state_message[11] = 0x000F0F0F;
479 vme_state_message[12] = 0x00;
480 vme_state_message[13] = 0x00;
482 vme_state_message[14] = 0x4a4a;
483 vme_state_message[15] = 0x0;
484 vme_state_message[16] = 0x4a4a4a4a;
485 vme_state_message[17] = 0x4a4a4a4a;
486 vme_state_message[18] = 0x21110100;
487 vme_state_message[19] = 0x61514131;
489 for(i = 20; i < 32; i++) {
490 vme_state_message[i] = 0;
492 //vme_state_message[16] = 0x42424242; //cost function LUT set 0 for Intra
494 gen6_vme_state_setup_fixup(ctx, encode_state, encoder_context, vme_state_message);
496 dri_bo_unmap( vme_context->vme_state.bo);
497 return VA_STATUS_SUCCESS;
501 gen6_vme_fill_vme_batchbuffer(VADriverContextP ctx,
502 struct encode_state *encode_state,
503 int mb_width, int mb_height,
505 int transform_8x8_mode_flag,
506 struct intel_encoder_context *encoder_context)
508 struct gen6_vme_context *vme_context = encoder_context->vme_context;
510 int mb_x = 0, mb_y = 0;
512 unsigned int *command_ptr;
514 dri_bo_map(vme_context->vme_batchbuffer.bo, 1);
515 command_ptr = vme_context->vme_batchbuffer.bo->virtual;
517 for (s = 0; s < encode_state->num_slice_params_ext; s++) {
518 VAEncSliceParameterBufferH264 *pSliceParameter = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[s]->buffer;
519 int slice_mb_begin = pSliceParameter->macroblock_address;
520 int slice_mb_number = pSliceParameter->num_macroblocks;
522 for (i = 0; i < slice_mb_number; ) {
523 int mb_count = i + slice_mb_begin;
524 mb_x = mb_count % mb_width;
525 mb_y = mb_count / mb_width;
527 number_mb_cmds = mb_width; // we must mark the slice edge.
528 } else if ( (i + 128 ) <= slice_mb_number) {
529 number_mb_cmds = 128;
531 number_mb_cmds = slice_mb_number - i;
534 *command_ptr++ = (CMD_MEDIA_OBJECT | (8 - 2));
535 *command_ptr++ = kernel;
542 *command_ptr++ = (mb_width << 16 | mb_y << 8 | mb_x);
543 *command_ptr++ = (number_mb_cmds << 16 | transform_8x8_mode_flag | ((i==0) << 1));
550 *command_ptr++ = MI_BATCH_BUFFER_END;
552 dri_bo_unmap(vme_context->vme_batchbuffer.bo);
555 static void gen6_vme_media_init(VADriverContextP ctx, struct intel_encoder_context *encoder_context)
557 struct i965_driver_data *i965 = i965_driver_data(ctx);
558 struct gen6_vme_context *vme_context = encoder_context->vme_context;
561 i965_gpe_context_init(ctx, &vme_context->gpe_context);
563 /* VME output buffer */
564 dri_bo_unreference(vme_context->vme_output.bo);
565 vme_context->vme_output.bo = NULL;
567 dri_bo_unreference(vme_context->vme_batchbuffer.bo);
568 vme_context->vme_batchbuffer.bo = NULL;
571 dri_bo_unreference(vme_context->vme_state.bo);
572 bo = dri_bo_alloc(i965->intel.bufmgr,
576 vme_context->vme_state.bo = bo;
579 static void gen6_vme_pipeline_programing(VADriverContextP ctx,
580 struct encode_state *encode_state,
581 struct intel_encoder_context *encoder_context)
583 struct gen6_vme_context *vme_context = encoder_context->vme_context;
584 struct intel_batchbuffer *batch = encoder_context->base.batch;
585 VAEncPictureParameterBufferH264 *pPicParameter = (VAEncPictureParameterBufferH264 *)encode_state->pic_param_ext->buffer;
586 VAEncSliceParameterBufferH264 *pSliceParameter = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[0]->buffer;
587 VAEncSequenceParameterBufferH264 *pSequenceParameter = (VAEncSequenceParameterBufferH264 *)encode_state->seq_param_ext->buffer;
588 int is_intra = pSliceParameter->slice_type == SLICE_TYPE_I;
589 int width_in_mbs = pSequenceParameter->picture_width_in_mbs;
590 int height_in_mbs = pSequenceParameter->picture_height_in_mbs;
592 gen6_vme_fill_vme_batchbuffer(ctx,
594 width_in_mbs, height_in_mbs,
595 is_intra ? AVC_VME_INTRA_SHADER : AVC_VME_INTER_SHADER,
596 pPicParameter->pic_fields.bits.transform_8x8_mode_flag,
599 intel_batchbuffer_start_atomic(batch, 0x1000);
600 gen6_gpe_pipeline_setup(ctx, &vme_context->gpe_context, batch);
601 BEGIN_BATCH(batch, 2);
602 OUT_BATCH(batch, MI_BATCH_BUFFER_START | (2 << 6));
604 vme_context->vme_batchbuffer.bo,
605 I915_GEM_DOMAIN_COMMAND, 0,
607 ADVANCE_BATCH(batch);
609 intel_batchbuffer_end_atomic(batch);
612 static VAStatus gen6_vme_prepare(VADriverContextP ctx,
613 struct encode_state *encode_state,
614 struct intel_encoder_context *encoder_context)
616 VAStatus vaStatus = VA_STATUS_SUCCESS;
617 VAEncSliceParameterBufferH264 *pSliceParameter = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[0]->buffer;
618 int is_intra = pSliceParameter->slice_type == SLICE_TYPE_I;
619 VAEncSequenceParameterBufferH264 *pSequenceParameter = (VAEncSequenceParameterBufferH264 *)encode_state->seq_param_ext->buffer;
620 struct gen6_vme_context *vme_context = encoder_context->vme_context;
622 if (!vme_context->h264_level ||
623 (vme_context->h264_level != pSequenceParameter->level_idc)) {
624 vme_context->h264_level = pSequenceParameter->level_idc;
626 /*Setup all the memory object*/
627 gen6_vme_surface_setup(ctx, encode_state, is_intra, encoder_context);
628 gen6_vme_interface_setup(ctx, encode_state, encoder_context);
629 gen6_vme_constant_setup(ctx, encode_state, encoder_context);
630 gen6_vme_vme_state_setup(ctx, encode_state, is_intra, encoder_context);
632 /*Programing media pipeline*/
633 gen6_vme_pipeline_programing(ctx, encode_state, encoder_context);
638 static VAStatus gen6_vme_run(VADriverContextP ctx,
639 struct encode_state *encode_state,
640 struct intel_encoder_context *encoder_context)
642 struct intel_batchbuffer *batch = encoder_context->base.batch;
644 intel_batchbuffer_flush(batch);
646 return VA_STATUS_SUCCESS;
649 static VAStatus gen6_vme_stop(VADriverContextP ctx,
650 struct encode_state *encode_state,
651 struct intel_encoder_context *encoder_context)
653 return VA_STATUS_SUCCESS;
657 gen6_vme_pipeline(VADriverContextP ctx,
659 struct encode_state *encode_state,
660 struct intel_encoder_context *encoder_context)
662 gen6_vme_media_init(ctx, encoder_context);
663 gen6_vme_prepare(ctx, encode_state, encoder_context);
664 gen6_vme_run(ctx, encode_state, encoder_context);
665 gen6_vme_stop(ctx, encode_state, encoder_context);
667 return VA_STATUS_SUCCESS;
671 gen6_vme_mpeg2_output_buffer_setup(VADriverContextP ctx,
672 struct encode_state *encode_state,
675 struct intel_encoder_context *encoder_context)
678 struct i965_driver_data *i965 = i965_driver_data(ctx);
679 struct gen6_vme_context *vme_context = encoder_context->vme_context;
680 VAEncSequenceParameterBufferMPEG2 *seq_param = (VAEncSequenceParameterBufferMPEG2 *)encode_state->seq_param_ext->buffer;
681 int width_in_mbs = ALIGN(seq_param->picture_width, 16) / 16;
682 int height_in_mbs = ALIGN(seq_param->picture_height, 16) / 16;
684 vme_context->vme_output.num_blocks = width_in_mbs * height_in_mbs;
685 vme_context->vme_output.pitch = 16; /* in bytes, always 16 */
688 vme_context->vme_output.size_block = INTRA_VME_OUTPUT_IN_BYTES;
690 vme_context->vme_output.size_block = INTER_VME_OUTPUT_IN_BYTES;
692 vme_context->vme_output.bo = dri_bo_alloc(i965->intel.bufmgr,
694 vme_context->vme_output.num_blocks * vme_context->vme_output.size_block,
696 assert(vme_context->vme_output.bo);
697 vme_context->vme_buffer_suface_setup(ctx,
698 &vme_context->gpe_context,
699 &vme_context->vme_output,
700 BINDING_TABLE_OFFSET(index),
701 SURFACE_STATE_OFFSET(index));
705 gen6_vme_mpeg2_output_vme_batchbuffer_setup(VADriverContextP ctx,
706 struct encode_state *encode_state,
708 struct intel_encoder_context *encoder_context)
711 struct i965_driver_data *i965 = i965_driver_data(ctx);
712 struct gen6_vme_context *vme_context = encoder_context->vme_context;
713 VAEncSequenceParameterBufferMPEG2 *seq_param = (VAEncSequenceParameterBufferMPEG2 *)encode_state->seq_param_ext->buffer;
714 int width_in_mbs = ALIGN(seq_param->picture_width, 16) / 16;
715 int height_in_mbs = ALIGN(seq_param->picture_height, 16) / 16;
717 vme_context->vme_batchbuffer.num_blocks = width_in_mbs * height_in_mbs + 1;
718 vme_context->vme_batchbuffer.size_block = 32; /* 4 OWORDs */
719 vme_context->vme_batchbuffer.pitch = 16;
720 vme_context->vme_batchbuffer.bo = dri_bo_alloc(i965->intel.bufmgr,
722 vme_context->vme_batchbuffer.num_blocks * vme_context->vme_batchbuffer.size_block,
724 vme_context->vme_buffer_suface_setup(ctx,
725 &vme_context->gpe_context,
726 &vme_context->vme_batchbuffer,
727 BINDING_TABLE_OFFSET(index),
728 SURFACE_STATE_OFFSET(index));
732 gen6_vme_mpeg2_surface_setup(VADriverContextP ctx,
733 struct encode_state *encode_state,
735 struct intel_encoder_context *encoder_context)
737 struct i965_driver_data *i965 = i965_driver_data(ctx);
738 struct object_surface *obj_surface;
739 VAEncPictureParameterBufferMPEG2 *pic_param = (VAEncPictureParameterBufferMPEG2 *)encode_state->pic_param_ext->buffer;
741 /*Setup surfaces state*/
742 /* current picture for encoding */
743 obj_surface = encode_state->input_yuv_object;
744 gen6_vme_source_surface_state(ctx, 0, obj_surface, encoder_context);
745 gen6_vme_media_source_surface_state(ctx, 4, obj_surface, encoder_context);
749 obj_surface = SURFACE(pic_param->forward_reference_picture);
751 if ( obj_surface->bo != NULL)
752 gen6_vme_source_surface_state(ctx, 1, obj_surface, encoder_context);
755 obj_surface = SURFACE(pic_param->backward_reference_picture);
756 if (obj_surface && obj_surface->bo != NULL)
757 gen6_vme_source_surface_state(ctx, 2, obj_surface, encoder_context);
761 gen6_vme_mpeg2_output_buffer_setup(ctx, encode_state, 3, is_intra, encoder_context);
762 gen6_vme_mpeg2_output_vme_batchbuffer_setup(ctx, encode_state, 5, encoder_context);
764 return VA_STATUS_SUCCESS;
768 gen6_vme_mpeg2_fill_vme_batchbuffer(VADriverContextP ctx,
769 struct encode_state *encode_state,
770 int mb_width, int mb_height,
772 int transform_8x8_mode_flag,
773 struct intel_encoder_context *encoder_context)
775 struct gen6_vme_context *vme_context = encoder_context->vme_context;
777 int mb_x = 0, mb_y = 0;
779 unsigned int *command_ptr;
781 dri_bo_map(vme_context->vme_batchbuffer.bo, 1);
782 command_ptr = vme_context->vme_batchbuffer.bo->virtual;
784 for (s = 0; s < encode_state->num_slice_params_ext; s++) {
785 VAEncSliceParameterBufferMPEG2 *slice_param = (VAEncSliceParameterBufferMPEG2 *)encode_state->slice_params_ext[s]->buffer;
787 for (j = 0; j < encode_state->slice_params_ext[s]->num_elements; j++) {
788 int slice_mb_begin = slice_param->macroblock_address;
789 int slice_mb_number = slice_param->num_macroblocks;
791 for (i = 0; i < slice_mb_number;) {
792 int mb_count = i + slice_mb_begin;
794 mb_x = mb_count % mb_width;
795 mb_y = mb_count / mb_width;
798 number_mb_cmds = mb_width;
799 } else if ((i + 128) <= slice_mb_number) {
800 number_mb_cmds = 128;
802 number_mb_cmds = slice_mb_number - i;
805 *command_ptr++ = (CMD_MEDIA_OBJECT | (8 - 2));
806 *command_ptr++ = kernel;
813 *command_ptr++ = (mb_width << 16 | mb_y << 8 | mb_x);
814 *command_ptr++ = ( (number_mb_cmds << 16) | transform_8x8_mode_flag | ((i == 0) << 1));
824 *command_ptr++ = MI_BATCH_BUFFER_END;
826 dri_bo_unmap(vme_context->vme_batchbuffer.bo);
830 gen6_vme_mpeg2_pipeline_programing(VADriverContextP ctx,
831 struct encode_state *encode_state,
833 struct intel_encoder_context *encoder_context)
835 struct gen6_vme_context *vme_context = encoder_context->vme_context;
836 struct intel_batchbuffer *batch = encoder_context->base.batch;
837 VAEncSequenceParameterBufferMPEG2 *seq_param = (VAEncSequenceParameterBufferMPEG2 *)encode_state->seq_param_ext->buffer;
838 int width_in_mbs = ALIGN(seq_param->picture_width, 16) / 16;
839 int height_in_mbs = ALIGN(seq_param->picture_height, 16) / 16;
841 gen6_vme_mpeg2_fill_vme_batchbuffer(ctx,
843 width_in_mbs, height_in_mbs,
844 MPEG2_VME_INTER_SHADER,
848 intel_batchbuffer_start_atomic(batch, 0x1000);
849 gen6_gpe_pipeline_setup(ctx, &vme_context->gpe_context, batch);
850 BEGIN_BATCH(batch, 2);
851 OUT_BATCH(batch, MI_BATCH_BUFFER_START | (2 << 6));
853 vme_context->vme_batchbuffer.bo,
854 I915_GEM_DOMAIN_COMMAND, 0,
856 ADVANCE_BATCH(batch);
858 intel_batchbuffer_end_atomic(batch);
862 gen6_vme_mpeg2_prepare(VADriverContextP ctx,
863 struct encode_state *encode_state,
864 struct intel_encoder_context *encoder_context)
866 VAStatus vaStatus = VA_STATUS_SUCCESS;
868 /*Setup all the memory object*/
869 gen6_vme_mpeg2_surface_setup(ctx, encode_state, 0, encoder_context);
870 gen6_vme_interface_setup(ctx, encode_state, encoder_context);
871 gen6_vme_vme_state_setup(ctx, encode_state, 0, encoder_context);
872 gen6_vme_constant_setup(ctx, encode_state, encoder_context);
874 /*Programing media pipeline*/
875 gen6_vme_mpeg2_pipeline_programing(ctx, encode_state, 0, encoder_context);
881 gen6_vme_mpeg2_pipeline(VADriverContextP ctx,
883 struct encode_state *encode_state,
884 struct intel_encoder_context *encoder_context)
886 struct i965_driver_data *i965 = i965_driver_data(ctx);
887 struct gen6_vme_context *vme_context = encoder_context->vme_context;
888 VAEncSliceParameterBufferMPEG2 *slice_param =
889 (VAEncSliceParameterBufferMPEG2 *)encode_state->slice_params_ext[0]->buffer;
890 VAEncSequenceParameterBufferMPEG2 *seq_param =
891 (VAEncSequenceParameterBufferMPEG2 *)encode_state->seq_param_ext->buffer;
893 /*No need of to exec VME for Intra slice */
894 if (slice_param->is_intra_slice) {
895 if(!vme_context->vme_output.bo) {
896 int w_in_mbs = ALIGN(seq_param->picture_width, 16) / 16;
897 int h_in_mbs = ALIGN(seq_param->picture_height, 16) / 16;
899 vme_context->vme_output.num_blocks = w_in_mbs * h_in_mbs;
900 vme_context->vme_output.pitch = 16; /* in bytes, always 16 */
901 vme_context->vme_output.size_block = INTRA_VME_OUTPUT_IN_BYTES;
902 vme_context->vme_output.bo = dri_bo_alloc(i965->intel.bufmgr,
903 "MPEG2 VME output buffer",
904 vme_context->vme_output.num_blocks
905 * vme_context->vme_output.size_block,
909 return VA_STATUS_SUCCESS;
912 gen6_vme_media_init(ctx, encoder_context);
913 gen6_vme_mpeg2_prepare(ctx, encode_state, encoder_context);
914 gen6_vme_run(ctx, encode_state, encoder_context);
915 gen6_vme_stop(ctx, encode_state, encoder_context);
917 return VA_STATUS_SUCCESS;
921 gen6_vme_context_destroy(void *context)
923 struct gen6_vme_context *vme_context = context;
925 i965_gpe_context_destroy(&vme_context->gpe_context);
927 dri_bo_unreference(vme_context->vme_output.bo);
928 vme_context->vme_output.bo = NULL;
930 dri_bo_unreference(vme_context->vme_state.bo);
931 vme_context->vme_state.bo = NULL;
933 dri_bo_unreference(vme_context->vme_batchbuffer.bo);
934 vme_context->vme_batchbuffer.bo = NULL;
939 Bool gen6_vme_context_init(VADriverContextP ctx, struct intel_encoder_context *encoder_context)
941 struct i965_driver_data *i965 = i965_driver_data(ctx);
942 struct gen6_vme_context *vme_context = calloc(1, sizeof(struct gen6_vme_context));
944 vme_context->gpe_context.surface_state_binding_table.length =
945 (SURFACE_STATE_PADDED_SIZE + sizeof(unsigned int)) * MAX_MEDIA_SURFACES_GEN6;
947 vme_context->gpe_context.idrt.max_entries = MAX_INTERFACE_DESC_GEN6;
948 vme_context->gpe_context.idrt.entry_size = sizeof(struct gen6_interface_descriptor_data);
949 vme_context->gpe_context.curbe.length = CURBE_TOTAL_DATA_LENGTH;
951 vme_context->gpe_context.vfe_state.max_num_threads = 60 - 1;
952 vme_context->gpe_context.vfe_state.num_urb_entries = 16;
953 vme_context->gpe_context.vfe_state.gpgpu_mode = 0;
954 vme_context->gpe_context.vfe_state.urb_entry_size = 59 - 1;
955 vme_context->gpe_context.vfe_state.curbe_allocation_size = CURBE_ALLOCATION_SIZE - 1;
957 if(encoder_context->profile == VAProfileH264Baseline ||
958 encoder_context->profile == VAProfileH264Main ||
959 encoder_context->profile == VAProfileH264High ){
960 vme_context->video_coding_type = VIDEO_CODING_AVC;
961 vme_context->vme_kernel_sum = AVC_VME_KERNEL_SUM;
963 } else if (encoder_context->profile == VAProfileMPEG2Simple ||
964 encoder_context->profile == VAProfileMPEG2Main ){
965 vme_context->video_coding_type = VIDEO_CODING_MPEG2;
966 vme_context->vme_kernel_sum = MPEG2_VME_KERNEL_SUM;
968 /* Unsupported encoding profile */
972 if (IS_GEN7(i965->intel.device_id)) {
973 if (vme_context->video_coding_type == VIDEO_CODING_AVC) {
974 i965_gpe_load_kernels(ctx,
975 &vme_context->gpe_context,
977 vme_context->vme_kernel_sum);
978 encoder_context->vme_pipeline = gen6_vme_pipeline;
981 i965_gpe_load_kernels(ctx,
982 &vme_context->gpe_context,
983 gen7_vme_mpeg2_kernels,
984 vme_context->vme_kernel_sum);
985 encoder_context->vme_pipeline = gen6_vme_mpeg2_pipeline;
989 vme_context->vme_surface2_setup = gen7_gpe_surface2_setup;
990 vme_context->vme_media_rw_surface_setup = gen7_gpe_media_rw_surface_setup;
991 vme_context->vme_buffer_suface_setup = gen7_gpe_buffer_suface_setup;
993 if (vme_context->video_coding_type == VIDEO_CODING_AVC) {
994 i965_gpe_load_kernels(ctx,
995 &vme_context->gpe_context,
997 vme_context->vme_kernel_sum);
999 encoder_context->vme_pipeline = gen6_vme_pipeline;
1001 /* Unsupported encoding profile for gen6 */
1005 vme_context->vme_surface2_setup = i965_gpe_surface2_setup;
1006 vme_context->vme_media_rw_surface_setup = i965_gpe_media_rw_surface_setup;
1007 vme_context->vme_buffer_suface_setup = i965_gpe_buffer_suface_setup;
1010 encoder_context->vme_context = vme_context;
1011 encoder_context->vme_context_destroy = gen6_vme_context_destroy;