import numpy as np

import time
import logging
import itertools

logging.basicConfig(level=0, format='%(message)s')
log = logging.getLogger(__name__)

import sigproc
from common import *

class Symbol(object):
    t       = np.arange(0, Nsym) * Ts
    carrier = [ np.exp(2j * np.pi * F * t) for F in frequencies ]

sym = Symbol()

data = open('data.send', 'r').read()

def write(fd, sym, n=1):
    fd.write(dumps(sym, n))

def start(sig, c):
    write(sig, c*0, n=100)
    write(sig, c*1, n=300)
    write(sig, c*0, n=100)

def train(sig, c):
    for i in range(20):
        write(sig, c*1, n=10)
        write(sig, c*0, n=10)
    write(sig, c*0, n=100)

def modulate(sig, bits):
    symbols_iter = sigproc.modulator.encode(list(bits))
    symbols_iter = itertools.chain(symbols_iter, itertools.repeat(0))
    carriers = np.array(sym.carrier) / len(sym.carrier)
    while True:
        symbols = itertools.islice(symbols_iter, len(sym.carrier))
        symbols = np.array(list(symbols))
        write(sig, np.dot(symbols, carriers))
        if all(symbols == 0):
            break

if __name__ == '__main__':

    import ecc
    bps = baud * sigproc.modulator.bits_per_symbol * len(sym.carrier)
    log.info('Running MODEM @ {:.1f} kbps'.format(bps / 1e3))

    with open('tx.int16', 'wb') as fd:
        start(fd, sym.carrier[carrier_index])
        for c in sym.carrier:
            train(fd, c)

        bits = to_bits(ecc.encode(data))
        modulate(fd, bits)


    from wave import play, record

    r = record('rx.int16')
    start = time.time()
    p = play(fd.name)
    p.wait()
    log.debug('Took %.2f seconds', time.time() - start)
    time.sleep(0.1)
    r.stop()