1 /* 2 * MJPEG decoder 3 * Copyright (c) 2000, 2001 Fabrice Bellard 4 * Copyright (c) 2003 Alex Beregszaszi 5 * Copyright (c) 2003-2004 Michael Niedermayer 6 * 7 * This file is part of FFmpeg. 8 * 9 * FFmpeg is free software; you can redistribute it and/or 10 * modify it under the terms of the GNU Lesser General Public 11 * License as published by the Free Software Foundation; either 12 * version 2.1 of the License, or (at your option) any later version. 13 * 14 * FFmpeg is distributed in the hope that it will be useful, 15 * but WITHOUT ANY WARRANTY; without even the implied warranty of 16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 17 * Lesser General Public License for more details. 18 * 19 * You should have received a copy of the GNU Lesser General Public 20 * License along with FFmpeg; if not, write to the Free Software 21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 22 */ 23 24 /** 25 * @file 26 * MJPEG decoder. 27 */ 28 29 #ifndef AVCODEC_MJPEGDEC_H 30 #define AVCODEC_MJPEGDEC_H 31 32 #include "libavutil/log.h" 33 #include "libavutil/pixdesc.h" 34 #include "libavutil/stereo3d.h" 35 36 #include "avcodec.h" 37 #include "blockdsp.h" 38 #include "get_bits.h" 39 #include "hpeldsp.h" 40 #include "idctdsp.h" 41 42 #undef near /* This file uses struct member 'near' which in windows.h is defined as empty. */ 43 44 #define MAX_COMPONENTS 4 45 46 typedef struct MJpegDecodeContext { 47 AVClass *class; 48 AVCodecContext *avctx; 49 GetBitContext gb; 50 int buf_size; 51 52 int start_code; /* current start code */ 53 int buffer_size; 54 uint8_t *buffer; 55 56 uint16_t quant_matrixes[4][64]; 57 VLC vlcs[3][4]; 58 int qscale[4]; ///< quantizer scale calculated from quant_matrixes 59 60 int org_height; /* size given at codec init */ 61 int first_picture; /* true if decoding first picture */ 62 int interlaced; /* true if interlaced */ 63 int bottom_field; /* true if bottom field */ 64 int lossless; 65 int ls; 66 int progressive; 67 int bayer; /* true if it's a bayer-encoded JPEG embedded in a DNG */ 68 int rgb; 69 uint8_t upscale_h[4]; 70 uint8_t upscale_v[4]; 71 int rct; /* standard rct */ 72 int pegasus_rct; /* pegasus reversible colorspace transform */ 73 int bits; /* bits per component */ 74 int colr; 75 int xfrm; 76 int adobe_transform; 77 78 int maxval; 79 int near; ///< near lossless bound (si 0 for lossless) 80 int t1,t2,t3; 81 int reset; ///< context halfing interval ?rename 82 83 int width, height; 84 int mb_width, mb_height; 85 int nb_components; 86 int block_stride[MAX_COMPONENTS]; 87 int component_id[MAX_COMPONENTS]; 88 int h_count[MAX_COMPONENTS]; /* horizontal and vertical count for each component */ 89 int v_count[MAX_COMPONENTS]; 90 int comp_index[MAX_COMPONENTS]; 91 int dc_index[MAX_COMPONENTS]; 92 int ac_index[MAX_COMPONENTS]; 93 int nb_blocks[MAX_COMPONENTS]; 94 int h_scount[MAX_COMPONENTS]; 95 int v_scount[MAX_COMPONENTS]; 96 int quant_sindex[MAX_COMPONENTS]; 97 int h_max, v_max; /* maximum h and v counts */ 98 int quant_index[4]; /* quant table index for each component */ 99 int last_dc[MAX_COMPONENTS]; /* last DEQUANTIZED dc (XXX: am I right to do that ?) */ 100 AVFrame *picture; /* picture structure */ 101 AVFrame *picture_ptr; /* pointer to picture structure */ 102 int got_picture; ///< we found a SOF and picture is valid, too. 103 int linesize[MAX_COMPONENTS]; ///< linesize << interlaced 104 int8_t *qscale_table; 105 DECLARE_ALIGNED(32, int16_t, block)[64]; 106 int16_t (*blocks[MAX_COMPONENTS])[64]; ///< intermediate sums (progressive mode) 107 uint8_t *last_nnz[MAX_COMPONENTS]; 108 uint64_t coefs_finished[MAX_COMPONENTS]; ///< bitmask of which coefs have been completely decoded (progressive mode) 109 int palette_index; 110 ScanTable scantable; 111 BlockDSPContext bdsp; 112 HpelDSPContext hdsp; 113 IDCTDSPContext idsp; 114 115 int restart_interval; 116 int restart_count; 117 118 int buggy_avid; 119 int cs_itu601; 120 int interlace_polarity; 121 int multiscope; 122 123 int mjpb_skiptosod; 124 125 int cur_scan; /* current scan, used by JPEG-LS */ 126 int flipped; /* true if picture is flipped */ 127 128 uint16_t (*ljpeg_buffer)[4]; 129 unsigned int ljpeg_buffer_size; 130 131 int extern_huff; 132 AVDictionary *exif_metadata; 133 134 AVStereo3D *stereo3d; ///!< stereoscopic information (cached, since it is read before frame allocation) 135 136 const AVPixFmtDescriptor *pix_desc; 137 138 uint8_t **iccdata; 139 int *iccdatalens; 140 int iccnum; 141 int iccread; 142 143 // Raw stream data for hwaccel use. 144 const uint8_t *raw_image_buffer; 145 size_t raw_image_buffer_size; 146 const uint8_t *raw_scan_buffer; 147 size_t raw_scan_buffer_size; 148 149 uint8_t raw_huffman_lengths[2][4][16]; 150 uint8_t raw_huffman_values[2][4][256]; 151 152 enum AVPixelFormat hwaccel_sw_pix_fmt; 153 enum AVPixelFormat hwaccel_pix_fmt; 154 void *hwaccel_picture_private; 155 } MJpegDecodeContext; 156 157 int ff_mjpeg_decode_init(AVCodecContext *avctx); 158 int ff_mjpeg_decode_end(AVCodecContext *avctx); 159 int ff_mjpeg_decode_frame(AVCodecContext *avctx, 160 void *data, int *got_frame, 161 AVPacket *avpkt); 162 int ff_mjpeg_decode_dqt(MJpegDecodeContext *s); 163 int ff_mjpeg_decode_dht(MJpegDecodeContext *s); 164 int ff_mjpeg_decode_sof(MJpegDecodeContext *s); 165 int ff_mjpeg_decode_sos(MJpegDecodeContext *s, 166 const uint8_t *mb_bitmask,int mb_bitmask_size, 167 const AVFrame *reference); 168 int ff_mjpeg_find_marker(MJpegDecodeContext *s, 169 const uint8_t **buf_ptr, const uint8_t *buf_end, 170 const uint8_t **unescaped_buf_ptr, int *unescaped_buf_size); 171 172 #endif /* AVCODEC_MJPEGDEC_H */ 173