From cb47f6fe368a0b3e22291396ef09d770e7364e71 Mon Sep 17 00:00:00 2001
From: Sadeep Madurange <sadeep@asciimx.com>
Date: Wed, 7 May 2025 15:58:55 +0800
Subject: wip: enroll.

---
 fpm.c  |  6 ++---
 fpm.h  |  8 +++----
 main.c |  7 +-----
 r503.c | 86 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 r503.h | 14 +++++++++++
 5 files changed, 106 insertions(+), 15 deletions(-)

diff --git a/fpm.c b/fpm.c
index a58045e..84a64fe 100644
--- a/fpm.c
+++ b/fpm.c
@@ -152,7 +152,7 @@ uint8_t fpm_init(void)
 	return check_pwd();
 }
 
-uint8_t fpm_getcfg(struct fpm_cfg *cfg)
+uint8_t fpm_get_cfg(struct fpm_cfg *cfg)
 {
 	uint16_t n;
 	uint8_t buf[MAXPDLEN];
@@ -179,7 +179,7 @@ uint8_t fpm_getcfg(struct fpm_cfg *cfg)
 	return 0;
 }
 
-uint8_t fpm_setpwd(uint32_t pwd)
+uint8_t fpm_set_pwd(uint32_t pwd)
 {
 	uint16_t n;
 	uint8_t buf[MAXPDLEN];
@@ -249,7 +249,7 @@ uint8_t fpm_match(void)
 	return buf[0] == OK;
 }
 
-uint8_t fpm_delete_all(void)
+uint8_t fpm_clear_db(void)
 {
 	uint16_t n;
 	uint8_t buf[MAXPDLEN];
diff --git a/fpm.h b/fpm.h
index f165958..0b9d303 100644
--- a/fpm.h
+++ b/fpm.h
@@ -15,16 +15,16 @@ struct fpm_cfg {
 
 uint8_t fpm_init(void);
 
-uint8_t fpm_getcfg(struct fpm_cfg *cfg);
+uint8_t fpm_get_cfg(struct fpm_cfg *cfg);
 
-uint8_t fpm_setpwd(uint32_t pwd);
+uint8_t fpm_set_pwd(uint32_t pwd);
 
-uint16_t fpm_getcount(void);
+uint16_t fpm_get_count(void);
 
 uint8_t fpm_enroll(void);
 
 uint8_t fpm_match(void);
 
-uint8_t fpm_delete_all(void);
+uint8_t fpm_clear_db(void);
 
 #endif  /* FPM_H */
diff --git a/main.c b/main.c
index ba5f6e6..5f51edb 100644
--- a/main.c
+++ b/main.c
@@ -40,12 +40,7 @@ int main(void)
 	
 	fpm_init();
 	if (fpm_get_count() == 0) {
-		for (;;) {
-			fpm_led_on(PURPLE);
-			_delay_ms(500);
-			fpm_led_off();
-			_delay_ms(500);
-		}
+		fpm_enroll();
 	}
 
     while (1) 
diff --git a/r503.c b/r503.c
index 434e3fe..500cf5c 100644
--- a/r503.c
+++ b/r503.c
@@ -110,6 +110,20 @@ bad_pkt:
 	return;
 }
 
+static inline void aura_breathe(COLOR color)
+{
+	uint16_t n;
+	uint8_t buf[5];
+	
+	buf[0] = 0x35;
+	buf[1] = 0x01;
+	buf[2] = 0x20;
+	buf[3] = color;
+	buf[4] = 0x00;
+
+	send(0x01, buf, 5);	
+}
+
 static inline uint8_t check_pwd(void)
 {
 	unsigned int n;
@@ -126,6 +140,38 @@ static inline uint8_t check_pwd(void)
 	return buf[0] == OK;
 }
 
+static inline uint8_t scan(void)
+{
+	uint16_t n, retries;
+	uint8_t buf[MAXPDLEN], led;
+	
+	retries = 0;
+	fpm_led_on(PURPLE);
+	led = 1;
+	_delay_ms(500);
+
+	do {
+		buf[0] = 0x01;
+		send(0x01, buf, 1);
+		recv(buf, &n);
+		if (buf[0] != OK) {
+			retries++;
+			if (led) {
+				fpm_led_off();
+				led = 0;
+			} else {
+				fpm_led_on(PURPLE);
+				led = 1;
+			}
+			_delay_ms(500);
+		}
+	} while(buf[0] != OK && retries < 100);
+
+	fpm_led_off();
+	led = 0;
+	return buf[0] == OK;
+}
+
 void fpm_led_on(COLOR color)
 {
 	uint16_t n;
@@ -138,7 +184,6 @@ void fpm_led_on(COLOR color)
 	buf[4] = 0x00;
 
 	send(0x01, buf, 5);	
-	recv(buf, &n);
 }
 
 void fpm_led_off(void)
@@ -153,7 +198,6 @@ void fpm_led_off(void)
 	buf[4] = 0x00;
 
 	send(0x01, buf, 5);	
-	recv(buf, &n);
 }
 
 uint8_t fpm_init(void)
@@ -172,6 +216,33 @@ uint8_t fpm_init(void)
 	return check_pwd();
 }
 
+uint8_t fpm_get_cfg(struct fpm_cfg *cfg)
+{
+	uint16_t n;
+	uint8_t buf[MAXPDLEN];
+
+	buf[0] = 0x0F;
+	send(0x01, buf, 1);
+	recv(buf, &n);
+
+	if (buf[0] == OK && n >= 17) {
+		cfg->status = ((uint16_t)buf[1] << 8) | buf[2];
+		cfg->sysid = ((uint16_t)buf[3] << 8) | buf[4];
+		cfg->cap = ((uint16_t)buf[5] << 8) | buf[6];
+		cfg->sec_level = ((uint16_t)buf[7] << 8) | buf[8];
+		cfg->addr[0] = buf[9];
+		cfg->addr[1] = buf[10];
+		cfg->addr[2] = buf[11];
+		cfg->addr[3] = buf[12];
+		cfg->pkt_size = ((uint16_t)buf[13] << 8) | buf[14];
+		cfg->pkt_size = 1 << (cfg->pkt_size + 5); 
+		cfg->baud = (((uint16_t)buf[15] << 8) | buf[16]);
+
+		return 1;
+	}
+	return 0;
+}
+
 uint8_t fpm_clear_db(void)
 {
 	uint16_t n;
@@ -200,3 +271,14 @@ uint16_t fpm_get_count(void)
 	}
 	return count;
 }
+
+uint8_t fpm_enroll(void)
+{
+	uint16_t n, retries;
+	uint8_t buf[MAXPDLEN], led;
+	
+	if (!scan())	
+		return 0;
+
+	return 1;
+}
diff --git a/r503.h b/r503.h
index e12b8f2..5582552 100644
--- a/r503.h
+++ b/r503.h
@@ -3,6 +3,16 @@
 
 #include <stdint.h>
 
+struct fpm_cfg {
+	uint16_t status;
+	uint16_t sysid;
+	uint16_t cap;
+	uint16_t sec_level;
+	uint8_t addr[4];
+	uint16_t pkt_size;
+	uint16_t baud;
+};
+
 typedef enum {
 	RED = 0x01,
 	BLUE = 0x02,
@@ -11,6 +21,8 @@ typedef enum {
 
 uint8_t fpm_init(void);
 
+uint8_t fpm_get_cfg(struct fpm_cfg *cfg);
+
 void fpm_led_on(COLOR color);
 
 void fpm_led_off(void);
@@ -19,4 +31,6 @@ uint8_t fpm_clear_db(void);
 
 uint16_t fpm_get_count(void);
 
+uint8_t fpm_enroll(void);
+
 #endif /* FPM_R50_H */
-- 
cgit v1.2.3