summaryrefslogtreecommitdiff
path: root/components/screen
diff options
context:
space:
mode:
Diffstat (limited to 'components/screen')
-rw-r--r--components/screen/CMakeLists.txt3
-rw-r--r--components/screen/screen.c183
-rw-r--r--components/screen/screen.h55
3 files changed, 241 insertions, 0 deletions
diff --git a/components/screen/CMakeLists.txt b/components/screen/CMakeLists.txt
new file mode 100644
index 0000000..df343b9
--- /dev/null
+++ b/components/screen/CMakeLists.txt
@@ -0,0 +1,3 @@
+idf_component_register(SRCS "screen.c"
+ INCLUDE_DIRS "."
+ PRIV_REQUIRES esp_driver_gpio esp_driver_i2c) \ No newline at end of file
diff --git a/components/screen/screen.c b/components/screen/screen.c
new file mode 100644
index 0000000..99f74b0
--- /dev/null
+++ b/components/screen/screen.c
@@ -0,0 +1,183 @@
+#include "screen.h"
+
+esp_err_t write_reg_bit_16(i2c_master_dev_handle_t mcp, uint8_t reg, bool val, uint8_t bit)
+{
+ uint8_t buf[2];
+
+ // Read the current value from the register
+ buf[0] = reg; // register address
+ ESP_ERROR_CHECK(i2c_master_transmit(mcp, &buf[0], sizeof(uint8_t), -1));
+ ESP_ERROR_CHECK(i2c_master_receive(mcp, &buf[1], sizeof(uint8_t), -1));
+ // Update the value in the buffer
+
+ buf[1] = (buf[1] & ~BV(bit)) | (val ? BV(bit) : 0);
+
+ // Write the updated value back to the register
+ buf[0] = reg; // Send register address
+ ESP_ERROR_CHECK(i2c_master_transmit(mcp, &buf[0], sizeof(uint16_t), -1));
+
+ return ESP_OK;
+}
+
+esp_err_t mcp_set_mode(i2c_master_dev_handle_t mcp, uint8_t reg, uint8_t pin, bool mode)
+{
+ return write_reg_bit_16(mcp, reg, mode, pin);
+}
+
+esp_err_t mcp_set_level(i2c_master_dev_handle_t mcp, uint8_t reg, uint8_t pin, bool val)
+{
+ return write_reg_bit_16(mcp, reg, val, pin);
+}
+
+void configure_screen(struct Screen* screen)
+{
+ i2c_master_bus_config_t i2c_bus_config = {
+ .clk_source = I2C_CLK_SRC_DEFAULT,
+ .i2c_port = 0,
+ .sda_io_num = SDA_PIN,
+ .scl_io_num = SCL_PIN,
+ .glitch_ignore_cnt = 7,
+ .flags.enable_internal_pullup = true,
+ };
+ ESP_ERROR_CHECK(i2c_new_master_bus(&i2c_bus_config, &(screen->i2c_bus)));
+
+ i2c_device_config_t mcp_cfg = {
+ .dev_addr_length = I2C_ADDR_BIT_LEN_7,
+ .device_address = MCP_I2C_ADDR,
+ .scl_speed_hz = 1000000,
+ };
+ ESP_ERROR_CHECK(i2c_master_bus_add_device(screen->i2c_bus, &mcp_cfg, &(screen->mcp)));
+
+ printf("I2C MCP Configured!\n");
+
+ screen->nixies[0].digit = 0;
+ screen->nixies[0].reg = REG_GPIOA;
+ screen->nixies[0].C = N0_P0;
+ screen->nixies[0].B = N0_P1;
+ screen->nixies[0].D = N0_P2;
+ screen->nixies[0].A = N0_P3;
+ screen->nixies[1].digit = 0;
+ screen->nixies[1].reg = REG_GPIOA;
+ screen->nixies[1].C = N1_P0;
+ screen->nixies[1].B = N1_P1;
+ screen->nixies[1].D = N1_P2;
+ screen->nixies[1].A = N1_P3;
+ screen->nixies[2].digit = 0;
+ screen->nixies[2].reg = REG_GPIOB;
+ screen->nixies[2].C = N2_P0;
+ screen->nixies[2].B = N2_P1;
+ screen->nixies[2].D = N2_P2;
+ screen->nixies[2].A = N2_P3;
+ screen->nixies[3].digit = 0;
+ screen->nixies[3].reg = REG_GPIOB;
+ screen->nixies[3].C = N3_P0;
+ screen->nixies[3].B = N3_P1;
+ screen->nixies[3].D = N3_P2;
+ screen->nixies[3].A = N3_P3;
+
+ for (uint8_t i = 0; i <= 3; ++i)
+ {
+ mcp_set_mode(screen->mcp, screen->nixies[i].reg, screen->nixies[i].D, MCP_GPIO_OUTPUT);
+ mcp_set_mode(screen->mcp, screen->nixies[i].reg, screen->nixies[i].C, MCP_GPIO_OUTPUT);
+ mcp_set_mode(screen->mcp, screen->nixies[i].reg, screen->nixies[i].B, MCP_GPIO_OUTPUT);
+ mcp_set_mode(screen->mcp, screen->nixies[i].reg, screen->nixies[i].A, MCP_GPIO_OUTPUT);
+ }
+
+
+ printf("MCP GPIO Configured!\n");
+ return;
+}
+
+void set_digit(i2c_master_dev_handle_t mcp, struct Nixie* n)
+{
+ switch (n->digit) {
+ case 0:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 0);
+ break;
+ case 1:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ case 2:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 1);
+ mcp_set_level(mcp, n->reg, n->A, 0);
+ break;
+ case 3:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 1);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ case 4:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 1);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 0);
+ break;
+ case 5:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 1);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ case 6:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 1);
+ mcp_set_level(mcp, n->reg, n->B, 1);
+ mcp_set_level(mcp, n->reg, n->A, 0);
+ break;
+ case 7:
+ mcp_set_level(mcp, n->reg, n->D, 0);
+ mcp_set_level(mcp, n->reg, n->C, 1);
+ mcp_set_level(mcp, n->reg, n->B, 1);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ case 8:
+ mcp_set_level(mcp, n->reg, n->D, 1);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 0);
+ break;
+ case 9:
+ mcp_set_level(mcp, n->reg, n->D, 1);
+ mcp_set_level(mcp, n->reg, n->C, 0);
+ mcp_set_level(mcp, n->reg, n->B, 0);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ default:
+ mcp_set_level(mcp, n->reg, n->D, 1);
+ mcp_set_level(mcp, n->reg, n->C, 1);
+ mcp_set_level(mcp, n->reg, n->B, 1);
+ mcp_set_level(mcp, n->reg, n->A, 1);
+ break;
+ return;
+ }
+}
+
+void test_screen(struct Screen* screen)
+{
+ while(1){
+ for (uint8_t i = 0; i <= 9; ++i) {
+ for (uint8_t j = 0; j < 4; ++j) {
+ screen->nixies[j].digit = (i + j) % 10; // Update the digit cyclically
+ set_digit(screen->mcp, &(screen->nixies[j])); // Call set_digit function
+ }
+ sleep(1);
+ }
+ }
+}
+
+void display(struct Screen* screen, char digits[4])
+{
+ for (uint8_t i = 0; i < 4; ++i) {
+ screen->nixies[i].digit = digits[i] - '0';
+ set_digit(screen->mcp, &(screen->nixies[i])); // Call set_digit function
+ }
+} \ No newline at end of file
diff --git a/components/screen/screen.h b/components/screen/screen.h
new file mode 100644
index 0000000..b8b951c
--- /dev/null
+++ b/components/screen/screen.h
@@ -0,0 +1,55 @@
+#ifndef SC_H
+#define SC_H
+
+#include <stdio.h>
+#include <unistd.h>
+#include "driver/gpio.h"
+#include "driver/i2c_master.h"
+
+#define MCP_I2C_ADDR CONFIG_MCP_I2C_ADDR
+#define SDA_PIN CONFIG_SDA_PIN
+#define SCL_PIN CONFIG_SCL_PIN
+#define N0_P0 CONFIG_N0_P0
+#define N0_P1 CONFIG_N0_P1
+#define N0_P2 CONFIG_N0_P2
+#define N0_P3 CONFIG_N0_P3
+#define N1_P0 CONFIG_N1_P0
+#define N1_P1 CONFIG_N1_P1
+#define N1_P2 CONFIG_N1_P2
+#define N1_P3 CONFIG_N1_P3
+#define N2_P0 CONFIG_N2_P0
+#define N2_P1 CONFIG_N2_P1
+#define N2_P2 CONFIG_N2_P2
+#define N2_P3 CONFIG_N2_P3
+#define N3_P0 CONFIG_N3_P0
+#define N3_P1 CONFIG_N3_P1
+#define N3_P2 CONFIG_N3_P2
+#define N3_P3 CONFIG_N3_P3
+#define MCP_GPIO_OUTPUT 0
+#define REG_IODIRA 0x00
+#define REG_IODIRB 0x01
+#define REG_GPIOA 0x12
+#define REG_GPIOB 0x13
+#define BV(x) (1 << (x))
+
+struct Nixie
+{
+ uint8_t digit;
+ uint8_t reg;
+ uint8_t D;
+ uint8_t C;
+ uint8_t B;
+ uint8_t A;
+};
+
+struct Screen
+{
+ struct Nixie nixies[4];
+ i2c_master_bus_handle_t i2c_bus;
+ i2c_master_dev_handle_t mcp;
+};
+
+void configure_screen(struct Screen* screen);
+void test_screen(struct Screen* screen);
+void display(struct Screen* screen, char digits[4]);
+#endif \ No newline at end of file
generated by cgit v1.2.3 (git 2.25.1) at 2024-10-30 01:27:22 +0000