examples/extract_mvs: Do not use stack packet

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2021-09-03 23:04:50 +02:00
parent 0230d125cb
commit f495604361
1 changed files with 13 additions and 5 deletions

View File

@ -124,7 +124,7 @@ static int open_codec_context(AVFormatContext *fmt_ctx, enum AVMediaType type)
int main(int argc, char **argv) int main(int argc, char **argv)
{ {
int ret = 0; int ret = 0;
AVPacket pkt = { 0 }; AVPacket *pkt = NULL;
if (argc != 2) { if (argc != 2) {
fprintf(stderr, "Usage: %s <video>\n", argv[0]); fprintf(stderr, "Usage: %s <video>\n", argv[0]);
@ -159,13 +159,20 @@ int main(int argc, char **argv)
goto end; goto end;
} }
pkt = av_packet_alloc();
if (!pkt) {
fprintf(stderr, "Could not allocate AVPacket\n");
ret = AVERROR(ENOMEM);
goto end;
}
printf("framenum,source,blockw,blockh,srcx,srcy,dstx,dsty,flags\n"); printf("framenum,source,blockw,blockh,srcx,srcy,dstx,dsty,flags\n");
/* read frames from the file */ /* read frames from the file */
while (av_read_frame(fmt_ctx, &pkt) >= 0) { while (av_read_frame(fmt_ctx, pkt) >= 0) {
if (pkt.stream_index == video_stream_idx) if (pkt->stream_index == video_stream_idx)
ret = decode_packet(&pkt); ret = decode_packet(pkt);
av_packet_unref(&pkt); av_packet_unref(pkt);
if (ret < 0) if (ret < 0)
break; break;
} }
@ -177,5 +184,6 @@ end:
avcodec_free_context(&video_dec_ctx); avcodec_free_context(&video_dec_ctx);
avformat_close_input(&fmt_ctx); avformat_close_input(&fmt_ctx);
av_frame_free(&frame); av_frame_free(&frame);
av_packet_free(&pkt);
return ret < 0; return ret < 0;
} }