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/mem_internal.h" 34 #include "libavutil/pixdesc.h" 35 #include "libavutil/stereo3d.h" 36 37 #include "avcodec.h" 38 #include "blockdsp.h" 39 #include "get_bits.h" 40 #include "hpeldsp.h" 41 #include "idctdsp.h" 42 43 #undef near /* This file uses struct member 'near' which in windows.h is defined as empty. */ 44 45 #define MAX_COMPONENTS 4 46 47 typedef struct ICCEntry { 48 uint8_t *data; 49 int length; 50 } ICCEntry; 51 52 struct JLSState; 53 54 typedef struct MJpegDecodeContext { 55 AVClass *class; 56 AVCodecContext *avctx; 57 GetBitContext gb; 58 int buf_size; 59 60 AVPacket *pkt; 61 62 int start_code; /* current start code */ 63 int buffer_size; 64 uint8_t *buffer; 65 66 uint16_t quant_matrixes[4][64]; 67 VLC vlcs[3][4]; 68 int qscale[4]; ///< quantizer scale calculated from quant_matrixes 69 70 int orig_height; /* size given at codec init */ 71 int first_picture; /* true if decoding first picture */ 72 int interlaced; /* true if interlaced */ 73 int bottom_field; /* true if bottom field */ 74 int lossless; 75 int ls; 76 int progressive; 77 int bayer; /* true if it's a bayer-encoded JPEG embedded in a DNG */ 78 int rgb; 79 uint8_t upscale_h[4]; 80 uint8_t upscale_v[4]; 81 int rct; /* standard rct */ 82 int pegasus_rct; /* pegasus reversible colorspace transform */ 83 int bits; /* bits per component */ 84 int colr; 85 int xfrm; 86 int adobe_transform; 87 88 int maxval; 89 int near; ///< near lossless bound (si 0 for lossless) 90 int t1,t2,t3; 91 int reset; ///< context halfing interval ?rename 92 93 int width, height; 94 int mb_width, mb_height; 95 int nb_components; 96 int block_stride[MAX_COMPONENTS]; 97 int component_id[MAX_COMPONENTS]; 98 int h_count[MAX_COMPONENTS]; /* horizontal and vertical count for each component */ 99 int v_count[MAX_COMPONENTS]; 100 int comp_index[MAX_COMPONENTS]; 101 int dc_index[MAX_COMPONENTS]; 102 int ac_index[MAX_COMPONENTS]; 103 int nb_blocks[MAX_COMPONENTS]; 104 int h_scount[MAX_COMPONENTS]; 105 int v_scount[MAX_COMPONENTS]; 106 int quant_sindex[MAX_COMPONENTS]; 107 int h_max, v_max; /* maximum h and v counts */ 108 int quant_index[4]; /* quant table index for each component */ 109 int last_dc[MAX_COMPONENTS]; /* last DEQUANTIZED dc (XXX: am I right to do that ?) */ 110 AVFrame *picture; /* picture structure */ 111 AVFrame *picture_ptr; /* pointer to picture structure */ 112 int got_picture; ///< we found a SOF and picture is valid, too. 113 int linesize[MAX_COMPONENTS]; ///< linesize << interlaced 114 int8_t *qscale_table; 115 DECLARE_ALIGNED(32, int16_t, block)[64]; 116 int16_t (*blocks[MAX_COMPONENTS])[64]; ///< intermediate sums (progressive mode) 117 uint8_t *last_nnz[MAX_COMPONENTS]; 118 uint64_t coefs_finished[MAX_COMPONENTS]; ///< bitmask of which coefs have been completely decoded (progressive mode) 119 int palette_index; 120 int force_pal8; 121 ScanTable scantable; 122 BlockDSPContext bdsp; 123 HpelDSPContext hdsp; 124 IDCTDSPContext idsp; 125 126 int restart_interval; 127 int restart_count; 128 129 int buggy_avid; 130 int cs_itu601; 131 int interlace_polarity; 132 int multiscope; 133 134 int mjpb_skiptosod; 135 136 int cur_scan; /* current scan, used by JPEG-LS */ 137 int flipped; /* true if picture is flipped */ 138 139 uint16_t (*ljpeg_buffer)[4]; 140 unsigned int ljpeg_buffer_size; 141 142 int extern_huff; 143 AVDictionary *exif_metadata; 144 145 AVStereo3D *stereo3d; ///!< stereoscopic information (cached, since it is read before frame allocation) 146 147 const AVPixFmtDescriptor *pix_desc; 148 149 ICCEntry *iccentries; 150 int iccnum; 151 int iccread; 152 153 AVFrame *smv_frame; 154 int smv_frames_per_jpeg; 155 int smv_next_frame; 156 157 // Raw stream data for hwaccel use. 158 const uint8_t *raw_image_buffer; 159 size_t raw_image_buffer_size; 160 const uint8_t *raw_scan_buffer; 161 size_t raw_scan_buffer_size; 162 163 uint8_t raw_huffman_lengths[2][4][16]; 164 uint8_t raw_huffman_values[2][4][256]; 165 166 enum AVPixelFormat hwaccel_sw_pix_fmt; 167 enum AVPixelFormat hwaccel_pix_fmt; 168 void *hwaccel_picture_private; 169 struct JLSState *jls_state; 170 } MJpegDecodeContext; 171 172 int ff_mjpeg_build_vlc(VLC *vlc, const uint8_t *bits_table, 173 const uint8_t *val_table, int is_ac, void *logctx); 174 int ff_mjpeg_decode_init(AVCodecContext *avctx); 175 int ff_mjpeg_decode_end(AVCodecContext *avctx); 176 int ff_mjpeg_receive_frame(AVCodecContext *avctx, AVFrame *frame); 177 int ff_mjpeg_decode_dqt(MJpegDecodeContext *s); 178 int ff_mjpeg_decode_dht(MJpegDecodeContext *s); 179 int ff_mjpeg_decode_sof(MJpegDecodeContext *s); 180 int ff_mjpeg_decode_sos(MJpegDecodeContext *s, 181 const uint8_t *mb_bitmask,int mb_bitmask_size, 182 const AVFrame *reference); 183 int ff_mjpeg_find_marker(MJpegDecodeContext *s, 184 const uint8_t **buf_ptr, const uint8_t *buf_end, 185 const uint8_t **unescaped_buf_ptr, int *unescaped_buf_size); 186 187 int ff_sp5x_process_packet(AVCodecContext *avctx, AVPacket *avpkt); 188 189 #endif /* AVCODEC_MJPEGDEC_H */ 190