annotate loco.c @ 4777:536588c41db6 libavcodec

typo
author michael
date Tue, 03 Apr 2007 13:53:25 +0000
parents ce643a22f049
children f99e40a7155b
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
1 /*
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
2 * LOCO codec
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
3 * Copyright (c) 2005 Konstantin Shishkov
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
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
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
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
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
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
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
15 * Lesser General Public License for more details.
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
16 *
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
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
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
20 *
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
21 */
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
22
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
23 /**
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
24 * @file loco.c
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
25 * LOCO codec.
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
26 */
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
27
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
28 #include "avcodec.h"
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
29 #include "common.h"
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
30 #include "bitstream.h"
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
31 #include "golomb.h"
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
32
2587
1e8fbc2b64e0 support some more color modes; patch by Kostya
melanson
parents: 2558
diff changeset
33 enum LOCO_MODE {LOCO_UNKN=0, LOCO_CYUY2=-1, LOCO_CRGB=-2, LOCO_CRGBA=-3, LOCO_CYV12=-4,
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
34 LOCO_YUY2=1, LOCO_UYVY=2, LOCO_RGB=3, LOCO_RGBA=4, LOCO_YV12=5};
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
35
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
36 typedef struct LOCOContext{
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
37 AVCodecContext *avctx;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
38 AVFrame pic;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
39 int lossy;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
40 int mode;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
41 } LOCOContext;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
42
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
43 typedef struct RICEContext{
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
44 GetBitContext gb;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
45 int save, run, run2; /* internal rice decoder state */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
46 int sum, count; /* sum and count for getting rice parameter */
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
47 int lossy;
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
48 }RICEContext;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
49
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
50 static int loco_get_rice_param(RICEContext *r)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
51 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
52 int cnt = 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
53 int val = r->count;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
54
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
55 while(r->sum > val && cnt < 9) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
56 val <<= 1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
57 cnt++;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
58 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
59
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
60 return cnt;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
61 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
62
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
63 static inline void loco_update_rice_param(RICEContext *r, int val)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
64 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
65 r->sum += val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
66 r->count++;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
67
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
68 if(r->count == 16) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
69 r->sum >>= 1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
70 r->count >>= 1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
71 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
72 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
73
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
74 static inline int loco_get_rice(RICEContext *r)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
75 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
76 int v;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
77 if (r->run > 0) { /* we have zero run */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
78 r->run--;
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
79 loco_update_rice_param(r, 0);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
80 return 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
81 }
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
82 v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0);
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
83 loco_update_rice_param(r, (v+1)>>1);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
84 if (!v) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
85 if (r->save >= 0) {
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
86 r->run = get_ur_golomb_jpegls(&r->gb, 2, INT_MAX, 0);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
87 if(r->run > 1)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
88 r->save += r->run + 1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
89 else
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
90 r->save -= 3;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
91 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
92 else
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
93 r->run2++;
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
94 } else {
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
95 v = ((v>>1) + r->lossy) ^ -(v&1);
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
96 if (r->run2 > 0) {
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
97 if (r->run2 > 2)
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
98 r->save += r->run2;
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
99 else
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
100 r->save -= 3;
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
101 r->run2 = 0;
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
102 }
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
103 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
104
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
105 return v;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
106 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
107
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
108 /* LOCO main predictor - LOCO-I/JPEG-LS predictor */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
109 static inline int loco_predict(uint8_t* data, int stride, int step)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
110 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
111 int a, b, c;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
112
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
113 a = data[-stride];
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
114 b = data[-step];
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
115 c = data[-stride - step];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
116
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
117 return mid_pred(a, a + b - c, b);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
118 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
119
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
120 static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
121 int stride, uint8_t *buf, int buf_size, int step)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
122 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
123 RICEContext rc;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
124 int val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
125 int i, j;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
126
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
127 init_get_bits(&rc.gb, buf, buf_size*8);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
128 rc.save = 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
129 rc.run = 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
130 rc.run2 = 0;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
131 rc.lossy = l->lossy;
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
132
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
133 rc.sum = 8;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
134 rc.count = 1;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
135
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
136 /* restore top left pixel */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
137 val = loco_get_rice(&rc);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
138 data[0] = 128 + val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
139 /* restore top line */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
140 for (i = 1; i < width; i++) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
141 val = loco_get_rice(&rc);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
142 data[i * step] = data[i * step - step] + val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
143 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
144 data += stride;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
145 for (j = 1; j < height; j++) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
146 /* restore left column */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
147 val = loco_get_rice(&rc);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
148 data[0] = data[-stride] + val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
149 /* restore all other pixels */
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
150 for (i = 1; i < width; i++) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
151 val = loco_get_rice(&rc);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
152 data[i * step] = loco_predict(&data[i * step], stride, step) + val;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
153 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
154 data += stride;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
155 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
156
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
157 return ((get_bits_count(&rc.gb) + 7) >> 3);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
158 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
159
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
160 static int decode_frame(AVCodecContext *avctx,
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
161 void *data, int *data_size,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
162 uint8_t *buf, int buf_size)
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
163 {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
164 LOCOContext * const l = avctx->priv_data;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
165 AVFrame * const p= (AVFrame*)&l->pic;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
166 int decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
167
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
168 if(p->data[0])
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
169 avctx->release_buffer(avctx, p);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
170
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
171 p->reference = 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
172 if(avctx->get_buffer(avctx, p) < 0){
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
173 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
174 return -1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
175 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
176 p->key_frame = 1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
177
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
178 switch(l->mode) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
179 case LOCO_CYUY2: case LOCO_YUY2: case LOCO_UYVY:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
180 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
181 p->linesize[0], buf, buf_size, 1);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
182 buf += decoded; buf_size -= decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
183 decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
184 p->linesize[1], buf, buf_size, 1);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
185 buf += decoded; buf_size -= decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
186 decoded = loco_decode_plane(l, p->data[2], avctx->width / 2, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
187 p->linesize[2], buf, buf_size, 1);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
188 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
189 case LOCO_CYV12: case LOCO_YV12:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
190 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
191 p->linesize[0], buf, buf_size, 1);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
192 buf += decoded; buf_size -= decoded;
2598
6eaebf1fbd74 Fix colors for YV12 case (u/v planes are swapped)
rtognimp
parents: 2587
diff changeset
193 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
194 p->linesize[2], buf, buf_size, 1);
6eaebf1fbd74 Fix colors for YV12 case (u/v planes are swapped)
rtognimp
parents: 2587
diff changeset
195 buf += decoded; buf_size -= decoded;
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
196 decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height / 2,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
197 p->linesize[1], buf, buf_size, 1);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
198 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
199 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
200 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
201 -p->linesize[0], buf, buf_size, 3);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
202 buf += decoded; buf_size -= decoded;
2599
3e36a706f5b7 Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents: 2598
diff changeset
203 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
204 -p->linesize[0], buf, buf_size, 3);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
205 buf += decoded; buf_size -= decoded;
2599
3e36a706f5b7 Fix upside-down picture for BGR24 images (fixes pig-loco-rgb.avi)
rtognimp
parents: 2598
diff changeset
206 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
207 -p->linesize[0], buf, buf_size, 3);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
208 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
209 case LOCO_RGBA:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
210 decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
211 p->linesize[0], buf, buf_size, 4);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
212 buf += decoded; buf_size -= decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
213 decoded = loco_decode_plane(l, p->data[0] + 1, avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
214 p->linesize[0], buf, buf_size, 4);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
215 buf += decoded; buf_size -= decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
216 decoded = loco_decode_plane(l, p->data[0] + 2, avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
217 p->linesize[0], buf, buf_size, 4);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
218 buf += decoded; buf_size -= decoded;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
219 decoded = loco_decode_plane(l, p->data[0] + 3, avctx->width, avctx->height,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
220 p->linesize[0], buf, buf_size, 4);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
221 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
222 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
223
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
224 *data_size = sizeof(AVFrame);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
225 *(AVFrame*)data = l->pic;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
226
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
227 return buf_size;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
228 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
229
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
230 static int decode_init(AVCodecContext *avctx){
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
231 LOCOContext * const l = avctx->priv_data;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
232 int version;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
233
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
234 l->avctx = avctx;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
235 if (avctx->extradata_size < 12) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
236 av_log(avctx, AV_LOG_ERROR, "Extradata size must be >= 12 instead of %i\n",
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
237 avctx->extradata_size);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
238 return -1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
239 }
4364
05e932ddaaa9 rename BE/LE_8/16/32 to AV_RL/B_8/16/32
alex
parents: 3947
diff changeset
240 version = AV_RL32(avctx->extradata);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
241 switch(version) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
242 case 1:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
243 l->lossy = 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
244 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
245 case 2:
4364
05e932ddaaa9 rename BE/LE_8/16/32 to AV_RL/B_8/16/32
alex
parents: 3947
diff changeset
246 l->lossy = AV_RL32(avctx->extradata + 8);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
247 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
248 default:
4364
05e932ddaaa9 rename BE/LE_8/16/32 to AV_RL/B_8/16/32
alex
parents: 3947
diff changeset
249 l->lossy = AV_RL32(avctx->extradata + 8);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
250 av_log(avctx, AV_LOG_INFO, "This is LOCO codec version %i, please upload file for study\n", version);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
251 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2599
diff changeset
252
4364
05e932ddaaa9 rename BE/LE_8/16/32 to AV_RL/B_8/16/32
alex
parents: 3947
diff changeset
253 l->mode = AV_RL32(avctx->extradata + 4);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
254 switch(l->mode) {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
255 case LOCO_CYUY2: case LOCO_YUY2: case LOCO_UYVY:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
256 avctx->pix_fmt = PIX_FMT_YUV422P;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
257 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
258 case LOCO_CRGB: case LOCO_RGB:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
259 avctx->pix_fmt = PIX_FMT_BGR24;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
260 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
261 case LOCO_CYV12: case LOCO_YV12:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
262 avctx->pix_fmt = PIX_FMT_YUV420P;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
263 break;
2587
1e8fbc2b64e0 support some more color modes; patch by Kostya
melanson
parents: 2558
diff changeset
264 case LOCO_CRGBA: case LOCO_RGBA:
4494
ce643a22f049 Replace deprecated PIX_FMT names by the newer variants.
diego
parents: 4364
diff changeset
265 avctx->pix_fmt = PIX_FMT_RGB32;
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
266 break;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
267 default:
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
268 av_log(avctx, AV_LOG_INFO, "Unknown colorspace, index = %i\n", l->mode);
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
269 return -1;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
270 }
2558
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
271 if(avctx->debug & FF_DEBUG_PICT_INFO)
2b01396ab483 optimize & simplify
michael
parents: 2530
diff changeset
272 av_log(avctx, AV_LOG_INFO, "lossy:%i, version:%i, mode: %i\n", l->lossy, version, l->mode);
2530
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
273
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
274 return 0;
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
275 }
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
276
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
277 AVCodec loco_decoder = {
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
278 "loco",
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
279 CODEC_TYPE_VIDEO,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
280 CODEC_ID_LOCO,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
281 sizeof(LOCOContext),
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
282 decode_init,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
283 NULL,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
284 NULL,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
285 decode_frame,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
286 CODEC_CAP_DR1,
eace30b70601 go LOCO, courtesy of Kostya Shishkov
melanson
parents:
diff changeset
287 };