Nick
4 years ago
5 changed files with 139 additions and 0 deletions
@ -0,0 +1,115 @@
@@ -0,0 +1,115 @@
|
||||
/*
|
||||
* Copyright (c) 2020 The ZMK Contributors |
||||
* |
||||
* SPDX-License-Identifier: MIT |
||||
*/ |
||||
|
||||
#define DT_DRV_COMPAT zmk_battery_voltage_divider |
||||
|
||||
#include <device.h> |
||||
#include <drivers/kscan.h> |
||||
#include <drivers/gpio.h> |
||||
#include <drivers/adc.h> |
||||
#include <drivers/sensor.h> |
||||
#include <logging/log.h> |
||||
#include <math.h> |
||||
|
||||
LOG_MODULE_DECLARE(zmk, CONFIG_ZMK_LOG_LEVEL); |
||||
|
||||
#if DT_HAS_COMPAT_STATUS_OKAY(DT_DRV_COMPAT) |
||||
|
||||
#define VBATT DT_PATH(vbatt) |
||||
|
||||
struct battery_config { |
||||
struct device *adc; |
||||
struct adc_channel_cfg acc; |
||||
struct adc_sequence as; |
||||
int16_t adc_raw; |
||||
}; |
||||
|
||||
static struct battery_config battery_config; |
||||
|
||||
static int lithium_ion_mv_to_pct(int16_t bat_mv) { |
||||
// Magic function that maps mV to this discharge graph from adafruit: https://learn.adafruit.com/li-ion-and-lipoly-batteries/voltages
|
||||
return round(106.818 + (-0.032685 - 106.818) / pow(1 + pow(bat_mv/3679.35, 58.979), 0.347386)); |
||||
} |
||||
|
||||
static void battery_read(struct k_work *workd) { |
||||
struct battery_config *cfg = &battery_config; |
||||
struct adc_sequence *as = &cfg->as; |
||||
|
||||
int rc = adc_read(cfg->adc, as); |
||||
as->calibrate = false; |
||||
if (rc == 0) { |
||||
int32_t val = cfg->adc_raw; |
||||
|
||||
adc_raw_to_millivolts(adc_ref_internal(cfg->adc), cfg->acc.gain, as->resolution, &val); |
||||
|
||||
rc = val * (uint64_t)DT_PROP(VBATT, full_ohms) / DT_PROP(VBATT, output_ohms); |
||||
LOG_DBG("ADC raw %d ~ %d mV => %d mV\n", cfg->adc_raw, val, rc); |
||||
int percent = lithium_ion_mv_to_pct(rc); |
||||
LOG_DBG("Percent: %d", percent); |
||||
} else { |
||||
LOG_DBG("Failed to read ADC: %d", rc); |
||||
} |
||||
} |
||||
|
||||
K_WORK_DEFINE(battery_work, battery_read); |
||||
|
||||
static void battery_handler(struct k_timer *timer) |
||||
{ |
||||
k_work_submit(&battery_work); |
||||
} |
||||
|
||||
K_TIMER_DEFINE(battery_tick, battery_handler, NULL); |
||||
|
||||
static int battery_setup(struct device *_arg) { |
||||
struct battery_config *cfg = &battery_config; |
||||
struct adc_sequence *as = &cfg->as; |
||||
struct adc_channel_cfg *acc = &cfg->acc; |
||||
|
||||
cfg->adc = device_get_binding(DT_IO_CHANNELS_LABEL(VBATT)); |
||||
|
||||
if (cfg->adc == NULL) { |
||||
LOG_ERR("ADC %s failed to retrieve", DT_IO_CHANNELS_LABEL(VBATT)); |
||||
return -ENOENT; |
||||
} |
||||
|
||||
*as = (struct adc_sequence){ |
||||
.channels = BIT(0), |
||||
.buffer = &cfg->adc_raw, |
||||
.buffer_size = sizeof(cfg->adc_raw), |
||||
.oversampling = 4, |
||||
.calibrate = true, |
||||
}; |
||||
|
||||
#ifdef CONFIG_ADC_NRFX_SAADC |
||||
*acc = (struct adc_channel_cfg){ |
||||
.gain = ADC_GAIN_1_5, |
||||
.reference = ADC_REF_INTERNAL, |
||||
.acquisition_time = ADC_ACQ_TIME(ADC_ACQ_TIME_MICROSECONDS, 40), |
||||
.input_positive = SAADC_CH_PSELP_PSELP_AnalogInput0 + DT_IO_CHANNELS_INPUT(VBATT), |
||||
}; |
||||
|
||||
as->resolution = 12; |
||||
#else |
||||
#error Unsupported ADC |
||||
#endif |
||||
|
||||
int adc_rc = adc_channel_setup(cfg->adc, acc); |
||||
LOG_DBG("AIN%u setup returned %d", DT_IO_CHANNELS_INPUT(VBATT), adc_rc); |
||||
|
||||
if (adc_rc != 0) { |
||||
return adc_rc; |
||||
} |
||||
|
||||
k_timer_start(&battery_tick, K_NO_WAIT, K_SECONDS(5)); |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
SYS_INIT(battery_setup, |
||||
APPLICATION, |
||||
CONFIG_APPLICATION_INIT_PRIORITY); |
||||
|
||||
#endif /* DT_HAS_COMPAT_STATUS_OKAY(DT_DRV_COMPAT) */ |
@ -0,0 +1,9 @@
@@ -0,0 +1,9 @@
|
||||
# Copyright (c) 2020 The ZMK Contributors |
||||
# SPDX-License-Identifier: MIT |
||||
|
||||
description: Battery SoC monitoring using voltage divider |
||||
|
||||
compatible: "zmk,battery-voltage-divider" |
||||
|
||||
include: voltage-divider.yaml |
||||
|
Loading…
Reference in new issue