avformat/evcdec: simplify probe function

Signed-off-by: James Almer <jamrial@gmail.com>
pull/389/head
James Almer 1 year ago
parent 43991dbd05
commit ca8a660176
  1. 29
      libavformat/evcdec.c

@ -35,14 +35,6 @@
#define RAW_PACKET_SIZE 1024
typedef struct EVCParserContext {
int got_sps;
int got_pps;
int got_idr;
int got_nonidr;
} EVCParserContext;
typedef struct EVCDemuxContext {
const AVClass *class;
AVRational framerate;
@ -103,10 +95,11 @@ static uint32_t read_nal_unit_length(const uint8_t *bits, int bits_size)
return nalu_len;
}
static int parse_nal_units(const AVProbeData *p, EVCParserContext *ev)
static int annexb_probe(const AVProbeData *p)
{
int nalu_type;
size_t nalu_size;
int got_sps = 0, got_pps = 0, got_idr = 0, got_nonidr = 0;
unsigned char *bits = (unsigned char *)p->buf;
int bytes_to_read = p->buf_size;
@ -123,27 +116,19 @@ static int parse_nal_units(const AVProbeData *p, EVCParserContext *ev)
nalu_type = get_nalu_type(bits, bytes_to_read);
if (nalu_type == EVC_SPS_NUT)
ev->got_sps++;
got_sps++;
else if (nalu_type == EVC_PPS_NUT)
ev->got_pps++;
got_pps++;
else if (nalu_type == EVC_IDR_NUT )
ev->got_idr++;
got_idr++;
else if (nalu_type == EVC_NOIDR_NUT)
ev->got_nonidr++;
got_nonidr++;
bits += nalu_size;
bytes_to_read -= nalu_size;
}
return 0;
}
static int annexb_probe(const AVProbeData *p)
{
EVCParserContext ev = {0};
int ret = parse_nal_units(p, &ev);
if (ret == 0 && ev.got_sps && ev.got_pps && (ev.got_idr || ev.got_nonidr > 3))
if (got_sps && got_pps && (got_idr || got_nonidr > 3))
return AVPROBE_SCORE_EXTENSION + 1; // 1 more than .mpg
return 0;

Loading…
Cancel
Save