avfilter/formats: Avoid using non public AV_SAMPLE_FMT_NB

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2014-05-27 18:22:09 +02:00
parent 0d26264fb4
commit 1853e2cba9
1 changed files with 2 additions and 2 deletions

View File

@ -376,7 +376,7 @@ AVFilterFormats *ff_planar_sample_fmts(void)
AVFilterFormats *ret = NULL; AVFilterFormats *ret = NULL;
int fmt; int fmt;
for (fmt = 0; fmt < AV_SAMPLE_FMT_NB; fmt++) for (fmt = 0; av_get_bytes_per_sample(fmt)>0; fmt++)
if (av_sample_fmt_is_planar(fmt)) if (av_sample_fmt_is_planar(fmt))
ff_add_format(&ret, fmt); ff_add_format(&ret, fmt);
@ -587,7 +587,7 @@ int ff_parse_sample_format(int *ret, const char *arg, void *log_ctx)
int sfmt = av_get_sample_fmt(arg); int sfmt = av_get_sample_fmt(arg);
if (sfmt == AV_SAMPLE_FMT_NONE) { if (sfmt == AV_SAMPLE_FMT_NONE) {
sfmt = strtol(arg, &tail, 0); sfmt = strtol(arg, &tail, 0);
if (*tail || (unsigned)sfmt >= AV_SAMPLE_FMT_NB) { if (*tail || av_get_bytes_per_sample(sfmt)<=0) {
av_log(log_ctx, AV_LOG_ERROR, "Invalid sample format '%s'\n", arg); av_log(log_ctx, AV_LOG_ERROR, "Invalid sample format '%s'\n", arg);
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }