# 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 spectra import Color
from sqlalchemy import Integer, ForeignKey, String, Float, Boolean
from sqlalchemy import Column
from sqlalchemy.orm import reconstructor

from . import Base


class Stripe(Base):
    """
    A stripe is the smallest controllable unit.
    """
    __tablename__ = "stripe"
    id = Column(Integer, primary_key=True)
    controller_id = Column(Integer, ForeignKey('controller.id'))
    name = Column(String)
    channel_r = Column(Integer)
    channel_g = Column(Integer)
    channel_b = Column(Integer)
    channel_r_gamma = Column(Float, default=2.8)
    channel_g_gamma = Column(Float, default=2.8)
    channel_b_gamma = Column(Float, default=2.8)
    rgb = Column(Boolean)

    @property
    def channels(self):
        return self.channel_r, self.channel_g, self.channel_b

    # TODO save channels to db

    @channels.setter
    def channels(self, t):
        self.channel_r, self.channel_g, self.channel_b = t

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._color = None
        self.gamma_correct = (self.channel_r_gamma, self.channel_g_gamma, self.channel_b_gamma)
        self.read_color()

    @reconstructor
    def init_on_load(self):
        self._color = None
        self.gamma_correct = (self.channel_r_gamma, self.channel_g_gamma, self.channel_b_gamma)
        self.read_color()

    def read_color(self):
        if self.controller:
            rc = tuple([float(self.controller.get_channel(channel)) for channel in self.channels])
            c = Color("rgb", rc[0], rc[1], rc[2])
            self._color = c.to("hsv")

    def __repr__(self):
        return "<Stripe id={}>".format(self.id)

    def set_color(self, c):
        self._color = c
        for channel, gamma_correct, value in zip(self.channels, self.gamma_correct, c.clamped_rgb):
            self.controller.set_channel(channel, value, gamma_correct)

    def get_color(self):
        return self._color

    def to_json(self):
        return {
            'id': self.id,
            'name': self.name,
            'rgb': self.rgb,
            'channel': self.channels
        }

    color = property(get_color, set_color)