d3d9: Fix copy and paste error in reset_enum_callback.
[wine] / dlls / iccvid / iccvid.c
1 /*
2  * Radius Cinepak Video Decoder
3  *
4  * Copyright 2001 Dr. Tim Ferguson (see below)
5  * Portions Copyright 2003 Mike McCormack for CodeWeavers
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
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
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 /* Copyright notice from original source:
23  * ------------------------------------------------------------------------
24  * Radius Cinepak Video Decoder
25  *
26  * Dr. Tim Ferguson, 2001.
27  * For more details on the algorithm:
28  *         http://www.csse.monash.edu.au/~timf/videocodec.html
29  *
30  * This is basically a vector quantiser with adaptive vector density.  The
31  * frame is segmented into 4x4 pixel blocks, and each block is coded using
32  * either 1 or 4 vectors.
33  *
34  * There are still some issues with this code yet to be resolved.  In
35  * particular with decoding in the strip boundaries.  However, I have not
36  * yet found a sequence it doesn't work on.  Ill keep trying :)
37  *
38  * You may freely use this source code.  I only ask that you reference its
39  * source in your projects documentation:
40  *       Tim Ferguson: http://www.csse.monash.edu.au/~timf/
41  * ------------------------------------------------------------------------ */
42
43 #include <stdarg.h>
44 #include "windef.h"
45 #include "winbase.h"
46 #include "wingdi.h"
47 #include "winuser.h"
48 #include "commdlg.h"
49 #include "vfw.h"
50 #include "mmsystem.h"
51 #include "iccvid_private.h"
52
53 #include "wine/debug.h"
54
55 WINE_DEFAULT_DEBUG_CHANNEL(iccvid);
56
57 static HINSTANCE ICCVID_hModule;
58
59 #define ICCVID_MAGIC mmioFOURCC('c', 'v', 'i', 'd')
60
61 #define DBUG    0
62 #define MAX_STRIPS 32
63
64 /* ------------------------------------------------------------------------ */
65 typedef struct
66 {
67     unsigned char y0, y1, y2, y3;
68     char u, v;
69     unsigned char r[4], g[4], b[4];
70 } cvid_codebook;
71
72 typedef struct {
73     cvid_codebook *v4_codebook[MAX_STRIPS];
74     cvid_codebook *v1_codebook[MAX_STRIPS];
75     unsigned int strip_num;
76 } cinepak_info;
77
78 typedef struct _ICCVID_Info
79 {
80     DWORD         dwMagic;
81     int           bits_per_pixel;
82     cinepak_info *cvinfo;
83 } ICCVID_Info;
84
85 static inline LPVOID heap_alloc( size_t size )
86 {
87     return HeapAlloc( GetProcessHeap(), 0, size );
88 }
89
90 static inline BOOL heap_free( LPVOID ptr )
91 {
92     return HeapFree( GetProcessHeap(), 0, ptr );
93 }
94
95
96 /* ------------------------------------------------------------------------ */
97 static unsigned char *in_buffer, uiclip[1024], *uiclp = NULL;
98
99 #define get_byte() *(in_buffer++)
100 #define skip_byte() in_buffer++
101 #define get_word() ((unsigned short)(in_buffer += 2, \
102     (in_buffer[-2] << 8 | in_buffer[-1])))
103 #define get_long() ((unsigned long)(in_buffer += 4, \
104     (in_buffer[-4] << 24 | in_buffer[-3] << 16 | in_buffer[-2] << 8 | in_buffer[-1])))
105
106
107 /* ---------------------------------------------------------------------- */
108 static inline void read_codebook(cvid_codebook *c, int mode)
109 {
110 int uvr, uvg, uvb;
111
112     if(mode)        /* black and white */
113         {
114         c->y0 = get_byte();
115         c->y1 = get_byte();
116         c->y2 = get_byte();
117         c->y3 = get_byte();
118         c->u = c->v = 0;
119
120         c->r[0] = c->g[0] = c->b[0] = c->y0;
121         c->r[1] = c->g[1] = c->b[1] = c->y1;
122         c->r[2] = c->g[2] = c->b[2] = c->y2;
123         c->r[3] = c->g[3] = c->b[3] = c->y3;
124         }
125     else            /* colour */
126         {
127         c->y0 = get_byte();  /* luma */
128         c->y1 = get_byte();
129         c->y2 = get_byte();
130         c->y3 = get_byte();
131         c->u = get_byte(); /* chroma */
132         c->v = get_byte();
133
134         uvr = c->v << 1;
135         uvg = -((c->u+1) >> 1) - c->v;
136         uvb = c->u << 1;
137
138         c->r[0] = uiclp[c->y0 + uvr]; c->g[0] = uiclp[c->y0 + uvg]; c->b[0] = uiclp[c->y0 + uvb];
139         c->r[1] = uiclp[c->y1 + uvr]; c->g[1] = uiclp[c->y1 + uvg]; c->b[1] = uiclp[c->y1 + uvb];
140         c->r[2] = uiclp[c->y2 + uvr]; c->g[2] = uiclp[c->y2 + uvg]; c->b[2] = uiclp[c->y2 + uvb];
141         c->r[3] = uiclp[c->y3 + uvr]; c->g[3] = uiclp[c->y3 + uvg]; c->b[3] = uiclp[c->y3 + uvb];
142         }
143 }
144
145
146 #define MAKECOLOUR32(r,g,b) (((r) << 16) | ((g) << 8) | (b))
147 /*#define MAKECOLOUR24(r,g,b) (((r) << 16) | ((g) << 8) | (b))*/
148 #define MAKECOLOUR16(r,g,b) (((r) >> 3) << 11)| (((g) >> 2) << 5)| (((b) >> 3) << 0)
149 #define MAKECOLOUR15(r,g,b) (((r) >> 3) << 10)| (((g) >> 3) << 5)| (((b) >> 3) << 0)
150
151 /* ------------------------------------------------------------------------ */
152 static void cvid_v1_32(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb)
153 {
154 unsigned long *vptr = (unsigned long *)frm;
155 #ifndef ORIGINAL
156 int row_inc = -stride/4;
157 #else
158 int row_inc = stride/4;
159 #endif
160 int x, y;
161
162     /* fill 4x4 block of pixels with colour values from codebook */
163     for (y = 0; y < 4; y++)
164     {
165         if (&vptr[y*row_inc] < (unsigned long *)limit) return;
166         for (x = 0; x < 4; x++)
167             vptr[y*row_inc + x] = MAKECOLOUR32(cb->r[x/2+(y/2)*2], cb->g[x/2+(y/2)*2], cb->b[x/2+(y/2)*2]);
168     }
169 }
170
171
172 /* ------------------------------------------------------------------------ */
173 static void cvid_v4_32(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb0,
174     cvid_codebook *cb1, cvid_codebook *cb2, cvid_codebook *cb3)
175 {
176 unsigned long *vptr = (unsigned long *)frm;
177 #ifndef ORIGINAL
178 int row_inc = -stride/4;
179 #else
180 int row_inc = stride/4;
181 #endif
182 int x, y;
183 cvid_codebook * cb[] = {cb0,cb1,cb2,cb3};
184
185     /* fill 4x4 block of pixels with colour values from codebooks */
186     for (y = 0; y < 4; y++)
187     {
188         if (&vptr[y*row_inc] < (unsigned long *)limit) return;
189         for (x = 0; x < 4; x++)
190             vptr[y*row_inc + x] = MAKECOLOUR32(cb[x/2+(y/2)*2]->r[x%2+(y%2)*2], cb[x/2+(y/2)*2]->g[x%2+(y%2)*2], cb[x/2+(y/2)*2]->b[x%2+(y%2)*2]);
191     }
192 }
193
194
195 /* ------------------------------------------------------------------------ */
196 static void cvid_v1_24(unsigned char *vptr, unsigned char *limit, int stride, cvid_codebook *cb)
197 {
198 #ifndef ORIGINAL
199 int row_inc = -stride;
200 #else
201 int row_inc = stride;
202 #endif
203 int x, y;
204
205     /* fill 4x4 block of pixels with colour values from codebook */
206     for (y = 0; y < 4; y++)
207     {
208         if (&vptr[y*row_inc] < limit) return;
209         for (x = 0; x < 4; x++)
210         {
211             vptr[y*row_inc + x*3 + 0] = cb->b[x/2+(y/2)*2];
212             vptr[y*row_inc + x*3 + 1] = cb->g[x/2+(y/2)*2];
213             vptr[y*row_inc + x*3 + 2] = cb->r[x/2+(y/2)*2];
214         }
215     }
216 }
217
218
219 /* ------------------------------------------------------------------------ */
220 static void cvid_v4_24(unsigned char *vptr, unsigned char *limit, int stride, cvid_codebook *cb0,
221     cvid_codebook *cb1, cvid_codebook *cb2, cvid_codebook *cb3)
222 {
223 #ifndef ORIGINAL
224 int row_inc = -stride;
225 #else
226 int row_inc = stride;
227 #endif
228 cvid_codebook * cb[] = {cb0,cb1,cb2,cb3};
229 int x, y;
230
231     /* fill 4x4 block of pixels with colour values from codebooks */
232     for (y = 0; y < 4; y++)
233     {
234         if (&vptr[y*row_inc] < limit) return;
235         for (x = 0; x < 4; x++)
236         {
237             vptr[y*row_inc + x*3 + 0] = cb[x/2+(y/2)*2]->b[x%2+(y%2)*2];
238             vptr[y*row_inc + x*3 + 1] = cb[x/2+(y/2)*2]->g[x%2+(y%2)*2];
239             vptr[y*row_inc + x*3 + 2] = cb[x/2+(y/2)*2]->r[x%2+(y%2)*2];
240         }
241     }
242 }
243
244
245 /* ------------------------------------------------------------------------ */
246 static void cvid_v1_16(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb)
247 {
248 unsigned short *vptr = (unsigned short *)frm;
249 #ifndef ORIGINAL
250 int row_inc = -stride/2;
251 #else
252 int row_inc = stride/2;
253 #endif
254 int x, y;
255
256     /* fill 4x4 block of pixels with colour values from codebook */
257     for (y = 0; y < 4; y++)
258     {
259         if (&vptr[y*row_inc] < (unsigned short *)limit) return;
260         for (x = 0; x < 4; x++)
261             vptr[y*row_inc + x] = MAKECOLOUR16(cb->r[x/2+(y/2)*2], cb->g[x/2+(y/2)*2], cb->b[x/2+(y/2)*2]);
262     }
263 }
264
265
266 /* ------------------------------------------------------------------------ */
267 static void cvid_v4_16(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb0,
268     cvid_codebook *cb1, cvid_codebook *cb2, cvid_codebook *cb3)
269 {
270 unsigned short *vptr = (unsigned short *)frm;
271 #ifndef ORIGINAL
272 int row_inc = -stride/2;
273 #else
274 int row_inc = stride/2;
275 #endif
276 cvid_codebook * cb[] = {cb0,cb1,cb2,cb3};
277 int x, y;
278
279     /* fill 4x4 block of pixels with colour values from codebooks */
280     for (y = 0; y < 4; y++)
281     {
282         if (&vptr[y*row_inc] < (unsigned short *)limit) return;
283         for (x = 0; x < 4; x++)
284             vptr[y*row_inc + x] = MAKECOLOUR16(cb[x/2+(y/2)*2]->r[x%2+(y%2)*2], cb[x/2+(y/2)*2]->g[x%2+(y%2)*2], cb[x/2+(y/2)*2]->b[x%2+(y%2)*2]);
285     }
286 }
287
288 /* ------------------------------------------------------------------------ */
289 static void cvid_v1_15(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb)
290 {
291 unsigned short *vptr = (unsigned short *)frm;
292 #ifndef ORIGINAL
293 int row_inc = -stride/2;
294 #else
295 int row_inc = stride/2;
296 #endif
297 int x, y;
298
299     /* fill 4x4 block of pixels with colour values from codebook */
300     for (y = 0; y < 4; y++)
301     {
302         if (&vptr[y*row_inc] < (unsigned short *)limit) return;
303         for (x = 0; x < 4; x++)
304             vptr[y*row_inc + x] = MAKECOLOUR15(cb->r[x/2+(y/2)*2], cb->g[x/2+(y/2)*2], cb->b[x/2+(y/2)*2]);
305     }
306 }
307
308
309 /* ------------------------------------------------------------------------ */
310 static void cvid_v4_15(unsigned char *frm, unsigned char *limit, int stride, cvid_codebook *cb0,
311     cvid_codebook *cb1, cvid_codebook *cb2, cvid_codebook *cb3)
312 {
313 unsigned short *vptr = (unsigned short *)frm;
314 #ifndef ORIGINAL
315 int row_inc = -stride/2;
316 #else
317 int row_inc = stride/2;
318 #endif
319 cvid_codebook * cb[] = {cb0,cb1,cb2,cb3};
320 int x, y;
321
322     /* fill 4x4 block of pixels with colour values from codebooks */
323     for (y = 0; y < 4; y++)
324     {
325         if (&vptr[y*row_inc] < (unsigned short *)limit) return;
326         for (x = 0; x < 4; x++)
327             vptr[y*row_inc + x] = MAKECOLOUR15(cb[x/2+(y/2)*2]->r[x%2+(y%2)*2], cb[x/2+(y/2)*2]->g[x%2+(y%2)*2], cb[x/2+(y/2)*2]->b[x%2+(y%2)*2]);
328     }
329 }
330
331
332 /* ------------------------------------------------------------------------
333  * Call this function once at the start of the sequence and save the
334  * returned context for calls to decode_cinepak().
335  */
336 static cinepak_info *decode_cinepak_init(void)
337 {
338     cinepak_info *cvinfo;
339     int i;
340
341     cvinfo = heap_alloc( sizeof (cinepak_info) );
342     if( !cvinfo )
343         return NULL;
344     cvinfo->strip_num = 0;
345
346     if(uiclp == NULL)
347     {
348         uiclp = uiclip+512;
349         for(i = -512; i < 512; i++)
350             uiclp[i] = (i < 0 ? 0 : (i > 255 ? 255 : i));
351     }
352
353     return cvinfo;
354 }
355
356 static void free_cvinfo( cinepak_info *cvinfo )
357 {
358     unsigned int i;
359
360     for( i=0; i<cvinfo->strip_num; i++ )
361     {
362         heap_free(cvinfo->v4_codebook[i]);
363         heap_free(cvinfo->v1_codebook[i]);
364     }
365     heap_free( cvinfo );
366 }
367
368 typedef void (*fn_cvid_v1)(unsigned char *frm, unsigned char *limit,
369                            int stride, cvid_codebook *cb);
370 typedef void (*fn_cvid_v4)(unsigned char *frm, unsigned char *limit, int stride,
371                            cvid_codebook *cb0, cvid_codebook *cb1,
372                            cvid_codebook *cb2, cvid_codebook *cb3);
373
374 /* ------------------------------------------------------------------------
375  * This function decodes a buffer containing a Cinepak encoded frame.
376  *
377  * context - the context created by decode_cinepak_init().
378  * buf - the input buffer to be decoded
379  * size - the size of the input buffer
380  * frame - the output frame buffer (24 or 32 bit per pixel)
381  * width - the width of the output frame
382  * height - the height of the output frame
383  * bit_per_pixel - the number of bits per pixel allocated to the output
384  *   frame (only 24 or 32 bpp are supported)
385  */
386 static void decode_cinepak(cinepak_info *cvinfo, unsigned char *buf, int size,
387            unsigned char *frame, unsigned int width, unsigned int height, int bit_per_pixel)
388 {
389     cvid_codebook *v4_codebook, *v1_codebook, *codebook = NULL;
390     unsigned long x, y, y_bottom, frame_flags, strips, cv_width, cv_height,
391                   cnum, strip_id, chunk_id, x0, y0, x1, y1, ci, flag, mask;
392     long len, top_size, chunk_size;
393     unsigned char *frm_ptr, *frm_end;
394     unsigned int i, cur_strip;
395     int d0, d1, d2, d3, frm_stride, bpp = 3;
396     fn_cvid_v1 cvid_v1 = cvid_v1_24;
397     fn_cvid_v4 cvid_v4 = cvid_v4_24;
398
399     x = y = 0;
400     y_bottom = 0;
401     in_buffer = buf;
402
403     frame_flags = get_byte();
404     len = get_byte() << 16;
405     len |= get_byte() << 8;
406     len |= get_byte();
407
408     switch(bit_per_pixel)
409         {
410         case 15:
411             bpp = 2;
412             cvid_v1 = cvid_v1_15;
413             cvid_v4 = cvid_v4_15;
414             break;
415         case 16:
416             bpp = 2;
417             cvid_v1 = cvid_v1_16;
418             cvid_v4 = cvid_v4_16;
419             break;
420         case 24:
421             bpp = 3;
422             cvid_v1 = cvid_v1_24;
423             cvid_v4 = cvid_v4_24;
424             break;
425         case 32:
426             bpp = 4;
427             cvid_v1 = cvid_v1_32;
428             cvid_v4 = cvid_v4_32;
429             break;
430         }
431
432     frm_stride = width * bpp;
433     frm_ptr = frame;
434     frm_end = frm_ptr + width * height * bpp;
435
436     if(len != size)
437         {
438         if(len & 0x01) len++; /* AVIs tend to have a size mismatch */
439         if(len != size)
440             {
441             ERR("CVID: corruption %d (QT/AVI) != %ld (CV)\n", size, len);
442             /* return; */
443             }
444         }
445
446     cv_width = get_word();
447     cv_height = get_word();
448     strips = get_word();
449
450     if(strips > cvinfo->strip_num)
451         {
452         if(strips >= MAX_STRIPS)
453             {
454             ERR("CVID: strip overflow (more than %d)\n", MAX_STRIPS);
455             return;
456             }
457
458         for(i = cvinfo->strip_num; i < strips; i++)
459             {
460             if((cvinfo->v4_codebook[i] = heap_alloc(sizeof(cvid_codebook) * 260)) == NULL)
461                 {
462                 ERR("CVID: codebook v4 alloc err\n");
463                 return;
464                 }
465
466             if((cvinfo->v1_codebook[i] = heap_alloc(sizeof(cvid_codebook) * 260)) == NULL)
467                 {
468                 ERR("CVID: codebook v1 alloc err\n");
469                 return;
470                 }
471             }
472         }
473     cvinfo->strip_num = strips;
474
475     TRACE("CVID: <%ld,%ld> strips %ld\n", cv_width, cv_height, strips);
476
477     for(cur_strip = 0; cur_strip < strips; cur_strip++)
478         {
479         v4_codebook = cvinfo->v4_codebook[cur_strip];
480         v1_codebook = cvinfo->v1_codebook[cur_strip];
481
482         if((cur_strip > 0) && (!(frame_flags & 0x01)))
483             {
484             memcpy(cvinfo->v4_codebook[cur_strip], cvinfo->v4_codebook[cur_strip-1], 260 * sizeof(cvid_codebook));
485             memcpy(cvinfo->v1_codebook[cur_strip], cvinfo->v1_codebook[cur_strip-1], 260 * sizeof(cvid_codebook));
486             }
487
488         strip_id = get_word();        /* 1000 = key strip, 1100 = iter strip */
489         top_size = get_word();
490         y0 = get_word();        /* FIXME: most of these are ignored at the moment */
491         x0 = get_word();
492         y1 = get_word();
493         x1 = get_word();
494
495         y_bottom += y1;
496         top_size -= 12;
497         x = 0;
498         if(x1 != width)
499             WARN("CVID: Warning x1 (%ld) != width (%d)\n", x1, width);
500
501         TRACE("   %d) %04lx %04ld <%ld,%ld> <%ld,%ld> yt %ld\n",
502               cur_strip, strip_id, top_size, x0, y0, x1, y1, y_bottom);
503
504         while(top_size > 0)
505             {
506             chunk_id  = get_word();
507             chunk_size = get_word();
508
509             TRACE("        %04lx %04ld\n", chunk_id, chunk_size);
510             top_size -= chunk_size;
511             chunk_size -= 4;
512
513             switch(chunk_id)
514                 {
515                     /* -------------------- Codebook Entries -------------------- */
516                 case 0x2000:
517                 case 0x2200:
518                     codebook = (chunk_id == 0x2200 ? v1_codebook : v4_codebook);
519                     cnum = chunk_size/6;
520                     for(i = 0; i < cnum; i++) read_codebook(codebook+i, 0);
521                     break;
522
523                 case 0x2400:
524                 case 0x2600:        /* 8 bit per pixel */
525                     codebook = (chunk_id == 0x2600 ? v1_codebook : v4_codebook);
526                     cnum = chunk_size/4;
527                     for(i = 0; i < cnum; i++) read_codebook(codebook+i, 1);
528                     break;
529
530                 case 0x2100:
531                 case 0x2300:
532                     codebook = (chunk_id == 0x2300 ? v1_codebook : v4_codebook);
533
534                     ci = 0;
535                     while(chunk_size > 0)
536                         {
537                         flag = get_long();
538                         chunk_size -= 4;
539
540                         for(i = 0; i < 32; i++)
541                             {
542                             if(flag & 0x80000000)
543                                 {
544                                 chunk_size -= 6;
545                                 read_codebook(codebook+ci, 0);
546                                 }
547
548                             ci++;
549                             flag <<= 1;
550                             }
551                         }
552                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
553                     break;
554
555                 case 0x2500:
556                 case 0x2700:        /* 8 bit per pixel */
557                     codebook = (chunk_id == 0x2700 ? v1_codebook : v4_codebook);
558
559                     ci = 0;
560                     while(chunk_size > 0)
561                         {
562                         flag = get_long();
563                         chunk_size -= 4;
564
565                         for(i = 0; i < 32; i++)
566                             {
567                             if(flag & 0x80000000)
568                                 {
569                                 chunk_size -= 4;
570                                 read_codebook(codebook+ci, 1);
571                                 }
572
573                             ci++;
574                             flag <<= 1;
575                             }
576                         }
577                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
578                     break;
579
580                     /* -------------------- Frame -------------------- */
581                 case 0x3000:
582                     while((chunk_size > 0) && (y < y_bottom))
583                         {
584                         flag = get_long();
585                         chunk_size -= 4;
586
587                         for(i = 0; i < 32; i++)
588                             {
589                             if(y >= y_bottom) break;
590                             if(flag & 0x80000000)    /* 4 bytes per block */
591                                 {
592                                 d0 = get_byte();
593                                 d1 = get_byte();
594                                 d2 = get_byte();
595                                 d3 = get_byte();
596                                 chunk_size -= 4;
597 #ifdef ORIGINAL
598                                 cvid_v4(frm_ptr + (y * frm_stride + x * bpp), frm_end, frm_stride, v4_codebook+d0, v4_codebook+d1, v4_codebook+d2, v4_codebook+d3);
599 #else
600                                 cvid_v4(frm_ptr + ((height - 1 - y) * frm_stride + x * bpp), frame, frm_stride, v4_codebook+d0, v4_codebook+d1, v4_codebook+d2, v4_codebook+d3);
601 #endif
602                                 }
603                             else        /* 1 byte per block */
604                                 {
605 #ifdef ORIGINAL
606                                 cvid_v1(frm_ptr + (y * frm_stride + x * bpp), frm_end, frm_stride, v1_codebook + get_byte());
607 #else
608                                 cvid_v1(frm_ptr + ((height - 1 - y) * frm_stride + x * bpp), frame, frm_stride, v1_codebook + get_byte());
609 #endif
610                                 chunk_size--;
611                                 }
612
613                             x += 4;
614                             if(x >= width)
615                                 {
616                                 x = 0;
617                                 y += 4;
618                                 }
619                             flag <<= 1;
620                             }
621                         }
622                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
623                     break;
624
625                 case 0x3100:
626                     while((chunk_size > 0) && (y < y_bottom))
627                         {
628                             /* ---- flag bits: 0 = SKIP, 10 = V1, 11 = V4 ---- */
629                         flag = get_long();
630                         chunk_size -= 4;
631                         mask = 0x80000000;
632
633                         while((mask) && (y < y_bottom))
634                             {
635                             if(flag & mask)
636                                 {
637                                 if(mask == 1)
638                                     {
639                                     if(chunk_size < 0) break;
640                                     flag = get_long();
641                                     chunk_size -= 4;
642                                     mask = 0x80000000;
643                                     }
644                                 else mask >>= 1;
645
646                                 if(flag & mask)        /* V4 */
647                                     {
648                                     d0 = get_byte();
649                                     d1 = get_byte();
650                                     d2 = get_byte();
651                                     d3 = get_byte();
652                                     chunk_size -= 4;
653 #ifdef ORIGINAL
654                                     cvid_v4(frm_ptr + (y * frm_stride + x * bpp), frm_end, frm_stride, v4_codebook+d0, v4_codebook+d1, v4_codebook+d2, v4_codebook+d3);
655 #else
656                                     cvid_v4(frm_ptr + ((height - 1 - y) * frm_stride + x * bpp), frame, frm_stride, v4_codebook+d0, v4_codebook+d1, v4_codebook+d2, v4_codebook+d3);
657 #endif
658                                     }
659                                 else        /* V1 */
660                                     {
661                                     chunk_size--;
662 #ifdef ORIGINAL
663                                     cvid_v1(frm_ptr + (y * frm_stride + x * bpp), frm_end, frm_stride, v1_codebook + get_byte());
664 #else
665                                     cvid_v1(frm_ptr + ((height - 1 - y) * frm_stride + x * bpp), frame, frm_stride, v1_codebook + get_byte());
666 #endif
667                                     }
668                                 }        /* else SKIP */
669
670                             mask >>= 1;
671                             x += 4;
672                             if(x >= width)
673                                 {
674                                 x = 0;
675                                 y += 4;
676                                 }
677                             }
678                         }
679
680                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
681                     break;
682
683                 case 0x3200:        /* each byte is a V1 codebook */
684                     while((chunk_size > 0) && (y < y_bottom))
685                         {
686 #ifdef ORIGINAL
687                         cvid_v1(frm_ptr + (y * frm_stride + x * bpp), frm_end, frm_stride, v1_codebook + get_byte());
688 #else
689                         cvid_v1(frm_ptr + ((height - 1 - y) * frm_stride + x * bpp), frame, frm_stride, v1_codebook + get_byte());
690 #endif
691                         chunk_size--;
692                         x += 4;
693                         if(x >= width)
694                             {
695                             x = 0;
696                             y += 4;
697                             }
698                         }
699                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
700                     break;
701
702                 default:
703                     ERR("CVID: unknown chunk_id %08lx\n", chunk_id);
704                     while(chunk_size > 0) { skip_byte(); chunk_size--; }
705                     break;
706                 }
707             }
708         }
709
710     if(len != size)
711         {
712         if(len & 0x01) len++; /* AVIs tend to have a size mismatch */
713         if(len != size)
714             {
715             long xlen;
716             skip_byte();
717             xlen = get_byte() << 16;
718             xlen |= get_byte() << 8;
719             xlen |= get_byte(); /* Read Len */
720             WARN("CVID: END INFO chunk size %d cvid size1 %ld cvid size2 %ld\n",
721                   size, len, xlen);
722             }
723         }
724 }
725
726 static void ICCVID_dump_BITMAPINFO(const BITMAPINFO * bmi)
727 {
728     TRACE(
729         "planes = %d\n"
730         "bpp    = %d\n"
731         "height = %d\n"
732         "width  = %d\n"
733         "compr  = %s\n",
734         bmi->bmiHeader.biPlanes,
735         bmi->bmiHeader.biBitCount,
736         bmi->bmiHeader.biHeight,
737         bmi->bmiHeader.biWidth,
738         debugstr_an( (const char *)&bmi->bmiHeader.biCompression, 4 ) );
739 }
740
741 static inline int ICCVID_CheckMask(RGBQUAD bmiColors[3], COLORREF redMask, COLORREF blueMask, COLORREF greenMask)
742 {
743     COLORREF realRedMask = MAKECOLOUR32(bmiColors[0].rgbRed, bmiColors[0].rgbGreen, bmiColors[0].rgbBlue);
744     COLORREF realBlueMask = MAKECOLOUR32(bmiColors[1].rgbRed, bmiColors[1].rgbGreen, bmiColors[1].rgbBlue);
745     COLORREF realGreenMask = MAKECOLOUR32(bmiColors[2].rgbRed, bmiColors[2].rgbGreen, bmiColors[2].rgbBlue);
746
747     TRACE("\nbmiColors[0] = 0x%08x\nbmiColors[1] = 0x%08x\nbmiColors[2] = 0x%08x\n",
748         realRedMask, realBlueMask, realGreenMask);
749         
750     if ((realRedMask == redMask) &&
751         (realBlueMask == blueMask) &&
752         (realGreenMask == greenMask))
753         return TRUE;
754     return FALSE;
755 }
756
757 static LRESULT ICCVID_DecompressQuery( ICCVID_Info *info, LPBITMAPINFO in, LPBITMAPINFO out )
758 {
759     TRACE("ICM_DECOMPRESS_QUERY %p %p %p\n", info, in, out);
760
761     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
762         return ICERR_BADPARAM;
763
764     TRACE("in: ");
765     ICCVID_dump_BITMAPINFO(in);
766
767     if( in->bmiHeader.biCompression != ICCVID_MAGIC )
768         return ICERR_BADFORMAT;
769
770     if( out )
771     {
772         TRACE("out: ");
773         ICCVID_dump_BITMAPINFO(out);
774
775         if( in->bmiHeader.biPlanes != out->bmiHeader.biPlanes )
776             return ICERR_BADFORMAT;
777         if( in->bmiHeader.biHeight != out->bmiHeader.biHeight )
778             return ICERR_BADFORMAT;
779         if( in->bmiHeader.biWidth != out->bmiHeader.biWidth )
780             return ICERR_BADFORMAT;
781
782         switch( out->bmiHeader.biBitCount )
783         {
784         case 16:
785             if ( out->bmiHeader.biCompression == BI_BITFIELDS )
786             {
787                 if ( !ICCVID_CheckMask(out->bmiColors, 0x7C00, 0x03E0, 0x001F) &&
788                      !ICCVID_CheckMask(out->bmiColors, 0xF800, 0x07E0, 0x001F) )
789                 {
790                     TRACE("unsupported output bit field(s) for 16-bit colors\n");
791                     return ICERR_BADFORMAT;
792                 }
793             }
794             break;
795         case 24:
796         case 32:
797             break;
798         default:
799             TRACE("unsupported output bitcount = %d\n", out->bmiHeader.biBitCount );
800             return ICERR_BADFORMAT;
801         }
802     }
803
804     return ICERR_OK;
805 }
806
807 static LRESULT ICCVID_DecompressGetFormat( ICCVID_Info *info, LPBITMAPINFO in, LPBITMAPINFO out )
808 {
809     DWORD size;
810
811     TRACE("ICM_DECOMPRESS_GETFORMAT %p %p %p\n", info, in, out);
812
813     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
814         return ICERR_BADPARAM;
815
816     size = in->bmiHeader.biSize;
817     if (in->bmiHeader.biBitCount <= 8)
818         size += in->bmiHeader.biClrUsed * sizeof(RGBQUAD);
819
820     if( out )
821     {
822         memcpy( out, in, size );
823         out->bmiHeader.biCompression = BI_RGB;
824         out->bmiHeader.biSizeImage = in->bmiHeader.biHeight
825                                    * in->bmiHeader.biWidth *4;
826         return ICERR_OK;
827     }
828     return size;
829 }
830
831 static LRESULT ICCVID_DecompressBegin( ICCVID_Info *info, LPBITMAPINFO in, LPBITMAPINFO out )
832 {
833     TRACE("ICM_DECOMPRESS_BEGIN %p %p %p\n", info, in, out);
834
835     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
836         return ICERR_BADPARAM;
837
838     info->bits_per_pixel = out->bmiHeader.biBitCount;
839
840     if (info->bits_per_pixel == 16)
841     {
842         if ( out->bmiHeader.biCompression == BI_BITFIELDS )
843         {
844             if ( ICCVID_CheckMask(out->bmiColors, 0x7C00, 0x03E0, 0x001F) )
845                 info->bits_per_pixel = 15;
846             else if ( ICCVID_CheckMask(out->bmiColors, 0xF800, 0x07E0, 0x001F) )
847                 info->bits_per_pixel = 16;
848             else
849             {
850                 TRACE("unsupported output bit field(s) for 16-bit colors\n");
851                 return ICERR_UNSUPPORTED;
852             }
853         }
854         else
855             info->bits_per_pixel = 15;
856     }
857
858     TRACE("bit_per_pixel = %d\n", info->bits_per_pixel);
859
860     if( info->cvinfo )
861         free_cvinfo( info->cvinfo );
862     info->cvinfo = decode_cinepak_init();
863
864     return ICERR_OK;
865 }
866
867 static LRESULT ICCVID_Decompress( ICCVID_Info *info, ICDECOMPRESS *icd, DWORD size )
868 {
869     LONG width, height;
870
871     TRACE("ICM_DECOMPRESS %p %p %d\n", info, icd, size);
872
873     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
874         return ICERR_BADPARAM;
875     if (info->cvinfo==NULL)
876     {
877         ERR("ICM_DECOMPRESS sent after ICM_DECOMPRESS_END\n");
878         return ICERR_BADPARAM;
879     }
880
881     width  = icd->lpbiInput->biWidth;
882     height = icd->lpbiInput->biHeight;
883
884     decode_cinepak(info->cvinfo, icd->lpInput, icd->lpbiInput->biSizeImage,
885                    icd->lpOutput, width, height, info->bits_per_pixel);
886
887     return ICERR_OK;
888 }
889
890 static LRESULT ICCVID_DecompressEx( ICCVID_Info *info, ICDECOMPRESSEX *icd, DWORD size )
891 {
892     LONG width, height;
893
894     TRACE("ICM_DECOMPRESSEX %p %p %d\n", info, icd, size);
895
896     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
897         return ICERR_BADPARAM;
898     if (info->cvinfo==NULL)
899     {
900         ERR("ICM_DECOMPRESSEX sent after ICM_DECOMPRESS_END\n");
901         return ICERR_BADPARAM;
902     }
903
904     /* FIXME: flags are ignored */
905
906     width  = icd->lpbiSrc->biWidth;
907     height = icd->lpbiSrc->biHeight;
908
909     decode_cinepak(info->cvinfo, icd->lpSrc, icd->lpbiSrc->biSizeImage,
910                    icd->lpDst, width, height, info->bits_per_pixel);
911
912     return ICERR_OK;
913 }
914
915 static LRESULT ICCVID_Close( ICCVID_Info *info )
916 {
917     if( (info==NULL) || (info->dwMagic!=ICCVID_MAGIC) )
918         return 0;
919     if( info->cvinfo )
920         free_cvinfo( info->cvinfo );
921     heap_free( info );
922     return 1;
923 }
924
925 static LRESULT ICCVID_GetInfo( ICCVID_Info *info, ICINFO *icinfo, DWORD dwSize )
926 {
927     if (!icinfo) return sizeof(ICINFO);
928     if (dwSize < sizeof(ICINFO)) return 0;
929
930     icinfo->dwSize = sizeof(ICINFO);
931     icinfo->fccType = ICTYPE_VIDEO;
932     icinfo->fccHandler = info ? info->dwMagic : ICCVID_MAGIC;
933     icinfo->dwFlags = 0;
934     icinfo->dwVersion = ICVERSION;
935     icinfo->dwVersionICM = ICVERSION;
936
937     LoadStringW(ICCVID_hModule, IDS_NAME, icinfo->szName, sizeof(icinfo->szName)/sizeof(WCHAR));
938     LoadStringW(ICCVID_hModule, IDS_DESCRIPTION, icinfo->szDescription, sizeof(icinfo->szDescription)/sizeof(WCHAR));
939     /* msvfw32 will fill icinfo->szDriver for us */
940
941     return sizeof(ICINFO);
942 }
943
944 static LRESULT ICCVID_DecompressEnd( ICCVID_Info *info )
945 {
946     if( info->cvinfo )
947     {
948         free_cvinfo( info->cvinfo );
949         info->cvinfo = NULL;
950     }
951     return ICERR_OK;
952 }
953
954 LRESULT WINAPI ICCVID_DriverProc( DWORD_PTR dwDriverId, HDRVR hdrvr, UINT msg,
955                                   LPARAM lParam1, LPARAM lParam2)
956 {
957     ICCVID_Info *info = (ICCVID_Info *) dwDriverId;
958
959     TRACE("%ld %p %d %ld %ld\n", dwDriverId, hdrvr, msg, lParam1, lParam2);
960
961     switch( msg )
962     {
963     case DRV_LOAD:
964         TRACE("Loaded\n");
965         return 1;
966     case DRV_ENABLE:
967         return 0;
968     case DRV_DISABLE:
969         return 0;
970     case DRV_FREE:
971         return 0;
972
973     case DRV_OPEN:
974     {
975         ICINFO *icinfo = (ICINFO *)lParam2;
976
977         TRACE("Opened\n");
978
979         if (icinfo && icinfo->fccType != ICTYPE_VIDEO) return 0;
980
981         info = heap_alloc( sizeof (ICCVID_Info) );
982         if( info )
983         {
984             info->dwMagic = ICCVID_MAGIC;
985             info->cvinfo = NULL;
986         }
987         return (LRESULT) info;
988     }
989
990     case DRV_CLOSE:
991         return ICCVID_Close( info );
992
993     case ICM_GETINFO:
994         return ICCVID_GetInfo( info, (ICINFO *)lParam1, (DWORD)lParam2 );
995
996     case ICM_DECOMPRESS_QUERY:
997         return ICCVID_DecompressQuery( info, (LPBITMAPINFO) lParam1,
998                                        (LPBITMAPINFO) lParam2 );
999     case ICM_DECOMPRESS_GET_FORMAT:
1000         return ICCVID_DecompressGetFormat( info, (LPBITMAPINFO) lParam1,
1001                                        (LPBITMAPINFO) lParam2 );
1002     case ICM_DECOMPRESS_BEGIN:
1003         return ICCVID_DecompressBegin( info, (LPBITMAPINFO) lParam1,
1004                                        (LPBITMAPINFO) lParam2 );
1005     case ICM_DECOMPRESS:
1006         return ICCVID_Decompress( info, (ICDECOMPRESS*) lParam1,
1007                                   (DWORD) lParam2 );
1008     case ICM_DECOMPRESSEX:
1009         return ICCVID_DecompressEx( info, (ICDECOMPRESSEX*) lParam1, 
1010                                   (DWORD) lParam2 );
1011
1012     case ICM_DECOMPRESS_END:
1013         return ICCVID_DecompressEnd( info );
1014
1015     case ICM_COMPRESS_QUERY:
1016         FIXME("compression not implemented\n");
1017         return ICERR_BADFORMAT;
1018
1019     case ICM_CONFIGURE:
1020         return ICERR_UNSUPPORTED;
1021
1022     default:
1023         FIXME("Unknown message: %04x %ld %ld\n", msg, lParam1, lParam2);
1024     }
1025     return ICERR_UNSUPPORTED;
1026 }
1027
1028 BOOL WINAPI DllMain(HINSTANCE hModule, DWORD dwReason, LPVOID lpReserved)
1029 {
1030     TRACE("(%p,%d,%p)\n", hModule, dwReason, lpReserved);
1031
1032     switch (dwReason)
1033     {
1034     case DLL_PROCESS_ATTACH:
1035         DisableThreadLibraryCalls(hModule);
1036         ICCVID_hModule = hModule;
1037         break;
1038
1039     case DLL_PROCESS_DETACH:
1040         break;
1041     }
1042     return TRUE;
1043 }