/*
 * libiec61883 - Linux IEEE 1394 streaming media library.
 * Copyright (C) 2004 Kristian Hogsberg, Dan Dennedy, and Dan Maas.
 * This file written by Dan Dennedy.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; 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 "iec61883.h"
#include "iec61883-private.h"
#include "tsbuffer.h"

#include <errno.h>
#include <stdio.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <assert.h>

#define MAX_PACKET_SIZE 2048 /* max 1394 iso packet size */
#define TSP_SPH_SIZE 192 /* size of transport stream packet plus source packet header */

iec61883_mpeg2_t
iec61883_mpeg2_xmit_init(raw1394handle_t handle, 
		iec61883_mpeg2_xmit_t get_data,
		void *callback_data)
{
	struct iec61883_mpeg2 *mpeg;

	assert (handle != NULL);
	mpeg = malloc(sizeof(struct iec61883_mpeg2));
	if (!mpeg) {
		errno = ENOMEM;
		return NULL;
	}

	mpeg->tsbuffer = NULL;
	mpeg->handle = handle;
	mpeg->put_data = NULL;
	mpeg->get_data = get_data;
	mpeg->callback_data = callback_data;
	mpeg->buffer_packets = 1000;
	mpeg->prebuffer_packets = 1000;
	mpeg->irq_interval = 250;
	mpeg->synch = 0;
	mpeg->speed = RAW1394_ISO_SPEED_200;

	raw1394_set_userdata (handle, mpeg);
	
	return mpeg;
}

iec61883_mpeg2_t
iec61883_mpeg2_recv_init(raw1394handle_t handle, 
		iec61883_mpeg2_recv_t put_data,
		void *callback_data)
{
	struct iec61883_mpeg2 *mpeg;

	mpeg = malloc(sizeof(struct iec61883_mpeg2));
	if (!mpeg) {
		errno = ENOMEM;
		return NULL;
	}

	mpeg->tsbuffer = NULL;
	mpeg->handle = handle;
	mpeg->put_data = put_data;
	mpeg->get_data = NULL;
	mpeg->callback_data = callback_data;
	mpeg->buffer_packets = 1000;
	mpeg->irq_interval = 250;
	mpeg->synch = 0;
	mpeg->speed = RAW1394_ISO_SPEED_200;

	raw1394_set_userdata (handle, mpeg);
	
	return mpeg;
}

static enum raw1394_iso_disposition
mpeg2_recv_handler (raw1394handle_t handle, 
		unsigned char *data,
		unsigned int len, 
		unsigned char channel,
		unsigned char tag, 
		unsigned char sy,
		unsigned int cycle, 
		unsigned int dropped)
{
	struct iec61883_mpeg2 *mpeg = raw1394_get_userdata (handle);
	enum raw1394_iso_disposition result = RAW1394_ISO_OK;
	
	/* check fields of CIP header for valid packet */
	unsigned short dbs_fn_qpc_sph = (htonl (* (unsigned long*) (data)) >> 10) & 0x3fff;
	unsigned char fmt = (htonl (* (unsigned long*) (data + 4)) >> 24) & 0x3f;
	
	assert (mpeg != NULL);
	mpeg->total_dropped += dropped;

	if (mpeg->put_data != NULL && /* only if callback registered */
		channel == mpeg->channel &&    /* only for selected channel */
		len >= TSP_SPH_SIZE + 8 &&     /* no empty packets */
		dbs_fn_qpc_sph == 0x01b1 &&    /* valid CIP header */
		fmt == 0x20 )
	{
		/* skip over CIP header and SPH */
		data += 12;

		/* write each TSP in the iso packet minus SPH */
		for (; len > IEC61883_MPEG2_TSP_SIZE; len -= TSP_SPH_SIZE, data += TSP_SPH_SIZE) {
			if (mpeg->put_data (data, IEC61883_MPEG2_TSP_SIZE, dropped, mpeg->callback_data) < 0) {
				result = RAW1394_ISO_ERROR;
				break;
			}
			dropped = 0; /* do not repeatedly report dropped */
		}
	}
	if (result == RAW1394_ISO_OK && dropped)
		result = RAW1394_ISO_DEFER;
			
	return result;
}

int
iec61883_mpeg2_recv_start(struct iec61883_mpeg2 *mpeg, int channel)
{
	int result = 0;
	
	assert (mpeg != NULL);
	result = raw1394_iso_recv_init (mpeg->handle, 
		mpeg2_recv_handler,
		mpeg->buffer_packets, 
		MAX_PACKET_SIZE + 8,
		channel,
		RAW1394_DMA_PACKET_PER_BUFFER,
		mpeg->irq_interval);
	
	if (result == 0) {
		mpeg->total_dropped = 0;
		mpeg->channel = channel;
		result = raw1394_iso_recv_start (mpeg->handle, -1, -1, 0);
	}
	return result;
}

static enum raw1394_iso_disposition
mpeg2_xmit_handler (raw1394handle_t handle, unsigned char *data,
                    unsigned int *len, unsigned char *tag,
                    unsigned char *sy, int cycle,
                    unsigned int dropped )
{
	struct iec61883_mpeg2 *mpeg = raw1394_get_userdata (handle);
	enum raw1394_iso_disposition result = RAW1394_ISO_OK;
	
	assert (mpeg != NULL);
	mpeg->total_dropped += dropped;
	
	if ( mpeg->tsbuffer != NULL ) {
		*len = tsbuffer_send_iso_cycle (mpeg->tsbuffer, data, cycle, 
			(raw1394_get_local_id (handle) & 0x3f), dropped);
		if (*len == 0)
			result = RAW1394_ISO_ERROR;
	}
	else
		result = RAW1394_ISO_ERROR;

	*tag = IEC61883_TAG_WITH_CIP;
	*sy = 0;

	return result;
}


int
iec61883_mpeg2_xmit_start (struct iec61883_mpeg2 *mpeg, int pid, int channel)
{
	int result = 0;
	
	assert (mpeg != NULL);
	if (mpeg->get_data != NULL) {
		mpeg->tsbuffer = tsbuffer_init (mpeg->get_data, mpeg->callback_data, pid);
		if (mpeg->tsbuffer != NULL) {
			if (raw1394_iso_xmit_init (mpeg->handle,
										mpeg2_xmit_handler,
										mpeg->buffer_packets,
										968,  /* max packets size  = 5 * 192 + 8 */
										channel,
										mpeg->speed,
										mpeg->irq_interval) == 0) {
				mpeg->total_dropped = 0;
				result = raw1394_iso_xmit_start (mpeg->handle, -1, mpeg->prebuffer_packets);
			} else
				result = -1;
		} else
			result = -1;
	} else
		result = -1;
	
	return result;
}

void
iec61883_mpeg2_xmit_stop (struct iec61883_mpeg2 *mpeg)
{
	assert (mpeg != NULL);
	if (mpeg->synch)
		raw1394_iso_xmit_sync (mpeg->handle);
	raw1394_iso_shutdown (mpeg->handle);
	tsbuffer_close (mpeg->tsbuffer);
	mpeg->tsbuffer = NULL;
}

void
iec61883_mpeg2_recv_stop (struct iec61883_mpeg2 *mpeg)
{
	assert (mpeg != NULL);
	if (mpeg->synch)
		raw1394_iso_recv_flush (mpeg->handle);
	raw1394_iso_shutdown (mpeg->handle);
}

void
iec61883_mpeg2_close (struct iec61883_mpeg2 *mpeg)
{
	assert (mpeg != NULL);
	if (mpeg->put_data)
		iec61883_mpeg2_recv_stop (mpeg);
	else if (mpeg->get_data)
		iec61883_mpeg2_xmit_stop (mpeg);
	free (mpeg);
}

unsigned int
iec61883_mpeg2_get_buffers(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->buffer_packets;
}

void
iec61883_mpeg2_set_buffers(iec61883_mpeg2_t mpeg2, unsigned int packets)
{
	assert (mpeg2 != NULL);
	mpeg2->buffer_packets = packets;
}

unsigned int
iec61883_mpeg2_get_prebuffers(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->prebuffer_packets;
}

void
iec61883_mpeg2_set_prebuffers(iec61883_mpeg2_t mpeg2, unsigned int packets)
{
	assert (mpeg2 != NULL);
	mpeg2->prebuffer_packets = packets;
}

unsigned int
iec61883_mpeg2_get_irq_interval(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->irq_interval;
}

void
iec61883_mpeg2_set_irq_interval(iec61883_mpeg2_t mpeg2, unsigned int packets)
{
	assert (mpeg2 != NULL);
	mpeg2->irq_interval = packets;
}

int
iec61883_mpeg2_get_synch(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->synch;
}

void
iec61883_mpeg2_set_synch(iec61883_mpeg2_t mpeg2, int synch)
{
	assert (mpeg2 != NULL);
	mpeg2->synch = synch;
}

int
iec61883_mpeg2_get_speed(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->speed;
}

void
iec61883_mpeg2_set_speed(iec61883_mpeg2_t mpeg2, int speed)
{
	assert (mpeg2 != NULL);
	mpeg2->speed = speed;
}

unsigned int
iec61883_mpeg2_get_dropped(iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->total_dropped;
}

void *
iec61883_mpeg2_get_callback_data (iec61883_mpeg2_t mpeg2)
{
	assert (mpeg2 != NULL);
	return mpeg2->callback_data;
}