codal_port/microbit_display: Add display.rotate(angle) method.

This takes an angle in degrees and rounds it towards the nearest multiple of 90. Signed-off-by: Damien George <[email protected]>
microbit-foundation · Sep 9, 2024 · 74e32fb · 74e32fb
1 parent 0df07b1
commit 74e32fb
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/codal_port/microbit_display.c b/src/codal_port/microbit_display.c
@@ -199,6 +199,24 @@ static mp_obj_t microbit_display_get_pixel_func(mp_obj_t self_in, mp_obj_t x_in,
 }
 MP_DEFINE_CONST_FUN_OBJ_3(microbit_display_get_pixel_obj, microbit_display_get_pixel_func);
 
+static mp_obj_t microbit_display_rotate(mp_obj_t self_in, mp_obj_t angle_in) {
+    (void)self_in;
+    mp_int_t angle_degrees = mp_obj_get_int(angle_in);
+
+    // Round angle towards nearest multiple of 90 degrees, within 0..359.
+    angle_degrees %= 360;
+    if (angle_degrees < 0) {
+        angle_degrees += 360;
+    }
+    unsigned int rotation = (angle_degrees + 45) / 90;
+
+    // Set the display rotation.
+    microbit_hal_display_rotate(rotation);
+
+    return mp_const_none;
+}
+MP_DEFINE_CONST_FUN_OBJ_2(microbit_display_rotate_obj, microbit_display_rotate);
+
 static const mp_rom_map_elem_t microbit_display_locals_dict_table[] = {
     { MP_ROM_QSTR(MP_QSTR_get_pixel), MP_ROM_PTR(&microbit_display_get_pixel_obj) },
     { MP_ROM_QSTR(MP_QSTR_set_pixel), MP_ROM_PTR(&microbit_display_set_pixel_obj) },
@@ -209,6 +227,7 @@ static const mp_rom_map_elem_t microbit_display_locals_dict_table[] = {
     { MP_ROM_QSTR(MP_QSTR_off), MP_ROM_PTR(&microbit_display_off_obj) },
     { MP_ROM_QSTR(MP_QSTR_is_on), MP_ROM_PTR(&microbit_display_is_on_obj) },
     { MP_ROM_QSTR(MP_QSTR_read_light_level),MP_ROM_PTR(&microbit_display_read_light_level_obj) },
+    { MP_ROM_QSTR(MP_QSTR_rotate),MP_ROM_PTR(&microbit_display_rotate_obj) },
 };
 static MP_DEFINE_CONST_DICT(microbit_display_locals_dict, microbit_display_locals_dict_table);