89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
"""
|
|
Created on 2019年5月21日
|
|
@author: weike32
|
|
@site: https://pyqt.site ,https://github.com/weike32
|
|
@email: 394967319@qq.com
|
|
@file: CopyContent
|
|
@description: 工具类
|
|
"""
|
|
import pyqtgraph as pg
|
|
from pyqtgraph.exporters.ImageExporter import ImageExporter, Exporter
|
|
from pyqtgraph.parametertree import Parameter
|
|
|
|
|
|
# 不用修改源码，重加载，解决右键保存图片异常
|
|
def widthChanged(self):
|
|
sr = self.getSourceRect()
|
|
ar = float(sr.height()) / sr.width()
|
|
self.params.param('height').setValue(int(self.params['width'] * ar), blockSignal=self.heightChanged)
|
|
|
|
|
|
def heightChanged(self):
|
|
sr = self.getSourceRect()
|
|
ar = float(sr.width()) / sr.height()
|
|
self.params.param('width').setValue(int(self.params['height'] * ar), blockSignal=self.widthChanged)
|
|
|
|
|
|
def New__init__(self, item):
|
|
Exporter.__init__(self, item)
|
|
tr = self.getTargetRect()
|
|
if isinstance(item, pg.Qt.QtGui.QGraphicsItem):
|
|
scene = item.scene()
|
|
else:
|
|
scene = item
|
|
bgbrush = scene.views()[0].backgroundBrush()
|
|
bg = bgbrush.color()
|
|
if bgbrush.style() == pg.Qt.QtCore.Qt.NoBrush:
|
|
bg.setAlpha(0)
|
|
|
|
self.params = Parameter(name='params', type='group', children=[
|
|
{'name': 'width', 'type': 'int', 'value': int(tr.width()), 'limits': (0, None)},
|
|
{'name': 'height', 'type': 'int', 'value': int(tr.height()), 'limits': (0, None)},
|
|
{'name': 'antialias', 'type': 'bool', 'value': True},
|
|
{'name': 'background', 'type': 'color', 'value': bg},
|
|
])
|
|
self.params.param('width').sigValueChanged.connect(self.widthChanged)
|
|
self.params.param('height').sigValueChanged.connect(self.heightChanged)
|
|
|
|
|
|
ImageExporter.heightChanged = heightChanged
|
|
ImageExporter.widthChanged = widthChanged
|
|
ImageExporter.__init__ = New__init__
|
|
|
|
|
|
# 解决自定义坐标轴密集显示
|
|
class MyStringAxis(pg.AxisItem):
|
|
def __init__(self, xdict, *args, **kwargs):
|
|
pg.AxisItem.__init__(self, *args, **kwargs)
|
|
self.xdict = xdict
|
|
|
|
def tickStrings(self, values, scale, spacing):
|
|
strings = []
|
|
for v in values:
|
|
vs = v * scale
|
|
if vs in self.xdict.keys():
|
|
vstr = self.xdict[vs]
|
|
else:
|
|
vstr = ""
|
|
strings.append(vstr)
|
|
return strings
|
|
|
|
|
|
# 禁止鼠标事件
|
|
class CustomViewBox(pg.ViewBox):
|
|
def __init__(self, *args, **kwds):
|
|
pg.ViewBox.__init__(self, *args, **kwds)
|
|
self.RectMode = 3
|
|
self.setMouseMode(self.RectMode)
|
|
|
|
def mouseClickEvent(self, ev):
|
|
if ev.button() == pg.QtCore.Qt.RightButton:
|
|
self.autoRange()
|
|
|
|
def mouseDragEvent(self, ev):
|
|
pg.ViewBox.mouseDragEvent(self, ev)
|
|
|
|
def wheelEvent(self, ev, axis=None):
|
|
pg.ViewBox.wheelEvent(self, ev, axis)
|