Source code for adafruit_ht16k33.matrix

# The MIT License (MIT)
#
# Copyright (c) 2016 Radomir Dopieralski & Tony DiCola for Adafruit Industries
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

"""
Matrix Displays
================

"""
from adafruit_ht16k33.ht16k33 import HT16K33

__version__ = "0.0.0-auto.0"
__repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_HT16K33.git"

[docs]class Matrix8x8(HT16K33): """A single matrix.""" _columns = 8 _rows = 8
[docs] def pixel(self, x, y, color=None): """Get or set the color of a given pixel.""" if not 0 <= x <= 7: return None if not 0 <= y <= 7: return None x = (x - 1) % 8 return super()._pixel(x, y, color)
def __getitem__(self, key): x, y = key return self.pixel(x, y) def __setitem__(self, key, value): x, y = key self.pixel(x, y, value) #pylint: disable=too-many-branches
[docs] def shift(self, x, y, rotate=False): """ Shift pixels by x and y :param rotate: (Optional) Rotate the shifted pixels to the left side (default=False) """ if x > 0: # Shift Right for _ in range(x): for row in range(0, self.rows): last_pixel = self[self.columns - 1, row] if rotate else 0 for col in range(self.columns - 1, 0, -1): self[col, row] = self[col - 1, row] self[0, row] = last_pixel elif x < 0: # Shift Left for _ in range(-x): for row in range(0, self.rows): last_pixel = self[0, row] if rotate else 0 for col in range(0, self.columns - 1): self[col, row] = self[col + 1, row] self[self.columns - 1, row] = last_pixel if y > 0: # Shift Up for _ in range(y): for col in range(0, self.columns): last_pixel = self[col, self.rows - 1] if rotate else 0 for row in range(self.rows - 1, 0, -1): self[col, row] = self[col, row - 1] self[col, 0] = last_pixel elif y < 0: # Shift Down for _ in range(-y): for col in range(0, self.columns): last_pixel = self[col, 0] if rotate else 0 for row in range(0, self.rows - 1): self[col, row] = self[col, row + 1] self[col, self.rows - 1] = last_pixel if self._auto_write: self.show()
#pylint: enable=too-many-branches
[docs] def shift_right(self, rotate=False): """ Shift all pixels right :param rotate: (Optional) Rotate the shifted pixels to the left side (default=False) """ self.shift(1, 0, rotate)
[docs] def shift_left(self, rotate=False): """ Shift all pixels left :param rotate: (Optional) Rotate the shifted pixels to the right side (default=False) """ self.shift(-1, 0, rotate)
[docs] def shift_up(self, rotate=False): """ Shift all pixels up :param rotate: (Optional) Rotate the shifted pixels to bottom (default=False) """ self.shift(0, 1, rotate)
[docs] def shift_down(self, rotate=False): """ Shift all pixels down :param rotate: (Optional) Rotate the shifted pixels to top (default=False) """ self.shift(0, -1, rotate)
[docs] def image(self, img): """Set buffer to value of Python Imaging Library image. The image should be in 1 bit mode and a size equal to the display size.""" imwidth, imheight = img.size if imwidth != self.columns or imheight != self.rows: raise ValueError('Image must be same dimensions as display ({0}x{1}).' \ .format(self.columns, self.rows)) # Grab all the pixels from the image, faster than getpixel. pixels = img.convert('1').load() # Iterate through the pixels for x in range(self.columns): # yes this double loop is slow, for y in range(self.rows): # but these displays are small! self.pixel(x, y, pixels[(x, y)]) if self._auto_write: self.show()
@property def columns(self): """Read-only property for number of columns""" return self._columns @property def rows(self): """Read-only property for number of rows""" return self._rows
[docs]class Matrix16x8(Matrix8x8): """The matrix wing.""" _columns = 16
[docs] def pixel(self, x, y, color=None): """Get or set the color of a given pixel.""" if not 0 <= x <= 15: return None if not 0 <= y <= 7: return None if x >= 8: x -= 8 y += 8 return super()._pixel(y, x, color)
[docs]class MatrixBackpack16x8(Matrix16x8): """A double matrix backpack."""
[docs] def pixel(self, x, y, color=None): """Get or set the color of a given pixel.""" if not 0 <= x <= 15: return None if not 0 <= y <= 7: return None return super()._pixel(x, y, color)
[docs]class Matrix8x8x2(Matrix8x8): """A bi-color matrix.""" LED_OFF = 0 LED_RED = 1 LED_GREEN = 2 LED_YELLOW = 3
[docs] def pixel(self, x, y, color=None): """Get or set the color of a given pixel.""" if not 0 <= x <= 7: return None if not 0 <= y <= 7: return None if color is not None: super()._pixel(y, x, (color >> 1) & 0x01) super()._pixel(y + 8, x, (color & 0x01)) else: return super()._pixel(y, x) | super()._pixel(y + 8, x) << 1 return None
[docs] def fill(self, color): """Fill the whole display with the given color.""" fill1 = 0xff if color & 0x01 else 0x00 fill2 = 0xff if color & 0x02 else 0x00 for i in range(8): self._set_buffer(i * 2, fill1) self._set_buffer(i * 2 + 1, fill2) if self._auto_write: self.show()
[docs] def image(self, img): """Set buffer to value of Python Imaging Library image. The image should be a size equal to the display size.""" imwidth, imheight = img.size if imwidth != self.columns or imheight != self.rows: raise ValueError('Image must be same dimensions as display ({0}x{1}).' \ .format(self.columns, self.rows)) # Grab all the pixels from the image, faster than getpixel. pixels = img.convert('RGB').load() # Iterate through the pixels for x in range(self.columns): # yes this double loop is slow, for y in range(self.rows): # but these displays are small! if pixels[(x, y)] == (255, 0, 0): self.pixel(x, y, self.LED_RED) elif pixels[(x, y)] == (0, 255, 0): self.pixel(x, y, self.LED_GREEN) elif pixels[(x, y)] == (255, 255, 0): self.pixel(x, y, self.LED_YELLOW) else: # Unknown color, default to LED off. self.pixel(x, y, self.LED_OFF) if self._auto_write: self.show()