#!/usr/bin/env python3

import math
import os


BAUD = 1200
BAUD_MARK = 1200
BAUD_SPACE = 2200

N_SHIFTS = BAUD // math.gcd(BAUD, BAUD_MARK, BAUD_SPACE)


sinus1200 = []
sinus2200 = []

for t in range(int(1000000/BAUD_MARK)):
    sinus1200.append(int((math.sin((t * BAUD_MARK / 1000000) * 2*math.pi) + 1) * 128))
for t in range(int(1000000/BAUD_SPACE)):
    sinus2200.append(int((math.sin((t * BAUD_SPACE / 1000000) * 2*math.pi) + 1) * 128))

phaseShift1200 = [int((N_SHIFTS-i) * 1000000 / (6*BAUD_MARK)) % len(sinus1200) for i in range(N_SHIFTS)]
phaseShift2200 = [int((N_SHIFTS-i) * 1000000 / (6*BAUD_SPACE)) % len(sinus2200) for i in range(N_SHIFTS)]

with open(os.path.join(os.path.dirname(__file__), 'include/afsk_sinus.hpp'), 'w') as f:
    f.write(f'''
#include <avr/pgmspace.h>

#define BAUD {BAUD}
#define BAUD_MARK {BAUD_MARK}
#define BAUD_SPACE {BAUD_SPACE}
#define INV_BAUD {int(1000000/BAUD)}
#define N_SHIFTS {N_SHIFTS}
#define N_SINUS_1200 {len(sinus1200)}
#define N_SINUS_2200 {len(sinus2200)}

const PROGMEM uint8_t SINUS1200[] = {{ {', '.join([str(i) for i in sinus1200])} }};
const PROGMEM uint8_t SINUS2200[] = {{ {', '.join([str(i) for i in sinus2200])} }};
const size_t PS1200[] = {{ {', '.join([str(i) for i in phaseShift1200])} }};
const size_t PS2200[] = {{ {', '.join([str(i) for i in phaseShift2200])} }};

void sendOne(uint32_t *us, uint8_t *phaseShift) {{
    uint32_t t;
    while ((t = micros() - *us) < INV_BAUD) {{
        OUT_PORT = pgm_read_byte(&SINUS1200[ (PS1200[*phaseShift] + t) % N_SINUS_1200 ]);
    }}
    *us += INV_BAUD;
}}

void sendZero(uint32_t *us, uint8_t *phaseShift) {{
    uint32_t t;
    while ((t = micros() - *us) < INV_BAUD) {{
        OUT_PORT = pgm_read_byte(&SINUS2200[ (PS2200[*phaseShift] + t) % N_SINUS_2200 ]);
    }}
    (*phaseShift)++;
    *us += INV_BAUD;
}}

''')


#from matplotlib import pyplot as plt
#
#phase_shift = 0
#
#signal = []
#
#
#def one():
#    global phase_shift
#    for i in range(int(1000000/BAUD)):
#        signal.append(sinus1200[int((phaseShift1200[phase_shift]+i) % len(sinus1200))])
#
#def zero():
#    global phase_shift
#    for i in range(int(1000000/BAUD)):
#        signal.append(sinus2200[int((phaseShift2200[phase_shift]+i) % len(sinus2200))])
#    phase_shift = (phase_shift + 1) % len(phaseShift2200)
#
#
#for i in range(20):
#    if i % 3 == 0:
#        one()
#    else:
#        zero()
#
#
#plt.plot(signal)
#plt.vlines([i*int(1000000/BAUD) for i in range(21)], ymin=0, ymax=255, color='red')
#plt.show()