An embedded device
Dependencies: Crypto
main.cpp
- Committer:
- tanyuzhuo
- Date:
- 2019-03-19
- Revision:
- 20:5bd9f9e406d1
- Parent:
- 18:e48c0910c71e
- Child:
- 21:34f440ae0227
File content as of revision 20:5bd9f9e406d1:
#include "mbed.h" #include "Crypto.h" //Photointerrupter input pins #define I1pin D3 #define I2pin D6 #define I3pin D5 //Incremental encoder input pins #define CHApin D12 #define CHBpin D11 //Motor Drive output pins //Mask in output byte #define L1Lpin D1 //0x01 #define L1Hpin A3 //0x02 #define L2Lpin D0 //0x04 #define L2Hpin A6 //0x08 #define L3Lpin D10 //0x10 #define L3Hpin D2 //0x20 #define PWMpin D9 //Motor current sense #define MCSPpin A1 #define MCSNpin A0 //Mapping from sequential drive states to motor phase outputs /* State L1 L2 L3 0 H - L 1 - H L 2 L H - 3 L - H 4 - L H 5 H L - 6 - - - 7 - - - */ //Drive state to output table const int8_t driveTable[] = {0x12,0x18,0x09,0x21,0x24,0x06,0x00,0x00}; //Mapping from interrupter inputs to sequential rotor states. 0x00 and 0x07 are not valid const int8_t stateMap[] = {0x07,0x05,0x03,0x04,0x01,0x00,0x02,0x07}; //const int8_t stateMap[] = {0x07,0x01,0x03,0x02,0x05,0x00,0x04,0x07}; //Alternative if phase order of input or drive is reversed //Phase lead to make motor spin const int8_t lead = -2; //2 for forwards, -2 for backwards //Status LED DigitalOut led1(LED1); //Photointerrupter inputs InterruptIn I1(I1pin); InterruptIn I2(I2pin); InterruptIn I3(I3pin); //Motor Drive outputs DigitalOut L1L(L1Lpin); DigitalOut L1H(L1Hpin); DigitalOut L2L(L2Lpin); DigitalOut L2H(L2Hpin); DigitalOut L3L(L3Lpin); DigitalOut L3H(L3Hpin); PwmOut d9 (PWMpin); int8_t orState = 0; //int8_t intState = 0; //int8_t intStateOld = 0; int32_t revoCounter = 0; //Counts the number of revolutions int32_t motorVelocity; //Phase lead to make motor spin //int8_t lead = -2; //2 for forwards, -2 for backwards typedef struct { int message; uint64_t data; float fdata; } mail_t; Mail<mail_t, 16> mail_box; Thread commandProcessorthread(osPriorityNormal,1024); Thread bitcointhread(osPriorityNormal,1024); Thread motorCtrlT(osPriorityNormal,1024); Thread commsOut(osPriorityNormal,1024); RawSerial pc(SERIAL_TX, SERIAL_RX); Queue<void, 8> inCharQ; Mutex newKey_mutex; uint64_t newKey = 0; volatile float newRev; volatile float maxSpeed = 300; uint32_t pulseWidth; float motorPosition_command; float motorPosition; void serialISR() { uint8_t newChar = pc.getc(); inCharQ.put((void*) newChar); } /***************************************************************/ //The following block should be moved to a library, but I don't the time to figure this out atm. //CommsOut.h enum message_keys { KEY_DECLINED = 0, ROTATION_CMD = 1, MAX_SPEED_CMD = 2, KEY_ECHO = 3, R_ECHO1 = 4, R_ECHO2 = 5, MAX_SPEED_ECHO1= 6, MAX_SPEED_ECHO2= 7, INVALID_CMD = 10, MOTOR_POS = 11, MOTOR_SPEED = 12, NONCE_DETECT = 14, ROTOR_ORG = 15, WELCOME = 20 }; void putMessage(int message, uint64_t data = 0, float fdata=0){ mail_t *mail = mail_box.alloc(); mail->message = message; mail->data = data; mail->fdata = fdata; mail_box.put(mail); } void commsOutFunc() { while (true) { osEvent evt = mail_box.get(); if (evt.status == osEventMail) { mail_t *mail = (mail_t*)evt.value.p; switch (mail->message) { case KEY_DECLINED : pc.printf("Not a valid key!\n\r"); break; case ROTATION_CMD : pc.printf("Rotation command\n\r"); break; case MAX_SPEED_CMD : pc.printf("Max speed command\n\r"); break; case R_ECHO1 : pc.printf("Rotor command:\n\r"); pc.printf("Full rotations: %d\n\r", mail->data); break; case R_ECHO2 : pc.printf("Partial rotation: %d\n\r", mail->data); break; case MAX_SPEED_ECHO1 : pc.printf("Max speed command:\n\r"); pc.printf("Max speed int: %d\n\r", mail->data); break; case MAX_SPEED_ECHO2 : pc.printf("Max speed decimal: %d\n\r", mail->data); break; case KEY_ECHO : pc.printf("Received key: %016llx\n\r", mail->data); break; case INVALID_CMD : pc.printf("Invalid command!\r\n"); break; case MOTOR_POS : pc.printf("Motor position: %f\r\n", mail->fdata); //pc.printf("{TIMEPLOT|%.2f|Motor Position", mail->fdata); break; case MOTOR_SPEED : pc.printf("Motor speed: %f\r\n", mail->fdata); break; case NONCE_DETECT : pc.printf("Successful nonce: %016x\n\r", mail->data); break; case WELCOME : pc.printf("Hello Pete\n\r"); break; case ROTOR_ORG : pc.printf("Rotor origin: %x\n\r", orState); } mail_box.free(mail); } } } //End of that block /***************************************************************/ /***************************************************************/ //The following block should also be moved to a library, but I still don't the time to figure this out atm. //CommsIn.h void commandDecoder(char* command) { int8_t sign =1; uint8_t index = 1; int intValue = 0; float decimalValue = 0; switch (command[0]) { case 'R' : // Check sign if (command[1] == '-'){ sign = -1; index++; } // Take first digit if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = command[index] - '0'; index++; } else { putMessage(INVALID_CMD); break; } //Try taking 2 more digits if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = intValue*10 + command[index] - '0'; index++; } if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = intValue*10 + command[index] - '0'; index++; } //Check for '.' if (command[index] == '.') { index++; //Check for decimals if (command[index] > ('0'-1) && command[index] < (1 + '9')) { decimalValue = (command[index] - '0')/10; index++; } if (command[index] > ('0'-1) && command[index] < (1 + '9')) { decimalValue = (decimalValue/10 + command[index] - '0')/10; } } putMessage(R_ECHO1, intValue); putMessage(R_ECHO2, (int) (100*decimalValue)); decimalValue = (decimalValue + intValue) * sign; //HERE SEND IT TO ANY GLOBAL VARIABLE YOU WISH break; case 'V' : // Take first digit if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = command[index] - '0'; index++; } else { putMessage(INVALID_CMD); break; } //Try taking 2 more digits if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = intValue*10 + command[index] - '0'; index++; } if (command[index] > ('0'-1) && command[index] < (1 + '9')) { intValue = intValue*10 + command[index] - '0'; index++; } //Check for '.' if (command[index] == '.') { index++; //Check for one decimal if (command[index] > ('0'-1) && command[index] < (1 + '9')) { decimalValue = (command[index] - '0')/10; } } putMessage(MAX_SPEED_ECHO1, intValue); putMessage(MAX_SPEED_ECHO2, (int) (100*decimalValue)); decimalValue = (decimalValue + intValue); //HERE SEND IT TO ANY GLOBAL VARIABLE YOU WISH // maxSpeed = decimalValue; break; case 'K' : ///pc.printf("Received key: %016llx\n\r", mail->data); break; case 'T' : //pc.printf("Received key: %016llx\n\r", mail->data); break; } } void commandProcessor() { pc.attach(&serialISR); char command[19]; char* number; //char k; uint64_t receivedKey; uint8_t index = 0; while(1) { osEvent newEvent = inCharQ.get(); uint8_t newChar = (uint8_t) newEvent.value.p; command[index] = newChar; index++; if (newChar == '\r') { command[17] = '\0'; if (command[0] == 'R') { putMessage(ROTATION_CMD); commandDecoder(command); } else if (command[0] == 'V') { putMessage(MAX_SPEED_CMD); } else if (command[0] == 'K') { if (index == 18){ // when index is 18 means you entered K and 16 digits number = command +1; //super bad solution, but I don't know how to work with strings in C receivedKey = strtoull(number, NULL, 16); putMessage(KEY_ECHO,receivedKey); //receivedKey = 2147483648; //sscanf(command, "%d", &receivedKey); //pc.printf("Received key: %016llx\n\r", receivedKey); newKey_mutex.lock(); newKey = receivedKey; newKey_mutex.unlock(); } else { putMessage(KEY_DECLINED); }; } else if (command[0] == 'T') { pc.printf("Melody command\n"); pc.printf("%s", command); } memset(command, 0, sizeof(command)); index = 0; } else { pc.printf("Current command: %s\n\r", command); //Not sure how to go around this one cause it requires passing string } } } void bitcoin(){ while(1) { SHA256 sha; uint8_t sequence[] = {0x45,0x6D,0x62,0x65,0x64,0x64,0x65,0x64, 0x20,0x53,0x79,0x73,0x74,0x65,0x6D,0x73, 0x20,0x61,0x72,0x65,0x20,0x66,0x75,0x6E, 0x20,0x61,0x6E,0x64,0x20,0x64,0x6F,0x20, 0x61,0x77,0x65,0x73,0x6F,0x6D,0x65,0x20, 0x74,0x68,0x69,0x6E,0x67,0x73,0x21,0x20, 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}; uint64_t* key = (uint64_t*) ((int) sequence + 48); uint64_t* nonce = (uint64_t*) ((int) sequence + 56); uint8_t hash[32]; Timer t; t.start(); unsigned currentTime = 0; unsigned currentCount= 0; for (unsigned i = 0; i <= UINT_MAX; i++) { (*nonce)++; newKey_mutex.lock(); *key = newKey; newKey_mutex.unlock(); sha.computeHash(hash, sequence, 64); if (hash[0] == 0 && hash[1] == 0) { //putMessage(nonce); pc.printf("Successful nonce: %016x\n\r", *nonce); } if ((unsigned) t.read() == currentTime) { //pc.printf("Hash rate: %d\n\r", i - currentCount); //pc.printf("Current key: %016llx\n\r", *key); currentTime++; currentCount = i; } } t.stop(); } } //Set a given drive state void motorOut(int8_t driveState){ //Lookup the output byte from the drive state. int8_t driveOut = driveTable[driveState & 0x07]; //Turn off first if (~driveOut & 0x01) L1L = 0; if (~driveOut & 0x02) L1H = 1; if (~driveOut & 0x04) L2L = 0; if (~driveOut & 0x08) L2H = 1; if (~driveOut & 0x10) L3L = 0; if (~driveOut & 0x20) L3H = 1; //Then turn on if (driveOut & 0x01) L1L = 1; if (driveOut & 0x02) L1H = 0; if (driveOut & 0x04) L2L = 1; if (driveOut & 0x08) L2H = 0; if (driveOut & 0x10) L3L = 1; if (driveOut & 0x20) L3H = 0; } //Convert photointerrupter inputs to a rotor state inline int8_t readRotorState(){ return stateMap[I1 + 2*I2 + 4*I3]; } int8_t motorHome() { //Put the motor in drive state 0 and wait for it to stabilize motorOut(0); wait(2.0); return readRotorState(); } // ISR to handle the updating of the motor position void motorISR() { static int8_t oldRotorState; //orState is subtracted from future rotor state inputs to align rotor and motor states int8_t rotorState = readRotorState(); motorOut((rotorState-orState+lead+6)%6); //+6 to make sure the remainder is positive if (rotorState - oldRotorState == 5) motorPosition --; else if (rotorState - oldRotorState == -5) motorPosition ++; else motorPosition += (rotorState - oldRotorState); //pc.printf ("motorpPosition %f\n\r", motorPosition); oldRotorState = rotorState; } /*void push() { intState = readRotorState(); if (intState != intStateOld) { intStateOld = intState; motorOut((intState - orState + lead +6) % 6); //+6 to make sure the remainder is positive } }*/ void motorCtrlTick(){ motorCtrlT.signal_set(0x1); } void motorCtrlFn(){ int32_t counter=0; static int32_t oldmotorPosition; // uint32_t Ts; // float Speed; // float kps = 25; //proportional constant for speed // Timer to count time passed between ticks to calculate velocity Timer motorTime; motorTime.start(); float motorPos; //float ki = ??; // integration constant, to be tested for friction Ticker motorCtrlTicker; motorCtrlTicker.attach_us(&motorCtrlTick,100000); while(1){ motorCtrlT.signal_wait(0x1); // errorSum= 0; // for(uint8_t i=9; i >0 ; i--){ // PrevErrorArray[i] = prevErrorArray[i-1]; // errorSum+= PrevErrorArray[i]; } // convert state change into rotations //Speed = maxSpeed*6; motorPos = motorPosition; //pc.printf ("motor Pos: %f\n\r", motorPos); motorVelocity = (motorPos - oldmotorPosition)/motorTime.read(); oldmotorPosition = motorPos; //equation for controls //Ts = kps*(Speed -abs(motorVelocity));//*errorSign; //pc.printf ("torque: %f\n\r", Ts); // Mp = ks*error + kd*(error - PrevError) /motorTime.read() + ki*errorSum; motorTime.reset(); // Serial output to monitor speed and position counter++; if(counter == 10){ counter = 0; //display velocity and motor position // pc.printf ("motor Pos: %f\n\r", (motorPosition/6.0)); putMessage(MOTOR_POS,0,(float)(motorPos/6.0)); putMessage(MOTOR_SPEED,0,(float)(motorVelocity/6.0)); } } int main() { //Serial pc(SERIAL_TX, SERIAL_RX); //Initialise bincoin mining and communication d9.period(0.002f); //Set PWM period in seconds d9.write(1); //PWM.period(0.002f); //Set PWM period in seconds //PWM.write(0.5); //Set PWM duty in % commandProcessorthread.start(commandProcessor); commsOut.start(commsOutFunc); motorCtrlT.start(motorCtrlFn); bitcointhread.start(bitcoin); putMessage(WELCOME); int8_t orState = motorHome(); putMessage(ROTOR_ORG); //pc.printf("Rotor origin: %x\n\r", orState); //Set PWM duty in % I1.rise(&motorISR); I2.rise(&motorISR); I3.rise(&motorISR); I1.fall(&motorISR); I2.fall(&motorISR); I3.fall(&motorISR); }