实现 MonoBitmap.draw()

Author: TakWolf

src/pixel_font_knife/mono_bitmap.py

@@ -1,4 +1,5 @@
 from collections import UserList
+from io import StringIO
 from os import PathLike
 from typing import Any
 
@@ -69,6 +70,16 @@ def resize(self, left: int = 0, right: int = 0, top: int = 0, bottom: int = 0) -
             bitmap.append(bitmap_row)
         return bitmap
 
+    def draw(self, white: str = '  ', black: str = '██', end: str | None = None) -> str:
+        text = StringIO()
+        for bitmap_row in self:
+            for alpha in bitmap_row:
+                text.write(white if alpha == 0 else black)
+            if end is not None:
+                text.write(end)
+            text.write('\n')
+        return text.getvalue()
+
     def save_png(
             self,
             file_path: str | bytes | PathLike[str] | PathLike[bytes],

tests/test_mono_bitmap.py

@@ -104,6 +104,16 @@ def test_resize():
     ])
 
 
+def test_draw():
+    bitmap = MonoBitmap([
+        [1, 1, 0, 0],
+        [0, 0, 1, 1],
+    ])
+    text = ('████    *\n'
+            '    ████*\n')
+    assert bitmap.draw(end='*') == text
+
+
 def test_load_save(glyphs_dir: Path, tmp_path: Path):
     black_load_dir = glyphs_dir.joinpath('black')
     black_save_dir = tmp_path.joinpath('black')