mirror of
https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git
synced 2024-11-13 10:03:32 +01:00
38 lines
913 B
Python
38 lines
913 B
Python
# vim: fdm=indent ts=4
|
|
"""
|
|
Library for this simulation
|
|
"""
|
|
import numpy as np
|
|
|
|
from earsim import Antenna
|
|
|
|
def sine_beacon(f, t, t0=0, amplitude=1, baseline=0, phase=0):
|
|
"""
|
|
Return a sine appropriate as a beacon
|
|
"""
|
|
return amplitude * np.cos(2*np.pi*f*(t+t0) + phase) + baseline
|
|
|
|
|
|
def distance(x1, x2):
|
|
"""
|
|
Calculate the Euclidean distance between two locations x1 and x2
|
|
"""
|
|
|
|
assert type(x1) in [Antenna]
|
|
x1 = np.array([x1.x, x1.y, x1.z])
|
|
|
|
assert type(x2) in [Antenna]
|
|
x2 = np.array([x2.x, x2.y, x2.z])
|
|
|
|
return np.sqrt( np.sum( (x1-x2)**2 ) )
|
|
|
|
def beacon_from(tx, rx, f, t=0, t0=0, c_light=3e8, radiate_rsq=True, amplitude=1,**kwargs):
|
|
dist = distance(tx,rx)
|
|
t0 = t0 + dist/c_light
|
|
|
|
if radiate_rsq:
|
|
if np.isclose(dist, 0):
|
|
dist = 1
|
|
amplitude *= 1/(dist**2)
|
|
|
|
return sine_beacon(f, t, t0=t0, amplitude=amplitude,**kwargs)
|