fix alphatest (HACK!!!)
[enigma2.git] / lib / gdi / gpixmap.cpp
1 #include <lib/gdi/gpixmap.h>
2 #include <lib/gdi/region.h>
3 #include <lib/gdi/accel.h>
4
5 gLookup::gLookup()
6         :size(0), lookup(0)
7 {
8 }
9
10 gLookup::gLookup(int size, const gPalette &pal, const gRGB &start, const gRGB &end)
11         :size(0), lookup(0)
12 {
13         build(size, pal, start, end);
14 }
15
16 void gLookup::build(int _size, const gPalette &pal, const gRGB &start, const gRGB &end)
17 {
18         if (lookup)
19         {
20                 delete [] lookup;
21                 lookup=0;
22                 size=0;
23         }
24         size=_size;
25         if (!size)
26                 return;
27         lookup=new gColor[size];
28         
29         for (int i=0; i<size; i++)
30         {
31                 gRGB col;
32                 if (i)
33                 {
34                         int rdiff=-start.r+end.r;
35                         int gdiff=-start.g+end.g;
36                         int bdiff=-start.b+end.b;
37                         int adiff=-start.a+end.a;
38                         rdiff*=i; rdiff/=(size-1);
39                         gdiff*=i; gdiff/=(size-1);
40                         bdiff*=i; bdiff/=(size-1);
41                         adiff*=i; adiff/=(size-1);
42                         col.r=start.r+rdiff;
43                         col.g=start.g+gdiff;
44                         col.b=start.b+bdiff;
45                         col.a=start.a+adiff;
46                 } else
47                         col=start;
48                 lookup[i]=pal.findColor(col);
49         }
50 }
51
52 gSurface::gSurface()
53 {
54         x = 0;
55         y = 0;
56         bpp = 0;
57         bypp = 0;
58         stride = 0;
59         data = 0;
60         data_phys = 0;
61         clut.colors = 0;
62         clut.data = 0;
63         type = 0;
64 }
65
66 gSurface::gSurface(eSize size, int _bpp, int accel)
67 {
68         x = size.width();
69         y = size.height();
70         bpp = _bpp;
71
72         switch (bpp)
73         {
74         case 8:
75                 bypp = 1;
76                 break;
77         case 15:
78         case 16:
79                 bypp = 2;
80                 break;
81         case 24:                // never use 24bit mode
82         case 32:
83                 bypp = 4;
84                 break;
85         default:
86                 bypp = (bpp+7)/8;
87         }
88
89         stride = x*bypp;
90         
91         data = 0;
92         data_phys = 0;
93         
94         if (accel)
95         {
96                 stride += 63;
97                 stride &=~63;
98                 
99                 if (gAccel::getInstance())
100                         eDebug("accel memory: %d", gAccel::getInstance()->accelAlloc(data, data_phys, y * stride));
101                 else
102                         eDebug("no accel available");
103         }
104         
105         clut.colors = 0;
106         clut.data = 0;
107
108         if (!data)
109                 data = malloc(y * stride);
110         
111         type = 1;
112 }
113
114 gSurface::~gSurface()
115 {
116         if (type)
117         {
118                 if (data_phys)
119                         gAccel::getInstance()->accelFree(data_phys);
120                 else
121                         free(data);
122
123                 delete[] clut.data;
124         }
125 }
126
127 gPixmap *gPixmap::lock()
128 {
129         contentlock.lock(1);
130         return this;
131 }
132
133 void gPixmap::unlock()
134 {
135         contentlock.unlock(1);
136 }
137
138 void gPixmap::fill(const gRegion &region, const gColor &color)
139 {
140         unsigned int i;
141         for (i=0; i<region.rects.size(); ++i)
142         {
143                 const eRect &area = region.rects[i];
144                 if ((area.height()<=0) || (area.width()<=0))
145                         continue;
146
147                 if (surface->bpp == 8)
148                 {
149                         for (int y=area.top(); y<area.bottom(); y++)
150                                 memset(((__u8*)surface->data)+y*surface->stride+area.left(), color.color, area.width());
151                 } else if (surface->bpp == 32)
152                 {
153                         __u32 col;
154
155                         if (surface->clut.data && color < surface->clut.colors)
156                                 col=(surface->clut.data[color].a<<24)|(surface->clut.data[color].r<<16)|(surface->clut.data[color].g<<8)|(surface->clut.data[color].b);
157                         else
158                                 col=0x10101*color;
159                         
160                         col^=0xFF000000;                        
161
162                         if (surface->data_phys && gAccel::getInstance())
163                                 if (!gAccel::getInstance()->fill(surface,  area, col))
164                                         continue;
165
166                         for (int y=area.top(); y<area.bottom(); y++)
167                         {
168                                 __u32 *dst=(__u32*)(((__u8*)surface->data)+y*surface->stride+area.left()*surface->bypp);
169                                 int x=area.width();
170                                 while (x--)
171                                         *dst++=col;
172                         }
173                 }       else
174                         eWarning("couldn't fill %d bpp", surface->bpp);
175         }
176 }
177
178 void gPixmap::blit(const gPixmap &src, ePoint pos, const gRegion &clip, int flag)
179 {
180         for (unsigned int i=0; i<clip.rects.size(); ++i)
181         {
182                 eRect area=eRect(pos, src.size());
183
184                 if (!(flag & blitAlphaTest))  // HACK !!
185                         area&=clip.rects[i];
186
187                 area&=eRect(ePoint(0, 0), size());
188                 if ((area.width()<0) || (area.height()<0))
189                         continue;
190
191                 eRect srcarea=area;
192                 srcarea.moveBy(-pos.x(), -pos.y());
193                 
194                 if ((surface->data_phys && src.surface->data_phys) && (gAccel::getInstance()))
195                         if (!gAccel::getInstance()->blit(surface, src.surface, area.topLeft(), srcarea, flag))
196                                 continue;
197                 flag &= ~ blitAlphaBlend;
198                 
199                 if ((surface->bpp == 8) && (src.surface->bpp==8))
200                 {
201                         __u8 *srcptr=(__u8*)src.surface->data;
202                         __u8 *dstptr=(__u8*)surface->data;
203         
204                         srcptr+=srcarea.left()*src.surface->bypp+srcarea.top()*src.surface->stride;
205                         dstptr+=area.left()*surface->bypp+area.top()*surface->stride;
206                         for (int y=0; y<area.height(); y++)
207                         {
208                                 if (flag & blitAlphaTest)
209                                 {
210                       // no real alphatest yet
211                                         int width=area.width();
212                                         unsigned char *src=(unsigned char*)srcptr;
213                                         unsigned char *dst=(unsigned char*)dstptr;
214                                                 // use duff's device here!
215                                         while (width--)
216                                         {
217                                                 if (!*src)
218                                                 {
219                                                         src++;
220                                                         dst++;
221                                                 } else
222                                                         *dst++=*src++;
223                                         }
224                                 } else
225                                         memcpy(dstptr, srcptr, area.width()*surface->bypp);
226                                 srcptr+=src.surface->stride;
227                                 dstptr+=surface->stride;
228                         }
229                 } else if ((surface->bpp == 32) && (src.surface->bpp==32))
230                 {
231                         __u32 *srcptr=(__u32*)src.surface->data;
232                         __u32 *dstptr=(__u32*)surface->data;
233         
234                         srcptr+=srcarea.left()+srcarea.top()*src.surface->stride/4;
235                         dstptr+=area.left()+area.top()*surface->stride/4;
236                         for (int y=0; y<area.height(); y++)
237                         {
238                                 if (flag & blitAlphaTest)
239                                 {
240                                         int width=area.width();
241                                         unsigned long *src=(unsigned long*)srcptr;
242                                         unsigned long *dst=(unsigned long*)dstptr;
243
244                                         while (width--)
245                                         {
246                                                 if (!((*src)&0xFF000000))
247                                                 {
248                                                         src++;
249                                                         dst++;
250                                                 } else
251                                                         *dst++=*src++;
252                                         }
253                                 } else if (flag & blitAlphaBlend)
254                                 {
255                                         // uh oh. this is only until hardware accel is working.
256                                         
257                                         int width=area.width();
258                                                         // ARGB color space!
259                                         unsigned char *src=(unsigned char*)srcptr;
260                                         unsigned char *dst=(unsigned char*)dstptr;
261
262 #define BLEND(x, y, a) (y + ((x-y) * a)/256)
263                                         while (width--)
264                                         {
265                                                 unsigned char sa = src[3];
266                                                 unsigned char sr = src[2];
267                                                 unsigned char sg = src[1];
268                                                 unsigned char sb = src[0];
269
270                                                 unsigned char da = dst[3];
271                                                 unsigned char dr = dst[2];
272                                                 unsigned char dg = dst[1];
273                                                 unsigned char db = dst[0];
274                                                 
275                                                 dst[3] = BLEND(0xFF, da, sa);
276                                                 dst[2] = BLEND(sr, dr, sa);
277                                                 dst[1] = BLEND(sg, dg, sa);
278                                                 dst[0] = BLEND(sb, db, sa);
279                                                 
280                                                 src += 4; dst += 4;
281                                         }
282                                 } else
283                                         memcpy(dstptr, srcptr, area.width()*surface->bypp);
284                                 srcptr+=src.surface->stride/4;
285                                 dstptr+=surface->stride/4;
286                         }
287                 } else if ((surface->bpp == 32) && (src.surface->bpp==8))
288                 {       
289                         __u8 *srcptr=(__u8*)src.surface->data;
290                         __u8 *dstptr=(__u8*)surface->data; // !!
291                         __u32 pal[256];
292
293                         for (int i=0; i<256; ++i)
294                         {
295                                 if (src.surface->clut.data && (i<src.surface->clut.colors))
296                                         pal[i]=(src.surface->clut.data[i].a<<24)|(src.surface->clut.data[i].r<<16)|(src.surface->clut.data[i].g<<8)|(src.surface->clut.data[i].b);
297                                 else
298                                         pal[i]=0x010101*i;
299                                 pal[i]^=0xFF000000;
300                         }
301         
302                         srcptr+=srcarea.left()*src.surface->bypp+srcarea.top()*src.surface->stride;
303                         dstptr+=area.left()*surface->bypp+area.top()*surface->stride;
304                         for (int y=0; y<area.height(); y++)
305                         {
306                                 if (flag & blitAlphaTest)
307                                 {
308                       // no real alphatest yet
309                                         int width=area.width();
310                                         unsigned char *src=(unsigned char*)srcptr;
311                                         __u32 *dst=(__u32*)dstptr;
312                                                 // use duff's device here!
313                                         while (width--)
314                                         {
315                                                 if (*src==0x01)
316                                                 {
317                                                         src++;
318                                                         dst++;
319                                                 } else
320                                                         *dst++=pal[*src++];
321                                         }
322                                 } else
323                                 {
324                                         int width=area.width();
325                                         unsigned char *src=(unsigned char*)srcptr;
326                                         __u32 *dst=(__u32*)dstptr;
327                                         while (width--)
328                                                 *dst++=pal[*src++];
329                                 }
330                                 srcptr+=src.surface->stride;
331                                 dstptr+=surface->stride;
332                         }
333                 } else
334                         eFatal("cannot blit %dbpp from %dbpp", surface->bpp, src.surface->bpp);
335         }
336 }
337
338 void gPixmap::mergePalette(const gPixmap &target)
339 {
340         if ((!surface->clut.colors) || (!target.surface->clut.colors))
341                 return;
342
343         gColor *lookup=new gColor[surface->clut.colors];
344
345         for (int i=0; i<surface->clut.colors; i++)
346                 lookup[i].color=target.surface->clut.findColor(surface->clut.data[i]);
347         
348         delete [] surface->clut.data;
349         surface->clut.colors=target.surface->clut.colors;
350         surface->clut.data=new gRGB[surface->clut.colors];
351         memcpy(surface->clut.data, target.surface->clut.data, sizeof(gRGB)*surface->clut.colors);
352
353         __u8 *dstptr=(__u8*)surface->data;
354
355         for (int ay=0; ay<surface->y; ay++)
356         {
357                 for (int ax=0; ax<surface->x; ax++)
358                         dstptr[ax]=lookup[dstptr[ax]];
359                 dstptr+=surface->stride;
360         }
361         
362         delete [] lookup;
363 }
364
365 static inline int sgn(int a)
366 {
367         if (a < 0)
368                 return -1;
369         else if (!a)
370                 return 0;
371         else
372                 return 1;
373 }
374
375 void gPixmap::line(const gRegion &clip, ePoint start, ePoint dst, gColor color)
376 {
377         __u8 *srf8 = 0;
378         __u32 *srf32 = 0; 
379         int stride = surface->stride;
380         
381         if (clip.rects.empty())
382                 return;
383                 
384         __u32 col;
385         if (surface->bpp == 8)
386         {
387                 srf8 = (__u8*)surface->data;
388         } else if (surface->bpp == 32)
389         {
390                 srf32 = (__u32*)surface->data;
391                 
392                 if (surface->clut.data && color < surface->clut.colors)
393                         col=(surface->clut.data[color].a<<24)|(surface->clut.data[color].r<<16)|(surface->clut.data[color].g<<8)|(surface->clut.data[color].b);
394                 else
395                         col=0x10101*color;
396                 col^=0xFF000000;                        
397         }
398         
399         int xa = start.x(), ya = start.y(), xb = dst.x(), yb = dst.y();
400         int dx, dy, x, y, s1, s2, e, temp, swap, i;
401         dy=abs(yb-ya);
402         dx=abs(xb-xa);
403         s1=sgn(xb-xa);
404         s2=sgn(yb-ya);
405         x=xa;
406         y=ya;
407         if (dy>dx)
408         {
409                 temp=dx;
410                 dx=dy;
411                 dy=temp;
412                 swap=1;
413         } else
414                 swap=0;
415         e = 2*dy-dx;
416         
417         int lasthit = 0;
418         for(i=1; i<=dx; i++)
419         {
420                                 /* i don't like this clipping loop, but the only */
421                                 /* other choice i see is to calculate the intersections */
422                                 /* before iterating through the pixels. */
423                                 
424                                 /* one could optimize this because of the ordering */
425                                 /* of the bands. */
426                                 
427                 lasthit = 0;
428                 int a = lasthit;
429                 
430                         /* if last pixel was invisble, first check bounding box */
431                 if (a == -1)
432                 {
433                                 /* check if we just got into the bbox again */
434                         if (clip.extends.contains(x, y))
435                                 lasthit = a = 0;
436                         else
437                                 goto fail;
438                 } else if (!clip.rects[a].contains(x, y))
439                 {
440                         do
441                         {
442                                 ++a;
443                                 if (a == clip.rects.size())
444                                         a = 0;
445                                 if (a == lasthit)
446                                 {
447                                         goto fail;
448                                         lasthit = -1;
449                                 }
450                         } while (!clip.rects[a].contains(x, y));
451                         lasthit = a;
452                 }
453                 
454                 if (srf8)
455                         srf8[y * stride + x] = color;
456                 if (srf32)
457                         srf32[y * stride/4 + x] = col;
458 fail:
459                 while (e>=0)
460                 {
461                         if (swap==1) x+=s1;
462                         else y+=s2;
463                         e-=2*dx;
464                 }
465     if (swap==1)
466         y+=s2;
467                 else
468                         x+=s1;
469                 e+=2*dy;
470         }
471 }
472
473 gColor gPalette::findColor(const gRGB &rgb) const
474 {
475                 /* grayscale? */
476         if (!data)
477                 return (rgb.r + rgb.g + rgb.b) / 3;
478         
479         int difference=1<<30, best_choice=0;
480         for (int t=0; t<colors; t++)
481         {
482                 int ttd;
483                 int td=(signed)(rgb.r-data[t].r); td*=td; td*=(255-data[t].a);
484                 ttd=td;
485                 if (ttd>=difference)
486                         continue;
487                 td=(signed)(rgb.g-data[t].g); td*=td; td*=(255-data[t].a);
488                 ttd+=td;
489                 if (ttd>=difference)
490                         continue;
491                 td=(signed)(rgb.b-data[t].b); td*=td; td*=(255-data[t].a);
492                 ttd+=td;
493                 if (ttd>=difference)
494                         continue;
495                 td=(signed)(rgb.a-data[t].a); td*=td; td*=255;
496                 ttd+=td;
497                 if (ttd>=difference)
498                         continue;
499                 if (!ttd)
500                         return t;
501                 difference=ttd;
502                 best_choice=t;
503         }
504         return best_choice;
505 }
506
507 DEFINE_REF(gPixmap);
508
509 gPixmap::~gPixmap()
510 {
511 }
512
513 gPixmap::gPixmap(gSurface *surface): surface(surface)
514 {
515 }
516
517 gPixmap::gPixmap(eSize size, int bpp, int accel)
518 {
519         surface = new gSurface(size, bpp, accel);
520 }
521