forked from xiaozhi/xiaozhi-esp32
315 lines
11 KiB
C
315 lines
11 KiB
C
|
#pragma once
|
|||
|
|
#include <vector>
|
|||
|
|
#include <functional>
|
|||
|
|
|
|||
|
|
#include <esp_timer.h>
|
|||
|
|
#include <driver/gpio.h>
|
|||
|
|
#include <esp_adc/adc_oneshot.h>
|
|||
|
|
#include "sdkconfig.h"
|
|||
|
|
#include "button.h"
|
|||
|
|
#include "board.h"
|
|||
|
|
#include "config.h"
|
|||
|
|
#include "assets/lang_config.h"
|
|||
|
|
#include <esp_sleep.h>
|
|||
|
|
|
|||
|
|
class PowerManager {
|
|||
|
|
private:
|
|||
|
|
esp_timer_handle_t timer_handle_;
|
|||
|
|
esp_timer_handle_t power_timer_handle_;
|
|||
|
|
std::function<void(bool)> on_charging_status_changed_;
|
|||
|
|
std::function<void(bool)> on_low_battery_status_changed_;
|
|||
|
|
|
|||
|
|
gpio_num_t charging_pin_ = GPIO_NUM_NC;
|
|||
|
|
std::vector<uint16_t> adc_values_;
|
|||
|
|
uint32_t battery_level_ = 30;
|
|||
|
|
bool is_charging_ = false;
|
|||
|
|
bool is_low_battery_ = false;
|
|||
|
|
int ticks_ = 0;
|
|||
|
|
adc_oneshot_unit_handle_t adc_handle_;
|
|||
|
|
const int kBatteryAdcInterval = 60;
|
|||
|
|
const int kBatteryAdcDataCount = 3;
|
|||
|
|
const int kLowBatteryLevel = 20;
|
|||
|
|
|
|||
|
|
// power
|
|||
|
|
bool pressed = false; // 是否按下电源键
|
|||
|
|
int PowerControl_ticks_ = 0; // 开机时长
|
|||
|
|
int press_ticks_ = 0; // 按下时的tick
|
|||
|
|
bool is_first_boot = true; //
|
|||
|
|
uint8_t PowerDec_level_ = 0; // 电源键电平
|
|||
|
|
const int power_off_ticks_ = 20; // 按键按下20/5秒准备关机;
|
|||
|
|
bool new_charging_status = false; // 插入usb时无法软件关机
|
|||
|
|
bool is_shutting_down_ = false; // 标记是否进入关机流程
|
|||
|
|
int shutdown_delay_ticks_ = 0; // 关机延迟计数
|
|||
|
|
uint8_t shutdown_ticks = 5; // 5/5s后关机 预留的关机操作
|
|||
|
|
bool shutdown_first_ = true;
|
|||
|
|
|
|||
|
|
// 旧版硬件软件关机逻辑
|
|||
|
|
void PowrSwitch() {
|
|||
|
|
PowerDec_level_ = gpio_get_level(Power_Dec);
|
|||
|
|
|
|||
|
|
// 确保开机后电源键松开再检测关机
|
|||
|
|
if (PowerDec_level_ == 1) {
|
|||
|
|
is_first_boot = false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (is_shutting_down_) {
|
|||
|
|
shutdown_delay_ticks_++;
|
|||
|
|
if (shutdown_delay_ticks_ >= shutdown_ticks) {
|
|||
|
|
shutdown();
|
|||
|
|
is_shutting_down_ = false;
|
|||
|
|
shutdown_delay_ticks_ = 0;
|
|||
|
|
}
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!is_first_boot) {
|
|||
|
|
PowerControl_ticks_++;
|
|||
|
|
if (PowerDec_level_ == 0 && pressed == false) {
|
|||
|
|
press_ticks_ = PowerControl_ticks_;
|
|||
|
|
pressed = true;
|
|||
|
|
}
|
|||
|
|
if ((press_ticks_ != 0) &&( PowerControl_ticks_ - press_ticks_ == power_off_ticks_ )&& (!new_charging_status)) {
|
|||
|
|
if (timer_handle_) {
|
|||
|
|
esp_timer_stop(timer_handle_);
|
|||
|
|
esp_timer_delete(timer_handle_);
|
|||
|
|
}
|
|||
|
|
is_shutting_down_ = true;
|
|||
|
|
shutdown_delay_ticks_ = 0;
|
|||
|
|
}
|
|||
|
|
if (PowerDec_level_ == 1 && press_ticks_!= 0) {
|
|||
|
|
PowerDec_level_ = gpio_get_level(Power_Dec);
|
|||
|
|
if (PowerDec_level_ == 1) {
|
|||
|
|
pressed = false;
|
|||
|
|
press_ticks_ = 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void CheckBatteryStatus() {
|
|||
|
|
int usb_usb_adc_value0;
|
|||
|
|
ESP_ERROR_CHECK(adc_oneshot_read(adc_handle_, POWER_USBIN_ADC_CHANNEL, &usb_usb_adc_value0));
|
|||
|
|
new_charging_status = (1500 < usb_usb_adc_value0 && usb_usb_adc_value0 < 4000);
|
|||
|
|
// ESP_LOGE("powercontrol", "USB ADC VALUE 1: %d", usb_usb_adc_value0);
|
|||
|
|
|
|||
|
|
if (new_charging_status != is_charging_) {
|
|||
|
|
ReadBatteryAdcData();
|
|||
|
|
is_charging_ = new_charging_status;
|
|||
|
|
if (on_charging_status_changed_) {
|
|||
|
|
on_charging_status_changed_(is_charging_);
|
|||
|
|
}
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果电池电量数据不足，则读取电池电量数据
|
|||
|
|
if (adc_values_.size() < kBatteryAdcDataCount) {
|
|||
|
|
ReadBatteryAdcData();
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果电池电量数据充足，则每 kBatteryAdcInterval 个 tick 读取一次电池电量数据
|
|||
|
|
ticks_++;
|
|||
|
|
if (ticks_ % kBatteryAdcInterval == 0) {
|
|||
|
|
ReadBatteryAdcData();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void ReadBatteryAdcData() {
|
|||
|
|
int adc_value;
|
|||
|
|
ESP_ERROR_CHECK(adc_oneshot_read(adc_handle_, POWER_BATTERY_ADC_CHANNEL, &adc_value));
|
|||
|
|
|
|||
|
|
// 将 ADC 值添加到队列中
|
|||
|
|
adc_values_.push_back(adc_value);
|
|||
|
|
if (adc_values_.size() > kBatteryAdcDataCount) {
|
|||
|
|
adc_values_.erase(adc_values_.begin());
|
|||
|
|
}
|
|||
|
|
uint32_t average_adc = 0;
|
|||
|
|
for (auto value : adc_values_) {
|
|||
|
|
average_adc += value;
|
|||
|
|
}
|
|||
|
|
average_adc /= adc_values_.size();
|
|||
|
|
|
|||
|
|
// 定义电池电量区间
|
|||
|
|
const struct {
|
|||
|
|
uint16_t adc;
|
|||
|
|
uint8_t level;
|
|||
|
|
} levels[] = {
|
|||
|
|
{1970, 0},
|
|||
|
|
{2062, 20},
|
|||
|
|
{2154, 40},
|
|||
|
|
{2246, 60},
|
|||
|
|
{2338, 80},
|
|||
|
|
{2430, 100}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 低于最低值时
|
|||
|
|
if (average_adc < levels[0].adc) {
|
|||
|
|
battery_level_ = 0;
|
|||
|
|
}
|
|||
|
|
// 高于最高值时
|
|||
|
|
else if (average_adc >= levels[5].adc) {
|
|||
|
|
battery_level_ = 100;
|
|||
|
|
} else {
|
|||
|
|
// 线性插值计算中间值
|
|||
|
|
for (int i = 0; i < 5; i++) {
|
|||
|
|
if (average_adc >= levels[i].adc && average_adc < levels[i+1].adc) {
|
|||
|
|
float ratio = static_cast<float>(average_adc - levels[i].adc) / (levels[i+1].adc - levels[i].adc);
|
|||
|
|
battery_level_ = levels[i].level + ratio * (levels[i+1].level - levels[i].level);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Check low battery status
|
|||
|
|
if (adc_values_.size() >= kBatteryAdcDataCount) {
|
|||
|
|
bool new_low_battery_status = battery_level_ <= kLowBatteryLevel;
|
|||
|
|
if (new_low_battery_status != is_low_battery_) {
|
|||
|
|
is_low_battery_ = new_low_battery_status;
|
|||
|
|
if (on_low_battery_status_changed_) {
|
|||
|
|
on_low_battery_status_changed_(is_low_battery_);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ESP_LOGI("PowerManager", "ADC value: %d average: %ld level: %ld", adc_value, average_adc, battery_level_);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
PowerManager(gpio_num_t pin) : charging_pin_(pin) {
|
|||
|
|
// 初始化电源键检测引脚
|
|||
|
|
gpio_config_t powerdecgpio_conf = {};
|
|||
|
|
powerdecgpio_conf.intr_type = GPIO_INTR_DISABLE;
|
|||
|
|
powerdecgpio_conf.mode = GPIO_MODE_INPUT;
|
|||
|
|
powerdecgpio_conf.pin_bit_mask = (1ULL << Power_Dec);
|
|||
|
|
powerdecgpio_conf.pull_down_en = GPIO_PULLDOWN_DISABLE;
|
|||
|
|
powerdecgpio_conf.pull_up_en = GPIO_PULLUP_DISABLE;
|
|||
|
|
gpio_config(&powerdecgpio_conf);
|
|||
|
|
|
|||
|
|
// 初始化电源控制引脚
|
|||
|
|
gpio_config_t powercontgpio_conf = {};
|
|||
|
|
powercontgpio_conf.intr_type = GPIO_INTR_DISABLE;
|
|||
|
|
powercontgpio_conf.mode = GPIO_MODE_OUTPUT;
|
|||
|
|
powercontgpio_conf.pin_bit_mask = (1ULL << Power_Control);
|
|||
|
|
powercontgpio_conf.pull_down_en = GPIO_PULLDOWN_ENABLE;
|
|||
|
|
powercontgpio_conf.pull_up_en = GPIO_PULLUP_DISABLE;
|
|||
|
|
gpio_config(&powercontgpio_conf);
|
|||
|
|
gpio_set_level(Power_Control, 1);
|
|||
|
|
ESP_LOGI("powercontrol", "turnded on ...");
|
|||
|
|
|
|||
|
|
// 创建电源控制检查定时器
|
|||
|
|
esp_timer_create_args_t power_timer_args = {
|
|||
|
|
.callback = [](void* arg) {
|
|||
|
|
PowerManager* self = static_cast<PowerManager*>(arg);
|
|||
|
|
self->PowrSwitch();
|
|||
|
|
},
|
|||
|
|
.arg = this,
|
|||
|
|
.dispatch_method = ESP_TIMER_TASK,
|
|||
|
|
.name = "power_cotrol_timer",
|
|||
|
|
.skip_unhandled_events = true,
|
|||
|
|
};
|
|||
|
|
ESP_ERROR_CHECK(esp_timer_create(&power_timer_args, &power_timer_handle_));
|
|||
|
|
ESP_ERROR_CHECK(esp_timer_start_periodic(power_timer_handle_, 200000));
|
|||
|
|
// 初始化充电引脚
|
|||
|
|
gpio_config_t io_conf = {};
|
|||
|
|
io_conf.intr_type = GPIO_INTR_DISABLE;
|
|||
|
|
io_conf.mode = GPIO_MODE_INPUT;
|
|||
|
|
io_conf.pin_bit_mask = (1ULL << charging_pin_);
|
|||
|
|
io_conf.pull_down_en = GPIO_PULLDOWN_DISABLE;
|
|||
|
|
io_conf.pull_up_en = GPIO_PULLUP_DISABLE;
|
|||
|
|
gpio_config(&io_conf);
|
|||
|
|
|
|||
|
|
// 创建电池电量检查定时器
|
|||
|
|
esp_timer_create_args_t timer_args = {
|
|||
|
|
.callback = [](void* arg) {
|
|||
|
|
PowerManager* self = static_cast<PowerManager*>(arg);
|
|||
|
|
self->CheckBatteryStatus();
|
|||
|
|
},
|
|||
|
|
.arg = this,
|
|||
|
|
.dispatch_method = ESP_TIMER_TASK,
|
|||
|
|
.name = "battery_check_timer",
|
|||
|
|
.skip_unhandled_events = true,
|
|||
|
|
};
|
|||
|
|
ESP_ERROR_CHECK(esp_timer_create(&timer_args, &timer_handle_));
|
|||
|
|
ESP_ERROR_CHECK(esp_timer_start_periodic(timer_handle_, 1000000));
|
|||
|
|
|
|||
|
|
adc_oneshot_unit_init_cfg_t init_config = {
|
|||
|
|
.unit_id = POWER_CBS_ADC_UNIT,
|
|||
|
|
.ulp_mode = ADC_ULP_MODE_DISABLE,
|
|||
|
|
};
|
|||
|
|
ESP_ERROR_CHECK(adc_oneshot_new_unit(&init_config, &adc_handle_));
|
|||
|
|
|
|||
|
|
adc_oneshot_chan_cfg_t chan_config = {
|
|||
|
|
.atten = ADC_ATTEN_DB_12,
|
|||
|
|
.bitwidth = ADC_BITWIDTH_12,
|
|||
|
|
};
|
|||
|
|
ESP_ERROR_CHECK(adc_oneshot_config_channel(adc_handle_, POWER_BATTERY_ADC_CHANNEL, &chan_config)); // 电池电量
|
|||
|
|
ESP_ERROR_CHECK(adc_oneshot_config_channel(adc_handle_, POWER_USBIN_ADC_CHANNEL, &chan_config)); // usb
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
~PowerManager() {
|
|||
|
|
if (timer_handle_) {
|
|||
|
|
esp_timer_stop(timer_handle_);
|
|||
|
|
esp_timer_delete(timer_handle_);
|
|||
|
|
}
|
|||
|
|
if (power_timer_handle_) {
|
|||
|
|
esp_timer_stop(power_timer_handle_);
|
|||
|
|
esp_timer_delete(power_timer_handle_);
|
|||
|
|
}
|
|||
|
|
if (adc_handle_ != nullptr) {
|
|||
|
|
adc_oneshot_del_unit(adc_handle_);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool IsCharging() {
|
|||
|
|
// 如果电量已经满了，则不再显示充电中
|
|||
|
|
if (battery_level_ == 100) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
return is_charging_;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool IsDischarging() {
|
|||
|
|
// 没有区分充电和放电，所以直接返回相反状态
|
|||
|
|
return !is_charging_;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
uint8_t GetBatteryLevel() {
|
|||
|
|
return battery_level_;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void OnLowBatteryStatusChanged(std::function<void(bool)> callback) {
|
|||
|
|
on_low_battery_status_changed_ = callback;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void OnChargingStatusChanged(std::function<void(bool)> callback) {
|
|||
|
|
on_charging_status_changed_ = callback;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void shutdown() {
|
|||
|
|
if (!new_charging_status && shutdown_first_)
|
|||
|
|
{
|
|||
|
|
shutdown_first_ = false; // 进入后置 false ，防止再次进入关机状态
|
|||
|
|
gpio_config_t shutdown_gpio_conf = {};
|
|||
|
|
shutdown_gpio_conf.intr_type = GPIO_INTR_DISABLE;
|
|||
|
|
shutdown_gpio_conf.mode = GPIO_MODE_OUTPUT;
|
|||
|
|
shutdown_gpio_conf.pin_bit_mask = (1ULL << Power_Dec);
|
|||
|
|
shutdown_gpio_conf.pull_down_en = GPIO_PULLDOWN_DISABLE;
|
|||
|
|
shutdown_gpio_conf.pull_up_en = GPIO_PULLUP_DISABLE;
|
|||
|
|
gpio_config(&shutdown_gpio_conf);
|
|||
|
|
gpio_set_level(DISPLAY_BACKLIGHT_PIN, 0);
|
|||
|
|
gpio_set_level(Power_Control, 0);
|
|||
|
|
for (int i=1;i<15;i++) {
|
|||
|
|
gpio_set_level(Power_Dec, 1);
|
|||
|
|
vTaskDelay(pdMS_TO_TICKS(100));
|
|||
|
|
gpio_set_level(Power_Dec, 0);
|
|||
|
|
vTaskDelay(pdMS_TO_TICKS(100));
|
|||
|
|
ESP_LOGI("PowerManager","触发开关机控制");
|
|||
|
|
}
|
|||
|
|
ESP_LOGI("PowerManager","关机失败，进入深睡眠");
|
|||
|
|
esp_deep_sleep_start();
|
|||
|
|
} else {
|
|||
|
|
ESP_LOGI("PowerManager","检测到插入usb，无法关机");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
};
|