import functools
import itertools
import numpy as np

import logging
log = logging.getLogger(__name__)

Fs = 32e3
Ts = 1.0 / Fs

frequencies = (1 + np.arange(8)) * 1e3
carrier_index = 0
Fc = frequencies[carrier_index]
Tc = 1.0 / Fc

symbols = np.array([complex(x, y)
                   for x in np.linspace(-1, 1, 8)
                   for y in np.linspace(-1, 1, 8)]) / np.sqrt(2)

Tsym = 1e-3
Nsym = int(Tsym / Ts)
baud = int(1/Tsym)

scaling = 32000.0  # out of 2**15
SATURATION_THRESHOLD = 1.0

LENGTH_FORMAT = '<I'


def to_bits(bytes_list):
    for val in bytes_list:
        for i in range(8):
            mask = 1 << i
            yield (1 if (val & mask) else 0)

bit_weights = [1 << i for i in range(8)]


def to_byte(bits):
    assert len(bits) == 8
    byte = int(np.dot(bits, bit_weights))
    return chr(byte)


def load(fileobj):
    return loads(fileobj.read())


def loads(data):
    x = np.fromstring(data, dtype='int16')
    x = x / scaling
    t = np.arange(len(x)) / Fs
    return t, x


def dumps(sym, n=1):
    sym = sym.imag * scaling
    sym = sym.astype('int16')
    data = sym.tostring()
    return data * n


def iterate(data, size, func=None):
    offset = 0
    data = iter(data)

    while True:
        buf = list(itertools.islice(data, size))
        if len(buf) < size:
            return

        buf = np.array(buf)
        result = func(buf) if func else buf
        yield offset, result
        offset += size


class Splitter(object):

    def __init__(self, iterable, n):
        self.iterable = iter(iterable)
        self.read = [True] * n
        self.last = None
        self.generators = [functools.partial(self._gen, i)() for i in range(n)]
        self.n = n

    def _gen(self, index):
        while True:
            if all(self.read):
                try:
                    self.last = self.iterable.next()
                except StopIteration:
                    return

                assert len(self.last) == self.n
                self.read = [False] * self.n

            if self.read[index]:
                raise IndexError(index)
            self.read[index] = True
            yield self.last[index]


def split(iterable, n):
    return Splitter(iterable, n).generators


def icapture(iterable, result):
    for i in iter(iterable):
        result.append(i)
        yield i

if __name__ == '__main__':

    import pylab
    t = pylab.linspace(0, Tsym, 1e3)
    x = pylab.sin(2 * pylab.pi * Fc * t)
    pylab.plot(t / Tsym, x)
    t = pylab.linspace(0, Tsym, Nsym + 1)
    x = pylab.sin(2 * pylab.pi * Fc * t)
    pylab.plot(t / Tsym, x, '.k')
    pylab.show()