pybricks.hubs.EV3Brick: Add display attribute.

This provides easy access to the brick display.

Refs pybricks/support#2154

Author: schodet

pybricks/hubs/pb_type_ev3brick.c

@@ -12,11 +12,13 @@
 #include <pybricks/hubs.h>
 #include <pybricks/util_mp/pb_obj_helper.h>
 #include <pybricks/util_pb/pb_error.h>
+#include <pybricks/media.h>
 
 typedef struct _hubs_EV3Brick_obj_t {
     mp_obj_base_t base;
     mp_obj_t battery;
     mp_obj_t buttons;
+    mp_obj_t display;
     mp_obj_t light;
     mp_obj_t system;
 } hubs_EV3Brick_obj_t;
@@ -49,6 +51,7 @@ static mp_obj_t hubs_EV3Brick_make_new(const mp_obj_type_t *type, size_t n_args,
 
     self->battery = MP_OBJ_FROM_PTR(&pb_module_battery);
     self->buttons = pb_type_Keypad_obj_new(pb_type_ev3brick_button_pressed);
+    self->display = pb_type_Image_display_obj_new();
     self->light = common_ColorLight_internal_obj_new(pbsys_status_light_main);
     self->system = MP_OBJ_FROM_PTR(&pb_type_System);
 
@@ -58,6 +61,7 @@ static mp_obj_t hubs_EV3Brick_make_new(const mp_obj_type_t *type, size_t n_args,
 static const pb_attr_dict_entry_t hubs_EV3Brick_attr_dict[] = {
     PB_DEFINE_CONST_ATTR_RO(MP_QSTR_battery, hubs_EV3Brick_obj_t, battery),
     PB_DEFINE_CONST_ATTR_RO(MP_QSTR_buttons, hubs_EV3Brick_obj_t, buttons),
+    PB_DEFINE_CONST_ATTR_RO(MP_QSTR_display, hubs_EV3Brick_obj_t, display),
     PB_DEFINE_CONST_ATTR_RO(MP_QSTR_light, hubs_EV3Brick_obj_t, light),
     PB_DEFINE_CONST_ATTR_RO(MP_QSTR_system, hubs_EV3Brick_obj_t, system),
     PB_ATTR_DICT_SENTINEL

pybricks/media.h

@@ -14,6 +14,8 @@
 
 extern const mp_obj_type_t pb_type_Image;
 
+mp_obj_t pb_type_Image_display_obj_new(void);
+
 #endif // PYBRICKS_PY_MEDIA_IMAGE
 
 #endif // PYBRICKS_PY_MEDIA

pybricks/media/pb_type_image.c

@@ -42,6 +42,15 @@ static int get_color(mp_obj_t obj) {
     return max - v * max / 100;
 }
 
+mp_obj_t pb_type_Image_display_obj_new(void) {
+    pb_type_Image_obj_t *self = mp_obj_malloc(pb_type_Image_obj_t, &pb_type_Image);
+    self->buffer = NULL;
+    self->update = true;
+    self->image = *pbdrv_display_get_image();
+
+    return MP_OBJ_FROM_PTR(self);
+}
+
 static mp_obj_t pb_type_Image_make_new(const mp_obj_type_t *type,
     size_t n_args, size_t n_kw, const mp_obj_t *args) {
     PB_PARSE_ARGS_CLASS(n_args, n_kw, args,