lib/h.263/mkvideo.c

   1 /* mkvideo.c
   2    Create a video file.
   3
   4    Part of the swftools package.
   5
   6    Copyright (c) 2003 Matthias Kramm <kramm@quiss.org> */
   7
   8 #include <stdlib.h>
   9 #include <stdio.h>
  10 #include <assert.h>
  11 #include <math.h>
  12 #include "../rfxswf.h"
  13 #include "h263tables.c"
  14 #include "swfvideo.h"
  15
  16 void swf_SetVideoStreamDefine(TAG*tag, VIDEOSTREAM*stream, U16 frames, U16 width, U16 height)
  17 {
  18     width=width&~15; height=height&~15;
  19     swf_SetU16(tag, frames);
  20     swf_SetU16(tag, width);
  21     swf_SetU16(tag, height);
  22     swf_SetU8(tag, 1); /* smoothing on */
  23     swf_SetU8(tag, 2); /* codec = h.263 sorenson spark */
  24
  25     memset(stream, 0, sizeof(VIDEOSTREAM));
  26     stream->olinex = width;
  27     width+=15;width&=~15;
  28     height+=15;height&=~15;
  29     stream->linex = width;
  30     stream->width = width;
  31     stream->height = height;
  32     stream->bbx = width/16;
  33     stream->bby = height/16;
  34     stream->current = (YUV*)malloc(width*height*sizeof(YUV));
  35     stream->oldpic = (YUV*)malloc(width*height*sizeof(YUV));
  36     stream->mvdx = (int*)malloc(stream->bbx*stream->bby*sizeof(int));
  37     stream->mvdy = (int*)malloc(stream->bbx*stream->bby*sizeof(int));
  38     stream->do_motion = 0;
  39
  40     memset(stream->oldpic, 0, width*height*sizeof(YUV));
  41     memset(stream->current, 0, width*height*sizeof(YUV));
  42 }
  43 void swf_VideoStreamClear(VIDEOSTREAM*stream)
  44 {
  45     free(stream->oldpic);stream->oldpic = 0;
  46     free(stream->current);stream->current = 0;
  47     free(stream->mvdx);stream->mvdx=0;
  48     free(stream->mvdy);stream->mvdy=0;
  49 }
  50
  51 typedef struct _block_t
  52 {
  53     int y1[64];
  54     int y2[64];
  55     int y3[64];
  56     int y4[64];
  57     int u[64];
  58     int v[64];
  59 } block_t;
  60
  61 typedef struct _fblock_t
  62 {
  63     double y1[64];
  64     double y2[64];
  65     double y3[64];
  66     double y4[64];
  67     double u[64];
  68     double v[64];
  69 } fblock_t;
  70
  71 static int zigzagtable[64] = {
  72     0, 1, 5, 6, 14, 15, 27, 28,
  73     2, 4, 7, 13, 16, 26, 29, 42,
  74     3, 8, 12, 17, 25, 30, 41, 43,
  75     9, 11, 18, 24, 31, 40, 44, 53,
  76     10, 19, 23, 32, 39, 45, 52, 54,
  77     20, 22, 33, 38, 46, 51, 55, 60,
  78     21, 34, 37, 47, 50, 56, 59, 61,
  79     35, 36, 48, 49, 57, 58, 62, 63};
  80
  81 static void fzigzag(double*src)
  82 {
  83     double tmp[64];
  84     int t;
  85     for(t=0;t<64;t++) {
  86         ((int*)&tmp[zigzagtable[t]])[0] = ((int*)&src[t])[0];
  87         ((int*)&tmp[zigzagtable[t]])[1] = ((int*)&src[t])[1];
  88     }
  89     memcpy(src, tmp, sizeof(double)*64);
  90 }
  91
  92 #define PI 3.14159265358979
  93 #define SQRT2 1.414214
  94 #define RSQRT2 (1.0/1.414214)
  95
  96 static double table[8][8] =
  97 {
  98 {0.707106781186548,0.707106781186548,0.707106781186548,0.707106781186548,0.707106781186548,0.707106781186548,0.707106781186548,0.707106781186548},
  99 {0.980785280403230,0.831469612302545,0.555570233019602,0.195090322016128,-0.195090322016128,-0.555570233019602,-0.831469612302545,-0.980785280403230},
 100 {0.923879532511287,0.382683432365090,-0.382683432365090,-0.923879532511287,-0.923879532511287,-0.382683432365090,0.382683432365090,0.923879532511287},
 101 {0.831469612302545,-0.195090322016128,-0.980785280403230,-0.555570233019602,0.555570233019602,0.980785280403230,0.195090322016129,-0.831469612302545},
 102 {0.707106781186548,-0.707106781186547,-0.707106781186548,0.707106781186547,0.707106781186548,-0.707106781186547,-0.707106781186547,0.707106781186547},
 103 {0.555570233019602,-0.980785280403230,0.195090322016128,0.831469612302545,-0.831469612302545,-0.195090322016128,0.980785280403231,-0.555570233019602},
 104 {0.382683432365090,-0.923879532511287,0.923879532511287,-0.382683432365090,-0.382683432365091,0.923879532511287,-0.923879532511286,0.382683432365090},
 105 {0.195090322016128,-0.555570233019602,0.831469612302545,-0.980785280403231,0.980785280403230,-0.831469612302545,0.555570233019602,-0.195090322016129}
 106 };
 107
 108 static void dct(double*src)
 109 {
 110     double tmp[64];
 111     int x,y,u,v,t;
 112
 113     for(v=0;v<8;v++)
 114     for(u=0;u<8;u++)
 115     {
 116         double c = 0;
 117         for(x=0;x<8;x++)
 118         {
 119             c+=table[u][x]*src[v*8+x];
 120         }
 121         tmp[v*8+u] = c;
 122     }
 123     for(u=0;u<8;u++)
 124     for(v=0;v<8;v++)
 125     {
 126         double c = 0;
 127         for(y=0;y<8;y++)
 128         {
 129             c+=table[v][y]*tmp[y*8+u];
 130         }
 131         src[v*8+u] = c*0.25;
 132     }
 133 }
 134
 135 static void idct(double*src)
 136 {
 137     double tmp[64];
 138     int x,y,u,v;
 139     for(y=0;y<8;y++)
 140     for(x=0;x<8;x++)
 141     {
 142         double c = 0;
 143         for(u=0;u<8;u++)
 144         {
 145             c+=table[u][x]*src[y*8+u];
 146         }
 147         tmp[y*8+x] = c;
 148     }
 149     for(y=0;y<8;y++)
 150     for(x=0;x<8;x++)
 151     {
 152         double c = 0;
 153         for(v=0;v<8;v++)
 154         {
 155             c+=table[v][y]*tmp[v*8+x];
 156         }
 157         src[y*8+x] = c*0.25;
 158     }
 159 }
 160
 161 static inline int truncate256(int a)
 162 {
 163     if(a>255) return 255;
 164     if(a<0) return 0;
 165     return a;
 166 }
 167
 168 static void getregion(fblock_t* bb, YUV*pic, int posx, int posy, int linex)
 169 {
 170     YUV*p1 = &pic[posy*linex+posx];
 171     YUV*p2 = p1;
 172     int y1=0, y2=0, y3=0, y4=0;
 173     int u=0,v=0;
 174     int x,y;
 175     for(y=0;y<8;y++) {
 176         for(x=0;x<8;x++) {
 177             bb->u[u++] = (p2[x*2].u + p2[x*2+1].u + p2[linex+x*2].u + p2[linex+x*2+1].u)/4;
 178             bb->v[v++] = (p2[x*2].v + p2[x*2+1].v + p2[linex+x*2].v + p2[linex+x*2+1].v)/4;
 179             bb->y1[y1++] = p1[x].y;
 180             bb->y2[y2++] = p1[x+8].y;
 181             bb->y3[y3++] = p1[linex*8+x].y;
 182             bb->y4[y4++] = p1[linex*8+x+8].y;
 183         }
 184         p1+=linex;
 185         p2+=linex*2;
 186     }
 187 }
 188 static void rgb2yuv(YUV*dest, RGBA*src, int dlinex, int slinex, int width, int height)
 189 {
 190     int x,y;
 191     for(y=0;y<height;y++) {
 192         for(x=0;x<width;x++) {
 193             int r,g,b;
 194             r = src[y*slinex+x].r;
 195             g = src[y*slinex+x].g;
 196             b = src[y*slinex+x].b;
 197             /*dest[y*dlinex+x].y = (r*0.299 + g*0.587 + b*0.114);
 198             dest[y*dlinex+x].u = (r*-0.169 + g*-0.332 + b*0.500 + 128.0);
 199             dest[y*dlinex+x].v = (r*0.500 + g*-0.419 + b*-0.0813 + 128.0);*/
 200             dest[y*dlinex+x].y = (r*((int)( 0.299*256)) + g*((int)( 0.587*256)) + b*((int)( 0.114 *256)))>>8;
 201             dest[y*dlinex+x].u = (r*((int)(-0.169*256)) + g*((int)(-0.332*256)) + b*((int)( 0.500 *256))+ 128*256)>>8;
 202             dest[y*dlinex+x].v = (r*((int)( 0.500*256)) + g*((int)(-0.419*256)) + b*((int)(-0.0813*256))+ 128*256)>>8;
 203         }
 204     }
 205 }
 206 static void copyregion(VIDEOSTREAM*s, YUV*dest, YUV*src, int bx, int by)
 207 {
 208     YUV*p1 = &src[by*s->linex*16+bx*16];
 209     YUV*p2 = &dest[by*s->linex*16+bx*16];
 210     int y;
 211     for(y=0;y<16;y++) {
 212         memcpy(p1, p2, 16*sizeof(YUV));
 213         p1+=s->linex;p2+=s->linex;
 214     }
 215 }
 216
 217 static void yuv2rgb(RGBA*dest, YUV*src, int linex, int width, int height)
 218 {
 219     int x,y;
 220     for(y=0;y<height;y++) {
 221         for(x=0;x<width;x++) {
 222             int u,v,yy;
 223             u = src[y*linex+x].u;
 224             v = src[y*linex+x].v;
 225             yy = src[y*linex+x].y;
 226             dest[y*linex+x].r = truncate256(yy + ((360*(v-128))>>8));
 227             dest[y*linex+x].g = truncate256(yy - ((88*(u-128)+183*(v-128))>>8));
 228             dest[y*linex+x].b = truncate256(yy + ((455 * (u-128))>>8));
 229         }
 230     }
 231 }
 232 static void copyblock(VIDEOSTREAM*s, YUV*dest, block_t*b, int bx, int by)
 233 {
 234     YUV*p1 = &dest[(by*16)*s->linex+bx*16];
 235     YUV*p2 = &dest[(by*16+8)*s->linex+bx*16];
 236     int x,y;
 237     for(y=0;y<8;y++) {
 238         for(x=0;x<8;x++) {
 239             int u,v,yy;
 240             p1[x+0].u = b->u[(y/2)*8+(x/2)];
 241             p1[x+0].v = b->v[(y/2)*8+(x/2)];
 242             p1[x+0].y = b->y1[y*8+x];
 243             p1[x+8].u = b->u[(y/2)*8+(x/2)+4];
 244             p1[x+8].v = b->v[(y/2)*8+(x/2)+4];
 245             p1[x+8].y = b->y2[y*8+x];
 246             p2[x+0].u = b->u[(y/2+4)*8+(x/2)];
 247             p2[x+0].v = b->v[(y/2+4)*8+(x/2)];
 248             p2[x+0].y = b->y3[y*8+x];
 249             p2[x+8].u = b->u[(y/2+4)*8+(x/2)+4];
 250             p2[x+8].v = b->v[(y/2+4)*8+(x/2)+4];
 251             p2[x+8].y = b->y4[y*8+x];
 252         }
 253         p1+=s->linex;
 254         p2+=s->linex;
 255     }
 256 }
 257
 258 static int compareregions(VIDEOSTREAM*s, int bx, int by)
 259 {
 260     int linex = s->width;
 261     YUV*p1 = &s->current[by*linex*16+bx*16];
 262     YUV*p2 = &s->oldpic[by*linex*16+bx*16];
 263     int diff = 0;
 264     int x,y;
 265     for(y=0;y<16;y++) {
 266         for(x=0;x<16;x++) {
 267             YUV*m = &p1[x];
 268             YUV*n = &p2[x];
 269             int y = m->y - n->y;
 270             int u = m->u - n->u;
 271             int v = m->v - n->v;
 272             diff += y*y+(u*u+v*v)/4;
 273         }
 274         p1+=linex;
 275         p2+=linex;
 276     }
 277     return diff/256;
 278 }
 279
 280 static inline int valtodc(int val)
 281 {
 282     assert(val>=0);
 283
 284     /* table 12/h.263 */
 285
 286     //val+=4; //round
 287     val/=8;
 288     /* TODO: what to do for zero values? skip the block? */
 289     if(val==0)
 290         return 1;
 291     if(val==128)
 292         return 255;
 293     if(val>254)
 294         return 254;
 295     return val;
 296 }
 297 static int dctoval(int dc)
 298 {
 299     int val;
 300     assert(dc>0);
 301     assert(dc!=128);
 302     assert(dc<256);
 303     /* table 12/h.263 */
 304     val = dc*8;
 305     if(val == 255*8)
 306         val = 128*8;
 307     return val;
 308 }
 309
 310 static int codehuffman(TAG*tag, struct huffcode*table, int index)
 311 {
 312     /* TODO: !optimize! */
 313     int i=0;
 314     while(table[index].code[i]) {
 315         if(table[index].code[i]=='0')
 316             swf_SetBits(tag, 0, 1);
 317         else
 318             swf_SetBits(tag, 1, 1);
 319         i++;
 320     }
 321     return i;
 322 }
 323
 324 static void quantize8x8(double*src, int*dest, int has_dc, int quant)
 325 {
 326     int t,pos=0;
 327     double q = 1.0/(quant*2);
 328     if(has_dc) {
 329         dest[0] = valtodc((int)src[0]); /*DC*/
 330         pos++;
 331     }
 332     for(t=pos;t<64;t++)
 333     {
 334         //dest[t] = (int)src[t];
 335     /* exact: if(quant&1){dest[t] = (dest[t]/quant - 1)/2;}else{dest[t] = ((dest[t]+1)/quant - 1)/2;} */
 336         //if(quant&1){dest[t] = (dest[t]/quant - 1)/2;}else{dest[t] = ((dest[t]+1)/quant - 1)/2;}
 337         //dest[t] = dest[t]/(quant*2);
 338         dest[t] = (int)(src[t]*q);
 339         /* TODO: warn if this happens- the video will be buggy */
 340         if(dest[t]>127) dest[t]=127;
 341         if(dest[t]<-127) dest[t]=-127;
 342     }
 343 }
 344
 345 static void dequantize8x8(int*b, int has_dc, int quant)
 346 {
 347     int t,pos=0;
 348     if(has_dc) {
 349         b[0] = dctoval(b[0]); //DC
 350         pos++;
 351     }
 352     for(t=pos;t<64;t++) {
 353         if(b[t]) {
 354             int sign = 0;
 355             if(b[t]<0) {
 356                 b[t] = -b[t];
 357                 sign = 1;
 358             }
 359
 360             if(quant&1) {
 361                 b[t] = quant*(2*b[t]+1); //-7,8,24,40
 362             } else {
 363                 b[t] = quant*(2*b[t]+1)-1; //-8,7,23,39
 364             }
 365
 366             if(sign)
 367                 b[t] = -b[t];
 368         }
 369
 370         /* paragraph 6.2.2, "clipping of reconstruction levels": */
 371         if(b[t]>2047) b[t]=2047;
 372         if(b[t]<-2048) b[t]=-2048;
 373     }
 374 }
 375
 376 static int hascoef(int*b, int has_dc)
 377 {
 378     int t;
 379     int pos=0;
 380     if(has_dc)
 381         pos++;
 382     for(t=pos;t<64;t++) {
 383         if(b[t])
 384             return 1;
 385     }
 386     return 0;
 387 }
 388
 389 static int coefbits8x8(int*bb, int has_dc)
 390 {
 391     int t;
 392     int pos=0;
 393     int bits=0;
 394     int last;
 395
 396     if(has_dc) {
 397         bits+=8;
 398         pos++;
 399     }
 400     for(last=63;last>=pos;last--) {
 401         if(bb[last])
 402             break;
 403     }
 404     if(last < pos)
 405         return bits;
 406     while(1) {
 407         int run=0, level=0, islast=0,t;
 408         while(!bb[pos] && pos<last) {
 409             pos++;
 410             run++;
 411         }
 412         if(pos==last)
 413             islast=1;
 414         level=bb[pos];
 415         if(level<0) level=-level;
 416         assert(level);
 417         for(t=0;t<RLE_ESCAPE;t++) {
 418             if(rle_params[t].run == run &&
 419                rle_params[t].level == level &&
 420                rle_params[t].last == islast) {
 421                 bits += rle[t].len + 1;
 422                 break;
 423             }
 424         }
 425         if(t==RLE_ESCAPE) {
 426             bits += rle[RLE_ESCAPE].len + 1 + 6 + 8;
 427         }
 428         if(islast)
 429             break;
 430         pos++;
 431     }
 432     return bits;
 433 }
 434
 435 static void encode8x8(TAG*tag, int*bb, int has_dc, int has_tcoef)
 436 {
 437     int t;
 438     int pos=0;
 439     int bits=0;
 440
 441     if(has_dc) {
 442         swf_SetBits(tag, bb[0], 8);
 443         pos++;
 444     }
 445
 446     if(has_tcoef) {
 447         int last;
 448         /* determine last non-null coefficient */
 449         for(last=63;last>=pos;last--) {
 450             /* TODO: we could leave out small coefficients
 451                      after a certain point (32?) */
 452             if(bb[last])
 453                 break;
 454         }
 455         /* blocks without coefficients should not be included
 456            in the cbpy/cbpc patterns: */
 457         assert(bb[last]);
 458
 459         while(1) {
 460             int run=0;
 461             int level=0;
 462             int islast=0;
 463             int sign=0;
 464             int t;
 465             while(!bb[pos] && pos<last) {
 466                 pos++;
 467                 run++;
 468             }
 469             if(pos==last)
 470                 islast=1;
 471             level=bb[pos];
 472             assert(level);
 473             if(level<0) {
 474                 level = -level;
 475                 sign = 1;
 476             }
 477             for(t=0;t<RLE_ESCAPE;t++) {
 478                 /* TODO: lookup table */
 479                 if(rle_params[t].run == run &&
 480                    rle_params[t].level == level &&
 481                    rle_params[t].last == islast) {
 482                     codehuffman(tag, rle, t);
 483                     swf_SetBits(tag, sign, 1);
 484                     break;
 485                 }
 486             }
 487             if(t==RLE_ESCAPE) {
 488                 codehuffman(tag, rle, RLE_ESCAPE);
 489                 level=bb[pos];
 490                 /* table 14/h.263 */
 491                 assert(level);
 492                 assert(level>=-127);
 493                 assert(level<=127);
 494
 495                 swf_SetBits(tag, islast, 1);
 496                 swf_SetBits(tag, run, 6);
 497                 swf_SetBits(tag, level, 8); //FIXME: fixme??
 498             }
 499
 500             if(islast)
 501                 break;
 502             pos++;
 503         }
 504     }
 505 }
 506
 507 static void dodct(fblock_t*fb)
 508 {
 509     int t;
 510     dct(fb->y1); dct(fb->y2); dct(fb->y3); dct(fb->y4);
 511     dct(fb->u);  dct(fb->v);
 512     fzigzag(fb->y1);
 513     fzigzag(fb->y2);
 514     fzigzag(fb->y3);
 515     fzigzag(fb->y4);
 516     fzigzag(fb->u);
 517     fzigzag(fb->v);
 518 }
 519
 520 static void doidct(block_t*b)
 521 {
 522     fblock_t fb;
 523     int t;
 524     for(t=0;t<64;t++) {
 525         fb.y1[t] = b->y1[zigzagtable[t]];
 526         fb.y2[t] = b->y2[zigzagtable[t]];
 527         fb.y3[t] = b->y3[zigzagtable[t]];
 528         fb.y4[t] = b->y4[zigzagtable[t]];
 529         fb.u[t] = b->u[zigzagtable[t]];
 530         fb.v[t] = b->v[zigzagtable[t]];
 531     }
 532     idct(fb.y1); idct(fb.y2); idct(fb.y3); idct(fb.y4);
 533     idct(fb.u);  idct(fb.v);
 534     for(t=0;t<64;t++) {
 535         b->y1[t] = fb.y1[t];
 536         b->y2[t] = fb.y2[t];
 537         b->y3[t] = fb.y3[t];
 538         b->y4[t] = fb.y4[t];
 539         b->u[t] = fb.u[t];
 540         b->v[t] = fb.v[t];
 541     }
 542 }
 543 static void truncateblock(block_t*b)
 544 {
 545     int t;
 546     for(t=0;t<64;t++) {
 547         b->y1[t] = truncate256(b->y1[t]);
 548         b->y2[t] = truncate256(b->y2[t]);
 549         b->y3[t] = truncate256(b->y3[t]);
 550         b->y4[t] = truncate256(b->y4[t]);
 551         b->u[t] = truncate256(b->u[t]);
 552         b->v[t] = truncate256(b->v[t]);
 553     }
 554 }
 555
 556 static void quantize(fblock_t*fb, block_t*b, int has_dc, int quant)
 557 {
 558     quantize8x8(fb->y1, b->y1, has_dc, quant);
 559     quantize8x8(fb->y2, b->y2, has_dc, quant);
 560     quantize8x8(fb->y3, b->y3, has_dc, quant);
 561     quantize8x8(fb->y4, b->y4, has_dc, quant);
 562     quantize8x8(fb->u, b->u, has_dc, quant);
 563     quantize8x8(fb->v, b->v, has_dc, quant);
 564 }
 565 static void dequantize(block_t*b, int has_dc, int quant)
 566 {
 567     dequantize8x8(b->y1, has_dc, quant);
 568     dequantize8x8(b->y2, has_dc, quant);
 569     dequantize8x8(b->y3, has_dc, quant);
 570     dequantize8x8(b->y4, has_dc, quant);
 571     dequantize8x8(b->u, has_dc, quant);
 572     dequantize8x8(b->v, has_dc, quant);
 573 }
 574
 575 static void getblockpatterns(block_t*b, int*cbpybits,int*cbpcbits, int has_dc)
 576 {
 577     *cbpybits = 0;
 578     *cbpcbits = 0;
 579
 580     *cbpybits|=hascoef(b->y1, has_dc)*8;
 581     *cbpybits|=hascoef(b->y2, has_dc)*4;
 582     *cbpybits|=hascoef(b->y3, has_dc)*2;
 583     *cbpybits|=hascoef(b->y4, has_dc)*1;
 584
 585     *cbpcbits|=hascoef(b->u, has_dc)*2;
 586     *cbpcbits|=hascoef(b->v, has_dc)*1;
 587 }
 588
 589 static void setQuant(TAG*tag, int dquant)
 590 {
 591     int code = 0;
 592     /* 00 01 10 11
 593        -1 -2 +1 +2
 594     */
 595     if(dquant == -1) {
 596         swf_SetBits(tag, 0x0, 2);
 597     } else if(dquant == -2) {
 598         swf_SetBits(tag, 0x1, 2);
 599     } else if(dquant == +1) {
 600         swf_SetBits(tag, 0x2, 2);
 601     } else if(dquant == +2) {
 602         swf_SetBits(tag, 0x3, 2);
 603     } else {
 604         assert(0*strlen("invalid dquant"));
 605     }
 606 }
 607
 608 static void change_quant(int quant, int*dquant)
 609 {
 610     /* TODO */
 611     *dquant = 0;
 612 }
 613
 614 static void encode_blockI(TAG*tag, VIDEOSTREAM*s, int bx, int by, int*quant)
 615 {
 616     fblock_t fb;
 617     block_t b;
 618     int dquant=0;
 619     int cbpcbits = 0, cbpybits=0;
 620
 621     getregion(&fb, s->current, bx*16, by*16, s->width);
 622     dodct(&fb);
 623
 624     change_quant(*quant, &dquant);
 625     *quant+=dquant;
 626     quantize(&fb, &b, 1, *quant);
 627
 628     //decode_blockI(s, &b, bx, by);
 629
 630     getblockpatterns(&b, &cbpybits, &cbpcbits, 1);
 631
 632     if(dquant) {
 633         codehuffman(tag, mcbpc_intra, 4+cbpcbits);
 634     } else {
 635         codehuffman(tag, mcbpc_intra, 0+cbpcbits);
 636     }
 637
 638     codehuffman(tag, cbpy, cbpybits);
 639
 640     if(dquant) {
 641         setQuant(tag, dquant);
 642     }
 643
 644     /* luminance */
 645     encode8x8(tag, b.y1, 1, cbpybits&8);
 646     encode8x8(tag, b.y2, 1, cbpybits&4);
 647     encode8x8(tag, b.y3, 1, cbpybits&2);
 648     encode8x8(tag, b.y4, 1, cbpybits&1);
 649
 650     /* chrominance */
 651     encode8x8(tag, b.u, 1, cbpcbits&2);
 652     encode8x8(tag, b.v, 1, cbpcbits&1);
 653
 654     /* reconstruct */
 655     dequantize(&b, 1, *quant);
 656     doidct(&b);
 657     truncateblock(&b);
 658     copyblock(s, s->current, &b, bx, by);
 659 }
 660
 661 static void yuvdiff(fblock_t*a, fblock_t*b)
 662 {
 663     int t;
 664     for(t=0;t<64;t++) {
 665         a->y1[t] = (a->y1[t] - b->y1[t]);
 666         a->y2[t] = (a->y2[t] - b->y2[t]);
 667         a->y3[t] = (a->y3[t] - b->y3[t]);
 668         a->y4[t] = (a->y4[t] - b->y4[t]);
 669         a->u[t]  = (a->u[t] - b->u[t]);
 670         a->v[t]  = (a->v[t] - b->v[t]);
 671     }
 672 }
 673
 674 static void predictmvd(VIDEOSTREAM*s, int bx, int by, int*px, int*py)
 675 {
 676     int i1,i2;
 677     int x1,y1,x2,y2,x3,y3;
 678     int x4,y4,p;
 679     if(bx) {x1=s->mvdx[by*s->bbx+bx-1];
 680             y1=s->mvdy[by*s->bbx+bx-1];
 681     } else {x1=y1=0;}
 682
 683     if(by) {x2=s->mvdx[(by-1)*s->bbx+bx];
 684             y2=s->mvdy[(by-1)*s->bbx+bx];
 685             if(bx<s->bbx-1) {
 686                 x3=s->mvdx[(by-1)*s->bbx+bx+1];
 687                 y3=s->mvdy[(by-1)*s->bbx+bx+1];
 688             } else {
 689                 x3=y3=0;
 690             }
 691            }
 692     else   {x2=x3=x1;y2=y3=y1;}
 693
 694            if((x1 <= x2 && x2 <= x3) ||
 695               (x3 <= x2 && x2 <= x1)) {
 696         x4=x2;
 697     } else if((x2 <= x1 && x1 <= x3) ||
 698               (x3 <= x1 && x1 <= x2)) {
 699         x4=x1;
 700     } else if((x1 <= x3 && x3 <= x2) ||
 701               (x2 <= x3 && x3 <= x1)) {
 702         x4=x3;
 703     }
 704
 705            if((y1 <= y2 && y2 <= y3) ||
 706               (y3 <= y2 && y2 <= y1)) {
 707         y4=y2;
 708     } else if((y2 <= y1 && y1 <= y3) ||
 709               (y3 <= y1 && y1 <= y2)) {
 710         y4=y1;
 711     } else if((y1 <= y3 && y3 <= y2) ||
 712               (y2 <= y3 && y3 <= y1)) {
 713         y4=y3;
 714     }
 715
 716     *px = x4;
 717     *py = y4;
 718     assert((x4>=-32 && x4<=31) && (y4>=-32 && y4<=31));
 719 }
 720
 721 static inline int mvd2index(int px, int py, int x, int y, int xy)
 722 {
 723     assert((x>=-32 && x<=31) && (y>=-32 && y<=31));
 724     assert((x&1)==0 && (y&1)==0);//for now
 725     assert((x&2)==0 && (y&2)==0);//for now(2)
 726
 727     x-=px;
 728     y-=py;
 729
 730     if(xy)
 731         x=y;
 732     x+=32;
 733
 734     /* (x&63) */
 735     if(x>63)
 736         x-=64;
 737     if(x<0)
 738         x+=64;
 739
 740     assert(x>=0 && x<64);
 741     return x;
 742 }
 743
 744 static int encode_blockP(TAG*tag, VIDEOSTREAM*s, int bx, int by, int*quant)
 745 {
 746     fblock_t fb;
 747     block_t b;
 748     int dquant=0;
 749     int has_mvd=0;
 750     int has_mvd24=0;
 751     int has_dc=1;
 752     int mode = 0;
 753     int cbpcbits = 0, cbpybits=0;
 754     int diff;
 755     int predictmvdx;
 756     int predictmvdy;
 757
 758     block_t b_i;
 759     int bits_i;
 760
 761     fblock_t fbold_v00;
 762     block_t b_v00;
 763     int bits_v00 = 65535;
 764     int x_v00=0;
 765     int y_v00=0;
 766
 767     diff = compareregions(s, bx, by);
 768     if(diff < 20 /*TODO: should be a parameter- good values are between 32 and 48 */) {
 769         swf_SetBits(tag, 1,1); /* cod=1, block skipped */
 770         /* copy the region from the last frame so that we have a complete reconstruction */
 771         copyregion(s, s->current, s->oldpic, bx, by);
 772         return 1;
 773     }
 774
 775     predictmvd(s,bx,by,&predictmvdx,&predictmvdy);
 776     getregion(&fb, s->current, bx*16, by*16, s->width);
 777
 778     { /* consider I-block */
 779         fblock_t fb_i;
 780         int y,c;
 781         memcpy(&fb_i, &fb, sizeof(fblock_t));
 782         dodct(&fb_i);
 783         quantize(&fb_i, &b_i, 1, *quant);
 784         getblockpatterns(&b_i, &y, &c, 1);
 785         bits_i = 1; //cod
 786         bits_i += mcbpc_inter[3*4+c].len;
 787         bits_i += cbpy[y].len;
 788         bits_i += coefbits8x8(b_i.y1, 1);
 789         bits_i += coefbits8x8(b_i.y2, 1);
 790         bits_i += coefbits8x8(b_i.y3, 1);
 791         bits_i += coefbits8x8(b_i.y4, 1);
 792         bits_i += coefbits8x8(b_i.u, 1);
 793         bits_i += coefbits8x8(b_i.v, 1);
 794     }
 795
 796     if(bx&&by&&(bx<s->bbx-1)&&(by<s->bby-1))
 797     { /* consider mvd(x,y)-block */
 798         fblock_t fbdiff;
 799         int y,c;
 800
 801         x_v00=0;
 802         y_v00=0;
 803
 804         if(s->do_motion) {
 805             int hx,hy;
 806             int bestx=0,besty=0,bestbits=65536;
 807
 808             for(hx=-8;hx<8;hx+=4)
 809             for(hy=-8;hy<8;hy+=4)
 810             {
 811                 block_t b;
 812                 fblock_t fbold;
 813                 int bits = 0;
 814                 memcpy(&fbdiff, &fb, sizeof(fblock_t));
 815                 getregion(&fbold, s->oldpic, bx*16+hx/2, by*16+hy/2, s->linex);
 816                 yuvdiff(&fbdiff, &fbold);
 817                 dodct(&fbdiff);
 818                 quantize(&fbdiff, &b, 0, *quant);
 819                 bits += coefbits8x8(b.y1, 0);
 820                 bits += coefbits8x8(b.y2, 0);
 821                 bits += coefbits8x8(b.y3, 0);
 822                 bits += coefbits8x8(b.y4, 0);
 823                 bits += coefbits8x8(b.u, 0);
 824                 bits += coefbits8x8(b.v, 0);
 825                 if(bits<bestbits) {
 826                     bestbits = bits;
 827                     bestx = hx;
 828                     besty = hy;
 829                 }
 830             }
 831             x_v00 = bestx;
 832             y_v00 = besty;
 833         }
 834
 835         memcpy(&fbdiff, &fb, sizeof(fblock_t));
 836         getregion(&fbold_v00, s->oldpic, bx*16+x_v00/2, by*16+y_v00/2, s->linex);
 837         yuvdiff(&fbdiff, &fbold_v00);
 838         dodct(&fbdiff);
 839         quantize(&fbdiff, &b_v00, 0, *quant);
 840         getblockpatterns(&b_v00, &y, &c, 0);
 841
 842         bits_v00 = 1; //cod
 843         bits_v00 += mcbpc_inter[0*4+c].len;
 844         bits_v00 += cbpy[y^15].len;
 845         bits_v00 += mvd[mvd2index(predictmvdx, predictmvdy, x_v00, y_v00, 0)].len; // (0,0)
 846         bits_v00 += mvd[mvd2index(predictmvdx, predictmvdy, x_v00, y_v00, 1)].len;
 847         bits_v00 += coefbits8x8(b_v00.y1, 0);
 848         bits_v00 += coefbits8x8(b_v00.y2, 0);
 849         bits_v00 += coefbits8x8(b_v00.y3, 0);
 850         bits_v00 += coefbits8x8(b_v00.y4, 0);
 851         bits_v00 += coefbits8x8(b_v00.u, 0);
 852         bits_v00 += coefbits8x8(b_v00.v, 0);
 853     }
 854
 855     if(bits_i > bits_v00)
 856     {
 857         /* mvd (0,0) block (mode=0) */
 858         int t;
 859         mode = 0; // mvd w/o mvd24
 860         has_dc = 0;
 861         memcpy(&b, &b_v00, sizeof(block_t));
 862
 863         getblockpatterns(&b, &cbpybits, &cbpcbits, has_dc);
 864         swf_SetBits(tag,0,1); // COD
 865         codehuffman(tag, mcbpc_inter, mode*4+cbpcbits);
 866         codehuffman(tag, cbpy, cbpybits^15);
 867
 868         /* vector */
 869         codehuffman(tag, mvd, mvd2index(predictmvdx, predictmvdy, x_v00, y_v00, 0));
 870         codehuffman(tag, mvd, mvd2index(predictmvdx, predictmvdy, x_v00, y_v00, 1));
 871         s->mvdx[by*s->bbx+bx] = x_v00;
 872         s->mvdy[by*s->bbx+bx] = y_v00;
 873
 874         /* luminance */
 875         encode8x8(tag, b.y1, has_dc, cbpybits&8);
 876         encode8x8(tag, b.y2, has_dc, cbpybits&4);
 877         encode8x8(tag, b.y3, has_dc, cbpybits&2);
 878         encode8x8(tag, b.y4, has_dc, cbpybits&1);
 879
 880         /* chrominance */
 881         encode8x8(tag, b.u, has_dc, cbpcbits&2);
 882         encode8x8(tag, b.v, has_dc, cbpcbits&1);
 883
 884         /* -- reconstruction -- */
 885         dequantize(&b, 0, *quant);
 886         doidct(&b);
 887         for(t=0;t<64;t++) {
 888             b.y1[t] = truncate256(b.y1[t] + (int)fbold_v00.y1[t]);
 889             b.y2[t] = truncate256(b.y2[t] + (int)fbold_v00.y2[t]);
 890             b.y3[t] = truncate256(b.y3[t] + (int)fbold_v00.y3[t]);
 891             b.y4[t] = truncate256(b.y4[t] + (int)fbold_v00.y4[t]);
 892             b.u[t] = truncate256(b.u[t] + (int)fbold_v00.u[t]);
 893             b.v[t] = truncate256(b.v[t] + (int)fbold_v00.v[t]);
 894         }
 895         copyblock(s, s->current, &b, bx, by);
 896         return bits_v00;
 897     } else {
 898         /* i block (mode=3) */
 899         mode = 3;
 900         has_dc = 1;
 901         memcpy(&b, &b_i, sizeof(block_t));
 902         getblockpatterns(&b, &cbpybits, &cbpcbits, has_dc);
 903         swf_SetBits(tag,0,1); // COD
 904         codehuffman(tag, mcbpc_inter, mode*4+cbpcbits);
 905         codehuffman(tag, cbpy, cbpybits);
 906
 907         /* luminance */
 908         encode8x8(tag, b.y1, has_dc, cbpybits&8);
 909         encode8x8(tag, b.y2, has_dc, cbpybits&4);
 910         encode8x8(tag, b.y3, has_dc, cbpybits&2);
 911         encode8x8(tag, b.y4, has_dc, cbpybits&1);
 912
 913         /* chrominance */
 914         encode8x8(tag, b.u, has_dc, cbpcbits&2);
 915         encode8x8(tag, b.v, has_dc, cbpcbits&1);
 916
 917         /* -- reconstruction -- */
 918         dequantize(&b, 1, *quant);
 919         doidct(&b);
 920         truncateblock(&b);
 921         copyblock(s, s->current, &b, bx, by);
 922         return bits_i;
 923     }
 924
 925     exit(1);
 926 #if 0
 927     dodct(&fb);
 928     quantize(&fb, &b, has_dc, *quant);
 929     getblockpatterns(&b, &cbpybits, &cbpcbits, has_dc);
 930
 931     if(!dquant && has_mvd && !has_mvd24 && !has_dc) mode = 0;
 932     else if(dquant && has_mvd && !has_mvd24 && !has_dc) mode = 1;
 933     else if(!dquant && has_mvd && has_mvd24 && !has_dc) mode = 2;
 934     else if(!dquant && !has_mvd && !has_mvd24 && has_dc) mode = 3;
 935     else if(dquant && !has_mvd && !has_mvd24 && has_dc) mode = 4;
 936     else exit(1);
 937
 938     swf_SetBits(tag,0,1); /* cod - 1 if we're not going to code this block*/
 939
 940     codehuffman(tag, mcbpc_inter, mode*4+cbpcbits);
 941     codehuffman(tag, cbpy, (mode==3 || mode==4)?cbpybits:cbpybits^15);
 942
 943     if(dquant) {
 944         setQuant(tag, dquant);
 945     }
 946
 947     if(has_mvd) {
 948         /* 0,0 */
 949         codehuffman(tag, mvd, 32);
 950         codehuffman(tag, mvd, 32);
 951     }
 952     if(has_mvd24) {
 953     }
 954
 955     /* luminance */
 956     encode8x8(tag, b.y1, has_dc, cbpybits&8);
 957     encode8x8(tag, b.y2, has_dc, cbpybits&4);
 958     encode8x8(tag, b.y3, has_dc, cbpybits&2);
 959     encode8x8(tag, b.y4, has_dc, cbpybits&1);
 960
 961     /* chrominance */
 962     encode8x8(tag, b.u, has_dc, cbpcbits&2);
 963     encode8x8(tag, b.v, has_dc, cbpcbits&1);
 964 #endif
 965 }
 966
 967 #define TYPE_IFRAME 0
 968 #define TYPE_PFRAME 1
 969
 970 static void writeHeader(TAG*tag, int width, int height, int frame, int quant, int type)
 971 {
 972     U32 i32;
 973     swf_SetU16(tag, frame);
 974     swf_SetBits(tag, 1, 17); /* picture start code*/
 975     swf_SetBits(tag, 0, 5); /* version=0, version 1 would optimize rle behaviour*/
 976     swf_SetBits(tag, frame, 8); /* time reference */
 977
 978     /* write dimensions, taking advantage of some predefined sizes
 979        if the opportunity presents itself */
 980     i32 = width<<16|height;
 981     switch(i32)
 982     {
 983         case 352<<16|288: swf_SetBits(tag, 2, 3);break;
 984         case 176<<16|144: swf_SetBits(tag, 3, 3);break;
 985         case 128<<16|96: swf_SetBits(tag, 4, 3);break;
 986         case 320<<16|240: swf_SetBits(tag, 5, 3);break;
 987         case 160<<16|120: swf_SetBits(tag, 6, 3);break;
 988         default:
 989             if(width>255 || height>255) {
 990                 swf_SetBits(tag, 1, 3);
 991                 swf_SetBits(tag, width, 16);
 992                 swf_SetBits(tag, height, 16);
 993             } else {
 994                 swf_SetBits(tag, 0, 3);
 995                 swf_SetBits(tag, width, 8);
 996                 swf_SetBits(tag, height, 8);
 997             }
 998     }
 999
1000     swf_SetBits(tag, type, 2); /* I-Frame or P-Frame */
1001     swf_SetBits(tag, 0, 1); /* No deblock filter */
1002     assert(quant>0);
1003     swf_SetBits(tag, quant, 5); /* quantizer (1-31), may be updated later on*/
1004     swf_SetBits(tag, 0, 1); /* No extra info */
1005 }
1006
1007 int stat_qdiff(double*b1, double*b2)
1008 {
1009     int x;
1010     double diff=0;
1011     for(x=0;x<64;x++) {
1012         double y1 = b1[x] - b2[x];
1013         diff += y1*y1;
1014     }
1015     return (int)(diff/64);
1016 }
1017
1018 int stat_absdiff(double*b1, double*b2)
1019 {
1020     int x;
1021     double diff=0;
1022     for(x=0;x<64;x++) {
1023         double y1 = b1[x] - b2[x];
1024         diff += fabs(y1);
1025     }
1026     return (int)(diff/64);
1027 }
1028
1029 int stat_absfreq(double*b1, double*b2)
1030 {
1031     int x;
1032     double diff=0;
1033     double d1[64],d2[64];
1034     memcpy(&d1, b1, 64*sizeof(double));
1035     dct(d1);
1036     memcpy(&d2, b2, 64*sizeof(double));
1037     dct(d2);
1038     for(x=0;x<64;x++) {
1039         double y1 = d1[x] - d2[x];
1040         diff += fabs(y1);
1041     }
1042     return (int)(diff/64);
1043 }
1044
1045 int stat_qfreq(double*b1, double*b2)
1046 {
1047     int x;
1048     double diff=0;
1049     double d1[64],d2[64];
1050     memcpy(&d1, b1, 64*sizeof(double));
1051     dct(d1);
1052     memcpy(&d2, b2, 64*sizeof(double));
1053     dct(d2);
1054     for(x=0;x<64;x++) {
1055         double y1 = d1[x] - d2[x];
1056         diff += y1*y1;
1057     }
1058     return (int)(diff/64);
1059 }
1060
1061 int stat_nonnull(double*b1, double*b2)
1062 {
1063     int x;
1064     int diff=0;
1065     double d1[64],d2[64];
1066     memcpy(&d1, b1, 64*sizeof(double));
1067     dct(d1);
1068     memcpy(&d2, b2, 64*sizeof(double));
1069     dct(d2);
1070     for(x=0;x<64;x++) {
1071         int y1 = (int)((d1[x] - d2[x])/9);
1072         if(y1)
1073             diff++;
1074     }
1075     return diff;
1076 }
1077
1078 void stat_filter(FILE*fi, double*d1, double*d2)
1079 {
1080     int x,y,xx,yy,b;
1081     for(b=3;b>=0;b--) {
1082         int d = 1<<b;
1083         double diff=0;
1084         for(x=0;x<8;x+=d)
1085         for(y=0;y<8;y+=d)
1086         {
1087             double add1=0,add2=0;
1088             for(xx=x;xx<x+d;xx++)
1089             for(yy=y;yy<y+d;yy++)
1090             {
1091                 add1 += d1[yy*8+xx];
1092                 add2 += d2[yy*8+xx];
1093             }
1094             diff += fabs(add1-add2);
1095         }
1096         fprintf(fi, "\t%d",(int)(diff/64));
1097     }
1098 }
1099
1100 void qstat_filter(FILE*fi, double*d1, double*d2)
1101 {
1102     int x,y,xx,yy,b;
1103     for(b=3;b>=0;b--) {
1104         int d = 1<<b;
1105         double diff=0;
1106         for(x=0;x<8;x+=d)
1107         for(y=0;y<8;y+=d)
1108         {
1109             double add1=0,add2=0;
1110             for(xx=x;xx<x+d;xx++)
1111             for(yy=y;yy<y+d;yy++)
1112             {
1113                 add1 += d1[yy*8+xx];
1114                 add2 += d2[yy*8+xx];
1115             }
1116             diff += (add1-add2)*(add1-add2);
1117         }
1118         fprintf(fi, "\t%d",(int)(diff/64));
1119     }
1120 }
1121
1122 void qqstat_filter(FILE*fi, double*d1, double*d2)
1123 {
1124     int x,y,xx,yy,b;
1125     for(b=3;b>=0;b--) {
1126         int d = 1<<b;
1127         double diff=0;
1128         for(x=0;x<8;x+=d)
1129         for(y=0;y<8;y+=d)
1130         {
1131             double add1=0,add2=0;
1132             for(xx=x;xx<x+d;xx++)
1133             for(yy=y;yy<y+d;yy++)
1134             {
1135                 add1 += d1[yy*8+xx]*d1[yy*8+xx];
1136                 add2 += d2[yy*8+xx]*d2[yy*8+xx];
1137             }
1138             diff += fabs(add1-add2);
1139         }
1140         fprintf(fi, "\t%d",(int)(diff/64));
1141     }
1142 }
1143
1144 void stat(FILE*fi, int*vals, double*yold, double*ynew)
1145 {
1146     int t;
1147     int bits = coefbits8x8(vals, 0);
1148     fprintf(fi, "%d\t%d\t%d\t%d\t%d\t%d", bits,
1149              stat_nonnull(ynew, yold),
1150              stat_qdiff(ynew,yold),
1151              stat_absdiff(ynew,yold),
1152              stat_absfreq(ynew,yold),
1153              stat_qfreq(ynew,yold));
1154     stat_filter(fi, ynew, yold);
1155     qqstat_filter(fi, ynew, yold);
1156     fprintf(fi, "\n");
1157 }
1158
1159 void dostat(VIDEOSTREAM*s)
1160 {
1161     int bx,by,bx2,by2;
1162     int quant = 9;
1163     int num = 0;
1164     FILE*fi = fopen("mvd.dat", "wb");
1165     fprintf(fi, "bits\tnonnull\tqdiff\tabsdiff\tabsfreq\tqfreq\tf1\tf2\tf4\tf8\tqf1\tqf2\tqf4\tqf8\n");
1166     for(by=0;by<s->bby;by++)
1167     for(bx=0;bx<s->bbx;bx++)
1168     {
1169         for(by2=0;by2<s->bby;by2++)
1170         for(bx2=0;bx2<bx;bx2++)
1171         {
1172             fblock_t fbnew,fbdiff,fbold;
1173             block_t b;
1174             int t, y,c,bits;
1175             getregion(&fbnew, s->current, bx*16, by*16, s->linex);
1176             memcpy(&fbdiff, &fbnew, sizeof(fblock_t));
1177             getregion(&fbold, s->current, bx2*16, by2*16, s->linex);
1178             yuvdiff(&fbdiff, &fbold);
1179             dodct(&fbdiff);
1180             quantize(&fbdiff, &b, 0, quant);
1181
1182             stat(fi, b.y1, fbnew.y1, fbold.y1);
1183             stat(fi, b.y2, fbnew.y2, fbold.y2);
1184             stat(fi, b.y3, fbnew.y3, fbold.y3);
1185             stat(fi, b.y4, fbnew.y4, fbold.y4);
1186             stat(fi, b.u, fbnew.u, fbold.u);
1187             stat(fi, b.v, fbnew.v, fbold.v);
1188
1189             num++;
1190             if(num==1000) {
1191                 fclose(fi);
1192                 exit(7);
1193             }
1194         }
1195         printf("%d\n", num);fflush(stdout);
1196     }
1197     fclose(fi);
1198     exit(7);
1199 }
1200
1201 void swf_SetVideoStreamIFrame(TAG*tag, VIDEOSTREAM*s, RGBA*pic, int quant)
1202 {
1203     int bx, by;
1204
1205     if(quant<1) quant=1;
1206     if(quant>31) quant=31;
1207
1208     writeHeader(tag, s->width, s->height, s->frame, quant, TYPE_IFRAME);
1209
1210     rgb2yuv(s->current, pic, s->linex, s->olinex, s->width, s->height);
1211
1212     //dostat(s);
1213
1214     for(by=0;by<s->bby;by++)
1215     {
1216         for(bx=0;bx<s->bbx;bx++)
1217         {
1218             encode_blockI(tag, s, bx, by, &quant);
1219         }
1220     }
1221     s->frame++;
1222     memcpy(s->oldpic, s->current, s->width*s->height*sizeof(YUV));
1223 }
1224
1225 void swf_SetVideoStreamPFrame(TAG*tag, VIDEOSTREAM*s, RGBA*pic, int quant)
1226 {
1227     int bx, by;
1228
1229     if(quant<1) quant=1;
1230     if(quant>31) quant=31;
1231
1232     writeHeader(tag, s->width, s->height, s->frame, quant, TYPE_PFRAME);
1233
1234     rgb2yuv(s->current, pic, s->linex, s->olinex, s->width, s->height);
1235     memset(s->mvdx, 0, s->bbx*s->bby*sizeof(int));
1236     memset(s->mvdy, 0, s->bbx*s->bby*sizeof(int));
1237
1238     for(by=0;by<s->bby;by++)
1239     {
1240         for(bx=0;bx<s->bbx;bx++)
1241         {
1242             encode_blockP(tag, s, bx, by, &quant);
1243         }
1244     }
1245     s->frame++;
1246     memcpy(s->oldpic, s->current, s->width*s->height*sizeof(YUV));
1247 #ifdef MAIN
1248     {
1249         int t;
1250         FILE*fi = fopen("test.ppm", "wb");
1251         yuv2rgb(pic, s->current, s->linex, s->width, s->height);
1252         fprintf(fi, "P6\n%d %d\n255\n", s->width, s->height);
1253         for(t=0;t<s->width*s->height;t++)
1254         {
1255             fwrite(&pic[t].r, 1, 1, fi);
1256             fwrite(&pic[t].g, 1, 1, fi);
1257             fwrite(&pic[t].b, 1, 1, fi);
1258         }
1259         fclose(fi);
1260     }
1261 #endif
1262 }
1263
1264 #ifdef MAIN
1265 #include "png.h"
1266 int main(int argn, char*argv[])
1267 {
1268     int fi;
1269     int t;
1270     SWF swf;
1271     TAG * tag;
1272     RGBA* pic, *pic2, rgb;
1273     SWFPLACEOBJECT obj;
1274     int width = 0;
1275     int height = 0;
1276     int frames = 50;
1277     int framerate = 29;
1278     unsigned char*data;
1279     char* fname = "/home/kramm/pics/peppers.png";
1280     VIDEOSTREAM stream;
1281     double d = 1.0;
1282
1283     memset(&stream, 0, sizeof(stream));
1284
1285     getPNG(fname, &width, &height, &data);
1286     pic = (RGBA*)malloc(width*height*sizeof(RGBA));
1287     pic2 = (RGBA*)malloc(width*height*sizeof(RGBA));
1288     memcpy(pic, data, width*height*sizeof(RGBA));
1289     free(data);
1290
1291     printf("Compressing %s, size %dx%d\n", fname, width, height);
1292
1293     memset(&swf,0,sizeof(SWF));
1294     memset(&obj,0,sizeof(obj));
1295
1296     swf.fileVersion    = 6;
1297     swf.frameRate      = framerate*256;
1298     swf.movieSize.xmax = 20*width;
1299     swf.movieSize.ymax = 20*height;
1300
1301     swf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
1302     tag = swf.firstTag;
1303     rgb.r = 0x00;rgb.g = 0x00;rgb.b = 0x00;
1304     swf_SetRGB(tag,&rgb);
1305
1306     tag = swf_InsertTag(tag, ST_DEFINEVIDEOSTREAM);
1307     swf_SetU16(tag, 33);
1308     swf_SetVideoStreamDefine(tag, &stream, frames, width, height);
1309     stream.do_motion = 1;
1310
1311     for(t=0;t<frames;t++)
1312     {
1313         int x,y;
1314         double xx,yy;
1315         for(y=0,yy=0;y<height;y++,yy+=d)  {
1316             RGBA*line = &pic[((int)yy)*width];
1317             for(x=0,xx=0;x<width;x++,xx+=d) {
1318                 pic2[y*width+x] = line[((int)xx)];
1319             }
1320         }
1321         printf("frame:%d\n", t);fflush(stdout);
1322
1323         tag = swf_InsertTag(tag, ST_VIDEOFRAME);
1324         swf_SetU16(tag, 33);
1325         if(t==0)
1326             swf_SetVideoStreamIFrame(tag, &stream, pic2, 9);
1327         else
1328             swf_SetVideoStreamPFrame(tag, &stream, pic2, 9);
1329
1330         tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
1331         swf_GetPlaceObject(0, &obj);
1332         if(t==0) {
1333             obj.depth = 1;
1334             obj.id = 33;
1335         } else {
1336             obj.move = 1;
1337             obj.depth = 1;
1338             obj.ratio = t;
1339         }
1340         swf_SetPlaceObject(tag,&obj);
1341
1342         tag = swf_InsertTag(tag, ST_SHOWFRAME);
1343         d-=0.005;
1344     }
1345
1346     swf_VideoStreamClear(&stream);
1347
1348     tag = swf_InsertTag(tag, ST_END);
1349
1350     fi = open("video3.swf", O_WRONLY|O_CREAT|O_TRUNC, 0644);
1351     if(swf_WriteSWC(fi,&swf)<0) {
1352         fprintf(stderr,"WriteSWF() failed.\n");
1353     }
1354     close(fi);
1355     swf_FreeTags(&swf);
1356 }
1357 #undef MAIN
1358 #endif