from __future__ import print_function
from math import log10
from obspy import UTCDateTime, read
from obspy.geodetics import gps2dist_azimuth
st = read("../data/LKBD.MSEED")
paz_le3d5s = {'gain': 1.009,
'poles': [-0.885 + 0.887j,
-0.885 - 0.887j,
-0.427 + 0j],
'sensitivity': 167364000.0,
'zeros': [0j, 0j, 0j]}
paz_wa = {'sensitivity': 2800, 'zeros': [0j], 'gain': 1,
'poles': [-6.2832 - 4.7124j, -6.2832 + 4.7124j]}
st.simulate(paz_remove=paz_le3d5s, paz_simulate=paz_wa, water_level=10)
t = UTCDateTime("2012-04-03T02:45:03")
st.trim(t, t + 50)
tr_n = st.select(component="N")[0]
ampl_n = max(abs(tr_n.data))
tr_e = st.select(component="E")[0]
ampl_e = max(abs(tr_e.data))
ampl = max(ampl_n, ampl_e)
sta_lat = 46.38703
sta_lon = 7.62714
event_lat = 46.218
event_lon = 7.706
epi_dist, az, baz = gps2dist_azimuth(event_lat, event_lon, sta_lat, sta_lon)
epi_dist = epi_dist / 1000
a = 0.018
b = 2.17
ml = log10(ampl * 1000) + a * epi_dist + b
print(ml)