From d740cec3ce564c43e34d7b45d6d32ed54a3ab004 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Magalh=C3=A3es?= <joamag@gmail.com>
Date: Sun, 5 Nov 2023 23:52:57 +0000
Subject: [PATCH] chore: new image method
---
src/python/boytacean/__init__.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/src/python/boytacean/__init__.py b/src/python/boytacean/__init__.py
index 89efe406..533633c5 100644
--- a/src/python/boytacean/__init__.py
+++ b/src/python/boytacean/__init__.py
@@ -1,4 +1,4 @@
-from PIL import Image
+from PIL.Image import Image, frombytes
from .boytacean import DISPLAY_WIDTH, DISPLAY_HEIGHT, CPU_FREQ, GameBoy as GameBoyRust
@@ -41,11 +41,13 @@ class GameBoy:
def frame_buffer(self):
return self._system.frame_buffer()
- def save_image(self, filename: str, format: str = "PNG"):
+ def image(self) -> Image:
frame_buffer = self._system.frame_buffer()
- image = Image.frombytes(
- "RGB", (DISPLAY_WIDTH, DISPLAY_HEIGHT), frame_buffer, "raw"
- )
+ image = frombytes("RGB", (DISPLAY_WIDTH, DISPLAY_HEIGHT), frame_buffer, "raw")
+ return image
+
+ def save_image(self, filename: str, format: str = "PNG"):
+ image = self.image()
image.save(filename, format=format)
@property
--
GitLab