]> git.sesse.net Git - ffmpeg/blob - libavcodec/smvjpegdec.c
Merge commit '17776638c392d104975aba169e17b186490e1d5e'
[ffmpeg] / libavcodec / smvjpegdec.c
1 /*
2  * SMV JPEG decoder
3  * Copyright (c) 2013 Ash Hughes
4  *
5  * This file is part of FFmpeg.
6  *
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.
11  *
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.
16  *
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
20  */
21
22 /**
23  * @file
24  * SMV JPEG decoder.
25  */
26
27 // #define DEBUG
28 #include "avcodec.h"
29 #include "libavutil/opt.h"
30 #include "libavutil/imgutils.h"
31 #include "mjpegdec.h"
32 #include "internal.h"
33
34 typedef struct SMVJpegDecodeContext {
35     MJpegDecodeContext jpg;
36     AVFrame *picture[2]; /* pictures array */
37     AVCodecContext* avctx;
38     int frames_per_jpeg;
39     int mjpeg_data_size;
40 } SMVJpegDecodeContext;
41
42 static inline void smv_img_pnt_plane(uint8_t      **dst, uint8_t *src,
43                                      int src_linesize, int height, int nlines)
44 {
45     if (!dst || !src)
46         return;
47     src += (nlines) * src_linesize * height;
48     *dst = src;
49 }
50
51 static inline void smv_img_pnt(uint8_t *dst_data[4], uint8_t *src_data[4],
52                                const int src_linesizes[4],
53                                enum AVPixelFormat pix_fmt, int width, int height,
54                                int nlines)
55 {
56     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
57     int i, planes_nb = 0;
58
59     if (desc->flags & AV_PIX_FMT_FLAG_HWACCEL)
60         return;
61
62     for (i = 0; i < desc->nb_components; i++)
63         planes_nb = FFMAX(planes_nb, desc->comp[i].plane + 1);
64
65     for (i = 0; i < planes_nb; i++) {
66         int h = height;
67         if (i == 1 || i == 2) {
68             h = FF_CEIL_RSHIFT(height, desc->log2_chroma_h);
69         }
70         smv_img_pnt_plane(&dst_data[i], src_data[i],
71             src_linesizes[i], h, nlines);
72     }
73     if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
74         desc->flags & AV_PIX_FMT_FLAG_PSEUDOPAL)
75         dst_data[1] = src_data[1];
76 }
77
78 static av_cold int smvjpeg_decode_end(AVCodecContext *avctx)
79 {
80     SMVJpegDecodeContext *s = avctx->priv_data;
81     MJpegDecodeContext *jpg = &s->jpg;
82     int ret;
83
84     jpg->picture_ptr = NULL;
85     av_frame_free(&s->picture[0]);
86     av_frame_free(&s->picture[1]);
87     ret = avcodec_close(s->avctx);
88     av_freep(&s->avctx);
89     return ret;
90 }
91
92 static av_cold int smvjpeg_decode_init(AVCodecContext *avctx)
93 {
94     SMVJpegDecodeContext *s = avctx->priv_data;
95     AVCodec *codec;
96     AVDictionary *thread_opt = NULL;
97     int ret = 0, r;
98
99     s->frames_per_jpeg = 0;
100
101     s->picture[0] = av_frame_alloc();
102     if (!s->picture[0])
103         return AVERROR(ENOMEM);
104
105     s->picture[1] = av_frame_alloc();
106     if (!s->picture[1]) {
107         av_frame_free(&s->picture[0]);
108         return AVERROR(ENOMEM);
109     }
110
111     s->jpg.picture_ptr      = s->picture[0];
112
113     if (avctx->extradata_size >= 4)
114         s->frames_per_jpeg = AV_RL32(avctx->extradata);
115
116     if (s->frames_per_jpeg <= 0) {
117         av_log(avctx, AV_LOG_ERROR, "Invalid number of frames per jpeg.\n");
118         ret = AVERROR_INVALIDDATA;
119     }
120
121     codec = avcodec_find_decoder(AV_CODEC_ID_MJPEG);
122     if (!codec) {
123         av_log(avctx, AV_LOG_ERROR, "MJPEG codec not found\n");
124         smvjpeg_decode_end(avctx);
125         return AVERROR_DECODER_NOT_FOUND;
126     }
127
128     s->avctx = avcodec_alloc_context3(codec);
129
130     av_dict_set(&thread_opt, "threads", "1", 0);
131     s->avctx->refcounted_frames = 1;
132     s->avctx->flags = avctx->flags;
133     s->avctx->idct_algo = avctx->idct_algo;
134     if ((r = ff_codec_open2_recursive(s->avctx, codec, &thread_opt)) < 0) {
135         av_log(avctx, AV_LOG_ERROR, "MJPEG codec failed to open\n");
136         ret = r;
137     }
138     av_dict_free(&thread_opt);
139
140     if (ret < 0)
141         smvjpeg_decode_end(avctx);
142     return ret;
143 }
144
145 static int smvjpeg_decode_frame(AVCodecContext *avctx, void *data, int *data_size,
146                             AVPacket *avpkt)
147 {
148     const AVPixFmtDescriptor *desc;
149     SMVJpegDecodeContext *s = avctx->priv_data;
150     AVFrame* mjpeg_data = s->picture[0];
151     int i, cur_frame = 0, ret = 0;
152
153     cur_frame = avpkt->pts % s->frames_per_jpeg;
154
155     /* Are we at the start of a block? */
156     if (!cur_frame) {
157         av_frame_unref(mjpeg_data);
158         ret = avcodec_decode_video2(s->avctx, mjpeg_data, &s->mjpeg_data_size, avpkt);
159         if (ret < 0) {
160             s->mjpeg_data_size = 0;
161             return ret;
162         }
163     } else if (!s->mjpeg_data_size)
164         return AVERROR(EINVAL);
165
166     desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
167     av_assert0(desc);
168
169     if (mjpeg_data->height % (s->frames_per_jpeg << desc->log2_chroma_h)) {
170         av_log(avctx, AV_LOG_ERROR, "Invalid height\n");
171         return AVERROR_INVALIDDATA;
172     }
173
174     /*use the last lot... */
175     *data_size = s->mjpeg_data_size;
176
177     avctx->pix_fmt = s->avctx->pix_fmt;
178
179     /* We shouldn't get here if frames_per_jpeg <= 0 because this was rejected
180        in init */
181     ret = ff_set_dimensions(avctx, mjpeg_data->width, mjpeg_data->height / s->frames_per_jpeg);
182     if (ret < 0) {
183         av_log(s, AV_LOG_ERROR, "Failed to set dimensions\n");
184         return ret;
185     }
186
187     if (*data_size) {
188         s->picture[1]->extended_data = NULL;
189         s->picture[1]->width         = avctx->width;
190         s->picture[1]->height        = avctx->height;
191         s->picture[1]->format        = avctx->pix_fmt;
192         /* ff_init_buffer_info(avctx, &s->picture[1]); */
193         smv_img_pnt(s->picture[1]->data, mjpeg_data->data, mjpeg_data->linesize,
194                     avctx->pix_fmt, avctx->width, avctx->height, cur_frame);
195         for (i = 0; i < AV_NUM_DATA_POINTERS; i++)
196             s->picture[1]->linesize[i] = mjpeg_data->linesize[i];
197
198         ret = av_frame_ref(data, s->picture[1]);
199     }
200
201     return ret;
202 }
203
204 static const AVClass smvjpegdec_class = {
205     .class_name = "SMVJPEG decoder",
206     .item_name  = av_default_item_name,
207     .version    = LIBAVUTIL_VERSION_INT,
208 };
209
210 AVCodec ff_smvjpeg_decoder = {
211     .name           = "smvjpeg",
212     .long_name      = NULL_IF_CONFIG_SMALL("SMV JPEG"),
213     .type           = AVMEDIA_TYPE_VIDEO,
214     .id             = AV_CODEC_ID_SMVJPEG,
215     .priv_data_size = sizeof(SMVJpegDecodeContext),
216     .init           = smvjpeg_decode_init,
217     .close          = smvjpeg_decode_end,
218     .decode         = smvjpeg_decode_frame,
219     .priv_class     = &smvjpegdec_class,
220 };