TOOLS/vivodump.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <inttypes.h>
   5
   6 #include "loader/wine/mmreg.h"
   7 #include "loader/wine/avifmt.h"
   8 #include "loader/wine/vfw.h"
   9
  10 #include "libmpdemux/muxer.h"
  11
  12 char *info_name;
  13 char *info_artist;
  14 char *info_genre;
  15 char *info_subject;
  16 char *info_copyright;
  17 char *info_sourceform;
  18 char *info_comment;
  19
  20 static const short h263_format[8][2] = {
  21     { 0, 0 },
  22     { 128, 96 },
  23     { 176, 144 },
  24     { 352, 288 },
  25     { 704, 576 },
  26     { 1408, 1152 },
  27     { 320, 240 }
  28 };
  29
  30 unsigned char* buffer;
  31 int bufptr=0;
  32 int bitcnt=0;
  33 unsigned char buf=0;
  34
  35 unsigned int x_get_bits(int n){
  36     unsigned int x=0;
  37     while(n-->0){
  38         if(!bitcnt){
  39             // fill buff
  40             buf=buffer[bufptr++];
  41             bitcnt=8;
  42         }
  43         //x=(x<<1)|(buf&1);buf>>=1;
  44         x=(x<<1)|(buf>>7);buf<<=1;
  45         --bitcnt;
  46     }
  47     return x;
  48 }
  49
  50 #define get_bits(xxx,n) x_get_bits(n)
  51 #define get_bits1(xxx) x_get_bits(1)
  52 #define skip_bits(xxx,n) x_get_bits(n)
  53 #define skip_bits1(xxx) x_get_bits(1)
  54
  55 int format;
  56 int width=320;
  57 int height=240;
  58
  59 /* most is hardcoded. should extend to handle all h263 streams */
  60 int h263_decode_picture_header(unsigned char *b_ptr)
  61 {
  62     int i;
  63
  64     for(i=0;i<16;i++) printf(" %02X",b_ptr[i]); printf("\n");
  65
  66     buffer=b_ptr;
  67     bufptr=bitcnt=buf=0;
  68
  69     /* picture header */
  70     if (get_bits(&s->gb, 22) != 0x20){
  71         printf("bad picture header\n");
  72         return -1;
  73     }
  74     skip_bits(&s->gb, 8); /* picture timestamp */
  75
  76     if (get_bits1(&s->gb) != 1){
  77         printf("bad marker\n");
  78         return -1;      /* marker */
  79     }
  80     if (get_bits1(&s->gb) != 0){
  81         printf("bad h263 id\n");
  82         return -1;      /* h263 id */
  83     }
  84     skip_bits1(&s->gb); /* split screen off */
  85     skip_bits1(&s->gb); /* camera  off */
  86     skip_bits1(&s->gb); /* freeze picture release off */
  87
  88     format = get_bits(&s->gb, 3);
  89
  90     if (format != 7) {
  91         printf("h263_plus = 0  format = %d\n",format);
  92         /* H.263v1 */
  93         width = h263_format[format][0];
  94         height = h263_format[format][1];
  95         printf("%d x %d\n",width,height);
  96 //        if (!width) return -1;
  97
  98         printf("pict_type=%d\n",get_bits1(&s->gb));
  99         printf("unrestricted_mv=%d\n",get_bits1(&s->gb));
 100 #if 1
 101         printf("SAC: %d\n",get_bits1(&s->gb));
 102         printf("advanced prediction mode: %d\n",get_bits1(&s->gb));
 103         printf("PB frame: %d\n",get_bits1(&s->gb));
 104 #else
 105         if (get_bits1(&s->gb) != 0)
 106             return -1;  /* SAC: off */
 107         if (get_bits1(&s->gb) != 0)
 108             return -1;  /* advanced prediction mode: off */
 109         if (get_bits1(&s->gb) != 0)
 110             return -1;  /* not PB frame */
 111 #endif
 112         printf("qscale=%d\n",get_bits(&s->gb, 5));
 113         skip_bits1(&s->gb);     /* Continuous Presence Multipoint mode: off */
 114     } else {
 115         printf("h263_plus = 1\n");
 116         /* H.263v2 */
 117         if (get_bits(&s->gb, 3) != 1){
 118             printf("H.263v2 A error\n");
 119             return -1;
 120         }
 121         if (get_bits(&s->gb, 3) != 6){ /* custom source format */
 122             printf("custom source format\n");
 123             return -1;
 124         }
 125         skip_bits(&s->gb, 12);
 126         skip_bits(&s->gb, 3);
 127         printf("pict_type=%d\n",get_bits(&s->gb, 3) + 1);
 128 //        if (s->pict_type != I_TYPE &&
 129 //            s->pict_type != P_TYPE)
 130 //            return -1;
 131         skip_bits(&s->gb, 7);
 132         skip_bits(&s->gb, 4); /* aspect ratio */
 133         width = (get_bits(&s->gb, 9) + 1) * 4;
 134         skip_bits1(&s->gb);
 135         height = get_bits(&s->gb, 9) * 4;
 136         printf("%d x %d\n",width,height);
 137         //if (height == 0)
 138         //    return -1;
 139         printf("qscale=%d\n",get_bits(&s->gb, 5));
 140     }
 141
 142     /* PEI */
 143     while (get_bits1(&s->gb) != 0) {
 144         skip_bits(&s->gb, 8);
 145     }
 146 //    s->f_code = 1;
 147 //    s->width = width;
 148 //    s->height = height;
 149     return 0;
 150 }
 151
 152 int postable[32768];
 153
 154 int main(int argc,char ** argv){
 155 int c;
 156 unsigned int head=-1;
 157 int pos=0;
 158 int frames=0;
 159 FILE *f;
 160 FILE *f2;
 161 muxer_t* avi;
 162 muxer_stream_t* mux;
 163 //unsigned char* buffer=malloc(0x200000);
 164 int i,len;
 165 int v_id=0;
 166 int flag=0;
 167 int flag2=0;
 168 int prefix=0;
 169
 170 // check if enough args were given
 171 if ( argc < 3 ){
 172     printf("Too few arguments given!\n"
 173            "Usage: %s <input_file> <output_file>\n", argv[0]);
 174
 175     return -1;
 176 }
 177 // input
 178 if(!(f=fopen(argv[1],"rb"))){
 179        printf("Couldn't open input file.\n");
 180        return -1;
 181 }
 182 // output
 183 if(!(f2=fopen(argv[2],"wb"))){
 184        printf("Couldn't open output file.\n");
 185        return -1;
 186 }
 187
 188 avi=muxer_new_muxer(MUXER_TYPE_AVI,f2);
 189 mux=muxer_new_stream(avi,MUXER_TYPE_VIDEO);
 190
 191 mux->buffer_size=0x200000;
 192 mux->buffer=malloc(mux->buffer_size);
 193
 194 mux->h.dwScale=1;
 195 mux->h.dwRate=10;
 196
 197 mux->bih=malloc(sizeof(BITMAPINFOHEADER));
 198 mux->bih->biSize=sizeof(BITMAPINFOHEADER);
 199 mux->bih->biPlanes=1;
 200 mux->bih->biBitCount=24;
 201 mux->bih->biCompression=0x6f766976;//      7669766f;
 202 muxer_write_header(avi);
 203
 204 /*
 205 c=fgetc(f); if(c) printf("error! not vivo file?\n");
 206 len=0;
 207 while((c=fgetc(f))>=0x80) len+=0x80*(c&0x0F);
 208 len+=c;
 209 printf("hdr1: %d\n",len);
 210 for(i=0;i<len;i++) fgetc(f);
 211 */
 212
 213 while((c=fgetc(f))>=0){
 214
 215     printf("%08X  %02X\n",ftell(f),c);
 216
 217     prefix=0;
 218     if(c==0x82){
 219         prefix=1;
 220         //continue;
 221         c=fgetc(f);
 222         printf("%08X  %02X\n",ftell(f),c);
 223     }
 224
 225     if(c==0x00){
 226         // header
 227         int len=0;
 228         while((c=fgetc(f))>=0x80) len+=0x80*(c&0x0F);
 229         len+=c;
 230         printf("header: 00 (%d)\n",len);
 231         for(i=0;i<len;i++) fgetc(f);
 232         continue;
 233     }
 234
 235     if((c&0xF0)==0x40){
 236         // audio
 237         len=24;
 238         if(prefix) len=fgetc(f);
 239         printf("audio: %02X (%d)\n",c,len);
 240         for(i=0;i<len;i++) fgetc(f);
 241         continue;
 242     }
 243     if((c&0xF0)==0x30){
 244         // audio
 245         len=40;
 246         if(prefix) len=fgetc(f);
 247         printf("audio: %02X (%d)\n",c,len);
 248         for(i=0;i<len;i++) fgetc(f);
 249         continue;
 250     }
 251     if(flag2 || (((c&0xF0)==0x10 || (c&0xF0)==0x20) && (c&0x0F)!=(v_id&0xF))){
 252         // end of frame:
 253         printf("Frame size: %d\n",mux->buffer_len);
 254         h263_decode_picture_header(mux->buffer);
 255         muxer_write_chunk(mux,mux->buffer_len,0x10);
 256         mux->buffer_len=0;
 257
 258         if((v_id&0xF0)==0x10) fprintf(stderr,"hmm. last video packet %02X\n",v_id);
 259     }
 260     flag2=0;
 261     if((c&0xF0)==0x10){
 262         // 128 byte
 263         len=128;
 264         if(prefix) len=fgetc(f);
 265         printf("video: %02X (%d)\n",c,len);
 266         fread(mux->buffer+mux->buffer_len,len,1,f);
 267         mux->buffer_len+=len;
 268     v_id=c;
 269         continue;
 270     }
 271     if((c&0xF0)==0x20){
 272         int len=fgetc(f);
 273         printf("video: %02X (%d)\n",c,len);
 274         fread(mux->buffer+mux->buffer_len,len,1,f);
 275         mux->buffer_len+=len;
 276         flag2=1;
 277     v_id=c;
 278         continue;
 279     }
 280     printf("error: %02X!\n",c);
 281     exit(1);
 282 }
 283
 284 if(!width) width=320;
 285 if(!height) height=240;
 286
 287 mux->bih->biWidth=width;
 288 mux->bih->biHeight=height;
 289 mux->bih->biSizeImage=3*width*height;
 290
 291 muxer_write_index(avi);
 292 fseek(f2,0,SEEK_SET);
 293 muxer_write_header(avi);
 294
 295 }