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

feature/port-to-py3 #11

Merged
marcel merged 59 commits from feature/port-to-py3 into develop 2022-03-21 15:30:06 +01:00
Conversation 0 Commits 59 Files Changed 45 +28 -17
1 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 2ccdfb8470 - Show all commits

45
pylot/core/util/array_map.py
View File

@ -2,6 +2,10 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os import os
import matplotlib
matplotlib.use('Qt5Agg')
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
import obspy import obspy
@ -11,6 +15,7 @@ from matplotlib.figure import Figure
from mpl_toolkits.axes_grid1.inset_locator import inset_axes from mpl_toolkits.axes_grid1.inset_locator import inset_axes
# from mpl_toolkits.basemap import Basemap # from mpl_toolkits.basemap import Basemap
from scipy.interpolate import griddata from scipy.interpolate import griddata
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
import cartopy.crs as ccrs import cartopy.crs as ccrs
import cartopy.feature as cf import cartopy.feature as cf
@ -71,7 +76,7 @@ class Array_map(QtWidgets.QWidget):
self.init_colormap() self.init_colormap()
self.connectSignals() self.connectSignals()
self.draw_everything() self.draw_everything()
self.canvas.setZoomBorders2content() # self.canvas.setZoomBorders2content()
def init_colormap(self): def init_colormap(self):
self.init_lat_lon_dimensions() self.init_lat_lon_dimensions()
@ -194,7 +199,7 @@ class Array_map(QtWidgets.QWidget):
self.annotations_box.stateChanged.connect(self.switch_annotations) self.annotations_box.stateChanged.connect(self.switch_annotations)
self.refresh_button.clicked.connect(self._refresh_drawings) self.refresh_button.clicked.connect(self._refresh_drawings)
self.canvas.mpl_connect('motion_notify_event', self.mouse_moved) self.canvas.mpl_connect('motion_notify_event', self.mouse_moved)
# self.zoom_id = self.basemap.ax.figure.canvas.mpl_connect('scroll_event', self.zoom) self.canvas.mpl_connect('scroll_event', self.zoom)
def _from_dict(self, function, key): def _from_dict(self, function, key):
return function(self.stations_dict.values(), key=lambda x: x[key])[key] return function(self.stations_dict.values(), key=lambda x: x[key])[key]
@ -230,14 +235,15 @@ class Array_map(QtWidgets.QWidget):
def init_graphics(self): def init_graphics(self):
if not self.figure: if not self.figure:
self.figure = Figure() self.figure = plt.figure()
self.status_label = QtWidgets.QLabel() self.status_label = QtWidgets.QLabel()
self.main_ax = self.figure.add_subplot(111) self.main_ax = self.figure.add_subplot(111)
#self.main_ax.set_facecolor('0.7') #self.main_ax.set_facecolor('0.7')
self.canvas = PylotCanvas(self.figure, parent=self._parent, multicursor=True, self.canvas = FigureCanvas(self.figure)
panZoomX=False, panZoomY=False)
# parent=self._parent, multicursor=True, panZoomX=False, panZoomY=False)
self.main_box = QtWidgets.QVBoxLayout() self.main_box = QtWidgets.QVBoxLayout()
self.setLayout(self.main_box) self.setLayout(self.main_box)
@ -356,6 +362,7 @@ class Array_map(QtWidgets.QWidget):
self.basemap = self.main_ax self.basemap = self.main_ax
# plt.show() # plt.show()
# self.show()
# self.figure._tight = True # self.figure._tight = True
# self.figure.tight_layout() # self.figure.tight_layout()
@ -567,17 +574,21 @@ class Array_map(QtWidgets.QWidget):
self.annotations = [] self.annotations = []
def zoom(self, event): def zoom(self, event):
map = self.basemap if not event.inaxes == self.canvas.axes:
xlim = map.get_xlim() return
ylim = map.get_ylim()
x, y = event.xdata, event.ydata
zoom = {'up': 1. / 2., zoom = {'up': 1. / 2.,
'down': 2.} 'down': 2.}
if not event.xdata or not event.ydata: # if not event.xdata or not event.ydata:
return # return
if event.button in zoom: if event.button in zoom:
m = self.basemap
xlim = m.get_xlim()
ylim = m.get_ylim()
x, y = event.xdata, event.ydata
factor = zoom[event.button] factor = zoom[event.button]
xdiff = (xlim[1] - xlim[0]) * factor xdiff = (xlim[1] - xlim[0]) * factor
xl = x - 0.5 * xdiff xl = x - 0.5 * xdiff
@ -586,12 +597,12 @@ class Array_map(QtWidgets.QWidget):
yb = y - 0.5 * ydiff yb = y - 0.5 * ydiff
yt = y + 0.5 * ydiff yt = y + 0.5 * ydiff
if xl < map.xmin or yb < map.ymin or xr > map.xmax or yt > map.ymax: #if xl < map.xmin or yb < map.ymin or xr > map.xmax or yt > map.ymax:
xl, xr = map.xmin, map.xmax # xl, xr = map.xmin, map.xmax
yb, yt = map.ymin, map.ymax # yb, yt = map.ymin, map.ymax
map.set_xlim(xl, xr) m.set_xlim(xl, xr)
map.set_ylim(yb, yt) m.set_ylim(yb, yt)
map.figure.canvas.draw() m.figure.canvas.draw_idle()
def _warn(self, message): def _warn(self, message):
self.qmb = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Icon.Warning, self.qmb = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Icon.Warning,