compact_wifi_board.cc 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. #include "wifi_board.h"
  2. #include "audio_codecs/no_audio_codec.h"
  3. #include "display/oled_display.h"
  4. #include "system_reset.h"
  5. #include "application.h"
  6. #include "button.h"
  7. #include "config.h"
  8. #include "iot/thing_manager.h"
  9. #include "led/single_led.h"
  10. #include "assets/lang_config.h"
  11. #include <wifi_station.h>
  12. #include <esp_log.h>
  13. #include <driver/i2c_master.h>
  14. #include <esp_lcd_panel_ops.h>
  15. #include <esp_lcd_panel_vendor.h>
  16. #ifdef SH1106
  17. #include <esp_lcd_panel_sh1106.h>
  18. #endif
  19. #define TAG "CompactWifiBoard"
  20. LV_FONT_DECLARE(font_puhui_14_1);
  21. LV_FONT_DECLARE(font_awesome_14_1);
  22. class CompactWifiBoard : public WifiBoard {
  23. private:
  24. i2c_master_bus_handle_t display_i2c_bus_;
  25. esp_lcd_panel_io_handle_t panel_io_ = nullptr;
  26. esp_lcd_panel_handle_t panel_ = nullptr;
  27. Display* display_ = nullptr;
  28. Button boot_button_;
  29. Button touch_button_;
  30. Button volume_up_button_;
  31. Button volume_down_button_;
  32. void InitializeDisplayI2c() {
  33. i2c_master_bus_config_t bus_config = {
  34. .i2c_port = (i2c_port_t)0,
  35. .sda_io_num = DISPLAY_SDA_PIN,
  36. .scl_io_num = DISPLAY_SCL_PIN,
  37. .clk_source = I2C_CLK_SRC_DEFAULT,
  38. .glitch_ignore_cnt = 7,
  39. .intr_priority = 0,
  40. .trans_queue_depth = 0,
  41. .flags = {
  42. .enable_internal_pullup = 1,
  43. },
  44. };
  45. ESP_ERROR_CHECK(i2c_new_master_bus(&bus_config, &display_i2c_bus_));
  46. }
  47. void InitializeSsd1306Display() {
  48. // SSD1306 config
  49. esp_lcd_panel_io_i2c_config_t io_config = {
  50. .dev_addr = 0x3C,
  51. .on_color_trans_done = nullptr,
  52. .user_ctx = nullptr,
  53. .control_phase_bytes = 1,
  54. .dc_bit_offset = 6,
  55. .lcd_cmd_bits = 8,
  56. .lcd_param_bits = 8,
  57. .flags = {
  58. .dc_low_on_data = 0,
  59. .disable_control_phase = 0,
  60. },
  61. .scl_speed_hz = 400 * 1000,
  62. };
  63. ESP_ERROR_CHECK(esp_lcd_new_panel_io_i2c_v2(display_i2c_bus_, &io_config, &panel_io_));
  64. ESP_LOGI(TAG, "Install SSD1306 driver");
  65. esp_lcd_panel_dev_config_t panel_config = {};
  66. panel_config.reset_gpio_num = -1;
  67. panel_config.bits_per_pixel = 1;
  68. esp_lcd_panel_ssd1306_config_t ssd1306_config = {
  69. .height = static_cast<uint8_t>(DISPLAY_HEIGHT),
  70. };
  71. panel_config.vendor_config = &ssd1306_config;
  72. #ifdef SH1106
  73. ESP_ERROR_CHECK(esp_lcd_new_panel_sh1106(panel_io_, &panel_config, &panel_));
  74. #else
  75. ESP_ERROR_CHECK(esp_lcd_new_panel_ssd1306(panel_io_, &panel_config, &panel_));
  76. #endif
  77. ESP_LOGI(TAG, "SSD1306 driver installed");
  78. // Reset the display
  79. ESP_ERROR_CHECK(esp_lcd_panel_reset(panel_));
  80. if (esp_lcd_panel_init(panel_) != ESP_OK) {
  81. ESP_LOGE(TAG, "Failed to initialize display");
  82. display_ = new NoDisplay();
  83. return;
  84. }
  85. // Set the display to on
  86. ESP_LOGI(TAG, "Turning display on");
  87. ESP_ERROR_CHECK(esp_lcd_panel_disp_on_off(panel_, true));
  88. display_ = new OledDisplay(panel_io_, panel_, DISPLAY_WIDTH, DISPLAY_HEIGHT, DISPLAY_MIRROR_X, DISPLAY_MIRROR_Y,
  89. {&font_puhui_14_1, &font_awesome_14_1});
  90. }
  91. void InitializeButtons() {
  92. boot_button_.OnClick([this]() {
  93. auto& app = Application::GetInstance();
  94. if (app.GetDeviceState() == kDeviceStateStarting && !WifiStation::GetInstance().IsConnected()) {
  95. ResetWifiConfiguration();
  96. }
  97. app.ToggleChatState();
  98. });
  99. touch_button_.OnPressDown([this]() {
  100. Application::GetInstance().StartListening();
  101. });
  102. touch_button_.OnPressUp([this]() {
  103. Application::GetInstance().StopListening();
  104. });
  105. volume_up_button_.OnClick([this]() {
  106. auto codec = GetAudioCodec();
  107. auto volume = codec->output_volume() + 10;
  108. if (volume > 100) {
  109. volume = 100;
  110. }
  111. codec->SetOutputVolume(volume);
  112. GetDisplay()->ShowNotification(Lang::Strings::VOLUME + std::to_string(volume));
  113. });
  114. volume_up_button_.OnLongPress([this]() {
  115. GetAudioCodec()->SetOutputVolume(100);
  116. GetDisplay()->ShowNotification(Lang::Strings::MAX_VOLUME);
  117. });
  118. volume_down_button_.OnClick([this]() {
  119. auto codec = GetAudioCodec();
  120. auto volume = codec->output_volume() - 10;
  121. if (volume < 0) {
  122. volume = 0;
  123. }
  124. codec->SetOutputVolume(volume);
  125. GetDisplay()->ShowNotification(Lang::Strings::VOLUME + std::to_string(volume));
  126. });
  127. volume_down_button_.OnLongPress([this]() {
  128. GetAudioCodec()->SetOutputVolume(0);
  129. GetDisplay()->ShowNotification(Lang::Strings::MUTED);
  130. });
  131. }
  132. // 物联网初始化,添加对 AI 可见设备
  133. void InitializeIot() {
  134. auto& thing_manager = iot::ThingManager::GetInstance();
  135. thing_manager.AddThing(iot::CreateThing("Speaker"));
  136. thing_manager.AddThing(iot::CreateThing("Lamp"));
  137. }
  138. public:
  139. CompactWifiBoard() :
  140. boot_button_(BOOT_BUTTON_GPIO),
  141. touch_button_(TOUCH_BUTTON_GPIO),
  142. volume_up_button_(VOLUME_UP_BUTTON_GPIO),
  143. volume_down_button_(VOLUME_DOWN_BUTTON_GPIO) {
  144. InitializeDisplayI2c();
  145. InitializeSsd1306Display();
  146. InitializeButtons();
  147. InitializeIot();
  148. }
  149. virtual Led* GetLed() override {
  150. static SingleLed led(BUILTIN_LED_GPIO);
  151. return &led;
  152. }
  153. virtual AudioCodec* GetAudioCodec() override {
  154. #ifdef AUDIO_I2S_METHOD_SIMPLEX
  155. static NoAudioCodecSimplex audio_codec(AUDIO_INPUT_SAMPLE_RATE, AUDIO_OUTPUT_SAMPLE_RATE,
  156. AUDIO_I2S_SPK_GPIO_BCLK, AUDIO_I2S_SPK_GPIO_LRCK, AUDIO_I2S_SPK_GPIO_DOUT, AUDIO_I2S_MIC_GPIO_SCK, AUDIO_I2S_MIC_GPIO_WS, AUDIO_I2S_MIC_GPIO_DIN);
  157. #else
  158. static NoAudioCodecDuplex audio_codec(AUDIO_INPUT_SAMPLE_RATE, AUDIO_OUTPUT_SAMPLE_RATE,
  159. AUDIO_I2S_GPIO_BCLK, AUDIO_I2S_GPIO_WS, AUDIO_I2S_GPIO_DOUT, AUDIO_I2S_GPIO_DIN);
  160. #endif
  161. return &audio_codec;
  162. }
  163. virtual Display* GetDisplay() override {
  164. return display_;
  165. }
  166. };
  167. DECLARE_BOARD(CompactWifiBoard);