PicoVector: Add a bounds method for polygon bounds.

pimoroni · Sep 12, 2023 · ae7e6e8 · ae7e6e8
1 parent 5f730ff
commit ae7e6e8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/micropython/modules/picovector/picovector.c b/micropython/modules/picovector/picovector.c
@@ -4,11 +4,13 @@
 
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(POLYGON__del__obj, POLYGON__del__);
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(POLYGON_centroid_obj, POLYGON_centroid);
+STATIC MP_DEFINE_CONST_FUN_OBJ_1(POLYGON_bounds_obj, POLYGON_bounds);
 
 
 STATIC const mp_rom_map_elem_t POLYGON_locals_dict_table[] = {
     { MP_ROM_QSTR(MP_QSTR___del__), MP_ROM_PTR(&POLYGON__del__obj) },
     { MP_ROM_QSTR(MP_QSTR_centroid), MP_ROM_PTR(&POLYGON_centroid_obj) },
+    { MP_ROM_QSTR(MP_QSTR_bounds), MP_ROM_PTR(&POLYGON_bounds_obj) },
 };
 
 STATIC MP_DEFINE_CONST_DICT(POLYGON_locals_dict, POLYGON_locals_dict_table);

diff --git a/micropython/modules/picovector/picovector.cpp b/micropython/modules/picovector/picovector.cpp
@@ -232,6 +232,18 @@ mp_obj_t POLYGON_centroid(mp_obj_t self_in) {
     return mp_obj_new_tuple(2, tuple);
 }
 
+mp_obj_t POLYGON_bounds(mp_obj_t self_in) {
+    _POLYGON_obj_t *self = MP_OBJ_TO_PTR2(self_in, _POLYGON_obj_t);
+
+    mp_obj_t tuple[4];
+    tuple[0] = mp_obj_new_int((int)(self->contour.bounds().x));
+    tuple[1] = mp_obj_new_int((int)(self->contour.bounds().y));
+    tuple[2] = mp_obj_new_int((int)(self->contour.bounds().w));
+    tuple[3] = mp_obj_new_int((int)(self->contour.bounds().h));
+
+    return mp_obj_new_tuple(4, tuple);
+}
+
 void POLYGON_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) {
     (void)kind;
     _POLYGON_obj_t *self = MP_OBJ_TO_PTR2(self_in, _POLYGON_obj_t);

diff --git a/micropython/modules/picovector/picovector.h b/micropython/modules/picovector/picovector.h
@@ -11,6 +11,7 @@ extern mp_obj_t REGULAR_POLYGON_make_new(const mp_obj_type_t *type, size_t n_arg
 extern mp_obj_t RECTANGLE_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *all_args);
 extern void POLYGON_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind);
 extern mp_obj_t POLYGON_centroid(mp_obj_t self_in);
+extern mp_obj_t POLYGON_bounds(mp_obj_t self_in);
 extern mp_obj_t POLYGON_getiter(mp_obj_t o_in, mp_obj_iter_buf_t *iter_buf);
 
 extern mp_obj_t POLYGON__del__(mp_obj_t self_in);