3 * Copyright (c) 2013 Ash Hughes
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
29 #include "libavutil/opt.h"
30 #include "libavutil/imgutils.h"
34 typedef struct SMVJpegDecodeContext {
35 MJpegDecodeContext jpg;
36 AVFrame *picture[2]; /* pictures array */
37 AVCodecContext* avctx;
39 } SMVJpegDecodeContext;
41 static inline void smv_img_pnt_plane(uint8_t **dst, uint8_t *src,
42 int src_linesize, int height, int nlines)
46 src += (nlines) * src_linesize * height;
50 static inline void smv_img_pnt(uint8_t *dst_data[4], uint8_t *src_data[4],
51 const int src_linesizes[4],
52 enum PixelFormat pix_fmt, int width, int height,
55 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
58 if (desc->flags & PIX_FMT_HWACCEL)
61 for (i = 0; i < desc->nb_components; i++)
62 planes_nb = FFMAX(planes_nb, desc->comp[i].plane + 1);
64 for (i = 0; i < planes_nb; i++) {
66 if (i == 1 || i == 2) {
67 h = FF_CEIL_RSHIFT(height, desc->log2_chroma_h);
69 smv_img_pnt_plane(&dst_data[i], src_data[i],
70 src_linesizes[i], h, nlines);
74 static av_cold int smvjpeg_decode_init(AVCodecContext *avctx)
76 SMVJpegDecodeContext *s = avctx->priv_data;
78 AVDictionary *thread_opt = NULL;
81 s->frames_per_jpeg = 0;
83 s->picture[0] = av_frame_alloc();
85 return AVERROR(ENOMEM);
87 s->picture[1] = av_frame_alloc();
89 return AVERROR(ENOMEM);
91 s->jpg.picture_ptr = s->picture[0];
93 if (avctx->extradata_size >= 4)
94 s->frames_per_jpeg = AV_RL32(avctx->extradata);
96 if (s->frames_per_jpeg <= 0) {
97 av_log(avctx, AV_LOG_ERROR, "Invalid number of frames per jpeg.\n");
101 avcodec_get_frame_defaults(s->picture[1]);
102 avctx->coded_frame = s->picture[1];
103 codec = avcodec_find_decoder(AV_CODEC_ID_MJPEG);
105 av_log(avctx, AV_LOG_ERROR, "MJPEG codec not found\n");
109 s->avctx = avcodec_alloc_context3(codec);
111 av_dict_set(&thread_opt, "threads", "1", 0);
112 s->avctx->flags = avctx->flags;
113 s->avctx->idct_algo = avctx->idct_algo;
114 if (ff_codec_open2_recursive(s->avctx, codec, &thread_opt) < 0) {
115 av_log(avctx, AV_LOG_ERROR, "MJPEG codec failed to open\n");
118 av_dict_free(&thread_opt);
123 static int smvjpeg_decode_frame(AVCodecContext *avctx, void *data, int *data_size,
126 SMVJpegDecodeContext *s = avctx->priv_data;
127 AVFrame* mjpeg_data = s->picture[0];
128 int i, cur_frame = 0, ret = 0;
130 cur_frame = avpkt->pts % s->frames_per_jpeg;
132 /* Are we at the start of a block? */
134 ret = avcodec_decode_video2(s->avctx, mjpeg_data, data_size, avpkt);
135 else /*use the last lot... */
136 *data_size = sizeof(AVPicture);
138 avctx->pix_fmt = s->avctx->pix_fmt;
140 /* We shouldn't get here if frames_per_jpeg <= 0 because this was rejected
142 avcodec_set_dimensions(avctx, mjpeg_data->width,
143 mjpeg_data->height / s->frames_per_jpeg);
145 s->picture[1]->extended_data = NULL;
146 s->picture[1]->width = avctx->width;
147 s->picture[1]->height = avctx->height;
148 s->picture[1]->format = avctx->pix_fmt;
149 /* ff_init_buffer_info(avctx, &s->picture[1]); */
150 smv_img_pnt(s->picture[1]->data, mjpeg_data->data, mjpeg_data->linesize,
151 avctx->pix_fmt, avctx->width, avctx->height, cur_frame);
152 for (i = 0; i < AV_NUM_DATA_POINTERS; i++)
153 s->picture[1]->linesize[i] = mjpeg_data->linesize[i];
155 ret = av_frame_ref(data, s->picture[1]);
160 static av_cold int smvjpeg_decode_end(AVCodecContext *avctx)
162 SMVJpegDecodeContext *s = avctx->priv_data;
163 MJpegDecodeContext *jpg = &s->jpg;
165 jpg->picture_ptr = NULL;
166 av_frame_free(&s->picture[1]);
167 ff_codec_close_recursive(s->avctx);
172 static const AVClass smvjpegdec_class = {
173 .class_name = "SMVJPEG decoder",
174 .item_name = av_default_item_name,
175 .version = LIBAVUTIL_VERSION_INT,
178 AVCodec ff_smvjpeg_decoder = {
180 .type = AVMEDIA_TYPE_VIDEO,
181 .id = AV_CODEC_ID_SMVJPEG,
182 .priv_data_size = sizeof(SMVJpegDecodeContext),
183 .init = smvjpeg_decode_init,
184 .close = smvjpeg_decode_end,
185 .decode = smvjpeg_decode_frame,
187 .long_name = NULL_IF_CONFIG_SMALL("SMV JPEG"),
188 .priv_class = &smvjpegdec_class,