V0.1

safe/punch/colorbar.py

@@ -0,0 +1,38 @@
+from PySide2.QtCore import *
+from PySide2.QtGui import *
+from PySide2.QtWidgets import *
+from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+import matplotlib.pyplot as plt
+import matplotlib as mpl
+import numpy as np
+
+
+class ColorMap(QWidget):
+ def __init__(self, minv, maxv, parent=None):
+ self.max = maxv
+ self.min = minv
+ super(ColorMap, self).__init__(parent)
+ self.figure = plt.figure()
+ self.canvas = FigureCanvas(self.figure)
+
+ self.draw_colormap()
+
+ # set the layout
+ layout = QVBoxLayout()
+ layout.addWidget(self.canvas)
+ self.setLayout(layout)
+
+ def draw_colormap(self):
+ # axes
+ ax = self.figure.add_axes([0.05, 0.10, 0.2, 0.8])
+ cmap = mpl.cm.jet
+ norm = mpl.colors.Normalize(vmin=self.min, vmax=self.max)
+ ticks_cm = np.linspace(self.min, self.max, 10, endpoint=True)
+ cb1 = mpl.colorbar.ColorbarBase(ax, cmap=cmap,
+ norm=norm,
+ ticks=ticks_cm,
+ orientation='vertical')
+ # label_cm = 'punch ratio'
+ # cb1.set_label(label=label_cm,weight='bold')
+ cb1.ax.tick_params(labelsize=8)
+ self.canvas.draw()