Parent Directory
| 
Revision Log
Revision 1501 - (view) (download)
| 1 : | edgomez | 547 | /***************************************************************************** |
| 2 : | chl | 376 | * |
| 3 : | edgomez | 547 | * XVID MPEG-4 VIDEO CODEC |
| 4 : | * - Console based decoding test application - | ||
| 5 : | chl | 376 | * |
| 6 : | edgomez | 851 | * Copyright(C) 2002-2003 Christoph Lampert |
| 7 : | edgomez | 1382 | * 2002-2003 Edouard Gomez <ed.gomez@free.fr> |
| 8 : | chl | 376 | * |
| 9 : | edgomez | 547 | * This program is free software; you can redistribute it and/or modify |
| 10 : | * it under the terms of the GNU General Public License as published by | ||
| 11 : | * the Free Software Foundation; either version 2 of the License, or | ||
| 12 : | * (at your option) any later version. | ||
| 13 : | chl | 376 | * |
| 14 : | edgomez | 547 | * This program is distributed in the hope that it will be useful, |
| 15 : | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 16 : | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 17 : | * GNU General Public License for more details. | ||
| 18 : | chl | 376 | * |
| 19 : | edgomez | 547 | * You should have received a copy of the GNU General Public License |
| 20 : | * along with this program; if not, write to the Free Software | ||
| 21 : | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
| 22 : | * | ||
| 23 : | edgomez | 1501 | * $Id: xvid_decraw.c,v 1.10.2.5 2004-07-17 11:41:41 edgomez Exp $ |
| 24 : | edgomez | 547 | * |
| 25 : | ****************************************************************************/ | ||
| 26 : | chl | 376 | |
| 27 : | edgomez | 547 | /***************************************************************************** |
| 28 : | chl | 376 | * |
| 29 : | edgomez | 547 | * Application notes : |
| 30 : | chl | 376 | * |
| 31 : | edgomez | 547 | * An MPEG-4 bitstream is read from an input file (or stdin) and decoded, |
| 32 : | * the speed for this is measured. | ||
| 33 : | edgomez | 1382 | * |
| 34 : | * The program is plain C and needs no libraries except for libxvidcore, | ||
| 35 : | * and maths-lib. | ||
| 36 : | chl | 376 | * |
| 37 : | edgomez | 1382 | * Use ./xvid_decraw -help for a list of options |
| 38 : | chl | 376 | * |
| 39 : | edgomez | 547 | ****************************************************************************/ |
| 40 : | chl | 376 | |
| 41 : | #include <stdio.h> | ||
| 42 : | #include <stdlib.h> | ||
| 43 : | edgomez | 547 | #include <string.h> |
| 44 : | #include <math.h> | ||
| 45 : | edgomez | 824 | #ifndef WIN32 |
| 46 : | edgomez | 547 | #include <sys/time.h> |
| 47 : | #else | ||
| 48 : | #include <time.h> | ||
| 49 : | #endif | ||
| 50 : | chl | 376 | |
| 51 : | edgomez | 547 | #include "xvid.h" |
| 52 : | chl | 376 | |
| 53 : | edgomez | 547 | /***************************************************************************** |
| 54 : | * Global vars in module and constants | ||
| 55 : | ****************************************************************************/ | ||
| 56 : | chl | 376 | |
| 57 : | edgomez | 547 | /* max number of frames */ |
| 58 : | #define ABS_MAXFRAMENR 9999 | ||
| 59 : | chl | 376 | |
| 60 : | edgomez | 1418 | #define USE_PNM 0 |
| 61 : | #define USE_TGA 1 | ||
| 62 : | |||
| 63 : | edgomez | 547 | static int XDIM = 0; |
| 64 : | static int YDIM = 0; | ||
| 65 : | static int ARG_SAVEDECOUTPUT = 0; | ||
| 66 : | static int ARG_SAVEMPEGSTREAM = 0; | ||
| 67 : | static char *ARG_INPUTFILE = NULL; | ||
| 68 : | edgomez | 1414 | static int CSP = XVID_CSP_I420; |
| 69 : | static int BPP = 1; | ||
| 70 : | edgomez | 1418 | static int FORMAT = USE_PNM; |
| 71 : | chl | 376 | |
| 72 : | edgomez | 547 | static char filepath[256] = "./"; |
| 73 : | static void *dec_handle = NULL; | ||
| 74 : | chl | 376 | |
| 75 : | edgomez | 1382 | #define BUFFER_SIZE (2*1024*1024) |
| 76 : | chl | 376 | |
| 77 : | edgomez | 547 | /***************************************************************************** |
| 78 : | * Local prototypes | ||
| 79 : | ****************************************************************************/ | ||
| 80 : | chl | 376 | |
| 81 : | edgomez | 547 | static double msecond(); |
| 82 : | static int dec_init(int use_assembler); | ||
| 83 : | static int dec_main(unsigned char *istream, | ||
| 84 : | edgomez | 851 | unsigned char *ostream, |
| 85 : | int istream_size, | ||
| 86 : | edgomez | 1382 | xvid_dec_stats_t *xvid_dec_stats); |
| 87 : | edgomez | 547 | static int dec_stop(); |
| 88 : | static void usage(); | ||
| 89 : | edgomez | 1433 | static int write_image(char *prefix, unsigned char *image); |
| 90 : | static int write_pnm(char *filename, unsigned char *image); | ||
| 91 : | static int write_tga(char *filename, unsigned char *image); | ||
| 92 : | edgomez | 547 | |
| 93 : | edgomez | 1382 | const char * type2str(int type) |
| 94 : | { | ||
| 95 : | if (type==XVID_TYPE_IVOP) | ||
| 96 : | return "I"; | ||
| 97 : | if (type==XVID_TYPE_PVOP) | ||
| 98 : | return "P"; | ||
| 99 : | if (type==XVID_TYPE_BVOP) | ||
| 100 : | return "B"; | ||
| 101 : | return "S"; | ||
| 102 : | } | ||
| 103 : | |||
| 104 : | edgomez | 547 | /***************************************************************************** |
| 105 : | * Main program | ||
| 106 : | ****************************************************************************/ | ||
| 107 : | |||
| 108 : | int main(int argc, char *argv[]) | ||
| 109 : | chl | 376 | { |
| 110 : | edgomez | 547 | unsigned char *mp4_buffer = NULL; |
| 111 : | unsigned char *mp4_ptr = NULL; | ||
| 112 : | unsigned char *out_buffer = NULL; | ||
| 113 : | edgomez | 1382 | int useful_bytes; |
| 114 : | xvid_dec_stats_t xvid_dec_stats; | ||
| 115 : | edgomez | 851 | |
| 116 : | edgomez | 547 | double totaldectime; |
| 117 : | |||
| 118 : | long totalsize; | ||
| 119 : | int status; | ||
| 120 : | |||
| 121 : | int use_assembler = 0; | ||
| 122 : | |||
| 123 : | char filename[256]; | ||
| 124 : | |||
| 125 : | FILE *in_file; | ||
| 126 : | int filenr; | ||
| 127 : | int i; | ||
| 128 : | |||
| 129 : | printf("xvid_decraw - raw mpeg4 bitstream decoder "); | ||
| 130 : | edgomez | 851 | printf("written by Christoph Lampert 2002-2003\n\n"); |
| 131 : | edgomez | 547 | |
| 132 : | /***************************************************************************** | ||
| 133 : | * Command line parsing | ||
| 134 : | ****************************************************************************/ | ||
| 135 : | |||
| 136 : | for (i=1; i< argc; i++) { | ||
| 137 : | |||
| 138 : | if (strcmp("-asm", argv[i]) == 0 ) { | ||
| 139 : | use_assembler = 1; | ||
| 140 : | edgomez | 1382 | } else if (strcmp("-d", argv[i]) == 0) { |
| 141 : | ARG_SAVEDECOUTPUT = 1; | ||
| 142 : | } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) { | ||
| 143 : | edgomez | 547 | i++; |
| 144 : | ARG_INPUTFILE = argv[i]; | ||
| 145 : | edgomez | 1382 | } else if (strcmp("-m", argv[i]) == 0) { |
| 146 : | ARG_SAVEMPEGSTREAM = 1; | ||
| 147 : | edgomez | 1414 | } else if (strcmp("-c", argv[i]) == 0 && i < argc - 1 ) { |
| 148 : | i++; | ||
| 149 : | if (strcmp(argv[i], "rgb16") == 0) { | ||
| 150 : | CSP = XVID_CSP_RGB555; | ||
| 151 : | BPP = 2; | ||
| 152 : | } else if (strcmp(argv[i], "rgb24") == 0) { | ||
| 153 : | CSP = XVID_CSP_BGR; | ||
| 154 : | BPP = 3; | ||
| 155 : | } else if (strcmp(argv[i], "rgb32") == 0) { | ||
| 156 : | CSP = XVID_CSP_BGRA; | ||
| 157 : | BPP = 4; | ||
| 158 : | } else if (strcmp(argv[i], "yv12") == 0) { | ||
| 159 : | CSP = XVID_CSP_YV12; | ||
| 160 : | BPP = 1; | ||
| 161 : | } else { | ||
| 162 : | CSP = XVID_CSP_I420; | ||
| 163 : | BPP = 1; | ||
| 164 : | } | ||
| 165 : | edgomez | 1418 | } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) { |
| 166 : | i++; | ||
| 167 : | if (strcmp(argv[i], "tga") == 0) { | ||
| 168 : | FORMAT = USE_TGA; | ||
| 169 : | } else { | ||
| 170 : | FORMAT = USE_PNM; | ||
| 171 : | } | ||
| 172 : | edgomez | 1382 | } else if (strcmp("-help", argv[i]) == 0) { |
| 173 : | edgomez | 547 | usage(); |
| 174 : | return(0); | ||
| 175 : | edgomez | 1382 | } else { |
| 176 : | edgomez | 547 | usage(); |
| 177 : | exit(-1); | ||
| 178 : | } | ||
| 179 : | chl | 376 | } |
| 180 : | edgomez | 547 | |
| 181 : | suxen_drol | 1430 | #if defined(_MSC_VER) |
| 182 : | if (ARG_INPUTFILE==NULL) { | ||
| 183 : | fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch.\n\n"); | ||
| 184 : | } | ||
| 185 : | #endif | ||
| 186 : | |||
| 187 : | edgomez | 547 | /***************************************************************************** |
| 188 : | * Values checking | ||
| 189 : | ****************************************************************************/ | ||
| 190 : | |||
| 191 : | if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) { | ||
| 192 : | in_file = stdin; | ||
| 193 : | } | ||
| 194 : | else { | ||
| 195 : | |||
| 196 : | in_file = fopen(ARG_INPUTFILE, "rb"); | ||
| 197 : | if (in_file == NULL) { | ||
| 198 : | fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE); | ||
| 199 : | edgomez | 1382 | return(-1); |
| 200 : | edgomez | 547 | } |
| 201 : | } | ||
| 202 : | |||
| 203 : | edgomez | 1418 | /* PNM/PGM format can't handle 16/32 bit data */ |
| 204 : | if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) { | ||
| 205 : | FORMAT = USE_TGA; | ||
| 206 : | } | ||
| 207 : | |||
| 208 : | edgomez | 547 | /***************************************************************************** |
| 209 : | * Memory allocation | ||
| 210 : | ****************************************************************************/ | ||
| 211 : | |||
| 212 : | /* Memory for encoded mp4 stream */ | ||
| 213 : | mp4_buffer = (unsigned char *) malloc(BUFFER_SIZE); | ||
| 214 : | mp4_ptr = mp4_buffer; | ||
| 215 : | if (!mp4_buffer) | ||
| 216 : | goto free_all_memory; | ||
| 217 : | |||
| 218 : | /***************************************************************************** | ||
| 219 : | * XviD PART Start | ||
| 220 : | ****************************************************************************/ | ||
| 221 : | |||
| 222 : | status = dec_init(use_assembler); | ||
| 223 : | if (status) { | ||
| 224 : | fprintf(stderr, | ||
| 225 : | edgomez | 1382 | "Decore INIT problem, return value %d\n", status); |
| 226 : | edgomez | 547 | goto release_all; |
| 227 : | } | ||
| 228 : | |||
| 229 : | |||
| 230 : | /***************************************************************************** | ||
| 231 : | * Main loop | ||
| 232 : | ****************************************************************************/ | ||
| 233 : | |||
| 234 : | edgomez | 851 | /* Fill the buffer */ |
| 235 : | edgomez | 1382 | useful_bytes = fread(mp4_buffer, 1, BUFFER_SIZE, in_file); |
| 236 : | edgomez | 559 | |
| 237 : | edgomez | 547 | totaldectime = 0; |
| 238 : | totalsize = 0; | ||
| 239 : | filenr = 0; | ||
| 240 : | mp4_ptr = mp4_buffer; | ||
| 241 : | |||
| 242 : | do { | ||
| 243 : | int used_bytes = 0; | ||
| 244 : | double dectime; | ||
| 245 : | |||
| 246 : | edgomez | 851 | /* |
| 247 : | * If the buffer is half empty or there are no more bytes in it | ||
| 248 : | * then fill it. | ||
| 249 : | */ | ||
| 250 : | edgomez | 1382 | if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) { |
| 251 : | int already_in_buffer = (mp4_buffer + BUFFER_SIZE - mp4_ptr); | ||
| 252 : | edgomez | 547 | |
| 253 : | edgomez | 851 | /* Move data if needed */ |
| 254 : | edgomez | 1382 | if (already_in_buffer > 0) |
| 255 : | memcpy(mp4_buffer, mp4_ptr, already_in_buffer); | ||
| 256 : | edgomez | 547 | |
| 257 : | edgomez | 851 | /* Update mp4_ptr */ |
| 258 : | mp4_ptr = mp4_buffer; | ||
| 259 : | edgomez | 547 | |
| 260 : | edgomez | 851 | /* read new data */ |
| 261 : | edgomez | 1382 | if(feof(in_file)) |
| 262 : | edgomez | 547 | break; |
| 263 : | |||
| 264 : | edgomez | 1382 | useful_bytes += fread(mp4_buffer + already_in_buffer, |
| 265 : | 1, BUFFER_SIZE - already_in_buffer, | ||
| 266 : | in_file); | ||
| 267 : | edgomez | 547 | |
| 268 : | } | ||
| 269 : | |||
| 270 : | |||
| 271 : | edgomez | 1382 | /* This loop is needed to handle VOL/NVOP reading */ |
| 272 : | edgomez | 851 | do { |
| 273 : | edgomez | 547 | |
| 274 : | edgomez | 851 | /* Decode frame */ |
| 275 : | dectime = msecond(); | ||
| 276 : | edgomez | 1382 | used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats); |
| 277 : | edgomez | 851 | dectime = msecond() - dectime; |
| 278 : | edgomez | 547 | |
| 279 : | edgomez | 1382 | /* Resize image buffer if needed */ |
| 280 : | if(xvid_dec_stats.type == XVID_TYPE_VOL) { | ||
| 281 : | |||
| 282 : | /* Check if old buffer is smaller */ | ||
| 283 : | if(XDIM*YDIM < xvid_dec_stats.data.vol.width*xvid_dec_stats.data.vol.height) { | ||
| 284 : | |||
| 285 : | /* Copy new witdh and new height from the vol structure */ | ||
| 286 : | XDIM = xvid_dec_stats.data.vol.width; | ||
| 287 : | YDIM = xvid_dec_stats.data.vol.height; | ||
| 288 : | |||
| 289 : | /* Free old output buffer*/ | ||
| 290 : | if(out_buffer) free(out_buffer); | ||
| 291 : | |||
| 292 : | /* Allocate the new buffer */ | ||
| 293 : | out_buffer = (unsigned char*)malloc(XDIM*YDIM*4); | ||
| 294 : | if(out_buffer == NULL) | ||
| 295 : | goto free_all_memory; | ||
| 296 : | |||
| 297 : | fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM); | ||
| 298 : | } | ||
| 299 : | edgomez | 851 | } |
| 300 : | edgomez | 547 | |
| 301 : | edgomez | 851 | /* Update buffer pointers */ |
| 302 : | edgomez | 1382 | if(used_bytes > 0) { |
| 303 : | mp4_ptr += used_bytes; | ||
| 304 : | useful_bytes -= used_bytes; | ||
| 305 : | edgomez | 559 | |
| 306 : | edgomez | 1382 | /* Total size */ |
| 307 : | totalsize += used_bytes; | ||
| 308 : | } | ||
| 309 : | edgomez | 547 | |
| 310 : | edgomez | 1382 | }while(xvid_dec_stats.type <= 0 && useful_bytes > 0); |
| 311 : | edgomez | 547 | |
| 312 : | edgomez | 1382 | /* Check if there is a negative number of useful bytes left in buffer |
| 313 : | * This means we went too far */ | ||
| 314 : | if(useful_bytes < 0) | ||
| 315 : | break; | ||
| 316 : | edgomez | 851 | |
| 317 : | edgomez | 1382 | /* Updated data - Count only usefull decode time */ |
| 318 : | edgomez | 851 | totaldectime += dectime; |
| 319 : | edgomez | 547 | |
| 320 : | edgomez | 851 | |
| 321 : | edgomez | 1382 | printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n", |
| 322 : | filenr, type2str(xvid_dec_stats.type), dectime, used_bytes); | ||
| 323 : | |||
| 324 : | edgomez | 851 | /* Save individual mpeg4 stream if required */ |
| 325 : | edgomez | 1382 | if(ARG_SAVEMPEGSTREAM) { |
| 326 : | edgomez | 547 | FILE *filehandle = NULL; |
| 327 : | |||
| 328 : | sprintf(filename, "%sframe%05d.m4v", filepath, filenr); | ||
| 329 : | filehandle = fopen(filename, "wb"); | ||
| 330 : | if(!filehandle) { | ||
| 331 : | fprintf(stderr, | ||
| 332 : | edgomez | 851 | "Error writing single mpeg4 stream to file %s\n", |
| 333 : | filename); | ||
| 334 : | edgomez | 547 | } |
| 335 : | else { | ||
| 336 : | edgomez | 1382 | fwrite(mp4_ptr-used_bytes, 1, used_bytes, filehandle); |
| 337 : | edgomez | 547 | fclose(filehandle); |
| 338 : | } | ||
| 339 : | } | ||
| 340 : | edgomez | 851 | |
| 341 : | edgomez | 547 | /* Save output frame if required */ |
| 342 : | if (ARG_SAVEDECOUTPUT) { | ||
| 343 : | edgomez | 1418 | sprintf(filename, "%sdec%05d", filepath, filenr); |
| 344 : | if(write_image(filename, out_buffer)) { | ||
| 345 : | edgomez | 547 | fprintf(stderr, |
| 346 : | edgomez | 1433 | "Error writing decoded frame %s\n", |
| 347 : | edgomez | 851 | filename); |
| 348 : | edgomez | 547 | } |
| 349 : | } | ||
| 350 : | |||
| 351 : | filenr++; | ||
| 352 : | |||
| 353 : | edgomez | 851 | } while ( (status>=0) && (filenr<ABS_MAXFRAMENR)); |
| 354 : | edgomez | 547 | |
| 355 : | /***************************************************************************** | ||
| 356 : | edgomez | 851 | * Flush decoder buffers |
| 357 : | ****************************************************************************/ | ||
| 358 : | |||
| 359 : | edgomez | 1382 | do { |
| 360 : | |||
| 361 : | edgomez | 851 | /* Fake vars */ |
| 362 : | edgomez | 1382 | int used_bytes; |
| 363 : | edgomez | 851 | double dectime; |
| 364 : | |||
| 365 : | edgomez | 1382 | do { |
| 366 : | dectime = msecond(); | ||
| 367 : | used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats); | ||
| 368 : | dectime = msecond() - dectime; | ||
| 369 : | }while(used_bytes>=0 && xvid_dec_stats.type <= 0); | ||
| 370 : | edgomez | 851 | |
| 371 : | edgomez | 1382 | if (used_bytes < 0) { /* XVID_ERR_END */ |
| 372 : | break; | ||
| 373 : | } | ||
| 374 : | edgomez | 851 | |
| 375 : | /* Updated data - Count only usefull decode time */ | ||
| 376 : | totaldectime += dectime; | ||
| 377 : | |||
| 378 : | /* Prints some decoding stats */ | ||
| 379 : | edgomez | 1382 | printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n", |
| 380 : | filenr, type2str(xvid_dec_stats.type), dectime, used_bytes); | ||
| 381 : | edgomez | 851 | |
| 382 : | /* Save output frame if required */ | ||
| 383 : | if (ARG_SAVEDECOUTPUT) { | ||
| 384 : | edgomez | 1418 | sprintf(filename, "%sdec%05d", filepath, filenr); |
| 385 : | if(write_image(filename, out_buffer)) { | ||
| 386 : | edgomez | 851 | fprintf(stderr, |
| 387 : | suxen_drol | 1430 | "Error writing decoded frame %s\n", |
| 388 : | edgomez | 851 | filename); |
| 389 : | } | ||
| 390 : | } | ||
| 391 : | |||
| 392 : | filenr++; | ||
| 393 : | |||
| 394 : | edgomez | 1382 | }while(1); |
| 395 : | edgomez | 851 | |
| 396 : | /***************************************************************************** | ||
| 397 : | edgomez | 547 | * Calculate totals and averages for output, print results |
| 398 : | ****************************************************************************/ | ||
| 399 : | |||
| 400 : | suxen_drol | 1430 | if (filenr>0) { |
| 401 : | totalsize /= filenr; | ||
| 402 : | totaldectime /= filenr; | ||
| 403 : | printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n", | ||
| 404 : | totaldectime, 1000/totaldectime, (int)totalsize); | ||
| 405 : | }else{ | ||
| 406 : | printf("Nothing was decoded!\n"); | ||
| 407 : | } | ||
| 408 : | edgomez | 547 | |
| 409 : | /***************************************************************************** | ||
| 410 : | * XviD PART Stop | ||
| 411 : | edgomez | 559 | ****************************************************************************/ |
| 412 : | edgomez | 547 | |
| 413 : | release_all: | ||
| 414 : | if (dec_handle) { | ||
| 415 : | status = dec_stop(); | ||
| 416 : | if (status) | ||
| 417 : | fprintf(stderr, "decore RELEASE problem return value %d\n", status); | ||
| 418 : | } | ||
| 419 : | |||
| 420 : | free_all_memory: | ||
| 421 : | free(out_buffer); | ||
| 422 : | free(mp4_buffer); | ||
| 423 : | |||
| 424 : | edgomez | 1382 | return(0); |
| 425 : | chl | 376 | } |
| 426 : | |||
| 427 : | edgomez | 547 | /***************************************************************************** |
| 428 : | * Usage function | ||
| 429 : | ****************************************************************************/ | ||
| 430 : | chl | 376 | |
| 431 : | edgomez | 547 | static void usage() |
| 432 : | chl | 376 | { |
| 433 : | edgomez | 547 | |
| 434 : | edgomez | 1382 | fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n"); |
| 435 : | edgomez | 547 | fprintf(stderr, "Options :\n"); |
| 436 : | fprintf(stderr, " -asm : use assembly optimizations (default=disabled)\n"); | ||
| 437 : | fprintf(stderr, " -i string : input filename (default=stdin)\n"); | ||
| 438 : | edgomez | 1382 | fprintf(stderr, " -d : save decoder output\n"); |
| 439 : | edgomez | 1418 | fprintf(stderr, " -c csp : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n"); |
| 440 : | fprintf(stderr, " -f format : choose output file format (tga, pnm, pgm)\n"); | ||
| 441 : | edgomez | 1382 | fprintf(stderr, " -m : save mpeg4 raw stream to individual files\n"); |
| 442 : | edgomez | 547 | fprintf(stderr, " -help : This help message\n"); |
| 443 : | fprintf(stderr, " (* means default)\n"); | ||
| 444 : | |||
| 445 : | } | ||
| 446 : | |||
| 447 : | /***************************************************************************** | ||
| 448 : | * "helper" functions | ||
| 449 : | ****************************************************************************/ | ||
| 450 : | |||
| 451 : | /* return the current time in milli seconds */ | ||
| 452 : | static double | ||
| 453 : | msecond() | ||
| 454 : | { | ||
| 455 : | edgomez | 824 | #ifndef WIN32 |
| 456 : | edgomez | 547 | struct timeval tv; |
| 457 : | gettimeofday(&tv, 0); | ||
| 458 : | edgomez | 1382 | return((double)tv.tv_sec*1.0e3 + (double)tv.tv_usec*1.0e-3); |
| 459 : | edgomez | 547 | #else |
| 460 : | clock_t clk; | ||
| 461 : | clk = clock(); | ||
| 462 : | edgomez | 1382 | return(clk * 1000 / CLOCKS_PER_SEC); |
| 463 : | edgomez | 547 | #endif |
| 464 : | } | ||
| 465 : | |||
| 466 : | /***************************************************************************** | ||
| 467 : | * output functions | ||
| 468 : | ****************************************************************************/ | ||
| 469 : | |||
| 470 : | edgomez | 1433 | static int write_image(char *prefix, unsigned char *image) |
| 471 : | edgomez | 1418 | { |
| 472 : | char filename[1024]; | ||
| 473 : | char *ext; | ||
| 474 : | int ret; | ||
| 475 : | |||
| 476 : | if (FORMAT == USE_PNM && BPP == 1) { | ||
| 477 : | ext = "pgm"; | ||
| 478 : | } else if (FORMAT == USE_PNM && BPP == 3) { | ||
| 479 : | ext = "pnm"; | ||
| 480 : | } else if (FORMAT == USE_TGA) { | ||
| 481 : | ext = "tga"; | ||
| 482 : | } else { | ||
| 483 : | fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used"); | ||
| 484 : | exit(-1); | ||
| 485 : | } | ||
| 486 : | |||
| 487 : | sprintf(filename, "%s.%s", prefix, ext); | ||
| 488 : | |||
| 489 : | if (FORMAT == USE_PNM) { | ||
| 490 : | ret = write_pnm(filename, image); | ||
| 491 : | } else { | ||
| 492 : | ret = write_tga(filename, image); | ||
| 493 : | } | ||
| 494 : | |||
| 495 : | return(ret); | ||
| 496 : | } | ||
| 497 : | |||
| 498 : | edgomez | 1433 | static int write_tga(char *filename, unsigned char *image) |
| 499 : | edgomez | 547 | { |
| 500 : | edgomez | 1414 | FILE * f; |
| 501 : | char hdr[18]; | ||
| 502 : | edgomez | 547 | |
| 503 : | edgomez | 1414 | f = fopen(filename, "wb"); |
| 504 : | if ( f == NULL) { | ||
| 505 : | return -1; | ||
| 506 : | } | ||
| 507 : | edgomez | 547 | |
| 508 : | edgomez | 1414 | hdr[0] = 0; /* ID length */ |
| 509 : | hdr[1] = 0; /* Color map type */ | ||
| 510 : | hdr[2] = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */ | ||
| 511 : | hdr[3] = 0; /* Color map specification (not used) */ | ||
| 512 : | hdr[4] = 0; /* Color map specification (not used) */ | ||
| 513 : | hdr[5] = 0; /* Color map specification (not used) */ | ||
| 514 : | hdr[6] = 0; /* Color map specification (not used) */ | ||
| 515 : | hdr[7] = 0; /* Color map specification (not used) */ | ||
| 516 : | hdr[8] = 0; /* LSB X origin */ | ||
| 517 : | hdr[9] = 0; /* MSB X origin */ | ||
| 518 : | hdr[10] = 0; /* LSB Y origin */ | ||
| 519 : | hdr[11] = 0; /* MSB Y origin */ | ||
| 520 : | hdr[12] = (XDIM>>0)&0xff; /* LSB Width */ | ||
| 521 : | hdr[13] = (XDIM>>8)&0xff; /* MSB Width */ | ||
| 522 : | if (BPP > 1) { | ||
| 523 : | hdr[14] = (YDIM>>0)&0xff; /* LSB Height */ | ||
| 524 : | hdr[15] = (YDIM>>8)&0xff; /* MSB Height */ | ||
| 525 : | } else { | ||
| 526 : | hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */ | ||
| 527 : | hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */ | ||
| 528 : | } | ||
| 529 : | hdr[16] = BPP*8; | ||
| 530 : | hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */ | ||
| 531 : | |||
| 532 : | /* Write header */ | ||
| 533 : | fwrite(hdr, 1, sizeof(hdr), f); | ||
| 534 : | chl | 376 | |
| 535 : | edgomez | 1414 | #ifdef ARCH_IS_LITTLE_ENDIAN |
| 536 : | /* write first plane */ | ||
| 537 : | fwrite(image, 1, XDIM*YDIM*BPP, f); | ||
| 538 : | #else | ||
| 539 : | { | ||
| 540 : | int i; | ||
| 541 : | for (i=0; i<width*height*BPP;i+=BPP) { | ||
| 542 : | if (BPP == 1) { | ||
| 543 : | fputc(image+i, f); | ||
| 544 : | } else if (BPP == 2) { | ||
| 545 : | fputc(image+i+1, f); | ||
| 546 : | fputc(image+i+0, f); | ||
| 547 : | } else if (BPP == 3) { | ||
| 548 : | fputc(image+i+2, f); | ||
| 549 : | fputc(image+i+1, f); | ||
| 550 : | fputc(image+i+0, f); | ||
| 551 : | } else if (BPP == 4) { | ||
| 552 : | fputc(image+i+3, f); | ||
| 553 : | fputc(image+i+2, f); | ||
| 554 : | fputc(image+i+1, f); | ||
| 555 : | fputc(image+i+0, f); | ||
| 556 : | } | ||
| 557 : | } | ||
| 558 : | } | ||
| 559 : | #endif | ||
| 560 : | edgomez | 547 | |
| 561 : | edgomez | 1414 | /* Write Y and V planes for YUV formats */ |
| 562 : | if (BPP == 1) { | ||
| 563 : | int i; | ||
| 564 : | edgomez | 547 | |
| 565 : | edgomez | 1414 | /* Write the two chrominance planes */ |
| 566 : | for (i=0; i<YDIM/2; i++) { | ||
| 567 : | edgomez | 1418 | fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f); |
| 568 : | fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f); | ||
| 569 : | edgomez | 1414 | } |
| 570 : | } | ||
| 571 : | edgomez | 547 | |
| 572 : | |||
| 573 : | edgomez | 1414 | /* Close the file */ |
| 574 : | fclose(f); | ||
| 575 : | edgomez | 547 | |
| 576 : | edgomez | 1414 | return(0); |
| 577 : | chl | 376 | } |
| 578 : | |||
| 579 : | edgomez | 1433 | static int write_pnm(char *filename, unsigned char *image) |
| 580 : | edgomez | 1418 | { |
| 581 : | FILE * f; | ||
| 582 : | |||
| 583 : | f = fopen(filename, "wb"); | ||
| 584 : | if ( f == NULL) { | ||
| 585 : | return -1; | ||
| 586 : | } | ||
| 587 : | |||
| 588 : | if (BPP == 1) { | ||
| 589 : | int i; | ||
| 590 : | fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2); | ||
| 591 : | |||
| 592 : | fwrite(image, 1, XDIM*YDIM, f); | ||
| 593 : | |||
| 594 : | for (i=0; i<YDIM/2;i++) { | ||
| 595 : | fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f); | ||
| 596 : | fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f); | ||
| 597 : | } | ||
| 598 : | } else if (BPP == 3) { | ||
| 599 : | int i; | ||
| 600 : | fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM); | ||
| 601 : | for (i=0; i<XDIM*YDIM*3; i+=3) { | ||
| 602 : | #ifdef ARCH_IS_LITTLE_ENDIAN | ||
| 603 : | fputc(image[i+2], f); | ||
| 604 : | fputc(image[i+1], f); | ||
| 605 : | fputc(image[i+0], f); | ||
| 606 : | #else | ||
| 607 : | fputc(image[i+0], f); | ||
| 608 : | fputc(image[i+1], f); | ||
| 609 : | fputc(image[i+2], f); | ||
| 610 : | #endif | ||
| 611 : | } | ||
| 612 : | } | ||
| 613 : | |||
| 614 : | fclose(f); | ||
| 615 : | |||
| 616 : | return 0; | ||
| 617 : | } | ||
| 618 : | edgomez | 1433 | |
| 619 : | edgomez | 547 | /***************************************************************************** |
| 620 : | * Routines for decoding: init decoder, use, and stop decoder | ||
| 621 : | ****************************************************************************/ | ||
| 622 : | chl | 376 | |
| 623 : | edgomez | 547 | /* init decoder before first run */ |
| 624 : | static int | ||
| 625 : | dec_init(int use_assembler) | ||
| 626 : | chl | 376 | { |
| 627 : | edgomez | 1382 | int ret; |
| 628 : | chl | 376 | |
| 629 : | edgomez | 1382 | xvid_gbl_init_t xvid_gbl_init; |
| 630 : | xvid_dec_create_t xvid_dec_create; | ||
| 631 : | chl | 376 | |
| 632 : | edgomez | 1501 | /* Reset the structure with zeros */ |
| 633 : | memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init_t)); | ||
| 634 : | memset(&xvid_dec_create, 0, sizeof(xvid_dec_create_t)); | ||
| 635 : | |||
| 636 : | edgomez | 1382 | /*------------------------------------------------------------------------ |
| 637 : | * XviD core initialization | ||
| 638 : | *----------------------------------------------------------------------*/ | ||
| 639 : | |||
| 640 : | /* Version */ | ||
| 641 : | xvid_gbl_init.version = XVID_VERSION; | ||
| 642 : | |||
| 643 : | /* Assembly setting */ | ||
| 644 : | if(use_assembler) | ||
| 645 : | suxen_drol | 860 | #ifdef ARCH_IS_IA64 |
| 646 : | edgomez | 1382 | xvid_gbl_init.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64; |
| 647 : | chl | 376 | #else |
| 648 : | edgomez | 1382 | xvid_gbl_init.cpu_flags = 0; |
| 649 : | chl | 376 | #endif |
| 650 : | edgomez | 1382 | else |
| 651 : | xvid_gbl_init.cpu_flags = XVID_CPU_FORCE; | ||
| 652 : | chl | 376 | |
| 653 : | edgomez | 1382 | xvid_global(NULL, 0, &xvid_gbl_init, NULL); |
| 654 : | chl | 376 | |
| 655 : | edgomez | 1382 | /*------------------------------------------------------------------------ |
| 656 : | * XviD encoder initialization | ||
| 657 : | *----------------------------------------------------------------------*/ | ||
| 658 : | edgomez | 851 | |
| 659 : | edgomez | 1382 | /* Version */ |
| 660 : | xvid_dec_create.version = XVID_VERSION; | ||
| 661 : | chl | 376 | |
| 662 : | edgomez | 1382 | /* |
| 663 : | * Image dimensions -- set to 0, xvidcore will resize when ever it is | ||
| 664 : | * needed | ||
| 665 : | */ | ||
| 666 : | xvid_dec_create.width = 0; | ||
| 667 : | xvid_dec_create.height = 0; | ||
| 668 : | |||
| 669 : | ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL); | ||
| 670 : | |||
| 671 : | dec_handle = xvid_dec_create.handle; | ||
| 672 : | |||
| 673 : | return(ret); | ||
| 674 : | chl | 376 | } |
| 675 : | |||
| 676 : | edgomez | 547 | /* decode one frame */ |
| 677 : | static int | ||
| 678 : | dec_main(unsigned char *istream, | ||
| 679 : | edgomez | 851 | unsigned char *ostream, |
| 680 : | int istream_size, | ||
| 681 : | edgomez | 1382 | xvid_dec_stats_t *xvid_dec_stats) |
| 682 : | edgomez | 547 | { |
| 683 : | chl | 376 | |
| 684 : | edgomez | 1382 | int ret; |
| 685 : | chl | 376 | |
| 686 : | edgomez | 1382 | xvid_dec_frame_t xvid_dec_frame; |
| 687 : | chl | 376 | |
| 688 : | edgomez | 1382 | /* Set version */ |
| 689 : | xvid_dec_frame.version = XVID_VERSION; | ||
| 690 : | xvid_dec_stats->version = XVID_VERSION; | ||
| 691 : | chl | 376 | |
| 692 : | edgomez | 1382 | /* No general flags to set */ |
| 693 : | xvid_dec_frame.general = 0; | ||
| 694 : | chl | 376 | |
| 695 : | edgomez | 1382 | /* Input stream */ |
| 696 : | xvid_dec_frame.bitstream = istream; | ||
| 697 : | xvid_dec_frame.length = istream_size; | ||
| 698 : | edgomez | 851 | |
| 699 : | edgomez | 1382 | /* Output frame structure */ |
| 700 : | xvid_dec_frame.output.plane[0] = ostream; | ||
| 701 : | edgomez | 1414 | xvid_dec_frame.output.stride[0] = XDIM*BPP; |
| 702 : | xvid_dec_frame.output.csp = CSP; | ||
| 703 : | edgomez | 1382 | |
| 704 : | ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats); | ||
| 705 : | |||
| 706 : | return(ret); | ||
| 707 : | chl | 376 | } |
| 708 : | |||
| 709 : | edgomez | 547 | /* close decoder to release resources */ |
| 710 : | static int | ||
| 711 : | dec_stop() | ||
| 712 : | chl | 376 | { |
| 713 : | edgomez | 1382 | int ret; |
| 714 : | edgomez | 547 | |
| 715 : | edgomez | 1382 | ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL); |
| 716 : | chl | 376 | |
| 717 : | edgomez | 1382 | return(ret); |
| 718 : | chl | 376 | } |
| No admin address has been configured | ViewVC Help |
| Powered by ViewVC 1.0.4 |