|
|
|
@@ -74,27 +74,27 @@ static int set_hwframe_ctx(AVCodecContext *ctx, AVBufferRef *hw_device_ctx) |
|
|
|
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; |
|
|
|
} |
|
|
|
|