259 lines
9.2 KiB
Python
259 lines
9.2 KiB
Python
|
import datetime, time, sys
|
|||
|
|
import numpy as np
|
|||
|
|
import pyqtgraph as pg
|
|||
|
|
from pyqtgraph.Qt import QtWidgets
|
|||
|
|
|
|||
|
|
def csv_to_nparray(fileName, ContainDateTime, timeRange):
|
|||
|
|
if ContainDateTime == 'True':
|
|||
|
|
ContainDateTime = True
|
|||
|
|
else:
|
|||
|
|
ContainDateTime = False
|
|||
|
|
Filter = 0
|
|||
|
|
itemMaxNum = 50
|
|||
|
|
if ContainDateTime:
|
|||
|
|
timeRange = timeRange.split(',')
|
|||
|
|
if len(timeRange) == 2:
|
|||
|
|
Filter = 1
|
|||
|
|
startTime = datetime.datetime.strptime(timeRange[0].strip(),'%Y/%m/%d %H:%M:%S')
|
|||
|
|
start_timestamp=time.mktime(startTime.timetuple())
|
|||
|
|
stopTime = datetime.datetime.strptime(timeRange[1].strip(),'%Y/%m/%d %H:%M:%S')
|
|||
|
|
stop_timestamp=time.mktime(stopTime.timetuple())
|
|||
|
|
else:
|
|||
|
|
Filter = 0
|
|||
|
|
|
|||
|
|
#读取文件
|
|||
|
|
with open(fileName, "r") as f:
|
|||
|
|
recordTotalNum = len(f.readlines())
|
|||
|
|
f.seek(0)
|
|||
|
|
|
|||
|
|
#读取数据文件中的标题行
|
|||
|
|
itemNames = []
|
|||
|
|
itemNames.append('SampleNO')
|
|||
|
|
if ContainDateTime == False:
|
|||
|
|
itemNames.append('DateTime')
|
|||
|
|
|
|||
|
|
tempStr = f.readline()
|
|||
|
|
tempStr = tempStr.replace(' ','')
|
|||
|
|
tempStr = tempStr.replace('"','')
|
|||
|
|
tempStr = tempStr.replace('\t','')
|
|||
|
|
tempStr = tempStr.replace('\n','')
|
|||
|
|
temp = tempStr.split(',')
|
|||
|
|
for i in range(len(temp)):
|
|||
|
|
temp[i] = temp[i].strip()
|
|||
|
|
if ContainDateTime == True:
|
|||
|
|
temp[0] = 'DateTime'
|
|||
|
|
itemNames.extend(temp)
|
|||
|
|
# print(itemNames)
|
|||
|
|
temp = [item for item in itemNames if len(item)>0]
|
|||
|
|
itemNames = temp[0:itemMaxNum]
|
|||
|
|
print(f"itemNames:{itemNames}")
|
|||
|
|
|
|||
|
|
#读取数据文件中的数据，以浮点数的形式读取
|
|||
|
|
rawData = np.array([])
|
|||
|
|
rawData_list = list(rawData)
|
|||
|
|
recordNum = 1
|
|||
|
|
for i in range(1, recordTotalNum-1):
|
|||
|
|
record = []
|
|||
|
|
line = f.readline()
|
|||
|
|
line = line.replace('"','')
|
|||
|
|
line = line.replace('/', ' ')
|
|||
|
|
line = line.replace('T', ' ')
|
|||
|
|
line = line.replace('Z', '')
|
|||
|
|
line = line.replace('\t', '')
|
|||
|
|
line = line.replace('\n', '')
|
|||
|
|
substrings = line.split(',')
|
|||
|
|
|
|||
|
|
if ContainDateTime:
|
|||
|
|
substrings[0] = substrings[0].split('.')[0]
|
|||
|
|
substrings[0] = substrings[0].replace('-',' ')
|
|||
|
|
substrings[0] = substrings[0].replace(':',' ')
|
|||
|
|
temp = substrings[0].split()
|
|||
|
|
date_str = ""
|
|||
|
|
for value in temp:
|
|||
|
|
if value == '':
|
|||
|
|
temp.remove(value)
|
|||
|
|
datetime_str = temp[0]+'-'+temp[1]+'-'+temp[2]+' '+temp[3]+':'+temp[4]+':'+temp[5]
|
|||
|
|
if datetime_str == '1970-1-1 1:1:1' or datetime_str == '1970-1-1 1:01:01':
|
|||
|
|
datetime_str = temp[0]+'-'+temp[1]+'-'+'2'+' '+temp[3]+':'+temp[4]+':'+temp[5]
|
|||
|
|
dt = datetime.datetime.strptime(datetime_str,'%Y-%m-%d %H:%M:%S')
|
|||
|
|
timestamp=time.mktime(dt.timetuple())
|
|||
|
|
else:
|
|||
|
|
datetime_str = '1971-1-2 1:1:1'
|
|||
|
|
dt = datetime.datetime.strptime(datetime_str,'%Y-%m-%d %H:%M:%S')
|
|||
|
|
timestamp=time.mktime(dt.timetuple())
|
|||
|
|
|
|||
|
|
record.append(recordNum)
|
|||
|
|
record.append(timestamp)
|
|||
|
|
|
|||
|
|
if ContainDateTime:
|
|||
|
|
for j in range(1, min(substrings.__len__(), itemMaxNum)):
|
|||
|
|
try:
|
|||
|
|
record.append(float(substrings[j].replace(' ','')))
|
|||
|
|
except:
|
|||
|
|
record.append(-1)
|
|||
|
|
pass
|
|||
|
|
else:
|
|||
|
|
for j in range(0, min(substrings.__len__(), itemMaxNum)):
|
|||
|
|
try:
|
|||
|
|
record.append(float(substrings[j].replace(' ','')))
|
|||
|
|
except:
|
|||
|
|
record.append(-1)
|
|||
|
|
pass
|
|||
|
|
if Filter:
|
|||
|
|
if record[1] >= start_timestamp and record[1] <= stop_timestamp:
|
|||
|
|
recordNum += 1
|
|||
|
|
rawData_list.append(record)
|
|||
|
|
else:
|
|||
|
|
recordNum += 1
|
|||
|
|
rawData_list.append(record)
|
|||
|
|
print(f"record:{record}")
|
|||
|
|
rawData = np.array(rawData_list)
|
|||
|
|
return itemNames, rawData
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
#解析参数
|
|||
|
|
fileName = sys.argv[1]
|
|||
|
|
timeRange = sys.argv[2]
|
|||
|
|
ContainDateTime = sys.argv[3]
|
|||
|
|
curveNum = int(sys.argv[4])
|
|||
|
|
|
|||
|
|
itemNamesRaw, recordsRaw = csv_to_nparray(fileName, ContainDateTime, timeRange)
|
|||
|
|
itemNames = itemNamesRaw
|
|||
|
|
records = recordsRaw
|
|||
|
|
|
|||
|
|
if ContainDateTime == 'True':
|
|||
|
|
DisplayTime = True
|
|||
|
|
else:
|
|||
|
|
DisplayTime = False
|
|||
|
|
|
|||
|
|
if DisplayTime == False:
|
|||
|
|
itemNames = np.delete(itemNames, 1, 0)
|
|||
|
|
# print(itemNames)
|
|||
|
|
|
|||
|
|
itemsNum = itemNames.__len__()
|
|||
|
|
# print(itemsNum)
|
|||
|
|
|
|||
|
|
if DisplayTime == False:
|
|||
|
|
records = np.delete(records, 1, 1)
|
|||
|
|
# print(records[:,:])
|
|||
|
|
|
|||
|
|
recordsNum = records.__len__()
|
|||
|
|
|
|||
|
|
if(itemsNum%curveNum == 0):
|
|||
|
|
rowNum = int(itemsNum/curveNum)
|
|||
|
|
else:
|
|||
|
|
rowNum = int(itemsNum/curveNum)+1
|
|||
|
|
|
|||
|
|
colors = ['#FFFFFF', '#FF0000', '#FFFF00', '#FF00FF', '#00FF00', '#00FFFF', '#C0FF00', '#FF0080', '#FF7F7F', '#7FFF7F','#FFFFFF']
|
|||
|
|
app = QtWidgets.QApplication([])
|
|||
|
|
axsY=[]
|
|||
|
|
axsX=[]
|
|||
|
|
vbs=[]
|
|||
|
|
vLine=[]
|
|||
|
|
# 创建视图
|
|||
|
|
pw = pg.GraphicsView()
|
|||
|
|
pw.setWindowTitle(fileName)
|
|||
|
|
pw.show()
|
|||
|
|
|
|||
|
|
# 创建图形布局
|
|||
|
|
l = pg.GraphicsLayout()
|
|||
|
|
# 设置视图中心小部件 为该布局
|
|||
|
|
pw.setCentralWidget(l)
|
|||
|
|
|
|||
|
|
label = pg.LabelItem(justify='left')
|
|||
|
|
l.addItem(label, row=0, col=0, rowspan=1, colspan=11)
|
|||
|
|
|
|||
|
|
labelStyle = {'font-size': '9pt'}
|
|||
|
|
for i in range(0, itemsNum-1):
|
|||
|
|
axsY.append(pg.AxisItem("left", pen=colors[i%10]))
|
|||
|
|
axsY[i].setLabel(itemNames[i+1], color=colors[i%10],**labelStyle)
|
|||
|
|
vbs.append(pg.ViewBox())
|
|||
|
|
l.addItem(axsY[i],row = int(i/curveNum)+1, col = i%curveNum, rowspan=1, colspan=1)
|
|||
|
|
l.addItem(vbs[i],row = int(i/curveNum)+1, col = 10, rowspan=1, colspan=1)
|
|||
|
|
axsY[i].linkToView(vbs[i])
|
|||
|
|
|
|||
|
|
if(i>0):
|
|||
|
|
vbs[i].setXLink(vbs[0])
|
|||
|
|
if(i%curveNum == 0):
|
|||
|
|
axsX.append(pg.AxisItem("bottom", pen=colors[10]))
|
|||
|
|
axsX[int(i/curveNum)].setLabel(itemNames[0], color=colors[10],**labelStyle)
|
|||
|
|
l.addItem(axsX[int(i/curveNum)],row = int(i/curveNum)+2, col = 10, rowspan=1, colspan=1)
|
|||
|
|
axsX[int(i/curveNum)].linkToView(vbs[0])
|
|||
|
|
|
|||
|
|
tempStr = ""
|
|||
|
|
if DisplayTime == True:
|
|||
|
|
tempStr += "<span style='font-size: 9pt; word-wrap: break-word; white-space: normal;'>NO.={SampleNO}, "
|
|||
|
|
BiasID = 0
|
|||
|
|
else:
|
|||
|
|
tempStr += "<span style='font-size: 9pt; word-wrap: break-word; white-space: normal;'>NO.={SampleNO}, <span style='color:white;'>DateTime={DateTime}</span>, "
|
|||
|
|
BiasID = 1
|
|||
|
|
for i in range(1, itemsNum):
|
|||
|
|
tempStr += "<span style='color:"
|
|||
|
|
tempStr += str(colors[(i-1)%10])
|
|||
|
|
tempStr += "'>"
|
|||
|
|
tempStr += itemNames[i]
|
|||
|
|
tempStr += "={"
|
|||
|
|
tempStr += itemNames[i]
|
|||
|
|
tempStr += "}</span>, "
|
|||
|
|
if(i%10 == 0):
|
|||
|
|
tempStr +="<br>"
|
|||
|
|
tempStr += "</span>"
|
|||
|
|
# print(tempStr)
|
|||
|
|
for i in range(0, rowNum):
|
|||
|
|
vLine.append(pg.InfiniteLine(angle=90, movable=True))
|
|||
|
|
vbs[i*curveNum-1].addItem(vLine[i])
|
|||
|
|
# pI.addItem(vLine)
|
|||
|
|
|
|||
|
|
def mouseMoved(evt):
|
|||
|
|
global tempStr, itemNamesRaw, recordsRaw, BiasID
|
|||
|
|
index = 0
|
|||
|
|
dic = {}
|
|||
|
|
pos = evt[0]
|
|||
|
|
# if l.sceneBoundingRect().contains(pos):
|
|||
|
|
for i in range(0, rowNum):
|
|||
|
|
if vbs[i*curveNum-1].sceneBoundingRect().contains(pos):
|
|||
|
|
mousePoint = vbs[i*curveNum-1].mapSceneToView(pos)
|
|||
|
|
index = int(mousePoint.x())
|
|||
|
|
|
|||
|
|
if index > 0 and index < recordsNum:
|
|||
|
|
dic[itemNamesRaw[0]] = index
|
|||
|
|
dic[itemNamesRaw[1]] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(recordsRaw[index-1,1]))
|
|||
|
|
for k in range(2, itemsNum+BiasID):
|
|||
|
|
dic[itemNamesRaw[k]] =recordsRaw[index-1,k]
|
|||
|
|
# print(dic)
|
|||
|
|
label.setText(tempStr.format(**dic))
|
|||
|
|
for i in range(0, rowNum):
|
|||
|
|
vLine[i].setPos(index)
|
|||
|
|
proxy = pg.SignalProxy(l.scene().sigMouseMoved, rateLimit=60, slot=mouseMoved)
|
|||
|
|
|
|||
|
|
# slot: 插槽：调整大小时更新视图
|
|||
|
|
def updateViews():
|
|||
|
|
# setGeometry设置几何图形
|
|||
|
|
# sceneBoundingRect场景边界矩形
|
|||
|
|
for i in range(0, itemsNum-1):
|
|||
|
|
vbs[i].setGeometry(vbs[i].sceneBoundingRect())
|
|||
|
|
|
|||
|
|
# 折线点值 （轴刻度范围自动按最大最小值生成）
|
|||
|
|
|
|||
|
|
# 给绘图项的视图框添加折线xy点的值，折线颜色
|
|||
|
|
for i in range(0, itemsNum-1):
|
|||
|
|
vbs[i].addItem(pg.PlotCurveItem(records[:,0], records[:,i+1], pen=colors[i%10]))
|
|||
|
|
#vbs[i].addItem(pg.ScatterPlotItem(records[:,0], records[:,i+1], pen=colors[i%10], symbolBrush=(255,0,0), symbolPen='w', size=2))
|
|||
|
|
|
|||
|
|
|
|||
|
|
# vbs[0].sigResized.connect(updateViews)
|
|||
|
|
|
|||
|
|
for i in range(0, itemsNum-1):
|
|||
|
|
vbs[i].enableAutoRange(axis= pg.ViewBox.XYAxes, enable=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 调整大小时更新
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 开始时自动调整一次以适应视图
|
|||
|
|
# enableAutoRange 启用自动范围
|
|||
|
|
|
|||
|
|
|
|||
|
|
updateViews()
|
|||
|
|
app.exec()
|