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