m-thesis-introduction/simulations/airshower_beacon_simulation/lib.py

32 lines
762 B
Python

"""
Library for this simulation
"""
import numpy as np
from earsim import Antenna
def sine_beacon(f, t, t0=0, amplitude=1, baseline=0):
return amplitude * np.sin(2*np.pi*f*(t-t0)) + 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:
amplitude *= 1/(dist**2)
return sine_beacon(f, t, t0=t0, amplitude=amplitude,**kwargs)