extmod/modframebuf: Add text size support.

jonathanhogg · jonathanhogg · commit 714793524ecc · 2022-01-13T13:25:52.000Z
Add a `size` parameter to the `text()` method. This does a fairly simple 
scaling of the built-in 8 pixel font for the moment, but could use 
different fonts in the future.
diff --git a/extmod/modframebuf.c b/extmod/modframebuf.c
@@ -574,6 +574,10 @@ STATIC mp_obj_t framebuf_text(size_t n_args, const mp_obj_t *args) {
     if (n_args >= 5) {
         col = mp_obj_get_int(args[4]);
     }
+    mp_int_t size = 8;
+    if (n_args >= 6) {
+        size = mp_obj_get_int(args[5]);
+    }
 
     // loop over chars
     for (; *str; ++str) {
@@ -585,13 +589,16 @@ STATIC mp_obj_t framebuf_text(size_t n_args, const mp_obj_t *args) {
         // get char data
         const uint8_t *chr_data = &font_petme128_8x8[(chr - 32) * 8];
         // loop over char data
-        for (int j = 0; j < 8; j++, x0++) {
+        for (int j = 0; j < size; j++, x0++) {
             if (0 <= x0 && x0 < self->width) { // clip x
-                uint vline_data = chr_data[j]; // each byte is a column of 8 pixels, LSB at top
-                for (int y = y0; vline_data; vline_data >>= 1, y++) { // scan over vertical column
-                    if (vline_data & 1) { // only draw if pixel set
+                uint vline_data = chr_data[j * 8 / size]; // each byte is a column of 8 pixels, LSB at top
+                if (vline_data) { // skip empty columns
+                    for (int i = 0, y = y0; i < size; i++, y++) {
                         if (0 <= y && y < self->height) { // clip y
-                            setpixel(self, x0, y, col);
+                            // scan over vertical column
+                            if (vline_data & (1 << (i * 8 / size))) { // only draw if pixel set
+                                setpixel(self, x0, y, col);
+                            }
                         }
                     }
                 }
@@ -600,7 +607,7 @@ STATIC mp_obj_t framebuf_text(size_t n_args, const mp_obj_t *args) {
     }
     return mp_const_none;
 }
-STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(framebuf_text_obj, 4, 5, framebuf_text);
+STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(framebuf_text_obj, 4, 6, framebuf_text);
 
 #if !MICROPY_ENABLE_DYNRUNTIME
 STATIC const mp_rom_map_elem_t framebuf_locals_dict_table[] = {
diff --git a/tests/extmod/framebuf1.py b/tests/extmod/framebuf1.py
@@ -94,6 +94,12 @@
     fbuf.text("hello", 0, 0, 0)  # clear
     print(buf)
 
+    # scaled text
+    fbuf.text("*", 0, 0, 1, 16)
+    print(buf)
+    fbuf.text("*", 0, 0, 0, 16)  # clear
+    print(buf)
+
     # char out of font range set to chr(127)
     fbuf.text(str(chr(31)), 0, 0)
     print(buf)
diff --git a/tests/extmod/framebuf1.py.exp b/tests/extmod/framebuf1.py.exp
@@ -18,6 +18,8 @@ bytearray(b'\x00\x00@\x00\x00\x00\x00\x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01')
 bytearray(b'\x00\x7f\x7f\x04\x04\x00\x00\x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
+bytearray(b'\xc0\xc0\xcc\xcc\xfc\x00\x00\x0c\x0c\x0f')
+bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
 bytearray(b'\xaaU\xaaU\xaa\x00\x00\x00\x00\x00')
 
 MONO_HLSB
@@ -40,6 +42,8 @@ bytearray(b'\x00\x00\x00\x00\x00\x00 \x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x08\x00')
 bytearray(b'``x````\x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
+bytearray(b'\x00\x0088\x08\x08\xf8\xf8\x08\x08')
+bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
 bytearray(b'P\xa8P\xa8P\xa8P\xa8\x00\x00')
 
 MONO_HMSB
@@ -62,6 +66,8 @@ bytearray(b'\x00\x00\x00\x00\x00\x00\x04\x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x10\x00')
 bytearray(b'\x06\x06\x1e\x06\x06\x06\x06\x00\x00\x00')
 bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
+bytearray(b'\x00\x00\x1c\x1c\x10\x10\x1f\x1f\x10\x10')
+bytearray(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
 bytearray(b'\n\x15\n\x15\n\x15\n\x15\x00\x00')
 
 ValueError
