Add arc

esphome · jesserockz · Aug 5, 2024 · Jul 22, 2024 · Jul 22, 2024 · Jul 22, 2024
commit 52a9b8a47edd3adad3e0fa19e75dea934e064271
@@ -21,6 +21,7 @@
 from esphome.helpers import write_file_if_changed
 
 from . import defines as df, helpers, lv_validation as lvalid
+from .arc import arc_spec
 from .automation import update_to_code
 from .btn import btn_spec
 from .defines import CONF_SKIP
@@ -57,7 +58,7 @@
 CODEOWNERS = ("@clydebarrow",)
 LOGGER = logging.getLogger(__name__)
 
-for w_type in (label_spec, obj_spec, btn_spec, bar_spec, slider_spec):
+for w_type in (label_spec, obj_spec, btn_spec, bar_spec, slider_spec, arc_spec):
     WIDGET_TYPES[w_type.name] = w_type
 
 WIDGET_SCHEMA = any_widget_schema()

@@ -0,0 +1,78 @@
+import esphome.config_validation as cv
+from esphome.const import (
+    CONF_MAX_VALUE,
+    CONF_MIN_VALUE,
+    CONF_MODE,
+    CONF_ROTATION,
+    CONF_VALUE,
+)
+
+from ...cpp_types import nullptr
+from .defines import (
+    ARC_MODES,
+    CONF_ADJUSTABLE,
+    CONF_ARC,
+    CONF_CHANGE_RATE,
+    CONF_END_ANGLE,
+    CONF_INDICATOR,
+    CONF_KNOB,
+    CONF_MAIN,
+    CONF_START_ANGLE,
+    literal,
+)
+from .lv_validation import angle, get_start_value, lv_float
+from .lvcode import lv, lv_obj
+from .types import LvNumber
+from .widget import Widget, WidgetType
+
+ARC_SCHEMA = cv.Schema(
+    {
+        cv.Optional(CONF_VALUE): lv_float,
+        cv.Optional(CONF_MIN_VALUE, default=0): cv.int_,
+        cv.Optional(CONF_MAX_VALUE, default=100): cv.int_,
+        cv.Optional(CONF_START_ANGLE, default=135): angle,
+        cv.Optional(CONF_END_ANGLE, default=45): angle,
+        cv.Optional(CONF_ROTATION, default=0.0): angle,
+        cv.Optional(CONF_ADJUSTABLE, default=False): bool,
+        cv.Optional(CONF_MODE, default="NORMAL"): ARC_MODES.one_of,
+        cv.Optional(CONF_CHANGE_RATE, default=720): cv.uint16_t,
+    }
+)
+
+ARC_MODIFY_SCHEMA = cv.Schema(
+    {
+        cv.Optional(CONF_VALUE): lv_float,
+    }
+)
+
+
+class ArcType(WidgetType):
+    def __init__(self):
+        super().__init__(
+            CONF_ARC,
+            LvNumber("lv_arc_t"),
+            parts=(CONF_MAIN, CONF_INDICATOR, CONF_KNOB),
+            schema=ARC_SCHEMA,
+            modify_schema=ARC_MODIFY_SCHEMA,
+        )
+
+    async def to_code(self, w: Widget, config):
+        if CONF_MIN_VALUE in config:
+            lv.arc_set_range(w.obj, config[CONF_MIN_VALUE], config[CONF_MAX_VALUE])
+            lv.arc_set_bg_angles(
+                w.obj, config[CONF_START_ANGLE] // 10, config[CONF_END_ANGLE] // 10
+            )
+            lv.arc_set_rotation(w.obj, config[CONF_ROTATION] // 10)
+            lv.arc_set_mode(w.obj, literal(config[CONF_MODE]))
+            lv.arc_set_change_rate(w.obj, config[CONF_CHANGE_RATE])
-        if CONF_MIN_VALUE in config:
-            lv.arc_set_range(w.obj, config[CONF_MIN_VALUE], config[CONF_MAX_VALUE])
-            lv.arc_set_bg_angles(
-                w.obj, config[CONF_START_ANGLE] // 10, config[CONF_END_ANGLE] // 10
-            )
-            lv.arc_set_rotation(w.obj, config[CONF_ROTATION] // 10)
-            lv.arc_set_mode(w.obj, literal(config[CONF_MODE]))
-            lv.arc_set_change_rate(w.obj, config[CONF_CHANGE_RATE])
+        lv.arc_set_range(w.obj, config[CONF_MIN_VALUE], config[CONF_MAX_VALUE])
+        lv.arc_set_bg_angles(
+            w.obj, config[CONF_START_ANGLE] // 10, config[CONF_END_ANGLE] // 10
+        )
+        lv.arc_set_rotation(w.obj, config[CONF_ROTATION] // 10)
+        lv.arc_set_mode(w.obj, literal(config[CONF_MODE]))
+        lv.arc_set_change_rate(w.obj, config[CONF_CHANGE_RATE])
-        if CONF_MIN_VALUE in config:
-            lv.arc_set_range(w.obj, config[CONF_MIN_VALUE], config[CONF_MAX_VALUE])
-            lv.arc_set_bg_angles(
-                w.obj, config[CONF_START_ANGLE] // 10, config[CONF_END_ANGLE] // 10
-            )
-            lv.arc_set_rotation(w.obj, config[CONF_ROTATION] // 10)
-            lv.arc_set_mode(w.obj, literal(config[CONF_MODE]))
-            lv.arc_set_change_rate(w.obj, config[CONF_CHANGE_RATE])
+        lv.arc_set_range(w.obj, config[CONF_MIN_VALUE], config[CONF_MAX_VALUE])
+        lv.arc_set_bg_angles(
+            w.obj, config[CONF_START_ANGLE] // 10, config[CONF_END_ANGLE] // 10
+        )
+        lv.arc_set_rotation(w.obj, config[CONF_ROTATION] // 10)
+        lv.arc_set_mode(w.obj, literal(config[CONF_MODE]))
+        lv.arc_set_change_rate(w.obj, config[CONF_CHANGE_RATE])
+
+        if config.get(CONF_ADJUSTABLE) is False:
+            lv_obj.remove_style(w.obj, nullptr, literal("LV_PART_KNOB"))
+            w.clear_flag("LV_OBJ_FLAG_CLICKABLE")
+
+        value = await get_start_value(config)
+        if value is not None:
+            lv.arc_set_value(w.obj, value)
+
+
+arc_spec = ArcType()
@@ -114,6 +114,7 @@ def extend(self, *choices):
 
 
 # Widgets
+CONF_ARC = "arc"
 CONF_BAR = "bar"
 CONF_LABEL = "label"
 CONF_SLIDER = "slider"

diff --git a/tests/components/lvgl/lvgl-package.yaml b/tests/components/lvgl/lvgl-package.yaml
@@ -139,6 +139,27 @@ lvgl:
 
     - id: page2
       widgets:
+        - arc:
+            align: left_mid
+            id: lv_arc
+            adjustable: true
+            on_value:
+              then:
+                - logger.log:
+                    format: "Arc value is %f"
+                    args: [x]
+            group: general
+            scroll_on_focus: true
+            value: 75
+            min_value: 1
+            max_value: 100
+            arc_color: 0xFF0000
+            indicator:
+              arc_color: 0xF000FF
+              pressed:
+                arc_color: 0xFFFF00
+              focused:
+                arc_color: 0x808080
         - bar:
             id: bar_id
             align: top_mid