FFmpeg  2.6.3
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Modules Pages
mjpegenc.c
Go to the documentation of this file.
1 /*
2  * MJPEG encoder
3  * Copyright (c) 2000, 2001 Fabrice Bellard
4  * Copyright (c) 2003 Alex Beregszaszi
5  * Copyright (c) 2003-2004 Michael Niedermayer
6  *
7  * Support for external huffman table, various fixes (AVID workaround),
8  * aspecting, new decode_frame mechanism and apple mjpeg-b support
9  * by Alex Beregszaszi
10  *
11  * This file is part of FFmpeg.
12  *
13  * FFmpeg is free software; you can redistribute it and/or
14  * modify it under the terms of the GNU Lesser General Public
15  * License as published by the Free Software Foundation; either
16  * version 2.1 of the License, or (at your option) any later version.
17  *
18  * FFmpeg is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21  * Lesser General Public License for more details.
22  *
23  * You should have received a copy of the GNU Lesser General Public
24  * License along with FFmpeg; if not, write to the Free Software
25  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26  */
27 
28 /**
29  * @file
30  * MJPEG encoder.
31  */
32 
33 #include "libavutil/pixdesc.h"
34 
35 #include "avcodec.h"
36 #include "mjpegenc_common.h"
37 #include "mpegvideo.h"
38 #include "mjpeg.h"
39 #include "mjpegenc.h"
40 
41 static uint8_t uni_ac_vlc_len[64 * 64 * 2];
42 static uint8_t uni_chroma_ac_vlc_len[64 * 64 * 2];
43 
44 static av_cold void init_uni_ac_vlc(const uint8_t huff_size_ac[256], uint8_t *uni_ac_vlc_len)
45 {
46  int i;
47 
48  for (i = 0; i < 128; i++) {
49  int level = i - 64;
50  int run;
51  if (!level)
52  continue;
53  for (run = 0; run < 64; run++) {
54  int len, code, nbits;
55  int alevel = FFABS(level);
56 
57  len = (run >> 4) * huff_size_ac[0xf0];
58 
59  nbits= av_log2_16bit(alevel) + 1;
60  code = ((15&run) << 4) | nbits;
61 
62  len += huff_size_ac[code] + nbits;
63 
64  uni_ac_vlc_len[UNI_AC_ENC_INDEX(run, i)] = len;
65  // We ignore EOB as its just a constant which does not change generally
66  }
67  }
68 }
69 
71 {
72  MJpegContext *m;
73 
74  if (s->width > 65500 || s->height > 65500) {
75  av_log(s, AV_LOG_ERROR, "JPEG does not support resolutions above 65500x65500\n");
76  return AVERROR(EINVAL);
77  }
78 
79  m = av_malloc(sizeof(MJpegContext));
80  if (!m)
81  return AVERROR(ENOMEM);
82 
83  s->min_qcoeff=-1023;
84  s->max_qcoeff= 1023;
85 
86  /* build all the huffman tables */
103 
110 
111  s->mjpeg_ctx = m;
112  return 0;
113 }
114 
116 {
117  av_freep(&s->mjpeg_ctx);
118 }
119 
120 static void encode_block(MpegEncContext *s, int16_t *block, int n)
121 {
122  int mant, nbits, code, i, j;
123  int component, dc, run, last_index, val;
124  MJpegContext *m = s->mjpeg_ctx;
125  uint8_t *huff_size_ac;
126  uint16_t *huff_code_ac;
127 
128  /* DC coef */
129  component = (n <= 3 ? 0 : (n&1) + 1);
130  dc = block[0]; /* overflow is impossible */
131  val = dc - s->last_dc[component];
132  if (n < 4) {
134  huff_size_ac = m->huff_size_ac_luminance;
135  huff_code_ac = m->huff_code_ac_luminance;
136  } else {
138  huff_size_ac = m->huff_size_ac_chrominance;
139  huff_code_ac = m->huff_code_ac_chrominance;
140  }
141  s->last_dc[component] = dc;
142 
143  /* AC coefs */
144 
145  run = 0;
146  last_index = s->block_last_index[n];
147  for(i=1;i<=last_index;i++) {
148  j = s->intra_scantable.permutated[i];
149  val = block[j];
150  if (val == 0) {
151  run++;
152  } else {
153  while (run >= 16) {
154  put_bits(&s->pb, huff_size_ac[0xf0], huff_code_ac[0xf0]);
155  run -= 16;
156  }
157  mant = val;
158  if (val < 0) {
159  val = -val;
160  mant--;
161  }
162 
163  nbits= av_log2_16bit(val) + 1;
164  code = (run << 4) | nbits;
165 
166  put_bits(&s->pb, huff_size_ac[code], huff_code_ac[code]);
167 
168  put_sbits(&s->pb, nbits, mant);
169  run = 0;
170  }
171  }
172 
173  /* output EOB only if not already 64 values */
174  if (last_index < 63 || run != 0)
175  put_bits(&s->pb, huff_size_ac[0], huff_code_ac[0]);
176 }
177 
178 void ff_mjpeg_encode_mb(MpegEncContext *s, int16_t block[12][64])
179 {
180  int i;
181  if (s->chroma_format == CHROMA_444) {
182  encode_block(s, block[0], 0);
183  encode_block(s, block[2], 2);
184  encode_block(s, block[4], 4);
185  encode_block(s, block[8], 8);
186  encode_block(s, block[5], 5);
187  encode_block(s, block[9], 9);
188 
189  if (16*s->mb_x+8 < s->width) {
190  encode_block(s, block[1], 1);
191  encode_block(s, block[3], 3);
192  encode_block(s, block[6], 6);
193  encode_block(s, block[10], 10);
194  encode_block(s, block[7], 7);
195  encode_block(s, block[11], 11);
196  }
197  } else {
198  for(i=0;i<5;i++) {
199  encode_block(s, block[i], i);
200  }
201  if (s->chroma_format == CHROMA_420) {
202  encode_block(s, block[5], 5);
203  } else {
204  encode_block(s, block[6], 6);
205  encode_block(s, block[5], 5);
206  encode_block(s, block[7], 7);
207  }
208  }
209 
210  s->i_tex_bits += get_bits_diff(s);
211 }
212 
213 // maximum over s->mjpeg_vsample[i]
214 #define V_MAX 2
216  const AVFrame *pic_arg, int *got_packet)
217 
218 {
219  MpegEncContext *s = avctx->priv_data;
220  AVFrame *pic;
221  int i, ret;
222  int chroma_h_shift, chroma_v_shift;
223 
224  av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &chroma_h_shift, &chroma_v_shift);
225 
226  //CODEC_FLAG_EMU_EDGE have to be cleared
228  return AVERROR(EINVAL);
229 
230  if ((avctx->height & 15) && avctx->strict_std_compliance > FF_COMPLIANCE_UNOFFICIAL) {
231  av_log(avctx, AV_LOG_ERROR,
232  "Heights which are not a multiple of 16 might fail with some decoders, "
233  "use vstrict=-1 / -strict -1 to use %d anyway.\n", avctx->height);
234  av_log(avctx, AV_LOG_WARNING, "If you have a device that plays AMV videos, please test if videos "
235  "with such heights work with it and report your findings to ffmpeg-devel@ffmpeg.org\n");
236  return AVERROR_EXPERIMENTAL;
237  }
238 
239  pic = av_frame_clone(pic_arg);
240  if (!pic)
241  return AVERROR(ENOMEM);
242  //picture should be flipped upside-down
243  for(i=0; i < 3; i++) {
244  int vsample = i ? 2 >> chroma_v_shift : 2;
245  pic->data[i] += pic->linesize[i] * (vsample * s->height / V_MAX - 1);
246  pic->linesize[i] *= -1;
247  }
248  ret = ff_mpv_encode_picture(avctx, pkt, pic, got_packet);
249  av_frame_free(&pic);
250  return ret;
251 }
252 
253 #if CONFIG_MJPEG_ENCODER
255 
256 AVCodec ff_mjpeg_encoder = {
257  .name = "mjpeg",
258  .long_name = NULL_IF_CONFIG_SMALL("MJPEG (Motion JPEG)"),
259  .type = AVMEDIA_TYPE_VIDEO,
260  .id = AV_CODEC_ID_MJPEG,
261  .priv_data_size = sizeof(MpegEncContext),
263  .encode2 = ff_mpv_encode_picture,
264  .close = ff_mpv_encode_end,
266  .pix_fmts = (const enum AVPixelFormat[]){
268  },
269  .priv_class = &mjpeg_class,
270 };
271 #endif
272 #if CONFIG_AMV_ENCODER
274 
275 AVCodec ff_amv_encoder = {
276  .name = "amv",
277  .long_name = NULL_IF_CONFIG_SMALL("AMV Video"),
278  .type = AVMEDIA_TYPE_VIDEO,
279  .id = AV_CODEC_ID_AMV,
280  .priv_data_size = sizeof(MpegEncContext),
282  .encode2 = amv_encode_picture,
283  .close = ff_mpv_encode_end,
284  .pix_fmts = (const enum AVPixelFormat[]){
286  },
287  .priv_class = &amv_class,
288 };
289 #endif
const char const char void * val
Definition: avisynth_c.h:672
const char * s
Definition: avisynth_c.h:669
This structure describes decoded (raw) audio or video data.
Definition: frame.h:163
static void put_sbits(PutBitContext *pb, int n, int32_t value)
Definition: put_bits.h:192
#define FF_MPV_GENERIC_CLASS(name)
Definition: mpegvideo.h:730
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:160
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:181
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
const uint8_t avpriv_mjpeg_bits_ac_luminance[17]
Definition: mjpeg.c:73
MJPEG encoder.
static uint8_t uni_chroma_ac_vlc_len[64 *64 *2]
Definition: mjpegenc.c:42
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1442
int min_qcoeff
minimum encodable coefficient
Definition: mpegvideo.h:441
const uint8_t avpriv_mjpeg_bits_dc_chrominance[17]
Definition: mjpeg.c:70
mpegvideo header.
const uint8_t avpriv_mjpeg_bits_ac_chrominance[17]
Definition: mjpeg.c:99
av_cold int ff_mjpeg_encode_init(MpegEncContext *s)
Definition: mjpegenc.c:70
uint8_t permutated[64]
Definition: idctdsp.h:31
uint8_t run
Definition: svq3.c:149
static AVPacket pkt
uint8_t * intra_ac_vlc_length
Definition: mpegvideo.h:444
#define UNI_AC_ENC_INDEX(run, level)
Definition: mpegvideo.h:451
AVCodec.
Definition: avcodec.h:3173
MJPEG encoder and decoder.
uint16_t huff_code_dc_chrominance[12]
Definition: mjpegenc.h:44
#define V_MAX
Definition: mjpegenc.c:214
uint16_t huff_code_ac_luminance[256]
Definition: mjpegenc.h:47
uint8_t
#define av_cold
Definition: attributes.h:74
#define av_malloc(s)
static uint8_t uni_ac_vlc_len[64 *64 *2]
Definition: mjpegenc.c:41
#define CODEC_CAP_INTRA_ONLY
Codec is intra only.
Definition: avcodec.h:884
static void encode_block(MpegEncContext *s, int16_t *block, int n)
Definition: mjpegenc.c:120
static int amv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic_arg, int *got_packet)
Definition: mjpegenc.c:215
#define CHROMA_420
Definition: mpegvideo.h:602
void ff_mjpeg_build_huffman_codes(uint8_t *huff_size, uint16_t *huff_code, const uint8_t *bits_table, const uint8_t *val_table)
Definition: mjpeg.c:127
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of PIX_FMT_YUV422P and setting color_...
Definition: pixfmt.h:81
uint8_t huff_size_dc_chrominance[12]
Definition: mjpegenc.h:43
int max_qcoeff
maximum encodable coefficient
Definition: mpegvideo.h:442
#define CHROMA_444
Definition: mpegvideo.h:604
#define av_log(a,...)
unsigned m
Definition: audioconvert.c:187
static int get_bits_diff(MpegEncContext *s)
Definition: mpegvideo.h:833
uint16_t huff_code_ac_chrominance[256]
Definition: mjpegenc.h:49
uint8_t huff_size_ac_luminance[256]
Definition: mjpegenc.h:46
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:175
int last_dc[3]
last DC values for MPEG1
Definition: mpegvideo.h:319
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:148
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
Definition: pixdesc.c:2057
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:180
uint8_t * intra_chroma_ac_vlc_length
Definition: mpegvideo.h:446
int flags
CODEC_FLAG_*.
Definition: avcodec.h:1333
uint8_t huff_size_ac_chrominance[256]
Definition: mjpegenc.h:48
const char * name
Name of the codec implementation.
Definition: avcodec.h:3180
Libavcodec external API header.
uint8_t * intra_ac_vlc_last_length
Definition: mpegvideo.h:445
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of PIX_FMT_YUV420P and setting color_...
Definition: pixfmt.h:80
ret
Definition: avfilter.c:974
const uint8_t avpriv_mjpeg_bits_dc_luminance[17]
Definition: mjpeg.c:65
const uint8_t avpriv_mjpeg_val_dc[12]
Definition: mjpeg.c:67
#define FFABS(a)
Definition: common.h:61
int block_last_index[12]
last non zero coefficient in block
Definition: mpegvideo.h:217
int n
Definition: avisynth_c.h:589
AVFrame * av_frame_clone(const AVFrame *src)
Create a new frame that references the same data as src.
Definition: frame.c:364
#define AVERROR_EXPERIMENTAL
Requested feature is flagged experimental. Set strict_std_compliance if you really want to use it...
Definition: error.h:72
struct MJpegContext * mjpeg_ctx
Definition: mpegvideo.h:555
#define FF_COMPLIANCE_UNOFFICIAL
Allow unofficial extensions.
Definition: avcodec.h:2545
void ff_mjpeg_encode_dc(PutBitContext *pb, int val, uint8_t *huff_size, uint16_t *huff_code)
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:191
uint8_t * intra_chroma_ac_vlc_last_length
Definition: mpegvideo.h:447
main external API structure.
Definition: avcodec.h:1239
ScanTable intra_scantable
Definition: mpegvideo.h:222
int height
picture size. must be a multiple of 16
Definition: mpegvideo.h:231
int ff_mpv_encode_init(AVCodecContext *avctx)
av_cold void ff_mjpeg_encode_close(MpegEncContext *s)
Definition: mjpegenc.c:115
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:174
uint8_t level
Definition: svq3.c:150
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_dlog(ac->avr,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> dc
MpegEncContext.
Definition: mpegvideo.h:212
const uint8_t avpriv_mjpeg_val_ac_luminance[]
Definition: mjpeg.c:75
struct AVCodecContext * avctx
Definition: mpegvideo.h:229
uint16_t huff_code_dc_luminance[12]
Definition: mjpegenc.h:42
PutBitContext pb
bit output
Definition: mpegvideo.h:285
uint8_t huff_size_dc_luminance[12]
Definition: mjpegenc.h:41
#define CODEC_FLAG_EMU_EDGE
Definition: avcodec.h:742
#define CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
Definition: avcodec.h:868
#define CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
Definition: avcodec.h:864
static av_cold void init_uni_ac_vlc(const uint8_t huff_size_ac[256], uint8_t *uni_ac_vlc_len)
Definition: mjpegenc.c:44
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of PIX_FMT_YUV444P and setting color_...
Definition: pixfmt.h:82
int ff_mpv_encode_end(AVCodecContext *avctx)
void * priv_data
Definition: avcodec.h:1281
const uint8_t avpriv_mjpeg_val_ac_chrominance[]
Definition: mjpeg.c:102
int len
#define av_log2_16bit
Definition: intmath.h:106
void ff_mjpeg_encode_mb(MpegEncContext *s, int16_t block[12][64])
Definition: mjpegenc.c:178
#define av_freep(p)
int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
AVPixelFormat
Pixel format.
Definition: pixfmt.h:66
This structure stores compressed data.
Definition: avcodec.h:1137
int strict_std_compliance
strictly follow the standard (MPEG4, ...).
Definition: avcodec.h:2541
static int16_t block[64]
Definition: dct-test.c:110