ECE 4180 - Final Project Team / Mbed 2 deprecated WalkieTalkie

Dependencies:   mbed 4DGL-uLCD-SE mbed-rtos nRF24L01P

Committer:
Nurchu
Date:
Wed Apr 25 15:39:03 2018 +0000
Revision:
33:5d86c111d9bc
Parent:
32:c40d581f50af
Child:
34:d73e95bbdbed
Child:
40:4f7d95c68d29
Fixed a few bugs

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Nurchu 0:c35b54fb9c3c 1 #include "mbed.h"
Nurchu 7:0ac1f1ca8aa6 2 #include "rtos.h"
Nurchu 20:e068469ffb89 3
Nurchu 2:dc046ff72566 4 #include "Microphone.h"
Nurchu 20:e068469ffb89 5 #include "Speaker.h"
drechtmann3 18:e2a8ea6f55d3 6 #include "HUD.h"
Nurchu 20:e068469ffb89 7 #include "nRF24L01P.h"
jacksacane 21:95009b231c1f 8 #include "uLCD_4DGL.h"
Nurchu 20:e068469ffb89 9
Nurchu 20:e068469ffb89 10 #include "CircularBuf.h"
Nurchu 20:e068469ffb89 11 #include "CircularBuf.cpp" // Hack to get templates to work
Nurchu 20:e068469ffb89 12
Nurchu 27:08d34e60b6d0 13 // How many times larger the buffer should be
Nurchu 33:5d86c111d9bc 14 #define FIFO_BUFFER_SCALE 250
Nurchu 27:08d34e60b6d0 15 // How maybe bytes to send over RF
Nurchu 27:08d34e60b6d0 16 #define DATA_PACKET_SIZE 32
Nurchu 27:08d34e60b6d0 17 // How quickly to sample the mic / play the speaker
Nurchu 33:5d86c111d9bc 18 #define SAMPLE_PERIOD 1/16000.0
Nurchu 20:e068469ffb89 19
Nurchu 20:e068469ffb89 20
Nurchu 27:08d34e60b6d0 21 Serial pc(USBTX, USBRX);
Nurchu 27:08d34e60b6d0 22 DigitalOut myled1(LED1); // Mic data sent over RF
Nurchu 27:08d34e60b6d0 23 DigitalOut myled2(LED2); // Speaker data recieved over RF
Nurchu 27:08d34e60b6d0 24 DigitalOut myled3(LED3); // Sampled mic / played the speaker
Nurchu 27:08d34e60b6d0 25 DigitalOut myled4(LED4); // Heartbeat
Nurchu 20:e068469ffb89 26
Nurchu 20:e068469ffb89 27 Speaker spkr(p18);
Nurchu 27:08d34e60b6d0 28 Microphone mic(p16);
Nurchu 27:08d34e60b6d0 29
Nurchu 33:5d86c111d9bc 30 //uLCD_4DGL uLCD(p28, p27, p29); // serial tx, serial rx, reset pin;
Nurchu 20:e068469ffb89 31
Nurchu 20:e068469ffb89 32 nRF24L01P my_nrf24l01p(p5, p6, p7, p8, p9, p10); // mosi, miso, sck, csn, ce, irq
Nurchu 27:08d34e60b6d0 33 CircularBuf<uint8_t> txbuff( FIFO_BUFFER_SCALE * DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 34 CircularBuf<uint8_t> rxbuff( FIFO_BUFFER_SCALE * DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 35 Ticker sampler; //10:41 am 4/20
Nurchu 20:e068469ffb89 36
Nurchu 29:0c6f3c0c992a 37 InterruptIn button(p12); //changed DitialIn to InterruptIn at 5:54 4/18/18
Nurchu 27:08d34e60b6d0 38 BusIn channel(p21, p22, p23, p24); // TODO: CHANGE THESE TO THE ACTUAL PINS NEEDED
Nurchu 20:e068469ffb89 39
jacksacane 21:95009b231c1f 40 int rfFreq;
jacksacane 21:95009b231c1f 41 int dataRate;
jacksacane 21:95009b231c1f 42 unsigned long long rxAddr, txAddr;
jacksacane 21:95009b231c1f 43
jacksacane 21:95009b231c1f 44 enum operatingMode {
jacksacane 21:95009b231c1f 45 RECEIVE = 0,
jacksacane 21:95009b231c1f 46 TRANSMIT
jacksacane 21:95009b231c1f 47 };
jacksacane 21:95009b231c1f 48
jacksacane 21:95009b231c1f 49 operatingMode mode;
jacksacane 21:95009b231c1f 50
Nurchu 27:08d34e60b6d0 51 // Cheap nonbranching min function
Nurchu 27:08d34e60b6d0 52 int min(int a, int b) {
Nurchu 27:08d34e60b6d0 53 return a * (int)(a <= b) + b * (int)(b < a);
Nurchu 27:08d34e60b6d0 54 }
drechtmann3 1:fc0a2c17e086 55
Nurchu 27:08d34e60b6d0 56 // Sets the frequency of the RF device based on which swtiches are flipped
Nurchu 27:08d34e60b6d0 57 // TODO: Make sure we don't have to restart the device or anything to change this
Nurchu 27:08d34e60b6d0 58 void setRFFrequency() {
Nurchu 29:0c6f3c0c992a 59 int channelNum = 0;//channel.read();
Nurchu 27:08d34e60b6d0 60
Nurchu 27:08d34e60b6d0 61 // TODO: Don't force it to the default RF frequency
Nurchu 27:08d34e60b6d0 62 channelNum = 2;
Nurchu 27:08d34e60b6d0 63
Nurchu 27:08d34e60b6d0 64 my_nrf24l01p.setRfFrequency(channelNum + NRF24L01P_MIN_RF_FREQUENCY);
Nurchu 27:08d34e60b6d0 65 }
drechtmann3 1:fc0a2c17e086 66
Nurchu 27:08d34e60b6d0 67 // Callback interrupt from the button to shift into transmit mode
Nurchu 27:08d34e60b6d0 68 void enterTransmitMode() {
Nurchu 27:08d34e60b6d0 69 mode = TRANSMIT;
Nurchu 33:5d86c111d9bc 70 txbuff.clear();
Nurchu 27:08d34e60b6d0 71 }
drechtmann3 1:fc0a2c17e086 72
Nurchu 27:08d34e60b6d0 73 // Callback interrupt from the button to shift into receive mode
Nurchu 27:08d34e60b6d0 74 void enterRecieveMode() {
Nurchu 27:08d34e60b6d0 75 mode = RECEIVE;
Nurchu 33:5d86c111d9bc 76 //TODO: Add this back in
Nurchu 33:5d86c111d9bc 77 //rxbuf.clear();
drechtmann3 10:a8fcfc869fa5 78 }
drechtmann3 1:fc0a2c17e086 79
Nurchu 27:08d34e60b6d0 80 // Called every SAMPLE_PERIOD ms to sample the mic or output data into the speaker
Nurchu 27:08d34e60b6d0 81 void sampleData()
Nurchu 12:efcfe4c0d9f2 82 {
Nurchu 27:08d34e60b6d0 83 // Depending on the mode, only sample the mic or output data to the speaker
Nurchu 27:08d34e60b6d0 84 if (mode == RECEIVE) {
Nurchu 27:08d34e60b6d0 85 // Get speaker sample from buffer
Nurchu 27:08d34e60b6d0 86 // If there is no data in the buffer, it will just output 0 to the write function
Nurchu 27:08d34e60b6d0 87 uint8_t speakerSample = 0;
Nurchu 27:08d34e60b6d0 88 rxbuff.pop(&speakerSample, 1);
Nurchu 27:08d34e60b6d0 89
Nurchu 27:08d34e60b6d0 90 // Output into the actual speaker
Nurchu 27:08d34e60b6d0 91 spkr.write(speakerSample);
Nurchu 27:08d34e60b6d0 92 } else {
Nurchu 27:08d34e60b6d0 93 // Get mic sample and place into buffer
Nurchu 33:5d86c111d9bc 94 uint8_t micSample = mic.getData();
Nurchu 27:08d34e60b6d0 95 txbuff.push(&micSample, 1);
Nurchu 12:efcfe4c0d9f2 96
Nurchu 27:08d34e60b6d0 97 // Make sure the speaker is actually off
Nurchu 27:08d34e60b6d0 98 spkr.turnOff();
Nurchu 27:08d34e60b6d0 99 }
Nurchu 27:08d34e60b6d0 100
Nurchu 27:08d34e60b6d0 101 // TODO: This will have to be removed later on once we actually crank up the sample rate
Nurchu 27:08d34e60b6d0 102 myled3 = !myled3;
Nurchu 27:08d34e60b6d0 103 }
drechtmann3 1:fc0a2c17e086 104
Nurchu 27:08d34e60b6d0 105 // Communicates to the other MBED using RF
Nurchu 27:08d34e60b6d0 106 void commThread()
Nurchu 27:08d34e60b6d0 107 {
Nurchu 29:0c6f3c0c992a 108 pc.printf("Enter commThread\n\r");
Nurchu 27:08d34e60b6d0 109 // We want this in it's own thread so we don't have to worry about the
Nurchu 27:08d34e60b6d0 110 // timings screwing anything else up
Nurchu 27:08d34e60b6d0 111 // It can't be in an interrupt because of that
Nurchu 27:08d34e60b6d0 112 while (true) {
Nurchu 27:08d34e60b6d0 113 // Change what we are sending based on what mode we are in
Nurchu 27:08d34e60b6d0 114 if (mode == RECEIVE) {
Nurchu 27:08d34e60b6d0 115 // Make sure something is there to read
Nurchu 27:08d34e60b6d0 116 if (my_nrf24l01p.readable( NRF24L01P_PIPE_P0 )) {
Nurchu 27:08d34e60b6d0 117 uint8_t spkrPacket[DATA_PACKET_SIZE];
Nurchu 27:08d34e60b6d0 118
Nurchu 27:08d34e60b6d0 119 // Remove entire packet of data from the bus
Nurchu 27:08d34e60b6d0 120 int numReceived = my_nrf24l01p.read( NRF24L01P_PIPE_P0, (char*) spkrPacket, DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 121
Nurchu 27:08d34e60b6d0 122 // Place into buffer to play speaker in another thread
Nurchu 27:08d34e60b6d0 123 // Only place into the buffer the number of bytes received
Nurchu 27:08d34e60b6d0 124 rxbuff.push(spkrPacket, min(DATA_PACKET_SIZE, numReceived));
Nurchu 27:08d34e60b6d0 125
Nurchu 27:08d34e60b6d0 126 pc.printf("Receiviing....\n\r");
Nurchu 27:08d34e60b6d0 127 myled2 = !myled2;
Nurchu 27:08d34e60b6d0 128 }
Nurchu 27:08d34e60b6d0 129 } else { // mode == TRANSMIT
Nurchu 27:08d34e60b6d0 130 if (txbuff.size() >= DATA_PACKET_SIZE) {
Nurchu 27:08d34e60b6d0 131 uint8_t micPacket[DATA_PACKET_SIZE];
Nurchu 27:08d34e60b6d0 132
Nurchu 27:08d34e60b6d0 133 // Pull an entire packet of data from the mic sample buffer
Nurchu 27:08d34e60b6d0 134 int numPopped = txbuff.pop(micPacket, DATA_PACKET_SIZE);
Nurchu 33:5d86c111d9bc 135 rxbuff.push(micPacket, DATA_PACKET_SIZE);
Nurchu 27:08d34e60b6d0 136
Nurchu 27:08d34e60b6d0 137 // Send the entire buffer to the other device
Nurchu 27:08d34e60b6d0 138 // TODO: We just assume that DATA_PACKET_SIZE bytes were popped, this may
Nurchu 27:08d34e60b6d0 139 // not be the case
Nurchu 27:08d34e60b6d0 140 my_nrf24l01p.write( NRF24L01P_PIPE_P0, (char*) micPacket, DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 141
Nurchu 33:5d86c111d9bc 142 //for (int i = 0; i < DATA_PACKET_SIZE; i++) {
Nurchu 33:5d86c111d9bc 143 // pc.printf("%x", micPacket[i]);
Nurchu 33:5d86c111d9bc 144 //}
Nurchu 33:5d86c111d9bc 145 //pc.printf("\n\r");
Nurchu 33:5d86c111d9bc 146
Nurchu 33:5d86c111d9bc 147 //pc.printf("Transmitting....\n\r");
Nurchu 27:08d34e60b6d0 148 myled1 = !myled1;
Nurchu 27:08d34e60b6d0 149 }
drechtmann3 1:fc0a2c17e086 150 }
Nurchu 27:08d34e60b6d0 151
Nurchu 33:5d86c111d9bc 152 //pc.printf("TX Size %d RX Size%d Mode %d\n\r", txbuff.size(), rxbuff.size(), mode);
Nurchu 27:08d34e60b6d0 153 Thread::yield();
Nurchu 33:5d86c111d9bc 154 //Thread::wait(10);
Nurchu 12:efcfe4c0d9f2 155 }
Nurchu 12:efcfe4c0d9f2 156 }
drechtmann3 1:fc0a2c17e086 157
Nurchu 27:08d34e60b6d0 158 // Displays the current info to the LCD display
jacksacane 21:95009b231c1f 159 void lcdThread()
jacksacane 21:95009b231c1f 160 {
Nurchu 29:0c6f3c0c992a 161 pc.printf("Enter lcdThread\n\r");
jacksacane 21:95009b231c1f 162 while (1) {
Nurchu 29:0c6f3c0c992a 163 /*
jacksacane 21:95009b231c1f 164 uLCD.locate(64, 20);
jacksacane 21:95009b231c1f 165 uLCD.printf("Frequency: %d MHz", rfFreq);
jacksacane 21:95009b231c1f 166 uLCD.locate(64, 40);
jacksacane 21:95009b231c1f 167 uLCD.printf("Data Rate: %d kbps", dataRate);
jacksacane 21:95009b231c1f 168 uLCD.locate(64, 60);
jacksacane 21:95009b231c1f 169 uLCD.printf("TX Address: 0x%010llX", txAddr);
jacksacane 21:95009b231c1f 170 uLCD.locate(64, 80);
jacksacane 21:95009b231c1f 171 uLCD.printf("RX Address: 0x%010llX", rxAddr);
jacksacane 21:95009b231c1f 172 uLCD.locate(64, 100);
Nurchu 27:08d34e60b6d0 173
jacksacane 21:95009b231c1f 174 switch (mode) {
jacksacane 21:95009b231c1f 175 case RECEIVE:
jacksacane 21:95009b231c1f 176 uLCD.printf("Mode: Receiving");
jacksacane 21:95009b231c1f 177 break;
jacksacane 21:95009b231c1f 178 case TRANSMIT:
jacksacane 21:95009b231c1f 179 uLCD.printf("Mode: Transmitting");
jacksacane 21:95009b231c1f 180 break;
jacksacane 21:95009b231c1f 181 }
Nurchu 29:0c6f3c0c992a 182 */
jacksacane 21:95009b231c1f 183 // Maybe add some graphics too idk
Nurchu 33:5d86c111d9bc 184 Thread::wait(500);
jacksacane 21:95009b231c1f 185 }
jacksacane 21:95009b231c1f 186 }
jacksacane 21:95009b231c1f 187
Nurchu 12:efcfe4c0d9f2 188 int main()
Nurchu 12:efcfe4c0d9f2 189 {
Nurchu 27:08d34e60b6d0 190 Thread lcd;
Nurchu 27:08d34e60b6d0 191 Thread comm;
jacksacane 21:95009b231c1f 192
Nurchu 33:5d86c111d9bc 193 pc.printf("Starting setup....\n\r");
Nurchu 33:5d86c111d9bc 194
Nurchu 27:08d34e60b6d0 195 // Set up the nrf24l01p
jacksacane 21:95009b231c1f 196 rfFreq = my_nrf24l01p.getRfFrequency();
jacksacane 21:95009b231c1f 197 dataRate = my_nrf24l01p.getAirDataRate();
jacksacane 21:95009b231c1f 198 rxAddr = my_nrf24l01p.getRxAddress();
jacksacane 21:95009b231c1f 199 txAddr = my_nrf24l01p.getTxAddress();
jacksacane 21:95009b231c1f 200
Nurchu 27:08d34e60b6d0 201 my_nrf24l01p.setTransferSize(DATA_PACKET_SIZE);
jacksacane 21:95009b231c1f 202
jacksacane 21:95009b231c1f 203 my_nrf24l01p.setReceiveMode();
jacksacane 21:95009b231c1f 204 my_nrf24l01p.enable();
jacksacane 21:95009b231c1f 205
Nurchu 27:08d34e60b6d0 206 pc.printf("Finished starting up....\n\r");
Nurchu 27:08d34e60b6d0 207
jacksacane 22:e835b3490280 208 mode = RECEIVE;
jacksacane 22:e835b3490280 209
jacksacane 21:95009b231c1f 210 // Initialize the uLCD
Nurchu 29:0c6f3c0c992a 211 //uLCD.baudrate(3000000);
Nurchu 29:0c6f3c0c992a 212 //uLCD.background_color(BLACK);
jacksacane 21:95009b231c1f 213
jacksacane 21:95009b231c1f 214 // Spawn threads
jacksacane 21:95009b231c1f 215 lcd.start(lcdThread);
Nurchu 27:08d34e60b6d0 216 comm.start(commThread);
jacksacane 21:95009b231c1f 217
Nurchu 27:08d34e60b6d0 218 // Setup the button to enter transmit mode when pushed down
Nurchu 27:08d34e60b6d0 219 // and recieve when release
jacksacane 28:9413eb50156d 220 button.mode(PullUp);
Nurchu 31:39d04aedc3e5 221 button.fall(&enterTransmitMode);
Nurchu 31:39d04aedc3e5 222 button.rise(&enterRecieveMode);
Nurchu 27:08d34e60b6d0 223
Nurchu 27:08d34e60b6d0 224 // Setup sampler to sample at a specific frequency
Nurchu 31:39d04aedc3e5 225 sampler.attach(&sampleData, SAMPLE_PERIOD);
Nurchu 27:08d34e60b6d0 226
Nurchu 29:0c6f3c0c992a 227 pc.printf("Enter Heartbeat\n\r");
Nurchu 27:08d34e60b6d0 228 // Heartbeat thread
drechtmann3 1:fc0a2c17e086 229 while (1) {
Nurchu 27:08d34e60b6d0 230 myled4 = !myled4;
Nurchu 27:08d34e60b6d0 231 Thread::wait(100);
Nurchu 12:efcfe4c0d9f2 232 }
Nurchu 12:efcfe4c0d9f2 233 }