Mercurial > libavcodec.hg
annotate jpeg_ls.c @ 4072:db5d1a289b56 libavcodec
common subexpression elimination
author | michael |
---|---|
date | Tue, 24 Oct 2006 08:52:51 +0000 |
parents | 00ad06966bff |
children | ce2ce063481c |
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 |
4072 | 251 temp= state->A[Q]; |
252 if(RItype) | |
253 temp += state->N[Q] >> 1; | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
254 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
255 for(k = 0; (state->N[Q] << k) < temp; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
256 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
257 #ifdef JLS_BROKEN |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
258 if(!show_bits_long(gb, 32))return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
259 #endif |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
260 ret = get_ur_golomb_jpegls(gb, k, state->limit - limit_add - 1, state->qbpp); |
2970 | 261 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
262 /* decode mapped error */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
263 map = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
264 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
|
265 map = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
266 ret += RItype + map; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
267 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
268 if(ret & 1){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
269 ret = map - ((ret + 1) >> 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
270 state->B[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
271 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
272 ret = ret >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
273 } |
2970 | 274 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
275 /* update state */ |
4001 | 276 state->A[Q] += FFABS(ret) - RItype; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
277 ret *= state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
278 if(state->N[Q] == state->reset){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
279 state->A[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
280 state->B[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
281 state->N[Q] >>=1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
282 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
283 state->N[Q]++; |
2970 | 284 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
285 return ret; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
286 } |
2970 | 287 |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
288 #define R(a, i ) (bits == 8 ? ((uint8_t*)(a))[i] : ((uint16_t*)(a))[i] ) |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
289 #define W(a, i, v) (bits == 8 ? (((uint8_t*)(a))[i]=v) : (((uint16_t*)(a))[i]=v)) |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
290 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
291 * Decode one line of image |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
292 */ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
293 static inline void ls_decode_line(JLSState *state, MJpegDecodeContext *s, void *last, void *dst, int last2, int w, int stride, int comp, int bits){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
294 int i, x = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
295 int Ra, Rb, Rc, Rd; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
296 int D0, D1, D2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
297 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
298 while(x < w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
299 int err, pred; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
300 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
301 /* compute gradients */ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
302 Ra = x ? R(dst, x - stride) : R(last, x); |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
303 Rb = R(last, x); |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
304 Rc = x ? R(last, x - stride) : last2; |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
305 Rd = (x >= w - stride) ? R(last, x) : R(last, x + stride); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
306 D0 = Rd - Rb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
307 D1 = Rb - Rc; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
308 D2 = Rc - Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
309 /* run mode */ |
4001 | 310 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
|
311 int r; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
312 int RItype; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
313 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
314 /* decode full runs while available */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
315 while(get_bits1(&s->gb)) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
316 int r; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
317 r = 1 << log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
318 if(x + r * stride > w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
319 r = (w - x) / stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
320 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
321 for(i = 0; i < r; i++) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
322 W(dst, x, Ra); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
323 x += stride; |
2970 | 324 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
325 /* if EOL reached, we stop decoding */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
326 if(r != (1 << log2_run[state->run_index[comp]])) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
327 return; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
328 if(state->run_index[comp] < 31) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
329 state->run_index[comp]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
330 if(x + stride > w) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
331 return; |
2970 | 332 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
333 /* decode aborted run */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
334 r = log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
335 if(r) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
336 r = get_bits_long(&s->gb, r); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
337 for(i = 0; i < r; i++) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
338 W(dst, x, Ra); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
339 x += stride; |
2970 | 340 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
341 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
342 /* decode run termination value */ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
343 Rb = R(last, x); |
4001 | 344 RItype = (FFABS(Ra - Rb) <= state->near) ? 1 : 0; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
345 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
|
346 if(state->run_index[comp]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
347 state->run_index[comp]--; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
348 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
349 if(state->near && RItype){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
350 pred = Ra + err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
351 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
352 if(Rb < Ra) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
353 pred = Rb - err; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
354 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
355 pred = Rb + err; |
2970 | 356 } |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
357 } else { /* regular mode */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
358 int context, sign; |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
359 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
360 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
|
361 pred = mid_pred(Ra, Ra + Rb - Rc, Rb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
362 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
363 if(context < 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
364 context = -context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
365 sign = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
366 }else{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
367 sign = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
368 } |
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 if(sign){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
371 pred = clip(pred - state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
372 err = -ls_get_code_regular(&s->gb, state, context); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
373 } else { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
374 pred = clip(pred + state->C[context], 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
375 err = ls_get_code_regular(&s->gb, state, context); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
376 } |
3063
f02d0b59279c
Remove all stray tabs and trailing whitespace, this time for good.
diego
parents:
3036
diff
changeset
|
377 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
378 /* 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
|
379 pred += err; |
4072 | 380 } |
381 if(state->near){ | |
382 if(pred < -state->near) | |
383 pred += state->range * state->twonear; | |
384 else if(pred > state->maxval + state->near) | |
385 pred -= state->range * state->twonear; | |
386 pred = clip(pred, 0, state->maxval); | |
387 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
388 |
4072 | 389 W(dst, x, pred); |
390 x += stride; | |
4069 | 391 } |
392 } | |
393 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
394 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
|
395 int i, t = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
396 uint8_t *zero, *last, *cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
397 JLSState *state; |
4069 | 398 int off = 0, stride = 1, width, shift; |
2970 | 399 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
400 zero = av_mallocz(s->picture.linesize[0]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
401 last = zero; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
402 cur = s->picture.data[0]; |
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 state = av_mallocz(sizeof(JLSState)); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
405 /* initialize JPEG-LS state from JPEG parameters */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
406 state->near = near; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
407 state->bpp = (s->bits < 2) ? 2 : s->bits; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
408 state->maxval = s->maxval; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
409 state->T1 = s->t1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
410 state->T2 = s->t2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
411 state->T3 = s->t3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
412 state->reset = s->reset; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
413 reset_ls_coding_parameters(state, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
414 ls_init_state(state); |
2970 | 415 |
4069 | 416 if(s->bits <= 8) |
417 shift = point_transform + (8 - s->bits); | |
418 else | |
419 shift = point_transform + (16 - s->bits); | |
420 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
421 // 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
|
422 // 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
|
423 if(ilv == 0) { /* separate planes */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
424 off = s->cur_scan - 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
425 stride = (s->nb_components > 1) ? 3 : 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
426 width = s->width * stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
427 cur += off; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
428 for(i = 0; i < s->height; i++) { |
4069 | 429 if(s->bits <= 8){ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
430 ls_decode_line(state, s, last, cur, t, width, stride, off, 8); |
4069 | 431 t = last[0]; |
432 }else{ | |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
433 ls_decode_line(state, s, last, cur, t, width, stride, off, 16); |
4069 | 434 t = *((uint16_t*)last); |
435 } | |
3078
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++) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
451 ls_decode_line(state, s, last + j, cur + j, Rc[j], width, 3, j, 8); |
3078
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 |
4069 | 469 if(shift){ /* we need to do point transform or normalize samples */ |
470 int x, w; | |
471 | |
472 w = s->width * s->nb_components; | |
473 | |
474 if(s->bits <= 8){ | |
475 uint8_t *src = s->picture.data[0]; | |
476 | |
477 for(i = 0; i < s->height; i++){ | |
478 for(x = off; x < w; x+= stride){ | |
479 src[x] <<= shift; | |
480 } | |
481 src += s->picture.linesize[0]; | |
482 } | |
483 }else{ | |
484 uint16_t *src = s->picture.data[0]; | |
485 | |
486 for(i = 0; i < s->height; i++){ | |
487 for(x = 0; x < w; x++){ | |
488 src[x] <<= shift; | |
489 } | |
490 src += s->picture.linesize[0]/2; | |
491 } | |
492 } | |
493 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
494 av_free(state); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
495 av_free(zero); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
496 |
2970 | 497 return 0; |
498 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
499 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
500 #if defined(CONFIG_ENCODERS) && defined(CONFIG_JPEGLS_ENCODER) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
501 /********** Encoder-specific functions **********/ |
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 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
504 * Encode error from regular symbol |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
505 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
506 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
|
507 int k; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
508 int val; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
509 int map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
510 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
511 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
|
512 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
513 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
|
514 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
515 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
516 err += state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
517 if(err >= ((state->range + 1) >> 1)) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
518 err -= state->range; |
4001 | 519 val = 2 * FFABS(err) - 1 - map; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
520 } else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
521 val = 2 * err + map; |
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 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
|
524 |
4001 | 525 state->A[Q] += FFABS(err); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
526 state->B[Q] += err * state->twonear; |
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 if(state->N[Q] == state->reset) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
529 state->A[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
530 state->B[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
531 state->N[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
532 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
533 state->N[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
534 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
535 if(state->B[Q] <= -state->N[Q]) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
536 state->B[Q] += state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
537 if(state->C[Q] > -128) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
538 state->C[Q]--; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
539 if(state->B[Q] <= -state->N[Q]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
540 state->B[Q] = -state->N[Q] + 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
541 }else if(state->B[Q] > 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
542 state->B[Q] -= state->N[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
543 if(state->C[Q] < 127) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
544 state->C[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
545 if(state->B[Q] > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
546 state->B[Q] = 0; |
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 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
549 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
550 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
551 * Encode error from run termination |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
552 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
553 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
|
554 int k; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
555 int val, map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
556 int Q = 365 + RItype; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
557 int temp; |
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 temp = state->A[Q]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
560 if(RItype) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
561 temp += state->N[Q] >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
562 for(k = 0; (state->N[Q] << k) < temp; k++); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
563 map = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
564 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
|
565 map = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
566 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
567 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
568 val = - (2 * err) - 1 - RItype + map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
569 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
570 val = 2 * err - RItype - map; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
571 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
|
572 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
573 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
574 state->B[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
575 state->A[Q] += (val + 1 - RItype) >> 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
576 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
577 if(state->N[Q] == state->reset) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
578 state->A[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
579 state->B[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
580 state->N[Q] >>= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
581 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
582 state->N[Q]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
583 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
584 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
585 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
586 * Encode run value as specified by JPEG-LS standard |
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 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
|
589 while(run >= (1 << log2_run[state->run_index[comp]])){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
590 put_bits(pb, 1, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
591 run -= 1 << log2_run[state->run_index[comp]]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
592 if(state->run_index[comp] < 31) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
593 state->run_index[comp]++; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
594 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
595 /* 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
|
596 if(!trail && run) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
597 put_bits(pb, 1, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
598 }else if(trail){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
599 put_bits(pb, 1, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
600 if(log2_run[state->run_index[comp]]) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
601 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
|
602 } |
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 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
605 /** |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
606 * Encode one line of image |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
607 */ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
608 static inline void ls_encode_line(JLSState *state, PutBitContext *pb, void *last, void *cur, int last2, int w, int stride, int comp, int bits){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
609 int x = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
610 int Ra, Rb, Rc, Rd; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
611 int D0, D1, D2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
612 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
613 while(x < w) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
614 int err, pred, sign; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
615 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
616 /* compute gradients */ |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
617 Ra = x ? R(cur, x - stride) : R(last, x); |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
618 Rb = R(last, x); |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
619 Rc = x ? R(last, x - stride) : last2; |
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
620 Rd = (x >= w - stride) ? R(last, x) : R(last, x + stride); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
621 D0 = Rd - Rb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
622 D1 = Rb - Rc; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
623 D2 = Rc - Ra; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
624 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
625 /* run mode */ |
4001 | 626 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
|
627 int RUNval, RItype, run; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
628 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
629 run = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
630 RUNval = Ra; |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
631 while(x < w && (FFABS(R(cur, x) - RUNval) <= state->near)){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
632 run++; |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
633 W(cur, x, Ra); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
634 x += stride; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
635 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
636 ls_encode_run(state, pb, run, comp, x < w); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
637 if(x >= w) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
638 return; |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
639 Rb = R(last, x); |
4001 | 640 RItype = (FFABS(Ra - Rb) <= state->near); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
641 pred = RItype ? Ra : Rb; |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
642 err = R(cur, x) - pred; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
643 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
644 if(!RItype && Ra > Rb) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
645 err = -err; |
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 if(state->near){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
648 if(err > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
649 err = (state->near + err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
650 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
651 err = -(state->near - err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
652 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
653 if(RItype || (Rb >= Ra)) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
654 Ra = clip(pred + err * state->twonear, 0, state->maxval); |
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 Ra = clip(pred - err * state->twonear, 0, state->maxval); |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
657 W(cur, x, Ra); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
658 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
659 if(err < 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
660 err += state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
661 if(err >= ((state->range + 1) >> 1)) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
662 err -= state->range; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
663 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
664 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
|
665 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
666 if(state->run_index[comp] > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
667 state->run_index[comp]--; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
668 } else { /* regular mode */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
669 int context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
670 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
671 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
|
672 pred = mid_pred(Ra, Ra + Rb - Rc, Rb); |
2970 | 673 |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
674 if(context < 0){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
675 context = -context; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
676 sign = 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
677 pred = clip(pred - state->C[context], 0, state->maxval); |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
678 err = pred - R(cur, x); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
679 }else{ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
680 sign = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
681 pred = clip(pred + state->C[context], 0, state->maxval); |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
682 err = R(cur, x) - pred; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
683 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
684 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
685 if(state->near){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
686 if(err > 0) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
687 err = (state->near + err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
688 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
689 err = -(state->near - err) / state->twonear; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
690 if(!sign) |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
691 Ra = clip(pred + err * state->twonear, 0, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
692 else |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
693 Ra = clip(pred - err * state->twonear, 0, state->maxval); |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
694 W(cur, x, Ra); |
4069 | 695 } |
696 | |
697 ls_encode_regular(state, pb, context, err); | |
698 } | |
4072 | 699 x += stride; |
4069 | 700 } |
701 } | |
702 | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
703 static void ls_store_lse(JLSState *state, PutBitContext *pb){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
704 /* 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
|
705 JLSState state2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
706 memset(&state2, 0, sizeof(JLSState)); |
4069 | 707 state2.bpp = state->bpp; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
708 state2.near = state->near; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
709 reset_ls_coding_parameters(&state2, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
710 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
|
711 return; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
712 /* store LSE type 1 */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
713 put_marker(pb, LSE); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
714 put_bits(pb, 16, 13); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
715 put_bits(pb, 8, 1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
716 put_bits(pb, 16, state->maxval); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
717 put_bits(pb, 16, state->T1); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
718 put_bits(pb, 16, state->T2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
719 put_bits(pb, 16, state->T3); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
720 put_bits(pb, 16, state->reset); |
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 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
723 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
|
724 JpeglsContext * const s = avctx->priv_data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
725 AVFrame *pict = data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
726 AVFrame * const p= (AVFrame*)&s->picture; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
727 const int near = avctx->prediction_method; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
728 PutBitContext pb, pb2; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
729 GetBitContext gb; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
730 uint8_t *buf2, *zero, *cur, *last; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
731 JLSState *state; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
732 int i, size; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
733 int comps; |
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 buf2 = av_malloc(buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
736 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
737 init_put_bits(&pb, buf, buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
738 init_put_bits(&pb2, buf2, buf_size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
739 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
740 *p = *pict; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
741 p->pict_type= FF_I_TYPE; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
742 p->key_frame= 1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
743 |
4069 | 744 if(avctx->pix_fmt == PIX_FMT_GRAY8 || avctx->pix_fmt == PIX_FMT_GRAY16) |
745 comps = 1; | |
746 else | |
747 comps = 3; | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
748 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
749 /* 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
|
750 put_marker(&pb, SOI); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
751 put_marker(&pb, SOF48); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
752 put_bits(&pb, 16, 8 + comps * 3); // header size depends on components |
4069 | 753 put_bits(&pb, 8, (avctx->pix_fmt == PIX_FMT_GRAY16) ? 16 : 8); // bpp |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
754 put_bits(&pb, 16, avctx->height); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
755 put_bits(&pb, 16, avctx->width); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
756 put_bits(&pb, 8, comps); // components |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
757 for(i = 1; i <= comps; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
758 put_bits(&pb, 8, i); // component ID |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
759 put_bits(&pb, 8, 0x11); // subsampling: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
760 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
|
761 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
762 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
763 put_marker(&pb, SOS); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
764 put_bits(&pb, 16, 6 + comps * 2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
765 put_bits(&pb, 8, comps); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
766 for(i = 1; i <= comps; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
767 put_bits(&pb, 8, i); // component ID |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
768 put_bits(&pb, 8, 0); // mapping index: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
769 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
770 put_bits(&pb, 8, near); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
771 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
|
772 put_bits(&pb, 8, 0); // point transform: none |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
773 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
774 state = av_mallocz(sizeof(JLSState)); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
775 /* initialize JPEG-LS state from JPEG parameters */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
776 state->near = near; |
4069 | 777 state->bpp = (avctx->pix_fmt == PIX_FMT_GRAY16) ? 16 : 8; |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
778 reset_ls_coding_parameters(state, 0); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
779 ls_init_state(state); |
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 ls_store_lse(state, &pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
782 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
783 zero = av_mallocz(p->linesize[0]); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
784 last = zero; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
785 cur = p->data[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
786 if(avctx->pix_fmt == PIX_FMT_GRAY8){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
787 int t = 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
788 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
789 for(i = 0; i < avctx->height; i++) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
790 ls_encode_line(state, &pb2, last, cur, t, avctx->width, 1, 0, 8); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
791 t = last[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
792 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
793 cur += p->linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
794 } |
4069 | 795 }else if(avctx->pix_fmt == PIX_FMT_GRAY16){ |
796 int t = 0; | |
797 | |
798 for(i = 0; i < avctx->height; i++) { | |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
799 ls_encode_line(state, &pb2, last, cur, t, avctx->width, 1, 0, 16); |
4069 | 800 t = *((uint16_t*)last); |
801 last = cur; | |
802 cur += p->linesize[0]; | |
803 } | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
804 }else if(avctx->pix_fmt == PIX_FMT_RGB24){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
805 int j, width; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
806 int Rc[3] = {0, 0, 0}; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
807 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
808 width = avctx->width * 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
809 for(i = 0; i < avctx->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
810 for(j = 0; j < 3; j++) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
811 ls_encode_line(state, &pb2, last + j, cur + j, Rc[j], width, 3, j, 8); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
812 Rc[j] = last[j]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
813 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
814 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
815 cur += s->picture.linesize[0]; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
816 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
817 }else if(avctx->pix_fmt == PIX_FMT_BGR24){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
818 int j, width; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
819 int Rc[3] = {0, 0, 0}; |
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 width = avctx->width * 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
822 for(i = 0; i < avctx->height; i++) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
823 for(j = 2; j >= 0; j--) { |
4071
00ad06966bff
merge 8 and 16bit functions, they are identical except the data type
michael
parents:
4069
diff
changeset
|
824 ls_encode_line(state, &pb2, last + j, cur + j, Rc[j], width, 3, j, 8); |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
825 Rc[j] = last[j]; |
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 last = cur; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
828 cur += s->picture.linesize[0]; |
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 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
831 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
832 av_free(zero); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
833 av_free(state); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
834 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
835 flush_put_bits(&pb2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
836 /* do escape coding */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
837 size = put_bits_count(&pb2) >> 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
838 init_get_bits(&gb, buf2, size); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
839 while(get_bits_count(&gb) < size * 8){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
840 int v; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
841 v = get_bits(&gb, 8); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
842 put_bits(&pb, 8, v); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
843 if(v == 0xFF){ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
844 v = get_bits(&gb, 7); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
845 put_bits(&pb, 8, v); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
846 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
847 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
848 align_put_bits(&pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
849 av_free(buf2); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
850 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
851 /* End of image */ |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
852 put_marker(&pb, EOI); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
853 flush_put_bits(&pb); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
854 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
855 emms_c(); |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
856 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
857 return put_bits_count(&pb) >> 3; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
858 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
859 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
860 static int encode_init_ls(AVCodecContext *ctx) { |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
861 JpeglsContext *c = (JpeglsContext*)ctx->priv_data; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
862 |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
863 c->avctx = ctx; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
864 ctx->coded_frame = &c->picture; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
865 |
4069 | 866 if(ctx->pix_fmt != PIX_FMT_GRAY8 && ctx->pix_fmt != PIX_FMT_GRAY16 && ctx->pix_fmt != PIX_FMT_RGB24 && ctx->pix_fmt != PIX_FMT_BGR24){ |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
867 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
|
868 return -1; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
869 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
870 return 0; |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
871 } |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
872 |
2970 | 873 AVCodec jpegls_encoder = { //FIXME avoid MPV_* lossless jpeg shouldnt need them |
874 "jpegls", | |
875 CODEC_TYPE_VIDEO, | |
876 CODEC_ID_JPEGLS, | |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
877 sizeof(JpeglsContext), |
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
878 encode_init_ls, |
2970 | 879 encode_picture_ls, |
3078
1e63c12c2c71
JPEG-LS codec by (Kostya | kostya.forjunk gmail com)
michael
parents:
3063
diff
changeset
|
880 NULL, |
4069 | 881 .pix_fmts= (enum PixelFormat[]){PIX_FMT_BGR24, PIX_FMT_RGB24, PIX_FMT_GRAY8, PIX_FMT_GRAY16, -1}, |
2970 | 882 }; |
883 #endif |