#include "JackNetTool.h"
#include "JackError.h"

using namespace std;

namespace Jack
{
// NetMidiBuffer**********************************************************************************

	NetMidiBuffer::NetMidiBuffer ( session_params_t* params, unsigned int nports, char* net_buffer )
	{
		fNPorts = nports;
		fMaxBufsize = fNPorts * sizeof ( sample_t ) * params->fPeriodSize ;
		fMaxPcktSize = params->fMtu - sizeof ( packet_header_t );
		fBuffer = new char[fMaxBufsize];
		fPortBuffer = new JackMidiBuffer* [fNPorts];
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
			fPortBuffer[port_index] = NULL;
		fNetBuffer = net_buffer;
	}

	NetMidiBuffer::~NetMidiBuffer()
	{
		delete[] fBuffer;
		delete[] fPortBuffer;
	}

	unsigned int NetMidiBuffer::GetSize()
	{
		return fMaxBufsize;
	}

	void NetMidiBuffer::DisplayEvents()
	{
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
		{
			for ( unsigned int event = 0; event < fPortBuffer[port_index]->event_count; event++ )
				if ( fPortBuffer[port_index]->IsValid() )
					jack_info ( "port %d : midi event %u/%u -> time : %u, size : %u",
						port_index + 1, event + 1, fPortBuffer[port_index]->event_count,
							fPortBuffer[port_index]->events[event].time, fPortBuffer[port_index]->events[event].size );
		}
	}

	int NetMidiBuffer::RenderFromJackPorts()
	{
		int pos = 0;
		int copy_size;
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
		{
			copy_size = sizeof ( JackMidiBuffer ) + fPortBuffer[port_index]->event_count * sizeof ( JackMidiEvent );
			memcpy ( fBuffer + pos, fPortBuffer[port_index], copy_size );
			pos += copy_size;
			memcpy ( fBuffer + pos, fPortBuffer[port_index] + ( fPortBuffer[port_index]->buffer_size - fPortBuffer[port_index]->write_pos ),
					fPortBuffer[port_index]->write_pos );
			pos += fPortBuffer[port_index]->write_pos;
		}
		return pos;
	}

	int NetMidiBuffer::RenderToJackPorts()
	{
		int pos = 0;
		int copy_size;
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
		{
			copy_size = sizeof ( JackMidiBuffer ) + reinterpret_cast<JackMidiBuffer*>(fBuffer + pos)->event_count * sizeof ( JackMidiEvent );
			memcpy ( fPortBuffer[port_index], fBuffer + pos, copy_size );
			pos += copy_size;
			memcpy ( fPortBuffer[port_index] + ( fPortBuffer[port_index]->buffer_size - fPortBuffer[port_index]->write_pos ),
					fBuffer + pos, fPortBuffer[port_index]->write_pos );
			pos += fPortBuffer[port_index]->write_pos;
		}
		return pos;
	}

	int NetMidiBuffer::RenderFromNetwork ( unsigned int subcycle, unsigned int copy_size )
	{
		memcpy ( fBuffer + subcycle * fMaxPcktSize, fNetBuffer, copy_size );
		return copy_size;
	}

	int NetMidiBuffer::RenderToNetwork ( unsigned int subcycle, unsigned int total_size )
	{
		int size = total_size - subcycle * fMaxPcktSize;
		int copy_size = ( size <= fMaxPcktSize ) ? size : fMaxPcktSize;
		memcpy ( fNetBuffer, fBuffer + subcycle * fMaxPcktSize, copy_size );
		return copy_size;
	}

// net audio buffer *********************************************************************************

	NetAudioBuffer::NetAudioBuffer ( session_params_t* params, unsigned int nports, char* net_buffer )
	{
		fNPorts = nports;
		fPeriodSize = params->fPeriodSize;
		fSubPeriodSize = params->fFramesPerPacket;
		fSubPeriodBytesSize = fSubPeriodSize * sizeof ( sample_t );
		fPortBuffer = new sample_t* [fNPorts];
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
			fPortBuffer[port_index] = NULL;
		fNetBuffer = net_buffer;
	}

	NetAudioBuffer::~NetAudioBuffer()
	{
		delete[] fPortBuffer;
	}

	unsigned int NetAudioBuffer::GetSize()
	{
		return fNPorts * fSubPeriodBytesSize;
	}

	void NetAudioBuffer::RenderFromJackPorts ( unsigned int subcycle )
	{
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
			memcpy ( fNetBuffer + port_index * fSubPeriodBytesSize, fPortBuffer[port_index] + subcycle * fSubPeriodSize, fSubPeriodBytesSize );
	}

	void NetAudioBuffer::RenderToJackPorts ( unsigned int subcycle )
	{
		for ( int port_index = 0; port_index < fNPorts; port_index++ )
			memcpy ( fPortBuffer[port_index] + subcycle * fSubPeriodSize, fNetBuffer + port_index * fSubPeriodBytesSize, fSubPeriodBytesSize );
	}

// SessionParams ************************************************************************************

	EXPORT void SessionParamsHToN ( session_params_t* params )
	{
		params->fPacketID = htonl ( params->fPacketID );
		params->fMtu = htonl ( params->fMtu );
		params->fID = htonl ( params->fID );
		params->fSendAudioChannels = htonl ( params->fSendAudioChannels );
		params->fReturnAudioChannels = htonl ( params->fReturnAudioChannels );
		params->fSendMidiChannels = htonl ( params->fSendMidiChannels );
		params->fReturnMidiChannels = htonl ( params->fReturnMidiChannels );
		params->fSampleRate = htonl ( params->fSampleRate );
		params->fPeriodSize = htonl ( params->fPeriodSize );
		params->fFramesPerPacket = htonl ( params->fFramesPerPacket );
		params->fBitdepth = htonl ( params->fBitdepth );
	}

	EXPORT void SessionParamsNToH ( session_params_t* params )
	{
		params->fPacketID = ntohl ( params->fPacketID );
		params->fMtu = ntohl ( params->fMtu );
		params->fID = ntohl ( params->fID );
		params->fSendAudioChannels = ntohl ( params->fSendAudioChannels );
		params->fReturnAudioChannels = ntohl ( params->fReturnAudioChannels );
		params->fSendMidiChannels = ntohl ( params->fSendMidiChannels );
		params->fReturnMidiChannels = ntohl ( params->fReturnMidiChannels );
		params->fSampleRate = ntohl ( params->fSampleRate );
		params->fPeriodSize = ntohl ( params->fPeriodSize );
		params->fFramesPerPacket = ntohl ( params->fFramesPerPacket );
		params->fBitdepth = ntohl ( params->fBitdepth );
	}

	EXPORT void SessionParamsDisplay ( session_params_t* params )
	{
		jack_info ( "********************Params********************" );
		jack_info ( "Protocol revision : %c", params->fProtocolVersion );
		jack_info ( "MTU : %u", params->fMtu );
		jack_info ( "Master name : %s", params->fMasterNetName );
		jack_info ( "Slave name : %s", params->fSlaveNetName );
		jack_info ( "ID : %u", params->fID );
		jack_info ( "Send channels (audio - midi) : %d - %d", params->fSendAudioChannels, params->fSendMidiChannels );
		jack_info ( "Return channels (audio - midi) : %d - %d", params->fReturnAudioChannels, params->fReturnMidiChannels );
		jack_info ( "Sample rate : %u frames per second", params->fSampleRate );
		jack_info ( "Period size : %u frames per period", params->fPeriodSize );
		jack_info ( "Frames per packet : %u", params->fFramesPerPacket );
		jack_info ( "Packet per period : %u", params->fPeriodSize / params->fFramesPerPacket );
		jack_info ( "Bitdepth (0 for float) : %u", params->fBitdepth );
		jack_info ( "Name : %s", params->fName );
		jack_info ( "**********************************************" );
	}

	EXPORT sync_packet_type_t GetPacketType ( session_params_t* params )
	{
		switch ( params->fPacketID )
		{
			case 0:
				return SLAVE_AVAILABLE;
			case 1:
				return SLAVE_SETUP;
			case 2:
				return START_MASTER;
			case 3:
				return START_SLAVE;
			case 4:
				return KILL_MASTER;
		}
		return INVALID;
	}

	EXPORT int SetPacketType ( session_params_t* params, sync_packet_type_t packet_type )
	{
		switch ( packet_type )
		{
			case INVALID:
				return -1;
			case SLAVE_AVAILABLE:
				params->fPacketID = 0;
				break;
			case SLAVE_SETUP:
				params->fPacketID = 1;
				break;
			case START_MASTER:
				params->fPacketID = 2;
				break;
			case START_SLAVE:
				params->fPacketID = 3;
				break;
			case KILL_MASTER:
				params->fPacketID = 4;
		}
		return 0;
	}

// Packet header **********************************************************************************

	EXPORT void PacketHeaderHToN ( packet_header_t* header )
	{
		header->fID = htonl ( header->fID );
		header->fMidiDataSize = htonl ( header->fMidiDataSize );
		header->fBitdepth = htonl ( header->fBitdepth );
		header->fNMidiPckt = htonl ( header->fNMidiPckt );
		header->fCycle = ntohl ( header->fCycle );
		header->fSubCycle = htonl ( header->fSubCycle );
	}

	EXPORT void PacketHeaderNToH ( packet_header_t* header )
	{
		header->fID = ntohl ( header->fID );
		header->fMidiDataSize = ntohl ( header->fMidiDataSize );
		header->fBitdepth = ntohl ( header->fBitdepth );
		header->fNMidiPckt = ntohl ( header->fNMidiPckt );
		header->fCycle = ntohl ( header->fCycle );
		header->fSubCycle = ntohl ( header->fSubCycle );
	}

	EXPORT void PacketHeaderDisplay ( packet_header_t* header )
	{
		jack_info ( "********************Header********************" );
		jack_info ( "Data type : %c", header->fDataType );
		jack_info ( "Data stream : %c", header->fDataStream );
		jack_info ( "ID : %u", header->fID );
		jack_info ( "Cycle : %u", header->fCycle );
		jack_info ( "SubCycle : %u", header->fSubCycle );
		jack_info ( "Midi packets : %u", header->fNMidiPckt );
		jack_info ( "Midi data size : %u", header->fMidiDataSize );
		jack_info ( "Last packet : '%c'", header->fIsLastPckt );
		jack_info ( "Bitdepth : %u (0 for float)", header->fBitdepth );
		jack_info ( "**********************************************" );
	}

// Utility *******************************************************************************************************

	EXPORT unsigned int SetFramesPerPacket ( session_params_t* params )
	{
		if ( !params->fSendAudioChannels && !params->fReturnAudioChannels )
			return ( params->fFramesPerPacket = params->fPeriodSize );
		unsigned int period = ( int ) powf ( 2.f, ( int ) log2 ( ( params->fMtu - sizeof ( packet_header_t ) )
			/ ( max ( params->fReturnAudioChannels, params->fSendAudioChannels ) * sizeof ( sample_t ) ) ) );
		( period > params->fPeriodSize ) ? params->fFramesPerPacket = params->fPeriodSize : params->fFramesPerPacket = period;
		return params->fFramesPerPacket;
	}

	EXPORT unsigned int GetNMidiPckt ( session_params_t* params, unsigned int data_size )
	{
		//even if there is no midi data, jack need an empty buffer to know there is no event to read
		//99% of the cases : all data in one packet
		if ( data_size <= ( params->fMtu - sizeof ( packet_header_t ) ) )
			return 1;
		//else, get the number of needed packets (simply slice the biiig buffer)
		unsigned int npckt = data_size / ( params->fMtu - sizeof ( packet_header_t ) );
		if ( data_size % ( params->fMtu - sizeof ( packet_header_t ) ) )
			return ++npckt;
		return npckt;
	}

	EXPORT int SetRxTimeout ( int* sockfd, session_params_t* params )
	{
		int ret;
		struct timeval timeout;
		float time = static_cast<float> ( params->fFramesPerPacket ) / static_cast<float> ( params->fSampleRate );
		timeout.tv_sec = ( int ) time;
		float usec = 1.25 * ( time - timeout.tv_sec ) * 1000000;
		timeout.tv_usec = ( int ) usec;
		if ( ( ret = setsockopt ( *sockfd, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof ( timeout ) ) ) < 0 )
			return ret;
		return timeout.tv_usec;
	}

// Packet *******************************************************************************************************

	EXPORT bool IsNextPacket ( packet_header_t* previous, packet_header_t* next, unsigned int subcycles )
	{
		//ignore first cycle
		if ( previous->fCycle <= 1 )
			return true;
		//same PcktID (cycle), next SubPcktID (subcycle)
		if ( ( previous->fSubCycle < ( subcycles - 1 ) ) && ( next->fCycle == previous->fCycle ) && ( next->fSubCycle == ( previous->fSubCycle + 1 ) ) )
			return true;
		//next PcktID (cycle), SubPcktID reset to 1 (first subcyle)
		if ( ( next->fCycle == ( previous->fCycle + 1 ) ) && ( previous->fSubCycle == ( subcycles - 1 ) ) && ( next->fSubCycle == 0 ) )
			return true;
		//else, next is'nt next, return false
		return false;
	}
}