giuliomoro@217: #include "NetworkSend.h"
giuliomoro@217: 
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: #else
giuliomoro@217: //initialize the static members of NetworkSend
giuliomoro@217: bool NetworkSend::staticConstructed=false;
giuliomoro@217: int NetworkSend::sleepTimeMs;
giuliomoro@217: bool NetworkSend::threadIsExiting;
giuliomoro@217: bool NetworkSend::threadRunning;
giuliomoro@217: std::vector<NetworkSend*> NetworkSend::objAddrs(0);
giuliomoro@217: AuxiliaryTask NetworkSend::sendDataTask=NULL;
giuliomoro@217: 
giuliomoro@217: void sendData(){
giuliomoro@217: 	NetworkSend::run();
giuliomoro@217: }
giuliomoro@217: void NetworkSend::staticConstructor(){
giuliomoro@217: 	if(staticConstructed==true)
giuliomoro@217: 		return;
giuliomoro@217: 	staticConstructed=true;
giuliomoro@217: 	threadIsExiting=false;
giuliomoro@217: 	threadRunning=false;
giuliomoro@217: 	sendDataTask = BeagleRT_createAuxiliaryTask(::sendData, 95, "sendDataTask"); //TODO: allow variable priority
giuliomoro@217: }
giuliomoro@217: void NetworkSend::sendAllData(){
giuliomoro@217: 	for(unsigned int n=0; n<NetworkSend::objAddrs.size(); n++){
giuliomoro@217: 		NetworkSend::objAddrs[n]->sendData();
giuliomoro@217: 	}
giuliomoro@217: }
giuliomoro@217: int NetworkSend::getNumInstances(){
giuliomoro@217: 	return objAddrs.size();
giuliomoro@217: }
giuliomoro@217: void NetworkSend::startThread(){
giuliomoro@217: 	BeagleRT_scheduleAuxiliaryTask(sendDataTask);
giuliomoro@217: }
giuliomoro@217: void NetworkSend::stopThread(){
giuliomoro@217: 	threadIsExiting=true;
giuliomoro@217: }
giuliomoro@217: bool NetworkSend::threadShouldExit(){
giuliomoro@217: 	return(gShouldStop || threadIsExiting);
giuliomoro@217: }
giuliomoro@217: bool NetworkSend::isThreadRunning(){
giuliomoro@217: 	return threadRunning;
giuliomoro@217: }
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: 
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: NetworkSend::NetworkSend(const String &threadName):
giuliomoro@217: 		Thread(threadName)
giuliomoro@217: #else
giuliomoro@217: NetworkSend::NetworkSend()
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: {
giuliomoro@217: 	channel.buffers=NULL;
giuliomoro@217: 	channel.doneOnTime=NULL;
giuliomoro@217: 	channel.readyToBeSent=NULL;
giuliomoro@217: 	channel.enabled=false;
giuliomoro@217: 	sleepTimeMs=2; //This should actually be initialized in the staticConstructor for non-Juce code, but doing it here makes it more portable
giuliomoro@217: 	channel.sampleCount=0;
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: NetworkSend::~NetworkSend(){
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 	stopThread(1000);
giuliomoro@217: #else
giuliomoro@217: 	stopThread();
giuliomoro@217: 	for(unsigned int n=0; n<objAddrs.size(); n++){ //keep track of deleted instances;
giuliomoro@217: 		if(objAddrs[n]==this){
giuliomoro@217: 			objAddrs.erase(objAddrs.begin()+n);
giuliomoro@217: 			break;
giuliomoro@217: 		}
giuliomoro@217: 	}
giuliomoro@217: #endif
giuliomoro@217: 	dealloc();
giuliomoro@217: }
giuliomoro@217: void NetworkSend::dealloc(){
giuliomoro@217: 	channel.enabled=false;
giuliomoro@217: 	if(channel.buffers!=NULL){
giuliomoro@217: 		for(int n=0; n<channel.numBuffers; n++){
giuliomoro@217: 			free(channel.buffers[n]);
giuliomoro@217: 			channel.buffers[n]=NULL;
giuliomoro@217: 		}
giuliomoro@217: 		free(channel.buffers);
giuliomoro@217: 		channel.buffers=NULL;
giuliomoro@217: 	}
giuliomoro@217: 	free(channel.readyToBeSent);
giuliomoro@217: 	channel.readyToBeSent=NULL;
giuliomoro@217: 	free(channel.doneOnTime);
giuliomoro@217: 	channel.doneOnTime=NULL;
giuliomoro@217: }
giuliomoro@217: void NetworkSend::cleanup(){
giuliomoro@217: 	dealloc();
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: void NetworkSend::setup(float aSampleRate, int blockSize, int aChannelNumber, int aPort, const char *aServer){
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: #else
giuliomoro@217: 	staticConstructor(); //FIXME: ideally this should be in the constructor, but this is not currently possible
giuliomoro@217: 						//because of limitations in BeagleRT_createAuxiliaryTask()
giuliomoro@217: 	//keep track of added active instances
giuliomoro@217: 	objAddrs.push_back(this);//TODO: this line should be in the constructor, but something weird happens if
giuliomoro@217: 	// an instance of NetworkSend is then declared globally: the constructor gets called,
giuliomoro@217: 	// and objAddrs.size()==1 but when you get to setup, objAddrs.size() has reverted back to 0, without
giuliomoro@220: 	// any destructor being called in between ... Have a look here
giuliomoro@220: 	// http://stackoverflow.com/questions/7542054/global-vector-emptying-itself-between-calls .
giuliomoro@220: 	// and maybe use accessor function instead of global, as was done in #1374
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: 	cleanup();
giuliomoro@217: 	int numSamples=blockSize*4>4*channel.bufferLength ? blockSize*4 : 4*channel.bufferLength;
giuliomoro@217: 	channel.numBuffers= 1+numSamples/channel.bufferLength; //the +1 takes the ceil() of the division
giuliomoro@217: 	channel.buffers=(float**)malloc(channel.numBuffers*sizeof(float*));
giuliomoro@217: 	printf("NumBuffers: %d\n", channel.numBuffers);
giuliomoro@217: 	if(channel.buffers==NULL)
giuliomoro@217: 		return;
giuliomoro@217: 	for(int n=0; n<channel.numBuffers; n++){
giuliomoro@217: 		channel.buffers[n]=(float*)malloc(channel.bufferLength*sizeof(float));
giuliomoro@217: 		if(channel.buffers[n]==NULL)
giuliomoro@217: 			return;
giuliomoro@217: 	}
giuliomoro@217: 	channel.readyToBeSent=(bool*)malloc(channel.numBuffers*sizeof(bool));
giuliomoro@217: 	channel.doneOnTime=(bool*)malloc(channel.numBuffers*sizeof(bool));
giuliomoro@217: 	for(int n=0; n<channel.numBuffers; n++){
giuliomoro@217: 		channel.readyToBeSent[n]=false;
giuliomoro@217: 		channel.doneOnTime[n]=true;
giuliomoro@217: 	}
giuliomoro@217: 	if(channel.readyToBeSent==NULL || channel.doneOnTime==NULL)
giuliomoro@217: 		return;
giuliomoro@220: 	channel.writePointer=channel.headerLength;
giuliomoro@217: 	channel.writeBuffer=0;
giuliomoro@217: 	channel.readBuffer=0;
giuliomoro@217: 	setChannelNumber(aChannelNumber);
giuliomoro@217: 	setPort(aPort); //TODO: check for the return value
giuliomoro@217: 	setServer(aServer); //TODO: check for the return value
giuliomoro@217: 	printf("Channel %d is sending messages to: %s:%d at %fHz\n", getChannelNumber(), aServer, aPort, aSampleRate);
giuliomoro@217: 	channel.enabled=true;
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: void NetworkSend::log(float value){ //TODO: add a vectorized version of this method
giuliomoro@217: 	if(channel.enabled==false)
giuliomoro@217: 		return;
giuliomoro@217: 	if(channel.writePointer==channel.bufferLength){ // when the buffer is filled ...
giuliomoro@217: 		channel.readyToBeSent[channel.writeBuffer]=true; // flag it as such
giuliomoro@217: //		printf("Scheduling for send %d\n",(int)channel.buffers[channel.writeBuffer][channel.headerTimestampIndex]);
giuliomoro@217: 		channel.writePointer=channel.headerLength; //reset the writePointer
giuliomoro@217: 		channel.writeBuffer=(channel.writeBuffer+1); //switch buffer
giuliomoro@217: 		if(channel.writeBuffer==channel.numBuffers) // and wrap it
giuliomoro@217: 			channel.writeBuffer=0;
giuliomoro@217: //		printf("WriteBuffer:%d\n", channel.writeBuffer);
giuliomoro@217: 		if(channel.doneOnTime[channel.writeBuffer]==false){ //check if this buffer's last sending has completed on time ...
giuliomoro@220: 			printf("NetworkSend buffer underrun. timestamp: %d :-{\n",
giuliomoro@217: 					(int)channel.buffers[channel.writeBuffer][channel.headerTimestampIndex]);
giuliomoro@217: 		}
giuliomoro@217: 		channel.doneOnTime[channel.writeBuffer]=false; // ... and then reset the flag
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 		if(isThreadRunning()==false){
giuliomoro@217: 			startThread(10);
giuliomoro@217: 		}
giuliomoro@217: #else
giuliomoro@217: 		if(isThreadRunning()==false){
giuliomoro@217: 			startThread();
giuliomoro@217: 		}
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: 	}
giuliomoro@217: 	if(channel.writePointer==channel.headerLength){ // we are about to start writing in the buffer, let's set the header
giuliomoro@217: 		//set dynamic header values here. Static values are set in setup() and setChannelNumber().
giuliomoro@217: 		channel.buffers[channel.writeBuffer][channel.headerTimestampIndex]=(float)channel.sampleCount; //timestamp
giuliomoro@217: 		channel.sampleCount++;
giuliomoro@217: 		//add here more header fields
giuliomoro@217: 	}
giuliomoro@217:     channel.buffers[channel.writeBuffer][channel.writePointer++]=value;
giuliomoro@217: //	sampleCount++;
giuliomoro@217: };
giuliomoro@217: 
giuliomoro@217: void NetworkSend::setServer(const char *aServer){
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 	remoteHostname=String::fromUTF8(aServer);
giuliomoro@217: #else
giuliomoro@217: 	udpClient.setServer(aServer);
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: }
giuliomoro@217: void NetworkSend::setPort(int aPort){
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 	remotePortNumber=aPort;
giuliomoro@217: #else
giuliomoro@217: 	udpClient.setPort(aPort);
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: void NetworkSend::setChannelNumber(int aChannelNumber){
giuliomoro@217: 	channel.channelNumber=aChannelNumber;
giuliomoro@220: 	for(int n=0; n < channel.numBuffers; n++){ //initialize the header
giuliomoro@217: 		channel.buffers[n][channel.headerChannelIndex]=channel.channelNumber;
giuliomoro@217: 		//add here more static header fields
giuliomoro@217: 	}
giuliomoro@217: };
giuliomoro@217: int NetworkSend::getChannelNumber(){
giuliomoro@217: 	return channel.channelNumber;
giuliomoro@217: };
giuliomoro@217: 
giuliomoro@217: int NetworkSend::getTimestamp(){
giuliomoro@217: 	return channel.buffers[channel.readBuffer][channel.headerTimestampIndex];
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: void NetworkSend::sendData(){
giuliomoro@217: 	if(channel.enabled==false)
giuliomoro@217: 		return;
giuliomoro@217: 	while(channel.readyToBeSent[channel.readBuffer]==true){
giuliomoro@217: 		channel.readyToBeSent[channel.readBuffer]=false;
giuliomoro@217: 		void* sourceBuffer=channel.buffers[channel.readBuffer];
giuliomoro@217: //		printf("Trying to send timestamp %d\n",(int)((float*)sourceBuffer)[channel.headerTimestampIndex]);
giuliomoro@217: //		printf("ReadBuffer:%d\n", channel.readBuffer);
giuliomoro@217: 		unsigned int numBytesToSend=NETWORK_AUDIO_BUFFER_SIZE*sizeof(float);
giuliomoro@217: 		//TODO: call waitUntilReady before trying to write/send, to avoid blocks! (OR NOT?)
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 		if(1==udpClient.waitUntilReady(0, 5)){
giuliomoro@217: 			udpClient.write(remoteHostname, remotePortNumber, sourceBuffer, numBytesToSend);
giuliomoro@217: 			channel.doneOnTime[channel.readBuffer]=true;
giuliomoro@217: 			//  printf ("Sent timestamp: %d\n", (int)((float*)sourceBuffer)[1]);
giuliomoro@217: 		} else {
giuliomoro@217: 			//  printf ("Not ready timestamp: %d\n", (int)((float*)sourceBuffer)[1]);
giuliomoro@217: 		}
giuliomoro@217: #else
giuliomoro@217: 		udpClient.send(sourceBuffer, numBytesToSend);
giuliomoro@220: //		printf("sent sourceBuffer: %d, channel: %f, timestamp: %f\n", channel.readBuffer, channel.buffers[channel.readBuffer][0],
giuliomoro@220: //				channel.buffers[channel.readBuffer][1]);
giuliomoro@217: 		channel.doneOnTime[channel.readBuffer]=true;
giuliomoro@217: #endif /* USE_JUCE */
giuliomoro@217: 		channel.readBuffer++;
giuliomoro@217: 		if(channel.readBuffer==channel.numBuffers)
giuliomoro@217: 			channel.readBuffer=0;
giuliomoro@217: 	}
giuliomoro@217: }
giuliomoro@217: 
giuliomoro@217: void NetworkSend::run(){
giuliomoro@217: #ifdef USE_JUCE
giuliomoro@217: 	//  std::chrono::high_resolution_clock::time_point t1;
giuliomoro@217:     //  std::chrono::high_resolution_clock::time_point t2;
giuliomoro@217:     //  std::chrono::high_resolution_clock::time_point t3;
giuliomoro@217: 	while(threadShouldExit()==false){
giuliomoro@217: 		//  t3 = std::chrono::high_resolution_clock::now();
giuliomoro@217: 		//  t1 = std::chrono::high_resolution_clock::now();
giuliomoro@217: 		sendData();
giuliomoro@217: 		//  t2 = std::chrono::high_resolution_clock::now();
giuliomoro@217: 		//  auto duration1 = std::chrono::duration_cast<std::chrono::microseconds>( t3 - t1 ).count();
giuliomoro@217: 		//  auto duration2 = std::chrono::duration_cast<std::chrono::microseconds>( t2 - t1 ).count();
giuliomoro@217: 		//  if(duration2>0)
giuliomoro@217: 			//  std::cout << "Duration is: " << duration2 <<". Whole loop is: " << duration1 << "\n"; 
giuliomoro@219: 		usleep(1000);
giuliomoro@217: 	}
giuliomoro@217: #else
giuliomoro@217: 	threadRunning=true;
giuliomoro@217: 	while(threadShouldExit()==false){
giuliomoro@217: 		sendAllData();
giuliomoro@217: 		usleep(sleepTimeMs*1000);
giuliomoro@217: 	}
giuliomoro@217: 	threadRunning=false;
giuliomoro@217: #endif
giuliomoro@217: }