from lark import Transformer
from .classes import *
from .common import flatten, sum_dict
from .defaults import default_durs
import operator


class ZiffersTransformer(Transformer):
    def start(self, items):
        return Sequence(values=items[0])

    def sequence(self, items):
        return flatten(items)

    def random_integer(self, s):
        val = s[0][1:-1].split(",")
        return RandomInteger(min=val[0], max=val[1], text=s[0].value)

    def range(self, s):
        val = s[0].split("..")
        return Range(start=val[0], end=val[1], text=s[0])

    def cycle(self, items):
        values = items[0]
        return Cyclic(values=values)

    def pc(self, s):
        if len(s) > 1:
            # Collect&sum prefixes from any order: _qee^s4 etc.
            result = sum_dict(s)
            return Pitch(**result)
        else:
            val = s[0]
            return Pitch(**val)

    def pitch(self, s):
        return {"pc": int(s[0].value), "text": s[0].value}

    def prefix(self, s):
        return s[0]

    def oct_change(self, s):
        octave = s[0]
        return [OctaveChange(oct=octave["oct"], text=octave["text"]), s[1]]

    def oct_mod(self, s):
        octave = s[0]
        return [OctaveMod(oct=octave["oct"], text=octave["text"]), s[1]]

    def escaped_octave(self, s):
        value = s[0][1:-1]
        return {"oct": int(value), "text": s[0].value}

    def octave(self, s):
        value = sum([1 if char == "^" else -1 for char in s[0].value])
        return {"oct": value, "text": s[0].value}

    def chord(self, s):
        return Chord(pcs=s, text="".join([val.text for val in s]))

    def dur_change(self, s):
        durs = s[0]
        return DurationChange(dur=durs[1], text=durs[0])

    def char_change(self, s):
        chars = ""
        durs = 0.0
        for (dchar, dots) in s:
            val = default_durs[dchar]
            if dots > 0:
                val = val * (2.0 - (1.0 / (2 * dots)))
            chars = chars + (dchar + "." * dots)
            durs = durs + val
        return [chars, durs]

    def dchar_not_prefix(self, s):
        dur = s[0].split(".", 1)
        dots = 0
        if len(dur) > 1:
            dots = len(dur[1]) + 1
        return [dur[0], dots]

    def escaped_decimal(self, s):
        val = s[0]
        val["text"] = "<" + val["text"] + ">"
        return val

    def random_pitch(self, s):
        return RandomPitch(text="?")

    def random_percent(self, s):
        return RandomPercent(text="%")

    def duration_chars(self, s):
        durations = [val[1] for val in s]
        characters = "".join([val[0] for val in s])
        return {"dur": sum(durations), "text": characters}

    def dotted_dur(self, s):
        key = s[0]
        val = default_durs[key]
        dots = len(s) - 1
        if dots > 0:
            val = val * (2.0 - (1.0 / (2 * dots)))
        return [key + "." * dots, val]

    def decimal(self, s):
        val = s[0]
        return {"dur": float(val), "text": val.value}

    def dot(self, s):
        return "."

    def dchar(self, s):
        chardur = s[0].value
        return chardur

    def WS(self, s):
        return Whitespace(text=s[0])

    def subdivision(self, items):
        values = flatten(items[0])
        return Subdivision(
            values=values, text="[" + "".join([val.text for val in values]) + "]"
        )

    def subitems(self, s):
        return s

    # Eval rules

    def eval(self, s):
        val = s[0]
        return Eval(values=val)

    def operation(self, s):
        return s

    def atom(self, s):
        val = s[0].value
        return Atom(value=val, text=val)

    # List rules

    def list(self, items):
        if len(items) > 1:
            prefixes = sum_dict(items[0:-1])
            values = items[-1]
            seq = ListSequence(values=values, wrap_start=prefixes["text"] + "(")
            seq.update_values(prefixes)
            return seq
        else:
            seq = ListSequence(values=items[0])
            return seq

    def repeated_list(self, items):
        if len(items) > 2:
            prefixes = sum_dict(items[0:-2])  # If there are prefixes
            if items[-1] != None:
                seq = RepeatedListSequence(
                    values=items[-2],
                    repeats=items[-1],
                    wrap_end=":" + items[-1].text + ")",
                )
            else:
                seq = RepeatedListSequence(
                    values=items[-2], repeats=Integer(text="1", value=1)
                )
            seq.update_values(prefixes)
            return seq
        else:
            if items[-1] != None:
                seq = RepeatedListSequence(
                    values=items[-2],
                    repeats=items[-1],
                    wrap_end=":" + items[-1].text + ")",
                )
            else:
                seq = RepeatedListSequence(
                    values=items[-2], repeats=Integer(text="1", value=1)
                )
            return seq

    def SIGNED_NUMBER(self, s):
        val = s.value
        return Integer(text=val, value=int(val))

    def number(self, s):
        return s

    def cyclic_number(self, s):
        return Cyclic(values=s)

    def lisp_operation(self, s):
        op = s[0]
        values = s[1:]
        return Operation(
            operator=op,
            values=values,
            text="(+" + "".join([v.text for v in values]) + ")",
        )

    def operator(self, s):
        val = s[0].value
        return Operator(text=val)

    def list_items(self, s):
        return Sequence(values=s)

    def list_op(self, s):
        return ListOperation(values=s)

    def euclid(self, s):
        params = s[1][1:-1].split(",")
        init = {"onset": s[0], "pulses": params[0], "length": params[1]}
        text = s[0].text + s[1]
        if len(params) > 2:
            init["rotate"] = params[2]
        if len(s) > 2:
            init["offset"] = s[2]
            text = text + s[2].text
        init["text"] = text
        return Euclid(**init)

    def euclid_operator(self, s):
        return s.value

    def repeat(self, s):
        if s[-1] != None:
            return RepeatedSequence(
                values=s[0], repeats=s[-1], wrap_end=":" + s[-1].text + "]"
            )
        else:
            return RepeatedSequence(values=s[0], repeats=Integer(value=1, text="1"))