/*
 * FreeBSD video grab interface
 * Copyright (c) 2002 Steve O'Hara-Smith
 * based on
 *           Linux video grab interface
 *           Copyright (c) 2000,2001 Gerard Lantau.
 * and
 *           simple_grab.c Copyright (c) 1999 Roger Hardiman
 *
 * GPL virus inherited:
 *
 * 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., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
#include "avformat.h"
#include <machine/ioctl_meteor.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <sys/time.h>
#include <signal.h>

typedef struct {
	int fd;
	int tuner_fd;
	int frame_format; /* see VIDEO_PALETTE_xxx */
	int width, height;
	int frame_rate;
	int frame_size;
} VideoData;

const char *video_device = "/dev/bktr0";

#define GRABBER_SETTLE_TIME 3
#define PAL 1
#define NTSC 2
/* PAL is 768 x 576. NTSC is 640 x 480 */
#define PAL_HEIGHT 576
#define NTSC_HEIGHT 480

#ifndef VIDEO_FORMAT
#define VIDEO_FORMAT NTSC
#endif

#ifndef VIDEO_INPUT
#define VIDEO_INPUT METEOR_INPUT_DEV0;
#endif

static UINT8 *video_buf;

static int signal_expected = 0;
static int unexpected_signals = 0;

static void catchsignal(int signal)
{
	if (!signal_expected) unexpected_signals++;
	signal_expected = 0;
	return;
}

static int bktr_init(AVFormatContext *s1,  AVFormatParameters *ap)
{
	VideoData *s = s1->priv_data;
	int width, height;
	int video_fd;
	int format = VIDEO_FORMAT;
	struct meteor_geomet geo;
	int c;
	struct sigaction act,old;

	memset(&act,0,sizeof(act));
	sigemptyset(&act.sa_mask);
	act.sa_handler  = catchsignal;
	sigaction(SIGUSR1,&act,&old);
	sigaction(SIGALRM,&act,&old);

	width = s->width;
	height = s->height;

	s->tuner_fd = open ("/dev/tuner0", O_RDWR);
	if (s->tuner_fd < 0) {
		perror("Warning: Tuner not opened continuing");
	}

	video_fd = open(video_device, O_RDWR);
	if (video_fd < 0) {
		perror(video_device);
		return -EIO;
	}
	s->fd=video_fd;
	geo.rows = height;
	geo.columns = width;
	geo.frames = 1;
	geo.oformat = METEOR_GEO_YUV_PACKED;

	if ((format == PAL) && (height <= (PAL_HEIGHT/2)))
		geo.oformat |= METEOR_GEO_EVEN_ONLY;
	if ((format == NTSC) && (height <= (NTSC_HEIGHT/2)))
		geo.oformat |= METEOR_GEO_EVEN_ONLY;

	if (ioctl(video_fd, METEORSETGEO, &geo) < 0) {
		perror ("METEORSETGEO");
		return -EIO;
	}

	switch (format) {
		case PAL:   c = METEOR_FMT_PAL; break;
		case NTSC:  c = METEOR_FMT_NTSC; break;
		default:    c = METEOR_FMT_PAL; break;
	}

	if (ioctl(video_fd, METEORSFMT, &c) < 0) {
		perror ("METEORSFMT");
		return -EIO;
	}

	c = VIDEO_INPUT;
	if (ioctl(video_fd, METEORSINPUT, &c) < 0) {
		perror ("METEORSINPUT");
		return -EIO;
	}
	video_buf = mmap((caddr_t)0, width*height*2, PROT_READ, MAP_SHARED,
	                 video_fd, (off_t) 0);
	if (video_buf == MAP_FAILED) {
		perror ("mmap");
		return -EIO;
	}
	c = METEOR_CAP_CONTINOUS;
	ioctl(s->fd, METEORCAPTUR, &c);
	c = SIGUSR1;
	signal_expected = 1;
	ioctl (s->fd, METEORSSIGNAL, &c);
	return 0;
}

static void bf_yuv422_to_yuv420p(UINT8 *lum, UINT8 *cb, UINT8 *cr,
                                 UINT8 *src, int width, int height)
{
	int x, y;
	UINT8 *p = src;
	for(y=0;y<height;y+=2) {
		for(x=0;x<width;x+=2) {
			lum[0] = p[1];
			cb[0] = p[0];
			lum[1] = p[3];
			cr[0] = p[2];
			p += 4;
			lum += 2;
			cb++;
			cr++;
		}
		for(x=0;x<width;x+=2) {
			lum[0] = p[1];
			lum[1] = p[3];
			p += 4;
			lum += 2;
		}
	}
}

/* note: we support only one picture read at a time */
static int grab_read_packet(AVFormatContext *s1, AVPacket *pkt)
{
	VideoData *s = s1->priv_data;
	int size, halfsize;
	sigset_t msig;
	UINT8 *lum, *cb, *cr;

	size = s->width * s->height;
	halfsize = size << 1;
	if (av_new_packet(pkt, size + halfsize) < 0)
		return -EIO;

	if (unexpected_signals > 0) {
		unexpected_signals--;
	} else {
		signal_expected = 1;
		sigemptyset (&msig);
		sigsuspend (&msig);
	}

	if (unexpected_signals & 1) {
		bzero (pkt->data,  size + halfsize);
	} else {
		lum = pkt->data;
		cb = lum + size;
		cr = cb + size/4;

		bf_yuv422_to_yuv420p (lum, cb, cr, video_buf, s->width, s->height);
	}
	return size + halfsize;
}

static int grab_read_header (AVFormatContext *s1,  AVFormatParameters *ap)
{
	VideoData *s = s1->priv_data;
	AVStream *st;
	int width, height;
	int frame_rate;

	if (!ap || ap->width <= 0 || ap->height <= 0 || ap->frame_rate <= 0)
		return -1;

	width = ap->width;
	height = ap->height;
	frame_rate = ap->frame_rate;
	st = av_new_stream(s1, 0);
	if (!st)
		return -ENOMEM;
	s1->priv_data = s;
	s1->nb_streams = 1;
	s1->streams[0] = st;

	s->width = width;
	s->height = height;
	s->frame_rate = frame_rate;
	s->frame_size = width*height*2;
	st->codec.pix_fmt = PIX_FMT_YUV420P;
	st->codec.codec_id = CODEC_ID_RAWVIDEO;
	st->codec.width = width;
	st->codec.height = height;
	st->codec.frame_rate = frame_rate;

	return bktr_init(s1, ap);
}

static int grab_read_close (AVFormatContext *s1)
{
	VideoData *s = s1->priv_data;

	int c = METEOR_CAP_STOP_CONT;
	ioctl(s->fd, METEORCAPTUR, &c);
	close(s->fd);
	close(s->tuner_fd);
	free(s);
	return 0;
}

AVInputFormat video_grab_device_format = {
	"video_grab_device",
	"video grab",
	 sizeof(VideoData),
	 NULL,
	grab_read_header,
	grab_read_packet,
	grab_read_close,
	flags: AVFMT_NOFILE,
};

int video_grab_init(void)
{
	av_register_input_format(&video_grab_device_format);
	return 0;
}