X-Git-Url: https://git.ayoreis.com/hackapet/Adafruit_Blinka_Displayio.git/blobdiff_plain/5cfe68b419b1e014ae334c500569d87b661e4281..cf46de4055b0ee82b9fe65c9a38aff2af1ab86ee:/displayio/bitmap.py diff --git a/displayio/bitmap.py b/displayio/bitmap.py index ffc6631..7ac075c 100644 --- a/displayio/bitmap.py +++ b/displayio/bitmap.py @@ -21,7 +21,7 @@ # THE SOFTWARE. """ -`displayio` +`displayio.bitmap` ================================================================================ displayio for Blinka @@ -35,11 +35,14 @@ displayio for Blinka """ -from displayio import Rectangle +from recordclass import recordclass +from PIL import Image __version__ = "0.0.0-auto.0" __repo__ = "https://github.com/adafruit/Adafruit_Blinka_displayio.git" +Rectangle = recordclass("Rectangle", "x1 y1 x2 y2") + class Bitmap: """Stores values of a certain size in a 2D array""" @@ -73,7 +76,7 @@ class Bitmap: ): raise NotImplementedError("Invalid bits per value") - self._data = (width * height) * [0] + self._image = Image.new("P", (width, height), 0) self._dirty_area = Rectangle(0, 0, width, height) def __getitem__(self, index): @@ -82,10 +85,16 @@ class Bitmap: an x,y tuple or an int equal to `y * width + x`. """ if isinstance(index, (tuple, list)): - index = (index[1] * self._width) + index[0] - if index >= len(self._data): + x, y = index + elif isinstance(index, int): + x = index % self._width + y = index // self._width + else: + raise TypeError("Index is not an int, list, or tuple") + + if x > self._image.width or y > self._image.height: raise ValueError("Index {} is out of range".format(index)) - return self._data[index] + return self._image.getpixel((x, y)) def __setitem__(self, index, value): """ @@ -101,7 +110,7 @@ class Bitmap: elif isinstance(index, int): x = index % self._width y = index // self._width - self._data[index] = value + self._image.putpixel((x, y), value) if self._dirty_area.x1 == self._dirty_area.x2: self._dirty_area.x1 = x self._dirty_area.x2 = x + 1 @@ -123,7 +132,7 @@ class Bitmap: def fill(self, value): """Fills the bitmap with the supplied palette index value.""" - self._data = (self._width * self._height) * [value] + self._image = Image.new("P", (self._width, self._height), value) self._dirty_area = Rectangle(0, 0, self._width, self._height) @property