Commit 67d28f4a authored by Anton Khirnov's avatar Anton Khirnov

examples/output: switch to the new encoding API

parent 7bf8db4d
...@@ -233,25 +233,37 @@ static AVFrame *get_audio_frame(OutputStream *ost) ...@@ -233,25 +233,37 @@ static AVFrame *get_audio_frame(OutputStream *ost)
static int encode_audio_frame(AVFormatContext *oc, OutputStream *ost, static int encode_audio_frame(AVFormatContext *oc, OutputStream *ost,
AVFrame *frame) AVFrame *frame)
{ {
AVPacket pkt = { 0 }; // data and size must be 0; int ret;
int got_packet;
av_init_packet(&pkt); ret = avcodec_send_frame(ost->enc, frame);
avcodec_encode_audio2(ost->enc, &pkt, frame, &got_packet); if (ret < 0) {
fprintf(stderr, "Error submitting a frame for encoding\n");
exit(1);
}
if (got_packet) { while (ret >= 0) {
pkt.stream_index = ost->st->index; 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. */ ret = avcodec_receive_packet(ost->enc, &pkt);
if (av_interleaved_write_frame(oc, &pkt) != 0) { if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
fprintf(stderr, "Error while writing audio frame\n"); fprintf(stderr, "Error encoding a video frame\n");
exit(1); 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) ...@@ -517,36 +529,41 @@ static int write_video_frame(AVFormatContext *oc, OutputStream *ost)
int ret; int ret;
AVCodecContext *c; AVCodecContext *c;
AVFrame *frame; AVFrame *frame;
AVPacket pkt = { 0 };
int got_packet = 0;
c = ost->enc; c = ost->enc;
frame = get_video_frame(ost); frame = get_video_frame(ost);
av_init_packet(&pkt);
/* encode the image */ /* encode the image */
ret = avcodec_encode_video2(c, &pkt, frame, &got_packet); ret = avcodec_send_frame(c, frame);
if (ret < 0) { if (ret < 0) {
fprintf(stderr, "Error encoding a video frame\n"); fprintf(stderr, "Error submitting a frame for encoding\n");
exit(1); exit(1);
} }
if (got_packet) { while (ret >= 0) {
av_packet_rescale_ts(&pkt, c->time_base, ost->st->time_base); AVPacket pkt = { 0 };
pkt.stream_index = ost->st->index;
/* Write the compressed frame to the media file. */ av_init_packet(&pkt);
ret = av_interleaved_write_frame(oc, &pkt);
}
if (ret != 0) { ret = avcodec_receive_packet(c, &pkt);
fprintf(stderr, "Error while writing video frame\n"); if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF) {
exit(1); 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) static void close_stream(AVFormatContext *oc, OutputStream *ost)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment