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