Mercurial > libavcodec.hg
annotate jpeg_ls.c @ 4038:16697bdf4ac2 libavcodec
copy cabac context onto the stack for the c code in decode_cabac_residual() (15% faster on P3 but still slower then the current asm)
author | michael |
---|---|
date | Tue, 17 Oct 2006 23:59:21 +0000 |
parents | 34fdffe98bd0 |
children | 0df780a93aa8 |
rev | line source |
---|---|
2970 | 1 /* |
2 * JPEG-LS encoder and decoder | |
3 * Copyright (c) 2003 Michael Niedermayer | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
4 * Copyright (c) 2006 Konstantin Shishkov |
2970 | 5 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
6 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
7 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
8 * FFmpeg is free software; you can redistribute it and/or |
2970 | 9 * modify it under the terms of the GNU Lesser General Public |
10 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
11 * version 2.1 of the License, or (at your option) any later version. |
2970 | 12 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
13 * FFmpeg is distributed in the hope that it will be useful, |
2970 | 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 * Lesser General Public License for more details. | |
17 * | |
18 * You should have received a copy of the GNU Lesser General Public | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3773
diff
changeset
|
19 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2970
diff
changeset
|
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
2970 | 21 */ |
22 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
23 #include "golomb.h" |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
24 |
2970 | 25 /** |
26 * @file jpeg_ls.c | |
27 * JPEG-LS encoder and decoder. | |
28 */ | |
29 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
30 typedef struct JpeglsContext{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
31 AVCodecContext *avctx; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
32 AVFrame picture; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
33 }JpeglsContext; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
34 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
35 typedef struct JLSState{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
36 int T1, T2, T3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
37 int A[367], B[367], C[365], N[367]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
38 int limit, reset, bpp, qbpp, maxval, range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
39 int near, twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
40 int run_index[3]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
41 }JLSState; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
42 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
43 static const uint8_t log2_run[32]={ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
44 0, 0, 0, 0, 1, 1, 1, 1, |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
45 2, 2, 2, 2, 3, 3, 3, 3, |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
46 4, 4, 5, 5, 6, 6, 7, 7, |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
47 8, 9,10,11,12,13,14,15 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
48 }; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
49 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
50 /* |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
51 * Uncomment this to significantly speed up decoding of broken JPEG-LS |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
52 * (or test broken JPEG-LS decoder) and slow down ordinary decoding a bit. |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
53 * |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
54 * There is no Golomb code with length >= 32 bits possible, so check and |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
55 * avoid situation of 32 zeros, FFmpeg Golomb decoder is painfully slow |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
56 * on this errors. |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
57 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
58 //#define JLS_BROKEN |
2970 | 59 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
60 /********** Functions for both encoder and decoder **********/ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
61 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
62 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
63 * Calculate initial JPEG-LS parameters |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
64 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
65 static void ls_init_state(JLSState *state){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
66 int i; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
67 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
68 state->twonear = state->near * 2 + 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
69 state->range = ((state->maxval + state->twonear - 1) / state->twonear) + 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
70 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
71 // QBPP = ceil(log2(RANGE)) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
72 for(state->qbpp = 0; (1 << state->qbpp) < state->range; state->qbpp++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
73 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
74 if(state->bpp < 8) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
75 state->limit = 16 + 2 * state->bpp - state->qbpp; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
76 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
77 state->limit = (4 * state->bpp) - state->qbpp; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
78 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
79 for(i = 0; i < 367; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
80 state->A[i] = (state->range + 32) >> 6; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
81 if(state->A[i] < 2) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
82 state->A[i] = 2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
83 state->N[i] = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
84 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
85 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
86 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
87 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
88 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
89 * Calculate quantized gradient value, used for context determination |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
90 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
91 static inline int quantize(JLSState *s, int v){ //FIXME optimize |
2970 | 92 if(v==0) return 0; |
93 if(v < 0){ | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
94 if(v <= -s->T3) return -4; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
95 if(v <= -s->T2) return -3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
96 if(v <= -s->T1) return -2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
97 if(v < -s->near) return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
98 return 0; |
2970 | 99 }else{ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
100 if(v <= s->near) return 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
101 if(v < s->T1) return 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
102 if(v < s->T2) return 2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
103 if(v < s->T3) return 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
104 return 4; |
2970 | 105 } |
106 } | |
107 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
108 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
109 * Custom value clipping function used in T1, T2, T3 calculation |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
110 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
111 static inline int iso_clip(int v, int vmin, int vmax){ |
2970 | 112 if(v > vmax || v < vmin) return vmin; |
113 else return v; | |
114 } | |
115 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
116 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
117 * Calculate JPEG-LS codec values |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
118 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
119 static void reset_ls_coding_parameters(JLSState *s, int reset_all){ |
2970 | 120 const int basic_t1= 3; |
121 const int basic_t2= 7; | |
122 const int basic_t3= 21; | |
123 int factor; | |
124 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
125 if(s->maxval==0 || reset_all) s->maxval= (1 << s->bpp) - 1; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
126 |
2970 | 127 if(s->maxval >=128){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
128 factor= (FFMIN(s->maxval, 4095) + 128)>>8; |
2970 | 129 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
130 if(s->T1==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
131 s->T1= iso_clip(factor*(basic_t1-2) + 2 + 3*s->near, s->near+1, s->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
132 if(s->T2==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
133 s->T2= iso_clip(factor*(basic_t2-3) + 3 + 5*s->near, s->T1, s->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
134 if(s->T3==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
135 s->T3= iso_clip(factor*(basic_t3-4) + 4 + 7*s->near, s->T2, s->maxval); |
2970 | 136 }else{ |
137 factor= 256 / (s->maxval + 1); | |
138 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
139 if(s->T1==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
140 s->T1= iso_clip(FFMAX(2, basic_t1/factor + 3*s->near), s->near+1, s->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
141 if(s->T2==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
142 s->T2= iso_clip(FFMAX(3, basic_t2/factor + 5*s->near), s->T1, s->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
143 if(s->T3==0 || reset_all) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
144 s->T3= iso_clip(FFMAX(4, basic_t3/factor + 6*s->near), s->T2, s->maxval); |
2970 | 145 } |
146 | |
147 if(s->reset==0 || reset_all) s->reset= 64; | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
148 // av_log(NULL, AV_LOG_DEBUG, "[JPEG-LS RESET] T=%i,%i,%i\n", s->T1, s->T2, s->T3); |
2970 | 149 } |
150 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
151 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
152 /********** Decoder-specific functions **********/ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
153 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
154 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
155 * Decode LSE block with initialization parameters |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
156 */ |
2970 | 157 static int decode_lse(MJpegDecodeContext *s) |
158 { | |
159 int len, id; | |
160 | |
161 /* XXX: verify len field validity */ | |
162 len = get_bits(&s->gb, 16); | |
163 id = get_bits(&s->gb, 8); | |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
164 |
2970 | 165 switch(id){ |
166 case 1: | |
167 s->maxval= get_bits(&s->gb, 16); | |
168 s->t1= get_bits(&s->gb, 16); | |
169 s->t2= get_bits(&s->gb, 16); | |
170 s->t3= get_bits(&s->gb, 16); | |
171 s->reset= get_bits(&s->gb, 16); | |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
172 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
173 // reset_ls_coding_parameters(s, 0); |
2970 | 174 //FIXME quant table? |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
175 break; |
2970 | 176 case 2: |
177 case 3: | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
178 av_log(s->avctx, AV_LOG_ERROR, "palette not supported\n"); |
2970 | 179 return -1; |
180 case 4: | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
181 av_log(s->avctx, AV_LOG_ERROR, "oversize image not supported\n"); |
2970 | 182 return -1; |
183 default: | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
184 av_log(s->avctx, AV_LOG_ERROR, "invalid id %d\n", id); |
2970 | 185 return -1; |
186 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
187 // av_log(s->avctx, AV_LOG_DEBUG, "ID=%i, T=%i,%i,%i\n", id, s->t1, s->t2, s->t3); |
2970 | 188 |
189 return 0; | |
190 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
191 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
192 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
193 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
194 * Get context-dependent Golomb code, decode it and update context |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
195 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
196 static inline int ls_get_code_regular(GetBitContext *gb, JLSState *state, int Q){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
197 int k, ret; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
198 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
199 for(k = 0; (state->N[Q] << k) < state->A[Q]; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
200 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
201 #ifdef JLS_BROKEN |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
202 if(!show_bits_long(gb, 32))return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
203 #endif |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
204 ret = get_ur_golomb_jpegls(gb, k, state->limit, state->qbpp); |
2970 | 205 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
206 /* decode mapped error */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
207 if(ret & 1) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
208 ret = -((ret + 1) >> 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
209 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
210 ret >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
211 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
212 /* for NEAR=0, k=0 and 2*B[Q] <= - N[Q] mapping is reversed */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
213 if(!state->near && !k && (2 * state->B[Q] <= -state->N[Q])) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
214 ret = -(ret + 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
215 |
4001 | 216 state->A[Q] += FFABS(ret); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
217 ret *= state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
218 state->B[Q] += ret; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
219 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
220 if(state->N[Q] == state->reset) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
221 state->A[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
222 state->B[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
223 state->N[Q] >>= 1; |
2970 | 224 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
225 state->N[Q]++; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
226 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
227 if(state->B[Q] <= -state->N[Q]) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
228 state->B[Q] += state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
229 if(state->C[Q] > -128) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
230 state->C[Q]--; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
231 if(state->B[Q] <= -state->N[Q]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
232 state->B[Q] = -state->N[Q] + 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
233 }else if(state->B[Q] > 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
234 state->B[Q] -= state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
235 if(state->C[Q] < 127) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
236 state->C[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
237 if(state->B[Q] > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
238 state->B[Q] = 0; |
2970 | 239 } |
240 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
241 return ret; |
2970 | 242 } |
243 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
244 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
245 * Get Golomb code, decode it and update state for run termination |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
246 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
247 static inline int ls_get_code_runterm(GetBitContext *gb, JLSState *state, int RItype, int limit_add){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
248 int k, ret, temp, map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
249 int Q = 365 + RItype; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
250 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
251 if(!RItype) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
252 temp = state->A[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
253 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
254 temp = state->A[Q] + (state->N[Q] >> 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
255 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
256 for(k = 0; (state->N[Q] << k) < temp; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
257 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
258 #ifdef JLS_BROKEN |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
259 if(!show_bits_long(gb, 32))return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
260 #endif |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
261 ret = get_ur_golomb_jpegls(gb, k, state->limit - limit_add - 1, state->qbpp); |
2970 | 262 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
263 /* decode mapped error */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
264 map = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
265 if(!k && (RItype || ret) && (2 * state->B[Q] < state->N[Q])) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
266 map = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
267 ret += RItype + map; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
268 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
269 if(ret & 1){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
270 ret = map - ((ret + 1) >> 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
271 state->B[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
272 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
273 ret = ret >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
274 } |
2970 | 275 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
276 /* update state */ |
4001 | 277 state->A[Q] += FFABS(ret) - RItype; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
278 ret *= state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
279 if(state->N[Q] == state->reset){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
280 state->A[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
281 state->B[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
282 state->N[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
283 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
284 state->N[Q]++; |
2970 | 285 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
286 return ret; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
287 } |
2970 | 288 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
289 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
290 * Decode one line of image |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
291 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
292 static inline void ls_decode_line(JLSState *state, MJpegDecodeContext *s, uint8_t *last, uint8_t *dst, int last2, int w, int stride, int comp){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
293 int i, x = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
294 int Ra, Rb, Rc, Rd; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
295 int D0, D1, D2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
296 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
297 while(x < w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
298 int err, pred; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
299 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
300 /* compute gradients */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
301 Ra = x ? dst[x - stride] : last[x]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
302 Rb = last[x]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
303 Rc = x ? last[x - stride] : last2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
304 Rd = (x >= w - stride) ? last[x] : last[x + stride]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
305 D0 = Rd - Rb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
306 D1 = Rb - Rc; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
307 D2 = Rc - Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
308 /* run mode */ |
4001 | 309 if((FFABS(D0) <= state->near) && (FFABS(D1) <= state->near) && (FFABS(D2) <= state->near)) { |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
310 int r; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
311 int RItype; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
312 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
313 /* decode full runs while available */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
314 while(get_bits1(&s->gb)) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
315 int r; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
316 r = 1 << log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
317 if(x + r * stride > w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
318 r = (w - x) / stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
319 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
320 for(i = 0; i < r; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
321 dst[x] = Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
322 x += stride; |
2970 | 323 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
324 /* if EOL reached, we stop decoding */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
325 if(r != (1 << log2_run[state->run_index[comp]])) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
326 return; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
327 if(state->run_index[comp] < 31) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
328 state->run_index[comp]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
329 if(x + stride > w) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
330 return; |
2970 | 331 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
332 /* decode aborted run */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
333 r = log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
334 if(r) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
335 r = get_bits_long(&s->gb, r); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
336 for(i = 0; i < r; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
337 dst[x] = Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
338 x += stride; |
2970 | 339 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
340 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
341 /* decode run termination value */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
342 Rb = last[x]; |
4001 | 343 RItype = (FFABS(Ra - Rb) <= state->near) ? 1 : 0; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
344 err = ls_get_code_runterm(&s->gb, state, RItype, log2_run[state->run_index[comp]]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
345 if(state->run_index[comp]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
346 state->run_index[comp]--; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
347 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
348 if(state->near && RItype){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
349 pred = Ra + err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
350 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
351 if(Rb < Ra) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
352 pred = Rb - err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
353 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
354 pred = Rb + err; |
2970 | 355 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
356 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
357 if(state->near){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
358 if(pred < -state->near) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
359 pred += state->range * state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
360 else if(pred > state->maxval + state->near) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
361 pred -= state->range * state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
362 pred = clip(pred, 0, state->maxval); |
2970 | 363 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
364 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
365 dst[x] = pred; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
366 x += stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
367 } else { /* regular mode */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
368 int context, sign; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
369 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
370 context = quantize(state, D0) * 81 + quantize(state, D1) * 9 + quantize(state, D2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
371 pred = mid_pred(Ra, Ra + Rb - Rc, Rb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
372 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
373 if(context < 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
374 context = -context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
375 sign = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
376 }else{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
377 sign = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
378 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
379 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
380 if(sign){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
381 pred = clip(pred - state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
382 err = -ls_get_code_regular(&s->gb, state, context); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
383 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
384 pred = clip(pred + state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
385 err = ls_get_code_regular(&s->gb, state, context); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
386 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
387 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
388 /* we have to do something more for near-lossless coding */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
389 pred += err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
390 if(state->near) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
391 if(pred < -state->near) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
392 pred += state->range * state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
393 else if(pred > state->maxval + state->near) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
394 pred -= state->range * state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
395 pred = clip(pred, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
396 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
397 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
398 dst[x] = pred; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
399 x += stride; |
2970 | 400 } |
401 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
402 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
403 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
404 static int ls_decode_picture(MJpegDecodeContext *s, int near, int point_transform, int ilv){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
405 int i, t = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
406 uint8_t *zero, *last, *cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
407 JLSState *state; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
408 int off, stride, width; |
2970 | 409 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
410 zero = av_mallocz(s->picture.linesize[0]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
411 last = zero; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
412 cur = s->picture.data[0]; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
413 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
414 state = av_mallocz(sizeof(JLSState)); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
415 /* initialize JPEG-LS state from JPEG parameters */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
416 state->near = near; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
417 state->bpp = (s->bits < 2) ? 2 : s->bits; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
418 state->maxval = s->maxval; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
419 state->T1 = s->t1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
420 state->T2 = s->t2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
421 state->T3 = s->t3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
422 state->reset = s->reset; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
423 reset_ls_coding_parameters(state, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
424 ls_init_state(state); |
2970 | 425 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
426 // av_log(s->avctx, AV_LOG_DEBUG, "JPEG-LS params: %ix%i NEAR=%i MV=%i T(%i,%i,%i) RESET=%i, LIMIT=%i, qbpp=%i, RANGE=%i\n",s->width,s->height,state->near,state->maxval,state->T1,state->T2,state->T3,state->reset,state->limit,state->qbpp, state->range); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
427 // av_log(s->avctx, AV_LOG_DEBUG, "JPEG params: ILV=%i Pt=%i BPP=%i, scan = %i\n", ilv, point_transform, s->bits, s->cur_scan); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
428 if(ilv == 0) { /* separate planes */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
429 off = s->cur_scan - 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
430 stride = (s->nb_components > 1) ? 3 : 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
431 width = s->width * stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
432 cur += off; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
433 for(i = 0; i < s->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
434 ls_decode_line(state, s, last, cur, t, width, stride, off); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
435 t = last[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
436 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
437 cur += s->picture.linesize[0]; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
438 |
2970 | 439 if (s->restart_interval && !--s->restart_count) { |
440 align_get_bits(&s->gb); | |
441 skip_bits(&s->gb, 16); /* skip RSTn */ | |
442 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
443 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
444 } else if(ilv == 1) { /* line interleaving */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
445 int j; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
446 int Rc[3] = {0, 0, 0}; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
447 memset(cur, 0, s->picture.linesize[0]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
448 width = s->width * 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
449 for(i = 0; i < s->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
450 for(j = 0; j < 3; j++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
451 ls_decode_line(state, s, last + j, cur + j, Rc[j], width, 3, j); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
452 Rc[j] = last[j]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
453 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
454 if (s->restart_interval && !--s->restart_count) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
455 align_get_bits(&s->gb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
456 skip_bits(&s->gb, 16); /* skip RSTn */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
457 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
458 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
459 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
460 cur += s->picture.linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
461 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
462 } else if(ilv == 2) { /* sample interleaving */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
463 av_log(s->avctx, AV_LOG_ERROR, "Sample interleaved images are not supported.\n"); |
3773 | 464 av_free(state); |
465 av_free(zero); | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
466 return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
467 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
468 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
469 av_free(state); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
470 av_free(zero); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
471 |
2970 | 472 return 0; |
473 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
474 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
475 #if defined(CONFIG_ENCODERS) && defined(CONFIG_JPEGLS_ENCODER) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
476 /********** Encoder-specific functions **********/ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
477 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
478 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
479 * Encode error from regular symbol |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
480 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
481 static inline void ls_encode_regular(JLSState *state, PutBitContext *pb, int Q, int err){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
482 int k; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
483 int val; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
484 int map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
485 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
486 for(k = 0; (state->N[Q] << k) < state->A[Q]; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
487 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
488 map = !state->near && !k && (2 * state->B[Q] <= -state->N[Q]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
489 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
490 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
491 err += state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
492 if(err >= ((state->range + 1) >> 1)) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
493 err -= state->range; |
4001 | 494 val = 2 * FFABS(err) - 1 - map; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
495 } else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
496 val = 2 * err + map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
497 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
498 set_ur_golomb_jpegls(pb, val, k, state->limit, state->qbpp); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
499 |
4001 | 500 state->A[Q] += FFABS(err); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
501 state->B[Q] += err * state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
502 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
503 if(state->N[Q] == state->reset) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
504 state->A[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
505 state->B[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
506 state->N[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
507 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
508 state->N[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
509 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
510 if(state->B[Q] <= -state->N[Q]) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
511 state->B[Q] += state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
512 if(state->C[Q] > -128) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
513 state->C[Q]--; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
514 if(state->B[Q] <= -state->N[Q]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
515 state->B[Q] = -state->N[Q] + 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
516 }else if(state->B[Q] > 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
517 state->B[Q] -= state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
518 if(state->C[Q] < 127) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
519 state->C[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
520 if(state->B[Q] > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
521 state->B[Q] = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
522 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
523 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
524 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
525 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
526 * Encode error from run termination |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
527 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
528 static inline void ls_encode_runterm(JLSState *state, PutBitContext *pb, int RItype, int err, int limit_add){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
529 int k; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
530 int val, map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
531 int Q = 365 + RItype; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
532 int temp; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
533 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
534 temp = state->A[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
535 if(RItype) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
536 temp += state->N[Q] >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
537 for(k = 0; (state->N[Q] << k) < temp; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
538 map = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
539 if(!k && err && (2 * state->B[Q] < state->N[Q])) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
540 map = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
541 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
542 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
543 val = - (2 * err) - 1 - RItype + map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
544 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
545 val = 2 * err - RItype - map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
546 set_ur_golomb_jpegls(pb, val, k, state->limit - limit_add - 1, state->qbpp); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
547 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
548 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
549 state->B[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
550 state->A[Q] += (val + 1 - RItype) >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
551 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
552 if(state->N[Q] == state->reset) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
553 state->A[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
554 state->B[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
555 state->N[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
556 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
557 state->N[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
558 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
559 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
560 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
561 * Encode run value as specified by JPEG-LS standard |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
562 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
563 static inline void ls_encode_run(JLSState *state, PutBitContext *pb, int run, int comp, int trail){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
564 while(run >= (1 << log2_run[state->run_index[comp]])){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
565 put_bits(pb, 1, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
566 run -= 1 << log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
567 if(state->run_index[comp] < 31) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
568 state->run_index[comp]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
569 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
570 /* if hit EOL, encode another full run, else encode aborted run */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
571 if(!trail && run) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
572 put_bits(pb, 1, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
573 }else if(trail){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
574 put_bits(pb, 1, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
575 if(log2_run[state->run_index[comp]]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
576 put_bits(pb, log2_run[state->run_index[comp]], run); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
577 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
578 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
579 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
580 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
581 * Encode one line of image |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
582 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
583 static inline void ls_encode_line(JLSState *state, PutBitContext *pb, uint8_t *last, uint8_t *cur, int last2, int w, int stride, int comp){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
584 int x = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
585 int Ra, Rb, Rc, Rd; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
586 int D0, D1, D2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
587 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
588 while(x < w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
589 int err, pred, sign; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
590 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
591 /* compute gradients */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
592 Ra = x ? cur[x - stride] : last[x]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
593 Rb = last[x]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
594 Rc = x ? last[x - stride] : last2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
595 Rd = (x >= w - stride) ? last[x] : last[x + stride]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
596 D0 = Rd - Rb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
597 D1 = Rb - Rc; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
598 D2 = Rc - Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
599 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
600 /* run mode */ |
4001 | 601 if((FFABS(D0) <= state->near) && (FFABS(D1) <= state->near) && (FFABS(D2) <= state->near)) { |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
602 int RUNval, RItype, run; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
603 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
604 run = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
605 RUNval = Ra; |
4001 | 606 while(x < w && (FFABS(cur[x] - RUNval) <= state->near)){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
607 run++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
608 cur[x] = Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
609 x += stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
610 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
611 ls_encode_run(state, pb, run, comp, x < w); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
612 if(x >= w) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
613 return; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
614 Rb = last[x]; |
4001 | 615 RItype = (FFABS(Ra - Rb) <= state->near); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
616 pred = RItype ? Ra : Rb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
617 err = cur[x] - pred; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
618 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
619 if(!RItype && Ra > Rb) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
620 err = -err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
621 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
622 if(state->near){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
623 if(err > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
624 err = (state->near + err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
625 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
626 err = -(state->near - err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
627 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
628 if(RItype || (Rb >= Ra)) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
629 Ra = clip(pred + err * state->twonear, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
630 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
631 Ra = clip(pred - err * state->twonear, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
632 cur[x] = Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
633 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
634 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
635 err += state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
636 if(err >= ((state->range + 1) >> 1)) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
637 err -= state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
638 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
639 ls_encode_runterm(state, pb, RItype, err, log2_run[state->run_index[comp]]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
640 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
641 if(state->run_index[comp] > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
642 state->run_index[comp]--; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
643 x += stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
644 } else { /* regular mode */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
645 int context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
646 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
647 context = quantize(state, D0) * 81 + quantize(state, D1) * 9 + quantize(state, D2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
648 pred = mid_pred(Ra, Ra + Rb - Rc, Rb); |
2970 | 649 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
650 if(context < 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
651 context = -context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
652 sign = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
653 pred = clip(pred - state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
654 err = pred - cur[x]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
655 }else{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
656 sign = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
657 pred = clip(pred + state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
658 err = cur[x] - pred; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
659 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
660 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
661 if(state->near){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
662 if(err > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
663 err = (state->near + err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
664 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
665 err = -(state->near - err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
666 if(!sign) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
667 Ra = clip(pred + err * state->twonear, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
668 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
669 Ra = clip(pred - err * state->twonear, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
670 cur[x] = Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
671 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
672 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
673 ls_encode_regular(state, pb, context, err); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
674 x += stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
675 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
676 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
677 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
678 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
679 static void ls_store_lse(JLSState *state, PutBitContext *pb){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
680 /* Test if we have default params and don't need to store LSE */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
681 JLSState state2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
682 memset(&state2, 0, sizeof(JLSState)); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
683 state2.bpp = 8; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
684 state2.near = state->near; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
685 reset_ls_coding_parameters(&state2, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
686 if(state->T1 == state2.T1 && state->T2 == state2.T2 && state->T3 == state2.T3 && state->reset == state2.reset) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
687 return; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
688 /* store LSE type 1 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
689 put_marker(pb, LSE); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
690 put_bits(pb, 16, 13); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
691 put_bits(pb, 8, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
692 put_bits(pb, 16, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
693 put_bits(pb, 16, state->T1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
694 put_bits(pb, 16, state->T2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
695 put_bits(pb, 16, state->T3); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
696 put_bits(pb, 16, state->reset); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
697 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
698 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
699 static int encode_picture_ls(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
700 JpeglsContext * const s = avctx->priv_data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
701 AVFrame *pict = data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
702 AVFrame * const p= (AVFrame*)&s->picture; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
703 const int near = avctx->prediction_method; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
704 PutBitContext pb, pb2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
705 GetBitContext gb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
706 uint8_t *buf2, *zero, *cur, *last; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
707 JLSState *state; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
708 int i, size; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
709 int comps; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
710 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
711 buf2 = av_malloc(buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
712 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
713 init_put_bits(&pb, buf, buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
714 init_put_bits(&pb2, buf2, buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
715 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
716 *p = *pict; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
717 p->pict_type= FF_I_TYPE; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
718 p->key_frame= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
719 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
720 comps = (avctx->pix_fmt == PIX_FMT_GRAY8) ? 1 : 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
721 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
722 /* write our own JPEG header, can't use mjpeg_picture_header */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
723 put_marker(&pb, SOI); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
724 put_marker(&pb, SOF48); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
725 put_bits(&pb, 16, 8 + comps * 3); // header size depends on components |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
726 put_bits(&pb, 8, 8); // bpp |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
727 put_bits(&pb, 16, avctx->height); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
728 put_bits(&pb, 16, avctx->width); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
729 put_bits(&pb, 8, comps); // components |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
730 for(i = 1; i <= comps; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
731 put_bits(&pb, 8, i); // component ID |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
732 put_bits(&pb, 8, 0x11); // subsampling: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
733 put_bits(&pb, 8, 0); // Tiq, used by JPEG-LS ext |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
734 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
735 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
736 put_marker(&pb, SOS); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
737 put_bits(&pb, 16, 6 + comps * 2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
738 put_bits(&pb, 8, comps); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
739 for(i = 1; i <= comps; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
740 put_bits(&pb, 8, i); // component ID |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
741 put_bits(&pb, 8, 0); // mapping index: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
742 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
743 put_bits(&pb, 8, near); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
744 put_bits(&pb, 8, (comps > 1) ? 1 : 0); // interleaving: 0 - plane, 1 - line |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
745 put_bits(&pb, 8, 0); // point transform: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
746 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
747 state = av_mallocz(sizeof(JLSState)); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
748 /* initialize JPEG-LS state from JPEG parameters */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
749 state->near = near; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
750 state->bpp = 8; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
751 reset_ls_coding_parameters(state, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
752 ls_init_state(state); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
753 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
754 ls_store_lse(state, &pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
755 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
756 zero = av_mallocz(p->linesize[0]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
757 last = zero; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
758 cur = p->data[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
759 if(avctx->pix_fmt == PIX_FMT_GRAY8){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
760 int t = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
761 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
762 for(i = 0; i < avctx->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
763 ls_encode_line(state, &pb2, last, cur, t, avctx->width, 1, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
764 t = last[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
765 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
766 cur += p->linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
767 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
768 }else if(avctx->pix_fmt == PIX_FMT_RGB24){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
769 int j, width; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
770 int Rc[3] = {0, 0, 0}; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
771 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
772 width = avctx->width * 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
773 for(i = 0; i < avctx->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
774 for(j = 0; j < 3; j++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
775 ls_encode_line(state, &pb2, last + j, cur + j, Rc[j], width, 3, j); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
776 Rc[j] = last[j]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
777 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
778 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
779 cur += s->picture.linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
780 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
781 }else if(avctx->pix_fmt == PIX_FMT_BGR24){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
782 int j, width; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
783 int Rc[3] = {0, 0, 0}; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
784 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
785 width = avctx->width * 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
786 for(i = 0; i < avctx->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
787 for(j = 2; j >= 0; j--) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
788 ls_encode_line(state, &pb2, last + j, cur + j, Rc[j], width, 3, j); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
789 Rc[j] = last[j]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
790 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
791 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
792 cur += s->picture.linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
793 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
794 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
795 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
796 av_free(zero); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
797 av_free(state); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
798 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
799 flush_put_bits(&pb2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
800 /* do escape coding */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
801 size = put_bits_count(&pb2) >> 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
802 init_get_bits(&gb, buf2, size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
803 while(get_bits_count(&gb) < size * 8){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
804 int v; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
805 v = get_bits(&gb, 8); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
806 put_bits(&pb, 8, v); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
807 if(v == 0xFF){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
808 v = get_bits(&gb, 7); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
809 put_bits(&pb, 8, v); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
810 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
811 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
812 align_put_bits(&pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
813 av_free(buf2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
814 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
815 /* End of image */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
816 put_marker(&pb, EOI); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
817 flush_put_bits(&pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
818 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
819 emms_c(); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
820 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
821 return put_bits_count(&pb) >> 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
822 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
823 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
824 static int encode_init_ls(AVCodecContext *ctx) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
825 JpeglsContext *c = (JpeglsContext*)ctx->priv_data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
826 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
827 c->avctx = ctx; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
828 ctx->coded_frame = &c->picture; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
829 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
830 if(ctx->pix_fmt != PIX_FMT_GRAY8 && ctx->pix_fmt != PIX_FMT_RGB24 && ctx->pix_fmt != PIX_FMT_BGR24){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
831 av_log(ctx, AV_LOG_ERROR, "Only grayscale and RGB24/BGR24 images are supported\n"); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
832 return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
833 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
834 return 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
835 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
836 |
2970 | 837 AVCodec jpegls_encoder = { //FIXME avoid MPV_* lossless jpeg shouldnt need them |
838 "jpegls", | |
839 CODEC_TYPE_VIDEO, | |
840 CODEC_ID_JPEGLS, | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
841 sizeof(JpeglsContext), |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
842 encode_init_ls, |
2970 | 843 encode_picture_ls, |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
844 NULL, |
3079 | 845 .pix_fmts= (enum PixelFormat[]){PIX_FMT_BGR24, PIX_FMT_RGB24, PIX_FMT_GRAY8, -1}, |
2970 | 846 }; |
847 #endif |