added setup.py

ledd/controller.py

@@ -0,0 +1,165 @@
+# LEDD Project
+# Copyright (C) 2015 LEDD Team
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from json import JSONEncoder
+
+import smbus
+from colour import Color
+
+
+PCA9685_SUBADR1 = 0x2
+PCA9685_SUBADR2 = 0x3
+PCA9685_SUBADR3 = 0x4
+
+PCA9685_MODE1 = 0x00
+PCA9685_MODE2 = 0x01
+PCA9685_PRESCALE = 0xFE
+PCA9685_RESET = 0xFE
+
+LED0_ON_L = 0x06
+LED0_ON_H = 0x07
+LED0_OFF_L = 0x08
+LED0_OFF_H = 0x09
+
+ALLLED_ON_L = 0xFA
+ALLLED_ON_H = 0xFB
+ALLLED_OFF_L = 0xFC
+ALLLED_OFF_H = 0xFD
+
+
+class Controller:
+    """
+    A controller controls a number of stripes.
+    """
+
+    @classmethod
+    def from_row(cls, db, row):
+        # load from db
+        return cls(db, pwm_freq=row["pwm_freq"], channels=row["channels"], i2c_device=row["i2c_device"],
+                   address=row["address"], cid=row["id"], from_db=True)
+
+    @staticmethod
+    def from_db(db):
+        l = []
+        cur = db.cursor()
+        for row in cur.execute("select * from controller"):
+            l.append(Controller.from_row(db, row))
+        cur.close()
+        return l
+
+    def save_to_db(self):
+        cur = self.db.cursor()
+        if self.id == -1:
+            cur.execute("INSERT INTO controller (pwm_freq, channels, i2c_device, address) VALUES (?,?,?,?)",
+                        (self.pwm_freq, self.channels, self.i2c_device, self.address))
+            self.id = cur.lastrowid
+        else:
+            cur.execute("UPDATE controller SET pwm_freq=?, channels=?, i2c_device=?, address=? WHERE id = ?",
+                        (self.pwm_freq, self.channels, self.i2c_device, self.address, self.id))
+        cur.close()
+        self.db.commit()
+
+    def __init__(self, db, channels, i2c_device, address, pwm_freq=-1, cid=-1, from_db=False):
+        self.pwm_freq = pwm_freq
+        self.channels = channels
+        self.i2c_device = i2c_device
+        self.bus = smbus.SMBus(i2c_device)
+        self.address = address
+        self.id = cid
+        self.db = db
+        self.stripes = []
+        self.load_stripes()
+        if not from_db:
+            self.save_to_db()
+
+    def load_stripes(self):
+        cur = self.db.cursor()
+        for stripe in cur.execute("select * from stripes where controller_id = ?", (self.id,)):
+            self.stripes.append(Stripe.from_db(self, stripe))
+        cur.close()
+
+    def __repr__(self):
+        return "<Controller stripes={} cid={}>".format(len(self.stripes), self.id)
+
+    def set_channel(self, channel, val):
+        self.bus.write_word_data(int(self.address, 16), LED0_OFF_L + 4 * channel, int(val * 4095))
+        self.bus.write_word_data(int(self.address, 16), LED0_ON_L + 4 * channel, 0)
+
+    def get_channel(self, channel):
+        return self.bus.read_word_data(self.address, LED0_OFF_L + 4 * channel)
+
+    def add_stripe(self, stripe):
+        self.stripes.append(stripe)
+
+
+class Stripe:
+    """
+    A stripe is the smallest controllable unit.
+    """
+
+    def __init__(self, controller, name, rgb, channels, sid=-1, from_db=False):
+        self.controller = controller
+        self.name = name
+        self.rgb = bool(rgb)
+        self.channels = channels
+        self.id = sid
+        self._color = Color()
+        self.gamma_correct = (2.8, 2.8, 2.8)  # TODO: add to DB
+        self.read_color()
+        if not from_db:
+            self.save_to_db()
+
+    def save_to_db(self):
+        cur = self.controller.db.cursor()
+        if self.id == -1:
+            cur.execute("INSERT INTO stripes DEFAULT VALUES")
+            self.id = cur.lastrowid
+        cur.execute(
+            "UPDATE stripes SET channel_r = ?, channel_g = ?, channel_b = ?,controller_id = ?, name = ? WHERE id = ?",
+            self.channels + [self.controller.id, self.name, self.id])
+        cur.close()
+        self.controller.db.commit()
+
+    def read_color(self):
+        self._color.rgb = [self.controller.get_channel(channel) ** (1 / 2.8) for channel in self.channels]
+
+    @classmethod
+    def from_db(cls, controller, row):
+        return cls(controller, name=row["name"], rgb=row["rgb"],
+                   channels=(row["channel_r"], row["channel_g"], row["channel_b"]), sid=row["id"], from_db=True)
+
+    def set_color(self, c):
+        self._color = c
+        for channel, gamma_correct, value in zip(self.channels, self.gamma_correct, c.rgb):
+            self.controller.set_channel(channel, value ** gamma_correct)
+
+    def get_color(self):
+        return self._color
+
+    color = property(get_color, set_color)
+
+
+class ControllerEncoder(JSONEncoder):
+    def default(self, o):
+        if isinstance(o, Controller):
+            return {
+                'id': o.id,
+                'pwm_freq': o.pwm_freq,
+                'channel': o.channels,
+                'address': o.address,
+                'stripes': o.stripes,
+                'i2c_device': o.i2c_device
+            }