* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
+#include <stdint.h>
+
+#include "libavutil/dict.h"
#include "libavutil/imgutils.h"
+#include "libavutil/log.h"
+#include "libavutil/pixfmt.h"
+
+#include "libavformat/avformat.h"
+
+#include "libavcodec/avcodec.h"
+
#include "lavfutils.h"
int ff_load_image(uint8_t *data[4], int linesize[4],
- int *w, int *h, enum PixelFormat *pix_fmt,
+ int *w, int *h, enum AVPixelFormat *pix_fmt,
const char *filename, void *log_ctx)
{
- AVInputFormat *iformat = NULL;
+ const AVInputFormat *iformat = NULL;
AVFormatContext *format_ctx = NULL;
- AVCodec *codec;
- AVCodecContext *codec_ctx;
- AVFrame *frame;
- int frame_decoded, ret = 0;
+ const AVCodec *codec;
+ AVCodecContext *codec_ctx = NULL;
+ AVCodecParameters *par;
+ AVFrame *frame = NULL;
+ int ret = 0;
AVPacket pkt;
+ AVDictionary *opt=NULL;
- av_register_all();
-
- iformat = av_find_input_format("image2");
+ iformat = av_find_input_format("image2pipe");
if ((ret = avformat_open_input(&format_ctx, filename, iformat, NULL)) < 0) {
av_log(log_ctx, AV_LOG_ERROR,
"Failed to open input file '%s'\n", filename);
return ret;
}
- codec_ctx = format_ctx->streams[0]->codec;
- codec = avcodec_find_decoder(codec_ctx->codec_id);
+ if ((ret = avformat_find_stream_info(format_ctx, NULL)) < 0) {
+ av_log(log_ctx, AV_LOG_ERROR, "Find stream info failed\n");
+ goto end;
+ }
+
+ par = format_ctx->streams[0]->codecpar;
+ codec = avcodec_find_decoder(par->codec_id);
if (!codec) {
av_log(log_ctx, AV_LOG_ERROR, "Failed to find codec\n");
ret = AVERROR(EINVAL);
goto end;
}
- if ((ret = avcodec_open2(codec_ctx, codec, NULL)) < 0) {
+ codec_ctx = avcodec_alloc_context3(codec);
+ if (!codec_ctx) {
+ av_log(log_ctx, AV_LOG_ERROR, "Failed to alloc video decoder context\n");
+ ret = AVERROR(ENOMEM);
+ goto end;
+ }
+
+ ret = avcodec_parameters_to_context(codec_ctx, par);
+ if (ret < 0) {
+ av_log(log_ctx, AV_LOG_ERROR, "Failed to copy codec parameters to decoder context\n");
+ goto end;
+ }
+
+ av_dict_set(&opt, "thread_type", "slice", 0);
+ if ((ret = avcodec_open2(codec_ctx, codec, &opt)) < 0) {
av_log(log_ctx, AV_LOG_ERROR, "Failed to open codec\n");
goto end;
}
- if (!(frame = avcodec_alloc_frame()) ) {
+ if (!(frame = av_frame_alloc()) ) {
av_log(log_ctx, AV_LOG_ERROR, "Failed to alloc frame\n");
ret = AVERROR(ENOMEM);
goto end;
goto end;
}
- ret = avcodec_decode_video2(codec_ctx, frame, &frame_decoded, &pkt);
- if (ret < 0 || !frame_decoded) {
+ ret = avcodec_send_packet(codec_ctx, &pkt);
+ av_packet_unref(&pkt);
+ if (ret < 0) {
+ av_log(log_ctx, AV_LOG_ERROR, "Error submitting a packet to decoder\n");
+ goto end;
+ }
+
+ ret = avcodec_receive_frame(codec_ctx, frame);
+ if (ret < 0) {
av_log(log_ctx, AV_LOG_ERROR, "Failed to decode image from file\n");
goto end;
}
- ret = 0;
*w = frame->width;
*h = frame->height;
goto end;
ret = 0;
- av_image_copy(data, linesize, (const uint8_t **)frame->data, frame->linesize, *pix_fmt, *w, *h);
+ av_image_copy2(data, linesize, frame->data, frame->linesize,
+ *pix_fmt, *w, *h);
end:
- if (codec_ctx)
- avcodec_close(codec_ctx);
- if (format_ctx)
- avformat_close_input(&format_ctx);
- av_freep(&frame);
+ avcodec_free_context(&codec_ctx);
+ avformat_close_input(&format_ctx);
+ av_frame_free(&frame);
+ av_dict_free(&opt);
if (ret < 0)
av_log(log_ctx, AV_LOG_ERROR, "Error loading image file '%s'\n", filename);