Secret wave function

secret_wave.py

import math as mt
import numpy as np


def sine(t, rising_time=mt.pi):
    return mt.sin(mt.pi * float(t) / rising_time)


def secret_wave(t, a, b):
    r = (t + float(a) / 2) % (a + b)

    t_prime = r - float(a) / 2

    if r <= a:
        return sine(t_prime, a)
    else:
        return sine(t_prime - float(a - b) / 2, b)

import matplotlib.pyplot as plt

x = np.arange(-10, 10, 0.1)
a, b = 4, 1
y = [secret_wave(t, a, b) for t in x]

plt.plot(x, y)

plt.axes().set_aspect('equal')

plt.show()