Commit dc971acf authored by Vittorio Giovara's avatar Vittorio Giovara

h264_parser: use enum values in h264_find_frame_end()

parent ba7dfe5c
...@@ -58,13 +58,15 @@ static int h264_find_frame_end(H264Context *h, const uint8_t *buf, ...@@ -58,13 +58,15 @@ static int h264_find_frame_end(H264Context *h, const uint8_t *buf,
else else
state >>= 1; // 2->1, 1->0, 0->0 state >>= 1; // 2->1, 1->0, 0->0
} else if (state <= 5) { } else if (state <= 5) {
int v = buf[i] & 0x1F; int nalu_type = buf[i] & 0x1F;
if (v == 6 || v == 7 || v == 8 || v == 9) { if (nalu_type == NAL_SEI || nalu_type == NAL_SPS ||
nalu_type == NAL_PPS || nalu_type == NAL_AUD) {
if (pc->frame_start_found) { if (pc->frame_start_found) {
i++; i++;
goto found; goto found;
} }
} else if (v == 1 || v == 2 || v == 5) { } else if (nalu_type == NAL_SLICE || nalu_type == NAL_DPA ||
nalu_type == NAL_IDR_SLICE) {
if (pc->frame_start_found) { if (pc->frame_start_found) {
state += 8; state += 8;
continue; continue;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment