added slow and fast fade animations

linux/caselights

@@ -7,6 +7,9 @@
 
 import sys
 import os.path
+import threading
+import time
+import colorsys
 import serial, serial.tools, serial.tools.list_ports
 from PyQt5 import QtWidgets, QtGui, QtCore
 from PyQt5.QtWidgets import QSystemTrayIcon, QAction, QMenu
@@ -29,8 +32,16 @@ class CaseLights():
         [ "White", "255", "255", "255", None ],
     ]
 
+    slowFadeUpdateFreq = 5
+    fastFadeUpdateFreq = 20
+    fadeSaturation = 1.0
+    fadeValue = 1.0
+    fadeHueCounter = 0
+
     usedPort = None
     serial = None
+    animation = None
+    animationRunning = False
     menu = None
     portMenu = None
     portActions = None
@@ -58,6 +69,18 @@ class CaseLights():
         colorMenu.triggered.connect(self.setStaticColor)
         self.menu.addMenu(colorMenu)
 
+        animMenu = QMenu("&Animations")
+        noFadeAction = QAction("Off")
+        noFadeAction.triggered.connect(self.animOff)
+        animMenu.addAction(noFadeAction)
+        slowFadeAction = QAction("Slow Fade")
+        slowFadeAction.triggered.connect(self.slowFadeOn)
+        animMenu.addAction(slowFadeAction)
+        fastFadeAction = QAction("Fast Fade")
+        fastFadeAction.triggered.connect(self.fastFadeOn)
+        animMenu.addAction(fastFadeAction)
+        self.menu.addMenu(animMenu)
+
         lightMenu = QMenu("&UV-Light")
         lightOnAction = QAction("O&n")
         lightOnAction.triggered.connect(self.lightsOn)
@@ -97,6 +120,8 @@ class CaseLights():
     def exit(self):
         if self.serial is not None:
             if self.serial.is_open:
+                print("stopping animations")
+                self.animOff()
                 print("turning off lights")
                 self.serial.write(b'RGB 0 0 0\n')
                 self.serial.write(b'UV 0\n')
@@ -162,20 +187,56 @@ class CaseLights():
             print("error connecting to: " + self.usedPort)
         return self.serial.is_open
 
+    def printRGBStrings(self, rs, gs, bs):
+        if self.serial.is_open:
+            r = str.encode(rs)
+            g = str.encode(gs)
+            b = str.encode(bs)
+            rgb = b'RGB ' + r + b' ' + g + b' ' + b + b'\n'
+            self.serial.write(rgb)
+        else:
+            print("not connected")
+
     def setStaticColor(self, action):
+        self.animOff()
         for color in self.staticColors:
             if color[4] is action:
-                if self.serial.is_open:
-                    r = str.encode(color[1])
-                    g = str.encode(color[2])
-                    b = str.encode(color[3])
-                    self.serial.write(b'RGB ' + r + b' ' + g + b' ' + b + b'\n')
-                else:
-                    print("not connected")
+                self.printRGBStrings(color[1], color[2], color[3])
                 return True
         print("color not found")
         return False
 
+    def hsvToRgb(self, h, s, v):
+        (r, g, b) = colorsys.hsv_to_rgb(h, s, v)
+        return (round(r * 255), round(g * 255), round(b * 255))
+
+    def fadeRunner(self, freq):
+        while self.animationRunning is True:
+            self.fadeHueCounter += 1
+            if self.fadeHueCounter >= 360:
+                self.fadeHueCounter = 0
+            (r, g, b) = self.hsvToRgb(self.fadeHueCounter / 360.0, self.fadeSaturation, self.fadeValue)
+            self.printRGBStrings(str(r), str(g), str(b))
+            time.sleep(1.0 / freq)
+
+    def slowFadeOn(self):
+        self.animOff()
+        self.animationRunning = True
+        self.animation = threading.Thread(target=self.fadeRunner, args=[self.slowFadeUpdateFreq])
+        self.animation.start()
+
+    def fastFadeOn(self):
+        self.animOff()
+        self.animationRunning = True
+        self.animation = threading.Thread(target=self.fadeRunner, args=[self.fastFadeUpdateFreq])
+        self.animation.start()
+
+    def animOff(self):
+        self.animationRunning = False
+        self.animation = None
+        self.printRGBStrings("0", "0", "0")
+        time.sleep(0.1)
+
     def lightsOn(self):
         if self.serial.is_open:
             self.serial.write(b'UV 1\n')