Parent Directory
| 
Revision Log
Revision 887 - (view) (download)
| 1 : | suxen_drol | 887 | /* |
| 2 : | rawdec: raw mpeg-4 bitstream decoder | ||
| 3 : | (c)2001-2003 peter ross <pross@xvid.org> | ||
| 4 : | |||
| 5 : | */ | ||
| 6 : | |||
| 7 : | |||
| 8 : | #include <stdio.h> | ||
| 9 : | #include <stdlib.h> | ||
| 10 : | #include <string.h> | ||
| 11 : | |||
| 12 : | #include <xvid.h> | ||
| 13 : | |||
| 14 : | |||
| 15 : | const static char * xvid_err(int err) | ||
| 16 : | { | ||
| 17 : | if (err == XVID_ERR_MEMORY) return "mem"; | ||
| 18 : | if (err == XVID_ERR_FORMAT) return "format"; | ||
| 19 : | if (err == XVID_ERR_VERSION) return "version"; | ||
| 20 : | return "unknown"; | ||
| 21 : | } | ||
| 22 : | |||
| 23 : | |||
| 24 : | /* returns true if the string contains scanf %'s chars | ||
| 25 : | (ignores %% sequences) */ | ||
| 26 : | |||
| 27 : | static int detect_sscanf(const char *s) | ||
| 28 : | { | ||
| 29 : | int percent = 0; | ||
| 30 : | while(*s){ | ||
| 31 : | if (percent){ | ||
| 32 : | if (*s!='%') | ||
| 33 : | return 1; | ||
| 34 : | percent=0; | ||
| 35 : | } | ||
| 36 : | if (*s=='%') | ||
| 37 : | percent = 1; | ||
| 38 : | s++; | ||
| 39 : | } | ||
| 40 : | return 0; | ||
| 41 : | } | ||
| 42 : | |||
| 43 : | |||
| 44 : | int main(int argc, char **argv) | ||
| 45 : | { | ||
| 46 : | char tmp[1024]; | ||
| 47 : | FILE * fbs, *fpgm; | ||
| 48 : | int fscanf; | ||
| 49 : | void * bs; | ||
| 50 : | int quit; | ||
| 51 : | int length; | ||
| 52 : | int pos; | ||
| 53 : | int frameno; | ||
| 54 : | int result; | ||
| 55 : | int width, height; | ||
| 56 : | |||
| 57 : | xvid_gbl_info_t info; | ||
| 58 : | xvid_gbl_init_t init; | ||
| 59 : | xvid_dec_create_t create; | ||
| 60 : | xvid_dec_frame_t frame; | ||
| 61 : | xvid_dec_stats_t stats; | ||
| 62 : | |||
| 63 : | memset(&info, 0, sizeof(info)); | ||
| 64 : | info.version = XVID_VERSION; | ||
| 65 : | |||
| 66 : | result = xvid_global(0, XVID_GBL_INFO, &info, 0); | ||
| 67 : | if (result < 0) | ||
| 68 : | { | ||
| 69 : | |||
| 70 : | printf("xvid_gbl_info error: %s\n", xvid_err(result)); | ||
| 71 : | return EXIT_FAILURE; | ||
| 72 : | } | ||
| 73 : | printf("xvidcore version %d.%d.%d (\"%s\")\n", | ||
| 74 : | XVID_MAJOR(info.actual_version), | ||
| 75 : | XVID_MINOR(info.actual_version), | ||
| 76 : | XVID_PATCH(info.actual_version), | ||
| 77 : | info.build); | ||
| 78 : | printf("cpu_flags:"); | ||
| 79 : | if ((info.cpu_flags & XVID_CPU_ASM)) printf(" ASM"); | ||
| 80 : | if ((info.cpu_flags & XVID_CPU_MMX)) printf(" MMX"); | ||
| 81 : | if ((info.cpu_flags & XVID_CPU_MMXEXT)) printf(" MMXEXT"); | ||
| 82 : | if ((info.cpu_flags & XVID_CPU_3DNOW)) printf(" 3DNOW"); | ||
| 83 : | if ((info.cpu_flags & XVID_CPU_3DNOWEXT)) printf(" 3DNOWEXT"); | ||
| 84 : | if ((info.cpu_flags & XVID_CPU_SSE)) printf(" SSE"); | ||
| 85 : | if ((info.cpu_flags & XVID_CPU_SSE2)) printf(" SSE2"); | ||
| 86 : | printf("\n"); | ||
| 87 : | printf("num_threads: %i\n", info.num_threads); | ||
| 88 : | |||
| 89 : | if (argc != 3 && argc != 5) | ||
| 90 : | { | ||
| 91 : | fprintf(stderr, | ||
| 92 : | "\nusage: rawdec bitstream outfile [width height]\n" | ||
| 93 : | "\n" | ||
| 94 : | "eg: rawdec foo.m4v [-|out.pgm|out%%05i.pgm]\n"); | ||
| 95 : | |||
| 96 : | return -1; | ||
| 97 : | } | ||
| 98 : | |||
| 99 : | /* open bitstream file & get length */ | ||
| 100 : | |||
| 101 : | fbs = fopen(argv[1], "rb"); | ||
| 102 : | if (fbs == NULL) | ||
| 103 : | { | ||
| 104 : | fprintf(stderr, "fatal: \"%s\" not found", argv[1]); | ||
| 105 : | return -1; | ||
| 106 : | } | ||
| 107 : | |||
| 108 : | if (argc == 5) | ||
| 109 : | { | ||
| 110 : | width = atoi(argv[3]); | ||
| 111 : | height = atoi(argv[4]); | ||
| 112 : | } | ||
| 113 : | else | ||
| 114 : | { | ||
| 115 : | width = height = 0; /* auto detect */ | ||
| 116 : | } | ||
| 117 : | |||
| 118 : | fseek(fbs, 0, SEEK_END); | ||
| 119 : | length = ftell(fbs); | ||
| 120 : | fseek(fbs, 0, SEEK_SET); | ||
| 121 : | |||
| 122 : | /* XXX: load entire bitstream file into memory */ | ||
| 123 : | |||
| 124 : | bs = malloc(length); | ||
| 125 : | if (bs == NULL) | ||
| 126 : | { | ||
| 127 : | fprintf(stderr, "fatal: bitstream malloc failed\n"); | ||
| 128 : | return -1; | ||
| 129 : | } | ||
| 130 : | |||
| 131 : | fread(bs, length, 1, fbs); | ||
| 132 : | fclose(fbs); | ||
| 133 : | |||
| 134 : | |||
| 135 : | /* otuput file stuff */ | ||
| 136 : | fscanf = detect_sscanf(argv[2]); | ||
| 137 : | if (argv[2][0] != '-' && !fscanf) | ||
| 138 : | { | ||
| 139 : | fpgm = fopen(argv[2], "wb"); | ||
| 140 : | if (fpgm == NULL){ | ||
| 141 : | fprintf(stderr,"aborting: \"%s\" write error\n", argv[2]); | ||
| 142 : | free(bs); | ||
| 143 : | return EXIT_FAILURE; | ||
| 144 : | } | ||
| 145 : | } | ||
| 146 : | |||
| 147 : | |||
| 148 : | /* init globals | ||
| 149 : | note: init.cpu_flags is optional | ||
| 150 : | */ | ||
| 151 : | |||
| 152 : | memset(&init, 0, sizeof(init)); | ||
| 153 : | init.version = XVID_VERSION; | ||
| 154 : | //init.cpu_flags = XVID_CPU_FORCE; | ||
| 155 : | result = xvid_global(0, XVID_GBL_INIT, &init, NULL); | ||
| 156 : | if (result < 0) | ||
| 157 : | { | ||
| 158 : | free(bs); | ||
| 159 : | printf("xvid_gbl_init error: %s\n", xvid_err(result)); | ||
| 160 : | return EXIT_FAILURE; | ||
| 161 : | } | ||
| 162 : | |||
| 163 : | |||
| 164 : | /* create decoder instance */ | ||
| 165 : | |||
| 166 : | memset(&create, 0, sizeof(create)); | ||
| 167 : | create.version = XVID_VERSION; | ||
| 168 : | create.width = width; | ||
| 169 : | create.height = height; | ||
| 170 : | result = xvid_decore(0, XVID_DEC_CREATE, &create, NULL); | ||
| 171 : | if (result < 0) | ||
| 172 : | { | ||
| 173 : | free(bs); | ||
| 174 : | fprintf(stderr, "xvid_dec_create error: %s\n", xvid_err(result)); | ||
| 175 : | return EXIT_FAILURE; | ||
| 176 : | } | ||
| 177 : | |||
| 178 : | |||
| 179 : | /* init other struct; we only have to do this once */ | ||
| 180 : | memset(&frame, 0, sizeof(frame)); | ||
| 181 : | frame.version = XVID_VERSION; | ||
| 182 : | |||
| 183 : | memset(&stats, 0, sizeof(stats)); | ||
| 184 : | stats.version = XVID_VERSION; | ||
| 185 : | |||
| 186 : | |||
| 187 : | /* decode frames ... */ | ||
| 188 : | quit = 0; | ||
| 189 : | pos = 0; | ||
| 190 : | frameno = 0; | ||
| 191 : | while (!quit) | ||
| 192 : | { | ||
| 193 : | int y; | ||
| 194 : | |||
| 195 : | if (pos < length) | ||
| 196 : | { | ||
| 197 : | frame.bitstream = (void*)((unsigned long)bs + pos); | ||
| 198 : | frame.length = length - pos; | ||
| 199 : | }else{ | ||
| 200 : | printf("**FLUSH LAST FRAME\n"); | ||
| 201 : | frame.length = -1; /* flush last frame */ | ||
| 202 : | quit = 1; | ||
| 203 : | } | ||
| 204 : | /* xvid will return pointers to it's internal image buffers */ | ||
| 205 : | frame.output.csp = XVID_CSP_INTERNAL; | ||
| 206 : | |||
| 207 : | result = xvid_decore(create.handle, XVID_DEC_DECODE, &frame, &stats); | ||
| 208 : | fprintf(stdout," #%i [%i/%i] length=%i\n", frameno, pos, length, frame.length); fflush(stderr); | ||
| 209 : | if (result < 0) | ||
| 210 : | { | ||
| 211 : | fprintf(stderr, "xvid_dec_decode error: %s\n", xvid_err(result)); | ||
| 212 : | break; | ||
| 213 : | } | ||
| 214 : | |||
| 215 : | /* increase position by the ammount consumed for this frame */ | ||
| 216 : | pos += result; | ||
| 217 : | |||
| 218 : | |||
| 219 : | if (stats.type == XVID_TYPE_NOTHING) /* no output */ | ||
| 220 : | { | ||
| 221 : | continue; | ||
| 222 : | } | ||
| 223 : | |||
| 224 : | if (stats.type == XVID_TYPE_VOL) /* vol/resize */ | ||
| 225 : | { | ||
| 226 : | printf("**RESIZE** %i,%i\n", stats.data.vol.width, stats.data.vol.height); | ||
| 227 : | width = stats.data.vol.width; | ||
| 228 : | height = stats.data.vol.height; | ||
| 229 : | printf("%i\n", frame.length); | ||
| 230 : | continue; | ||
| 231 : | } | ||
| 232 : | |||
| 233 : | /* write output image to pgm file */ | ||
| 234 : | if (argv[2][0] == '-') | ||
| 235 : | { | ||
| 236 : | fpgm = stdout; | ||
| 237 : | }else if (fscanf) { | ||
| 238 : | sprintf(tmp,argv[2], frameno); | ||
| 239 : | fpgm = fopen(tmp, "wb"); | ||
| 240 : | if (fpgm == NULL) { | ||
| 241 : | fprintf(stderr,"aborting: \"%s\" write error\n", tmp); | ||
| 242 : | break; | ||
| 243 : | } | ||
| 244 : | } | ||
| 245 : | |||
| 246 : | sprintf(tmp, "P5\n%i %i\n255\n", width, height * 3 / 2); | ||
| 247 : | fwrite(tmp, strlen(tmp) , 1, fpgm); | ||
| 248 : | |||
| 249 : | for (y = height; y; y--) { | ||
| 250 : | fwrite(frame.output.plane[0], width, 1, fpgm); | ||
| 251 : | frame.output.plane[0] = (void*)((char *)frame.output.plane[0] + frame.output.stride[0]); | ||
| 252 : | } | ||
| 253 : | |||
| 254 : | for (y = height >> 1; y; y--) { | ||
| 255 : | fwrite(frame.output.plane[1], width / 2, 1, fpgm); | ||
| 256 : | frame.output.plane[1] = (void*)((char *)frame.output.plane[1] + frame.output.stride[1]); | ||
| 257 : | |||
| 258 : | fwrite(frame.output.plane[2], width / 2, 1, fpgm); | ||
| 259 : | frame.output.plane[2] = (void*)((char *)frame.output.plane[2] + frame.output.stride[2]); | ||
| 260 : | } | ||
| 261 : | |||
| 262 : | if (argv[2][0] != '-' && fscanf) { | ||
| 263 : | fclose(fpgm); | ||
| 264 : | } | ||
| 265 : | |||
| 266 : | |||
| 267 : | /* finally, increment frame num */ | ||
| 268 : | frameno++; | ||
| 269 : | } | ||
| 270 : | |||
| 271 : | |||
| 272 : | /* cleanup & exit */ | ||
| 273 : | |||
| 274 : | if (argv[2][0] != '-' && !fscanf) | ||
| 275 : | { | ||
| 276 : | fclose(fpgm); | ||
| 277 : | } | ||
| 278 : | |||
| 279 : | free(bs); | ||
| 280 : | |||
| 281 : | result = xvid_decore(create.handle, XVID_DEC_DESTROY, NULL, NULL); | ||
| 282 : | if (result < 0) | ||
| 283 : | { | ||
| 284 : | fprintf(stderr, "xvid_dec_destroy error: %s\n", xvid_err(result)); | ||
| 285 : | return EXIT_FAILURE; | ||
| 286 : | } | ||
| 287 : | |||
| 288 : | return EXIT_SUCCESS; | ||
| 289 : | } |
| No admin address has been configured | ViewVC Help |
| Powered by ViewVC 1.0.4 |