mjpegdec: use the AVFrame API properly.
[libav.git] / libavcodec / mjpegdec.h
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 Libav.
8 *
9 * Libav 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 * Libav 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 Libav; 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
34 #include "avcodec.h"
35 #include "get_bits.h"
36 #include "dsputil.h"
37 #include "hpeldsp.h"
38
39 #define MAX_COMPONENTS 4
40
41 typedef struct MJpegDecodeContext {
42 AVClass *class;
43 AVCodecContext *avctx;
44 GetBitContext gb;
45
46 int start_code; /* current start code */
47 int buffer_size;
48 uint8_t *buffer;
49
50 int16_t quant_matrixes[4][64];
51 VLC vlcs[3][4];
52 int qscale[4]; ///< quantizer scale calculated from quant_matrixes
53
54 int org_height; /* size given at codec init */
55 int first_picture; /* true if decoding first picture */
56 int interlaced; /* true if interlaced */
57 int bottom_field; /* true if bottom field */
58 int lossless;
59 int ls;
60 int progressive;
61 int rgb;
62 int rct; /* standard rct */
63 int pegasus_rct; /* pegasus reversible colorspace transform */
64 int bits; /* bits per component */
65
66 int maxval;
67 int near; ///< near lossless bound (si 0 for lossless)
68 int t1,t2,t3;
69 int reset; ///< context halfing intervall ?rename
70
71 int width, height;
72 int mb_width, mb_height;
73 int nb_components;
74 int block_stride[MAX_COMPONENTS];
75 int component_id[MAX_COMPONENTS];
76 int h_count[MAX_COMPONENTS]; /* horizontal and vertical count for each component */
77 int v_count[MAX_COMPONENTS];
78 int comp_index[MAX_COMPONENTS];
79 int dc_index[MAX_COMPONENTS];
80 int ac_index[MAX_COMPONENTS];
81 int nb_blocks[MAX_COMPONENTS];
82 int h_scount[MAX_COMPONENTS];
83 int v_scount[MAX_COMPONENTS];
84 int h_max, v_max; /* maximum h and v counts */
85 int quant_index[4]; /* quant table index for each component */
86 int last_dc[MAX_COMPONENTS]; /* last DEQUANTIZED dc (XXX: am I right to do that ?) */
87 AVFrame *picture; /* picture structure */
88 AVFrame *picture_ptr; /* pointer to picture structure */
89 int got_picture; ///< we found a SOF and picture is valid, too.
90 int linesize[MAX_COMPONENTS]; ///< linesize << interlaced
91 int8_t *qscale_table;
92 DECLARE_ALIGNED(16, int16_t, block)[64];
93 int16_t (*blocks[MAX_COMPONENTS])[64]; ///< intermediate sums (progressive mode)
94 uint8_t *last_nnz[MAX_COMPONENTS];
95 uint64_t coefs_finished[MAX_COMPONENTS]; ///< bitmask of which coefs have been completely decoded (progressive mode)
96 ScanTable scantable;
97 DSPContext dsp;
98 HpelDSPContext hdsp;
99
100 int restart_interval;
101 int restart_count;
102
103 int buggy_avid;
104 int cs_itu601;
105 int interlace_polarity;
106
107 int mjpb_skiptosod;
108
109 int cur_scan; /* current scan, used by JPEG-LS */
110 int flipped; /* true if picture is flipped */
111
112 uint16_t (*ljpeg_buffer)[4];
113 unsigned int ljpeg_buffer_size;
114
115 int extern_huff;
116 } MJpegDecodeContext;
117
118 int ff_mjpeg_decode_init(AVCodecContext *avctx);
119 int ff_mjpeg_decode_end(AVCodecContext *avctx);
120 int ff_mjpeg_decode_frame(AVCodecContext *avctx,
121 void *data, int *got_frame,
122 AVPacket *avpkt);
123 int ff_mjpeg_decode_dqt(MJpegDecodeContext *s);
124 int ff_mjpeg_decode_dht(MJpegDecodeContext *s);
125 int ff_mjpeg_decode_sof(MJpegDecodeContext *s);
126 int ff_mjpeg_decode_sos(MJpegDecodeContext *s,
127 const uint8_t *mb_bitmask, const AVFrame *reference);
128 int ff_mjpeg_find_marker(MJpegDecodeContext *s,
129 const uint8_t **buf_ptr, const uint8_t *buf_end,
130 const uint8_t **unescaped_buf_ptr, int *unescaped_buf_size);
131
132 #endif /* AVCODEC_MJPEGDEC_H */