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()