1
0
Fork 0

Arduino: Add full Arduino code

This is the full working Arduino code, for writing and loading.

This shouldn not be used by students.
This commit is contained in:
Teo-CD 2023-11-12 20:58:05 +00:00
parent f34d81ad87
commit dfe7b6b5d0
2 changed files with 215 additions and 0 deletions

View file

@ -0,0 +1,117 @@
#include <Wire.h>
#include <inttypes.h>
static const int eeprom_addr = 0b1010000;
// Send a two bytes address via I²C which selects an address within the EEPROM.
void write_addr(uint16_t addr) {
Wire.write((uint8_t)(addr >> 8));
Wire.write((uint8_t)addr);
}
void write_page(uint8_t page[128], uint16_t page_addr) {
uint8_t bytes_written = 0;
for (int i = 0; i < 5; i++) {
int ret;
int write_bytes = min(128 - bytes_written, 30);
do {
Wire.beginTransmission(eeprom_addr);
write_addr(page_addr + bytes_written);
Wire.write(page + bytes_written, write_bytes);
ret = Wire.endTransmission();
delay(5);
} while (ret != 0);
bytes_written += write_bytes;
}
}
bool check_sumcomplement(uint8_t data[128], uint16_t checksum) {
uint16_t sum = 0;
for (int i = 0; i<128; i += 2) {
sum += (data[i+1] << 8) | data[i];
}
sum += checksum;
return !sum;
}
volatile uint8_t pageData[128] = {};
volatile uint16_t pageChecksum = 0;
volatile uint16_t pageAddr = 0;
volatile bool pageOk = false;
volatile bool reicivingCart = false;
volatile uint32_t last_page_update = 0;
void setup() {
Serial.begin(115200);
pinMode(LED_BUILTIN, OUTPUT);
Wire.begin();
}
void loop() {
if (reicivingCart && millis() - last_page_update > 1000) {
Serial.print("TO");
reicivingCart = false;
pageOk = false;
pageAddr = 0;
digitalWrite(LED_BUILTIN, HIGH);
delay(500);
digitalWrite(LED_BUILTIN, LOW);
delay(500);
digitalWrite(LED_BUILTIN, HIGH);
delay(500);
digitalWrite(LED_BUILTIN, LOW);
}
if (pageOk) {
pageOk = false;
bool pageValid = check_sumcomplement((uint8_t *)pageData, pageChecksum);
Serial.print(pageValid ? "ACK" : "NAK");
if (pageValid) {
write_page(pageData, pageAddr);
pageAddr += 128;
}
/* for (int i = 0; i<128; i++) {
if (i%16 == 0)
Serial.println();
Serial.print(pageData[i], HEX);
Serial.print(" ");
}
Serial.println();
Serial.println(pageChecksum, HEX);*/
Serial.print("OK");
}
}
void serialEvent() {
static size_t read = 0;
digitalWrite(LED_BUILTIN, HIGH);
if (Serial.available() == 5) {
char control[4] = {};
Serial.readBytesUntil('\n', control, 4);
Serial.read(); /* Drop the '\n' */
if (!strncmp(control, "CART", 4)) {
reicivingCart = true;
Serial.print("OK");
} else if (!strncmp(control, "DONE", 4)) {
reicivingCart = false;
} else {
Serial.print("KO");
}
pageOk = false;
pageAddr = 0;
read = 0;
}
while (reicivingCart && Serial.available() && read < 128) {
read += Serial.readBytes((char *)pageData+read, min(128-read, 64));
}
if (reicivingCart && read >= 128) {
pageOk = true;
Serial.readBytes((char*)&pageChecksum, 2);
read = 0;
}
digitalWrite(LED_BUILTIN, LOW);
last_page_update = millis();
}

View file

@ -0,0 +1,98 @@
#include <Wire.h>
#include <inttypes.h>
// Arduino constants
static constexpr uint8_t i2c_buff_size = 32;
static constexpr uint8_t serial_buff_size = 64;
// EEPROM constants
static constexpr int page_count = 512;
static constexpr int page_size = 128;
static constexpr int i2c_buff_per_page = page_size / i2c_buff_size;
static constexpr uint8_t eeprom_addr = 0b1010000;
uint8_t pageData[page_size] = {};
uint16_t pageAddr = 0;
// Send a two bytes address via I²C which selects an address within the EEPROM.
void write_addr(uint16_t addr) {
Wire.write((uint8_t)(addr >> 8));
Wire.write((uint8_t)addr);
}
// When the receiver sums on its side, it can just add this sum and check that is 0.
uint16_t sum_complement(const uint8_t data[page_size]) {
uint16_t sum = 0;
for (int i = 0; i<page_size; i += 2) {
sum += (data[i+1] << 8) | data[i];
}
return -sum;
}
void setup() {
Serial.begin(115200);
pinMode(LED_BUILTIN, OUTPUT);
Wire.begin();
}
/*
* Read data from the Serial port until a full message is read.
* This function is blocking and will not return early, use with caution.
* Use '\n' as a delimiter for a complete message, making it exit
* early if it is encountered before the count is done (invalid message).
*/
void read_message(char *buffer, const uint8_t match_len) {
uint8_t read_bytes = 0;
do {
// readBytesUntil should always be positive or zero, despite the documentation.
read_bytes += Serial.readBytesUntil('\n', buffer + read_bytes, match_len - read_bytes);
} while(read_bytes < match_len || Serial.peek() == '\n');
if (Serial.peek() == '\n')
Serial.read(); // Consume the delimiter if it is there.
}
void loop() {
char ret[4];
// Wait until we receive the start message.
do {
memset(ret, 0, 4);
read_message(ret, 2);
} while (strncmp(ret, "GO", 2) != 0);
// Reset the EEPROM internal address to 0. It autoincrements after.
Wire.beginTransmission(eeprom_addr);
write_addr(0);
Wire.endTransmission(false);
digitalWrite(LED_BUILTIN, HIGH);
/*
* We don't know the size of the cartridge, and it could be the whole EEPROM.
* Read and send all the pages (64kiB), it should not impact the final PNG.
*/
for (int i = 0; i < page_count; i++) {
memset(pageData, 0, page_size);
// Arduino I2C buffer is 32 bytes, so we need multiple requests per page.
for (int j = 0; j < i2c_buff_per_page; j++) {
Wire.requestFrom(eeprom_addr, i2c_buff_size);
for (int k = 0; k < i2c_buff_size; k++) {
pageData[i2c_buff_size*j + k] = Wire.read();
}
}
// Try to send the full page via serial until the recipient ACKs.
uint16_t page_checksum = sum_complement(pageData);
do {
// The internal Serial buffer is smaller than a page, so send in bursts.
size_t written = 0;
while (written < page_size) {
written += Serial.write(pageData + written, min(page_size - written, serial_buff_size));
}
Serial.write((uint8_t*)&page_checksum, 2);
memset(ret, 0, 4);
read_message(ret, 3);
} while(strncmp(ret, "ACK", 3) != 0);
}
digitalWrite(LED_BUILTIN, LOW);
}