From 92d86240f4b6155097a38b80f6714d8bf619adaf Mon Sep 17 00:00:00 2001
From: Chris Crocker-White <chriscw@balena.io>
Date: Wed, 23 Mar 2022 16:25:10 -0700
Subject: [PATCH] Add support for RPi RGB LED Matrix displays

Change-type: minor
Signed-off-by: Chris Crocker-White <chriscw@balena.io>
---
 luma/led_matrix/device.py | 104 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 103 insertions(+), 1 deletion(-)

diff --git a/luma/led_matrix/device.py b/luma/led_matrix/device.py
index 49f3abf..c8aeb8e 100644
--- a/luma/led_matrix/device.py
+++ b/luma/led_matrix/device.py
@@ -42,7 +42,7 @@
 from luma.led_matrix.segment_mapper import dot_muncher, regular
 
 
-__all__ = ["max7219", "ws2812", "neopixel", "neosegment", "apa102", "unicornhathd"]
+__all__ = ["max7219", "ws2812", "neopixel", "neosegment", "apa102", "unicornhathd", "rpirgbledmatrix"]
 
 
 class max7219(device):
@@ -622,3 +622,105 @@ def contrast(self, value):
         self._brightness = value
         if self._last_image is not None:
             self.display(self._last_image)
+
+class rpirgbledmatrix(device):
+    """
+    Interface to the Python bindings of the rpi-rgb-led-matrix library
+    https://github.com/hzeller/rpi-rgb-led-matrix
+    """
+    def __init__(self,
+        led_gpio_mapping="regular",
+        width=32, # Total matrix width, will be divided by chain to obtain panel width
+        height=32, # Total matrix height, will be divided by parallel to obtain panel height
+        led_chain = 1,
+        led_parallel = 1,
+        led_row_addr_type = 0,
+        led_multiplexing = 0,
+        led_pwm_bits = 11,
+        led_brightness = 100,
+        led_scan_mode = 1,
+        led_pwm_lsb_nanoseconds = 130,
+        led_rgb_sequence = "RGB",
+        led_pixel_mapper = "",
+        led_panel_type = "",
+        led_show_refresh = None,
+        led_slowdown_gpio = 1,
+        led_disable_hardware_pulse = None,
+        led_drop_privileges = True,
+        rotate=0, **kwargs):
+        super(rpirgbledmatrix, self).__init__(const=None, serial_interface=noop)
+
+        self.capabilities(width, height, rotate, mode="RGB")
+
+        from rgbmatrix import RGBMatrixOptions
+        self.options = RGBMatrixOptions()
+
+        self.options.hardware_mapping = led_gpio_mapping
+        self.options.rows = height / led_parallel
+        self.options.cols = width / led_chain
+        self.options.chain_length = led_chain
+        self.options.parallel = led_parallel
+        self.options.row_address_type = led_row_addr_type
+        self.options.multiplexing = led_multiplexing
+        self.options.pwm_bits = led_pwm_bits
+        self.options.brightness = led_brightness
+        self.options.scan_mode = led_scan_mode
+        self.options.pwm_lsb_nanoseconds = led_pwm_lsb_nanoseconds
+        self.options.led_rgb_sequence = led_rgb_sequence
+        self.options.pixel_mapper_config = led_pixel_mapper
+        self.options.panel_type = led_panel_type
+
+        if led_show_refresh != None:
+          self.options.show_refresh_rate = 1
+
+        if led_slowdown_gpio != None:
+            self.options.gpio_slowdown = led_slowdown_gpio
+
+        if led_disable_hardware_pulse != None:
+            self.options.led_disable_hardware_pulse = True
+
+        self.options.drop_privileges = led_drop_privileges
+
+
+        rlm = self._rlm = self.__rlm__()
+
+
+    def __rlm__(self):
+        from rgbmatrix import RGBMatrix
+        matrix = RGBMatrix(options = self.options)
+
+        return matrix
+
+    def display(self, image):
+        """
+        Copies an image to the LED matrix
+        """
+        assert(image.mode == self.mode)
+        assert(image.size == self.size)
+
+        self._rlm.SetImage(image)
+
+    def hide(self):
+        """
+        Turns the display off - since this is an LED Matrix this is achieved
+        by clearing the display
+        """
+        self.clear()
+
+    def clear(self):
+        """
+        Clears the LED matrix
+        """
+        self._rlm.Clear()
+
+    def contrast(self, value):
+        """
+        Sets the LED intensity to the desired level, in the range 0-255.
+
+        :param level: Desired contrast level in the range of 0-255.
+        :type level: int
+        """
+        max_brightness = self.options.brightness
+        scalar = max_brightness / value
+
+        self._rlm.brightness = round(value * scalar)
\ No newline at end of file