yuv colorspace/range + prefs, ffmpeg colorrange probe, x11 direct force colormodel...
[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   "grey8": {
329     "i8": {
330       "r": " int32_t y = *inp++, u = 0x80, v = 0x80;",
331       "w": " *out++ = y; (void)u; (void)v;",
332     },
333     "i16": {
334       "r": " int32_t iy = *inp++, y = (iy<<8) | iy, u = 0x8000, v = 0x8000;",
335       "w": " *out++ = y>>8; (void)u; (void)v;",
336     },
337   },
338
339   "grey16": {
340     "i8": {
341       "r": " int32_t y = *inp++>>8, u = 0x80, v = 0x80;",
342       "w": " *out++ = (y<<8) | y; (void)u; (void)v;",
343     },
344     "i16": {
345       "r": " int32_t y = *inp++, u = 0x8000, v = 0x8000;",
346       "w": " *out++ = y; (void)u; (void)v;",
347     },
348   },
349
350   # alpha component
351   "a8": {
352     "i8": {
353       "r": " z_int a = *inp++;",
354       "w": " *out++ = a;",
355     },
356     "i16": {
357       "r": " z_int a = *inp++; a = (a<<8) | a;",
358       "w": " *out++ = a>>8;",
359     },
360     "fp": {
361       "r": " z_float fa = fclp(*inp++,256);",
362       "w": " *out++ = clp(256,a)",
363     },
364   },
365   "a16": {
366     "i8": {
367       "r": " z_int a = *inp++>>8;",
368       "w": " *out++ = (a<<8) | a;",
369     },
370     "i16": {
371       "r": " z_int a = *inp++;",
372       "w": " *out++ = a;",
373     },
374     "fp": {
375       "r": " z_float fa = fclp(*inp++,65536);",
376       "w": " *out++ = clp(65536,a);",
377     },
378   },
379   "afp": {
380     "i8": {
381       "r": " z_int a = clp(256,*inp++);",
382       "w": " *out++ = fclp(a,256);",
383     },
384     "i16": {
385       "r": " z_int a = clp(65536,*inp++);",
386       "w": " *out++ = fclp(a,65536);",
387     },
388     "fp": {
389       "r": " z_float fa = *inp++;",
390       "w": " *out++ = fa;",
391     },
392   },
393   "afpp": {
394     "i8": {
395       "r": " z_int a = clp(256,*aip++);",
396       "w": " *aop++ = fclp(a,256);",
397     },
398     "i16": {
399       "r": " z_int a = clp(65536,*aip++);",
400       "w": " *aop++ = fclp(a,65536);",
401     },
402     "fp": {
403       "r": " z_float fa = *aip++;",
404       "w": " *aop++ = fa;",
405     },
406   },
407   # alpha blend rgb/black, yuv/black, rgb/bg_color
408   "brgb": {
409     "i8": " r = r*a/0xffu; g = g*a/0xffu; b = b*a/0xffu;",
410     "i16": " r = r*a/0xffffu; g = g*a/0xffffu; b = b*a/0xffffu;",
411     "fp": " r *= fa; g *= fa; b *= fa;",
412   },
413   "byuv": {
414     "i8": " z_int a1 = 0xffu-a;\n" +
415           " y = y*a/0xffu; u = (u*a + 0x80u*a1)/0xffu; v = (v*a + 0x80u*a1)/0xffu;",
416     "i16": " z_int a1 = 0xffffu-a;\n" +
417            " y = y*a/0xffffu; u = (u*a + 0x8000u*a1)/0xffffu; v = (v*a + 0x8000u*a1)/0xffffu;",
418   },
419   "bbg": {
420     "i8": " int a1 = 0xffu-a;\n" +
421         " r = (r*a + bg_r*a1)/0xffu; g = (g*a + bg_g*a1)/0xffu; b = (b*a + bg_b*a1)/0xffu;",
422     "i16": "int a1 = 0xffffu-a;\n" +
423         " r = (r*a + bg_r*a1)/0xffffu; g = (g*a + bg_g*a1)/0xffffu; b = (b*a + bg_b*a1)/0xffffu;",
424     "fp": " float a1 = 1-fa;\n" +
425         " r = (r*fa + bg_r*a1); g = (g*fa + bg_g*a1); b = (b*fa + bg_b*a1);",
426   },
427 }
428
429 cmodels = []
430 bcmodels = {}
431 layout = {}
432 dtype = { None: None }
433 special = {}
434 mx_bcmdl = -1
435
436 def add_cmodel(n, nm, typ=None, *args):
437   global cmodels, bcmodels, layout, dtype, mx_bcmdl
438   cmodels += [nm,]
439   bcmodels[n] = nm
440   if( n > mx_bcmdl ): mx_bcmdl = n
441   dtype[nm] = typ
442   layout[nm] = args
443
444 def specialize(fr_cmdl, to_cmdl, fn):
445   global special
446   special[(fr_cmdl, to_cmdl)] = fn
447
448 add_cmodel( 0, "bc_transparency")
449 add_cmodel( 1, "bc_compressed")
450
451 add_cmodel( 2, "bc_rgb8", "i8", "rgb8")
452 add_cmodel( 3, "bc_rgb565", "i8", "rgb565")
453 add_cmodel( 4, "bc_bgr565", "i8", "bgr565")
454 add_cmodel( 5, "bc_bgr888", "i8", "bgr888")
455 add_cmodel( 6, "bc_bgr8888", "i8", "bgr8888")
456
457 add_cmodel( 9, "bc_rgb888", "i8", "rgb888")
458 add_cmodel(10, "bc_rgba8888", "i8", "rgb888", "a8")
459 add_cmodel(20, "bc_argb8888", "i8", "a8", "rgb888")
460 add_cmodel(21, "bc_abgr8888", "i8", "a8", "bgr888")
461 add_cmodel(11, "bc_rgb161616", "i16", "rgb161616")
462 add_cmodel(12, "bc_rgba16161616", "i16", "rgb161616", "a16")
463 add_cmodel(13, "bc_yuv888", "i8", "yuv888")
464 add_cmodel(14, "bc_yuva8888", "i8", "yuv888", "a8")
465 add_cmodel(15, "bc_yuv161616", "i16", "yuv161616")
466 add_cmodel(16, "bc_yuva16161616", "i16", "yuv161616", "a16")
467 add_cmodel(35, "bc_ayuv16161616", "i16", "a16", "yuv161616")
468
469 add_cmodel(18, "bc_uvy422", "i8", "uyvy8888")
470 add_cmodel(19, "bc_yuv422", "i8", "yuyv8888")
471 add_cmodel(22, "bc_a8")
472 add_cmodel(23, "bc_a16")
473 add_cmodel(31, "bc_a_float")
474 add_cmodel(24, "bc_yuv101010", "i16", "yuv10101010")
475 add_cmodel(25, "bc_vyu888", "i8", "vyu888")
476 add_cmodel(26, "bc_uyva8888", "i8", "uyv888", "a8")
477 add_cmodel(29, "bc_rgb_float", "fp", "rgbfloat")
478 add_cmodel(30, "bc_rgba_float", "fp", "rgbfloat", "afp")
479
480 add_cmodel( 7, "bc_yuv420p", "i8", "yuv420p")
481 add_cmodel( 8, "bc_yuv422p", "i8", "yuv422p")
482 add_cmodel(27, "bc_yuv444p", "i8", "yuv444p")
483 add_cmodel(17, "bc_yuv411p", "i8", "yuv411p")
484 add_cmodel(28, "bc_yuv410p", "i8", "yuv410p")
485 add_cmodel(32, "bc_rgb_floatp", "fp", "rgbfltp")
486 add_cmodel(33, "bc_rgba_floatp", "fp", "rgbfltp", "afpp")
487 add_cmodel(34, "bc_yuv420pi", "i8", "yuv420pi")
488
489 add_cmodel(36, "bc_grey8", "i8", "grey8")
490 add_cmodel(37, "bc_grey16", "i16", "grey16")
491
492 specialize("bc_rgba8888", "bc_transparency", "XFER_rgba8888_to_transparency")
493
494 ctype = {
495   "i8": "uint8_t",
496   "i16": "uint16_t",
497   "fp": "float",
498 }
499
500 adata = {
501   "i8": " z_int a=0xff;",
502   "i16": " z_int a=0xffff;",
503   "fp": " z_float fa=1;",
504 }
505
506 def has_alpha(nm):
507   return nm in ["bc_rgba8888", "bc_argb8888", "bc_abgr8888", \
508     "bc_rgba16161616", "bc_yuva8888", "bc_yuva16161616", "bc_ayuv16161616", \
509     "bc_uyva8888", "bc_rgba_float", "bc_rgba_floatp",]
510
511 def has_bgcolor(fr_cmdl,to_cmdl):
512   return fr_cmdl in ["bc_rgba8888"] and \
513     to_cmdl in ["bc_rgba888", "bc_bgr565", "bc_rgb565", \
514      "bc_bgr888", "bc_rgb888", "bc_bgr8888"]
515
516 def is_specialized(fr_cmdl,to_cmdl):
517   global special
518   return special.get((fr_cmdl, to_cmdl)) is not None
519
520 def is_rgb(nm):
521   return nm in [ "bc_rgb8", "bc_rgb565", "bc_bgr565", \
522     "bc_bgr888", "bc_bgr8888", "bc_rgb888", "bc_rgba8888", \
523     "bc_argb8888", "bc_abgr8888", "bc_rgb", "bc_rgb161616", \
524     "bc_rgba16161616", "bc_rgb_float", "bc_rgba_float", \
525     "bc_rgb_floatp", "bc_rgba_floatp", ]
526
527 def is_yuv(nm):
528   return nm in [ "bc_yuv888", "bc_yuva8888", "bc_yuv161616", \
529     "bc_yuva16161616", "bc_ayuv16161616", "bc_yuv422", "bc_uvy422", "bc_yuv101010", \
530     "bc_vyu888", "bc_uyva8888", "bc_yuv420p", "bc_yuv420pi", "bc_yuv422p", \
531     "bc_yuv444p", "bc_yuv411p", "bc_yuv410p", "bc_grey8", "bc_grey16" ]
532
533 def is_planar(nm):
534   return nm in [ "bc_yuv420p", "bc_yuv420pi", "bc_yuv422p", "bc_yuv444p", \
535     "bc_yuv411p", "bc_yuv410p", "bc_rgb_floatp", "bc_rgba_floatp", ]
536
537 def is_float(nm):
538   return nm in ["bc_rgb_float", "bc_rgba_float", "bc_rgb_floatp", "bc_rgba_floatp", ]
539
540 def gen_xfer_proto(pfx, cls, fr_cmdl, to_cmdl):
541   global dtype, ctype
542   print "%svoid %sxfer_%s_to_%s" % (pfx, cls, fr_cmdl[3:], to_cmdl[3:]),
543   ityp = dtype[fr_cmdl];  fr_typ = ctype[ityp];
544   otyp = dtype[to_cmdl];  to_typ = ctype[otyp];
545   print "(unsigned y0, unsigned y1)",
546
547 def gen_xfer_fn(fr_cmdl, to_cmdl):
548   global layout, dtype, adata
549   ityp = dtype[fr_cmdl];  otyp = dtype[to_cmdl]
550   if( ityp is None or otyp is None ): return
551   # xfr fn header
552   gen_xfer_proto("", class_qual, fr_cmdl, to_cmdl);
553   # xfr fn body
554   print "{"
555   # loops / pointer preload
556   in_cmdl = fr_cmdl[3:] if is_planar(fr_cmdl) else "flat";
557   out_cmdl = to_cmdl[3:] if is_planar(to_cmdl) else "flat";
558   print " xfer_%s_row_out(%s) xfer_%s_row_in(%s)" % \
559      (out_cmdl, ctype[otyp], in_cmdl, ctype[ityp],)
560   # load inp
561   if( is_float(to_cmdl) and is_yuv(fr_cmdl) ):
562     for ic in layout[fr_cmdl]: print "%s" % (base[ic][ityp]['r']),
563     if( ityp == "i8" ):
564       print "\n float r, g, b; YUV::yuv.yuv_to_rgb_8(r, g, b, y, u, v);",
565     elif( ityp == "i16" ):
566       print "\n float r, g, b; YUV::yuv.yuv_to_rgb_16(r, g, b, y, u, v);",
567     if( has_alpha(fr_cmdl) or has_alpha(to_cmdl) ):
568       if( not has_alpha(fr_cmdl) ):
569         print " z_float fa = 1;",
570       elif( ityp == "i8" ):
571         print " float fa = fclp(a,256);",
572       elif( ityp == "i16" ):
573         print " float fa = fclp(a,65536);",
574   else:
575     for ic in layout[fr_cmdl]: print "%s" % (base[ic][otyp]['r']),
576     if( has_alpha(to_cmdl) and not has_alpha(fr_cmdl) ):
577       print "%s" % (adata[otyp]),
578   print ""
579   # xfer
580   if( is_rgb(fr_cmdl) and is_yuv(to_cmdl) ):
581     if( otyp == "i8" ):
582       print " int32_t y, u, v;  YUV::yuv.rgb_to_yuv_8(r, g, b, y, u, v);"
583     elif( otyp == "i16" ):
584       print " int32_t y, u, v;  YUV::yuv.rgb_to_yuv_16(r, g, b, y, u, v);"
585   elif( is_yuv(fr_cmdl) and is_rgb(to_cmdl)):
586     if( otyp == "i8" ):
587       print " int32_t r, g, b;  YUV::yuv.yuv_to_rgb_8(r, g, b, y, u, v);"
588     elif( otyp == "i16" ):
589       print " int32_t r, g, b;  YUV::yuv.yuv_to_rgb_16(r, g, b, y, u, v);"
590   # blend
591   if( has_bgcolor(fr_cmdl,to_cmdl) ):
592     print "%s" % (base["bbg"][otyp])
593   elif( has_alpha(fr_cmdl) and not has_alpha(to_cmdl) ):
594     if( is_rgb(to_cmdl) ):
595       print "%s" % (base["brgb"][otyp])
596     elif( is_yuv(to_cmdl) ):
597       print "%s" % (base["byuv"][otyp])
598   # store out
599   for oc in layout[to_cmdl]:
600     print "%s" % (base[oc][otyp]['w']),
601   print "xfer_end"
602   print "}"
603   print ""
604
605 # output code file
606 class_qual = "BC_Xfer::"
607
608 print "#include \"xfer.h\""
609 print ""
610
611 for fr_cmdl in cmodels:
612   ityp = dtype[fr_cmdl]
613   for to_cmdl in cmodels:
614     otyp = dtype[to_cmdl]
615     if( is_specialized(fr_cmdl, to_cmdl) ):
616       print "  void %s(unsigned y0, unsigned y1);" % (special[(fr_cmdl, to_cmdl)])
617       continue
618     if( ityp is None or otyp is None ): continue
619     gen_xfer_proto("  ", "", fr_cmdl, to_cmdl);
620     print ";"
621 # end of class definition
622 print "};"
623 print ""
624
625 # xfer functions
626 for fr_cmdl in cmodels:
627   for to_cmdl in cmodels:
628     gen_xfer_fn(fr_cmdl, to_cmdl)
629
630 # transfer switch
631 print ""
632 print "void %sxfer()" % class_qual
633 print "{"
634 mx_no = mx_bcmdl + 1
635 print "  static xfer_fn xfns[%d][%d] = {" % (mx_no, mx_no)
636 for fr_no in range(mx_no):
637   fr_cmdl = bcmodels.get(fr_no)
638   ityp = dtype[fr_cmdl]
639   print "  { // %s" % (fr_cmdl.upper() if ityp else "None")
640   n = 0
641   for to_no in range(mx_no):
642     to_cmdl = bcmodels.get(to_no)
643     otyp = dtype[to_cmdl]
644     xfn = special[(fr_cmdl, to_cmdl)] if( is_specialized(fr_cmdl, to_cmdl) ) else \
645       "xfer_%s_to_%s" % (fr_cmdl[3:], to_cmdl[3:]) if ( ityp and otyp ) else None
646     if( n > 72 ): print ""; n = 0
647     if( n == 0 ): print "   ",; n += 4
648     fn = "&%s%s" % (class_qual, xfn) if( xfn ) else "0"
649     print "%s, " % (fn),
650     n += len(fn) + 3
651   print "}, "
652 print "  }; "
653 print "  xfn = xfns[in_colormodel][out_colormodel];"
654 print "  xfer_slices(out_w*out_h/0x80000+1);"
655 print "}"
656 print ""
657 print "#include \"xfer.C\""
658 print ""
659