# Copyright (C) 2014, 2015 Robert Jordens <jordens@gmail.com>

import unittest

import numpy as np

from artiq.wavesynth import coefficients, compute_samples


class TestSplineCoef(unittest.TestCase):
    def setUp(self):
        self.x = np.arange(5.)
        self.y = np.sin(2*np.pi*self.x/5) + np.arange(2)[:, None]
        self.s = coefficients.SplineSource(self.x, self.y, order=4)

    def test_get_segment(self):
        return list(self.s.get_segment(start=1.5, stop=3.2, scale=.01))

    def test_synth(self):
        d = self.test_get_segment()
        d[0]["trigger"] = True
        return compute_samples.Synthesizer(self.y.shape[0], [d, d + d])

    def drive(self, s):
        y = []
        for f in 0, 1, None, 0:
            if f is not None:
                s.select(f)
            y += s.trigger()[0]
        return y

    def test_run(self):
        return self.drive(self.test_synth())

    def test_compare(self):
        scale = 100
        d = list(self.s.get_segment(start=0, stop=4, scale=1/scale))
        d[0]["trigger"] = True
        s = compute_samples.Synthesizer(self.y.shape[0], [d])
        s.select(0)
        y = s.trigger()[0]
        np.testing.assert_almost_equal(y[::scale], self.y[0, :-1])

    @unittest.skip("manual/visual test")
    def test_plot(self):
        import matplotlib.pyplot as plt
        y = self.test_run()
        plt.step(np.arange(len(y)), y)
        plt.show()