mirror of https://git.ffmpeg.org/ffmpeg.git
avcodec/hevc_parser: move hevc_find_frame_end() down in the file
Reduces differences with libav.
This commit is contained in:
parent
35f3df0d76
commit
214f4133c4
|
@ -49,49 +49,6 @@ typedef struct HEVCParserContext {
|
||||||
int pocTid0;
|
int pocTid0;
|
||||||
} HEVCParserContext;
|
} HEVCParserContext;
|
||||||
|
|
||||||
/**
|
|
||||||
* Find the end of the current frame in the bitstream.
|
|
||||||
* @return the position of the first byte of the next frame, or END_NOT_FOUND
|
|
||||||
*/
|
|
||||||
static int hevc_find_frame_end(AVCodecParserContext *s, const uint8_t *buf,
|
|
||||||
int buf_size)
|
|
||||||
{
|
|
||||||
int i;
|
|
||||||
ParseContext *pc = s->priv_data;
|
|
||||||
|
|
||||||
for (i = 0; i < buf_size; i++) {
|
|
||||||
int nut;
|
|
||||||
|
|
||||||
pc->state64 = (pc->state64 << 8) | buf[i];
|
|
||||||
|
|
||||||
if (((pc->state64 >> 3 * 8) & 0xFFFFFF) != START_CODE)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
nut = (pc->state64 >> 2 * 8 + 1) & 0x3F;
|
|
||||||
// Beginning of access unit
|
|
||||||
if ((nut >= HEVC_NAL_VPS && nut <= HEVC_NAL_AUD) || nut == HEVC_NAL_SEI_PREFIX ||
|
|
||||||
(nut >= 41 && nut <= 44) || (nut >= 48 && nut <= 55)) {
|
|
||||||
if (pc->frame_start_found) {
|
|
||||||
pc->frame_start_found = 0;
|
|
||||||
return i - 5;
|
|
||||||
}
|
|
||||||
} else if (nut <= HEVC_NAL_RASL_R ||
|
|
||||||
(nut >= HEVC_NAL_BLA_W_LP && nut <= HEVC_NAL_CRA_NUT)) {
|
|
||||||
int first_slice_segment_in_pic_flag = buf[i] >> 7;
|
|
||||||
if (first_slice_segment_in_pic_flag) {
|
|
||||||
if (!pc->frame_start_found) {
|
|
||||||
pc->frame_start_found = 1;
|
|
||||||
} else { // First slice of next frame found
|
|
||||||
pc->frame_start_found = 0;
|
|
||||||
return i - 5;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return END_NOT_FOUND;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int hevc_parse_slice_header(AVCodecParserContext *s, H2645NAL *nal,
|
static int hevc_parse_slice_header(AVCodecParserContext *s, H2645NAL *nal,
|
||||||
AVCodecContext *avctx)
|
AVCodecContext *avctx)
|
||||||
{
|
{
|
||||||
|
@ -289,6 +246,50 @@ static inline int parse_nal_units(AVCodecParserContext *s, const uint8_t *buf,
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Find the end of the current frame in the bitstream.
|
||||||
|
* @return the position of the first byte of the next frame, or END_NOT_FOUND
|
||||||
|
*/
|
||||||
|
static int hevc_find_frame_end(AVCodecParserContext *s, const uint8_t *buf,
|
||||||
|
int buf_size)
|
||||||
|
{
|
||||||
|
HEVCParserContext *ctx = s->priv_data;
|
||||||
|
ParseContext *pc = &ctx->pc;
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < buf_size; i++) {
|
||||||
|
int nut;
|
||||||
|
|
||||||
|
pc->state64 = (pc->state64 << 8) | buf[i];
|
||||||
|
|
||||||
|
if (((pc->state64 >> 3 * 8) & 0xFFFFFF) != START_CODE)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
nut = (pc->state64 >> 2 * 8 + 1) & 0x3F;
|
||||||
|
// Beginning of access unit
|
||||||
|
if ((nut >= HEVC_NAL_VPS && nut <= HEVC_NAL_AUD) || nut == HEVC_NAL_SEI_PREFIX ||
|
||||||
|
(nut >= 41 && nut <= 44) || (nut >= 48 && nut <= 55)) {
|
||||||
|
if (pc->frame_start_found) {
|
||||||
|
pc->frame_start_found = 0;
|
||||||
|
return i - 5;
|
||||||
|
}
|
||||||
|
} else if (nut <= HEVC_NAL_RASL_R ||
|
||||||
|
(nut >= HEVC_NAL_BLA_W_LP && nut <= HEVC_NAL_CRA_NUT)) {
|
||||||
|
int first_slice_segment_in_pic_flag = buf[i] >> 7;
|
||||||
|
if (first_slice_segment_in_pic_flag) {
|
||||||
|
if (!pc->frame_start_found) {
|
||||||
|
pc->frame_start_found = 1;
|
||||||
|
} else { // First slice of next frame found
|
||||||
|
pc->frame_start_found = 0;
|
||||||
|
return i - 5;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return END_NOT_FOUND;
|
||||||
|
}
|
||||||
|
|
||||||
static int hevc_parse(AVCodecParserContext *s,
|
static int hevc_parse(AVCodecParserContext *s,
|
||||||
AVCodecContext *avctx,
|
AVCodecContext *avctx,
|
||||||
const uint8_t **poutbuf, int *poutbuf_size,
|
const uint8_t **poutbuf, int *poutbuf_size,
|
||||||
|
|
Loading…
Reference in New Issue