marcel/pylot
3
0
Fork 0
Code Issues 17 Pull Requests 1 Releases Wiki Activity

small bugfixes and widget added

Browse Source
This commit is contained in:
Marcel Paffrath 2017-04-11 15:16:53 +02:00
parent bf63348151
commit 2e6d39c52e
2 changed files with 41 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
map_projection.py
View File

@ -3,36 +3,36 @@ import matplotlib.pyplot as plt
import numpy as np import numpy as np
import obspy import obspy
from matplotlib import cm from matplotlib import cm
from PySide import QtCore, QtGui
from matplotlib.backends.backend_qt4agg import NavigationToolbar2QT as NavigationToolbar
#import QtPyLoT #import QtPyLoT
from pylot.core.util.dataprocessing import read_metadata from pylot.core.util.dataprocessing import read_metadata
from scipy.interpolate import griddata from scipy.interpolate import griddata
#pf=QtPyLoT.main() #pf=QtPyLoT.main()
def onpick(event): def onpick(event):
ind = event.ind ind = event.ind
print(ind) for index in ind:
print(station_names[index])
def get_metadata(path): def get_metadata(path):
metadata=read_metadata(path) metadata=read_metadata(path)
parser=metadata[1] parser=metadata[1]
return parser return parser
def get_station_names(parser): def get_station_names_lat_lon(parser):
station_names=[] station_names=[]
for station in parser.stations:
station_names.append(station[0].station_call_letters)
return station_names
def get_lat_lon(parser):
lat=[] lat=[]
lon=[] lon=[]
for station in parser.stations: for station in parser.stations:
station_name=station[0].station_call_letters
if not station_name in station_names:
station_names.append(station_name)
lat.append(station[0].latitude) lat.append(station[0].latitude)
lon.append(station[0].longitude) lon.append(station[0].longitude)
return lat, lon return station_names, lat, lon
def get_picks(pf, station_names): def get_picks(pf, station_names):
picks=[] picks=[]
@ -78,8 +78,8 @@ def get_x_y_dim(x, y):
ydim = max(y) - min(y) ydim = max(y) - min(y)
return xdim, ydim return xdim, ydim
def init_map(projection, resolution='l'): def init_map(projection, ax, resolution='l'):
m = Basemap(projection=projection, resolution = resolution) m = Basemap(projection=projection, resolution = resolution, ax=ax)
m.drawmapboundary(fill_color='darkblue') m.drawmapboundary(fill_color='darkblue')
m.drawcountries() m.drawcountries()
m.drawstates() m.drawstates()
@ -116,11 +116,24 @@ def add_cbar(ax, scatter, label):
cbar.set_label(label) cbar.set_label(label)
return cbar return cbar
qwidget = QtGui.QWidget()
QtGui.QVBoxLayout()
main_box = QtGui.QVBoxLayout()
qwidget.setLayout(main_box)
fig = plt.figure()
main_ax = fig.add_subplot(111)
canvas = main_ax.figure.canvas
main_box.addWidget(canvas)
toolbar = NavigationToolbar(canvas, qwidget)
main_box.addWidget(toolbar)
parser = get_metadata('/data/Geothermie/Insheim/STAT_INFO/MAGS2_net.dless') parser = get_metadata('/data/Geothermie/Insheim/STAT_INFO/MAGS2_net.dless')
station_names = get_station_names(parser) station_names, lat, lon = get_station_names_lat_lon(parser)
lat, lon = get_lat_lon(parser)
picks = get_picks(pf, station_names) picks = get_picks(pf, station_names)
picks_rel = get_picks_rel(picks) picks_rel = get_picks_rel(picks)
@ -131,7 +144,7 @@ londim, latdim = get_lon_lat_dim(lon, lat)
x, y = m(lon, lat) x, y = m(lon, lat)
xdim, ydim = get_x_y_dim(x, y) xdim, ydim = get_x_y_dim(x, y)
m = init_map('mill', 'l') m = init_map(projection='mill', ax=main_ax, resolution='l')
lataxis, lonaxis = get_lat_lon_axis(lat, lon) lataxis, lonaxis = get_lat_lon_axis(lat, lon)
latgrid, longrid = get_lat_lon_grid(lataxis, lonaxis) latgrid, longrid = get_lat_lon_grid(lataxis, lonaxis)
@ -143,17 +156,21 @@ contourf = draw_contour_filled(picks_no_nan, longrid, latgrid, picksgrid_no_nan)
sc = m.scatter(lon, lat, s=50, facecolor='none', latlon=True, zorder=10, picker=True, edgecolor='m', label='Not Picked') sc = m.scatter(lon, lat, s=50, facecolor='none', latlon=True, zorder=10, picker=True, edgecolor='m', label='Not Picked')
sc_picked = m.scatter(lon_no_nan, lat_no_nan, s=50, c=picks_no_nan, latlon=True, zorder=11, label='Picked') sc_picked = m.scatter(lon_no_nan, lat_no_nan, s=50, c=picks_no_nan, latlon=True, zorder=11, label='Picked')
ax = plt.gca() # IMPROVE!!!! annotate_ax(main_ax, x, y, station_names)
annotate_ax(ax, x, y, station_names) main_ax.legend()
ax.legend() connect_pick(main_ax, onpick)
connect_pick(ax, onpick) cbar = add_cbar(main_ax, sc_picked, label='Time relative to first onset [s]')
cbar = add_cbar(ax, sc_picked, label='Time relative to first onset [s]') canvas.draw()
qwidget.show()
# ax.set_xlim(min(x)-0.5*xdim, max(x)+0.5*xdim) # ax.set_xlim(min(x)-0.5*xdim, max(x)+0.5*xdim)
# ax.set_ylim(min(y)-0.5*ydim, max(y)+0.5*ydim) # ax.set_ylim(min(y)-0.5*ydim, max(y)+0.5*ydim)
plt.show()
#plt.show()

2
pylot/RELEASE-VERSION
View File

@ -1 +1 @@
f1e3-dirty bf63-dirty