giuliomoro@230: /*
giuliomoro@230:  * render.cpp
giuliomoro@230:  *
giuliomoro@230:  *  Created on: Oct 24, 2014
giuliomoro@230:  *      Author: parallels
giuliomoro@230:  */
giuliomoro@230: 
giuliomoro@301: #include <Bela.h>
giuliomoro@230: #include <cmath>
giuliomoro@230: #include <Utilities.h>
giuliomoro@230: #include <I2c_Codec.h>
giuliomoro@230: #include <PRU.h>
giuliomoro@230: #include <stdio.h>
giuliomoro@230: #include "z_libpd.h"
giuliomoro@338: #include "z_queued.h"
giuliomoro@230: #include <UdpServer.h>
giuliomoro@324: #include <Midi.h>
giuliomoro@324: 
giuliomoro@230: 
giuliomoro@230: // setup() is called once before the audio rendering starts.
giuliomoro@230: // Use it to perform any initialisation and allocation which is dependent
giuliomoro@230: // on the period size or sample rate.
giuliomoro@230: //
giuliomoro@230: // userData holds an opaque pointer to a data structure that was passed
giuliomoro@230: // in from the call to initAudio().
giuliomoro@230: //
giuliomoro@230: // Return true on success; returning false halts the program.
giuliomoro@340: unsigned int gLibpdBlockSize;  //make sure this matches the one used to compile libpd
giuliomoro@230: 
giuliomoro@337: unsigned int gChannelsInUse = 10;
giuliomoro@230: int gBufLength;
giuliomoro@230: 
giuliomoro@230: float* gInBuf;
giuliomoro@230: float* gOutBuf;
giuliomoro@230: 
giuliomoro@230: void pdnoteon(int ch, int pitch, int vel) {
giuliomoro@230:   printf("noteon: %d %d %d\n", ch, pitch, vel);
giuliomoro@230: }
giuliomoro@230: 
giuliomoro@301: void Bela_printHook(const char *recv){
giuliomoro@230: 	rt_printf("%s", recv);
giuliomoro@230: }
giuliomoro@230: 
giuliomoro@230: UdpServer udpServer;
giuliomoro@230: 
giuliomoro@340: void libpdReadFilesLoop(){
giuliomoro@230:     while(!gShouldStop){
giuliomoro@337:     	// check for modified sockets/file descriptors
giuliomoro@337:     	// (libpd would normally do this every block WITHIN the audio thread)
giuliomoro@337:     	// not sure if this is thread-safe at the moment
giuliomoro@230:     	libpd_sys_microsleep(0);
giuliomoro@230:         usleep(1000);
giuliomoro@230:     }
giuliomoro@230: }
giuliomoro@340: 
giuliomoro@325: #define PARSE_MIDI
giuliomoro@340: AuxiliaryTask libpdReadFilesTask;
giuliomoro@340: AuxiliaryTask libpdProcessMessageQueueTask;
giuliomoro@340: AuxiliaryTask libpdProcessMidiQueueTask;
giuliomoro@324: Midi midi;
giuliomoro@337: 
giuliomoro@301: bool setup(BelaContext *context, void *userData)
giuliomoro@230: {
giuliomoro@324: 	midi.readFrom(0);
giuliomoro@324: 	midi.writeTo(0);
giuliomoro@325: #ifdef PARSE_MIDI
giuliomoro@324: 	midi.enableParser(true);
giuliomoro@325: #else
giuliomoro@325: 	midi.enableParser(false);
giuliomoro@325: #endif /* PARSE_MIDI */
giuliomoro@232: 	gChannelsInUse = min((int)(context->analogChannels+context->audioChannels), (int)gChannelsInUse);
giuliomoro@230: 	udpServer.bindToPort(1234);
giuliomoro@230: 
giuliomoro@340: 	gLibpdBlockSize = libpd_blocksize();
giuliomoro@337: 	// check that we are not running with a blocksize smaller than gLibPdBlockSize
giuliomoro@230: 	// it would still work, but the load would be executed unevenly between calls to render
giuliomoro@340: 	if(context->audioFrames < gLibpdBlockSize){
giuliomoro@340: 		fprintf(stderr, "Error: minimum block size must be %d\n", gLibpdBlockSize);
giuliomoro@230: 		return false;
giuliomoro@230: 	}
giuliomoro@230: 	// init pd
giuliomoro@338: 	libpd_set_queued_printhook(Bela_printHook); // set this before calling libpd_init
giuliomoro@338: 	libpd_set_queued_noteonhook(pdnoteon);
giuliomoro@340: 	//TODO: add hooks for other midi events and generate MIDI output appropriately
giuliomoro@338: 	libpd_queued_init();
giuliomoro@340: 	//TODO: ideally, we would analyse the ASCII of the patch file and find the in/outs to use
giuliomoro@230: 	libpd_init_audio(gChannelsInUse, gChannelsInUse, context->audioSampleRate);
giuliomoro@230: 
giuliomoro@230: 	libpd_start_message(1); // one entry in list
giuliomoro@230: 	libpd_add_float(1.0f);
giuliomoro@230: 	libpd_finish_message("pd", "dsp");
giuliomoro@230: 
giuliomoro@340: 	gBufLength = max(gLibpdBlockSize, context->audioFrames);
giuliomoro@230: 	unsigned int bufferSize = sizeof(float)*gChannelsInUse*gBufLength;
giuliomoro@230: 	gInBuf = (float*)malloc(bufferSize);
giuliomoro@230: 	gOutBuf = (float*)malloc(bufferSize);
giuliomoro@230: 	// no need to memset to zero
giuliomoro@230: 
giuliomoro@230: 	char file[] = "_main.pd";
giuliomoro@230: 	char folder[] = "./";
giuliomoro@230: 	// open patch       [; pd open file folder(
giuliomoro@230: 	libpd_openfile(file, folder);
giuliomoro@230:     
giuliomoro@340: 	libpdReadFilesTask = Bela_createAuxiliaryTask(libpdReadFilesLoop, 60, "libpdReadFiles");
giuliomoro@340: 	Bela_scheduleAuxiliaryTask(libpdReadFilesTask);
giuliomoro@340: 
giuliomoro@340: 	// Higher priority for the midi queue and lower priority for the message queue. Adjust to taste
giuliomoro@340: 	libpdProcessMidiQueueTask = Bela_createAuxiliaryTask(libpd_queued_receive_midi_messages, 80, "libpdProcessMidiQueue");
giuliomoro@340: 	libpdProcessMessageQueueTask = Bela_createAuxiliaryTask(libpd_queued_receive_pd_messages, 70, "libpdProcessMessageQueue");
giuliomoro@230: 	return true;
giuliomoro@230: }
giuliomoro@230: 
giuliomoro@230: // render() is called regularly at the highest priority by the audio engine.
giuliomoro@230: // Input and output are given from the audio hardware and the other
giuliomoro@230: // ADCs and DACs (if available). If only audio is available, numMatrixFrames
giuliomoro@230: // will be 0.
giuliomoro@301: BelaContext *c;
giuliomoro@301: void render(BelaContext *context, void *userData)
giuliomoro@230: {
giuliomoro@324: 	int num;
giuliomoro@341: 	// the safest thread-safe option to handle MIDI input is to process the MIDI buffer
giuliomoro@341: 	// from the audio thread.
giuliomoro@325: #ifdef PARSE_MIDI
giuliomoro@324: 	while((num = midi.getParser()->numAvailableMessages()) > 0){
giuliomoro@324: 		static MidiChannelMessage message;
giuliomoro@324: 		message = midi.getParser()->getNextChannelMessage();
giuliomoro@341: 		//message.prettyPrint(); // use this to print beautified message (channel, data bytes)
giuliomoro@324: 		switch(message.getType()){
giuliomoro@325: 			case kmmNoteOn:
giuliomoro@325: 			{
giuliomoro@324: 				int noteNumber = message.getDataByte(0);
giuliomoro@324: 				int velocity = message.getDataByte(1);
giuliomoro@324: 				int channel = message.getChannel();
giuliomoro@324: 				libpd_noteon(channel, noteNumber, velocity);
giuliomoro@325: 				break;
giuliomoro@324: 			}
giuliomoro@325: 			case kmmNoteOff:
giuliomoro@325: 			{
giuliomoro@325: 				/* PureData does not seem to handle noteoff messages as per the MIDI specs,
giuliomoro@325: 				 * so that the noteoff velocity is ignored. Here we convert them to noteon
giuliomoro@325: 				 * with a velocity of 0.
giuliomoro@325: 				 */
giuliomoro@325: 				int noteNumber = message.getDataByte(0);
giuliomoro@325: //				int velocity = message.getDataByte(1); // would be ignored by Pd
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				libpd_noteon(channel, noteNumber, 0);
giuliomoro@325: 				break;
giuliomoro@324: 			}
giuliomoro@325: 			case kmmControlChange:
giuliomoro@325: 			{
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				int controller = message.getDataByte(0);
giuliomoro@325: 				int value = message.getDataByte(1);
giuliomoro@325: 				libpd_controlchange(channel, controller, value);
giuliomoro@325: 				break;
giuliomoro@325: 			}
giuliomoro@325: 			case kmmProgramChange:
giuliomoro@325: 			{
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				int program = message.getDataByte(0);
giuliomoro@325: 				libpd_programchange(channel, program);
giuliomoro@325: 				break;
giuliomoro@325: 			}
giuliomoro@325: 			case kmmPolyphonicKeyPressure:
giuliomoro@325: 			{
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				int pitch = message.getDataByte(0);
giuliomoro@325: 				int value = message.getDataByte(1);
giuliomoro@325: 				libpd_polyaftertouch(channel, pitch, value);
giuliomoro@325: 				break;
giuliomoro@325: 			}
giuliomoro@325: 			case kmmChannelPressure:
giuliomoro@325: 			{
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				int value = message.getDataByte(0);
giuliomoro@325: 				libpd_aftertouch(channel, value);
giuliomoro@325: 				break;
giuliomoro@325: 			}
giuliomoro@325: 			case kmmPitchBend:
giuliomoro@325: 			{
giuliomoro@325: 				int channel = message.getChannel();
giuliomoro@325: 				int value =  (message.getDataByte(1) << 7)| message.getDataByte(0);
giuliomoro@325: 				libpd_pitchbend(channel, value);
giuliomoro@325: 				break;
giuliomoro@325: 			}
giuliomoro@337: 			case kmmNone:
giuliomoro@337: 			case kmmAny:
giuliomoro@337: 				break;
giuliomoro@324: 		}
giuliomoro@324: 	}
giuliomoro@325: #else
giuliomoro@325: 	int input;
giuliomoro@325: 	while((input = midi.getInput()) >= 0){
giuliomoro@325: 		libpd_midibyte(0, input);
giuliomoro@325: 	}
giuliomoro@325: #endif /* PARSE_MIDI */
giuliomoro@340: 
giuliomoro@341: 	unsigned int inW = 0;
giuliomoro@341: 	unsigned int outR = 0;
giuliomoro@230: /*
giuliomoro@230:  *	NOTE: if you are only using audio (or only analogs) and you are using interleaved buffers
giuliomoro@337:  *	 and the blocksize of Bela is the same as gLibPdBlockSize, then you probably
giuliomoro@230:  *	 do not need the for loops before and after libpd_process_float, so you can save quite some
giuliomoro@230:  *	 memory operations.
giuliomoro@230:  */
giuliomoro@337: 	static unsigned int analogChannelsInUse = min(context->analogChannels, gChannelsInUse - context->audioChannels);
giuliomoro@341: 	static unsigned int numberOfPdBlocksToProcess = gBufLength / gLibpdBlockSize;
giuliomoro@232: 	// rt_printf("channelsInUse: %d, analogChannels in Use: %d\n", gChannelsInUse, analogChannelsInUse);
giuliomoro@341: 	for(unsigned int j = 0; j < numberOfPdBlocksToProcess; ++j){
giuliomoro@341: 		int oldInW = inW;
giuliomoro@341: 		for(unsigned int n = 0; n < gLibpdBlockSize; ++n){ //pd buffers are interleaved
giuliomoro@341: 			for(unsigned int ch = 0; ch < context->audioChannels; ++ch){ //first two channels are audio
giuliomoro@341: 				gInBuf[inW++] = audioRead(context, n, ch);
giuliomoro@232: 			}
giuliomoro@341: 			// then analogs
giuliomoro@341: 			// this loop resamples by ZOH, as needed, using m
giuliomoro@341: 			if(context->analogChannels == 8 ){ //hold the value for two frames
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					gInBuf[inW++] = analogRead(context, n/2, analogCh); // n/2 wil be the same for n and n+1 when n is even
giuliomoro@341: 				}
giuliomoro@341: 			} else if(context->analogChannels == 4){ //write every frame
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					gInBuf[inW++] = analogRead(context, n, analogCh);
giuliomoro@341: 				}
giuliomoro@341: 			} else if(context->analogChannels == 2){ //drop every other frame
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					gInBuf[inW++] = analogRead(context, n*2, analogCh);
giuliomoro@341: 				}
giuliomoro@232: 			}
giuliomoro@230: 		}
giuliomoro@338: 		// TODO: see if we can rewrite libpd_process_float so that it takes a buffer
giuliomoro@340: 		// of interleaved channels of arbitrary length rather than a series of
giuliomoro@338: 		// stacked buffers of size gLibPdBlockSize as it currently does.
giuliomoro@341: 		// This would allow to use the buffers as they are rather than having to copy them
giuliomoro@341: 		libpd_process_float(1, &gInBuf[oldInW], &gOutBuf[oldInW]);
giuliomoro@341: 		for(unsigned int n = 0; n < gLibpdBlockSize; ++n){ //pd buffers are interleaved
giuliomoro@341: 			unsigned int outAudioFrame = n + j * gLibpdBlockSize;
giuliomoro@341: 			for(unsigned int ch = 0; ch < context->audioChannels; ++ch){
giuliomoro@341: 				audioWrite(context, outAudioFrame, ch, gOutBuf[outR++]);
giuliomoro@341: 			}
giuliomoro@341: 			//and analogs
giuliomoro@341: 			if(context->analogChannels == 8){
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					float analogOut = gOutBuf[outR++];
giuliomoro@341: 					if((n&1) == 0){//write every two frames
giuliomoro@341: 						analogWrite(context, outAudioFrame/2, analogCh, analogOut);
giuliomoro@341: 					} else {
giuliomoro@341: 						// discard this sample
giuliomoro@341: 					}
giuliomoro@341: 				}
giuliomoro@341: 			} else if(context->analogChannels == 4){ //write every frame
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					float analogOut = gOutBuf[outR++];
giuliomoro@341: 					analogWrite(context, outAudioFrame, analogCh, analogOut);
giuliomoro@341: 				}
giuliomoro@341: 			} else if(context->analogChannels == 2){ //write twice every frame
giuliomoro@341: 				for(unsigned int analogCh = 0; analogCh < analogChannelsInUse; ++analogCh){
giuliomoro@341: 					float analogOut = gOutBuf[outR++];
giuliomoro@341: 					analogWrite(context, 2*outAudioFrame, analogCh, analogOut);
giuliomoro@341: 					analogWrite(context, 2*outAudioFrame + 1, analogCh, analogOut);
giuliomoro@232: 				}
giuliomoro@232: 			}
giuliomoro@230: 		}
giuliomoro@230: 	}
giuliomoro@340: 	Bela_scheduleAuxiliaryTask(libpdProcessMidiQueueTask);
giuliomoro@340: 	Bela_scheduleAuxiliaryTask(libpdProcessMessageQueueTask);
giuliomoro@230: }
giuliomoro@232: 
giuliomoro@230: // cleanup() is called once at the end, after the audio has stopped.
giuliomoro@230: // Release any resources that were allocated in setup().
giuliomoro@230: 
giuliomoro@301: void cleanup(BelaContext *context, void *userData)
giuliomoro@230: {
giuliomoro@338: 	libpd_queued_release();
giuliomoro@230: 	free(gInBuf);
giuliomoro@230: 	free(gOutBuf);
giuliomoro@230: }