def test_fit_spectra():
config = get_config()
datapath = os.path.join("data", "testdata", "demo", "ci38457511", "raw")
datadir = pkg_resources.resource_filename("gmprocess", datapath)
event = get_event_object("ci38457511")
sc = StreamCollection.from_directory(datadir)
for st in sc:
st = signal_split(st, event)
end_conf = config["windows"]["signal_end"]
st = signal_end(st,
event_time=event.time,
event_lon=event.longitude,
event_lat=event.latitude,
event_mag=event.magnitude,
**end_conf)
st = compute_snr(st, 30)
st = get_corner_frequencies(st,
event,
method="constant",
constant={
"highpass": 0.08,
"lowpass": 20.0
})
for st in sc:
spectrum.fit_spectra(st, event)
def test_fit_spectra():
config = get_config()
datapath = os.path.join('data', 'testdata', 'demo', 'ci38457511', 'raw')
datadir = pkg_resources.resource_filename('gmprocess', datapath)
event = get_event_object('ci38457511')
sc = StreamCollection.from_directory(datadir)
for st in sc:
st = signal_split(st, event)
end_conf = config['windows']['signal_end']
st = signal_end(st,
event_time=event.time,
event_lon=event.longitude,
event_lat=event.latitude,
event_mag=event.magnitude,
**end_conf)
st = compute_snr(st, 30)
st = get_corner_frequencies(st,
method='constant',
constant={
'highpass': 0.08,
'lowpass': 20.0
})
for st in sc:
spectrum.fit_spectra(st, event)
def test_corner_frequencies():
# Default config has 'constant' corner frequency method, so the need
# here is to force the 'snr' method.
data_files, origin = read_data_dir("geonet", "us1000778i", "*.V1A")
streams = []
for f in data_files:
streams += read_data(f)
sc = StreamCollection(streams)
config = get_config()
window_conf = config["windows"]
processed_streams = sc.copy()
for st in processed_streams:
if st.passed:
# Estimate noise/signal split time
event_time = origin.time
event_lon = origin.longitude
event_lat = origin.latitude
st = signal_split(st, origin)
# Estimate end of signal
end_conf = window_conf["signal_end"]
event_mag = origin.magnitude
print(st)
st = signal_end(st,
event_time=event_time,
event_lon=event_lon,
event_lat=event_lat,
event_mag=event_mag,
**end_conf)
wcheck_conf = window_conf["window_checks"]
st = window_checks(
st,
min_noise_duration=wcheck_conf["min_noise_duration"],
min_signal_duration=wcheck_conf["min_signal_duration"],
)
pconfig = config["processing"]
# Run SNR check
# I think we don't do this anymore.
test = [d for d in pconfig if list(d.keys())[0] == "compute_snr"]
snr_config = test[0]["compute_snr"]
snr_config["check"]["min_freq"] = 0.2
for stream in processed_streams:
stream = compute_snr(stream, mag=origin.magnitude, **snr_config)
# Run get_corner_frequencies
test = [
d for d in pconfig if list(d.keys())[0] == "get_corner_frequencies"
]
cf_config = test[0]["get_corner_frequencies"]
snr_config = cf_config["snr"]
# With same_horiz False
snr_config["same_horiz"] = False
lp = []
hp = []
for stream in processed_streams:
if not stream.passed:
continue
stream = get_corner_frequencies(stream, method="snr", snr=snr_config)
if stream[0].hasParameter("corner_frequencies"):
cfdict = stream[0].getParameter("corner_frequencies")
lp.append(cfdict["lowpass"])
hp.append(cfdict["highpass"])
np.testing.assert_allclose(np.sort(hp),
[0.00467919, 0.00584742, 0.01026485],
atol=1e-6)
st = processed_streams.select(station="HSES")[0]
lps = [tr.getParameter("corner_frequencies")["lowpass"] for tr in st]
hps = [tr.getParameter("corner_frequencies")["highpass"] for tr in st]
np.testing.assert_allclose(np.sort(lps), [100.0, 100.0, 100.0], atol=1e-6)
np.testing.assert_allclose(np.sort(hps),
[0.00542478, 0.01026485, 0.02527502],
atol=1e-6)
# With same_horiz True
snr_config["same_horiz"] = True
lp = []
hp = []
for stream in processed_streams:
if not stream.passed:
continue
stream = get_corner_frequencies(stream, method="snr", snr=snr_config)
if stream[0].hasParameter("corner_frequencies"):
cfdict = stream[0].getParameter("corner_frequencies")
lp.append(cfdict["lowpass"])
hp.append(cfdict["highpass"])
np.testing.assert_allclose(np.sort(hp),
[0.00467919, 0.01026485, 0.01787214],
atol=1e-6)
st = processed_streams.select(station="HSES")[0]
lps = [tr.getParameter("corner_frequencies")["lowpass"] for tr in st]
hps = [tr.getParameter("corner_frequencies")["highpass"] for tr in st]
np.testing.assert_allclose(np.sort(lps), [100.0, 100.0, 100.0], atol=1e-6)
np.testing.assert_allclose(np.sort(hps), [0.010265, 0.010265, 0.025275],
atol=1e-6)
def test_corner_frequencies():
# Default config has 'constant' corner frequency method, so the need
# here is to force the 'snr' method.
data_files, origin = read_data_dir('geonet', 'us1000778i', '*.V1A')
streams = []
for f in data_files:
streams += read_data(f)
sc = StreamCollection(streams)
config = get_config()
window_conf = config['windows']
processed_streams = sc.copy()
for st in processed_streams:
if st.passed:
# Estimate noise/signal split time
event_time = origin.time
event_lon = origin.longitude
event_lat = origin.latitude
st = signal_split(st, origin)
# Estimate end of signal
end_conf = window_conf['signal_end']
event_mag = origin.magnitude
print(st)
st = signal_end(st,
event_time=event_time,
event_lon=event_lon,
event_lat=event_lat,
event_mag=event_mag,
**end_conf)
wcheck_conf = window_conf['window_checks']
st = window_checks(
st,
min_noise_duration=wcheck_conf['min_noise_duration'],
min_signal_duration=wcheck_conf['min_signal_duration'])
pconfig = config['processing']
# Run SNR check
# I think we don't do this anymore.
test = [d for d in pconfig if list(d.keys())[0] == 'compute_snr']
snr_config = test[0]['compute_snr']
snr_config['check']['min_freq'] = 0.2
for stream in processed_streams:
stream = compute_snr(stream, mag=origin.magnitude, **snr_config)
# Run get_corner_frequencies
test = [
d for d in pconfig if list(d.keys())[0] == 'get_corner_frequencies'
]
cf_config = test[0]['get_corner_frequencies']
snr_config = cf_config['snr']
# With same_horiz False
snr_config['same_horiz'] = False
lp = []
hp = []
for stream in processed_streams:
if not stream.passed:
continue
stream = get_corner_frequencies(stream, method="snr", snr=snr_config)
if stream[0].hasParameter('corner_frequencies'):
cfdict = stream[0].getParameter('corner_frequencies')
lp.append(cfdict['lowpass'])
hp.append(cfdict['highpass'])
np.testing.assert_allclose(np.sort(hp), [0.003052, 0.003052, 0.010265],
atol=1e-6)
st = processed_streams.select(station='HSES')[0]
lps = [tr.getParameter('corner_frequencies')['lowpass'] for tr in st]
hps = [tr.getParameter('corner_frequencies')['highpass'] for tr in st]
np.testing.assert_allclose(np.sort(lps), [100., 100., 100.], atol=1e-6)
np.testing.assert_allclose(np.sort(hps), [0.003052, 0.010265, 0.025275],
atol=1e-6)
# With same_horiz True
snr_config['same_horiz'] = True
lp = []
hp = []
for stream in processed_streams:
if not stream.passed:
continue
stream = get_corner_frequencies(stream, method="snr", snr=snr_config)
if stream[0].hasParameter('corner_frequencies'):
cfdict = stream[0].getParameter('corner_frequencies')
lp.append(cfdict['lowpass'])
hp.append(cfdict['highpass'])
np.testing.assert_allclose(np.sort(hp), [0.003052, 0.010265, 0.017872],
atol=1e-6)
st = processed_streams.select(station='HSES')[0]
lps = [tr.getParameter('corner_frequencies')['lowpass'] for tr in st]
hps = [tr.getParameter('corner_frequencies')['highpass'] for tr in st]
np.testing.assert_allclose(np.sort(lps), [100., 100., 100.], atol=1e-6)
np.testing.assert_allclose(np.sort(hps), [0.010265, 0.010265, 0.025275],
atol=1e-6)
def test_corner_frequencies_magnitude():
# Default config has 'constant' corner frequency method, so the need
# here is to force the 'magnitude' method.
data_files, origin = read_data_dir("geonet", "us1000778i", "*.V1A")
streams = []
for f in data_files:
streams += read_data(f)
sc = StreamCollection(streams)
config = get_config()
window_conf = config["windows"]
processed_streams = sc.copy()
for st in processed_streams:
if st.passed:
# Estimate noise/signal split time
event_time = origin.time
event_lon = origin.longitude
event_lat = origin.latitude
st = signal_split(st, origin)
# Estimate end of signal
end_conf = window_conf["signal_end"]
event_mag = origin.magnitude
print(st)
st = signal_end(
st,
event_time=event_time,
event_lon=event_lon,
event_lat=event_lat,
event_mag=event_mag,
**end_conf
)
wcheck_conf = window_conf["window_checks"]
st = window_checks(
st,
min_noise_duration=wcheck_conf["min_noise_duration"],
min_signal_duration=wcheck_conf["min_signal_duration"],
)
pconfig = config["processing"]
# Run get_corner_frequencies
test = [d for d in pconfig if list(d.keys())[0] == "get_corner_frequencies"]
cf_config = test[0]["get_corner_frequencies"]
mag_config = cf_config["magnitude"]
lp = []
hp = []
for stream in processed_streams:
if not stream.passed:
continue
stream = get_corner_frequencies(
stream, origin, method="magnitude", magnitude=mag_config
)
if stream[0].hasParameter("corner_frequencies"):
cfdict = stream[0].getParameter("corner_frequencies")
lp.append(cfdict["lowpass"])
hp.append(cfdict["highpass"])