VAStatus vld_dec_allocate_colocated_buffer(context_DEC_p ctx, object_surface_p obj_surface, uint32_t size)
{
psb_buffer_p buf;
VAStatus vaStatus;
psb_surface_p surface = obj_surface->psb_surface;
int index = GET_SURFACE_INFO_colocated_index(surface);
if (!index) {
index = ctx->colocated_buffers_idx;
if (index >= ctx->colocated_buffers_size) {
return VA_STATUS_ERROR_UNKNOWN;
}
drv_debug_msg(VIDEO_DEBUG_GENERAL, "Allocating colocated buffer for surface %08x size = %08x\n", surface, size);
buf = &(ctx->colocated_buffers[index]);
vaStatus = psb_buffer_create(ctx->obj_context->driver_data, size, psb_bt_vpu_only, buf);
if (VA_STATUS_SUCCESS != vaStatus) {
return vaStatus;
}
ctx->colocated_buffers_idx++;
SET_SURFACE_INFO_colocated_index(surface, index + 1); /* 0 means unset, index is offset by 1 */
} else {
buf = &(ctx->colocated_buffers[index - 1]);
if (buf->size < size) {
psb_buffer_destroy(buf);
vaStatus = psb_buffer_create(ctx->obj_context->driver_data, size, psb_bt_vpu_only, buf);
if (VA_STATUS_SUCCESS != vaStatus) {
return vaStatus;
}
SET_SURFACE_INFO_colocated_index(surface, index); /* replace the original buffer */
}
}
return VA_STATUS_SUCCESS;
}
VAStatus vld_dec_CreateContext(context_DEC_p ctx, object_context_p obj_context)
{
VAStatus vaStatus = VA_STATUS_SUCCESS;
ctx->obj_context = obj_context;
ctx->split_buffer_pending = FALSE;
ctx->slice_param_list_size = 8;
ctx->slice_param_list = (object_buffer_p*) calloc(1, sizeof(object_buffer_p) * ctx->slice_param_list_size);
ctx->slice_param_list_idx = 0;
if (NULL == ctx->slice_param_list) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
return vaStatus;
}
ctx->colocated_buffers_size = obj_context->num_render_targets;
ctx->colocated_buffers_idx = 0;
ctx->colocated_buffers = (psb_buffer_p) calloc(1, sizeof(struct psb_buffer_s) * ctx->colocated_buffers_size);
if (NULL == ctx->colocated_buffers) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
free(ctx->slice_param_list);
}
if (vaStatus == VA_STATUS_SUCCESS) {
vaStatus = psb_buffer_create(obj_context->driver_data,
AUX_LINE_BUFFER_VLD_SIZE,
psb_bt_cpu_vpu,
&ctx->aux_line_buffer_vld);
DEBUG_FAILURE;
}
return vaStatus;
}
static VAStatus tng_JPEG_CreateContext(
object_context_p obj_context,
object_config_p obj_config) {
VAStatus vaStatus = VA_STATUS_SUCCESS;
context_JPEG_p ctx;
/* Validate flag */
/* Validate picture dimensions */
vaStatus = tng__JPEG_check_legal_picture(obj_context, obj_config);
if (VA_STATUS_SUCCESS != vaStatus) {
DEBUG_FAILURE;
return vaStatus;
}
ctx = (context_JPEG_p) calloc(1, sizeof(struct context_JPEG_s));
if (NULL == ctx) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
return vaStatus;
}
obj_context->format_data = (void*) ctx;
ctx->obj_context = obj_context;
ctx->pic_params = NULL;
ctx->dec_ctx.begin_slice = tng__JPEG_begin_slice;
ctx->dec_ctx.process_slice = tng__JPEG_process_slice_data;
ctx->dec_ctx.end_slice = tng__JPEG_end_slice;
ctx->dec_ctx.process_buffer = tng_JPEG_process_buffer;
ctx->dec_ctx.preload_buffer = NULL;
switch (obj_config->profile) {
case VAProfileJPEGBaseline:
drv_debug_msg(VIDEO_DEBUG_GENERAL, "JPEG_PROFILE_BASELINE\n");
ctx->profile = JPEG_PROFILE_BASELINE;
break;
default:
ASSERT(0 == 1);
vaStatus = VA_STATUS_ERROR_UNKNOWN;
}
ctx->vlctable_buffer_size = 1984 * 2;
if (vaStatus == VA_STATUS_SUCCESS) {
vaStatus = psb_buffer_create(obj_context->driver_data,
ctx->vlctable_buffer_size,
psb_bt_cpu_vpu,
&ctx->vlc_packed_table);
DEBUG_FAILURE;
}
if (vaStatus == VA_STATUS_SUCCESS) {
vaStatus = vld_dec_CreateContext(&ctx->dec_ctx, obj_context);
DEBUG_FAILURE;
}
if (vaStatus != VA_STATUS_SUCCESS) {
tng_JPEG_DestroyContext(obj_context);
}
return vaStatus;
}
static VAStatus vsp_vp8_process_seqence_param(
psb_driver_data_p driver_data,
context_VPP_p ctx,
object_buffer_p obj_buffer)
{
VAStatus vaStatus = VA_STATUS_SUCCESS;
vsp_cmdbuf_p cmdbuf = ctx->obj_context->vsp_cmdbuf;
int i;
int ref_frame_width, ref_frame_height, ref_chroma_height, ref_size;
VAEncSequenceParameterBufferVP8 *va_seq =
(VAEncSequenceParameterBufferVP8 *) obj_buffer->buffer_data;
struct VssVp8encSequenceParameterBuffer *seq = &ctx->vp8_seq_param;
struct VssVp8encSequenceParameterBuffer *seq_to_firmware =
(struct VssVp8encSequenceParameterBuffer *)cmdbuf->seq_param_p;
struct ref_frame_surface *ref =
(struct ref_frame_surface*)cmdbuf->ref_param_p;
seq->frame_width = va_seq->frame_width;
seq->frame_height = va_seq->frame_height;
seq->rc_target_bitrate = va_seq->bits_per_second / 1000;
seq->max_intra_rate = 100 * ctx->max_frame_size /
(va_seq->bits_per_second / seq->frame_rate);
/* FIXME: API doc says max 5000, but for current default test vector we still use 6000 */
seq->kf_mode = va_seq->kf_auto; /* AUTO */
seq->kf_max_dist = va_seq->kf_max_dist;
seq->kf_min_dist = va_seq->kf_min_dist;
seq->error_resilient = va_seq->error_resilient;
ref_frame_width = (seq->frame_width + 2 * 32 + 63) & (~63);
ref_frame_height = (seq->frame_height + 2 * 32 + 63) & (~63);
ref_chroma_height = (ref_frame_height / 2 + 63) & (~63);
ref_size = ref_frame_width * (ref_frame_height + ref_chroma_height);
for (i = 0; i < 4; i++) {
seq->ref_frame_buffers[i].surface_id = va_seq->reference_frames[i];
seq->ref_frame_buffers[i].width = ref_frame_width;
seq->ref_frame_buffers[i].height = ref_frame_height;
}
for (i = 0; i < 4; i++) {
object_surface_p ref_surf = SURFACE(va_seq->reference_frames[i]);
if (!ref_surf)
return VA_STATUS_ERROR_UNKNOWN;
ref_surf->is_ref_surface = 2;
if (ref_surf->psb_surface->size < ref_size) {
/* re-alloc buffer */
ref_surf->psb_surface->size = ref_size;
psb_buffer_destroy(&ref_surf->psb_surface->buf);
vaStatus = psb_buffer_create(driver_data, ref_surf->psb_surface->size, psb_bt_surface, &ref_surf->psb_surface->buf);
if (VA_STATUS_SUCCESS != vaStatus)
return VA_STATUS_ERROR_ALLOCATION_FAILED;
}
vsp_cmdbuf_reloc_pic_param(&(seq->ref_frame_buffers[i].base),
0,
&(ref_surf->psb_surface->buf),
cmdbuf->param_mem_loc, seq);
}
*seq_to_firmware = *seq;
vsp_cmdbuf_insert_command(cmdbuf, CONTEXT_VP8_ID, &cmdbuf->param_mem,
VssVp8encSetSequenceParametersCommand,
ctx->seq_param_offset,
sizeof(struct VssVp8encSequenceParameterBuffer));
ctx->vp8_seq_cmd_send = 1;
return vaStatus;
}
static VAStatus vsp_VP8_CreateContext(
object_context_p obj_context,
object_config_p obj_config)
{
VAStatus vaStatus = VA_STATUS_SUCCESS;
/* currently vp8 will use vpp's context since they will use the same cmdbuf */
context_VPP_p ctx;
int i;
ctx = (context_VPP_p) calloc(1, sizeof(struct context_VPP_s));
if (NULL == ctx) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
return vaStatus;
}
//set default VP8 sequence params
vsp_VP8_set_default_params(&ctx->vp8_seq_param);
ctx->temporal_layer_number = 1;
for (i = 0; i < obj_config->attrib_count; i++) {
if (obj_config->attrib_list[i].type == VAConfigAttribRateControl) {
if (obj_config->attrib_list[i].value == VA_RC_VCM)
ctx->vp8_seq_param.rc_end_usage = VP8_ENC_SMOOTH_CBR_HRD;
else
ctx->vp8_seq_param.rc_end_usage = VP8_ENC_CBR;
}
}
/* set size */
ctx->param_sz = 0;
ctx->pic_param_sz = ALIGN_TO_128(sizeof(struct VssVp8encPictureParameterBuffer));
ctx->param_sz += ctx->pic_param_sz;
ctx->seq_param_sz = ALIGN_TO_128(sizeof(struct VssVp8encSequenceParameterBuffer));
ctx->param_sz += ctx->seq_param_sz;
ctx->ref_param_sz = ALIGN_TO_128(sizeof(ref_frame_surface));
ctx->param_sz += ctx->ref_param_sz;
/* set offset */
ctx->pic_param_offset = 0;
ctx->seq_param_offset = ctx->pic_param_sz;
ctx->ref_param_offset = ctx->pic_param_sz + ctx->seq_param_sz;
ctx->context_buf = (psb_buffer_p) calloc(1, sizeof(struct psb_buffer_s));
if (NULL == ctx->context_buf) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
goto out;
}
vaStatus = psb_buffer_create(obj_context->driver_data, VSP_VP8ENC_STATE_SIZE, psb_bt_vpu_only, ctx->context_buf);
if (VA_STATUS_SUCCESS != vaStatus) {
goto out;
}
obj_context->format_data = (void*) ctx;
ctx->obj_context = obj_context;
return vaStatus;
out:
vsp_VP8_DestroyContext(obj_context);
if (ctx)
free(ctx);
return vaStatus;
}
static VAStatus vsp_vp8_process_seqence_param(
psb_driver_data_p driver_data,
context_VPP_p ctx,
object_buffer_p obj_buffer)
{
VAStatus vaStatus = VA_STATUS_SUCCESS;
vsp_cmdbuf_p cmdbuf = ctx->obj_context->vsp_cmdbuf;
int i;
int ref_frame_width, ref_frame_height;
VAEncSequenceParameterBufferVP8 *va_seq =
(VAEncSequenceParameterBufferVP8 *) obj_buffer->buffer_data;
struct VssVp8encSequenceParameterBuffer *seq =
(struct VssVp8encSequenceParameterBuffer *)cmdbuf->seq_param_p;
ref_frame_surface *ref =
(struct ref_frame_surface*)cmdbuf->ref_param_p;
ctx->frame_width = va_seq->frame_width;
ctx->frame_height = va_seq->frame_height;
/*cmd structures initializations*/
seq->frame_width = va_seq->frame_width;
seq->frame_height = va_seq->frame_height;
seq->rc_target_bitrate = va_seq->bits_per_second / 1000;
seq->max_intra_rate = 0;
seq->rc_undershoot_pct = ctx->rc_undershoot;
seq->rc_overshoot_pct = 100;
/* FIXME: API doc says max 5000, but for current default test vector we still use 6000 */
seq->rc_buf_sz = ctx->buffer_size;
seq->rc_buf_initial_sz = ctx->initial_buffer_fullness;
seq->rc_buf_optimal_sz = ctx->optimal_buffer_fullness;
seq->rc_min_quantizer = ctx->min_qp;
seq->rc_max_quantizer = ctx->max_qp;
seq->kf_max_dist = va_seq->kf_max_dist;
seq->kf_min_dist = va_seq->kf_min_dist;
seq->frame_rate = ctx->frame_rate;
seq->error_resilient = va_seq->error_resilient;
seq->num_token_partitions = 2; // (log2: 2^2 = 4)
seq->rc_end_usage = (ctx->rc_mode == VA_RC_CBR) ? VP8_ENC_CBR_HRD : VP8_ENC_CBR; /* CBR */
seq->kf_mode = va_seq->kf_auto; /* AUTO */
seq->cyclic_intra_refresh = 0;
seq->concatenate_partitions = 1; //Make 0 not to concatenate partitions
ref_frame_width = (ctx->frame_width + 2 * 32 + 63) & (~63);
ref_frame_height = (ctx->frame_height + 2 * 32 + 63) & (~63);
int chroma_height = (ref_frame_height / 2 + 63) & (~63);
int ref_size = ref_frame_width * (ref_frame_height + chroma_height);
seq->recon_buffer_mode = vss_vp8enc_seq_param_recon_buffer_mode_per_seq;
for (i = 0; i < 4; i++) {
seq->ref_frame_buffers[i].surface_id = va_seq->reference_frames[i];
seq->ref_frame_buffers[i].width = ref_frame_width;
seq->ref_frame_buffers[i].height = ref_frame_height;
}
for (i = 0; i < 4; i++) {
object_surface_p ref_surf = SURFACE(va_seq->reference_frames[i]);
if (!ref_surf)
return VA_STATUS_ERROR_UNKNOWN;
ref_surf->is_ref_surface = 2;
if (ref_surf->psb_surface->size < ref_size) {
/* re-alloc buffer */
ref_surf->psb_surface->size = ref_size;
psb_buffer_destroy(&ref_surf->psb_surface->buf);
vaStatus = psb_buffer_create(driver_data, ref_surf->psb_surface->size, psb_bt_surface, &ref_surf->psb_surface->buf);
if (VA_STATUS_SUCCESS != vaStatus)
return VA_STATUS_ERROR_ALLOCATION_FAILED;
}
vsp_cmdbuf_reloc_pic_param(&(seq->ref_frame_buffers[i].base),
0,
&(ref_surf->psb_surface->buf),
cmdbuf->param_mem_loc, seq);
}
ctx->frame_rate = seq->frame_rate;
ctx->bits_per_second = va_seq->bits_per_second;
ctx->vp8_seq_param = * seq;
vsp_cmdbuf_insert_command(cmdbuf, CONTEXT_VP8_ID, &cmdbuf->param_mem,
VssVp8encSetSequenceParametersCommand,
ctx->seq_param_offset,
sizeof(struct VssVp8encSequenceParameterBuffer));
ctx->vp8_seq_cmd_send = 1;
return vaStatus;
}
static VAStatus vsp_VP8_CreateContext(
object_context_p obj_context,
object_config_p obj_config)
{
VAStatus vaStatus = VA_STATUS_SUCCESS;
/* currently vp8 will use vpp's context since they will use the same cmdbuf */
context_VPP_p ctx;
int i;
ctx = (context_VPP_p) calloc(1, sizeof(struct context_VPP_s));
if (NULL == ctx) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
return vaStatus;
}
for (i = 0; i < obj_config->attrib_count; i++) {
if (obj_config->attrib_list[i].type == VAConfigAttribRateControl) {
ctx->rc_mode = obj_config->attrib_list[i].value;
break;
}
}
/* set size */
ctx->param_sz = 0;
ctx->pic_param_sz = ALIGN_TO_128(sizeof(struct VssVp8encPictureParameterBuffer));
ctx->param_sz += ctx->pic_param_sz;
ctx->seq_param_sz = ALIGN_TO_128(sizeof(struct VssVp8encSequenceParameterBuffer));
ctx->param_sz += ctx->seq_param_sz;
ctx->ref_param_sz = ALIGN_TO_128(sizeof(ref_frame_surface));
ctx->param_sz += ctx->ref_param_sz;
/* set offset */
ctx->pic_param_offset = 0;
ctx->seq_param_offset = ctx->pic_param_sz;
ctx->ref_param_offset = ctx->pic_param_sz + ctx->seq_param_sz;
ctx->min_qp = 4;
ctx->max_qp = 63;
ctx->rc_undershoot = 100;
ctx->buffer_size = 6000;
ctx->initial_buffer_fullness = 4000;
ctx->optimal_buffer_fullness = 5000;
ctx->context_buf = (psb_buffer_p) calloc(1, sizeof(struct psb_buffer_s));
if (NULL == ctx->context_buf) {
vaStatus = VA_STATUS_ERROR_ALLOCATION_FAILED;
DEBUG_FAILURE;
goto out;
}
vaStatus = psb_buffer_create(obj_context->driver_data, VSP_VP8ENC_STATE_SIZE, psb_bt_vpu_only, ctx->context_buf);
if (VA_STATUS_SUCCESS != vaStatus) {
goto out;
}
obj_context->format_data = (void*) ctx;
ctx->obj_context = obj_context;
return vaStatus;
out:
vsp_VP8_DestroyContext(obj_context);
if (ctx)
free(ctx);
return vaStatus;
}
