ECE 4180 - Final Project Team / Mbed 2 deprecated WalkieTalkie

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

Committer:
Nurchu
Date:
Sun Apr 22 00:48:48 2018 +0000
Revision:
27:08d34e60b6d0
Parent:
26:57731422b129
Child:
28:9413eb50156d
Child:
29:0c6f3c0c992a
Fixed the code so it should almost work now

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 27:08d34e60b6d0 14 #define FIFO_BUFFER_SCALE 4
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 27:08d34e60b6d0 18 #define SAMPLE_PERIOD 0.1
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 27:08d34e60b6d0 30 uLCD_4DGL uLCD(p27, p28, p30); // 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 27:08d34e60b6d0 37 InterruptIn button(p18); //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 27:08d34e60b6d0 59 int channelNum = 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 27:08d34e60b6d0 70 }
drechtmann3 1:fc0a2c17e086 71
Nurchu 27:08d34e60b6d0 72 // Callback interrupt from the button to shift into receive mode
Nurchu 27:08d34e60b6d0 73 void enterRecieveMode() {
Nurchu 27:08d34e60b6d0 74 mode = RECEIVE;
drechtmann3 10:a8fcfc869fa5 75 }
drechtmann3 1:fc0a2c17e086 76
Nurchu 27:08d34e60b6d0 77 // Called every SAMPLE_PERIOD ms to sample the mic or output data into the speaker
Nurchu 27:08d34e60b6d0 78 void sampleData()
Nurchu 12:efcfe4c0d9f2 79 {
Nurchu 27:08d34e60b6d0 80 // Depending on the mode, only sample the mic or output data to the speaker
Nurchu 27:08d34e60b6d0 81 if (mode == RECEIVE) {
Nurchu 27:08d34e60b6d0 82 // Get speaker sample from buffer
Nurchu 27:08d34e60b6d0 83 // If there is no data in the buffer, it will just output 0 to the write function
Nurchu 27:08d34e60b6d0 84 uint8_t speakerSample = 0;
Nurchu 27:08d34e60b6d0 85 rxbuff.pop(&speakerSample, 1);
Nurchu 27:08d34e60b6d0 86
Nurchu 27:08d34e60b6d0 87 // Output into the actual speaker
Nurchu 27:08d34e60b6d0 88 spkr.write(speakerSample);
Nurchu 27:08d34e60b6d0 89 } else {
Nurchu 27:08d34e60b6d0 90 // Get mic sample and place into buffer
Nurchu 27:08d34e60b6d0 91 uint8_t micSample = mic.getData();
Nurchu 27:08d34e60b6d0 92 txbuff.push(&micSample, 1);
Nurchu 12:efcfe4c0d9f2 93
Nurchu 27:08d34e60b6d0 94 // Make sure the speaker is actually off
Nurchu 27:08d34e60b6d0 95 spkr.turnOff();
Nurchu 27:08d34e60b6d0 96 }
Nurchu 27:08d34e60b6d0 97
Nurchu 27:08d34e60b6d0 98 // TODO: This will have to be removed later on once we actually crank up the sample rate
Nurchu 27:08d34e60b6d0 99 pc.printf("Sampling....\n\r");
Nurchu 27:08d34e60b6d0 100 myled3 = !myled3;
Nurchu 27:08d34e60b6d0 101 }
drechtmann3 1:fc0a2c17e086 102
Nurchu 27:08d34e60b6d0 103 // Communicates to the other MBED using RF
Nurchu 27:08d34e60b6d0 104 void commThread()
Nurchu 27:08d34e60b6d0 105 {
Nurchu 27:08d34e60b6d0 106 // We want this in it's own thread so we don't have to worry about the
Nurchu 27:08d34e60b6d0 107 // timings screwing anything else up
Nurchu 27:08d34e60b6d0 108 // It can't be in an interrupt because of that
Nurchu 27:08d34e60b6d0 109 while (true) {
Nurchu 27:08d34e60b6d0 110 // Change what we are sending based on what mode we are in
Nurchu 27:08d34e60b6d0 111 if (mode == RECEIVE) {
Nurchu 27:08d34e60b6d0 112 // Make sure something is there to read
Nurchu 27:08d34e60b6d0 113 if (my_nrf24l01p.readable( NRF24L01P_PIPE_P0 )) {
Nurchu 27:08d34e60b6d0 114 uint8_t spkrPacket[DATA_PACKET_SIZE];
Nurchu 27:08d34e60b6d0 115
Nurchu 27:08d34e60b6d0 116 // Remove entire packet of data from the bus
Nurchu 27:08d34e60b6d0 117 int numReceived = my_nrf24l01p.read( NRF24L01P_PIPE_P0, (char*) spkrPacket, DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 118
Nurchu 27:08d34e60b6d0 119 // Place into buffer to play speaker in another thread
Nurchu 27:08d34e60b6d0 120 // Only place into the buffer the number of bytes received
Nurchu 27:08d34e60b6d0 121 rxbuff.push(spkrPacket, min(DATA_PACKET_SIZE, numReceived));
Nurchu 27:08d34e60b6d0 122
Nurchu 27:08d34e60b6d0 123 pc.printf("Receiviing....\n\r");
Nurchu 27:08d34e60b6d0 124 myled2 = !myled2;
Nurchu 27:08d34e60b6d0 125 }
Nurchu 27:08d34e60b6d0 126 } else { // mode == TRANSMIT
Nurchu 27:08d34e60b6d0 127 if (txbuff.size() >= DATA_PACKET_SIZE) {
Nurchu 27:08d34e60b6d0 128 uint8_t micPacket[DATA_PACKET_SIZE];
Nurchu 27:08d34e60b6d0 129
Nurchu 27:08d34e60b6d0 130 // Pull an entire packet of data from the mic sample buffer
Nurchu 27:08d34e60b6d0 131 int numPopped = txbuff.pop(micPacket, DATA_PACKET_SIZE);
Nurchu 27:08d34e60b6d0 132
Nurchu 27:08d34e60b6d0 133 // Send the entire buffer to the other device
Nurchu 27:08d34e60b6d0 134 // TODO: We just assume that DATA_PACKET_SIZE bytes were popped, this may
Nurchu 27:08d34e60b6d0 135 // not be the case
Nurchu 27:08d34e60b6d0 136 my_nrf24l01p.write( NRF24L01P_PIPE_P0, (char*) micPacket, DATA_PACKET_SIZE );
Nurchu 27:08d34e60b6d0 137
Nurchu 27:08d34e60b6d0 138 pc.printf("Transmitting....\n\r");
Nurchu 27:08d34e60b6d0 139 myled1 = !myled1;
Nurchu 27:08d34e60b6d0 140 }
drechtmann3 1:fc0a2c17e086 141 }
Nurchu 27:08d34e60b6d0 142
Nurchu 27:08d34e60b6d0 143 Thread::yield();
Nurchu 12:efcfe4c0d9f2 144 }
Nurchu 12:efcfe4c0d9f2 145 }
drechtmann3 1:fc0a2c17e086 146
Nurchu 27:08d34e60b6d0 147 // Displays the current info to the LCD display
jacksacane 21:95009b231c1f 148 void lcdThread()
jacksacane 21:95009b231c1f 149 {
jacksacane 21:95009b231c1f 150 while (1) {
jacksacane 21:95009b231c1f 151 uLCD.locate(64, 20);
jacksacane 21:95009b231c1f 152 uLCD.printf("Frequency: %d MHz", rfFreq);
jacksacane 21:95009b231c1f 153 uLCD.locate(64, 40);
jacksacane 21:95009b231c1f 154 uLCD.printf("Data Rate: %d kbps", dataRate);
jacksacane 21:95009b231c1f 155 uLCD.locate(64, 60);
jacksacane 21:95009b231c1f 156 uLCD.printf("TX Address: 0x%010llX", txAddr);
jacksacane 21:95009b231c1f 157 uLCD.locate(64, 80);
jacksacane 21:95009b231c1f 158 uLCD.printf("RX Address: 0x%010llX", rxAddr);
jacksacane 21:95009b231c1f 159 uLCD.locate(64, 100);
Nurchu 27:08d34e60b6d0 160
jacksacane 21:95009b231c1f 161 switch (mode) {
jacksacane 21:95009b231c1f 162 case RECEIVE:
jacksacane 21:95009b231c1f 163 uLCD.printf("Mode: Receiving");
jacksacane 21:95009b231c1f 164 break;
jacksacane 21:95009b231c1f 165 case TRANSMIT:
jacksacane 21:95009b231c1f 166 uLCD.printf("Mode: Transmitting");
jacksacane 21:95009b231c1f 167 break;
jacksacane 21:95009b231c1f 168 }
Nurchu 27:08d34e60b6d0 169
jacksacane 21:95009b231c1f 170 // Maybe add some graphics too idk
Nurchu 27:08d34e60b6d0 171 Thread::wait(50);
jacksacane 21:95009b231c1f 172 }
jacksacane 21:95009b231c1f 173 }
jacksacane 21:95009b231c1f 174
Nurchu 12:efcfe4c0d9f2 175 int main()
Nurchu 12:efcfe4c0d9f2 176 {
Nurchu 27:08d34e60b6d0 177 Thread lcd;
Nurchu 27:08d34e60b6d0 178 Thread comm;
jacksacane 21:95009b231c1f 179
Nurchu 27:08d34e60b6d0 180 // Set up the nrf24l01p
jacksacane 21:95009b231c1f 181 rfFreq = my_nrf24l01p.getRfFrequency();
jacksacane 21:95009b231c1f 182 dataRate = my_nrf24l01p.getAirDataRate();
jacksacane 21:95009b231c1f 183 rxAddr = my_nrf24l01p.getRxAddress();
jacksacane 21:95009b231c1f 184 txAddr = my_nrf24l01p.getTxAddress();
jacksacane 21:95009b231c1f 185
Nurchu 27:08d34e60b6d0 186 my_nrf24l01p.setTransferSize(DATA_PACKET_SIZE);
jacksacane 21:95009b231c1f 187
jacksacane 21:95009b231c1f 188 my_nrf24l01p.setReceiveMode();
jacksacane 21:95009b231c1f 189 my_nrf24l01p.enable();
jacksacane 21:95009b231c1f 190
Nurchu 27:08d34e60b6d0 191 pc.printf("Finished starting up....\n\r");
Nurchu 27:08d34e60b6d0 192
jacksacane 22:e835b3490280 193 mode = RECEIVE;
jacksacane 22:e835b3490280 194
jacksacane 21:95009b231c1f 195 // Initialize the uLCD
jacksacane 21:95009b231c1f 196 uLCD.baudrate(3000000);
jacksacane 21:95009b231c1f 197 uLCD.background_color(BLACK);
jacksacane 21:95009b231c1f 198
jacksacane 21:95009b231c1f 199 // Spawn threads
jacksacane 21:95009b231c1f 200 lcd.start(lcdThread);
Nurchu 27:08d34e60b6d0 201 comm.start(commThread);
jacksacane 21:95009b231c1f 202
Nurchu 27:08d34e60b6d0 203 // Setup the button to enter transmit mode when pushed down
Nurchu 27:08d34e60b6d0 204 // and recieve when release
Nurchu 27:08d34e60b6d0 205 button.mode(PullDown);
Nurchu 27:08d34e60b6d0 206 button.fall(&enterTransmitMode);
Nurchu 27:08d34e60b6d0 207 button.rise(&enterRecieveMode);
Nurchu 27:08d34e60b6d0 208
Nurchu 27:08d34e60b6d0 209 // Setup sampler to sample at a specific frequency
Nurchu 27:08d34e60b6d0 210 sampler.attach(&sampleData, SAMPLE_PERIOD);
Nurchu 27:08d34e60b6d0 211
Nurchu 27:08d34e60b6d0 212 // Heartbeat thread
drechtmann3 1:fc0a2c17e086 213 while (1) {
Nurchu 27:08d34e60b6d0 214 myled4 = !myled4;
Nurchu 27:08d34e60b6d0 215 Thread::wait(100);
Nurchu 12:efcfe4c0d9f2 216 }
Nurchu 12:efcfe4c0d9f2 217 }