在STM32上使用HAL库和SSD1306 OLED显示屏实现菜单功能

#include "main.h"

#include <string.h>

#include <stdio.h>

#include <stdlib.h>



#include "ssd1306.h"



#define OLED_RESET_PIN GPIO_PIN_15

#define OLED_RESET_PORT GPIOA

#define OLED_I2C_PORT hi2c1



SSD1306 display(&hi2c1, OLED_RESET_PORT, OLED_RESET_PIN);



// 菜单节点

struct MenuItem

{

  const char* name;  // 节点名称

  MenuItem* next;    // 下一个节点的指针

  void (*function)();  // 节点功能函数的指针

};



// 菜单类

class Menu

{

public:

  Menu();

  void AddItem(const char* name, void (*function)());

  void Run();



private:

  MenuItem* mHead;

};



// 菜单构造函数

Menu::Menu()

  : mHead(nullptr)

{

}



// 添加菜单节点

void Menu::AddItem(const char* name, void (*function)())

{

  MenuItem* item = new MenuItem;

  item->name = name;

  item->function = function;

  item->next = mHead;

  mHead = item;

}



// 运行菜单系统

void Menu::Run()

{

  display.clear();



  MenuItem* current = mHead;

  char input[32];



  // 遍历链表，打印菜单选项

  int y = 0;

  while (current != nullptr)

  {

    display.setCursor(0, y);

    display.println(current->name);

    y += 10;

    current = current->next;

  }

  display.display();



  // 等待用户输入选择

  HAL_UART_Transmit(&huart2, (uint8_t*)"Enter your choice: ", strlen("Enter your choice: "), HAL_MAX_DELAY);

  HAL_UART_Receive(&huart2, (uint8_t*)input, sizeof(input), HAL_MAX_DELAY);

  input[strcspn(input, "\r\n")] = 0;  // 去除输入末尾的回车符



  // 查找用户输入对应的菜单节点，并执行功能函数

  current = mHead;

  while (current != nullptr)

  {

    if (strcmp(current->name, input) == 0)

    {

      current->function();

      return;

    }

    current = current->next;

  }



  HAL_UART_Transmit(&huart2, (uint8_t*)"Invalid choice.\r\n", strlen("Invalid choice.\r\n"), HAL_MAX_DELAY);

}



// 菜单功能函数1

void Function1()

{

  HAL_UART_Transmit(&huart2, (uint8_t*)"Function 1 executed.\r\n", strlen("Function 1 executed.\r\n"), HAL_MAX_DELAY);

}



// 菜单功能函数2

void Function2()

{

  HAL_UART_Transmit(&huart2, (uint8_t*)"Function 2 executed.\r\n", strlen("Function 2 executed.\r\n"), HAL_MAX_DELAY);

}



int main()

{

  // 初始化硬件和HAL库

  HAL_Init();

  SystemClock_Config();

  MX_GPIO_Init();

  MX_I2C1_Init();

  MX_USART2_UART_Init();



  // 初始化OLED显示屏

  display.init();

  display.setFont(ArialMT_Plain_10);



  // 创建菜单对象，添加菜单选项

  Menu menu;

menu.AddItem("Option 1", Function1);

menu.AddItem("Option 2", Function2);



// 循环执行菜单

while (1)

{

menu.Run();

}

}