[Libva] [PATCH 3/3] Expose h264 encoder motion vector buffer on IVB

[Xiang, Haihao]

Please add a version number (v2, v3...) to your patch and state what is
changed.


> Get motion vectors of vme output by FEI interface.
> Set mv to be 0x8000 when intra MB.
> 
> Signed-off-by: Zhong Li <zhong.li at intel.com>
> ---
>  src/gen6_mfc.c |   72 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
>  src/gen6_vme.h |    2 ++
>  2 files changed, 74 insertions(+)
> 
> diff --git a/src/gen6_mfc.c b/src/gen6_mfc.c
> index 21db0a7..17a81c7 100644
> --- a/src/gen6_mfc.c
> +++ b/src/gen6_mfc.c
> @@ -1357,6 +1357,77 @@ gen6_mfc_avc_pipeline_programing(VADriverContextP ctx,
>      dri_bo_unreference(slice_batch_bo);
>  }
>  
> +static void
> +gen6_mfc_get_slice_motion_vector(VADriverContextP ctx,
> +		struct encode_state *encode_state,
> +		struct intel_encoder_context *encoder_context,
> +		int slice_index,
> +		int misc_param_idx
> +		)
> +{
> +    struct i965_driver_data *i965 = i965_driver_data(ctx);
> +    struct gen6_vme_context *vme_context = encoder_context->vme_context;
> +    VAEncMiscParameterBuffer *misc_param = (VAEncMiscParameterBuffer *)encode_state->misc_param[misc_param_idx]->buffer;
> +    VAEncMiscParameterFEIFrameControlH264Intel *pFeiFrameControlParameter = (VAEncMiscParameterFEIFrameControlH264Intel *)misc_param->data;
> +    VAEncSliceParameterBufferH264 *pSliceParameter = (VAEncSliceParameterBufferH264 *)encode_state->slice_params_ext[slice_index]->buffer;
> +    unsigned int *msg = NULL;
> +    int i;
> +    int slice_type = intel_avc_enc_slice_type_fixup(pSliceParameter->slice_type);
> +    int is_intra = slice_type == SLICE_TYPE_I;
> +    struct object_buffer *mv_buf = BUFFER(pFeiFrameControlParameter->mv_data);
> +
> +    if (mv_buf && mv_buf->buffer_store && mv_buf->buffer_store->buffer) {
> +	unsigned char *mv_buffer = mv_buf->buffer_store->buffer;
> +	unsigned int *intra_mv = (unsigned int *)malloc(INTER_VME_OUTPUT_MV_IN_BYTES);
> +
> +	dri_bo_map(vme_context->vme_output.bo , 1);
> +	msg = (unsigned int *)vme_context->vme_output.bo->virtual;
> +
> +	for (i = 0; i < 32; i++) {
> +	  intra_mv[i] = 0x80008000; //for intra mb, set mv.x and mv.y to be 0x8000
> +	}
> +
> +	if (is_intra) {
> +	    for (i = 0; i < pSliceParameter->num_macroblocks; i++) {
> +		memcpy(mv_buffer, intra_mv, INTER_VME_OUTPUT_MV_IN_BYTES);
> +		mv_buffer += INTER_VME_OUTPUT_MV_IN_BYTES;
> +	    }
> +	} else {
> +	    msg += pSliceParameter->macroblock_address * INTER_VME_OUTPUT_IN_DWS;
> +	    msg += 32; /* the first 32 DWs are MVs */
> +
> +	    for (i = 0; i < pSliceParameter->num_macroblocks; i++) {
> +		if (msg[0] & INTRA_MB_FLAG_MASK) {
> +		    memcpy(mv_buffer, intra_mv, INTER_VME_OUTPUT_MV_IN_BYTES);
> +		} else {
> +		    memcpy(mv_buffer, (char *)(msg - 32), INTER_VME_OUTPUT_MV_IN_BYTES);
> +		}
> +		mv_buffer += INTER_VME_OUTPUT_MV_IN_BYTES;
> +		msg += INTER_VME_OUTPUT_IN_DWS;
> +	    }

Why does msg directly point to MVs ? It is easy to understand if msg
points to MVs.

> +	}
> +
> +	free(intra_mv);
> +	dri_bo_unmap(vme_context->vme_output.bo);
> +    }
> +}
> +
> +static void
> +gen6_mfc_get_motion_vector(VADriverContextP ctx,
> +		struct encode_state *encode_state,
> +		struct intel_encoder_context *encoder_context)
> +{
> +    int i;
> +    int misc_param_idx = va_enc_misc_param_type_to_idx(VAEncMiscParameterTypeFEIFrameControlIntel);
> +
> +    if (encode_state->misc_param[misc_param_idx]
> +	&& encode_state->misc_param[misc_param_idx]->buffer) {
> +	for (i = 0; i < encode_state->num_slice_params_ext; i++) {
> +	    gen6_mfc_get_slice_motion_vector(ctx, encode_state, encoder_context, i, misc_param_idx);
> +	}
> +    }
> +}
> +
>  VAStatus
>  gen6_mfc_avc_encode_picture(VADriverContextP ctx, 
>                              struct encode_state *encode_state,
> @@ -1373,6 +1444,7 @@ gen6_mfc_avc_encode_picture(VADriverContextP ctx,
>          /*Programing bcs pipeline*/
>          gen6_mfc_avc_pipeline_programing(ctx, encode_state, encoder_context);	//filling the pipeline
>          gen6_mfc_run(ctx, encode_state, encoder_context);
> +        gen6_mfc_get_motion_vector(ctx, encode_state, encoder_context);
>          if (rate_control_mode == VA_RC_CBR /*|| rate_control_mode == VA_RC_VBR*/) {
>              gen6_mfc_stop(ctx, encode_state, encoder_context, &current_frame_bits_size);
>              sts = intel_mfc_brc_postpack(encode_state, mfc_context, current_frame_bits_size);
> diff --git a/src/gen6_vme.h b/src/gen6_vme.h
> index d461982..db3fb27 100644
> --- a/src/gen6_vme.h
> +++ b/src/gen6_vme.h
> @@ -40,6 +40,8 @@
>  #define INTRA_VME_OUTPUT_IN_DWS         (INTRA_VME_OUTPUT_IN_BYTES / 4)
>  #define INTER_VME_OUTPUT_IN_BYTES       160     /* the first 128 bytes for MVs and the last 32 bytes for other info */
>  #define INTER_VME_OUTPUT_IN_DWS         (INTER_VME_OUTPUT_IN_BYTES / 4)
> +#define INTER_VME_OUTPUT_MV_IN_BYTES    128
> +#define INTER_VME_OUTPUT_MV_IN_DWs      (INTER_VME_OUTPUT_MV_IN_BYTES / 4)
>  
>  #define MAX_INTERFACE_DESC_GEN6         MAX_GPE_KERNELS
>  #define MAX_MEDIA_SURFACES_GEN6         34