beer/models.py at master · tcjansen/beer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import numpy as np
import matplotlib.pyplot as plt

def flat(x, a0):
	return np.zeros(len(x)) + (a0 * 1e-6 + 1)

def therm(x, a0, a1):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos(x * 2 * np.pi)

def ellip(x, a0, a1):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos(4 * np.pi * x)

def shifted_therm(x, a0, a1, phi):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos((x + phi) * 2 * np.pi)

def therm_ellip(x, a0, a1, a2):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos(x * 2 * np.pi) - 10**a2 * np.cos(4 * np.pi * x)

def therm_beam(x, a0, a1, a2):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos(x * 2 * np.pi) + 10**a2 * np.sin(2 * np.pi * x)

def ellip_beam(x, a0, a1, a2):
	return (a0 * 1e-6 + 1) - 10**a1 * np.cos(4 * np.pi * x) + 10**a2 * np.sin(2 * np.pi * x)