static GstVaapiDecoderStatus
decode_scan (GstVaapiDecoderJpeg * decoder, GstJpegSegment * seg)
{
GstVaapiDecoderJpegPrivate *const priv = &decoder->priv;
GstVaapiPicture *const picture = priv->current_picture;
GstVaapiSlice *slice;
VASliceParameterBufferJPEGBaseline *slice_param;
GstJpegScanHdr scan_hdr;
guint scan_hdr_size, scan_data_size;
guint i, h_max, v_max, mcu_width, mcu_height;
if (!VALID_STATE (decoder, GOT_SOF))
return GST_VAAPI_DECODER_STATUS_SUCCESS;
scan_hdr_size = (seg->data[seg->offset] << 8) | seg->data[seg->offset + 1];
scan_data_size = seg->size - scan_hdr_size;
memset (&scan_hdr, 0, sizeof (scan_hdr));
if (!gst_jpeg_segment_parse_scan_header (seg, &scan_hdr)) {
GST_ERROR ("failed to parse scan header");
return GST_VAAPI_DECODER_STATUS_ERROR_BITSTREAM_PARSER;
}
slice = GST_VAAPI_SLICE_NEW (JPEGBaseline, decoder,
seg->data + seg->offset + scan_hdr_size, scan_data_size);
if (!slice) {
GST_ERROR ("failed to allocate slice");
return GST_VAAPI_DECODER_STATUS_ERROR_ALLOCATION_FAILED;
}
gst_vaapi_picture_add_slice (picture, slice);
if (!VALID_STATE (decoder, GOT_HUF_TABLE))
gst_jpeg_get_default_huffman_tables (&priv->huf_tables);
// Update VA Huffman table if it changed for this scan
if (huffman_tables_updated (&priv->huf_tables)) {
slice->huf_table = GST_VAAPI_HUFFMAN_TABLE_NEW (JPEGBaseline, decoder);
if (!slice->huf_table) {
GST_ERROR ("failed to allocate Huffman tables");
huffman_tables_reset (&priv->huf_tables);
return GST_VAAPI_DECODER_STATUS_ERROR_ALLOCATION_FAILED;
}
fill_huffman_table (slice->huf_table, &priv->huf_tables);
huffman_tables_reset (&priv->huf_tables);
}
slice_param = slice->param;
slice_param->num_components = scan_hdr.num_components;
for (i = 0; i < scan_hdr.num_components; i++) {
slice_param->components[i].component_selector =
scan_hdr.components[i].component_selector;
slice_param->components[i].dc_table_selector =
scan_hdr.components[i].dc_selector;
slice_param->components[i].ac_table_selector =
scan_hdr.components[i].ac_selector;
}
slice_param->restart_interval = priv->mcu_restart;
slice_param->slice_horizontal_position = 0;
slice_param->slice_vertical_position = 0;
get_max_sampling_factors (&priv->frame_hdr, &h_max, &v_max);
mcu_width = 8 * h_max;
mcu_height = 8 * v_max;
if (scan_hdr.num_components == 1) { // Non-interleaved
const guint Csj = slice_param->components[0].component_selector;
const GstJpegFrameComponent *const fcp =
get_component (&priv->frame_hdr, Csj);
if (!fcp || fcp->horizontal_factor == 0 || fcp->vertical_factor == 0) {
GST_ERROR ("failed to validate image component %u", Csj);
return GST_VAAPI_DECODER_STATUS_ERROR_INVALID_PARAMETER;
}
mcu_width /= fcp->horizontal_factor;
mcu_height /= fcp->vertical_factor;
}
slice_param->num_mcus =
((priv->frame_hdr.width + mcu_width - 1) / mcu_width) *
((priv->frame_hdr.height + mcu_height - 1) / mcu_height);
priv->decoder_state |= GST_JPEG_VIDEO_STATE_GOT_SOS;
return GST_VAAPI_DECODER_STATUS_SUCCESS;
}
static gboolean
parse_jpeg_segment (GstJpegSegment * segment)
{
switch (segment->marker) {
case GST_JPEG_MARKER_SOF0:
case GST_JPEG_MARKER_SOF1:
case GST_JPEG_MARKER_SOF2:
case GST_JPEG_MARKER_SOF3:
case GST_JPEG_MARKER_SOF9:
case GST_JPEG_MARKER_SOF10:
case GST_JPEG_MARKER_SOF11:{
GstJpegFrameHdr hdr;
int i;
if (!gst_jpeg_segment_parse_frame_header (segment, &hdr)) {
g_printerr ("Failed to parse frame header!\n");
return FALSE;
}
g_print ("\t\twidth x height = %u x %u\n", hdr.width, hdr.height);
g_print ("\t\tsample precision = %u\n", hdr.sample_precision);
g_print ("\t\tnum components = %u\n", hdr.num_components);
for (i = 0; i < hdr.num_components; ++i) {
g_print ("\t\t%d: id=%d, h=%d, v=%d, qts=%d\n", i,
hdr.components[i].identifier, hdr.components[i].horizontal_factor,
hdr.components[i].vertical_factor,
hdr.components[i].quant_table_selector);
}
break;
}
case GST_JPEG_MARKER_DHT:{
GstJpegHuffmanTables ht;
if (!gst_jpeg_segment_parse_huffman_table (segment, &ht)) {
g_printerr ("Failed to parse huffman table!\n");
return FALSE;
}
break;
}
case GST_JPEG_MARKER_DQT:{
GstJpegQuantTables qt;
if (!gst_jpeg_segment_parse_quantization_table (segment, &qt)) {
g_printerr ("Failed to parse quantization table!\n");
return FALSE;
}
break;
}
case GST_JPEG_MARKER_SOS:{
GstJpegScanHdr hdr;
int i;
if (!gst_jpeg_segment_parse_scan_header (segment, &hdr)) {
g_printerr ("Failed to parse scan header!\n");
return FALSE;
}
g_print ("\t\tnum components = %u\n", hdr.num_components);
for (i = 0; i < hdr.num_components; ++i) {
g_print ("\t\t %d: cs=%d, dcs=%d, acs=%d\n", i,
hdr.components[i].component_selector,
hdr.components[i].dc_selector, hdr.components[i].ac_selector);
}
}
case GST_JPEG_MARKER_COM:
/* gst_util_dump_mem (segment->data + segment->offset, segment->size); */
break;
default:
if (segment->marker >= GST_JPEG_MARKER_APP_MIN
&& segment->marker <= GST_JPEG_MARKER_APP_MAX) {
guint n = segment->marker - GST_JPEG_MARKER_APP_MIN;
if (app_segments[n] == NULL)
app_segments[n] = gst_buffer_new ();
gst_buffer_append_memory (app_segments[n],
gst_memory_new_wrapped (GST_MEMORY_FLAG_READONLY,
(guint8 *) segment->data + segment->offset,
segment->size, 0, segment->size, NULL, NULL));
}
break;
}
return TRUE;
}
