Initial Commit

figure_canvas.py

@@ -0,0 +1,95 @@
+from __future__ import print_function
+from future.builtins import super
+
+import os
+import sys
+import numpy
+import matplotlib
+from matplotlib.figure import Figure
+from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg
+from PyQt5.QtGui import QIcon
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import QSizePolicy, QMainWindow, QApplication, QAction
+from PyQt5.QtWidgets import QMenu, QWidget, QVBoxLayout, QMessageBox
+from PyQt5.QtWidgets import QSlider, QHBoxLayout, QLabel, QDialog
+from PyQt5.QtWidgets import QDialogButtonBox
+
+
+class GravityModelCanvas(FigureCanvasQTAgg):
+
+    def __init__(self, parent=None, width=5, height=4, dpi=100):
+        self.fig = Figure(figsize=(width, height), dpi=dpi)
+        super().__init__(self.fig)
+        self.setParent(parent)
+
+        self._x, self._z = None, None
+        self._min_depth, self._max_depth = 0., 35000.
+
+        self.predicted = None
+        self.data = None
+        self.polygons = None
+
+        self._plotted = False
+
+        FigureCanvasQTAgg.setSizePolicy(self,
+                                        QSizePolicy.Expanding,
+                                        QSizePolicy.Expanding)
+        FigureCanvasQTAgg.updateGeometry(self)
+
+    @property
+    def x(self):
+        return self._x
+
+    @x.setter
+    def x(self, new_value):
+        self._x = new_value
+
+    @property
+    def z(self):
+        return self._z
+
+    @z.setter
+    def z(self, new_value):
+        self._z = new_value
+
+    @property
+    def min_depth(self):
+        return self._min_depth
+
+    @min_depth.setter
+    def min_depth(self, new_value):
+        self._min_depth = new_value
+
+    @property
+    def max_depth(self):
+        return self._max_depth
+
+    @max_depth.setter
+    def max_depth(self, new_value):
+        self._max_depth = new_value
+
+    def update_plot(self):
+        if self._plotted:
+            pass
+        else:
+            self._figure_setup()
+
+    def _figure_setup(self, **kwargs):
+        self.dataax, self.modelax = self.fig.subplots(2, 1, sharex=True)
+        if self.data is not None:
+            self.data_line, = self.dataax.plot(self.x, self.data, '.k')
+        self.dataax.set_ylabel('Gravity anomaly [mGal]')
+        self.dataax.set_xlim(self.x.min(), self.x.max())
+        self.dataax.set_ylim((-200, 200))
+        self.dataax.grid(True)
+        self.modelax.set_xlabel('x [m]')
+        self.modelax.set_xlim(self.x.min(), self.x.max())
+        self.modelax.set_ylim(self.min_depth, self.max_depth)
+        self.modelax.grid(True)
+        self.modelax.invert_yaxis()
+        self.modelax.set_ylabel('z [m]')
+        self.fig.subplots_adjust(top=0.95, left=0.1, right=0.95, bottom=0.1,
+                                 hspace=0.1)
+        self.figure = self.fig
+        self.canvas = self.fig.canvas
+        self.fig.canvas.draw()