21 |
* along with this program; if not, write to the Free Software |
* along with this program; if not, write to the Free Software |
22 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 |
* |
* |
24 |
* $Id: xvid_encraw.c,v 1.11.2.27 2003-06-09 13:49:25 edgomez Exp $ |
* $Id: xvid_encraw.c,v 1.11.2.28 2003-06-25 23:23:21 edgomez Exp $ |
25 |
* |
* |
26 |
****************************************************************************/ |
****************************************************************************/ |
27 |
|
|
50 |
|
|
51 |
#include "xvid.h" |
#include "xvid.h" |
52 |
|
|
53 |
|
#undef READ_PNM |
54 |
|
|
55 |
/***************************************************************************** |
/***************************************************************************** |
56 |
* Quality presets |
* Quality presets |
150 |
static int ARG_DEBUG = 0; |
static int ARG_DEBUG = 0; |
151 |
static int ARG_VOPDEBUG = 0; |
static int ARG_VOPDEBUG = 0; |
152 |
|
|
153 |
|
#ifndef READ_PNM |
154 |
#define IMAGE_SIZE(x,y) ((x)*(y)*3/2) |
#define IMAGE_SIZE(x,y) ((x)*(y)*3/2) |
155 |
|
#else |
156 |
|
#define IMAGE_SIZE(x,y) ((x)*(y)*3) |
157 |
|
#endif |
158 |
|
|
159 |
#define MAX(A,B) ( ((A)>(B)) ? (A) : (B) ) |
#define MAX(A,B) ( ((A)>(B)) ? (A) : (B) ) |
160 |
#define SMALL_EPS (1e-10) |
#define SMALL_EPS (1e-10) |
185 |
static double msecond(); |
static double msecond(); |
186 |
|
|
187 |
/* PGM related functions */ |
/* PGM related functions */ |
188 |
|
#ifndef READ_PNM |
189 |
static int read_pgmheader(FILE * handle); |
static int read_pgmheader(FILE * handle); |
190 |
static int read_pgmdata(FILE * handle, |
static int read_pgmdata(FILE * handle, |
191 |
unsigned char *image); |
unsigned char *image); |
192 |
|
#else |
193 |
|
static int read_pnmheader(FILE * handle); |
194 |
|
static int read_pnmdata(FILE * handle, |
195 |
|
unsigned char *image); |
196 |
|
#endif |
197 |
static int read_yuvdata(FILE * handle, |
static int read_yuvdata(FILE * handle, |
198 |
unsigned char *image); |
unsigned char *image); |
199 |
|
|
384 |
} |
} |
385 |
|
|
386 |
if (ARG_INPUTTYPE) { |
if (ARG_INPUTTYPE) { |
387 |
|
#ifndef READ_PNM |
388 |
if (read_pgmheader(in_file)) { |
if (read_pgmheader(in_file)) { |
389 |
|
#else |
390 |
|
if (read_pnmheader(in_file)) { |
391 |
|
#endif |
392 |
fprintf(stderr, |
fprintf(stderr, |
393 |
"Wrong input format, I want YUV encapsulated in PGM\n"); |
"Wrong input format, I want YUV encapsulated in PGM\n"); |
394 |
return (-1); |
return (-1); |
454 |
if (!result) { |
if (!result) { |
455 |
if (ARG_INPUTTYPE) { |
if (ARG_INPUTTYPE) { |
456 |
/* read PGM data (YUV-format) */ |
/* read PGM data (YUV-format) */ |
457 |
|
#ifndef READ_PNM |
458 |
result = read_pgmdata(in_file, in_buffer); |
result = read_pgmdata(in_file, in_buffer); |
459 |
|
#else |
460 |
|
result = read_pnmdata(in_file, in_buffer); |
461 |
|
#endif |
462 |
} else { |
} else { |
463 |
/* read raw data (YUV-format) */ |
/* read raw data (YUV-format) */ |
464 |
result = read_yuvdata(in_file, in_buffer); |
result = read_yuvdata(in_file, in_buffer); |
554 |
|
|
555 |
/* Read the header if it's pgm stream */ |
/* Read the header if it's pgm stream */ |
556 |
if (!result && ARG_INPUTTYPE) |
if (!result && ARG_INPUTTYPE) |
557 |
|
#ifndef READ_PNM |
558 |
result = read_pgmheader(in_file); |
result = read_pgmheader(in_file); |
559 |
|
#else |
560 |
|
result = read_pnmheader(in_file); |
561 |
|
#endif |
562 |
} while (1); |
} while (1); |
563 |
|
|
564 |
|
|
700 |
* |
* |
701 |
*****************************************************************************/ |
*****************************************************************************/ |
702 |
|
|
703 |
|
#ifndef READ_PNM |
704 |
static int |
static int |
705 |
read_pgmheader(FILE * handle) |
read_pgmheader(FILE * handle) |
706 |
{ |
{ |
756 |
|
|
757 |
return (0); |
return (0); |
758 |
} |
} |
759 |
|
#else |
760 |
|
static int |
761 |
|
read_pnmheader(FILE * handle) |
762 |
|
{ |
763 |
|
int bytes, xsize, ysize, depth; |
764 |
|
char dummy[2]; |
765 |
|
|
766 |
|
bytes = fread(dummy, 1, 2, handle); |
767 |
|
|
768 |
|
if ((bytes < 2) || (dummy[0] != 'P') || (dummy[1] != '6')) |
769 |
|
return (1); |
770 |
|
|
771 |
|
fscanf(handle, "%d %d %d", &xsize, &ysize, &depth); |
772 |
|
if ((xsize > 1440) || (ysize > 2880) || (depth != 255)) { |
773 |
|
fprintf(stderr, "%d %d %d\n", xsize, ysize, depth); |
774 |
|
return (2); |
775 |
|
} |
776 |
|
|
777 |
|
XDIM = xsize; |
778 |
|
YDIM = ysize; |
779 |
|
|
780 |
|
return (0); |
781 |
|
} |
782 |
|
|
783 |
|
static int |
784 |
|
read_pnmdata(FILE * handle, |
785 |
|
unsigned char *image) |
786 |
|
{ |
787 |
|
int i; |
788 |
|
char dummy; |
789 |
|
|
790 |
|
/* read Y component of picture */ |
791 |
|
fread(image, 1, XDIM * YDIM * 3, handle); |
792 |
|
|
793 |
|
/* I don't know why, but this seems needed */ |
794 |
|
fread(&dummy, 1, 1, handle); |
795 |
|
|
796 |
|
return (0); |
797 |
|
} |
798 |
|
#endif |
799 |
|
|
800 |
static int |
static int |
801 |
read_yuvdata(FILE * handle, |
read_yuvdata(FILE * handle, |
1052 |
/* Initialize input image fields */ |
/* Initialize input image fields */ |
1053 |
if (image) { |
if (image) { |
1054 |
xvid_enc_frame.input.plane[0] = image; |
xvid_enc_frame.input.plane[0] = image; |
1055 |
|
#ifndef READ_PNM |
1056 |
xvid_enc_frame.input.csp = XVID_CSP_I420; |
xvid_enc_frame.input.csp = XVID_CSP_I420; |
1057 |
xvid_enc_frame.input.stride[0] = XDIM; |
xvid_enc_frame.input.stride[0] = XDIM; |
1058 |
|
#else |
1059 |
|
xvid_enc_frame.input.csp = XVID_CSP_BGR; |
1060 |
|
xvid_enc_frame.input.stride[0] = XDIM*3; |
1061 |
|
#endif |
1062 |
} else { |
} else { |
1063 |
xvid_enc_frame.input.csp = XVID_CSP_NULL; |
xvid_enc_frame.input.csp = XVID_CSP_NULL; |
1064 |
} |
} |