andrewm@0: /*
andrewm@0:  * PRU.cpp
andrewm@0:  *
andrewm@0:  * Code for communicating with the Programmable Realtime Unit (PRU)
andrewm@0:  * on the BeagleBone AM335x series processors. The PRU loads and runs
andrewm@0:  * a separate code image compiled from an assembly file. Here it is
andrewm@0:  * used to handle audio and SPI ADC/DAC data.
andrewm@0:  *
andrewm@0:  * This code is specific to the PRU code in the assembly file; for example,
andrewm@0:  * it uses certain GPIO resources that correspond to that image.
andrewm@0:  *
andrewm@0:  *  Created on: May 27, 2014
andrewm@0:  *      Author: andrewm
andrewm@0:  */
andrewm@0: 
andrewm@0: #include "../include/PRU.h"
andrewm@0: #include "../include/prussdrv.h"
andrewm@0: #include "../include/pruss_intc_mapping.h"
giuliomoro@19: #include "../include/digital_gpio_mapping.h"
andrewm@0: #include "../include/GPIOcontrol.h"
andrewm@0: #include "../include/render.h"
andrewm@15: #include "../include/pru_rtaudio_bin.h"
andrewm@0: 
andrewm@0: #include <iostream>
andrewm@0: #include <stdlib.h>
andrewm@0: #include <cstdio>
andrewm@0: #include <cerrno>
andrewm@0: #include <fcntl.h>
andrewm@0: #include <sys/mman.h>
giuliomoro@16: #include <unistd.h>
andrewm@0: 
andrewm@0: // Xenomai-specific includes
andrewm@0: #include <sys/mman.h>
andrewm@0: #include <native/task.h>
andrewm@0: #include <native/timer.h>
andrewm@0: #include <rtdk.h>
andrewm@0: 
andrewm@0: using namespace std;
andrewm@0: 
andrewm@0: #define PRU_MEM_MCASP_OFFSET 0x2000  // Offset within PRU-SHARED RAM
andrewm@0: #define PRU_MEM_MCASP_LENGTH 0x2000  // Length of McASP memory, in bytes
andrewm@0: #define PRU_MEM_DAC_OFFSET 0x0     // Offset within PRU0 RAM
andrewm@0: #define PRU_MEM_DAC_LENGTH 0x2000  // Length of ADC+DAC memory, in bytes
andrewm@0: #define PRU_MEM_COMM_OFFSET 0x0    // Offset within PRU-SHARED RAM
giuliomoro@19: #define PRU_MEM_DIGITAL_OFFSET 0x1000 //Offset within PRU-SHARED RAM
giuliomoro@19: #define MEM_DIGITAL_BUFFER1_OFFSET 0x400 //Start pointer to DIGITAL_BUFFER1, which is 256 words.
giuliomoro@16: 											// 256 is the maximum number of frames allowed
andrewm@0: #define PRU_SHOULD_STOP 	0
andrewm@0: #define PRU_CURRENT_BUFFER  1
andrewm@0: #define PRU_BUFFER_FRAMES   2
andrewm@0: #define PRU_SHOULD_SYNC     3
andrewm@0: #define PRU_SYNC_ADDRESS    4
andrewm@0: #define PRU_SYNC_PIN_MASK   5
andrewm@0: #define PRU_LED_ADDRESS		6
andrewm@0: #define PRU_LED_PIN_MASK	7
andrewm@0: #define PRU_FRAME_COUNT		8
andrewm@0: #define PRU_USE_SPI			9
andrewm@12: #define PRU_SPI_NUM_CHANNELS 10
giuliomoro@38: #define PRU_USE_DIGITAL    11
giuliomoro@16: 
giuliomoro@19: short int digitalPins[NUM_DIGITALS]={
giuliomoro@16: 		GPIO_NO_BIT_0,
giuliomoro@16: 		GPIO_NO_BIT_1,
giuliomoro@16: 		GPIO_NO_BIT_2,
giuliomoro@16: 		GPIO_NO_BIT_3,
giuliomoro@16: 		GPIO_NO_BIT_4,
giuliomoro@16: 		GPIO_NO_BIT_5,
giuliomoro@16: 		GPIO_NO_BIT_6,
giuliomoro@16: 		GPIO_NO_BIT_7,
giuliomoro@16: 		GPIO_NO_BIT_8,
giuliomoro@16: 		GPIO_NO_BIT_9,
giuliomoro@16: 		GPIO_NO_BIT_10,
giuliomoro@16: 		GPIO_NO_BIT_11,
giuliomoro@16: 		GPIO_NO_BIT_12,
giuliomoro@16: 		GPIO_NO_BIT_13,
giuliomoro@16: 		GPIO_NO_BIT_14,
giuliomoro@16: 		GPIO_NO_BIT_15,
giuliomoro@16: };
andrewm@0: 
andrewm@12: #define PRU_SAMPLE_INTERVAL_NS 11338	// 88200Hz per SPI sample = 11.338us
andrewm@0: 
andrewm@0: #define GPIO0_ADDRESS 		0x44E07000
andrewm@0: #define GPIO1_ADDRESS 		0x4804C000
andrewm@0: #define GPIO_SIZE			0x198
andrewm@0: #define GPIO_CLEARDATAOUT 	(0x190 / 4)
andrewm@0: #define GPIO_SETDATAOUT 	(0x194 / 4)
andrewm@0: 
andrewm@0: #define TEST_PIN_GPIO_BASE	GPIO0_ADDRESS	// Use GPIO0(31) for debugging
andrewm@0: #define TEST_PIN_MASK		(1 << 31)
andrewm@0: #define TEST_PIN2_MASK		(1 << 26)
andrewm@0: 
andrewm@0: #define USERLED3_GPIO_BASE  GPIO1_ADDRESS // GPIO1(24) is user LED 3
andrewm@0: #define USERLED3_PIN_MASK   (1 << 24)
andrewm@0: 
andrewm@0: const unsigned int PRU::kPruGPIODACSyncPin = 5;	// GPIO0(5); P9-17
andrewm@0: const unsigned int PRU::kPruGPIOADCSyncPin = 48; // GPIO1(16); P9-15
andrewm@0: 
andrewm@0: const unsigned int PRU::kPruGPIOTestPin = 60;	// GPIO1(28); P9-12
andrewm@0: const unsigned int PRU::kPruGPIOTestPin2 = 31;	// GPIO0(31); P9-13
andrewm@0: const unsigned int PRU::kPruGPIOTestPin3 = 26;	// GPIO0(26); P8-14
andrewm@0: 
andrewm@0: extern int gShouldStop;
andrewm@0: extern int gRTAudioVerbose;
andrewm@0: 
andrewm@0: // Constructor: specify a PRU number (0 or 1)
andrewm@0: PRU::PRU()
andrewm@0: : pru_number(0), running(false), spi_enabled(false), gpio_enabled(false), led_enabled(false),
andrewm@12:   gpio_test_pin_enabled(false), spi_num_channels(0), xenomai_gpio_fd(-1), xenomai_gpio(0)
andrewm@0: {
andrewm@0: 
andrewm@0: }
andrewm@0: 
andrewm@0: // Destructor
andrewm@0: PRU::~PRU()
andrewm@0: {
andrewm@0: 	if(running)
andrewm@0: 		disable();
andrewm@0: 	if(gpio_enabled)
andrewm@0: 		cleanupGPIO();
andrewm@0: 	if(xenomai_gpio_fd >= 0)
andrewm@0: 		close(xenomai_gpio_fd);
andrewm@0: }
andrewm@0: 
andrewm@0: // Prepare the GPIO pins needed for the PRU
andrewm@0: // If include_test_pin is set, the GPIO output
andrewm@0: // is also prepared for an output which can be
andrewm@0: // viewed on a scope. If include_led is set,
andrewm@0: // user LED 3 on the BBB is taken over by the PRU
andrewm@0: // to indicate activity
giuliomoro@19: int PRU::prepareGPIO(int use_spi, int use_digital, int include_test_pin, int include_led)
andrewm@0: {
andrewm@0: 	if(use_spi) {
andrewm@0: 		// Prepare DAC CS/ pin: output, high to begin
andrewm@0: 		if(gpio_export(kPruGPIODACSyncPin)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Warning: couldn't export DAC sync pin\n";
andrewm@0: 		}
andrewm@0: 		if(gpio_set_dir(kPruGPIODACSyncPin, OUTPUT_PIN)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set direction on DAC sync pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		if(gpio_set_value(kPruGPIODACSyncPin, HIGH)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set value on DAC sync pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		// Prepare ADC CS/ pin: output, high to begin
andrewm@0: 		if(gpio_export(kPruGPIOADCSyncPin)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Warning: couldn't export ADC sync pin\n";
andrewm@0: 		}
andrewm@0: 		if(gpio_set_dir(kPruGPIOADCSyncPin, OUTPUT_PIN)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set direction on ADC sync pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		if(gpio_set_value(kPruGPIOADCSyncPin, HIGH)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set value on ADC sync pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		spi_enabled = true;
andrewm@0: 	}
andrewm@0: 
giuliomoro@19: 	if(use_digital){
giuliomoro@19: 		printf("gNumDigitalChannels: %d;\n",gNumDigitalChannels);
giuliomoro@19: 		for(int i=0; i<gNumDigitalChannels; i++){
giuliomoro@19: 			if(gpio_export(digitalPins[i])) {
giuliomoro@16: 				if(gRTAudioVerbose)
giuliomoro@38: 					cerr << "Warning: couldn't export digital GPIO pin " << digitalPins[i] << "\n"; // this is left as a warning because if the pin has been exported by somebody else, can still be used
giuliomoro@16: 			}
giuliomoro@38: 			if(gpio_set_dir(digitalPins[i], INPUT_PIN)) {
giuliomoro@16: 				if(gRTAudioVerbose)
giuliomoro@38: 					cerr << "Error: Couldn't set direction on digital GPIO pin " << digitalPins[i] << "\n";
giuliomoro@16: 				return -1;
giuliomoro@16: 			}
giuliomoro@16: 		}
giuliomoro@19: 		digital_enabled=true;
giuliomoro@16: 	}
giuliomoro@16: 
andrewm@0: 	if(include_test_pin) {
andrewm@0: 		// Prepare GPIO test output (for debugging), low to begin
andrewm@0: 		if(gpio_export(kPruGPIOTestPin)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Warning: couldn't export GPIO test pin\n";
andrewm@0: 		}
andrewm@0: 		if(gpio_set_dir(kPruGPIOTestPin, OUTPUT_PIN)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set direction on GPIO test pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		if(gpio_set_value(kPruGPIOTestPin, LOW)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set value on GPIO test pin\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		if(gpio_export(kPruGPIOTestPin2)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Warning: couldn't export GPIO test pin 2\n";
andrewm@0: 		}
andrewm@0: 		if(gpio_set_dir(kPruGPIOTestPin2, OUTPUT_PIN)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set direction on GPIO test pin 2\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		if(gpio_set_value(kPruGPIOTestPin2, LOW)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set value on GPIO test pin 2\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		if(gpio_export(kPruGPIOTestPin3)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Warning: couldn't export GPIO test pin 3\n";
andrewm@0: 		}
andrewm@0: 		if(gpio_set_dir(kPruGPIOTestPin3, OUTPUT_PIN)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set direction on GPIO test pin 3\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		if(gpio_set_value(kPruGPIOTestPin3, LOW)) {
andrewm@0: 			if(gRTAudioVerbose)
andrewm@0: 				cout << "Couldn't set value on GPIO test pin 3\n";
andrewm@0: 			return -1;
andrewm@0: 		}
andrewm@0: 		gpio_test_pin_enabled = true;
andrewm@0: 	}
andrewm@0: 
andrewm@0: 	if(include_led) {
andrewm@0: 		// Turn off system function for LED3 so it can be reused by PRU
andrewm@0: 		led_set_trigger(3, "none");
andrewm@0: 		led_enabled = true;
andrewm@0: 	}
andrewm@0: 
andrewm@0: 	gpio_enabled = true;
andrewm@0: 
andrewm@0: 	return 0;
andrewm@0: }
andrewm@0: 
andrewm@0: // Clean up the GPIO at the end
andrewm@0: void PRU::cleanupGPIO()
andrewm@0: {
andrewm@0: 	if(!gpio_enabled)
andrewm@0: 		return;
andrewm@0: 	if(spi_enabled) {
andrewm@0: 		gpio_unexport(kPruGPIODACSyncPin);
andrewm@0: 		gpio_unexport(kPruGPIOADCSyncPin);
andrewm@0: 	}
giuliomoro@19: 	if(digital_enabled){
giuliomoro@19: 		for(int i=0; i<gNumDigitalChannels; i++){
giuliomoro@19: 			gpio_unexport(digitalPins[i]);
giuliomoro@16: 		}
giuliomoro@16: 	}
andrewm@0: 	if(gpio_test_pin_enabled) {
andrewm@0: 		gpio_unexport(kPruGPIOTestPin);
andrewm@0: 		gpio_unexport(kPruGPIOTestPin2);
andrewm@0: 		gpio_unexport(kPruGPIOTestPin3);
andrewm@0: 	}
andrewm@0: 	if(led_enabled) {
andrewm@0: 		// Set LED back to default eMMC status
andrewm@0: 		// TODO: make it go back to its actual value before this program,
andrewm@0: 		// rather than the system default
andrewm@0: 		led_set_trigger(3, "mmc1");
andrewm@0: 	}
andrewm@0: 	gpio_enabled = gpio_test_pin_enabled = false;
andrewm@0: }
andrewm@0: 
andrewm@0: // Initialise and open the PRU
andrewm@12: int PRU::initialise(int pru_num, int frames_per_buffer, int spi_channels, bool xenomai_test_pin)
andrewm@0: {
andrewm@0: 	uint32_t *pruMem = 0;
andrewm@0: 
andrewm@0: 	if(!gpio_enabled) {
andrewm@0: 		rt_printf("initialise() called before GPIO enabled\n");
andrewm@0: 		return 1;
andrewm@0: 	}
andrewm@0: 
andrewm@0: 	pru_number = pru_num;
andrewm@0: 
andrewm@12: 	/* Set number of SPI ADC / DAC channels to use. This implicitly
andrewm@12: 	 * also determines the sample rate relative to the audio clock
andrewm@12: 	 * (half audio clock for 8 channels, full audio clock for 4,
andrewm@12: 	 * double audio clock for 2)
andrewm@12: 	 */
andrewm@12: 	spi_num_channels = spi_channels;
andrewm@12: 
andrewm@0:     /* Initialize structure used by prussdrv_pruintc_intc   */
andrewm@0:     /* PRUSS_INTC_INITDATA is found in pruss_intc_mapping.h */
andrewm@0:     tpruss_intc_initdata pruss_intc_initdata = PRUSS_INTC_INITDATA;
andrewm@0: 
andrewm@0:     /* Allocate and initialize memory */
andrewm@0:     prussdrv_init();
andrewm@15:     if(prussdrv_open(pru_number == 0 ? PRU_EVTOUT_0 : PRU_EVTOUT_1)) {
andrewm@0:     	rt_printf("Failed to open PRU driver\n");
andrewm@0:     	return 1;
andrewm@0:     }
andrewm@0: 
andrewm@0:     /* Map PRU's INTC */
andrewm@0:     prussdrv_pruintc_init(&pruss_intc_initdata);
andrewm@0: 
andrewm@0:     spi_buffer_frames = frames_per_buffer;
andrewm@12:     audio_buffer_frames = spi_buffer_frames * spi_num_channels / 4;
giuliomoro@19:     digital_buffer_frames = audio_buffer_frames;
andrewm@0: 
andrewm@0:     /* Map PRU memory to pointers */
andrewm@0: 	prussdrv_map_prumem (PRUSS0_SHARED_DATARAM, (void **)&pruMem);
andrewm@0:     pru_buffer_comm = (uint32_t *)&pruMem[PRU_MEM_COMM_OFFSET/sizeof(uint32_t)];
andrewm@0: 	pru_buffer_audio_dac = (int16_t *)&pruMem[PRU_MEM_MCASP_OFFSET/sizeof(uint32_t)];
andrewm@0: 
andrewm@12: 	/* ADC memory starts 2(ch)*2(buffers)*bufsize samples later */
andrewm@12: 	pru_buffer_audio_adc = &pru_buffer_audio_dac[4 * audio_buffer_frames];
andrewm@0: 
andrewm@0: 	if(spi_enabled) {
andrewm@0: 		prussdrv_map_prumem (pru_number == 0 ? PRUSS0_PRU0_DATARAM : PRUSS0_PRU1_DATARAM, (void **)&pruMem);
andrewm@0: 		pru_buffer_spi_dac = (uint16_t *)&pruMem[PRU_MEM_DAC_OFFSET/sizeof(uint32_t)];
andrewm@0: 
andrewm@12: 		/* ADC memory starts after N(ch)*2(buffers)*bufsize samples */
andrewm@12: 		pru_buffer_spi_adc = &pru_buffer_spi_dac[2 * spi_num_channels * spi_buffer_frames];
andrewm@0: 	}
andrewm@0: 	else {
andrewm@0: 		pru_buffer_spi_dac = pru_buffer_spi_adc = 0;
andrewm@0: 	}
andrewm@0: 
giuliomoro@19: 	if(digital_enabled) {
giuliomoro@16: 		prussdrv_map_prumem (PRUSS0_SHARED_DATARAM, (void **)&pruMem);
giuliomoro@19: 		pru_buffer_digital = (uint32_t *)&pruMem[PRU_MEM_DIGITAL_OFFSET/sizeof(uint32_t)];
giuliomoro@16: 	}
giuliomoro@16: 	else {
giuliomoro@19: 		pru_buffer_digital = 0;
giuliomoro@16: 	}
andrewm@0:     /* Set up flags */
andrewm@0:     pru_buffer_comm[PRU_SHOULD_STOP] = 0;
andrewm@0:     pru_buffer_comm[PRU_CURRENT_BUFFER] = 0;
andrewm@0:     pru_buffer_comm[PRU_BUFFER_FRAMES] = spi_buffer_frames;
andrewm@0:     pru_buffer_comm[PRU_SHOULD_SYNC] = 0;
andrewm@0:     pru_buffer_comm[PRU_SYNC_ADDRESS] = 0;
andrewm@0:     pru_buffer_comm[PRU_SYNC_PIN_MASK] = 0;
andrewm@0:     if(led_enabled) {
andrewm@0:     	pru_buffer_comm[PRU_LED_ADDRESS] = USERLED3_GPIO_BASE;
andrewm@0:     	pru_buffer_comm[PRU_LED_PIN_MASK] = USERLED3_PIN_MASK;
andrewm@0:     }
andrewm@0:     else {
andrewm@0:     	pru_buffer_comm[PRU_LED_ADDRESS] = 0;
andrewm@0:     	pru_buffer_comm[PRU_LED_PIN_MASK] = 0;
andrewm@0:     }
andrewm@0:     if(spi_enabled) {
andrewm@0:     	pru_buffer_comm[PRU_USE_SPI] = 1;
andrewm@12:     	pru_buffer_comm[PRU_SPI_NUM_CHANNELS] = spi_num_channels;
andrewm@0:     }
andrewm@0:     else {
andrewm@0:     	pru_buffer_comm[PRU_USE_SPI] = 0;
andrewm@12:     	pru_buffer_comm[PRU_SPI_NUM_CHANNELS] = 0;
andrewm@0:     }
giuliomoro@19:     if(digital_enabled) {
giuliomoro@38:     	pru_buffer_comm[PRU_USE_DIGITAL] = 1;
giuliomoro@38: //TODO: add mask
giuliomoro@16:     }
giuliomoro@16:     else {
giuliomoro@38:     	pru_buffer_comm[PRU_USE_DIGITAL] = 0;
giuliomoro@38: 
giuliomoro@16:     }
andrewm@0: 
giuliomoro@38:     /* Clear ADC and DAC memory.*/
giuliomoro@38:     //TODO: this initialisation should only address the memory effectively used by these buffers, i.e.:depend on the number of frames
giuliomoro@38:     //  (otherwise might cause issues if we move memory locations later on)
andrewm@0:     if(spi_enabled) {
andrewm@0: 		for(int i = 0; i < PRU_MEM_DAC_LENGTH / 2; i++)
andrewm@0: 			pru_buffer_spi_dac[i] = 0;
giuliomoro@38: 		if(digital_enabled){
giuliomoro@38: 			for(int i = 0; i < PRU_MEM_DIGITAL_OFFSET*2; i++)
giuliomoro@38: 				pru_buffer_digital[i] = 0x0000ffff; // set to all inputs, to avoid unexpected spikes
giuliomoro@38: 		}
andrewm@0:     }
andrewm@0: 	for(int i = 0; i < PRU_MEM_MCASP_LENGTH / 2; i++)
andrewm@0: 		pru_buffer_audio_dac[i] = 0;
giuliomoro@16: //TODO: maybe the lines below are to be deleted, as we removed the test code from pru_rtaudio.p ?
andrewm@0: 	/* If using GPIO test pin for Xenomai (for debugging), initialise the pointer now */
andrewm@0: 	if(xenomai_test_pin && xenomai_gpio_fd < 0) {
andrewm@0: 		xenomai_gpio_fd = open("/dev/mem", O_RDWR);
andrewm@0: 		if(xenomai_gpio_fd < 0)
andrewm@0: 			rt_printf("Unable to open /dev/mem for GPIO test pin\n");
andrewm@0: 		else {
andrewm@0: 			xenomai_gpio = (uint32_t *)mmap(0, GPIO_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, xenomai_gpio_fd, TEST_PIN_GPIO_BASE);
andrewm@0: 			if(xenomai_gpio == MAP_FAILED) {
andrewm@0: 				rt_printf("Unable to map GPIO address for test pin\n");
andrewm@0: 				xenomai_gpio = 0;
andrewm@0: 				close(xenomai_gpio_fd);
andrewm@0: 				xenomai_gpio_fd = -1;
andrewm@0: 			}
andrewm@0: 		}
andrewm@0: 	}
andrewm@0: 
andrewm@0: 	return 0;
andrewm@0: }
andrewm@0: 
andrewm@0: // Run the code image in the specified file
giuliomoro@16: int PRU::start(char * const filename)
andrewm@0: {
andrewm@0: 	/* Clear any old interrupt */
andrewm@15: 	if(pru_number == 0)
andrewm@15: 		prussdrv_pru_clear_event(PRU_EVTOUT_0, PRU0_ARM_INTERRUPT);
andrewm@15: 	else
andrewm@15: 		prussdrv_pru_clear_event(PRU_EVTOUT_1, PRU1_ARM_INTERRUPT);
giuliomoro@16: 	/* Load and execute binary on PRU */
giuliomoro@16: 	if(filename[0] == '\0') { //if the string is empty, load the embedded code
giuliomoro@16: 		if(gRTAudioVerbose)
giuliomoro@16: 			rt_printf("Using embedded PRU code\n");
giuliomoro@16: 		if(prussdrv_exec_code(pru_number, PRUcode, sizeof(PRUcode))) {
giuliomoro@16: 			rt_printf("Failed to execute PRU code\n");
giuliomoro@16: 			return 1;
giuliomoro@16: 		}
giuliomoro@16: 	} else {
giuliomoro@16: 		if(gRTAudioVerbose)
giuliomoro@16: 			rt_printf("Using PRU code from %s\n",filename);
giuliomoro@16: 		if(prussdrv_exec_program(pru_number, filename)) {
giuliomoro@16: 			rt_printf("Failed to execute PRU code from %s\n", filename);
giuliomoro@16: 			return 1;
giuliomoro@16: 		}
giuliomoro@16: 	}
andrewm@0: 
andrewm@0:     running = true;
andrewm@0:     return 0;
andrewm@0: }
giuliomoro@16: uint32_t empty[1024]={0x0};
andrewm@0: 
andrewm@0: // Main loop to read and write data from/to PRU
andrewm@0: void PRU::loop()
andrewm@0: {
andrewm@0: 	// Polling interval is 1/4 of the period
andrewm@12: 	RTIME sleepTime = PRU_SAMPLE_INTERVAL_NS * (spi_num_channels / 2) * spi_buffer_frames / 4;
andrewm@0: 	float *audioInBuffer, *audioOutBuffer;
giuliomoro@23: 	float *analogInBuffer, *analogOutBuffer, *lastAnalogOutFrame;
giuliomoro@23: 	uint32_t *digitalBuffer0, *digitalBuffer1, *lastDigitalBuffer;
andrewm@0: 
andrewm@0: 	audioInBuffer = (float *)malloc(2 * audio_buffer_frames * sizeof(float));
andrewm@0: 	audioOutBuffer = (float *)malloc(2 * audio_buffer_frames * sizeof(float));
giuliomoro@19: 	analogInBuffer = (float *)malloc(spi_num_channels * spi_buffer_frames * sizeof(float));
giuliomoro@19: 	analogOutBuffer = (float *)malloc(spi_num_channels * spi_buffer_frames * sizeof(float));
giuliomoro@23: 	lastAnalogOutFrame = (float *)malloc(spi_num_channels * sizeof(float));
giuliomoro@19: 	digitalBuffer0 = pru_buffer_digital; 
giuliomoro@19: 	digitalBuffer1 = pru_buffer_digital+MEM_DIGITAL_BUFFER1_OFFSET/sizeof(uint32_t);
giuliomoro@23:     digital_buffer_frames = digital_enabled ? audio_buffer_frames : 0; //TODO: find a more elegant solution for when the digital is disabled e.g.:
giuliomoro@19:     																	// - embed in the digitalWrite/Read macros a check whether digital is enabled
giuliomoro@16:     																	// - allocate some memory in ARM just to allow render() to run regardless.
giuliomoro@19:     																	//     in this case it can be digitalBuffer0 == digitalBuffer1
giuliomoro@19: 	printf("digital_buffer_frames: %d;\n",digital_buffer_frames);
giuliomoro@23:     lastDigitalBuffer = (uint32_t *)malloc(digital_buffer_frames*sizeof(uint32_t)); //temp buffer to hold previous states
andrewm@0: 	if(audioInBuffer == 0 || audioOutBuffer == 0) {
giuliomoro@16: 		rt_printf("Error: couldn't allocate audio buffers\n");
andrewm@0: 		return;
andrewm@0: 	}
giuliomoro@23: 	if(analogInBuffer == 0 || analogOutBuffer == 0 || lastAnalogOutFrame == 0) {
giuliomoro@19: 		rt_printf("Error: couldn't allocate analog buffers\n");
giuliomoro@16: 		return;
giuliomoro@16: 	}
giuliomoro@23: 	if(lastDigitalBuffer == 0) {
giuliomoro@19: 		rt_printf("Error: couldn't allocate digital buffers\n");
giuliomoro@16: 		return;
giuliomoro@23: 	}
giuliomoro@38: 
giuliomoro@38: 	for(unsigned int n=0; n<digital_buffer_frames; n++){ //initialize lastDigitalFrames to all inputs
giuliomoro@38: 		lastDigitalBuffer[n]= 0x0000ffff;
giuliomoro@38: 	}
andrewm@0: 	while(!gShouldStop) {
andrewm@0: 		// Wait for PRU to move to buffer 1
andrewm@0: 		while(pru_buffer_comm[PRU_CURRENT_BUFFER] == 0 && !gShouldStop) {
andrewm@0: 			rt_task_sleep(sleepTime);
andrewm@0: 		}
andrewm@0: 		if(gShouldStop)
andrewm@0: 			break;
andrewm@0: 
andrewm@0: 		if(xenomai_gpio != 0) {
andrewm@0: 			// Set the test pin high
andrewm@0: 			xenomai_gpio[GPIO_SETDATAOUT] = TEST_PIN_MASK;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		// Render from/to buffer 0
andrewm@0: 
andrewm@0: 		// Convert short (16-bit) samples to float
andrewm@0: 		for(unsigned int n = 0; n < 2 * audio_buffer_frames; n++)
andrewm@0: 			audioInBuffer[n] = (float)pru_buffer_audio_adc[n] / 32768.0;
giuliomoro@16: 		if(spi_enabled) {
giuliomoro@16: 			for(unsigned int n = 0; n < spi_num_channels * spi_buffer_frames; n++)
giuliomoro@19: 				analogInBuffer[n] = (float)pru_buffer_spi_adc[n] / 65536.0;
giuliomoro@23: 			//initialize the output buffer with the values that were in the last frame of the previous output
giuliomoro@23: 			for(int n = 0; n < spi_num_channels; n++){
giuliomoro@23: 				for(unsigned int j = 0; j < spi_buffer_frames; j++){
giuliomoro@23: 					analogOutBuffer[j*spi_buffer_frames + n] = lastAnalogOutFrame[n];
giuliomoro@23: 				}
giuliomoro@23: 			}
giuliomoro@23:         //use past digital values to initialize the array properly.
giuliomoro@23: 		//For each frame:
giuliomoro@23:         //- pins previously set as outputs will keep the output value they had in the last frame of the previous buffer,
giuliomoro@16:         //- pins previously set as inputs will carry the newly read input value
giuliomoro@19:             if(digital_enabled){
giuliomoro@19:                 for(unsigned int n = 0; n < digital_buffer_frames; n++){ 
giuliomoro@23:                     uint16_t inputs=lastDigitalBuffer[n]&0xffff;//half-word, has 1 for inputs and 0 for outputs
giuliomoro@16: //                    printf("inputs: 0x%x\n",inputs);
giuliomoro@16:                     uint16_t outputs=~inputs; //half-word has 1 for outputs and 0 for inputs;
giuliomoro@23:                     digitalBuffer0[n]=(lastDigitalBuffer[digital_buffer_frames-1]&(outputs<<16))| //keep output values set in the last frame of the previous buffer
giuliomoro@19:                                          (digitalBuffer0[n]&(inputs<<16))   | //inputs from current digitalBuffer0[n];
giuliomoro@23:                                          (lastDigitalBuffer[n]&(inputs));     //keep pin configuration from previous digitalBuffer1[n]
giuliomoro@19: //                    digitalBuffer0[n]=digitalBufferTemp[n]; //ignores inputs
giuliomoro@16:                 }
giuliomoro@16:             }
giuliomoro@19: 			render(spi_buffer_frames, digital_buffer_frames, audio_buffer_frames, audioInBuffer, audioOutBuffer,
giuliomoro@19: 					analogInBuffer, analogOutBuffer, digitalBuffer0);
giuliomoro@23: 			//remember the content of the lastAnalogOutFrame
giuliomoro@23: 			for(int n = 0; n < spi_num_channels; n++){
giuliomoro@23: 				lastAnalogOutFrame[n] = analogOutBuffer[spi_buffer_frames*(spi_buffer_frames-1) + n];
giuliomoro@23: 			}
giuliomoro@16: 			for(unsigned int n = 0; n < spi_num_channels * spi_buffer_frames; n++) {
giuliomoro@19: 				int out = analogOutBuffer[n] * 65536.0;
giuliomoro@16: 				if(out < 0) out = 0;
giuliomoro@16: 				else if(out > 65535) out = 65535;
giuliomoro@16: 				pru_buffer_spi_dac[n] = (uint16_t)out;
giuliomoro@16: 			}
giuliomoro@19:             if(digital_enabled){ // keep track of past digital values
giuliomoro@19:                 for(unsigned int n = 0; n < digital_buffer_frames; n++){ 
giuliomoro@23:                     lastDigitalBuffer[n]=digitalBuffer0[n];
giuliomoro@16:                 }
giuliomoro@16:             }
giuliomoro@16: 		}
andrewm@0: 		else
giuliomoro@19: 			render(0, 0, audio_buffer_frames, audioInBuffer, audioOutBuffer, 0, 0, 0); // we still pass digitalBuffer, just it is unused
giuliomoro@16:         // Convert float back to short
andrewm@0: 		for(unsigned int n = 0; n < 2 * audio_buffer_frames; n++) {
andrewm@0: 			int out = audioOutBuffer[n] * 32768.0;
andrewm@0: 			if(out < -32768) out = -32768;
andrewm@0: 			else if(out > 32767) out = 32767;
andrewm@0: 			pru_buffer_audio_dac[n] = (int16_t)out;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		if(xenomai_gpio != 0) {
andrewm@0: 			// Set the test pin high
andrewm@0: 			xenomai_gpio[GPIO_CLEARDATAOUT] = TEST_PIN_MASK;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		// Wait for PRU to move to buffer 0
andrewm@0: 		while(pru_buffer_comm[PRU_CURRENT_BUFFER] != 0 && !gShouldStop) {
andrewm@0: 			rt_task_sleep(sleepTime);
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		if(gShouldStop)
andrewm@0: 			break;
andrewm@0: 
andrewm@0: 		if(xenomai_gpio != 0) {
andrewm@0: 			// Set the test pin high
andrewm@0: 			xenomai_gpio[GPIO_SETDATAOUT] = TEST_PIN_MASK;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		// Render from/to buffer 1
andrewm@0: 
andrewm@0: 		// Convert short (16-bit) samples to float
andrewm@0: 		for(unsigned int n = 0; n < 2 * audio_buffer_frames; n++)
andrewm@0: 			audioInBuffer[n] = (float)pru_buffer_audio_adc[n + audio_buffer_frames * 2] / 32768.0;
andrewm@0: 
giuliomoro@16: 		if(spi_enabled) {
giuliomoro@23: 			 //convert input values TODO: move to PRU
giuliomoro@23: 			for(unsigned int n = 0; n < spi_num_channels * spi_buffer_frames; n++){
giuliomoro@19: 				analogInBuffer[n] = (float)pru_buffer_spi_adc[n + spi_buffer_frames * spi_num_channels] / 65536.0;
giuliomoro@23: 			}
giuliomoro@23: 			//initialize the output buffer with the values that were in the last frame of the previous output
giuliomoro@23: 			for(int n = 0; n < spi_num_channels; n++){
giuliomoro@23: 				for(unsigned int j = 0; j < spi_buffer_frames; j++){
giuliomoro@23: 					analogOutBuffer[j*spi_buffer_frames + n] = lastAnalogOutFrame[n];
giuliomoro@23: 				}
giuliomoro@23: 			}
giuliomoro@19:             if(digital_enabled){
giuliomoro@23:                 for(unsigned int n = 0; n < digital_buffer_frames; n++){
giuliomoro@23:                     uint16_t inputs=lastDigitalBuffer[n]&0xffff;//half-word, has 1 for inputs and 0 for outputs
giuliomoro@16:                     uint16_t outputs=~inputs; //half-word has 1 for outputs and one for inputs;
giuliomoro@23:                     digitalBuffer1[n]=(lastDigitalBuffer[digital_buffer_frames-1]&(outputs<<16))| //keep output values set in the last frame of the  previous buffer
giuliomoro@19:                                          (digitalBuffer1[n]&(inputs<<16))   | //inputs from current digitalBuffer1[n];
giuliomoro@23:                                          (lastDigitalBuffer[n]&(inputs));     //keep pin configuration from previous digitalBuffer1[n]
giuliomoro@19: //                    digitalBuffer1[n]=digitalBufferTemp[n]; //ignores inputs
giuliomoro@16:                 }
giuliomoro@16:             }
giuliomoro@19: 			render(spi_buffer_frames, digital_buffer_frames, audio_buffer_frames, audioInBuffer, audioOutBuffer,
giuliomoro@19: 					analogInBuffer, analogOutBuffer, digitalBuffer1);
giuliomoro@23: 			//remember the content of the lastAnalogOutFrame
giuliomoro@23: 			for(int n = 0; n < spi_num_channels; n++){
giuliomoro@23: 				lastAnalogOutFrame[n] = analogOutBuffer[spi_buffer_frames*(spi_buffer_frames-1) + n];
giuliomoro@23: 			}
giuliomoro@23: 
giuliomoro@16: 			for(unsigned int n = 0; n < spi_num_channels * spi_buffer_frames; n++) {
giuliomoro@19: 				int out = analogOutBuffer[n] * 65536.0;
giuliomoro@16: 				if(out < 0) out = 0;
giuliomoro@16: 				else if(out > 65535) out = 65535;
giuliomoro@16: 				pru_buffer_spi_dac[n + spi_buffer_frames * spi_num_channels] = (uint16_t)out;
giuliomoro@16: 			}
giuliomoro@19:             if(digital_enabled){ // keep track of past digital values
giuliomoro@19:                 for(unsigned int n = 0; n < digital_buffer_frames; n++){ 
giuliomoro@23:                     lastDigitalBuffer[n]=digitalBuffer1[n];
giuliomoro@16:                 }
giuliomoro@16:             }
giuliomoro@16: 		}
andrewm@0: 		else
giuliomoro@19: 			render(0, 0, audio_buffer_frames, audioInBuffer, audioOutBuffer, 0, 0, 0); // we still pass digitalBuffer, just it is unused
andrewm@0: 
andrewm@0: 		// Convert float back to short
andrewm@0: 		for(unsigned int n = 0; n < 2 * audio_buffer_frames; n++) {
andrewm@0: 			int out = audioOutBuffer[n] * 32768.0;
andrewm@0: 			if(out < -32768) out = -32768;
andrewm@0: 			else if(out > 32767) out = 32767;
andrewm@0: 			pru_buffer_audio_dac[n + audio_buffer_frames * 2] = (int16_t)out;
andrewm@0: 		}
andrewm@0: 
andrewm@0: 		if(xenomai_gpio != 0) {
andrewm@0: 			// Set the test pin high
andrewm@0: 			xenomai_gpio[GPIO_CLEARDATAOUT] = TEST_PIN_MASK;
andrewm@0: 		}
andrewm@0: 	}
andrewm@0: 
andrewm@0: 	// Tell PRU to stop
andrewm@0: 	pru_buffer_comm[PRU_SHOULD_STOP] = 1;
andrewm@0: 
giuliomoro@19: 	free(analogOutBuffer);
andrewm@0: 	free(audioInBuffer);
andrewm@0: 	free(audioOutBuffer);
giuliomoro@19: 	free(analogInBuffer);
giuliomoro@23: 	free(lastAnalogOutFrame);
giuliomoro@23:     free(lastDigitalBuffer);
andrewm@0: }
andrewm@0: 
andrewm@0: // Wait for an interrupt from the PRU indicate it is finished
andrewm@0: void PRU::waitForFinish()
andrewm@0: {
andrewm@0: 	if(!running)
andrewm@0: 		return;
andrewm@15:     prussdrv_pru_wait_event (pru_number == 0 ? PRU_EVTOUT_0 : PRU_EVTOUT_1);
andrewm@15: 	if(pru_number == 0)
andrewm@15: 		prussdrv_pru_clear_event(PRU_EVTOUT_0, PRU0_ARM_INTERRUPT);
andrewm@15: 	else
andrewm@15: 		prussdrv_pru_clear_event(PRU_EVTOUT_1, PRU1_ARM_INTERRUPT);
andrewm@0: }
andrewm@0: 
andrewm@0: // Turn off the PRU when done
andrewm@0: void PRU::disable()
andrewm@0: {
andrewm@0:     /* Disable PRU and close memory mapping*/
andrewm@0:     prussdrv_pru_disable(pru_number);
andrewm@0:     prussdrv_exit();
andrewm@0: 	running = false;
andrewm@0: }
andrewm@0: 
andrewm@0: // Debugging
andrewm@0: void PRU::setGPIOTestPin()
andrewm@0: {
andrewm@0: 	if(!xenomai_gpio)
andrewm@0: 		return;
andrewm@0: 	xenomai_gpio[GPIO_SETDATAOUT] = TEST_PIN2_MASK;
andrewm@0: }
andrewm@0: 
andrewm@0: void PRU::clearGPIOTestPin()
andrewm@0: {
andrewm@0: 	if(!xenomai_gpio)
andrewm@0: 		return;
andrewm@0: 	xenomai_gpio[GPIO_CLEARDATAOUT] = TEST_PIN2_MASK;
andrewm@0: }