def catalog_no_magnitude(self):
""" get a events with no magnitudes (should just fill with NaN) """
t1 = obspy.UTCDateTime("2099-04-01T00-01-00")
ori = ev.Origin(time=t1, latitude=47.1, longitude=-100.22)
event = ev.Event(origins=[ori])
cat = ev.Catalog(events=[event])
return events_to_df(cat)
def css_to_catalog(method: Union[dict, str, Path], **kwargs) -> obspy.Catalog:
"""
Convert a CSS 3.0 directory or dict of dataframes to a catalog.
Parameters
----------
method
A path to a css 3.0 directory or a dict of dataframes create with
dir_to_df.
Returns
-------
obspy.Catalog
"""
if isinstance(method, dict):
df_dict = method
else:
df_dict = dir_to_df(method)
if kwargs:
df_dict = filter_df_dict(df_dict, **kwargs)
# get events
events = df_dict['event'].apply(_create_event, axis=1)
df_dict['origin'].apply(_create_origin, axis=1)
df_dict['origerr'].apply(_create_origin_quality, axis=1)
_ = df_dict['arrival'].apply(_create_pick, axis=1)
df_dict['assoc'].apply(_create_arrival, axis=1)
df_dict['netmag'].apply(_create_magnitude, axis=1)
# df_dict['stamag'].apply(_create_stationmag, axis=1)
# attach pick to catalog
for event in events:
attach_picks(event)
return oe.Catalog(events=list(events.values))
def test_empty_catalog(self):
""" ensure returns empty df with required columns """
empty_cat = ev.Catalog()
df = station_magnitudes_to_dataframe(empty_cat)
assert isinstance(df, pd.DataFrame)
assert not len(df)
assert set(df.columns).issubset(STATION_MAGNITUDE_COLUMNS)
def test_empty_catalog(self):
""" ensure returns empty df with required columns """
empty_cat = ev.Catalog()
df = arrivals_to_dataframe(empty_cat)
assert isinstance(df, pd.DataFrame)
assert not len(df)
assert set(df.columns).issubset(ARRIVAL_COLUMNS)
def dummy_cat(self):
cat = ev.Catalog()
eve = ev.Event()
eve.origins.append(ev.Origin(time=UTCDateTime()))
eve.magnitudes = mag_generator(["ML", "Md", "MW"])
cat.append(eve)
return cat
def _saveCatalog(self, filename=None):
'''
Saves the catalog to filename
'''
if self.savefile is None and filename is None:
return self._saveCatalogDlg()
if filename is not None:
savefile = filename
else:
savefile = self.savefile
cat = event.Catalog()
cat.events.append(event.Event(picks=self._picks))
def dummy_cat(self):
scnls1 = ["UK.STA1..HHZ", "UK.STA2..HHZ"]
cat = ev.Catalog()
eve1 = ev.Event()
eve1.origins.append(ev.Origin(time=UTCDateTime()))
eve1.amplitudes = make_amplitudes(scnls1)
eve1.station_magnitudes = sm_generator(amplitudes=eve1.amplitudes)
scnls2 = ["UK.STA3..HHZ", "UK.STA4..HHZ", "UK.STA5..HHZ"]
eve2 = ev.Event()
eve2.origins.append(ev.Origin(time=UTCDateTime()))
eve2.station_magnitudes = sm_generator(scnls=scnls2)
cat.events = [eve1, eve2]
return cat
def dummy_cat(self):
scnls1 = ["UK.STA1..HHZ", "UK.STA2..HHZ"]
cat = ev.Catalog()
eve1 = ev.Event()
eve1.origins.append(ev.Origin(time=UTCDateTime()))
eve1.picks = pick_generator(scnls1)
eve1.amplitudes = amp_generator(picks=eve1.picks)
scnls2 = ["UK.STA3..HHZ", "UK.STA4..HHZ", "UK.STA5..HHZ"]
eve2 = ev.Event()
eve2.origins.append(ev.Origin(time=UTCDateTime()))
eve2.amplitudes = amp_generator(scnls=scnls2)
cat.events = [eve1, eve2]
return cat
def _saveCatalog(self, filename=None):
'''
Saves the catalog to filename
'''
if self.savefile is None and filename is None:
return self._saveCatalogDlg()
if filename is not None:
savefile = filename
else:
savefile = self.savefile
cat = event.Catalog()
cat.events.append(event.Event(picks=self._picks))
cat.write(savefile, format='QUAKEML')
print 'Picks saved as %s' % savefile
def dummy_cat(self):
scnls1 = ["UK.STA1..HHZ", "UK.STA2..HHZ"]
cat = ev.Catalog()
eve1 = ev.Event()
eve1.origins.append(ev.Origin(time=UTCDateTime()))
eve1.preferred_origin_id = eve1.origins[0].resource_id
picks = pick_generator(scnls1)
eve1.picks = picks
eve1.preferred_origin().arrivals = make_arrivals(picks)
scnls2 = ["UK.STA3..HHZ", "UK.STA4..HHZ", "UK.STA5..HHZ"]
eve2 = ev.Event()
eve2.origins.append(ev.Origin(time=UTCDateTime()))
eve2.preferred_origin_id = eve2.origins[0].resource_id
picks = pick_generator(scnls2)
eve2.picks = picks
eve2.preferred_origin().arrivals = make_arrivals(picks)
cat.events = [eve1, eve2]
return cat
def test_empty_catalog(self):
""" ensure returns empty df with required columns """
df = amplitudes_to_dataframe(ev.Catalog())
assert isinstance(df, pd.DataFrame)
assert not len(df)
assert set(df.columns).issubset(AMPLITUDE_COLUMNS)
def no_origin(self):
"""Return a catalog with no origin."""
cat = ev.Catalog()
cat.append(ev.Event())
return cat
def catalog_empty(self):
""" get a with one blank event """
event = ev.Event()
cat = ev.Catalog(events=[event])
return events_to_df(cat)
def empty_cat(self):
return ev.Catalog()
def no_origin(self):
cat = ev.Catalog()
cat.append(ev.Event())
return cat
def empty_cat(self):
"""Return an empty catalog"""
return ev.Catalog()
