static int encode_write(AVCodecContext *avctx, AVFrame *frame, FILE *fout)
{
int ret = 0;
- AVPacket enc_pkt;
+ AVPacket *enc_pkt;
- av_init_packet(&enc_pkt);
- enc_pkt.data = NULL;
- enc_pkt.size = 0;
+ if (!(enc_pkt = av_packet_alloc()))
+ return AVERROR(ENOMEM);
if ((ret = avcodec_send_frame(avctx, frame)) < 0) {
fprintf(stderr, "Error code: %s\n", av_err2str(ret));
goto end;
}
while (1) {
- ret = avcodec_receive_packet(avctx, &enc_pkt);
+ ret = avcodec_receive_packet(avctx, enc_pkt);
if (ret)
break;
- enc_pkt.stream_index = 0;
- ret = fwrite(enc_pkt.data, enc_pkt.size, 1, fout);
- av_packet_unref(&enc_pkt);
+ enc_pkt->stream_index = 0;
+ ret = fwrite(enc_pkt->data, enc_pkt->size, 1, fout);
+ av_packet_unref(enc_pkt);
}
end:
+ av_packet_free(&enc_pkt);
ret = ((ret == AVERROR(EAGAIN)) ? 0 : -1);
return ret;
}
FILE *fin = NULL, *fout = NULL;
AVFrame *sw_frame = NULL, *hw_frame = NULL;
AVCodecContext *avctx = NULL;
- AVCodec *codec = NULL;
+ const AVCodec *codec = NULL;
const char *enc_name = "h264_vaapi";
if (argc < 5) {
sw_frame->width = width;
sw_frame->height = height;
sw_frame->format = AV_PIX_FMT_NV12;
- if ((err = av_frame_get_buffer(sw_frame, 32)) < 0)
+ if ((err = av_frame_get_buffer(sw_frame, 0)) < 0)
goto close;
if ((err = fread((uint8_t*)(sw_frame->data[0]), size, 1, fin)) <= 0)
break;