from rdflib import Literal, URIRef

from light9.namespaces import L9, DEV

from webcolors import rgb_to_hex, hex_to_rgb

from colorsys import hsv_to_rgb

import math

from noise import pnoise1

import logging

from light9.effect.settings import DeviceSettings

from light9.effect.scale import scale

from typing import Dict, Tuple, Any

import random

random.seed(0)

log = logging.getLogger('effecteval')

log.info("reload effecteval")

def literalColor(rnorm, gnorm, bnorm):

    return Literal(

        rgb_to_hex([int(rnorm * 255),

                    int(gnorm * 255),

                    int(bnorm * 255)]))

def literalColorHsv(h, s, v):

    return literalColor(*hsv_to_rgb(h, s, v))

def nsin(x):

    return (math.sin(x * (2 * math.pi)) + 1) / 2

def ncos(x):

    return (math.cos(x * (2 * math.pi)) + 1) / 2

    duty = .3

    offset = 0

    f = (((songTime + offset) * rate) % 1.0)

    c = (f < duty) * strength

    col = rgb_to_hex([int(c * 255), int(c * 255), int(c * 255)])

    return {(L9['device/colorStrip'], L9['color']): Literal(col)}

def effect_lightning(effectSettings, strength, songTime, noteTime):

    devs = [

        L9['device/veryLow1'], L9['device/veryLow2'], L9['device/veryLow3'],

        L9['device/veryLow4'], L9['device/veryLow5'], L9['device/backlight1'],

        L9['device/backlight2'], L9['device/backlight3'],

        L9['device/backlight4'], L9['device/backlight5'], L9['device/down2'],

        L9['device/down3'], L9['device/down4'], L9['device/hexLow3'],

        L9['device/hexLow5'], L9['device/postL1'], L9['device/postR1']

    ]

    out = {}

    col = rgb_to_hex([int(255 * strength)] * 3)

    for i, dev in enumerate(devs):

        n = noise(songTime * 8 + i * 6.543)

        if n > .4:

            out[(dev, L9['color'])] = col

    return out

'''

copies from effectloop.py, which this should replace

'''

from louie import dispatcher

from rdflib import URIRef

from twisted.internet import reactor

from twisted.internet import defer

from twisted.internet.defer import Deferred, inlineCallbacks

from twisted.internet.inotify import INotify

from twisted.python.filepath import FilePath

import cyclone.sse

import logging, bisect, time

import traceback

from typing import Any, Callable, Dict, List, Tuple, cast, Union

from light9.ascoltami.musictime_client import MusicTime

from light9.effect import effecteval

from light9.effect.settings import DeviceSettings

from light9.effect.simple_outputs import SimpleOutputs

from light9.namespaces import L9, RDF

from light9.newtypes import DeviceUri, DeviceAttr, NoteUri, Curve, Song

from rdfdb.syncedgraph import SyncedGraph

from standardservice.scalessetup import gatherProcessStats

from greplin import scales

import imp

log = logging.getLogger('sequencer')

gatherProcessStats()

updateStats = scales.collection(

    '/update',

    scales.PmfStat('s0_getMusic', recalcPeriod=1),

    scales.PmfStat('s1_eval', recalcPeriod=1),

    #scales.PmfStat('s3_send_client', recalcPeriod=1),

    scales.PmfStat('s3_send', recalcPeriod=1),

    scales.PmfStat('updateLoopLatency', recalcPeriod=1),

    scales.DoubleStat('updateLoopLatencyGoal'),

    scales.RecentFpsStat('updateFps'),

    scales.DoubleStat('goalFps'),

)

compileStats = scales.collection(

    '/compile',

    scales.PmfStat('graph', recalcPeriod=1),

    scales.PmfStat('song', recalcPeriod=1),

)

class Note(object):

    def __init__(self, graph: SyncedGraph, uri: NoteUri, effectevalModule,

                 simpleOutputs):

        g = self.graph = graph

        self.uri = uri

        self.effectEval = effectevalModule.EffectEval(

            graph, g.value(uri, L9['effectClass']), simpleOutputs)

        self.baseEffectSettings: Dict[URIRef, Any] = {}  # {effectAttr: value}

        for s in g.objects(uri, L9['setting']):

            settingValues = dict(g.predicate_objects(s))

            ea = settingValues[L9['effectAttr']]

        def floatVal(s, p):

            return float(g.value(s, p).toPython())

        originTime = floatVal(uri, L9['originTime'])

        self.points: List[Tuple[float, float]] = []

        for curve in g.objects(uri, L9['curve']):

            self.points.extend(

                self.getCurvePoints(curve, L9['strength'], originTime))

        self.points.sort()

    def getCurvePoints(self, curve: Curve, attr,

                       originTime: float) -> List[Tuple[float, float]]:

        points = []

        po = list(self.graph.predicate_objects(curve))

        if dict(po).get(L9['attr'], None) != attr:

            return []

        for point in [row[1] for row in po if row[0] == L9['point']]:

            po2 = dict(self.graph.predicate_objects(point))

            points.append(

                (originTime + float(po2[L9['time']]), float(po2[L9['value']])))

        return points

    def activeAt(self, t: float) -> bool:

        if i == -1:

            return self.points[0][1]

        if self.points[i][0] > t:

            return self.points[i][1]

        if i >= len(self.points) - 1:

            return self.points[i][1]

        p1, p2 = self.points[i], self.points[i + 1]

        frac = (t - p1[0]) / (p2[0] - p1[0])

        y = p1[1] + (p2[1] - p1[1]) * frac

        return y

    def outputSettings(

            self,

            t: float) -> Tuple[List[Tuple[DeviceUri, DeviceAttr, float]], Dict]:

        """

        list of (device, attr, value), and a report for web

        """

        report = {

            'note': str(self.uri),

            'effectClass': self.effectEval.effect,

        }

        effectSettings: Dict[DeviceAttr, Union[float, str]] = dict(

            for da, v in self.baseEffectSettings.items())

        effectSettings[L9['strength']] = self.evalCurve(t)

        def prettyFormat(x: Union[float, str]):

            if isinstance(x, float):

                return round(x, 4)

            return x

        report['effectSettings'] = dict(

            (str(k), prettyFormat(v))

            for k, v in sorted(effectSettings.items()))

        report['nonZero'] = cast(float, effectSettings[L9['strength']]) > 0

        out, evalReport = self.effectEval.outputFromEffect(

            list(effectSettings.items()),

            songTime=t,

            # note: not using origin here since it's going away

            noteTime=t - self.points[0][0])

        report['devicesAffected'] = len(out.devices())

        return out, report

class CodeWatcher(object):

    def __init__(self, onChange):

    @inlineCallbacks

    def updateLoop(self) -> None:

        frameStart = time.time()

        try:

            sec = yield self.update()

        except Exception as e:

            self.lastLoopSucceeded = False

            traceback.print_exc()

            log.warn('updateLoop: %r', e)

            reactor.callLater(1, self.updateLoop)

        else:

            took = time.time() - frameStart

            updateStats.updateLoopLatency = took

            if not self.lastLoopSucceeded:

                log.info('Sequencer.update is working')

                self.lastLoopSucceeded = True

            delay = max(0, 1 / self.fps - took)

            reactor.callLater(delay, self.updateLoop)

    @updateStats.updateFps.rate()

    @inlineCallbacks

    def update(self) -> Deferred:

        with updateStats.s0_getMusic.time():

            musicState = self.music.getLatest()

            if not musicState.get('song') or not isinstance(

                    musicState.get('t'), float):

                return defer.succeed(0.0)

            song = Song(URIRef(musicState['song']))

            dispatcher.send('state',

                            update={

                                'song': str(song),

                                't': musicState['t']

                            })

        with updateStats.s1_eval.time():

            settings = []

            songNotes = sorted(self.notes.get(song, []),

                               key=lambda n: n.uri)

            noteReports = []

            for note in songNotes:

                try:

                    s, report = note.outputSettings(musicState['t'])

                except Exception:

                    traceback.print_exc()

                    raise

                noteReports.append(report)