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