Flipdot-Array

"""
READ FIRST
It is not recommended to change this file if you do not know what you are doing.
Doing so can cause PERMANENT DAMAGE to the display or the RPi (if done incorrectly).
Edit at your own risk.

This is a python-based interface for the display.
"""
#import RPi.GPIO as GPIO
import numpy as np
import atexit
import signal
import time

class Display:
    def __init__(self):
        self._ser= 18
        self._oe = 23
        self._rclk = 24
        self._srclk = 25
        self._srclr = 8
        # 21 cols, 7 rows
        self.shown = np.zeros((21,7),dtype=np.bool_)

        #GPIO.setmode(GPIO.BCM)
        #GPIO.setup(self._ser, GPIO.OUT)
        #GPIO.setup(self._oe, GPIO.OUT)
        #GPIO.setup(self._rclk, GPIO.OUT)
        #GPIO.setup(self._srclk, GPIO.OUT)
        #GPIO.setup(self._srclr, GPIO.OUT)
        atexit.register(self._cleanup)
        signal.signal(signal.SIGTERM, self._cleanup)
        self.all_off(force=True)
        self._disable()
        self._clear()
    
    def _cleanup(self):
        self._disable()
        self._clear()
        #GPIO.cleanup()

    def _disable(self):
        pass
        #GPIO.output(self._oe, GPIO.HIGH)

    def _enable(self):
        pass
        #GPIO.output(self._oe, GPIO.LOW)

    def _clear(self):
        pass
        #GPIO.output(self._srclr, GPIO.LOW)
        #GPIO.output(self._srclr, GPIO.HIGH)
        #GPIO.output(self._ser, GPIO.LOW)

    def write_dot(self, x, y, state, force=False):
        self._disable()
        self._clear()
        if not (0 <= x < 21 and 0 <= y < 7):
            return

        serial_data = 0
        if state and (not self.shown[x][y] or force):
            serial_data = serial_data | (1 << y)  #set row
            serial_data = serial_data | (1 << (x + 24 + (8 * (x // 8))))  #set col
            self.shown[x][y] = True
        elif not state and (self.shown[x][y] or force):
            serial_data = serial_data | (1 << y+8)  #set row
            serial_data = serial_data | (1 << (x + 16 + (8 * (x // 8))))  #set col
            self.shown[x][y] = False

        if serial_data > 0:
            print(f"{x} {y} {int(state)} {serial_data}") #{" "*(2-((x-10>=0) + (y-10>=0)))} {serial_data:064b}")
            for i in range(63,-1,-1):
                pass
                #GPIO.output(self._ser, GPIO.HIGH if serial_data & (1 << i) else GPIO.LOW)
                #GPIO.output(self._srclk, GPIO.HIGH)
                #GPIO.output(self._srclk, GPIO.LOW)

            #move data into registers
            #GPIO.output(self._rclk, GPIO.HIGH)
            #GPIO.output(self._rclk, GPIO.LOW)
            #GPIO.output(self._ser, GPIO.LOW)

            self._enable()
            time.sleep(0.0002)
            self._disable()

    def all_off(self, force=False):
        for y in range(7):
            for x in range(21):
                self.write_dot(x, y, False, force)
        self._disable()
        self._clear()

    def all_on(self, force=False):
        for y in range(7):
            for x in range(21):
                self.write_dot(x, y, True, force)
        self._disable()
        self._clear()

    def write_display(self, new_display, start_x=0, start_y=0, delay=0, force=False, horizontal="WE", vertical="NS", direction="LR"):
        x_range = range(start_x, min(21, len(new_display) + start_x)) if direction == "LR" else range(min(20, len(new_display) + start_x), start_x-1,-1)
        y_range = range(start_y, 7) if direction == "TB" else range(6, start_y-1,-1)
        for x in x_range:
            for y in range(start_y, 7):
                time.sleep(delay)
                self.write_dot(x, y, bool(new_display[x - start_x] & (1 << y)), force) 
        self._disable()
        self._clear()
    
    def write_display_boolarray(self, new_display, start_x=0, start_y=0, delay=0, force=False, horizontal="WE", vertical="NS", direction="LR"):
        new_display_int = []
        for x in new_display:
            new_display_int.append(0)
            for y in x:
                new_display_int[-1] = new_display_int[-1] | (1 << y)
        self.write_display(new_display_int, start_x, start_y, delay, force, horizontal, vertical, direction)