35 lines
928 B
Python
35 lines
928 B
Python
|
from SpectralParameters import SpectralParams
|
||
|
|
|
||
|
|
|
||
|
|
class FModel:
|
||
|
|
def __init__(self, freq_bins, spectral_parameters: SpectralParams):
|
||
|
|
self.freq = freq_bins
|
||
|
|
self.sp_par = spectral_parameters
|
||
|
|
|
||
|
|
def fmodel(self):
|
||
|
|
"""Function return given model"""
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
class FBrune(FModel):
|
||
|
|
"""
|
||
|
|
Generate Brune's model source spectrum.
|
||
|
|
"""
|
||
|
|
def __init__(self, freq_bins, spectral_parameters: SpectralParams):
|
||
|
|
super().__init__(freq_bins, spectral_parameters)
|
||
|
|
|
||
|
|
def fmodel(self):
|
||
|
|
return self.sp_par.mo / (1 + (self.freq / self.sp_par.fo) ** 2)
|
||
|
|
|
||
|
|
|
||
|
|
class FBoatwright(FModel):
|
||
|
|
"""
|
||
|
|
Generate Boatwright's model source spectrum.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, freq_bins, spectral_parameters: SpectralParams):
|
||
|
|
super().__init__(freq_bins, spectral_parameters)
|
||
|
|
|
||
|
|
def fmodel(self):
|
||
|
|
return self.sp_par.mo / (1 + (self.freq / self.sp_par.fo) ** 4) ** 0.5
|