/*
 *  extract_rgb.c
 *
 *  Copyright (C) Thomas Östreich - June 2001
 *
 *  This file is part of transcode, a video stream processing tool
 *      
 *  transcode 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, or (at your option)
 *  any later version.
 *   
 *  transcode 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 GNU Make; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. 
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/stat.h>
#include <sys/mman.h>

#include "ioaux.h"
#include "avilib.h"

extern void import_exit(int ret);


/* ------------------------------------------------------------ 
 *
 * rgb extract thread
 *
 * magic: TC_MAGIC_AVI 
 *        TC_MAGIC_RAW  <-- default
 *
 * ------------------------------------------------------------*/


void extract_rgb(info_t *ipipe)
{

    avi_t *avifile=NULL;
    char *video;
    
    int key, error=0;

    long frames, bytes, n;


    /* ------------------------------------------------------------ 
     *
     * AVI
     *
     * ------------------------------------------------------------*/
    
    switch(ipipe->magic) {
	
    case TC_MAGIC_AVI:

	// scan file
	if (ipipe->nav_seek_file) {
	  if(NULL == (avifile = AVI_open_indexfd(ipipe->fd_in,0,ipipe->nav_seek_file))) {
	    AVI_print_error("AVI open");
	    import_exit(1);
	  }
	} else {
	  if(NULL == (avifile = AVI_open_fd(ipipe->fd_in,1))) {
	    AVI_print_error("AVI open");
	    import_exit(1);
	  }
	}
	
	// read video info;
	
	frames =  AVI_video_frames(avifile);
        if (ipipe->frame_limit[1] < frames)
        {
                frames=ipipe->frame_limit[1];
        }


	if(ipipe->verbose & TC_STATS) fprintf(stderr, "(%s) %ld video frames\n", __FILE__, frames);

	// allocate space, assume max buffer size
	if((video = (char *)calloc(1, SIZE_RGB_FRAME))==NULL) {
	    fprintf(stderr, "(%s) out of memory", __FILE__);
	    error=1;
	    break;
	}

        (int)AVI_set_video_position(avifile,ipipe->frame_limit[0]);
        for (n=ipipe->frame_limit[0]; n<=frames; ++n) {
	    
	    // video
	    if((bytes = AVI_read_frame(avifile, video, &key))<0) {
		error=1;
		break;
	    }
	    if(p_write(ipipe->fd_out, video, bytes)!=bytes) {
		error=1;
		break;
	    }
	}
	
	free(video);
	
	break;

	
	/* ------------------------------------------------------------ 
	 *
	 * RAW
	 *
	 * ------------------------------------------------------------*/
	
	    
    case TC_MAGIC_RAW:
    
    default:

	if(ipipe->magic == TC_MAGIC_UNKNOWN)
	    fprintf(stderr, "(%s) no file type specified, assuming %s\n", 
		    __FILE__, filetype(TC_MAGIC_RAW));
	
	error=p_readwrite(ipipe->fd_in, ipipe->fd_out);
	
	break;
    }
}		

