summaryrefslogtreecommitdiff
path: root/src/spi_flash.c
diff options
context:
space:
mode:
authorDawsyn Schraiber <[email protected]>2024-05-09 02:05:35 -0400
committerGitHub <[email protected]>2024-05-09 02:05:35 -0400
commit93acde052369568beaefb0d99629d8797f5c191f (patch)
treea3fb96ddad2d289aa7f8bf410c60cf6289bca7a1 /src/spi_flash.c
parent5f68c7a1b5c8dec82d4a2e1e12443a41b5196b1d (diff)
downloadactive-drag-system-93acde052369568beaefb0d99629d8797f5c191f.tar.gz
active-drag-system-93acde052369568beaefb0d99629d8797f5c191f.tar.bz2
active-drag-system-93acde052369568beaefb0d99629d8797f5c191f.zip
Raspberry Pi Pico (#12)
* Adding a 90% completed, compilable but untested ADS * Made basic changes to actuator & sensor. Also added motor class * Removed unnecessary .cpp files * Updated sensor & actuator classes, finished ads, added variable time step to kalman filter, set up all tests for future assertions * Relocated 'main' to 'active-drag-system.cpp'. Added more info to README * Removed main.cpp * Added more details to README * Changed some function parameters from pass-by-pointer to pass-by-reference. Also removed the std namespace * Started writing the test cases * Updated the .gitignore file * Removed some files that should be gitignored * Up to date with Jazz's pull request * Test Launch Branch Created; PRU Servo Control with Test Program * Added I2C device class and register IDs for MPL [INCOMPLETE SENSOR IMPLEMENTATION] Needs actual data getting function implementation for both sensors and register IDs for BNO, will implement shortly. * Partial implementation of MPL sensor Added startup method, still needs fleshed out data getters and setters and finished I2C implementation. MOST LIKELY WILL HAVE COMPILATION ISSUES. * *Hypothetically* complete MPL implementation NEEDS HARDWARE TESTING * IMU Header and init() method implementation Needs like, all data handling still lol * Hypothetically functional (Definitely won't compile) * We ball? * Conversion to Raspberry Pi Pico Build System; Removed Beaglebone specific code; Simple blinking example in ADS source file; builds for Pico W * Rearranged build so dependent upon cmake file already existing in pico-sdk; current executable prints current altitude, velocity, and time taken to read and calculate said values; ~320 us to do so * Altimeter interrupt callback for Pad to Boost State; dummy templates for other callbacks with comments describing potential implementation details * Altimeter interrupts relatively finished; need to test with vacuum chamber to verify behavior * Established interrupt pins as pullup and active-low; adjusted callback functions to properly use function pointers; still need to verify interrupt system with vacuum chamber * Removed weird artifact in .gitignore, adjust CMakeLists to auto pull pico sdk, added Dockerfile * added Docker dev container file * modified CMakeLists to auto pull sdk if not already downloaded, add build.sh script, fixed Dockerfile * added bno055 support * changed bno055 lin accel struct to use float instead of double * added bno055 support not tested, but compiles, fixed CMakLists to before I messed with it * added absolute quaternion output from bno055 * Added Euler and aboslute linear accelration * Flash implementation for data logging; each log entry is 32 bytes long * added base pwm functions and started on apogee detection * State machine verified functional with logging capabilities; currently on same core * Ooops missed double define, renamed LOOP_HZ to LOOP_PERIOD; State machine functional after merge still * Simple test program to see servo PWM range; logging with semaphores for safe multithreading * Kalman filters generously provided from various sources for temporary replacement; minimum deployment 30 percent; state machine functionality restored; multithreading logging verified; altimeter broke and replaced * Stop logging on END state; provide deployment function with AGL instead of ASL altitude * Various minimal changes; Flash size from 1MB to 8MB; M1939 to M2500T burn time; pin assignments for new PCB; External Status LED to Internal Status LED --------- Co-authored-by: Jazz Jackson <[email protected]> Co-authored-by: Cian Capacci <[email protected]> Co-authored-by: Gregory Wainer <[email protected]>
Diffstat (limited to 'src/spi_flash.c')
-rw-r--r--src/spi_flash.c267
1 files changed, 267 insertions, 0 deletions
diff --git a/src/spi_flash.c b/src/spi_flash.c
new file mode 100644
index 0000000..32babc5
--- /dev/null
+++ b/src/spi_flash.c
@@ -0,0 +1,267 @@
+/**
+ * Copyright (c) 2020 Raspberry Pi (Trading) Ltd.
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+
+// Example of reading/writing an external serial flash using the PL022 SPI interface
+
+#include "spi_flash.h"
+
+static inline void cs_select(uint cs_pin) {
+ asm volatile("nop \n nop \n nop"); // FIXME
+ gpio_put(cs_pin, 0);
+ asm volatile("nop \n nop \n nop"); // FIXME
+}
+
+static inline void cs_deselect(uint cs_pin) {
+ asm volatile("nop \n nop \n nop"); // FIXME
+ gpio_put(cs_pin, 1);
+ asm volatile("nop \n nop \n nop"); // FIXME
+}
+
+void __not_in_flash_func(flash_read)(spi_inst_t *spi, uint cs_pin, uint32_t addr, uint8_t* dest, size_t len) {
+ cs_select(cs_pin);
+ uint8_t cmdbuf[4] = {
+ FLASH_CMD_READ,
+ addr >> 16,
+ addr >> 8,
+ addr
+ };
+ spi_write_blocking(spi, cmdbuf, 4);
+ spi_read_blocking(spi, 0, dest, len);
+ cs_deselect(cs_pin);
+}
+
+void __not_in_flash_func(flash_write_enable)(spi_inst_t *spi, uint cs_pin) {
+ cs_select(cs_pin);
+ uint8_t cmd = FLASH_CMD_WRITE_EN;
+ spi_write_blocking(spi, &cmd, 1);
+ cs_deselect(cs_pin);
+}
+
+void __not_in_flash_func(flash_wait_done)(spi_inst_t *spi, uint cs_pin) {
+ uint8_t status;
+ do {
+ cs_select(cs_pin);
+ uint8_t buf[2] = {FLASH_CMD_STATUS, 0};
+ spi_write_read_blocking(spi, buf, buf, 2);
+ cs_deselect(cs_pin);
+ status = buf[1];
+ } while (status & FLASH_STATUS_BUSY_MASK);
+}
+
+void __not_in_flash_func(flash_sector_erase)(spi_inst_t *spi, uint cs_pin, uint32_t addr) {
+ uint8_t cmdbuf[4] = {
+ FLASH_CMD_SECTOR_ERASE,
+ addr >> 16,
+ addr >> 8,
+ addr
+ };
+ flash_write_enable(spi, cs_pin);
+ cs_select(cs_pin);
+ spi_write_blocking(spi, cmdbuf, 4);
+ cs_deselect(cs_pin);
+ flash_wait_done(spi, cs_pin);
+}
+
+void __not_in_flash_func(flash_block_erase)(spi_inst_t *spi, uint cs_pin, uint32_t addr) {
+ uint8_t cmdbuf[4] = {
+ FLASH_CMD_BLOCK_ERASE,
+ addr >> 16,
+ addr >> 8,
+ addr
+ };
+ flash_write_enable(spi, cs_pin);
+ cs_select(cs_pin);
+ spi_write_blocking(spi, cmdbuf, 4);
+ cs_deselect(cs_pin);
+ flash_wait_done(spi, cs_pin);
+}
+
+
+void __not_in_flash_func(flash_erase)(spi_inst_t *spi, uint cs_pin) {
+ uint8_t cmdbuf[1] = {
+ FLASH_CMD_CHIP_ERASE
+ };
+ flash_write_enable(spi, cs_pin);
+ cs_select(cs_pin);
+ spi_write_blocking(spi, cmdbuf, 1);
+ cs_deselect(cs_pin);
+ flash_wait_done(spi, cs_pin);
+}
+
+void __not_in_flash_func(flash_page_program)(spi_inst_t *spi, uint cs_pin, uint32_t addr, uint8_t* src) {
+ uint8_t cmdbuf[4] = {
+ FLASH_CMD_PAGE_PROGRAM,
+ addr >> 16,
+ addr >> 8,
+ addr
+ };
+ flash_write_enable(spi, cs_pin);
+ cs_select(cs_pin);
+ spi_write_blocking(spi, cmdbuf, 4);
+ spi_write_blocking(spi, src, FLASH_PAGE_SIZE);
+ cs_deselect(cs_pin);
+ flash_wait_done(spi, cs_pin);
+}
+
+void __not_in_flash_func(flash_write)(spi_inst_t *spi, uint cs_pin, uint32_t addr, uint8_t* src, size_t size) {
+ uint8_t cmdbuf[4] = {
+ FLASH_CMD_PAGE_PROGRAM,
+ addr >> 16,
+ addr >> 8,
+ addr
+ };
+ flash_write_enable(spi, cs_pin);
+ cs_select(cs_pin);
+ spi_write_blocking(spi, cmdbuf, 4);
+ spi_write_blocking(spi, src, size);
+ cs_deselect(cs_pin);
+ flash_wait_done(spi, cs_pin);
+}
+
+void write_entry(uint8_t* data_entry) {
+ flash_read(spi_default, PICO_DEFAULT_SPI_CSN_PIN, base_addr, page_buffer, FLASH_PAGE_SIZE);
+ for (uint16_t i = 0; i < FLASH_PAGE_SIZE; i += PACKET_SIZE) {
+ if (page_buffer[i] == 0xFF) {
+ base_addr += i;
+ break;
+ }
+ if ((i + PACKET_SIZE) == FLASH_PAGE_SIZE) {
+ base_addr += FLASH_PAGE_SIZE;
+ flash_read(spi_default, PICO_DEFAULT_SPI_CSN_PIN, base_addr, page_buffer, FLASH_PAGE_SIZE);
+ i = 0;
+ }
+ }
+ flash_write(spi_default, PICO_DEFAULT_SPI_CSN_PIN, base_addr, data_entry, PACKET_SIZE);
+ base_addr += PACKET_SIZE;
+}
+
+
+#ifdef FLASH_TEST
+ #include "pico/multicore.h"
+ #include <stdint.h>
+ #include <inttypes.h>
+ #include <stdio.h>
+
+
+ void printbuf(uint8_t buf[FLASH_PAGE_SIZE]) {
+ for (int i = 0; i < FLASH_PAGE_SIZE; ++i) {
+ if (i % 16 == 15)
+ printf("%02x\n", buf[i]);
+ else
+ printf("%02x ", buf[i]);
+ }
+ }
+
+
+
+ void core1_entry() {
+ printf("SPI flash example\n");
+
+ // Enable SPI 0 at 1 MHz and connect to GPIOs
+ spi_init(spi_default, 1000 * 1000 * 60);
+ gpio_set_function(PICO_DEFAULT_SPI_RX_PIN, GPIO_FUNC_SPI);
+ gpio_set_function(PICO_DEFAULT_SPI_TX_PIN, GPIO_FUNC_SPI);
+ gpio_set_function(PICO_DEFAULT_SPI_SCK_PIN, GPIO_FUNC_SPI);
+
+ // Chip select is active-low, so we'll initialise it to a driven-high state
+ gpio_init(PICO_DEFAULT_SPI_CSN_PIN);
+ gpio_set_dir(PICO_DEFAULT_SPI_CSN_PIN, GPIO_OUT);
+ gpio_put(PICO_DEFAULT_SPI_CSN_PIN, 1);
+
+ // flash_erase(spi_default, PICO_DEFAULT_SPI_CSN_PIN);
+ // flash_sector_erase(spi_default, PICO_DEFAULT_SPI_CSN_PIN, 0);
+ printf("Before program:\n");
+ flash_read(spi_default, PICO_DEFAULT_SPI_CSN_PIN, 0, page_buffer, FLASH_PAGE_SIZE);
+ printbuf(page_buffer);
+
+ uint8_t entry[PACKET_SIZE];
+
+ printf("Written Data:\n");
+ printf("time (us)\t|\tstate\t|\tdep pcnt\t|\talt (m)\t|\tvel (m/s)\t|\tempty\n");
+ for (uint16_t i = 0; i < 500; i++) {
+ absolute_time_t now = get_absolute_time();
+ uint64_t now_us= to_us_since_boot(now);
+ float altitude = 10.0f * i;
+ float velocity = 5.0f * i;
+ uint8_t deploy_percent = (i*1000) / 200;
+ printf("%"PRIu64"\t|\t%"PRIu8"\t|\t%"PRIu8"\t|\t%4.2f\t|\t%4.2f\t|\tDAWSYN_SCHRAIB\n", now_us, (uint8_t)(i), deploy_percent, altitude, velocity);
+ uint32_t alt_bits = *((uint32_t *)&altitude);
+ uint32_t vel_bits = *((uint32_t *)&velocity);
+ entry[0] = now_us >> 56;
+ entry[1] = now_us >> 48;
+ entry[2] = now_us >> 40;
+ entry[3] = now_us >> 32;
+ entry[4] = now_us >> 24;
+ entry[5] = now_us >> 16;
+ entry[6] = now_us >> 8;
+ entry[7] = now_us;
+ entry[8] = i;
+ entry[9] = deploy_percent;
+ entry[10] = alt_bits >> 24;
+ entry[11] = alt_bits >> 16;
+ entry[12] = alt_bits >> 8;
+ entry[13] = alt_bits;
+ entry[14] = vel_bits >> 24;
+ entry[15] = vel_bits >> 16;
+ entry[16] = vel_bits >> 8;
+ entry[17] = vel_bits;
+ entry[18] = 'D';
+ entry[19] = 'A';
+ entry[20] = 'W';
+ entry[21] = 'S';
+ entry[22] = 'Y';
+ entry[23] = 'N';
+ entry[24] = '_';
+ entry[25] = 'S';
+ entry[26] = 'C';
+ entry[27] = 'H';
+ entry[28] = 'R';
+ entry[29] = 'A';
+ entry[30] = 'I';
+ entry[31] = 'B';
+ write_entry(entry);
+ }
+
+ printf("After program:\n");
+ flash_read(spi_default, PICO_DEFAULT_SPI_CSN_PIN, 0, page_buffer, FLASH_PAGE_SIZE);
+ printbuf(page_buffer);
+
+ printf("\nRead Data:\n");
+ printf("time (us)\t|\tstate\t|\tdep pcnt\t|\talt (m)\t|\tvel (m/s)\t|\tempty\n");
+ for (uint32_t i = 0; i < base_addr; i += PACKET_SIZE) {
+ flash_read(spi_default, PICO_DEFAULT_SPI_CSN_PIN, i, entry, PACKET_SIZE);
+ uint64_t now_us = (((uint64_t)entry[0] << 56) | ((uint64_t)entry[1] << 48) | \
+ ((uint64_t)entry[2] << 40) | ((uint64_t)entry[3] << 32) | \
+ ((uint64_t)entry[4] << 24) | ((uint64_t)entry[5] << 16) | \
+ ((uint64_t)entry[6] << 8) | ((uint64_t)entry[7]));
+
+ uint8_t state = entry[8];
+ uint8_t deploy_percent = entry[9];
+
+ uint32_t alt_bits = (entry[10] << 24) | (entry[11] << 16) | (entry[12] << 8) | (entry[13]);
+ uint32_t vel_bits = (entry[14] << 24) | (entry[15] << 16) | (entry[16] << 8) | (entry[17]);
+ float altitude = *(float *)(&alt_bits);
+ float velocity = *(float *)(&vel_bits);
+ printf("%"PRIu64"\t|\t%"PRIu8"\t|\t%"PRIu8"\t|\t%4.2f\t|\t%4.2f\t|\t%c%c%c%c%c%c%c%c%c%c%c%c%c%c\n", \
+ now_us, state, deploy_percent, altitude, velocity, \
+ entry[18],entry[19],entry[20],entry[21],entry[22],entry[23],entry[24],entry[25],entry[26],entry[27],entry[28],entry[29],entry[30],entry[31]);
+ }
+
+ }
+
+ int main() {
+ // Enable UART so we can print status output
+ stdio_init_all();
+ getchar();
+
+ multicore_launch_core1(core1_entry);
+
+ while (1) {
+ tight_loop_contents();
+ }
+
+ }
+#endif