|
|
@@ -233,25 +233,37 @@ static AVFrame *get_audio_frame(OutputStream *ost) |
|
|
|
static int encode_audio_frame(AVFormatContext *oc, OutputStream *ost, |
|
|
|
AVFrame *frame) |
|
|
|
{ |
|
|
|
AVPacket pkt = { 0 }; // data and size must be 0; |
|
|
|
int got_packet; |
|
|
|
int ret; |
|
|
|
|
|
|
|
av_init_packet(&pkt); |
|
|
|
avcodec_encode_audio2(ost->enc, &pkt, frame, &got_packet); |
|
|
|
ret = avcodec_send_frame(ost->enc, frame); |
|
|
|
if (ret < 0) { |
|
|
|
fprintf(stderr, "Error submitting a frame for encoding\n"); |
|
|
|
exit(1); |
|
|
|
} |
|
|
|
|
|
|
|
if (got_packet) { |
|
|
|
pkt.stream_index = ost->st->index; |
|
|
|
while (ret >= 0) { |
|
|
|
AVPacket pkt = { 0 }; // data and size must be 0; |
|
|
|
|
|
|
|
av_packet_rescale_ts(&pkt, ost->enc->time_base, ost->st->time_base); |
|
|
|
av_init_packet(&pkt); |
|
|
|
|
|
|
|
/* Write the compressed frame to the media file. */ |
|
|
|
if (av_interleaved_write_frame(oc, &pkt) != 0) { |
|
|
|
fprintf(stderr, "Error while writing audio frame\n"); |
|
|
|
ret = avcodec_receive_packet(ost->enc, &pkt); |
|
|
|
if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) { |
|
|
|
fprintf(stderr, "Error encoding a video frame\n"); |
|
|
|
exit(1); |
|
|
|
} else if (ret >= 0) { |
|
|
|
av_packet_rescale_ts(&pkt, ost->enc->time_base, ost->st->time_base); |
|
|
|
pkt.stream_index = ost->st->index; |
|
|
|
|
|
|
|
/* Write the compressed frame to the media file. */ |
|
|
|
ret = av_interleaved_write_frame(oc, &pkt); |
|
|
|
if (ret < 0) { |
|
|
|
fprintf(stderr, "Error while writing video frame\n"); |
|
|
|
exit(1); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
return (frame || got_packet) ? 0 : 1; |
|
|
|
return ret == AVERROR_EOF; |
|
|
|
} |
|
|
|
|
|
|
|
/* |
|
|
@@ -517,36 +529,41 @@ static int write_video_frame(AVFormatContext *oc, OutputStream *ost) |
|
|
|
int ret; |
|
|
|
AVCodecContext *c; |
|
|
|
AVFrame *frame; |
|
|
|
AVPacket pkt = { 0 }; |
|
|
|
int got_packet = 0; |
|
|
|
|
|
|
|
c = ost->enc; |
|
|
|
|
|
|
|
frame = get_video_frame(ost); |
|
|
|
|
|
|
|
av_init_packet(&pkt); |
|
|
|
|
|
|
|
/* encode the image */ |
|
|
|
ret = avcodec_encode_video2(c, &pkt, frame, &got_packet); |
|
|
|
ret = avcodec_send_frame(c, frame); |
|
|
|
if (ret < 0) { |
|
|
|
fprintf(stderr, "Error encoding a video frame\n"); |
|
|
|
fprintf(stderr, "Error submitting a frame for encoding\n"); |
|
|
|
exit(1); |
|
|
|
} |
|
|
|
|
|
|
|
if (got_packet) { |
|
|
|
av_packet_rescale_ts(&pkt, c->time_base, ost->st->time_base); |
|
|
|
pkt.stream_index = ost->st->index; |
|
|
|
while (ret >= 0) { |
|
|
|
AVPacket pkt = { 0 }; |
|
|
|
|
|
|
|
/* Write the compressed frame to the media file. */ |
|
|
|
ret = av_interleaved_write_frame(oc, &pkt); |
|
|
|
} |
|
|
|
av_init_packet(&pkt); |
|
|
|
|
|
|
|
if (ret != 0) { |
|
|
|
fprintf(stderr, "Error while writing video frame\n"); |
|
|
|
exit(1); |
|
|
|
ret = avcodec_receive_packet(c, &pkt); |
|
|
|
if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) { |
|
|
|
fprintf(stderr, "Error encoding a video frame\n"); |
|
|
|
exit(1); |
|
|
|
} else if (ret >= 0) { |
|
|
|
av_packet_rescale_ts(&pkt, c->time_base, ost->st->time_base); |
|
|
|
pkt.stream_index = ost->st->index; |
|
|
|
|
|
|
|
/* Write the compressed frame to the media file. */ |
|
|
|
ret = av_interleaved_write_frame(oc, &pkt); |
|
|
|
if (ret < 0) { |
|
|
|
fprintf(stderr, "Error while writing video frame\n"); |
|
|
|
exit(1); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
return (frame || got_packet) ? 0 : 1; |
|
|
|
return ret == AVERROR_EOF; |
|
|
|
} |
|
|
|
|
|
|
|
static void close_stream(AVFormatContext *oc, OutputStream *ost) |
|
|
|