avcodec/nvenc: support external context in sw mode
[ffmpeg.git] / libavcodec / cfhd.h
1 /*
2  * Copyright (c) 2015 Kieran Kunhya
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #ifndef AVCODEC_CFHD_H
22 #define AVCODEC_CFHD_H
23
24 #include <stdint.h>
25
26 #include "libavutil/avassert.h"
27
28 #include "avcodec.h"
29 #include "get_bits.h"
30
31 #define VLC_BITS 9
32 #define NB_VLC_TABLE_9 (71+3)
33 #define NB_VLC_TABLE_18 (263+1)
34
35 typedef struct CFHD_RL_VLC_ELEM {
36     int16_t level;
37     int8_t len;
38     uint16_t run;
39 } CFHD_RL_VLC_ELEM;
40
41 #define DWT_LEVELS 3
42
43 typedef struct SubBand {
44     int level;
45     int orientation;
46     int stride;
47     int a_width;
48     int width;
49     int a_height;
50     int height;
51     int pshift;
52     int quant;
53     uint8_t *ibuf;
54 } SubBand;
55
56 typedef struct Plane {
57     int width;
58     int height;
59     ptrdiff_t stride;
60
61     int16_t *idwt_buf;
62     int16_t *idwt_tmp;
63
64     /* TODO: merge this into SubBand structure */
65     int16_t *subband[10];
66     int16_t *l_h[8];
67
68     SubBand band[DWT_LEVELS][4];
69 } Plane;
70
71 typedef struct CFHDContext {
72     AVCodecContext *avctx;
73
74     CFHD_RL_VLC_ELEM table_9_rl_vlc[2088];
75     VLC vlc_9;
76
77     CFHD_RL_VLC_ELEM table_18_rl_vlc[4572];
78     VLC vlc_18;
79
80     GetBitContext gb;
81
82     int chroma_x_shift;
83     int chroma_y_shift;
84
85     int coded_width;
86     int coded_height;
87     int coded_format;
88
89     int a_width;
90     int a_height;
91     int a_format;
92
93     int bpc;
94     int channel_cnt;
95     int subband_cnt;
96     int channel_num;
97     uint8_t lowpass_precision;
98     uint16_t quantisation;
99     int wavelet_depth;
100     int pshift;
101
102     int codebook;
103     int subband_num;
104     int level;
105     int subband_num_actual;
106
107     uint8_t prescale_shift[3];
108     Plane plane[4];
109
110 } CFHDContext;
111
112 int ff_cfhd_init_vlcs(CFHDContext *s);
113
114 #endif /* AVCODEC_CFHD_H */