/****
Multi-channel simulator: designed to mimic the acquisition of signals from a digitizer and send them out a socket using the OSC protocol. 
Creates arbitrary number of channels with arbitrary buffer size at an arbitrary (simulated) digitization rate.  The content of each channel 
is a continuous sine wave.

Note: By default, each message contains each channel's buffer, which is sent as soon as it is filled.  To bundle the messages (to send 
all channels at once per buffer), uncomment the appropriate lines.

Brian D. Allen
MIT Media Lab - Synth Neuro Group
http://www.syntheticneurobiology.org
2/26/2010

OSC code (using OSCpack) modified from Ross Bencina's example at http://www.audiomulch.com/~rossb/code/oscpack/
*****/

#include "osc/OscOutboundPacketStream.h"
#include "ip/UdpSocket.h"
#include <iostream>
#include <stdio.h>
#include <windows.h>
#include <sstream>
#include <string>
#include <vector>
#include "HighResTimer.h"
#include <conio.h>
#include <math.h>

#define ADDRESS "127.0.0.1"
#define PORT 9000

#define PI 3.14159

#define OSC_BUFFER_SIZE 5012 // This buffer size is set to accomodate 1000 floats. S smaller buffer could be used for less floats (use p.Size() to check packet size)
#define DAQ_BUFFER_SIZE 100 // Number of floats 
#define DAQ_RATE 1000.0
#define DAQ_CHANNELS 16
using namespace std;

UdpTransmitSocket transmitSocket( IpEndpointName( ADDRESS, PORT ) );
char buffer[OSC_BUFFER_SIZE];
osc::OutboundPacketStream p( buffer, OSC_BUFFER_SIZE );

Timer timer;
double t0 = timer.seconds();
double previous_buffer = t0;
double current_time;
double delta;
double loop_time;
int counter = 0;
vector <string> chan_num_str_vect;

int main(int argc, char* argv[])
{
	// create the channel names
	for (int chan=0; chan<DAQ_CHANNELS; ++chan) {
				stringstream chan_ss;
				chan_ss << chan;
				string s = "/chan";
				s = s + chan_ss.str();
				chan_num_str_vect.push_back(s);
	}
	
	cout << "Streaming data of " << DAQ_CHANNELS << " channels at " << DAQ_RATE << " Hz, " << "with buffer size of " << DAQ_BUFFER_SIZE << "." << endl;
	cout << "Close this window to stop streaming" << endl;

	while (true) {
		current_time = timer.seconds() - t0;
		delta = current_time - previous_buffer;
		if (delta >= ((double)DAQ_BUFFER_SIZE / DAQ_RATE)) {
			previous_buffer = current_time;
			//cout << current_time << endl;
			//p << osc::BeginBundleImmediate; ********* If you're bundling the messages, uncomment this
			for (int chan=0; chan<DAQ_CHANNELS; ++chan) {
				p << osc::BeginMessage( chan_num_str_vect.at(chan).c_str() );
				for (int ii=0; ii<DAQ_BUFFER_SIZE; ++ii) {
					if (ii==(DAQ_BUFFER_SIZE - 1)) {
						p << (float) sin((double) counter / (double) (DAQ_RATE * DAQ_CHANNELS) * 2.0 * PI);
						counter++;
						p << osc::EndMessage;	
						transmitSocket.Send( p.Data(), p.Size() ); //****** If you're bundling, don't transmit the packet stream yet
						p.Clear(); //****** If you're bundling, don't clear the packet stream yet
					}
					else {
						p << (float) sin((double) counter / (double) (DAQ_RATE * DAQ_CHANNELS) * 2.0 * PI);
						counter++;
					}
				}
			}
			/*p << osc::EndMessage
			<< osc::EndBundle;
			transmitSocket.Send( p.Data(), p.Size() );
			p.Clear();*///******** If you're bundling, uncomment this
		}
		loop_time = timer.seconds() - current_time;
		if (loop_time > ((double) DAQ_BUFFER_SIZE / (double) DAQ_RATE)) {
			cout << "The (simulated) acquisition + OSC send loop is not executing fast enough.  Try decreasing the DAQ_RATE or increasing the DAQ_BUFFER_SIZE" << endl; 
		}
	}
	
	return 0;
}