2 * Copyright © 2010 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 extern Bool gen6_mfc_context_init(VADriverContextP ctx, struct intel_encoder_context *encoder_context);
44 extern Bool gen6_vme_context_init(VADriverContextP ctx, struct intel_encoder_context *encoder_context);
45 extern Bool gen7_mfc_context_init(VADriverContextP ctx, struct intel_encoder_context *encoder_context);
48 i965_DestroySurfaces(VADriverContextP ctx,
49 VASurfaceID *surface_list,
52 i965_CreateSurfaces(VADriverContextP ctx,
57 VASurfaceID *surfaces);
60 intel_encoder_check_yuv_surface(VADriverContextP ctx,
62 struct encode_state *encode_state,
63 struct intel_encoder_context *encoder_context)
65 struct i965_driver_data *i965 = i965_driver_data(ctx);
66 struct i965_surface src_surface, dst_surface;
67 struct object_surface *obj_surface;
71 /* releae the temporary surface */
72 if (encoder_context->is_tmp_id) {
73 i965_DestroySurfaces(ctx, &encoder_context->input_yuv_surface, 1);
74 encode_state->input_yuv_object = NULL;
77 encoder_context->is_tmp_id = 0;
78 obj_surface = SURFACE(encode_state->current_render_target);
79 assert(obj_surface && obj_surface->bo);
81 if (!obj_surface || !obj_surface->bo)
82 return VA_STATUS_ERROR_INVALID_PARAMETER;
84 if (obj_surface->fourcc == VA_FOURCC('N', 'V', '1', '2')) {
85 unsigned int tiling = 0, swizzle = 0;
87 dri_bo_get_tiling(obj_surface->bo, &tiling, &swizzle);
89 if (tiling == I915_TILING_Y) {
90 encoder_context->input_yuv_surface = encode_state->current_render_target;
91 encode_state->input_yuv_object = obj_surface;
92 return VA_STATUS_SUCCESS;
98 rect.width = obj_surface->orig_width;
99 rect.height = obj_surface->orig_height;
101 src_surface.base = (struct object_base *)obj_surface;
102 src_surface.type = I965_SURFACE_TYPE_SURFACE;
103 src_surface.flags = I965_SURFACE_FLAG_FRAME;
105 status = i965_CreateSurfaces(ctx,
106 obj_surface->orig_width,
107 obj_surface->orig_height,
110 &encoder_context->input_yuv_surface);
111 assert(status == VA_STATUS_SUCCESS);
113 if (status != VA_STATUS_SUCCESS)
116 obj_surface = SURFACE(encoder_context->input_yuv_surface);
117 encode_state->input_yuv_object = obj_surface;
119 i965_check_alloc_surface_bo(ctx, obj_surface, 1, VA_FOURCC('N', 'V', '1', '2'), SUBSAMPLE_YUV420);
121 dst_surface.base = (struct object_base *)obj_surface;
122 dst_surface.type = I965_SURFACE_TYPE_SURFACE;
123 dst_surface.flags = I965_SURFACE_FLAG_FRAME;
125 status = i965_image_processing(ctx,
130 assert(status == VA_STATUS_SUCCESS);
132 encoder_context->is_tmp_id = 1;
134 return VA_STATUS_SUCCESS;
138 intel_encoder_check_avc_parameter(VADriverContextP ctx,
139 struct encode_state *encode_state,
140 struct intel_encoder_context *encoder_context)
142 struct i965_driver_data *i965 = i965_driver_data(ctx);
143 struct object_surface *obj_surface;
144 struct object_buffer *obj_buffer;
145 VAEncPictureParameterBufferH264 *pic_param = (VAEncPictureParameterBufferH264 *)encode_state->pic_param_ext->buffer;
148 assert(!(pic_param->CurrPic.flags & VA_PICTURE_H264_INVALID));
150 if (pic_param->CurrPic.flags & VA_PICTURE_H264_INVALID)
153 obj_surface = SURFACE(pic_param->CurrPic.picture_id);
154 assert(obj_surface); /* It is possible the store buffer isn't allocated yet */
159 encode_state->reconstructed_object = obj_surface;
160 obj_buffer = BUFFER(pic_param->coded_buf);
161 assert(obj_buffer && obj_buffer->buffer_store && obj_buffer->buffer_store->bo);
163 if (!obj_buffer || !obj_buffer->buffer_store || !obj_buffer->buffer_store->bo)
166 encode_state->coded_buf_object = obj_buffer;
168 for (i = 0; i < 16; i++) {
169 if (pic_param->ReferenceFrames[i].flags & VA_PICTURE_H264_INVALID ||
170 pic_param->ReferenceFrames[i].picture_id == VA_INVALID_SURFACE)
173 obj_surface = SURFACE(pic_param->ReferenceFrames[i].picture_id);
180 encode_state->reference_objects[i] = obj_surface;
182 encode_state->reference_objects[i] = NULL; /* FIXME: Warning or Error ??? */
187 encode_state->reference_objects[i] = NULL;
189 return VA_STATUS_SUCCESS;
192 return VA_STATUS_ERROR_INVALID_PARAMETER;
196 intel_encoder_check_mpeg2_parameter(VADriverContextP ctx,
197 struct encode_state *encode_state,
198 struct intel_encoder_context *encoder_context)
200 struct i965_driver_data *i965 = i965_driver_data(ctx);
201 VAEncPictureParameterBufferMPEG2 *pic_param = (VAEncPictureParameterBufferMPEG2 *)encode_state->pic_param_ext->buffer;
202 struct object_surface *obj_surface;
203 struct object_buffer *obj_buffer;
206 obj_surface = SURFACE(pic_param->reconstructed_picture);
207 assert(obj_surface); /* It is possible the store buffer isn't allocated yet */
212 encode_state->reconstructed_object = obj_surface;
213 obj_buffer = BUFFER(pic_param->coded_buf);
214 assert(obj_buffer && obj_buffer->buffer_store && obj_buffer->buffer_store->bo);
216 if (!obj_buffer || !obj_buffer->buffer_store || !obj_buffer->buffer_store->bo)
219 encode_state->coded_buf_object = obj_buffer;
221 if (pic_param->picture_type == VAEncPictureTypeIntra) {
222 } else if (pic_param->picture_type == VAEncPictureTypePredictive) {
223 assert(pic_param->forward_reference_picture != VA_INVALID_SURFACE);
224 obj_surface = SURFACE(pic_param->forward_reference_picture);
225 assert(obj_surface && obj_surface->bo);
227 if (!obj_surface || !obj_surface->bo)
230 encode_state->reference_objects[i++] = obj_surface;
231 } else if (pic_param->picture_type == VAEncPictureTypeBidirectional) {
232 assert(pic_param->forward_reference_picture != VA_INVALID_SURFACE);
233 obj_surface = SURFACE(pic_param->forward_reference_picture);
234 assert(obj_surface && obj_surface->bo);
236 if (!obj_surface || !obj_surface->bo)
239 encode_state->reference_objects[i++] = obj_surface;
241 assert(pic_param->backward_reference_picture != VA_INVALID_SURFACE);
242 obj_surface = SURFACE(pic_param->backward_reference_picture);
243 assert(obj_surface && obj_surface->bo);
245 if (!obj_surface || !obj_surface->bo)
248 encode_state->reference_objects[i++] = obj_surface;
253 encode_state->reference_objects[i] = NULL;
255 return VA_STATUS_SUCCESS;
258 return VA_STATUS_ERROR_INVALID_PARAMETER;
262 intel_encoder_sanity_check_input(VADriverContextP ctx,
264 struct encode_state *encode_state,
265 struct intel_encoder_context *encoder_context)
270 case VAProfileH264ConstrainedBaseline:
271 case VAProfileH264Main:
272 case VAProfileH264High:
273 vaStatus = intel_encoder_check_avc_parameter(ctx, encode_state, encoder_context);
276 case VAProfileMPEG2Simple:
277 case VAProfileMPEG2Main:
278 vaStatus = intel_encoder_check_mpeg2_parameter(ctx, encode_state, encoder_context);
282 vaStatus = VA_STATUS_ERROR_UNSUPPORTED_PROFILE;
286 if (vaStatus != VA_STATUS_SUCCESS)
289 vaStatus = intel_encoder_check_yuv_surface(ctx, profile, encode_state, encoder_context);
296 intel_encoder_end_picture(VADriverContextP ctx,
298 union codec_state *codec_state,
299 struct hw_context *hw_context)
301 struct intel_encoder_context *encoder_context = (struct intel_encoder_context *)hw_context;
302 struct encode_state *encode_state = &codec_state->encode;
305 vaStatus = intel_encoder_sanity_check_input(ctx, profile, encode_state, encoder_context);
307 if (vaStatus != VA_STATUS_SUCCESS)
310 encoder_context->mfc_brc_prepare(encode_state, encoder_context);
312 vaStatus = encoder_context->vme_pipeline(ctx, profile, encode_state, encoder_context);
314 if (vaStatus == VA_STATUS_SUCCESS)
315 encoder_context->mfc_pipeline(ctx, profile, encode_state, encoder_context);
316 return VA_STATUS_SUCCESS;
320 intel_encoder_context_destroy(void *hw_context)
322 struct intel_encoder_context *encoder_context = (struct intel_encoder_context *)hw_context;
324 encoder_context->mfc_context_destroy(encoder_context->mfc_context);
325 encoder_context->vme_context_destroy(encoder_context->vme_context);
326 intel_batchbuffer_free(encoder_context->base.batch);
327 free(encoder_context);
330 typedef Bool (* hw_init_func)(VADriverContextP, struct intel_encoder_context *);
332 static struct hw_context *
333 intel_enc_hw_context_init(VADriverContextP ctx,
334 struct object_config *obj_config,
335 hw_init_func vme_context_init,
336 hw_init_func mfc_context_init)
338 struct intel_driver_data *intel = intel_driver_data(ctx);
339 struct intel_encoder_context *encoder_context = calloc(1, sizeof(struct intel_encoder_context));
342 encoder_context->base.destroy = intel_encoder_context_destroy;
343 encoder_context->base.run = intel_encoder_end_picture;
344 encoder_context->base.batch = intel_batchbuffer_new(intel, I915_EXEC_RENDER, 0);
345 encoder_context->input_yuv_surface = VA_INVALID_SURFACE;
346 encoder_context->is_tmp_id = 0;
347 encoder_context->rate_control_mode = VA_RC_NONE;
349 switch (obj_config->profile) {
350 case VAProfileMPEG2Simple:
351 case VAProfileMPEG2Main:
352 encoder_context->codec = CODEC_MPEG2;
355 case VAProfileH264ConstrainedBaseline:
356 case VAProfileH264Main:
357 case VAProfileH264High:
358 encoder_context->codec = CODEC_H264;
367 for (i = 0; i < obj_config->num_attribs; i++) {
368 if (obj_config->attrib_list[i].type == VAConfigAttribRateControl) {
369 encoder_context->rate_control_mode = obj_config->attrib_list[i].value;
371 if (encoder_context->codec == CODEC_MPEG2 &&
372 encoder_context->rate_control_mode & VA_RC_CBR) {
373 WARN_ONCE("Don't support CBR for MPEG-2 encoding\n");
374 encoder_context->rate_control_mode &= ~VA_RC_CBR;
381 vme_context_init(ctx, encoder_context);
382 assert(encoder_context->vme_context);
383 assert(encoder_context->vme_context_destroy);
384 assert(encoder_context->vme_pipeline);
386 mfc_context_init(ctx, encoder_context);
387 assert(encoder_context->mfc_context);
388 assert(encoder_context->mfc_context_destroy);
389 assert(encoder_context->mfc_pipeline);
391 return (struct hw_context *)encoder_context;
395 gen6_enc_hw_context_init(VADriverContextP ctx, struct object_config *obj_config)
397 return intel_enc_hw_context_init(ctx, obj_config, gen6_vme_context_init, gen6_mfc_context_init);
401 gen7_enc_hw_context_init(VADriverContextP ctx, struct object_config *obj_config)
403 return intel_enc_hw_context_init(ctx, obj_config, gen7_vme_context_init, gen7_mfc_context_init);
407 gen75_enc_hw_context_init(VADriverContextP ctx, struct object_config *obj_config)
409 return intel_enc_hw_context_init(ctx, obj_config, gen75_vme_context_init, gen75_mfc_context_init);
413 gen8_enc_hw_context_init(VADriverContextP ctx, struct object_config *obj_config)
415 return intel_enc_hw_context_init(ctx, obj_config, gen8_vme_context_init, gen8_mfc_context_init);