0c730de07c6ff6b61ab3a3bebef5450b52e1d5b3
[goodguy/history.git] / cinelerra-5.1 / guicast / bccmdl.py
1 #!/bin/python
2
3 base = {
4   "rgb8": {
5     "i8": {
6       "r": " uint32_t in = *inp, r = ((in>>6)&3)*0x55u, g = ((in>>3)&7)*0x24u, b = (in&7)*0x24u;",
7       "w": " uint32_t ot = (r&0xc0u) | ((g>>2)&0x38u) | ((b>>5)&0x07u);\n" +
8            " *out++ = ot;",
9     },
10     "i16": {
11       "r": " uint32_t in = *inp, r = ((in>>6)&3)*0x5555u, g = ((in>>3)&7)*0x2492u, b = (in&7)*0x2492u;",
12       "w": " uint32_t ot = ((r>>8)&0xc0u) | ((g>>10) & 0x38u) >> 2) | ((b>>13)&0x07u);\n" +
13            " *out++ = ot;",
14     },
15     "fp": {
16       "r": " int in = *inp; float r = (in>>6)/3.f, g = ((in>>3)&0x07u)/7.f, b = (in&0x07u)/7.f;",
17       "w": " int vr = clp(4,r), vg = clp(8,g), vb = clp(8,b);\n" +
18            " *out++ = ((vr<<6)&0xc0u) | ((vg<<3)&0x38u) | (vb&0x7u);",
19     },
20   },
21   "rgb565": {
22     "i8": {
23       "r": " uint32_t in = *(uint16_t*)inp;\n" +
24            " int r = (in>>8)&0xf8u, g = (in>>3)&0xfcu, b = (in&0x1fu)<<3;",
25       "w": " uint32_t ot = ((r<<8) & 0xf800u) | ((g<<3) & 0x07e0u) | ((b>>3) & 0x001fu);\n" +
26            " *(uint16_t*)out = ot; out += sizeof(uint16_t);",
27     },
28     "i16": {
29       "r": " uint32_t in = *(uint16_t*)inp;\n" +
30            " int r = in&0xf800u, g = (in<<5)&0xfc00u, b = (in<<11)&0xf800u;",
31       "w": " uint32_t ot = (r&0xf800u) | ((g>>5) & 0x07e0u) | ((b>>11) & 0x001fu);\n" +
32            " *out++ = ot;",
33     },
34     "fp": {
35       "r": " uint32_t in = *(uint16_t*)inp;\n" +
36            " float r = (in>>11)/31.f, g = ((in>>5)&0x3fu)/63.f, b = (in&0x1fu)/31.f;",
37       "w": " uint32_t vr = clp(32,r), vg = clp(64,g), vb = clp(32,b);\n" +
38            " *out++ = ((vr<<11)&0xf800u) | ((vg<<6)&0x07e0u) | (vb&0x001fu);",
39     },
40   },
41   "rgb888": {
42     "i8": {
43       "r": " int r = *inp++, g = *inp++, b = *inp++;",
44       "w": " *out++ = r; *out++ = g; *out++ = b;",
45     },
46     "i16": {
47       "r": " int r = *inp++*0x101u, g = *inp++*0x101u, b = *inp++*0x101u;",
48       "w": " *out++ = r>>8; *out++ = g>>8; *out++ = b>>8;",
49     },
50     "fp": {
51       "r": " float r = fclp(*inp++,256), g=fclp(*inp++,256), b = fclp(*inp++,256);",
52       "w": " *out++ = clp(256,r); *out++ = clp(256,g); *out++ = clp(256,b);",
53     },
54   },
55   "rgb161616": {
56     "i8": {
57       "r": " int r = *inp++>>8, g = *inp++>>8, b = *inp++>>8;",
58       "w": " *out++ = r*0x101u; *out++ = g*0x101u; *out++ = b*0x101u;",
59     },
60     "i16": {
61       "r": " int r = *inp++, g = *inp++, b = *inp++;",
62       "w": " *out++ = r; *out++ = g; *out++ = b;",
63     },
64     "fp": {
65       "r": " float r = fclp(*inp++,65536), g=fclp(*inp++,65536), b = fclp(*inp++,65536);",
66       "w": " *out++ = clp(65536,r); *out++ = clp(65536,g); *out++ = clp(65536,b);",
67     },
68   },
69   "rgbfloat": {
70     "i8": {
71       "r": " int r = clp(256,*inp++), g = clp(256,*inp++), b = clp(256,*inp++);",
72       "w": " *out++ = fclp(r,256); *out++ = fclp(g,256); *out++ = fclp(b,256);",
73     },
74     "i16": {
75       "r": " int r = clp(65536,*inp++), g = clp(65536,*inp++), b = clp(65536,*inp++);",
76       "w": " *out++ = fclp(r,65536); *out++ = fclp(g,65536); *out++ = fclp(b,65536);",
77     },
78     "fp": {
79       "r": " float r = *inp++, g=*inp++, b = *inp++;",
80       "w": " *out++ = r; *out++ = g; *out++ = b;",
81     },
82   },
83
84   "bgr565": {
85     "i8": {
86       "r": " uint32_t in = *(uint16_t*)inp;\n" +
87            " int b = (in>>8)&0xf8u, g = (in>>3)&0xfcu, r = (in&0x1fu)<<3;",
88       "w": " uint32_t ot = ((b&0xf8u)<<8) | ((g&0xfcu)<<3) | ((r&0xf8u)>>3);\n" +
89            " *(uint16_t*)out = ot; out += sizeof(uint16_t)/sizeof(*out);",
90     },
91     "i16": {
92       "r": " uint32_t in = *(uint16_t*)inp;\n" +
93            " int b = in&0xf800u, g = (in<<5)&0xfc00u, r = (in<<11)&0xf800u;",
94       "w": " uint32_t ot = (b&0xf800u) | ((g>>5) & 0x07e0u) << 3) | (r>>11);\n" +
95            " *out++ = ot;",
96     },
97     "fp": {
98       "r": " uint32_t in = *(uint16_t*)inp;\n" +
99            " float b = (in>>11)/31.f, g = ((in>>5)&0x3fu)/63.f, r = (in&0x1fu)/31.f;",
100       "w": " uint32_t vb = clp(32,b), vg = clp(64,g), vr = clp(32,r);\n" +
101            " *out++ = ((vb<<11)&0xf800u) | ((vg<<6)&0x03e0u) | (vr&0x001fu);",
102     },
103   },
104   "bgr888": {
105     "i8": {
106       "r": " int b = *inp++, g = *inp++, r = *inp++;",
107       "w": " *out++ = b; *out++ = g; *out++ = r;",
108     },
109     "i16": {
110       "r": " int b = *inp++*0x101u, g = *inp++*0x101u, r = *inp++*0x101u;",
111       "w": " *out.16++ = b>>8; *out.16++ = g>>8; *out.16++ = r>>8;",
112     },
113     "fp": {
114       "r": " float b = fclp(*inp++,256), g=fclp(*inp++,256), r = fclp(*inp++,256);",
115       "w": " *out++ = clp(256,b); *out++ = clp(256,g); *out++ = clp(256,r);",
116     },
117   },
118   "bgr8888": {
119     "i8": {
120       "r": " int b = *inp++, g = *inp++, r = *inp++;",
121       "w": " *out++ = b; *out++ = g; *out++ = r; ++out;",
122     },
123     "i16": {
124       "r": " int b = *inp++*0x101u, g = *inp++*0x101u, r = *inp++*0x101u;",
125       "w": " *out++ = b>>8; *out++ = g>>8; *out++ = r>>8; ++out;",
126     },
127     "fp": {
128       "r": " float b = fclp(*inp++,256), g=fclp(*inp++,256), r = fclp(*inp++,256);",
129       "w": " *out++ = clp(256,b); *out++ = clp(256,g); *out++ = clp(256,r); ++out;",
130     },
131   },
132   "bgr161616": {
133     "i8": {
134       "r": " int b = *inp++>>8, g = *inp++>>8, r = *inp++>>8;",
135       "w": " *out++ = b*0x101u; *out++ = g*0x101u; *out++ = r*0x101u;",
136     },
137     "i16": {
138       "r": " int b = *inp++, g = *inp++, r = *inp++;",
139       "w": " *out++ = b; *out++ = g; *out++ = r;",
140     },
141     "fp": {
142       "r": " float b = fclp(*inp++,65536), g=fclp(*inp++,65536), r = fclp(*inp++,65536);",
143       "w": " *out++ = clp(65536,b); *out++ = clp(65536,g); *out++ = clp(65536,r);",
144     },
145   },
146   "bgrfloat": {
147     "i8": {
148       "r": " int b = clp(256,*inp++), g = clp(256,*inp++), r = clp(256,*inp++);",
149       "w": " *out++ = fclp(b,256); *out++ = fclp(g,256); *out++ = fclp(r,256);",
150     },
151     "i16": {
152       "r": " int b = clp(65536,*inp++), g = clp(65536,*inp++), r = clp(65536,*inp++);",
153       "w": " *out++ = fclp(b,65536); *out++ = fclp(g,65536); *out++ = fclp(r,65536);",
154     },
155     "fp": {
156       "r": " float b = *inp++, g=*inp++, r = *inp++;",
157       "w": " *out++ = b; *out++ = g; *out++ = r;",
158     },
159   },
160
161   "yuv888": {
162     "i8": {
163       "r": " int32_t y = *inp++*0x010101u, u = *inp++, v = *inp++;",
164       "w": " *out++ = y; *out++ = u; *out++ = v;",
165     },
166     "i16": {
167       "r": " int32_t y = *inp++*0x010101u, u = *inp++<<8, v = *inp++<<8;",
168       "w": " *out++ = y>>8; *out++ = u>>8; *out++ = v>>8;",
169     },
170   },
171   "yuv161616": {
172     "i8": {
173       "r": " int32_t iy = *inp++, y = (iy<<8) | (iy>>8), u = *inp++>>8, v = *inp++>>8;",
174       "w": " *out++ = y<<8; *out++ = u<<8; *out++ = v<<8;",
175     },
176     "i16": {
177       "r": " int32_t iy = *inp++, y = (iy<<8) | (iy>>8), u = *inp++, v = *inp++;",
178       "w": " *out++ = y; *out++ = u; *out++ = v;",
179     },
180   },
181
182   "yuyv8888": {
183     "i8": {
184       "r": " int32_t iy = inp[(i&1)<<1], y = iy*0x010101u, u = inp[1], v = inp[3];",
185       "w": " if( !(j&1) ) { *out++ = y; *out = u; out[2] = v; }\n" +
186            " else { *out++ = u; *out++= y; *out++ = v; }",
187     },
188     "i16": {
189       "r": " int32_t iy = inp[(i&1)<<1], y = iy*0x010101u, u = inp[1]<<8, v = inp[3]<<8;",
190       "w": " if( !(j&1) ) { *out++ = y>>8; *out = u>>8; out[2] = v>>8; }\n" +
191            " else { *out++ = u>>8; *out++= y>>8; *out++ = v>>8; }",
192     },
193   },
194
195   "uyvy8888": {
196     "i8": {
197       "r": " int32_t u = inp[0], iy = inp[((i&1)<<1)+1], y = iy*0x010101u, v = inp[2];",
198       "w": " if( !(j&1) ) { *out++ = u; *out++ = y; *out++ = v; *out = y; }\n" +
199            " else { *out++= y; }",
200     },
201     "i16": {
202       "r": " int32_t u = inp[0]<<8, iy = inp[((i&1)<<1)+1], y = iy*0x010101u, v = inp[2]<<8;",
203       "w": " if( !(j&1) ) { *out++ = u>>8; *out++ = y>>8; *out++ = v>>8; *out = y>>8; }\n" +
204            " else { *out++= y>>8; }",
205     },
206   },
207
208   "yuv10101010": {
209     "i8": {
210       "r": " uint32_t it = *(uint32_t*)inp;\n" +
211            " int32_t y = ((it>>22)&0x3ffu)*0x4010u, u = (it>>14)&0xffu, v = (it>>4)&0xffu;",
212       "w": " uint32_t ot = (y<<24) | (u<<14) | (v<<4);\n" +
213            " *(uint32_t*)out = ot; out += sizeof(uint32_t)/sizeof(*out);",
214     },
215     "i16": {
216       "r": " uint32_t it = *(uint32_t*)inp;\n" +
217            " int32_t y = ((it>>22)&0x3ffu)*0x4010u, u = (it>>6)&0xffc0u, v = (it<<4)&0xffc0u;",
218       "w": " uint32_t ot = ((y&0xffc0u)<<16) | ((u&0xffc0u)<<6) | ((v&0xffc0u)>>4);\n" +
219            " *(uint32_t*)out = ot; out += sizeof(uint32_t)/sizeof(*out);",
220     },
221   },
222
223   "vyu888": {
224     "i8": {
225       "r": " int32_t v = *inp++, y = *inp++*0x010101u, u = *inp++;",
226       "w": " *out++ = v; *out++ = y; *out++ = u;",
227     },
228     "i16": {
229       "r": " int32_t v = *inp++<<8, y = *inp++*0x010101u, u = *inp++<<8;",
230       "w": " *out++ = v>>8; *out++ = y>>8; *out++ = u>>8;",
231     },
232   },
233
234   "uyv888": {
235     "i8": {
236       "r": " int32_t u = *inp++, y = *inp++*0x010101u, v = *inp++;",
237       "w": " *out++ = u; *out++ = y; *out++ = v;",
238     },
239     "i16": {
240       "r": " int32_t u = *inp++<<8, y = *inp++*0x010101u, v = *inp++<<8;",
241       "w": " *out++ = u>>8; *out++ = y>>8; *out++ = v>>8;",
242     },
243   },
244
245   "yuv420p": {
246     "i8": {
247       "r": " int32_t y = *yip*0x010101u, u = *uip, v = *vip;",
248       "w": " yop[j] = y;  uop[j/2] = u;  vop[j/2] = v;",
249     },
250     "i16": {
251       "r": " int32_t y = *yip*0x010101u, u = *uip<<8, v = *vip<<8;",
252       "w": " yop[j] = y>>8;  uop[j/2] = u>>8;  vop[j/2] = v>>8;",
253     },
254   },
255
256   "yuv422p": {
257     "i8": {
258       "r": " int32_t y = *yip*0x010101u, u = *uip, v = *vip;",
259       "w": " yop[j] = y;  uop[j/2] = u;  vop[j/2] = v;",
260     },
261     "i16": {
262       "r": " int32_t y = *yip*0x010101u, u = *uip<<8, v = *vip<<8;",
263       "w": " yop[j] = y>>8;  uop[j/2] = u>>8;  vop[j/2] = v>>8;",
264     },
265   },
266
267   "yuv444p": {
268     "i8": {
269       "r": " int32_t y = *yip*0x010101u, u = *uip, v = *vip;",
270       "w": " yop[j] = y;  uop[j] = u;  vop[j] = v;",
271     },
272     "i16": {
273       "r": " int32_t y = *yip*0x010101u, u = *uip<<8, v = *vip<<8;",
274       "w": " yop[j] = y>>8;  uop[j] = u>>8;  vop[j] = v>>8;",
275     },
276   },
277
278   "yuv411p": {
279     "i8": {
280       "r": " int32_t y = *yip*0x010101u, u = *uip, v = *vip;",
281       "w": " yop[j] = y;  uop[j/4] = u;  vop[j/4] = v;",
282     },
283     "i16": {
284       "r": " int32_t y = *yip*0x010101u, u = *uip<<8, v = *vip<<8;",
285       "w": " yop[j] = y>>8;  uop[j/4] = u>>8;  vop[j/4] = v>>8;",
286     },
287   },
288
289   "yuv410p": {
290     "i8": {
291       "r": " int32_t y = *yip*0x010101u, u = *uip, v = *vip;",
292       "w": " yop[j] = y;  uop[j/4] = u;  vop[j/4] = v;",
293     },
294     "i16": {
295       "r": " int32_t y = *yip*0x010101u, u = *uip<<8, v = *vip<<8;",
296       "w": " yop[j] = y>>8;  uop[j/4] = u>>8;  vop[j/4] = v>>8;",
297     },
298   },
299
300   "rgbfltp": {
301     "i8": {
302       "r": " int r = clp(256,*rip++), g = clp(256,*gip++), b = clp(256,*bip++);",
303       "w": " *rop++ = fclp(r,256); *gop++ = fclp(g,256); *bop++ = fclp(b,256);",
304     },
305     "i16": {
306       "r": " int r = clp(65536,*rip++), g = clp(65536,*gip++), b = clp(65536,*bip++);",
307       "w": " *rop++ = fclp(r,65536); *gop++ = fclp(g,65536); *bop++ = fclp(b,65536);",
308     },
309     "fp": {
310       "r": " float r = *rip++, g = *gip++, b = *bip++;",
311       "w": " *rop++ = r; *gop++ = g; *bop++ = b;",
312     },
313   },
314
315   # alpha component
316   "a8": {
317     "i8": {
318       "r": " z_int a = *inp++;",
319       "w": " *out++ = a;",
320     },
321     "i16": {
322       "r": " z_int a = *inp++<<8;",
323       "w": " *out++ = a>>8;",
324     },
325     "fp": {
326       "r": " z_float fa = fclp(*inp++,256);",
327       "w": " *out++ = clp(256,a)",
328     },
329   },
330   "a16": {
331     "i8": {
332       "r": " z_int a = *inp++>>8;",
333       "w": " *out++ = a<<8;",
334     },
335     "i16": {
336       "r": " z_int a = *inp++;",
337       "w": " *out++ = a;",
338     },
339     "fp": {
340       "r": " z_float fa = fclp(*inp++,65536);",
341       "w": " *out++ = clp(65536,a);",
342     },
343   },
344   "afp": {
345     "i8": {
346       "r": " z_int a = clp(256,*inp++);",
347       "w": " *out++ = fclp(a,256);",
348     },
349     "i16": {
350       "r": " z_int a = clp(65536,*inp++);",
351       "w": " *out++ = fclp(a,65536);",
352     },
353     "fp": {
354       "r": " z_float fa = *inp++;",
355       "w": " *out++ = fa;",
356     },
357   },
358   "afpp": {
359     "i8": {
360       "r": " z_int a = clp(256,*aip++);",
361       "w": " *aop++ = fclp(a,256);",
362     },
363     "i16": {
364       "r": " z_int a = clp(65536,*aip++);",
365       "w": " *aop++ = fclp(a,65536);",
366     },
367     "fp": {
368       "r": " z_float fa = *aip++;",
369       "w": " *aop++ = fa;",
370     },
371   },
372   # alpha blend rgb/black, yuv/black, rgb/bg_color
373   "brgb": {
374     "i8": " r = r*a/0xffu; g = g*a/0xffu; b = b*a/0xffu;",
375     "i16": " r = r*a/0xffffu; g = g*a/0xffffu; b = b*a/0xffffu;",
376     "fp": " r *= fa; g *= fa; b *= fa;",
377   },
378   "byuv": {
379     "i8": " z_int a1 = 0xffu-a;\n" +
380           " y = y*a/0xffu; u = (u*a + 0x80u*a1)/0xffu; v = (v*a + 0x80u*a1)/0xffu;",
381     "i16": " z_int a1 = 0xffffu-a;\n" +
382            " y = y*a/0xffffu; u = (u*a + 0x8000u*a1)/0xffffu; v = (v*a + 0x8000u*a1)/0xffffu;",
383   },
384   "bbg": {
385     "i8": " int a1 = 0xffu-a;\n" +
386         " r = (r*a + bg_r*a1)/0xffu; g = (g*a + bg_g*a1)/0xffu; b = (b*a + bg_b*a1)/0xffu;",
387     "i16": "int a1 = 0xffffu-a;\n" +
388         " r = (r*a + bg_r*a1)/0xffffu; g = (g*a + bg_g*a1)/0xffffu; b = (b*a + bg_b*a1)/0xffffu;",
389     "fp": " float a1 = 1-fa;\n" +
390         " r = (r*fa + bg_r*a1); g = (g*fa + bg_g*a1); b = (b*fa + bg_b*a1);",
391   },
392 }
393
394 cmodels = []
395 bcmodels = {}
396 layout = {}
397 dtype = { None: None }
398 special = {}
399 mx_bcmdl = -1
400
401 def add_cmodel(n, nm, typ=None, *args):
402   global cmodels, bcmodels, layout, dtype, mx_bcmdl
403   cmodels += [nm,]
404   bcmodels[n] = nm
405   if( n > mx_bcmdl ): mx_bcmdl = n
406   dtype[nm] = typ
407   layout[nm] = args
408
409 def specialize(fr_cmdl, to_cmdl, fn):
410   global special
411   special[(fr_cmdl, to_cmdl)] = fn
412
413 add_cmodel( 0, "bc_transparency")
414 add_cmodel( 1, "bc_compressed")
415
416 add_cmodel( 2, "bc_rgb8", "i8", "rgb8")
417 add_cmodel( 3, "bc_rgb565", "i8", "rgb565")
418 add_cmodel( 4, "bc_bgr565", "i8", "bgr565")
419 add_cmodel( 5, "bc_bgr888", "i8", "bgr888")
420 add_cmodel( 6, "bc_bgr8888", "i8", "bgr8888")
421
422 add_cmodel( 9, "bc_rgb888", "i8", "rgb888")
423 add_cmodel(10, "bc_rgba8888", "i8", "rgb888", "a8")
424 add_cmodel(20, "bc_argb8888", "i8", "a8", "rgb888")
425 add_cmodel(21, "bc_abgr8888", "i8", "a8", "bgr888")
426 add_cmodel(11, "bc_rgb161616", "i16", "rgb161616")
427 add_cmodel(12, "bc_rgba16161616", "i16", "rgb161616", "a16")
428 add_cmodel(13, "bc_yuv888", "i8", "yuv888")
429 add_cmodel(14, "bc_yuva8888", "i8", "yuv888", "a8")
430 add_cmodel(15, "bc_yuv161616", "i16", "yuv161616")
431 add_cmodel(16, "bc_yuva16161616", "i16", "yuv161616", "a16")
432
433 add_cmodel(18, "bc_yuv422", "i8", "yuyv8888")
434 add_cmodel(19, "bc_uvy422", "i8", "uyvy8888")
435 add_cmodel(22, "bc_a8")
436 add_cmodel(23, "bc_a16")
437 add_cmodel(31, "bc_a_float")
438 add_cmodel(24, "bc_yuv101010", "i16", "yuv10101010")
439 add_cmodel(25, "bc_vyu888", "i8", "vyu888")
440 add_cmodel(26, "bc_uyva8888", "i8", "uyv888", "a8")
441 add_cmodel(29, "bc_rgb_float", "fp", "rgbfloat")
442 add_cmodel(30, "bc_rgba_float", "fp", "rgbfloat", "afp")
443
444 add_cmodel( 7, "bc_yuv420p", "i8", "yuv420p")
445 add_cmodel( 8, "bc_yuv422p", "i8", "yuv422p")
446 add_cmodel(27, "bc_yuv444p", "i8", "yuv444p")
447 add_cmodel(17, "bc_yuv411p", "i8", "yuv411p")
448 add_cmodel(28, "bc_yuv410p", "i8", "yuv410p")
449 add_cmodel(32, "bc_rgb_floatp", "fp", "rgbfltp")
450 add_cmodel(33, "bc_rgba_floatp", "fp", "rgbfltp", "afpp")
451
452 specialize("bc_rgba8888", "bc_transparency", "XFER_rgba8888_to_transparency")
453
454 ctype = {
455   "i8": "uint8_t",
456   "i16": "uint16_t",
457   "fp": "float",
458 }
459
460 adata = {
461   "i8": " z_int a=0xff;",
462   "i16": " z_int a=0xffff;",
463   "fp": " z_float fa=1;",
464 }
465
466 def has_alpha(nm):
467   return nm in ["bc_rgba8888", "bc_argb8888", "bc_abgr8888", \
468     "bc_rgba16161616", "bc_yuva8888", "bc_yuva16161616", \
469     "bc_uyva8888", "bc_rgba_float", "bc_rgba_floatp",]
470
471 def has_bgcolor(fr_cmdl,to_cmdl):
472   return fr_cmdl in ["bc_rgba8888"] and \
473     to_cmdl in ["bc_rgba888", "bc_bgr565", "bc_rgb565", \
474      "bc_bgr888", "bc_rgb888", "bc_bgr8888"]
475
476 def is_specialized(fr_cmdl,to_cmdl):
477   global special
478   return special.get((fr_cmdl, to_cmdl)) is not None
479
480 def is_rgb(nm):
481   return nm in [ "bc_rgb8", "bc_rgb565", "bc_bgr565", \
482     "bc_bgr888", "bc_bgr8888", "bc_rgb888", "bc_rgba8888", \
483     "bc_argb8888", "bc_abgr8888", "bc_rgb", "bc_rgb161616", \
484     "bc_rgba16161616", "bc_rgb_float", "bc_rgba_float", \
485     "bc_rgb_floatp", "bc_rgba_floatp", ]
486
487 def is_yuv(nm):
488   return nm in [ "bc_yuv888", "bc_yuva8888", "bc_yuv161616", \
489     "bc_yuva16161616", "bc_yuv422", "bc_uvy422", "bc_yuv101010", \
490     "bc_vyu888", "bc_uyva8888", "bc_yuv420p", "bc_yuv422p", \
491     "bc_yuv444p", "bc_yuv411p", "bc_yuv410p", ]
492
493 def is_planar(nm):
494   return nm in [ "bc_yuv420p", "bc_yuv422p", "bc_yuv444p", \
495     "bc_yuv411p", "bc_yuv410p", "bc_rgb_floatp", "bc_rgba_floatp", ]
496
497 def is_float(nm):
498   return nm in ["bc_rgb_float", "bc_rgba_float", "bc_rgb_floatp", "bc_rgba_floatp", ]
499
500 def gen_xfer_proto(pfx, cls, fr_cmdl, to_cmdl):
501   global dtype, ctype
502   print "%svoid %sxfer_%s_to_%s" % (pfx, cls, fr_cmdl[3:], to_cmdl[3:]),
503   ityp = dtype[fr_cmdl];  fr_typ = ctype[ityp];
504   otyp = dtype[to_cmdl];  to_typ = ctype[otyp];
505   print "(unsigned y0, unsigned y1)",
506
507 def gen_xfer_fn(fr_cmdl, to_cmdl):
508   global layout, dtype, adata
509   ityp = dtype[fr_cmdl];  otyp = dtype[to_cmdl]
510   if( ityp is None or otyp is None ): return
511   # xfr fn header
512   gen_xfer_proto("", class_qual, fr_cmdl, to_cmdl);
513   # xfr fn body
514   print "{"
515   # loops / pointer preload
516   in_cmdl = fr_cmdl[3:] if is_planar(fr_cmdl) else "flat";
517   out_cmdl = to_cmdl[3:] if is_planar(to_cmdl) else "flat";
518   print " xfer_%s_row_out(%s) xfer_%s_row_in(%s)" % \
519      (out_cmdl, ctype[otyp], in_cmdl, ctype[ityp],)
520   # load inp
521   if( is_float(to_cmdl) and is_yuv(fr_cmdl) ):
522     for ic in layout[fr_cmdl]: print "%s" % (base[ic][ityp]['r']),
523     if( ityp == "i8" ):
524       print "\n float fy = y/16777215.f, r, g, b; YUV_TO_FLOAT(fy, u, v, r, g, b);",
525     elif( ityp == "i16" ):
526       print "\n float fy = y/16777215.f, r, g, b; YUV16_TO_RGB_FLOAT(fy, u, v, r, g, b);",
527     if( has_alpha(fr_cmdl) or has_alpha(to_cmdl) ):
528       if( not has_alpha(fr_cmdl) ):
529         print " z_float fa = 1;",
530       elif( ityp == "i8" ):
531         print " float fa = fclp(a,256);",
532       elif( ityp == "i16" ):
533         print " float fa = fclp(a,65536);",
534   else:
535     for ic in layout[fr_cmdl]: print "%s" % (base[ic][otyp]['r']),
536     if( has_alpha(to_cmdl) and not has_alpha(fr_cmdl) ):
537       print "%s" % (adata[otyp]),
538   print ""
539   # xfer
540   if( is_rgb(fr_cmdl) and is_yuv(to_cmdl) ):
541     if( otyp == "i8" ):
542       print " int32_t y, u, v;  RGB_TO_YUV(y, u, v, r, g, b);"
543     elif( otyp == "i16" ):
544       print " int32_t y, u, v;  RGB_TO_YUV16(y, u, v, r, g, b);"
545   elif( is_yuv(fr_cmdl) and is_rgb(to_cmdl)):
546     if( otyp == "i8" ):
547       print " int32_t r, g, b;  YUV_TO_RGB(y, u, v, r, g, b);"
548     elif( otyp == "i16" ):
549       print " int32_t r, g, b;  YUV_TO_RGB16(y, u, v, r, g, b);"
550   elif( is_yuv(fr_cmdl) and is_yuv(to_cmdl) ):
551     if( otyp == "i8" ):
552       print " y >>= 16;",
553     elif( otyp == "i16" ):
554       print " y >>= 8;",
555   # blend
556   if( has_bgcolor(fr_cmdl,to_cmdl) ):
557     print "%s" % (base["bbg"][otyp])
558   elif( has_alpha(fr_cmdl) and not has_alpha(to_cmdl) ):
559     if( is_rgb(to_cmdl) ):
560       print "%s" % (base["brgb"][otyp])
561     elif( is_yuv(to_cmdl) ):
562       print "%s" % (base["byuv"][otyp])
563   # store out
564   for oc in layout[to_cmdl]:
565     print "%s" % (base[oc][otyp]['w']),
566   print "xfer_end"
567   print "}"
568   print ""
569
570 # output code file
571 class_qual = "BC_Xfer::"
572
573 print "#include \"xfer.h\""
574 print ""
575
576 for fr_cmdl in cmodels:
577   ityp = dtype[fr_cmdl]
578   for to_cmdl in cmodels:
579     otyp = dtype[to_cmdl]
580     if( is_specialized(fr_cmdl, to_cmdl) ):
581       print "  void %s(unsigned y0, unsigned y1);" % (special[(fr_cmdl, to_cmdl)])
582       continue
583     if( ityp is None or otyp is None ): continue
584     gen_xfer_proto("  ", "", fr_cmdl, to_cmdl);
585     print ";"
586 # end of class definition
587 print "};"
588 print ""
589
590 # xfer functions
591 for fr_cmdl in cmodels:
592   for to_cmdl in cmodels:
593     gen_xfer_fn(fr_cmdl, to_cmdl)
594
595 # transfer switch
596 print ""
597 print "void %sxfer()" % class_qual
598 print "{"
599 mx_no = mx_bcmdl + 1
600 print "  static xfer_fn xfns[%d][%d] = {" % (mx_no, mx_no)
601 for fr_no in range(mx_no):
602   fr_cmdl = bcmodels.get(fr_no)
603   ityp = dtype[fr_cmdl]
604   print "  { // %s" % (fr_cmdl.upper() if ityp else "None")
605   n = 0
606   for to_no in range(mx_no):
607     to_cmdl = bcmodels.get(to_no)
608     otyp = dtype[to_cmdl]
609     xfn = special[(fr_cmdl, to_cmdl)] if( is_specialized(fr_cmdl, to_cmdl) ) else \
610       "xfer_%s_to_%s" % (fr_cmdl[3:], to_cmdl[3:]) if ( ityp and otyp ) else None
611     if( n > 72 ): print ""; n = 0
612     if( n == 0 ): print "   ",; n += 4
613     fn = "&%s%s" % (class_qual, xfn) if( xfn ) else "0"
614     print "%s, " % (fn),
615     n += len(fn) + 3
616   print "}, "
617 print "  }; "
618 print "  xfn = xfns[in_colormodel][out_colormodel];"
619 print "  xfer_slices(out_w*out_h/0x80000+1);"
620 print "}"
621 print ""
622 print "#include \"xfer.C\""
623 print ""
624