blob: 67a0e4c32c5a8f761262a8b8a73f5dfc3973a02b
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 */ |
115 |