2 * Copyright 2012 Stefano Sabatini <stefasab gmail com>
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 #include "libavutil/imgutils.h"
22 #include "lavfutils.h"
24 int ff_load_image(uint8_t *data
[4], int linesize
[4],
25 int *w
, int *h
, enum AVPixelFormat
*pix_fmt
,
26 const char *filename
, void *log_ctx
)
28 AVInputFormat
*iformat
= NULL
;
29 AVFormatContext
*format_ctx
= NULL
;
31 AVCodecContext
*codec_ctx
;
33 int frame_decoded
, ret
= 0;
40 iformat
= av_find_input_format("image2");
41 if ((ret
= avformat_open_input(&format_ctx
, filename
, iformat
, NULL
)) < 0) {
42 av_log(log_ctx
, AV_LOG_ERROR
,
43 "Failed to open input file '%s'\n", filename
);
47 if ((ret
= avformat_find_stream_info(format_ctx
, NULL
)) < 0) {
48 av_log(log_ctx
, AV_LOG_ERROR
, "Find stream info failed\n");
52 codec_ctx
= format_ctx
->streams
[0]->codec
;
53 codec
= avcodec_find_decoder(codec_ctx
->codec_id
);
55 av_log(log_ctx
, AV_LOG_ERROR
, "Failed to find codec\n");
56 ret
= AVERROR(EINVAL
);
60 if ((ret
= avcodec_open2(codec_ctx
, codec
, NULL
)) < 0) {
61 av_log(log_ctx
, AV_LOG_ERROR
, "Failed to open codec\n");
65 if (!(frame
= av_frame_alloc()) ) {
66 av_log(log_ctx
, AV_LOG_ERROR
, "Failed to alloc frame\n");
67 ret
= AVERROR(ENOMEM
);
71 ret
= av_read_frame(format_ctx
, &pkt
);
73 av_log(log_ctx
, AV_LOG_ERROR
, "Failed to read frame from file\n");
77 ret
= avcodec_decode_video2(codec_ctx
, frame
, &frame_decoded
, &pkt
);
78 if (ret
< 0 || !frame_decoded
) {
79 av_log(log_ctx
, AV_LOG_ERROR
, "Failed to decode image from file\n");
87 *pix_fmt
= frame
->format
;
89 if ((ret
= av_image_alloc(data
, linesize
, *w
, *h
, *pix_fmt
, 16)) < 0)
93 av_image_copy(data
, linesize
, (const uint8_t **)frame
->data
, frame
->linesize
, *pix_fmt
, *w
, *h
);
97 avcodec_close(codec_ctx
);
98 avformat_close_input(&format_ctx
);
102 av_log(log_ctx
, AV_LOG_ERROR
, "Error loading image file '%s'\n", filename
);