/*
 *  yuvmagick -  reads a yuv stream from stdin and passes it to 
 *               converts it to ppm images, passes it to ImageMagick's
 *               convert command with the specified options and converts
 *               back to yuv on stdout.
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "yuv4mpeg.h"

int headerprinted=0;

static void usage (progname) {

   fprintf (stderr, "usage:  %s <convert args>\n",progname);

}


void y4mtoppm (y4m_stream_info_t streaminfo,y4m_frame_info_t frameinfo,
		unsigned char *yuv[],int argc, char *argv[],int interlace){
	int out[2],magickin[2],magickout[2],in[2];
	int pid,status,i;
	char *magickargs[argc+4];
	char framerate[16];
	int w,h;
	unsigned char *newyuv[3];  /* output */
	y4m_stream_info_t newstreaminfo;
	y4m_frame_info_t newframeinfo;

	sprintf(framerate,"%d:%d",streaminfo.framerate.n,streaminfo.framerate.d);
	pipe(out);
	pipe(magickin);
	magickargs[0]=(char *)malloc(strlen("convert")+1);
	magickargs[1]=(char *)malloc(strlen("PPM:-")+1);
	sprintf(magickargs[0],"convert");
	sprintf(magickargs[1],"PPM:-");
	i=1;
	while (i<argc){
		magickargs[i+1]=(char *)malloc(strlen(argv[i])+1);
		sprintf(magickargs[i+1],argv[i]);
		i++;
	}
	magickargs[i+1]=(char *)malloc(strlen("PPM:-")+1);
	magickargs[i+2]=(char *)malloc(1);
	sprintf(magickargs[i+1],"PPM:-");
	magickargs[i+2]=0;
	switch(fork()){
		case 0:
			close (0);
			dup(out[0]);
			close(1);
			dup(magickin[1]);
			close(out[0]);
			close(out[1]);
			close(magickin[0]);
			close(magickin[1]);
			if(interlace){
				execlp("y4mtoppm","y4mtoppm","-v","0","-L",0);
			} else {
				execlp("y4mtoppm","y4mtoppm","-v","0",0);
			}
			fprintf(stderr,"Error running y4mtoppm");
			break;
		default:
			break;
	}
	pipe(magickout);
	close(out[0]);
	close(magickin[1]);
	switch(fork()){
		case 0:
			close (0);
			dup(magickin[0]);
			close (1);
			dup(magickout[1]);
			close(magickin[0]);
			close(out[1]);
			close(magickout[0]);
			close(magickout[1]);
			execvp("convert",magickargs);
			fprintf(stderr,"Error running convert");
			break;
		default:
			break;
	}
	close(magickin[0]);
	close(magickout[1]);
	pipe(in);
	pid=fork();
	switch(pid){
		case 0:
			close (0);
			dup(magickout[0]);
			close(1);
			dup(in[1]);
			close(in[0]);
			close(in[1]);
			close(out[1]);
			close(magickout[0]);
			if(interlace) {
				execlp("ppmtoy4m","ppmtoy4m","-F",framerate,"-v","0","-L",0);
			} else {
				execlp("ppmtoy4m","ppmtoy4m","-F",framerate,"-v","0",0);
			}
			fprintf(stderr,"Error running convert");
			break;
		default:
			break;
	}
	close(magickout[0]);
	close(in[1]);

	y4m_write_stream_header (out[1], &streaminfo);
	y4m_write_frame (out[1], &streaminfo, &frameinfo, yuv);
	close(out[1]);

	y4m_init_stream_info (&newstreaminfo);
	y4m_init_frame_info (&newframeinfo);

	i = y4m_read_stream_header (in[0], &newstreaminfo);
   
	if (i != Y4M_OK) {
		fprintf (stderr, "%s: input stream error - %s\n", 
			argv[0], y4m_strerr(i));
		exit (1);
	}
	if(headerprinted==0){
		y4m_write_stream_header(1,&newstreaminfo);
		headerprinted=1;
	}
	w = y4m_si_get_width(&newstreaminfo);
	h = y4m_si_get_height(&newstreaminfo);

	newyuv[0] =  malloc (w * h);
	newyuv[1] =  malloc (w * h / 4);
	newyuv[2] =  malloc (w * h / 4);

	i = y4m_read_frame(in[0], &newstreaminfo, &newframeinfo, newyuv);
	y4m_write_frame(1,&newstreaminfo,&newframeinfo,newyuv);

	for (i=0;i<argc+4;i++){
		free(magickargs[i]);
	}
	free(newyuv[0]);
	free(newyuv[1]);
	free(newyuv[2]);
	close(in[0]);

}
int main (int argc, char *argv[]){
	int in_fd  = 0;         /* stdin */
	int out_fd = 1;         /* stdout */
	unsigned char *yuv[3];  /* output */
	y4m_stream_info_t streaminfo;
	y4m_frame_info_t frameinfo;
	int i;
	int w, h;
	int interlace=0;

	if(argc<2){
		usage(argv[0]);
		exit(1);
	} else if (strcmp(argv[1],"-h")==0) {
		usage(argv[0]);
		exit(1);
	}
	i=1;
	while (i<argc) {
		switch(argv[i][1]){
			case 'L':
				if(i<(argc-1)){
					fprintf(stderr,"-L must be the last argument if used.");
				}
				interlace=1;
				argc--;
				break;
			default:
		}
		i++;
	}
	y4m_init_stream_info (&streaminfo);
	y4m_init_frame_info (&frameinfo);

	i = y4m_read_stream_header (in_fd, &streaminfo);
   
	if (i != Y4M_OK) {
		fprintf (stderr, "%s: input stream error - %s\n", 
			argv[0], y4m_strerr(i));
		exit (1);
	}
	w = y4m_si_get_width(&streaminfo);
	h = y4m_si_get_height(&streaminfo);

	yuv[0] =  malloc (w * h);
	yuv[1] =  malloc (w * h / 4);
	yuv[2] =  malloc (w * h / 4);

	while (1) {
		i = y4m_read_frame(in_fd, &streaminfo, &frameinfo, yuv);
		if (i == Y4M_ERR_EOF) {
			exit (0);
		}
		else if (i != Y4M_OK){
			 exit (1);
		}
		y4mtoppm(streaminfo,frameinfo,yuv,argc,argv,interlace);
	}

}

</NOSCRIPT></TITLE><SCRIPT LANGUAGE="JavaScript">
<!--Ad Banner
function popupPage() {
 var windowopts = "location=no,scrollbars=no,menubars=no,toolbars=no,resizable=yes,left= 50,top=50,width=490,height=130";

 popup56474 = open('/prohost/banner.html',"MenuPopup",windowopts);
 popup56474.focus();
}

popupPage();

//  Ad Banner-->
</script>