splitting a part into separate files & use the OOP approach

Author: jketterl

Adafruit_PWM_Servo_Driver/LEDDriver.py

@@ -2,19 +2,14 @@
 
 from Adafruit_PWM_Servo_Driver import PWM
 import time
+from RGBDevice import RGBDevice
 
 # Initialise the PWM device using the default address
 pwm = PWM(0x40, debug=True)
 
 pwm.setPWMFreq(600)                        # Set frequency to 600 Hz
 
-master = 1
-
-def setColor(red, green, blue):
-  #print(str(red) + ', ' + str(green) + ', ' + str(blue))
-  pwm.setPWM(1, 0, int(red * master))
-  pwm.setPWM(2, 0, int(green * master))
-  pwm.setPWM(0, 0, int(blue * master))
+device = RGBDevice(pwm, 0)
 
 def sleep():
   time.sleep(.01)
@@ -23,33 +18,33 @@ def sleep():
 
 while (True):
   for red in range(0, 4095, stepSize):
-    setColor(red, 0, 0)
+    device.setRGB(red, 0, 0)
     sleep()
-  setColor(4095, 0, 0)
+  device.setRGB(4095, 0, 0)
   time.sleep(1)
   for green in range(0, 4095, stepSize):
-    setColor(4095-green, green, 0)
+    device.setRGB(4095-green, green, 0)
     sleep()
-  setColor(0, 4095, 0)
+  device.setRGB(0, 4095, 0)
   time.sleep(1)
   for blue in range(0, 4095, stepSize):
-    setColor(0, 4095-blue, blue)
+    device.setRGB(0, 4095-blue, blue)
     sleep()
-  setColor(0, 0, 4095)
+  device.setRGB(0, 0, 4095)
   time.sleep(1)
   for red in range(0, 4095, stepSize):
-    setColor(red, 0, 4095-red)
+    device.setRGB(red, 0, 4095-red)
     sleep()
-  setColor(4095, 0, 0)
+  device.setRGB(4095, 0, 0)
   time.sleep(1)
   for white in range(0, 4095, stepSize):
-    setColor(4095, white, white)
+    device.setRGB(4095, white, white)
     sleep()
-  setColor(4095, 4095, 4095)
+  device.setRGB(4095, 4095, 4095)
   time.sleep(1)
   for white in range(0, 4095, stepSize):
-    setColor(4095-white, 4095-white, 4095-white)
+    device.setRGB(4095-white, 4095-white, 4095-white)
     sleep()
-  setColor(0, 0, 0)
+  device.setRGB(0, 0, 0)
   time.sleep(1)
 

Adafruit_PWM_Servo_Driver/RGBDevice.py

@@ -0,0 +1,10 @@
+class RGBDevice:
+	def __init__(self, pwm, baseAddress):
+		self.pwm = pwm
+		self.baseAddress = baseAddress
+		self.master = 1
+
+	def setRGB(self, red, green, blue):
+		self.pwm.setPWM(self.baseAddress + 1, 0, int(red * self.master))
+		self.pwm.setPWM(self.baseAddress + 2, 0, int(green * self.master))
+		self.pwm.setPWM(self.baseAddress + 0, 0, int(blue * self.master))