pylot/map_projection.py

57 lines
1.2 KiB
Python
Raw Normal View History

from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
import numpy as np
import obspy
#import QtPyLoT
from pylot.core.util.dataprocessing import read_metadata
#pf=QtPyLoT.main()
def onpick(event):
ind = event.ind
print(ind)
metadata=read_metadata('/data/Geothermie/Insheim/STAT_INFO/MAGS2_net.dless')
parser=metadata[1]
lat=[]
lon=[]
stcl=[]
picks=[]
picks_rel=[]
for station in parser.stations:
stcl.append(station[0].station_call_letters)
lat.append(station[0].latitude)
lon.append(station[0].longitude)
for station in stcl:
try:
picks.append(pf.autopicks[station]['P']['mpp'])
except:
picks.append(np.nan)
minp = min(picks)
for pick in picks:
if type(pick) is obspy.core.utcdatetime.UTCDateTime:
pick -= minp
picks_rel.append(pick)
m = Basemap(projection='hammer',lon_0=0,lat_0=0)
m.drawmapboundary(fill_color='aqua')
m.drawcountries()
m.drawstates()
m.fillcontinents(color='grey', lake_color='aqua')
m.drawcoastlines()
m.scatter(lon, lat, s=50, c=picks_rel, latlon=True, zorder=10, picker=True)
ax = plt.gca() # IMPROVE!!!!
ax.figure.canvas.mpl_connect('pick_event', onpick)
plt.show()