228 lines
7.3 KiB
Python
228 lines
7.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
Created on Tue May 28 15:40:48 2024
|
|||
|
|
|
|||
|
|
@author: WANGXIBAO
|
|||
|
|
"""
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
import csv
|
|||
|
|
import math
|
|||
|
|
import os
|
|||
|
|
import re
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
from PyQt5.QtChart import QChart, QValueAxis, QLineSeries
|
|||
|
|
from PyQt5.QtCore import QThread, pyqtSignal, Qt
|
|||
|
|
from PyQt5.QtGui import QPen
|
|||
|
|
from PyQt5.QtWidgets import QMessageBox
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 波形显示
|
|||
|
|
class QChartViewPlot(QChart):
|
|||
|
|
|
|||
|
|
# 相位振动波形
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super(QChartViewPlot, self).__init__(parent)
|
|||
|
|
self.window = parent
|
|||
|
|
self.xRange = 60
|
|||
|
|
self.counter = 0
|
|||
|
|
self.seriesList = []
|
|||
|
|
self.legend().show()
|
|||
|
|
|
|||
|
|
self.axisX = QValueAxis()
|
|||
|
|
self.axisX.setRange(0, self.xRange)
|
|||
|
|
self.addAxis(self.axisX, Qt.AlignBottom)
|
|||
|
|
# self.setAxisX(self.axisX, series)
|
|||
|
|
# 增加X轴网格设置
|
|||
|
|
self.axisX.setTickCount(13)
|
|||
|
|
grid_pen_x = QPen(Qt.black, 0.5, Qt.DotLine) # 设置网格线的颜色、宽度和样式
|
|||
|
|
self.axisX.setGridLinePen(grid_pen_x)
|
|||
|
|
|
|||
|
|
self.y_min = 0
|
|||
|
|
self.y_max = 100
|
|||
|
|
self.axisY = QValueAxis()
|
|||
|
|
self.axisY.setRange(self.y_min, self.y_max)
|
|||
|
|
self.addAxis(self.axisY, Qt.AlignLeft)
|
|||
|
|
# self.setAxisY(self.axisY, series)
|
|||
|
|
# 增加Y轴网格设置
|
|||
|
|
self.axisY.setTickCount(11)
|
|||
|
|
grid_pen_y = QPen(Qt.black, 0.5, Qt.DotLine) # 设置网格线的颜色、宽度和样式
|
|||
|
|
self.axisY.setGridLinePen(grid_pen_y)
|
|||
|
|
|
|||
|
|
#self.series = QSplineSeries()
|
|||
|
|
self.series = QLineSeries()
|
|||
|
|
self.series.setName("波形")
|
|||
|
|
self.series.setUseOpenGL(True)
|
|||
|
|
self.addSeries(self.series)
|
|||
|
|
|
|||
|
|
|
|||
|
|
self.series.attachAxis(self.axisX)
|
|||
|
|
self.series.attachAxis(self.axisY)
|
|||
|
|
|
|||
|
|
self.curretValue = 0
|
|||
|
|
self.windowAverage =0
|
|||
|
|
self.windowStd = 0
|
|||
|
|
|
|||
|
|
def handle_update(self, ydata):
|
|||
|
|
# 更新y值
|
|||
|
|
if self.counter < self.xRange:
|
|||
|
|
self.series.append(self.counter, ydata)
|
|||
|
|
self.counter += 1
|
|||
|
|
points = self.series.pointsVector()
|
|||
|
|
self.y_min = min(points, key=lambda point: point.y()).y()
|
|||
|
|
self.y_max = max(points, key=lambda point: point.y()).y()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
points = self.series.pointsVector()
|
|||
|
|
for i in range(self.xRange - 1):
|
|||
|
|
points[i].setY(points[i + 1].y())
|
|||
|
|
points[-1].setY(ydata)
|
|||
|
|
self.y_min = min(points, key=lambda point: point.y()).y()
|
|||
|
|
self.y_max = max(points, key=lambda point: point.y()).y()
|
|||
|
|
self.series.replace(points)
|
|||
|
|
|
|||
|
|
# 计算总和
|
|||
|
|
total_sum = sum(point.y() for point in points)
|
|||
|
|
# 计算点的数量
|
|||
|
|
num_points = len(points)
|
|||
|
|
# 计算均值，确保除数不为零
|
|||
|
|
self.windowAverage = total_sum / num_points if num_points > 0 else 0
|
|||
|
|
print("windowMean:",self.windowAverage)
|
|||
|
|
|
|||
|
|
# 计算每个点与均值的差的平方，然后求和
|
|||
|
|
variance_sum = sum((point.y() - self.windowAverage) ** 2 for point in points)
|
|||
|
|
# 计算方差，除以数据点的总数
|
|||
|
|
variance = variance_sum / len(points)
|
|||
|
|
self.axisY.setRange(math.floor(self.y_min-0.1),math.ceil( self.y_max+0.1))
|
|||
|
|
# 计算标准差，即方差的平方根
|
|||
|
|
self.windowStd = variance ** 0.5
|
|||
|
|
print("standard_deviation", self.windowStd)
|
|||
|
|
|
|||
|
|
def clearSeries(self):
|
|||
|
|
self.series.clear()
|
|||
|
|
self.counter = 0
|
|||
|
|
# 使用线程不断更新波形数据
|
|||
|
|
|
|||
|
|
class UpdateDataThread(QThread):
|
|||
|
|
_signal_update = pyqtSignal(float) # 信号
|
|||
|
|
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super(UpdateDataThread, self).__init__(parent)
|
|||
|
|
self.is_exit = False
|
|||
|
|
self.x_range = 1024
|
|||
|
|
self.flag = 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
print("run start")
|
|||
|
|
while not self.is_exit: # 直接在循环中检查退出标志
|
|||
|
|
# for i in range(self.x_range):
|
|||
|
|
# print(i,self.sin.get_data(i))
|
|||
|
|
|
|||
|
|
# self._signal_update.emit(str(self.sin.get_data(i))) # 发射信号
|
|||
|
|
# time.sleep(0.01) # 实际应用中推荐使用QTimer代替
|
|||
|
|
|
|||
|
|
# 这里应该有一个 flag 默认是 0 外部调用可以传紧来1，没循环完自己置零，while 只起到监控左右
|
|||
|
|
|
|||
|
|
if self.flag == 1:
|
|||
|
|
#print("flag",self.flag)
|
|||
|
|
|
|||
|
|
#data = self.setReceiveData
|
|||
|
|
#print("发数",datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f"))
|
|||
|
|
print("data receive ",self.setReceiveData)
|
|||
|
|
# print(f"{data:.3f}")
|
|||
|
|
try:
|
|||
|
|
|
|||
|
|
self._signal_update.emit(self.setReceiveData)
|
|||
|
|
self.flag = 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
except Exception:
|
|||
|
|
QMessageBox.critical(self, '数据异常', '数据格式不对，不能绘图')
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def SetReceiveData(self,data_in):
|
|||
|
|
self.setReceiveData = data_in
|
|||
|
|
|
|||
|
|
def SetFlag(self,flag):
|
|||
|
|
self.flag = flag
|
|||
|
|
|
|||
|
|
def SetFilenameCsv(self,filename):
|
|||
|
|
self.filenameCsv = filename
|
|||
|
|
|
|||
|
|
def SetPlotItem(self,ind): #设置显示项目
|
|||
|
|
self.ind = ind
|
|||
|
|
|
|||
|
|
def stop(self):
|
|||
|
|
self.is_exit = True # 设置退出标志
|
|||
|
|
print("停止绘图")
|
|||
|
|
def restart(self):
|
|||
|
|
self.is_exit = False
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class GetDataTF(): #读取费加罗或者自定义数据
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
super().__init__()
|
|||
|
|
self.indOfReturn =0
|
|||
|
|
self.TIME_FORMAT = "%Y-%m-%d %H:%M:%S"
|
|||
|
|
self.headStr = ""
|
|||
|
|
self.rowTitle=[]
|
|||
|
|
self.csv_buffer = []
|
|||
|
|
|
|||
|
|
def SetConfig(self,config_in):
|
|||
|
|
self.regex= re.compile(config_in[0]) #正则表达
|
|||
|
|
self.headStr = config_in[1]
|
|||
|
|
self.rowTitleStr = config_in[2]
|
|||
|
|
#print(self.rowTitleStr)
|
|||
|
|
self.rowTitle = self.rowTitleStr.split(',')
|
|||
|
|
|
|||
|
|
|
|||
|
|
def Transdata(self,data):
|
|||
|
|
if data[:2] ==self.headStr:
|
|||
|
|
|
|||
|
|
self.data2csv = self.regex.findall(data)
|
|||
|
|
|
|||
|
|
returnData = float(self.data2csv[self.indOfReturn])
|
|||
|
|
print("returnData",returnData)
|
|||
|
|
return returnData
|
|||
|
|
else:
|
|||
|
|
return "noNum"
|
|||
|
|
|
|||
|
|
def SaveCsv(self,filenameCsv,num):
|
|||
|
|
# 打开一个文件用于写入，如果文件不存在则创建
|
|||
|
|
# 增加num 写入文件间隔
|
|||
|
|
if os.path.isfile(filenameCsv) == 0:
|
|||
|
|
# 文件为空，需要写入表头
|
|||
|
|
with open(filenameCsv, mode='w', newline='') as file:
|
|||
|
|
writer = csv.writer(file)
|
|||
|
|
writer.writerow(self.rowTitle)
|
|||
|
|
else:
|
|||
|
|
timeCrvt=time.strftime(self.TIME_FORMAT, time.localtime())
|
|||
|
|
self.data2csv.insert(0, timeCrvt)
|
|||
|
|
print(self.data2csv)
|
|||
|
|
self.csv_buffer.append(self.data2csv)
|
|||
|
|
if len(self.csv_buffer)>= num:
|
|||
|
|
with open(filenameCsv, mode='a', newline='') as file:
|
|||
|
|
# 创建一个写入器对象
|
|||
|
|
writer = csv.writer(file)
|
|||
|
|
# 写入数据行
|
|||
|
|
writer.writerows(self.csv_buffer)
|
|||
|
|
self.csv_buffer.clear()
|
|||
|
|
|
|||
|
|
def IndOfReturn(self,ind):
|
|||
|
|
self.indOfReturn = ind
|
|||
|
|
|
|||
|
|
|