✨ (drivers): Add getLevel()

leka · Jun 21, 2021 · f8de43f · f8de43f
1 parent 8d12317
commit f8de43f
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 5 deletions.
diff --git a/drivers/LKCoreBattery/include/LKCoreBattery.h b/drivers/LKCoreBattery/include/LKCoreBattery.h
@@ -12,12 +12,30 @@ namespace leka {
 class LKCoreBattery
 {
   public:
+	enum class BatteryLevel
+	{
+		high,
+		medium,
+		low
+	};
+
 	explicit LKCoreBattery(PinName pin) : _pin {mbed::AnalogIn(pin)} {};
 
-	float readVoltage(void);
+	auto readVoltage(void) -> float;
+	auto getLevel() -> BatteryLevel;
 
   private:
+	auto averageVoltage(int number_of_points__for_average) -> float;
+
 	mbed::AnalogIn _pin;
+
+	float high_battery_level_step = 12.00;
+	float low_battery_level_step  = 10.00;
+
+	float battery_level_sensor_coefficient = 3.33;
+	float resistor1						   = 169;
+	float resistor2						   = 47;
+	float _voltage_divider_coefficient	   = resistor2 / (resistor2 + resistor1);
 };
 
 }	// namespace leka

diff --git a/drivers/LKCoreBattery/source/LKCoreBattery.cpp b/drivers/LKCoreBattery/source/LKCoreBattery.cpp
@@ -6,7 +6,29 @@
 
 namespace leka {
 
-float LKCoreBattery::readVoltage(void)
+LKCoreBattery::BatteryLevel LKCoreBattery ::getLevel()
+{
+	auto batteryVoltage = averageVoltage(20);
+
+	if (batteryVoltage < (low_battery_level_step / battery_level_sensor_coefficient) * _voltage_divider_coefficient) {
+		return BatteryLevel::low;
+	}
+	if (batteryVoltage > (high_battery_level_step / battery_level_sensor_coefficient) * _voltage_divider_coefficient) {
+		return BatteryLevel::high;
+	}
+	return BatteryLevel::medium;
+}
+
+auto LKCoreBattery::averageVoltage(int number_of_points__for_average) -> float
+{
+	float average = 0;
+	for (int i = 0; i < number_of_points__for_average; ++i) {
+		average += readVoltage();
+	}
+
+	return average / points_number_for_average;
+}
+auto LKCoreBattery::readVoltage(void) -> float
 {
 	return _pin.read();
 }

diff --git a/drivers/LKCoreBattery/tests/LKCoreBattery_test.cpp b/drivers/LKCoreBattery/tests/LKCoreBattery_test.cpp
@@ -22,23 +22,26 @@ TEST(LKCoreBatteryTest, initialization)
 	ASSERT_NE(&battery, nullptr);
 }
 
-TEST(LKCoreBatteryTest, readMinVoltage)
+TEST(LKCoreBatteryTest, readLowVoltage)
 {
 	auto expected = test_set_Voltage(0.0f);
 
 	ASSERT_EQ(expected, battery.readVoltage());
+	ASSERT_EQ(LKCoreBattery::BatteryLevel::low, battery.getLevel());
 }
 
 TEST(LKCoreBatteryTest, readMiddleVoltage)
 {
-	auto expected = test_set_Voltage(0.25f);
+	auto expected = test_set_Voltage(0.50f);
 
 	ASSERT_EQ(expected, battery.readVoltage());
+	ASSERT_EQ(LKCoreBattery::BatteryLevel::medium, battery.getLevel());
 }
 
-TEST(LKCoreBatteryTest, readMaxVoltage)
+TEST(LKCoreBatteryTest, readHighVoltage)
 {
 	auto expected = test_set_Voltage(1.0f);
 
 	ASSERT_EQ(expected, battery.readVoltage());
+	ASSERT_EQ(LKCoreBattery::BatteryLevel::high, battery.getLevel());
 }