forked from xiaozhi/xiaozhi-esp32
218 lines
7.2 KiB
C++
218 lines
7.2 KiB
C++
#include "wifi_board.h"
|
||
#include "audio_codecs/es8311_audio_codec.h"
|
||
#include "display/ssd1306_display.h"
|
||
#include "application.h"
|
||
#include "button.h"
|
||
#include "config.h"
|
||
#include "iot/thing_manager.h"
|
||
#include "led/single_led.h"
|
||
#include "settings.h"
|
||
#include "font_awesome_symbols.h"
|
||
|
||
#include <wifi_station.h>
|
||
#include <esp_log.h>
|
||
#include <esp_efuse_table.h>
|
||
#include <driver/i2c_master.h>
|
||
#include <esp_timer.h>
|
||
#include <esp_pm.h>
|
||
|
||
#define TAG "XminiC3Board"
|
||
|
||
LV_FONT_DECLARE(font_puhui_14_1);
|
||
LV_FONT_DECLARE(font_awesome_14_1);
|
||
|
||
class XminiC3Board : public WifiBoard {
|
||
private:
|
||
i2c_master_bus_handle_t codec_i2c_bus_;
|
||
Button boot_button_;
|
||
bool press_to_talk_enabled_ = false;
|
||
esp_timer_handle_t power_save_timer_ = nullptr;
|
||
bool sleep_mode_enabled_ = false;
|
||
int power_save_ticks_ = 0;
|
||
|
||
void InitializePowerSaveTimer() {
|
||
esp_timer_create_args_t power_save_timer_args = {
|
||
.callback = [](void *arg) {
|
||
auto board = static_cast<XminiC3Board*>(arg);
|
||
board->PowerSaveCheck();
|
||
},
|
||
.arg = this,
|
||
.dispatch_method = ESP_TIMER_TASK,
|
||
.name = "power_save_timer",
|
||
.skip_unhandled_events = false,
|
||
};
|
||
ESP_ERROR_CHECK(esp_timer_create(&power_save_timer_args, &power_save_timer_));
|
||
ESP_ERROR_CHECK(esp_timer_start_periodic(power_save_timer_, 1000000));
|
||
}
|
||
|
||
void PowerSaveCheck() {
|
||
// 如果待机超过一定时间，则进入睡眠模式
|
||
const int seconds_to_sleep = 120;
|
||
auto& app = Application::GetInstance();
|
||
if (app.GetDeviceState() != kDeviceStateIdle) {
|
||
power_save_ticks_ = 0;
|
||
return;
|
||
}
|
||
|
||
power_save_ticks_++;
|
||
if (power_save_ticks_ >= seconds_to_sleep) {
|
||
EnableSleepMode(true);
|
||
}
|
||
}
|
||
|
||
void EnableSleepMode(bool enable) {
|
||
power_save_ticks_ = 0;
|
||
if (!sleep_mode_enabled_ && enable) {
|
||
ESP_LOGI(TAG, "Enabling sleep mode");
|
||
auto display = GetDisplay();
|
||
display->SetChatMessage("system", "");
|
||
display->SetEmotion("sleepy");
|
||
// 如果是LCD，还可以调节屏幕亮度
|
||
|
||
auto codec = GetAudioCodec();
|
||
codec->EnableInput(false);
|
||
|
||
esp_pm_config_t pm_config = {
|
||
.max_freq_mhz = 160,
|
||
.min_freq_mhz = 40,
|
||
.light_sleep_enable = true,
|
||
};
|
||
esp_pm_configure(&pm_config);
|
||
sleep_mode_enabled_ = true;
|
||
} else if (sleep_mode_enabled_ && !enable) {
|
||
esp_pm_config_t pm_config = {
|
||
.max_freq_mhz = 160,
|
||
.min_freq_mhz = 160,
|
||
.light_sleep_enable = false,
|
||
};
|
||
esp_pm_configure(&pm_config);
|
||
ESP_LOGI(TAG, "Disabling sleep mode");
|
||
|
||
auto codec = GetAudioCodec();
|
||
codec->EnableInput(true);
|
||
sleep_mode_enabled_ = false;
|
||
}
|
||
}
|
||
|
||
|
||
void InitializeCodecI2c() {
|
||
// Initialize I2C peripheral
|
||
i2c_master_bus_config_t i2c_bus_cfg = {
|
||
.i2c_port = I2C_NUM_0,
|
||
.sda_io_num = AUDIO_CODEC_I2C_SDA_PIN,
|
||
.scl_io_num = AUDIO_CODEC_I2C_SCL_PIN,
|
||
.clk_source = I2C_CLK_SRC_DEFAULT,
|
||
.glitch_ignore_cnt = 7,
|
||
.intr_priority = 0,
|
||
.trans_queue_depth = 0,
|
||
.flags = {
|
||
.enable_internal_pullup = 1,
|
||
},
|
||
};
|
||
ESP_ERROR_CHECK(i2c_new_master_bus(&i2c_bus_cfg, &codec_i2c_bus_));
|
||
}
|
||
|
||
void InitializeButtons() {
|
||
boot_button_.OnClick([this]() {
|
||
auto& app = Application::GetInstance();
|
||
if (app.GetDeviceState() == kDeviceStateStarting && !WifiStation::GetInstance().IsConnected()) {
|
||
ResetWifiConfiguration();
|
||
}
|
||
if (!press_to_talk_enabled_) {
|
||
app.ToggleChatState();
|
||
}
|
||
});
|
||
boot_button_.OnPressDown([this]() {
|
||
EnableSleepMode(false);
|
||
if (press_to_talk_enabled_) {
|
||
Application::GetInstance().StartListening();
|
||
}
|
||
});
|
||
boot_button_.OnPressUp([this]() {
|
||
if (press_to_talk_enabled_) {
|
||
Application::GetInstance().StopListening();
|
||
}
|
||
});
|
||
}
|
||
|
||
// 物联网初始化，添加对 AI 可见设备
|
||
void InitializeIot() {
|
||
Settings settings("vendor");
|
||
press_to_talk_enabled_ = settings.GetInt("press_to_talk", 0) != 0;
|
||
|
||
auto& thing_manager = iot::ThingManager::GetInstance();
|
||
thing_manager.AddThing(iot::CreateThing("Speaker"));
|
||
thing_manager.AddThing(iot::CreateThing("PressToTalk"));
|
||
}
|
||
|
||
public:
|
||
XminiC3Board() : boot_button_(BOOT_BUTTON_GPIO) {
|
||
// 把 ESP32C3 的 VDD SPI 引脚作为普通 GPIO 口使用
|
||
esp_efuse_write_field_bit(ESP_EFUSE_VDD_SPI_AS_GPIO);
|
||
|
||
InitializeCodecI2c();
|
||
InitializeButtons();
|
||
InitializePowerSaveTimer();
|
||
InitializeIot();
|
||
}
|
||
|
||
virtual Led* GetLed() override {
|
||
static SingleLed led_strip(BUILTIN_LED_GPIO);
|
||
return &led_strip;
|
||
}
|
||
|
||
virtual Display* GetDisplay() override {
|
||
static Ssd1306Display display(codec_i2c_bus_, DISPLAY_WIDTH, DISPLAY_HEIGHT, DISPLAY_MIRROR_X, DISPLAY_MIRROR_Y,
|
||
&font_puhui_14_1, &font_awesome_14_1);
|
||
return &display;
|
||
}
|
||
|
||
virtual AudioCodec* GetAudioCodec() override {
|
||
static Es8311AudioCodec audio_codec(codec_i2c_bus_, I2C_NUM_0, AUDIO_INPUT_SAMPLE_RATE, AUDIO_OUTPUT_SAMPLE_RATE,
|
||
AUDIO_I2S_GPIO_MCLK, AUDIO_I2S_GPIO_BCLK, AUDIO_I2S_GPIO_WS, AUDIO_I2S_GPIO_DOUT, AUDIO_I2S_GPIO_DIN,
|
||
AUDIO_CODEC_PA_PIN, AUDIO_CODEC_ES8311_ADDR);
|
||
return &audio_codec;
|
||
}
|
||
|
||
void SetPressToTalkEnabled(bool enabled) {
|
||
press_to_talk_enabled_ = enabled;
|
||
|
||
Settings settings("vendor", true);
|
||
settings.SetInt("press_to_talk", enabled ? 1 : 0);
|
||
ESP_LOGI(TAG, "Press to talk enabled: %d", enabled);
|
||
}
|
||
|
||
bool IsPressToTalkEnabled() {
|
||
return press_to_talk_enabled_;
|
||
}
|
||
};
|
||
|
||
DECLARE_BOARD(XminiC3Board);
|
||
|
||
|
||
namespace iot {
|
||
|
||
class PressToTalk : public Thing {
|
||
public:
|
||
PressToTalk() : Thing("PressToTalk", "控制对话模式，一种是长按对话，一种是单击后连续对话。") {
|
||
// 定义设备的属性
|
||
properties_.AddBooleanProperty("enabled", "true 表示长按说话模式，false 表示单击说话模式", []() -> bool {
|
||
auto board = static_cast<XminiC3Board*>(&Board::GetInstance());
|
||
return board->IsPressToTalkEnabled();
|
||
});
|
||
|
||
// 定义设备可以被远程执行的指令
|
||
methods_.AddMethod("SetEnabled", "启用或禁用长按说话模式，调用前需要经过用户确认", ParameterList({
|
||
Parameter("enabled", "true 表示长按说话模式，false 表示单击说话模式", kValueTypeBoolean, true)
|
||
}), [](const ParameterList& parameters) {
|
||
bool enabled = parameters["enabled"].boolean();
|
||
auto board = static_cast<XminiC3Board*>(&Board::GetInstance());
|
||
board->SetPressToTalkEnabled(enabled);
|
||
});
|
||
}
|
||
};
|
||
|
||
} // namespace iot
|
||
|
||
DECLARE_THING(PressToTalk);
|