hevc: Use switch instead of if-nests in decode_nal_sei_message

Makes simpler to add support for more SEI types.

Signed-off-by: Luca Barbato <lu_zero@gentoo.org>
This commit is contained in:
Luca Barbato 2015-07-25 15:26:29 +02:00
parent 2cd841c077
commit 043f46f574
1 changed files with 21 additions and 13 deletions

View File

@ -53,7 +53,7 @@ enum HEVC_SEI_TYPE {
SEI_TYPE_CONTENT_LIGHT_LEVEL_INFO = 144,
};
static void decode_nal_sei_decoded_picture_hash(HEVCContext *s)
static int decode_nal_sei_decoded_picture_hash(HEVCContext *s)
{
int cIdx, i;
GetBitContext *gb = &s->HEVClc.gb;
@ -72,9 +72,10 @@ static void decode_nal_sei_decoded_picture_hash(HEVCContext *s)
skip_bits(gb, 32);
}
}
return 0;
}
static void decode_nal_sei_frame_packing_arrangement(HEVCContext *s)
static int decode_nal_sei_frame_packing_arrangement(HEVCContext *s)
{
GetBitContext *gb = &s->HEVClc.gb;
@ -97,9 +98,10 @@ static void decode_nal_sei_frame_packing_arrangement(HEVCContext *s)
skip_bits1(gb); // frame_packing_arrangement_persistance_flag
}
skip_bits1(gb); // upsampled_aspect_ratio_flag
return 0;
}
static void decode_nal_sei_display_orientation(HEVCContext *s)
static int decode_nal_sei_display_orientation(HEVCContext *s)
{
GetBitContext *gb = &s->HEVClc.gb;
@ -112,6 +114,8 @@ static void decode_nal_sei_display_orientation(HEVCContext *s)
s->sei_anticlockwise_rotation = get_bits(gb, 16);
skip_bits1(gb); // display_orientation_persistence_flag
}
return 0;
}
static int decode_nal_sei_message(HEVCContext *s)
@ -133,22 +137,26 @@ static int decode_nal_sei_message(HEVCContext *s)
payload_size += byte;
}
if (s->nal_unit_type == NAL_SEI_PREFIX) {
if (payload_type == 256) // Mismatched value from HM 8.1
decode_nal_sei_decoded_picture_hash(s);
else if (payload_type == SEI_TYPE_FRAME_PACKING)
decode_nal_sei_frame_packing_arrangement(s);
else if (payload_type == SEI_TYPE_DISPLAY_ORIENTATION)
decode_nal_sei_display_orientation(s);
else {
switch (payload_type) {
case 256: // Mismatched value from HM 8.1
return decode_nal_sei_decoded_picture_hash(s);
case SEI_TYPE_FRAME_PACKING:
return decode_nal_sei_frame_packing_arrangement(s);
case SEI_TYPE_DISPLAY_ORIENTATION:
return decode_nal_sei_display_orientation(s);
default:
av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
skip_bits(gb, 8 * payload_size);
return 0;
}
} else { /* nal_unit_type == NAL_SEI_SUFFIX */
if (payload_type == SEI_TYPE_DECODED_PICTURE_HASH)
decode_nal_sei_decoded_picture_hash(s);
else {
switch (payload_type) {
case SEI_TYPE_DECODED_PICTURE_HASH:
return decode_nal_sei_decoded_picture_hash(s);
default:
av_log(s->avctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", payload_type);
skip_bits(gb, 8 * payload_size);
return 0;
}
}
return 0;