#! /usr/bin/python3
# -*- coding: utf8 -*-

# load needed modules from luma
from luma.core.interface.serial import i2c, spi
from luma.core.render import canvas
from luma.oled.device import sh1106
from luma.core import lib
from luma.core.sprite_system import framerate_regulator

# load needed modules from RPi.GPIO
import RPi.GPIO as GPIO

# load needed modules from PIL
from PIL import ImageFont, ImageDraw, Image

# load more needed modules
import time
import sys
import math
import colorsys

# load randommodules
from random import randint, gauss

#GPIO define pins
RST_PIN  = 25 #Reset
CS_PIN   = 8
DC_PIN   = 24

serial = spi(device=0, port=0, bus_speed_hz = 8000000, transfer_size = 4096, gpio_DC = DC_PIN, gpio_RST = RST_PIN)
device = sh1106(serial, rotate=2) #sh1106

FALLOFF = 1.9
SCAN_SPEED = 3


def main():
    start_time = time.time()
    regulator = framerate_regulator(fps=10)

    while True:
        with regulator:
            delta = (time.time() - start_time)

            # Offset is a sine wave derived from the time delta
            # we use this to animate both the hue and larson scan
            # so they are kept in sync with each other
            offset = (math.sin(delta * SCAN_SPEED) + 1) / 2

            # Use offset to pick the right colour from the hue wheel
            hue = int(round(offset * 360))

            # Now we generate a value from 0 to 7
            offset = int(round(offset * device.width))

            with canvas(device, dither=True) as draw:
                for x in range(device.width):
                    sat = 1.0

                    val = (device.width - 1) - (abs(offset - x) * FALLOFF)
                    val /= (device.width - 1)  # Convert to 0.0 to 1.0
                    val = max(val, 0.0)  # Ditch negative values

                    xhue = hue  # Grab hue for this pixel
                    xhue += (1 - val) * 10  # Use the val offset to give a slight colour trail variation
                    xhue %= 360  # Clamp to 0-359
                    xhue /= 360.0  # Convert to 0.0 to 1.0

                    r, g, b = [int(c * 255) for c in colorsys.hsv_to_rgb(xhue, sat, val)]
                    draw.line((x, 0, x, device.height), fill=(r, g, b, int(val * 255)))


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        pass
        

GPIO.cleanup()