|
1 | 1 | #include "app/state.h" |
2 | | - |
| 2 | +#include "app/utils/devices.h" |
| 3 | +#include "app/utils/packets.h" |
| 4 | +#include "devices/imu.h" |
| 5 | +#include "devices/gnss.h" |
| 6 | +#include "devices/barometer.h" |
| 7 | +#include "devices/actuator.h" |
| 8 | +#include "devices/adc.h" |
| 9 | +#include "devices/magnetometer.h" |
| 10 | +#include "devices/temperature.h" |
| 11 | +#include "devices/radio.h" |
| 12 | +#include "peripheral/spi.h" |
| 13 | +#include "peripheral/errc.h" |
| 14 | +#include "peripheral/systick.h" |
3 | 15 |
|
4 | 16 | // -- ( ) FIRING: Deliver propelants to manifold |
5 | 17 | // ~~~ ( ) Open main propelant valves |
6 | 18 | // ~~~ ( ) Execute control loop for engine burn |
7 | 19 | // ~~~ ( ) Send and record telemetry |
8 | 20 |
|
| 21 | +#define ADC_PACKET_INDEX 0x00 |
| 22 | +#define LOOP_PERIOD_MS 100U |
| 23 | +#define COMM_PERIOD_MS 1000U |
| 24 | +#define SENSOR_CYCLES_PER_COMM (COMM_PERIOD_MS / LOOP_PERIOD_MS) |
| 25 | + |
| 26 | +/** Buffers for packets */ |
| 27 | +static uint8_t gnss_packet[PACKET_GNSS_SIZE]; |
| 28 | +static uint8_t sensor_packet[PACKET_SENSOR_SIZE]; |
| 29 | +static uint8_t adc_packet[PACKET_ADC_MAX_SIZE]; |
| 30 | +static uint8_t state_packet[PACKET_STATE_MAX_SIZE]; |
| 31 | +static uint8_t comm_packet[PACKET_COMM_MAX_SIZE]; |
| 32 | +static uint8_t rx_packet[PACKET_RX_MAX_SIZE]; |
| 33 | +static comm_packet_t uplink_comm; |
| 34 | + |
| 35 | +static uint32_t elapsed_ms; |
| 36 | +static uint32_t processor_time_ms; |
| 37 | +static uint16_t ping_id; |
| 38 | + |
9 | 39 | bool fire_state_init(){ |
| 40 | + enum ti_errc_t errc; |
| 41 | + |
| 42 | + radio_init(&radio_dev, &radio_spi_config, &radio_config, &errc); |
| 43 | + if (errc && errc != TI_ERRC_NONE) { |
| 44 | + TI_SET_ERRC(&errc, errc, "Failed to initialize radio"); |
| 45 | + } |
| 46 | + errc = imu_init(&imu_dev1); |
| 47 | + if (errc && errc != TI_ERRC_NONE) { |
| 48 | + TI_SET_ERRC(&errc, errc, "Failed to initialize IMU 1"); |
| 49 | + } |
| 50 | + errc = imu_init(&imu_dev2); |
| 51 | + if (errc && errc != TI_ERRC_NONE) { |
| 52 | + TI_SET_ERRC(&errc, errc, "Failed to initialize IMU 2"); |
| 53 | + } |
| 54 | + gnss_init(&gnss_dev, &errc); |
| 55 | + if (errc && errc != TI_ERRC_NONE) { |
| 56 | + TI_SET_ERRC(&errc, errc, "Failed to initialize GNSS"); |
| 57 | + } |
| 58 | + barometer_init(&barometer_dev1, &errc); |
| 59 | + if (errc && errc != TI_ERRC_NONE) { |
| 60 | + TI_SET_ERRC(&errc, errc, "Failed to initialize barometer 1"); |
| 61 | + } |
| 62 | + barometer_init(&barometer_dev2, &errc); |
| 63 | + if (errc && errc != TI_ERRC_NONE) { |
| 64 | + TI_SET_ERRC(&errc, errc, "Failed to initialize barometer 2"); |
| 65 | + } |
| 66 | + temperature_init(&temperature_dev1, &errc); |
| 67 | + if (errc && errc != TI_ERRC_NONE) { |
| 68 | + TI_SET_ERRC(&errc, errc, "Failed to initialize temperature 1"); |
| 69 | + } |
| 70 | + temperature_init(&temperature_dev2, &errc); |
| 71 | + if (errc && errc != TI_ERRC_NONE) { |
| 72 | + TI_SET_ERRC(&errc, errc, "Failed to initialize temperature 2"); |
| 73 | + } |
| 74 | + magnetometer_init(&magnetometer_dev1); |
| 75 | + if (errc && errc != TI_ERRC_NONE) { |
| 76 | + TI_SET_ERRC(&errc, errc, "Failed to initialize magnetometer 1"); |
| 77 | + } |
| 78 | + magnetometer_init(&magnetometer_dev2); |
| 79 | + if (errc && errc != TI_ERRC_NONE) { |
| 80 | + TI_SET_ERRC(&errc, errc, "Failed to initialize magnetometer 2"); |
| 81 | + } |
| 82 | + |
| 83 | + adc_init(&adc_dev, &errc); |
| 84 | + if (errc && errc != TI_ERRC_NONE) { |
| 85 | + TI_SET_ERRC(&errc, errc, "Failed to initialize ADC"); |
| 86 | + } |
10 | 87 |
|
11 | 88 | return 1; |
12 | 89 | } |
13 | 90 |
|
14 | 91 | int fire_state_run(){ |
| 92 | + static const uint8_t comm_tags[] = {0}; |
| 93 | + |
| 94 | + for (uint8_t cycle = 0; cycle < SENSOR_CYCLES_PER_COMM; cycle++) { |
| 95 | + enum ti_errc_t errc; |
| 96 | + size_t adc_packet_len; |
| 97 | + size_t state_packet_len; |
| 98 | + |
| 99 | + gnss_pvt_t gnss_pvt; |
| 100 | + gnss_get_pvt(&gnss_dev, &gnss_pvt, &errc); |
| 101 | + if (errc && errc != TI_ERRC_NONE) { |
| 102 | + TI_SET_ERRC(&errc, errc, "Failed to get GNSS PVT data"); |
| 103 | + } |
| 104 | + |
| 105 | + build_gnss_packet(&gnss_pvt, gnss_packet, sizeof(gnss_packet), &errc); |
| 106 | + send_packet_radio_flash(&radio_dev, gnss_packet, sizeof(gnss_packet), &errc); |
| 107 | + if (errc && errc != TI_ERRC_NONE) { |
| 108 | + TI_SET_ERRC(&errc, errc, "Failed to transmit GNSS packet"); |
| 109 | + } |
| 110 | + |
| 111 | + struct imu_result imu_result1; |
| 112 | + struct imu_result imu_result2; |
| 113 | + errc = imu_transfer(&imu_dev1, &imu_result1); |
| 114 | + if (errc && errc != TI_ERRC_NONE) { |
| 115 | + TI_SET_ERRC(&errc, errc, "Failed to transfer IMU 1 data"); |
| 116 | + } |
| 117 | + errc = imu_transfer(&imu_dev2, &imu_result2); |
| 118 | + if (errc && errc != TI_ERRC_NONE) { |
| 119 | + TI_SET_ERRC(&errc, errc, "Failed to transfer IMU 2 data"); |
| 120 | + } |
| 121 | + |
| 122 | + temperature_result_t temperature_result1; |
| 123 | + temperature_result_t temperature_result2; |
| 124 | + temperature_read_temp(&temperature_dev1, &temperature_result1, &errc); |
| 125 | + if (errc && errc != TI_ERRC_NONE) { |
| 126 | + TI_SET_ERRC(&errc, errc, "Failed to read temperature 1 data"); |
| 127 | + } |
| 128 | + temperature_read_temp(&temperature_dev2, &temperature_result2, &errc); |
| 129 | + if (errc && errc != TI_ERRC_NONE) { |
| 130 | + TI_SET_ERRC(&errc, errc, "Failed to read temperature 2 data"); |
| 131 | + } |
| 132 | + barometer_result_t barometer_result1; |
| 133 | + barometer_result_t barometer_result2; |
| 134 | + barometer_result1 = get_barometer_data(&barometer_dev1, &errc); |
| 135 | + if (errc && errc != TI_ERRC_NONE) { |
| 136 | + TI_SET_ERRC(&errc, errc, "Failed to read barometer 1 data"); |
| 137 | + } |
| 138 | + barometer_result2 = get_barometer_data(&barometer_dev2, &errc); |
| 139 | + if (errc && errc != TI_ERRC_NONE) { |
| 140 | + TI_SET_ERRC(&errc, errc, "Failed to read barometer 2 data"); |
| 141 | + } |
| 142 | + struct magnetometer_result_t magnetometer_result1; |
| 143 | + struct magnetometer_result_t magnetometer_result2; |
| 144 | + errc = magnetometer_read(&magnetometer_dev1, &magnetometer_result1); |
| 145 | + if (errc && errc != TI_ERRC_NONE) { |
| 146 | + TI_SET_ERRC(&errc, errc, "Failed to transfer magnetometer 1 data"); |
| 147 | + } |
| 148 | + errc = magnetometer_read(&magnetometer_dev2, &magnetometer_result2); |
| 149 | + if (errc && errc != TI_ERRC_NONE) { |
| 150 | + TI_SET_ERRC(&errc, errc, "Failed to transfer magnetometer 2 data"); |
| 151 | + } |
| 152 | + |
| 153 | + build_sensor_packet(&imu_result1, |
| 154 | + &imu_result2, |
| 155 | + &barometer_result1, |
| 156 | + &barometer_result2, |
| 157 | + &magnetometer_result1, |
| 158 | + &magnetometer_result2, |
| 159 | + &temperature_result1, |
| 160 | + &temperature_result2, |
| 161 | + sensor_packet, |
| 162 | + sizeof(sensor_packet), |
| 163 | + &errc); |
| 164 | + send_packet_radio_flash(&radio_dev, sensor_packet, sizeof(sensor_packet), &errc); |
| 165 | + if (errc && errc != TI_ERRC_NONE) { |
| 166 | + TI_SET_ERRC(&errc, errc, "Failed to transmit sensor data packet"); |
| 167 | + } |
| 168 | + |
| 169 | + build_adc_packet(adc_channels, adc_channel_count, ADC_PACKET_INDEX, adc_packet, sizeof(adc_packet), &errc); |
| 170 | + if (errc && errc != TI_ERRC_NONE) { |
| 171 | + TI_SET_ERRC(&errc, errc, "Failed to build ADC packet"); |
| 172 | + } |
| 173 | + |
| 174 | + adc_packet_len = (size_t)(10U + (3U * adc_channel_count)); |
| 175 | + send_packet_radio_flash(&radio_dev, adc_packet, adc_packet_len, &errc); |
| 176 | + if (errc && errc != TI_ERRC_NONE) { |
| 177 | + TI_SET_ERRC(&errc, errc, "Failed to transmit ADC packet"); |
| 178 | + } |
| 179 | + |
| 180 | + build_state_packet(valve_states, |
| 181 | + VALVE_COUNT, |
| 182 | + servo_states, |
| 183 | + SERVO_COUNT, |
| 184 | + state_packet, |
| 185 | + sizeof(state_packet), |
| 186 | + &errc); |
| 187 | + state_packet_len = (size_t)(9U + ((VALVE_COUNT + 7U) / 8U) + (2U * SERVO_COUNT)); |
| 188 | + send_packet_radio_flash(&radio_dev, state_packet, state_packet_len, &errc); |
| 189 | + if (errc && errc != TI_ERRC_NONE) { |
| 190 | + TI_SET_ERRC(&errc, errc, "Failed to transmit state packet"); |
| 191 | + } |
| 192 | + |
| 193 | + elapsed_ms += LOOP_PERIOD_MS; |
| 194 | + processor_time_ms += LOOP_PERIOD_MS; |
| 195 | + systick_delay(LOOP_PERIOD_MS); |
| 196 | + } |
| 197 | + |
| 198 | + { |
| 199 | + enum ti_errc_t errc; |
| 200 | + size_t comm_packet_len; |
| 201 | + |
| 202 | + build_comm_packet(ping_id, |
| 203 | + 0, |
| 204 | + processor_time_ms, |
| 205 | + uplink_comm.command_id, |
| 206 | + 0, |
| 207 | + comm_tags, |
| 208 | + 1, |
| 209 | + comm_packet, |
| 210 | + sizeof(comm_packet), |
| 211 | + &comm_packet_len, |
| 212 | + &errc); |
| 213 | + if (errc && errc != TI_ERRC_NONE) { |
| 214 | + TI_SET_ERRC(&errc, errc, "Failed to build comm packet"); |
| 215 | + } |
| 216 | + |
| 217 | + send_packet_radio_flash(&radio_dev, comm_packet, comm_packet_len, &errc); |
| 218 | + if (errc && errc != TI_ERRC_NONE) { |
| 219 | + TI_SET_ERRC(&errc, errc, "Failed to transmit comm packet"); |
| 220 | + } |
| 221 | + |
| 222 | + receive_uplink_comm_packet(&radio_dev, rx_packet, sizeof(rx_packet), &uplink_comm, &errc); |
| 223 | + if (errc && errc != TI_ERRC_NONE) { |
| 224 | + TI_SET_ERRC(&errc, errc, "Failed to receive or parse uplink comm packet"); |
| 225 | + } |
| 226 | + } |
| 227 | + |
| 228 | + ping_id++; |
| 229 | + elapsed_ms = 0; |
| 230 | + |
15 | 231 | return 6; |
16 | 232 | } |
17 | 233 |
|
|
0 commit comments