2017-12-12 15:12:07 +01:00
|
|
|
#include "decoder.h"
|
|
|
|
|
|
|
|
#include <libavformat/avformat.h>
|
|
|
|
|
|
|
|
#include "events.h"
|
2019-03-02 15:16:55 +01:00
|
|
|
#include "video_buffer.h"
|
2019-11-24 11:53:00 +01:00
|
|
|
#include "util/log.h"
|
2017-12-12 15:12:07 +01:00
|
|
|
|
2019-03-02 20:09:56 +01:00
|
|
|
void
|
2021-02-23 19:59:43 +01:00
|
|
|
decoder_init(struct decoder *decoder, struct video_buffer *vb) {
|
2019-03-02 16:43:43 +01:00
|
|
|
decoder->video_buffer = vb;
|
2018-02-15 22:56:32 +01:00
|
|
|
}
|
|
|
|
|
2019-03-02 23:52:22 +01:00
|
|
|
bool
|
2019-03-03 11:59:31 +01:00
|
|
|
decoder_open(struct decoder *decoder, const AVCodec *codec) {
|
2019-03-02 16:43:43 +01:00
|
|
|
decoder->codec_ctx = avcodec_alloc_context3(codec);
|
|
|
|
if (!decoder->codec_ctx) {
|
2018-02-13 10:10:18 +01:00
|
|
|
LOGC("Could not allocate decoder context");
|
2019-03-02 23:52:22 +01:00
|
|
|
return false;
|
2017-12-12 15:12:07 +01:00
|
|
|
}
|
|
|
|
|
2019-03-02 16:43:43 +01:00
|
|
|
if (avcodec_open2(decoder->codec_ctx, codec, NULL) < 0) {
|
|
|
|
LOGE("Could not open codec");
|
|
|
|
avcodec_free_context(&decoder->codec_ctx);
|
2019-03-02 23:52:22 +01:00
|
|
|
return false;
|
2017-12-12 15:12:07 +01:00
|
|
|
}
|
|
|
|
|
2019-03-02 23:52:22 +01:00
|
|
|
return true;
|
2019-03-02 16:43:43 +01:00
|
|
|
}
|
2018-10-11 07:12:36 +02:00
|
|
|
|
2019-03-02 20:09:56 +01:00
|
|
|
void
|
|
|
|
decoder_close(struct decoder *decoder) {
|
2019-03-02 16:43:43 +01:00
|
|
|
avcodec_close(decoder->codec_ctx);
|
|
|
|
avcodec_free_context(&decoder->codec_ctx);
|
|
|
|
}
|
2017-12-12 15:12:07 +01:00
|
|
|
|
2019-03-02 23:52:22 +01:00
|
|
|
bool
|
2019-03-03 11:59:31 +01:00
|
|
|
decoder_push(struct decoder *decoder, const AVPacket *packet) {
|
2019-03-02 16:43:43 +01:00
|
|
|
int ret;
|
|
|
|
if ((ret = avcodec_send_packet(decoder->codec_ctx, packet)) < 0) {
|
|
|
|
LOGE("Could not send video packet: %d", ret);
|
2019-03-02 23:52:22 +01:00
|
|
|
return false;
|
2017-12-12 15:12:07 +01:00
|
|
|
}
|
2019-03-02 16:43:43 +01:00
|
|
|
ret = avcodec_receive_frame(decoder->codec_ctx,
|
2021-02-19 21:16:57 +01:00
|
|
|
decoder->video_buffer->producer_frame);
|
2019-03-02 16:43:43 +01:00
|
|
|
if (!ret) {
|
|
|
|
// a frame was received
|
2021-02-23 19:59:43 +01:00
|
|
|
video_buffer_producer_offer_frame(decoder->video_buffer);
|
2019-03-02 16:43:43 +01:00
|
|
|
} else if (ret != AVERROR(EAGAIN)) {
|
|
|
|
LOGE("Could not receive video frame: %d", ret);
|
2019-03-02 23:52:22 +01:00
|
|
|
return false;
|
2018-10-11 07:12:36 +02:00
|
|
|
}
|
2019-03-02 23:52:22 +01:00
|
|
|
return true;
|
2017-12-12 15:12:07 +01:00
|
|
|
}
|