Mercurial > libavcodec.hg
annotate loco.c @ 8917:422a9dbebb53 libavcodec
Almost cosmetics.
Move some assignments after the checks.
author | iive |
---|---|
date | Sun, 15 Feb 2009 10:14:16 +0000 |
parents | e9d9d946f213 |
children | 54bc8a2727b0 |
rev | line source |
---|---|
2530 | 1 /* |
2 * LOCO codec | |
3 * Copyright (c) 2005 Konstantin Shishkov | |
4 * | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
2530 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * 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:
3036
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
2530 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
2530 | 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * 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:
3036
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2967
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
2530 | 20 */ |
2967 | 21 |
2530 | 22 /** |
8718
e9d9d946f213
Use full internal pathname in doxygen @file directives.
diego
parents:
8627
diff
changeset
|
23 * @file libavcodec/loco.c |
2530 | 24 * LOCO codec. |
25 */ | |
2967 | 26 |
2530 | 27 #include "avcodec.h" |
28 #include "bitstream.h" | |
29 #include "golomb.h" | |
8627
d6bab465b82c
moves mid_pred() into mathops.h (with arch specific code split by directory)
aurel
parents:
7040
diff
changeset
|
30 #include "mathops.h" |
2530 | 31 |
2587 | 32 enum LOCO_MODE {LOCO_UNKN=0, LOCO_CYUY2=-1, LOCO_CRGB=-2, LOCO_CRGBA=-3, LOCO_CYV12=-4, |
2530 | 33 LOCO_YUY2=1, LOCO_UYVY=2, LOCO_RGB=3, LOCO_RGBA=4, LOCO_YV12=5}; |
34 | |
35 typedef struct LOCOContext{ | |
36 AVCodecContext *avctx; | |
37 AVFrame pic; | |
38 int lossy; | |
39 int mode; | |
40 } LOCOContext; | |
41 | |
42 typedef struct RICEContext{ | |
43 GetBitContext gb; | |
44 int save, run, run2; /* internal rice decoder state */ | |
45 int sum, count; /* sum and count for getting rice parameter */ | |
2558 | 46 int lossy; |
2530 | 47 }RICEContext; |
48 | |
49 static int loco_get_rice_param(RICEContext *r) | |
50 { | |
51 int cnt = 0; | |
52 int val = r->count; | |
2967 | 53 |
2530 | 54 while(r->sum > val && cnt < 9) { |
55 val <<= 1; | |
56 cnt++; | |
57 } | |
2967 | 58 |
2530 | 59 return cnt; |
60 } | |
61 | |
62 static inline void loco_update_rice_param(RICEContext *r, int val) | |
63 { | |
64 r->sum += val; | |
65 r->count++; | |
2967 | 66 |
2530 | 67 if(r->count == 16) { |
68 r->sum >>= 1; | |
69 r->count >>= 1; | |
70 } | |
71 } | |
72 | |
73 static inline int loco_get_rice(RICEContext *r) | |
74 { | |
75 int v; | |
76 if (r->run > 0) { /* we have zero run */ | |
77 r->run--; | |
2558 | 78 loco_update_rice_param(r, 0); |
2530 | 79 return 0; |
80 } | |
2558 | 81 v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0); |
82 loco_update_rice_param(r, (v+1)>>1); | |
2530 | 83 if (!v) { |
84 if (r->save >= 0) { | |
2558 | 85 r->run = get_ur_golomb_jpegls(&r->gb, 2, INT_MAX, 0); |
2530 | 86 if(r->run > 1) |
87 r->save += r->run + 1; | |
88 else | |
89 r->save -= 3; | |
90 } | |
91 else | |
92 r->run2++; | |
2558 | 93 } else { |
94 v = ((v>>1) + r->lossy) ^ -(v&1); | |
95 if (r->run2 > 0) { | |
96 if (r->run2 > 2) | |
97 r->save += r->run2; | |
98 else | |
99 r->save -= 3; | |
100 r->run2 = 0; | |
101 } | |
2530 | 102 } |
2967 | 103 |
2530 | 104 return v; |
105 } | |
106 | |
107 /* LOCO main predictor - LOCO-I/JPEG-LS predictor */ | |
108 static inline int loco_predict(uint8_t* data, int stride, int step) | |
109 { | |
110 int a, b, c; | |
2967 | 111 |
2530 | 112 a = data[-stride]; |
113 b = data[-step]; | |
114 c = data[-stride - step]; | |
2967 | 115 |
2558 | 116 return mid_pred(a, a + b - c, b); |
2530 | 117 } |
118 | |
119 static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int height, | |
6218 | 120 int stride, const uint8_t *buf, int buf_size, int step) |
2530 | 121 { |
122 RICEContext rc; | |
123 int val; | |
124 int i, j; | |
2967 | 125 |
2530 | 126 init_get_bits(&rc.gb, buf, buf_size*8); |
127 rc.save = 0; | |
128 rc.run = 0; | |
129 rc.run2 = 0; | |
2967 | 130 rc.lossy = l->lossy; |
131 | |
2530 | 132 rc.sum = 8; |
133 rc.count = 1; | |
2967 | 134 |
2530 | 135 /* restore top left pixel */ |
136 val = loco_get_rice(&rc); | |
137 data[0] = 128 + val; | |
138 /* restore top line */ | |
139 for (i = 1; i < width; i++) { | |
140 val = loco_get_rice(&rc); | |
141 data[i * step] = data[i * step - step] + val; | |
142 } | |
143 data += stride; | |
144 for (j = 1; j < height; j++) { | |
145 /* restore left column */ | |
146 val = loco_get_rice(&rc); | |
147 data[0] = data[-stride] + val; | |
148 /* restore all other pixels */ | |
149 for (i = 1; i < width; i++) { | |
150 val = loco_get_rice(&rc); | |
151 data[i * step] = loco_predict(&data[i * step], stride, step) + val; | |
152 } | |
153 data += stride; | |
154 } | |
2967 | 155 |
6750 | 156 return (get_bits_count(&rc.gb) + 7) >> 3; |
2530 | 157 } |
158 | |
2967 | 159 static int decode_frame(AVCodecContext *avctx, |
2530 | 160 void *data, int *data_size, |
6218 | 161 const uint8_t *buf, int buf_size) |
2530 | 162 { |
163 LOCOContext * const l = avctx->priv_data; | |
164 AVFrame * const p= (AVFrame*)&l->pic; | |
165 int decoded; | |
166 | |
167 if(p->data[0]) | |
168 avctx->release_buffer(avctx, p); | |
169 | |
170 p->reference = 0; | |
171 if(avctx->get_buffer(avctx, p) < 0){ | |
172 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | |
173 return -1; | |
174 } | |
175 p->key_frame = 1; | |
176 | |
177 switch(l->mode) { | |
178 case LOCO_CYUY2: case LOCO_YUY2: case LOCO_UYVY: | |
179 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height, | |
180 p->linesize[0], buf, buf_size, 1); | |
181 buf += decoded; buf_size -= decoded; | |
182 decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height, | |
183 p->linesize[1], buf, buf_size, 1); | |
184 buf += decoded; buf_size -= decoded; | |
185 decoded = loco_decode_plane(l, p->data[2], avctx->width / 2, avctx->height, | |
186 p->linesize[2], buf, buf_size, 1); | |
187 break; | |
188 case LOCO_CYV12: case LOCO_YV12: | |
189 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height, | |
190 p->linesize[0], buf, buf_size, 1); | |
191 buf += decoded; buf_size -= decoded; | |
2598
6eaebf1fbd74
Fix colors for YV12 case (u/v planes are swapped)
rtognimp
parents:
2587
diff
changeset
|
192 decoded = loco_decode_plane(l, p->data[2], avctx->width / 2, avctx->height / 2, |
6eaebf1fbd74
Fix colors for YV12 case (u/v planes are swapped)
rtognimp
parents:
2587
diff
changeset
|
193 p->linesize[2], buf, buf_size, 1); |
6eaebf1fbd74
Fix colors for YV12 case (u/v planes are swapped)
rtognimp
parents:
2587
diff
changeset
|
194 buf += decoded; buf_size -= decoded; |
2530 | 195 decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height / 2, |
196 p->linesize[1], buf, buf_size, 1); | |
197 break; | |
198 case LOCO_CRGB: case LOCO_RGB: | |
2599
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
199 decoded = loco_decode_plane(l, p->data[0] + p->linesize[0]*(avctx->height-1), avctx->width, avctx->height, |
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
200 -p->linesize[0], buf, buf_size, 3); |
2530 | 201 buf += decoded; buf_size -= decoded; |
2599
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
202 decoded = loco_decode_plane(l, p->data[0] + p->linesize[0]*(avctx->height-1) + 1, avctx->width, avctx->height, |
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
203 -p->linesize[0], buf, buf_size, 3); |
2530 | 204 buf += decoded; buf_size -= decoded; |
2599
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
205 decoded = loco_decode_plane(l, p->data[0] + p->linesize[0]*(avctx->height-1) + 2, avctx->width, avctx->height, |
3e36a706f5b7
Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents:
2598
diff
changeset
|
206 -p->linesize[0], buf, buf_size, 3); |
2530 | 207 break; |
208 case LOCO_RGBA: | |
209 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height, | |
210 p->linesize[0], buf, buf_size, 4); | |
211 buf += decoded; buf_size -= decoded; | |
212 decoded = loco_decode_plane(l, p->data[0] + 1, avctx->width, avctx->height, | |
213 p->linesize[0], buf, buf_size, 4); | |
214 buf += decoded; buf_size -= decoded; | |
215 decoded = loco_decode_plane(l, p->data[0] + 2, avctx->width, avctx->height, | |
216 p->linesize[0], buf, buf_size, 4); | |
217 buf += decoded; buf_size -= decoded; | |
218 decoded = loco_decode_plane(l, p->data[0] + 3, avctx->width, avctx->height, | |
219 p->linesize[0], buf, buf_size, 4); | |
220 break; | |
221 } | |
222 | |
223 *data_size = sizeof(AVFrame); | |
224 *(AVFrame*)data = l->pic; | |
2967 | 225 |
2530 | 226 return buf_size; |
227 } | |
228 | |
6517
48759bfbd073
Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents:
6218
diff
changeset
|
229 static av_cold int decode_init(AVCodecContext *avctx){ |
2530 | 230 LOCOContext * const l = avctx->priv_data; |
231 int version; | |
232 | |
233 l->avctx = avctx; | |
234 if (avctx->extradata_size < 12) { | |
235 av_log(avctx, AV_LOG_ERROR, "Extradata size must be >= 12 instead of %i\n", | |
236 avctx->extradata_size); | |
237 return -1; | |
238 } | |
4364 | 239 version = AV_RL32(avctx->extradata); |
2530 | 240 switch(version) { |
241 case 1: | |
242 l->lossy = 0; | |
243 break; | |
244 case 2: | |
4364 | 245 l->lossy = AV_RL32(avctx->extradata + 8); |
2530 | 246 break; |
247 default: | |
4364 | 248 l->lossy = AV_RL32(avctx->extradata + 8); |
2530 | 249 av_log(avctx, AV_LOG_INFO, "This is LOCO codec version %i, please upload file for study\n", version); |
250 } | |
2967 | 251 |
4364 | 252 l->mode = AV_RL32(avctx->extradata + 4); |
2530 | 253 switch(l->mode) { |
254 case LOCO_CYUY2: case LOCO_YUY2: case LOCO_UYVY: | |
255 avctx->pix_fmt = PIX_FMT_YUV422P; | |
256 break; | |
257 case LOCO_CRGB: case LOCO_RGB: | |
258 avctx->pix_fmt = PIX_FMT_BGR24; | |
259 break; | |
260 case LOCO_CYV12: case LOCO_YV12: | |
261 avctx->pix_fmt = PIX_FMT_YUV420P; | |
262 break; | |
2587 | 263 case LOCO_CRGBA: case LOCO_RGBA: |
4494
ce643a22f049
Replace deprecated PIX_FMT names by the newer variants.
diego
parents:
4364
diff
changeset
|
264 avctx->pix_fmt = PIX_FMT_RGB32; |
2530 | 265 break; |
266 default: | |
267 av_log(avctx, AV_LOG_INFO, "Unknown colorspace, index = %i\n", l->mode); | |
268 return -1; | |
269 } | |
2558 | 270 if(avctx->debug & FF_DEBUG_PICT_INFO) |
271 av_log(avctx, AV_LOG_INFO, "lossy:%i, version:%i, mode: %i\n", l->lossy, version, l->mode); | |
2530 | 272 |
273 return 0; | |
274 } | |
275 | |
276 AVCodec loco_decoder = { | |
277 "loco", | |
278 CODEC_TYPE_VIDEO, | |
279 CODEC_ID_LOCO, | |
280 sizeof(LOCOContext), | |
281 decode_init, | |
282 NULL, | |
283 NULL, | |
284 decode_frame, | |
285 CODEC_CAP_DR1, | |
7040
e943e1409077
Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents:
6750
diff
changeset
|
286 .long_name = NULL_IF_CONFIG_SMALL("LOCO"), |
2530 | 287 }; |