www.pudn.com > mpeg2dec050512.rar > store.c
/* store.c, picture output routines */
/* Copyright (C) 1996, MPEG Software Simulation Group. All Rights Reserved. */
/*
* Disclaimer of Warranty
*
* These software programs are available to the user without any license fee or
* royalty on an "as is" basis. The MPEG Software Simulation Group disclaims
* any and all warranties, whether express, implied, or statuary, including any
* implied warranties or merchantability or of fitness for a particular
* purpose. In no event shall the copyright-holder be liable for any
* incidental, punitive, or consequential damages of any kind whatsoever
* arising from the use of these programs.
*
* This disclaimer of warranty extends to the user of these programs and user's
* customers, employees, agents, transferees, successors, and assigns.
*
* The MPEG Software Simulation Group does not represent or warrant that the
* programs furnished hereunder are free of infringement of any third-party
* patents.
*
* Commercial implementations of MPEG-1 and MPEG-2 video, including shareware,
* are subject to royalty fees to patent holders. Many of these patents are
* general enough such that they are unavoidable regardless of implementation
* design.
*
*/
#include
#include
#include
#include "config.h"
#include "global.h"
/* private prototypes */
static void store_one _ANSI_ARGS_((char *outname, unsigned char *src[],
int offset, int incr, int height));
static void store_yuv _ANSI_ARGS_((char *outname, unsigned char *src[],
int offset, int incr, int height));
static void store_sif _ANSI_ARGS_((char *outname, unsigned char *src[],
int offset, int incr, int height));
static void store_ppm_tga _ANSI_ARGS_((char *outname, unsigned char *src[],
int offset, int incr, int height, int tgaflag));
static void store_yuv1 _ANSI_ARGS_((char *name, unsigned char *src,
int offset, int incr, int width, int height));
static void putbyte _ANSI_ARGS_((int c));
static void putword _ANSI_ARGS_((int w));
static void conv422to444 _ANSI_ARGS_((unsigned char *src, unsigned char *dst));
static void conv420to422 _ANSI_ARGS_((unsigned char *src, unsigned char *dst));
#define OBFRSIZE 4096
static unsigned char obfr[OBFRSIZE];
static unsigned char *optr;
static int outfile;
/*
* store a picture as either one frame or two fields
*/
void Write_Frame(src,frame)
unsigned char *src[];
int frame;
{
char outname[FILENAME_LENGTH];
if (progressive_sequence || progressive_frame || Frame_Store_Flag)
{
/* progressive */
sprintf(outname,Output_Picture_Filename,frame,'f');
store_one(outname,src,0,Coded_Picture_Width,vertical_size);
}
else
{
/* interlaced */
sprintf(outname,Output_Picture_Filename,frame,'a');
store_one(outname,src,0,Coded_Picture_Width<<1,vertical_size>>1);
sprintf(outname,Output_Picture_Filename,frame,'b');
store_one(outname,src,
Coded_Picture_Width,Coded_Picture_Width<<1,vertical_size>>1);
}
}
/*
* store one frame or one field
*/
static void store_one(outname,src,offset,incr,height)
char *outname;
unsigned char *src[];
int offset, incr, height;
{
switch (Output_Type)
{
case T_YUV:
store_yuv(outname,src,offset,incr,height);
break;
case T_SIF:
store_sif(outname,src,offset,incr,height);
break;
case T_TGA:
store_ppm_tga(outname,src,offset,incr,height,1);
break;
case T_PPM:
store_ppm_tga(outname,src,offset,incr,height,0);
break;
#ifdef DISPLAY
case T_X11:
dither(src);
break;
#endif
default:
break;
}
}
/* separate headerless files for y, u and v */
static void store_yuv(outname,src,offset,incr,height)
char *outname;
unsigned char *src[];
int offset,incr,height;
{
int hsize;
char tmpname[FILENAME_LENGTH];
hsize = horizontal_size;
sprintf(tmpname,"%s.Y",outname);
store_yuv1(tmpname,src[0],offset,incr,hsize,height);
if (chroma_format!=CHROMA444)
{
offset>>=1; incr>>=1; hsize>>=1;
}
if (chroma_format==CHROMA420)
{
height>>=1;
}
sprintf(tmpname,"%s.U",outname);
store_yuv1(tmpname,src[1],offset,incr,hsize,height);
sprintf(tmpname,"%s.V",outname);
store_yuv1(tmpname,src[2],offset,incr,hsize,height);
}
/* auxiliary routine */
static void store_yuv1(name,src,offset,incr,width,height)
char *name;
unsigned char *src;
int offset,incr,width,height;
{
int i, j;
unsigned char *p;
if (!Quiet_Flag)
fprintf(stderr,"saving %s\n",name);
if ((outfile = open(name,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
{
sprintf(Error_Text,"Couldn't create %s\n",name);
Error(Error_Text);
}
optr=obfr;
for (i=0; i>1)
*Coded_Picture_Height)))
Error("malloc failed");
if (!(v422 = (unsigned char *)malloc((Coded_Picture_Width>>1)
*Coded_Picture_Height)))
Error("malloc failed");
}
conv420to422(src[1],u422);
conv420to422(src[2],v422);
}
strcat(outname,".SIF");
if (!Quiet_Flag)
fprintf(stderr,"saving %s\n",outname);
if ((outfile = open(outname,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
{
sprintf(Error_Text,"Couldn't create %s\n",outname);
Error(Error_Text);
}
optr = obfr;
for (i=0; i>1) + (incr>>1)*i;
pv = v422 + (offset>>1) + (incr>>1)*i;
for (j=0; j>1)
*Coded_Picture_Height)))
Error("malloc failed");
if (!(v422 = (unsigned char *)malloc((Coded_Picture_Width>>1)
*Coded_Picture_Height)))
Error("malloc failed");
}
if (!(u444 = (unsigned char *)malloc(Coded_Picture_Width
*Coded_Picture_Height)))
Error("malloc failed");
if (!(v444 = (unsigned char *)malloc(Coded_Picture_Width
*Coded_Picture_Height)))
Error("malloc failed");
}
if (chroma_format==CHROMA420)
{
conv420to422(src[1],u422);
conv420to422(src[2],v422);
conv422to444(u422,u444);
conv422to444(v422,v444);
}
else
{
conv422to444(src[1],u444);
conv422to444(src[2],v444);
}
}
strcat(outname,tgaflag ? ".tga" : ".ppm");
if (!Quiet_Flag)
fprintf(stderr,"saving %s\n",outname);
if ((outfile = open(outname,O_CREAT|O_TRUNC|O_WRONLY|O_BINARY,0666))==-1)
{
sprintf(Error_Text,"Couldn't create %s\n",outname);
Error(Error_Text);
}
optr = obfr;
if (tgaflag)
{
/* TGA header */
for (i=0; i<12; i++)
putbyte(tga24[i]);
putword(horizontal_size); putword(height);
putbyte(tga24[12]); putbyte(tga24[13]);
}
else
{
/* PPM header */
sprintf(header,"P6\n%d %d\n255\n",horizontal_size,height);
for (i=0; header[i]!=0; i++)
putbyte(header[i]);
}
/* matrix coefficients */
crv = Inverse_Table_6_9[matrix_coefficients][0];
cbu = Inverse_Table_6_9[matrix_coefficients][1];
cgu = Inverse_Table_6_9[matrix_coefficients][2];
cgv = Inverse_Table_6_9[matrix_coefficients][3];
for (i=0; i>16];
g = Clip[(y - cgu*u - cgv*v + 32768)>>16];
b = Clip[(y + cbu*u + 32786)>>16];
if (tgaflag)
{
putbyte(b); putbyte(g); putbyte(r);
}
else
{
putbyte(r); putbyte(g); putbyte(b);
}
}
}
if (optr!=obfr)
write(outfile,obfr,optr-obfr);
close(outfile);
}
static void putbyte(c)
int c;
{
*optr++ = c;
if (optr == obfr+OBFRSIZE)
{
write(outfile,obfr,OBFRSIZE);
optr = obfr;
}
}
static void putword(w)
int w;
{
putbyte(w); putbyte(w>>8);
}
/* horizontal 1:2 interpolation filter */
static void conv422to444(src,dst)
unsigned char *src,*dst;
{
int i, i2, w, j, im3, im2, im1, ip1, ip2, ip3;
w = Coded_Picture_Width>>1;
if (base.MPEG2_Flag)
{
for (j=0; j>8];
}
src+= w;
dst+= Coded_Picture_Width;
}
}
else
{
for (j=0; j>8];
dst[i2+1] = Clip[(int)( 5*src[ip3]
-21*src[ip2]
+70*src[ip1]
+228*src[i]
-37*src[im1]
+11*src[im2]+128)>>8];
}
src+= w;
dst+= Coded_Picture_Width;
}
}
}
/* vertical 1:2 interpolation filter */
static void conv420to422(src,dst)
unsigned char *src,*dst;
{
int w, h, i, j, j2;
int jm6, jm5, jm4, jm3, jm2, jm1, jp1, jp2, jp3, jp4, jp5, jp6, jp7;
w = Coded_Picture_Width>>1;
h = Coded_Picture_Height>>1;
if (progressive_frame)
{
/* intra frame */
for (i=0; i>8];
dst[w*(j2+1)] = Clip[(int)( 3*src[w*jp3]
-16*src[w*jp2]
+67*src[w*jp1]
+227*src[w*j]
-32*src[w*jm1]
+7*src[w*jm2]+128)>>8];
}
src++;
dst++;
}
}
else
{
/* intra field */
for (i=0; i>8];
/* Polyphase FIR filter coefficients (*256): 11 -38 192 113 -30 8 */
/* New polyphase FIR filter coefficients (*256):7 -35 194 110 -24 4 */
dst[w*(j2+2)] = Clip[(int)( 7*src[w*jm4]
-35*src[w*jm2]
+194*src[w*j]
+110*src[w*jp2]
-24*src[w*jp4]
+4*src[w*jp6]+128)>>8];
/* bottom field */
jm5 = (j<5) ? 1 : j-5;
jm3 = (j<3) ? 1 : j-3;
jm1 = (j<1) ? 1 : j-1;
jp1 = (j>8];
dst[w*(j2+3)] = Clip[(int)( 1*src[w*jp7]
-7*src[w*jp5]
+30*src[w*jp3]
+248*src[w*jp1]
-21*src[w*jm1]
+5*src[w*jm3]+128)>>8];
}
src++;
dst++;
}
}
}