jedioutcast/utils/roq2/mpeg/store.c

696 lines
18 KiB
C
Raw Normal View History

2013-04-04 18:02:27 +00:00
/* 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 <io.h>
#include <stdlib.h>
#include <fcntl.h>
#include "config.h"
#include "global.h"
#include "im.h"
#include "sdsc.h"
#include "Zim.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_nxbitmap _ANSI_ARGS_((char *outname, unsigned char *src[],
int offset, int incr, int height));
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);
if (globalCurrentRoqImage) //eliminate a memory leak
{
ZimFree(globalCurrentRoqImage);
globalCurrentRoqImage = NULL;
}
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
case T_NXBitmapImageRep:
store_nxbitmap(outname,src,offset,incr,height);
break;
case T_NoWrite: //don't write anything
break;
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<height; i++)
{
p = src + offset + incr*i;
for (j=0; j<width; j++)
putbyte(*p++);
}
if (optr!=obfr)
write(outfile,obfr,optr-obfr);
close(outfile);
}
/*
* store as headerless file in U,Y,V,Y format
*/
static void store_sif (outname,src,offset,incr,height)
char *outname;
unsigned char *src[];
int offset, incr, height;
{
int i,j;
unsigned char *py, *pu, *pv;
static unsigned char *u422, *v422;
if (chroma_format==CHROMA444)
Error("4:4:4 not supported for SIF format");
if (chroma_format==CHROMA422)
{
u422 = src[1];
v422 = src[2];
}
else
{
if (!u422)
{
if (!(u422 = (unsigned char *)malloc((Coded_Picture_Width>>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<height; i++)
{
py = src[0] + offset + incr*i;
pu = u422 + (offset>>1) + (incr>>1)*i;
pv = v422 + (offset>>1) + (incr>>1)*i;
for (j=0; j<horizontal_size; j+=2)
{
putbyte(*pu++);
putbyte(*py++);
putbyte(*pv++);
putbyte(*py++);
}
}
if (optr!=obfr)
write(outfile,obfr,optr-obfr);
close(outfile);
}
/*
* store as PPM (PBMPLUS) or uncompressed Truevision TGA ('Targa') file
*/
static void store_ppm_tga(outname,src,offset,incr,height,tgaflag)
char *outname;
unsigned char *src[];
int offset, incr, height;
int tgaflag;
{
int i, j;
int y, u, v, r, g, b;
int crv, cbu, cgu, cgv;
unsigned char *py, *pu, *pv;
static unsigned char tga24[14] = {0,0,2,0,0,0,0, 0,0,0,0,0,24,32};
char header[FILENAME_LENGTH];
static unsigned char *u422, *v422, *u444, *v444;
if (chroma_format==CHROMA444)
{
u444 = src[1];
v444 = src[2];
}
else
{
if (!u444)
{
if (chroma_format==CHROMA420)
{
if (!(u422 = (unsigned char *)malloc((Coded_Picture_Width>>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<height; i++)
{
py = src[0] + offset + incr*i;
pu = u444 + offset + incr*i;
pv = v444 + offset + incr*i;
for (j=0; j<horizontal_size; j++)
{
u = *pu++ - 128;
v = *pv++ - 128;
y = 76309 * (*py++ - 16); /* (255/219)*65536 */
r = Clip[(y + crv*v + 32768)>>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);
}
/*
* store into globalCurrentRoqImage as TagTable with ImVfb entry
*/
static void store_nxbitmap(outname,src,offset,incr,height)
char *outname;
unsigned char *src[];
int offset, incr, height;
{
int i, j;
int y, u, v, r, g, b;
int crv, cbu, cgu, cgv;
unsigned char *py, *pu, *pv;
static unsigned char tga24[14] = {0,0,2,0,0,0,0, 0,0,0,0,0,24,32};
char header[FILENAME_LENGTH];
static unsigned char *u422, *v422, *u444, *v444;
ImVfb *vfb;
ImVfbPtr p;
globalCurrentRoqImage = ZimImageTagTableNew(globalRoqWidth, globalRoqHeight, IMVFBRGB);
TagEntryQValue(TagTableQDirect(globalCurrentRoqImage,"image vfb", 0), &vfb);
if (chroma_format==CHROMA444)
{
u444 = src[1];
v444 = src[2];
}
else
{
if (!u444)
{
if (chroma_format==CHROMA420)
{
if (!(u422 = (unsigned char *)malloc((Coded_Picture_Width>>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);
}
}
/* 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];
p = ImVfbQPtr(vfb, 0, 0);
for (i=0; i<height; i++)
{
py = src[0] + offset + incr*i;
pu = u444 + offset + incr*i;
pv = v444 + offset + incr*i;
for (j=0; j<horizontal_size; j++)
{
u = *pu++ - 128;
v = *pv++ - 128;
y = 76309 * (*py++ - 16); /* (255/219)*65536 */
r = Clip[(y + crv*v + 32768)>>16];
g = Clip[(y - cgu*u - cgv*v + 32768)>>16];
b = Clip[(y + cbu*u + 32786)>>16];
if ((j>=globalRoqHorizOffset) &&
(j< globalRoqHorizOffset + globalRoqWidth) &&
(i>=globalRoqVertOffset) &&
(i< globalRoqVertOffset + globalRoqHeight))
{ /*TRUNCATION TO 16 BIT!!!! Move into ParamImSeq when appropriate*/
ImVfbSRed(vfb, p, r & 0xfc); // 5 bits
ImVfbSGreen(vfb, p, g & 0xf8);// 6 bits
ImVfbSBlue(vfb, p, b & 0xfc);// 5 bits
p = ImVfbQNext(vfb, p);
}
}
}
}
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<Coded_Picture_Height; j++)
{
for (i=0; i<w; i++)
{
i2 = i<<1;
im2 = (i<2) ? 0 : i-2;
im1 = (i<1) ? 0 : i-1;
ip1 = (i<w-1) ? i+1 : w-1;
ip2 = (i<w-2) ? i+2 : w-1;
ip3 = (i<w-3) ? i+3 : w-1;
/* FIR filter coefficients (*256): 21 0 -52 0 159 256 159 0 -52 0 21 */
/* even samples (0 0 256 0 0) */
dst[i2] = src[i];
/* odd samples (21 -52 159 159 -52 21) */
dst[i2+1] = Clip[(int)(21*(src[im2]+src[ip3])
-52*(src[im1]+src[ip2])
+159*(src[i]+src[ip1])+128)>>8];
}
src+= w;
dst+= Coded_Picture_Width;
}
}
else
{
for (j=0; j<Coded_Picture_Height; j++)
{
for (i=0; i<w; i++)
{
i2 = i<<1;
im3 = (i<3) ? 0 : i-3;
im2 = (i<2) ? 0 : i-2;
im1 = (i<1) ? 0 : i-1;
ip1 = (i<w-1) ? i+1 : w-1;
ip2 = (i<w-2) ? i+2 : w-1;
ip3 = (i<w-3) ? i+3 : w-1;
/* FIR filter coefficients (*256): 5 -21 70 228 -37 11 */
dst[i2] = Clip[(int)( 5*src[im3]
-21*src[im2]
+70*src[im1]
+228*src[i]
-37*src[ip1]
+11*src[ip2]+128)>>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<w; i++)
{
for (j=0; j<h; j++)
{
j2 = j<<1;
jm3 = (j<3) ? 0 : j-3;
jm2 = (j<2) ? 0 : j-2;
jm1 = (j<1) ? 0 : j-1;
jp1 = (j<h-1) ? j+1 : h-1;
jp2 = (j<h-2) ? j+2 : h-1;
jp3 = (j<h-3) ? j+3 : h-1;
/* FIR filter coefficients (*256): 5 -21 70 228 -37 11 */
/* New FIR filter coefficients (*256): 3 -16 67 227 -32 7 */
dst[w*j2] = Clip[(int)( 3*src[w*jm3]
-16*src[w*jm2]
+67*src[w*jm1]
+227*src[w*j]
-32*src[w*jp1]
+7*src[w*jp2]+128)>>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<w; i++)
{
for (j=0; j<h; j+=2)
{
j2 = j<<1;
/* top field */
jm6 = (j<6) ? 0 : j-6;
jm4 = (j<4) ? 0 : j-4;
jm2 = (j<2) ? 0 : j-2;
jp2 = (j<h-2) ? j+2 : h-2;
jp4 = (j<h-4) ? j+4 : h-2;
jp6 = (j<h-6) ? j+6 : h-2;
/* Polyphase FIR filter coefficients (*256): 2 -10 35 242 -18 5 */
/* New polyphase FIR filter coefficients (*256): 1 -7 30 248 -21 5 */
dst[w*j2] = Clip[(int)( 1*src[w*jm6]
-7*src[w*jm4]
+30*src[w*jm2]
+248*src[w*j]
-21*src[w*jp2]
+5*src[w*jp4]+128)>>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<h-1) ? j+1 : h-1;
jp3 = (j<h-3) ? j+3 : h-1;
jp5 = (j<h-5) ? j+5 : h-1;
jp7 = (j<h-7) ? j+7 : h-1;
/* 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+1)] = Clip[(int)( 7*src[w*jp5]
-35*src[w*jp3]
+194*src[w*jp1]
+110*src[w*jm1]
-24*src[w*jm3]
+4*src[w*jm5]+128)>>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++;
}
}
}