veejay/veejay-ng/libel/vj-dv.c

/* 
 * Linux VeeJay
 *
 * Copyright(C)2002-2004 Niels Elburg <nelburg@looze.net>
 *
 * 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
 */

/** \defgroup dvvideo DV Video
 *
 *
 * This module provides basic support for decoding and encoding raw DV files
*/
#include <config.h>
#include <libvjmsg/vj-common.h>
#ifdef SUPPORT_READ_DV2
#include <libdv/dv.h>
#include <veejay/defs.h>
#include <libel/vj-el.h>
#include <stdint.h>
#include <libel/vj-dv.h>
#include <libel/vj-avcodec.h>
#include <string.h>


#define NTSC_W 720
#define NTSC_H 480
#define PAL_W 720
#define PAL_H 576
#define DV_PAL_SIZE 144000
#define DV_NTSC_SIZE 120000
#define DV_AUDIO_MAX_SAMPLES 1944

#ifdef STRICT_CHECKING
#include <assert.h>
#endif

int	is_dv_resolution(int w, int h)
{
	if( h == NTSC_H && w == NTSC_W )
		return 1;
	if( h == PAL_H && w == PAL_W )
		return 1;
	return 0;
}

/* init the dv decoder and decode buffer*/
vj_dv_decoder *vj_dv_decoder_init(int quality, int width, int height, int pixel_format)
{
	int dv_q = DV_QUALITY_COLOR;
	vj_dv_decoder *d = (vj_dv_decoder*)vj_malloc(sizeof(vj_dv_decoder));
	if(!d) return NULL;
	d->decoder = dv_decoder_new( 1,1,0 );
	if( quality == 0 )
		dv_q = DV_QUALITY_FASTEST;
	if( quality == 1 )
		dv_q = DV_QUALITY_BEST;

	d->decoder->quality = dv_q;
	d->dv_video = (uint8_t*) vj_malloc(sizeof(uint8_t) * width * height * 4);
    	memset( d->dv_video, 0, (width*height*4));
	d->fmt = pixel_format;
	d->audio = 0; // audio off
	return d;
}

int	vj_dv_scan_frame( vj_dv_decoder *d, uint8_t * input_buf )
{
	if (dv_parse_header(d->decoder, input_buf) < 0)
	{
		veejay_msg(0, "Unable to read DV header");
		return -1;
	}
	if( d->decoder->system == e_dv_system_none )
	{
		veejay_msg(0, "No valid PAL or NTSC video frame detected");
		return -1;
	}
	if (d->decoder->sampling == e_dv_sample_411 || d->decoder->sampling == e_dv_sample_422)
		return 1;
	if( d->decoder->sampling == e_dv_sample_420 )
		return 0;
	return -1;
}
/* init the dv encoder and encode buffer */
vj_dv_encoder *vj_dv_init_encoder(int w, int h, int norm, int pixel_format)
{
	vj_dv_encoder *e = (vj_dv_encoder*) vj_malloc(sizeof(vj_dv_encoder));
	if(!e) return NULL;
	e->encoder = dv_encoder_new(0,0,0);
	e->encoder->isPAL = (norm == 0 ? 1: 0);
	e->encoder->is16x9 = (((float)w / h >= 1.777) ? 1: 0);
	e->encoder->vlc_encode_passes = 3;
    	e->encoder->static_qno = 0;
    	e->encoder->force_dct = DV_DCT_AUTO;
    	e->fmt = pixel_format;

    	e->dv_video =
	(uint8_t *) vj_malloc(sizeof(uint8_t) * 
			   (e->encoder->isPAL ?
			    	DV_PAL_SIZE : DV_NTSC_SIZE));
	memset( e->dv_video, 0 ,
		(e->encoder->isPAL ? DV_PAL_SIZE: DV_NTSC_SIZE ) );
	return e;
}


/* encode frame to dv format, dv frame will be in output_buf */
int vj_dv_encode_frame(vj_dv_encoder *encoder, uint8_t *input_buf[3], uint8_t *output_buf)
{

    time_t now = time(NULL);
    uint8_t *pixels[3];
    int w=0; int h = 0;
    if (!input_buf)
		return 0;

    pixels[0] = (uint8_t *) encoder->dv_video;

    if (encoder->encoder->isPAL)
    {
		h = PAL_H;
		w = PAL_W;
    }
    else
    {
		h = NTSC_H;
		w = NTSC_W;
    }

	if( encoder->fmt == FMT_420)
	{	
		pixels[1] = (uint8_t *) encoder->dv_video + (w * h);
		pixels[2] = (uint8_t *) encoder->dv_video + (w * h * 5) / 4;
    	yuv420p_to_yuv422(input_buf, encoder->dv_video, w, h );	
	}
    else
    {  // convert 422 planar to packed
		int off = w * h / 2;
		pixels[1] = (uint8_t *) encoder->dv_video + (w * h );
		pixels[2] = (uint8_t *) encoder->dv_video + (w * h) + off;
		yuv422p_to_yuv422(input_buf,encoder->dv_video,w,h);
    }	
  
    dv_encode_full_frame( encoder->encoder, pixels, e_dv_color_yuv,
			 output_buf);
    dv_encode_metadata(output_buf, encoder->encoder->isPAL,
		       encoder->encoder->is16x9, &now, 0);
    dv_encode_timecode(output_buf, encoder->encoder->isPAL, 0);

    if(encoder->encoder->isPAL) return DV_PAL_SIZE;
    return DV_NTSC_SIZE;
}

void vj_dv_free_encoder(vj_dv_encoder *e)
{
	if(e)
	{
		if(e->encoder)
			dv_encoder_free( e->encoder);
		if(e->dv_video)
			free(e->dv_video);
		free(e);
	}
}

void vj_dv_free_decoder(vj_dv_decoder *d) {
	if(d->decoder)
		dv_decoder_free( d->decoder );
	if(d->dv_video)
		free(d->dv_video);
	if(d) 
		free(d);
}

void	vj_dv_decoder_set_audio(vj_dv_decoder *d, int audio)
{
}

void	   vj_dv_decoder_get_audio(vj_dv_decoder *d, uint8_t *audio_buf)
{

	if(!d->audio) return;

	int n_samples = dv_get_num_samples( d->decoder);
	int channels  = dv_get_num_channels( d->decoder );
	int i,j;
	int16_t *ch0  = d->audio_buffers[0];
	int16_t *ch1  = d->audio_buffers[1];
	// convert short to uint8_t, 
	// interleave audio into single buffer
	for(i = 0; i < n_samples; i ++ )
	{
		*(audio_buf)   = ch0[i] & 0xff;			//lo
		*(audio_buf+1) = (ch0[i] >> 8) & 0xff; 		//hi
		*(audio_buf+2) = ch1[i] & 0xff;			//lo
		*(audio_buf+3) = (ch1[i] >> 8) & 0xff;		//hi
	}


}

int vj_dv_decode_frame(vj_dv_decoder *d, uint8_t * input_buf, uint8_t * Y,
		       uint8_t * Cb, uint8_t * Cr, int width, int height, int fmt)
{

    int pitches[3];

    if (!input_buf)
		return 0;

    if (dv_parse_header(d->decoder, input_buf) < 0)
	{
		return 0;
	}
    if (!((d->decoder->num_dif_seqs == 10)
	  || (d->decoder->num_dif_seqs == 12)))
	return 0;

    /*
       switch(vj_dv_decoder->sampling) {
       case e_dv_sample_411:fprintf(stderr,"sample 411\n"); break;
       case e_dv_sample_422:fprintf(stderr,"sample 422\n");break;
       case e_dv_sample_420:fprintf(stderr,"sample 420\n");break;
       default: fprintf(stderr,"unknown\n");break;
       }
     */
/*
    if (d->decoder->sampling == e_dv_sample_411 ||
		d->decoder->sampling == e_dv_sample_422 ||
		d->decoder->sampling == e_dv_sample_420)
	{
		pitches[0] = width * 2;
		pitches[1] = 0;
		pitches[2] = 0;
		uint8_t *pixels[3] = { Y , Cb, Cr };

		dv_decode_full_frame(d->decoder, input_buf,
				     e_dv_color_yuv, pixels, pitches);

		if(fmt == FMT_420)
			yuy2toyv12(Y, Cb, Cr, d->dv_video, width, height);
		else
			yuy2toyv16(Y, Cb, Cr, d->dv_video, width, height);

		return 1;
    }
*/
	pitches[0] = width * 2;
	pitches[1] = 0;
	pitches[2] = 0;


	if( d->decoder->sampling == e_dv_sample_420 )
	{
		uint8_t *pixels[3];
		pixels[0] = d->dv_video;
		pixels[1] = d->dv_video + (width * height);
		pixels[2] = d->dv_video + (width * height * 5)/4;
		dv_decode_full_frame( d->decoder, input_buf, e_dv_color_yuv,
				pixels,pitches);
		if(fmt==FMT_422)
			yuy2toyv16( Y,Cb,Cr, d->dv_video, width ,height );
		else
			yuy2toyv12( Y,Cb,Cr, d->dv_video, width, height );

		return 1;
	}

	if( d->decoder->sampling == e_dv_sample_422 )
	{	
		uint8_t *pixels[3];
		pixels[0] = d->dv_video;
		pixels[1] = d->dv_video + (width * height);
		pixels[2] = d->dv_video + (width * height) + (width * height/2);
		dv_decode_full_frame( d->decoder, input_buf, e_dv_color_yuv,
				pixels,pitches);

		if(fmt==FMT_422)
			yuy2toyv16( Y,Cb,Cr, d->dv_video, width ,height );
		else
			yuy2toyv12( Y,Cb,Cr, d->dv_video, width, height );
		return 1;
	}

    return 0;
}

#endif