319 lines
13 KiB
Python
319 lines
13 KiB
Python
|
import os, time, csv, struct, sys, shutil
|
|||
|
|
import numpy as np
|
|||
|
|
import xlwings as xw
|
|||
|
|
import pandas as pd
|
|||
|
|
from datetime import datetime
|
|||
|
|
sys.path.append(r'.\UserScripts')
|
|||
|
|
import CRC
|
|||
|
|
from win32com.client import Dispatch
|
|||
|
|
import pythoncom, report
|
|||
|
|
|
|||
|
|
# 将bytearray以字为单位转换成16进制字符串，方便显示
|
|||
|
|
def wordData2HexStr(data):
|
|||
|
|
ret = ' '.join(data[i:i+2].hex() for i in range(0, len(data), 2))
|
|||
|
|
return ret.upper()
|
|||
|
|
|
|||
|
|
# 获取当前时间字符串
|
|||
|
|
def nowStr():
|
|||
|
|
now = datetime.now()
|
|||
|
|
ret = now.strftime('%H:%M:%S.') + f"{now.microsecond // 1000:03d}"
|
|||
|
|
return ret
|
|||
|
|
|
|||
|
|
def checkValue(data: bytes) -> int:
|
|||
|
|
crc = 0xFFFF
|
|||
|
|
length = len(data)
|
|||
|
|
if length % 2 != 0:
|
|||
|
|
return 0
|
|||
|
|
for i in range(0, length, 2):
|
|||
|
|
val = data[i] * 256 + data[i + 1]
|
|||
|
|
crc = crc ^ val
|
|||
|
|
return crc
|
|||
|
|
|
|||
|
|
def sendCmd(hexStr):
|
|||
|
|
attr = cmdInfo['attr']
|
|||
|
|
try:
|
|||
|
|
data = bytearray().fromhex(hexStr) #将十六进制字符串转成字节数据发送
|
|||
|
|
if attr['req_is_check'] == True:
|
|||
|
|
crc = CRC.crc8(data)
|
|||
|
|
data = data + bytearray(crc.to_bytes(1, 'big'))
|
|||
|
|
send(data)
|
|||
|
|
except Exception as e:
|
|||
|
|
log_e(f"Error in sendCmd. {str(e)}")
|
|||
|
|
|
|||
|
|
# 启动函数，命令开始会调用此函数
|
|||
|
|
def start():
|
|||
|
|
global startTime, recvData, runMode, parseStr, timeDelta, dataPoints, sdimDatas, deviceInfo, keys, workMode
|
|||
|
|
global modeName, avePointsNum, deviceInfo, fo, RECORDSNUM, recordNum, fileName, SN,istoxlsx
|
|||
|
|
try:
|
|||
|
|
|
|||
|
|
log_i(f"[{devInfo['name']}] {cmdInfo['name']}".center(80, '-'))
|
|||
|
|
attr = cmdInfo['attr']
|
|||
|
|
if not attr.__contains__('params') or not attr['params']:
|
|||
|
|
attr['params'] = scanf('输入参数', '请输入记录条数, 平均点数, 是否新建文件(1-是, 0-否),是否导出报告(1-是, 0-否)', '1, 10, 1,0')
|
|||
|
|
# attr['params'] = '0.005， 100'
|
|||
|
|
if not attr['params']:
|
|||
|
|
raise ValueError(f"Cancel.")
|
|||
|
|
params = attr['params']
|
|||
|
|
values = params.replace(' ','').split(',')
|
|||
|
|
RECORDSNUM = int(values[0])
|
|||
|
|
recordNum = 0
|
|||
|
|
isCreateFile = bool(int(values[2]))
|
|||
|
|
istoxlsx = bool(int(values[3]))
|
|||
|
|
# log(str(isCreateFile))
|
|||
|
|
cmdInfo['attr']['rsp_timeout'] = 400
|
|||
|
|
if RECORDSNUM <= 0:
|
|||
|
|
raise ValueError(f"Wrong params.")
|
|||
|
|
|
|||
|
|
avePointsNum = int(values[1])
|
|||
|
|
modeName = {'0x0001':'Calibration','0x0002':'Verification', '0x0003':'CalibI', '0x0004':'CalibII'}
|
|||
|
|
deviceInfo = _G.get(devInfo['name'])
|
|||
|
|
if not deviceInfo:
|
|||
|
|
deviceInfo = {'data':{}}
|
|||
|
|
|
|||
|
|
# if 'workMode' in deviceInfo['data']:
|
|||
|
|
# workMode = deviceInfo['data']['workMode']
|
|||
|
|
# if workMode not in modeName:
|
|||
|
|
# raise ValueError(f"Please set tool to right mode.")
|
|||
|
|
# else:
|
|||
|
|
workMode = '0x0003'
|
|||
|
|
|
|||
|
|
SN = _G.get('SN')
|
|||
|
|
|
|||
|
|
if not SN:
|
|||
|
|
log(f"Unable to read SN number,Set the default SN number to 123.")
|
|||
|
|
SN = 123
|
|||
|
|
|
|||
|
|
keysList = {'0x0001':['GX_AD','GX_T','GY_AD','GY_T','GZ_AD','GZ_T','Pt','MX_AD','MY_AD','MZ_AD','M_T','INC','GTF','GT','BRD_T','AZI','MT'],
|
|||
|
|
'0x0002':['GX','GY','GZ','SINC','DIP','SAZI','MX','MY','MZ','MTF','RPM','INC','GTF','GT','MODE','AZI','MT'],
|
|||
|
|
'0x0003':['GX_AD','GX_T','GY_AD','GY_T','GZ_AD','GZ_T','Pt','MX_AD','MY_AD','MZ_AD','M_T','INC','GTF','BRD_T','GT','MT','AZI'],
|
|||
|
|
'0x0004':['GX_AD','GX_T','GY_AD','GY_T','GZ_AD','GZ_T','Pt','MX_AD','MY_AD','MZ_AD','M_T','INC','GTF','CURR_AD','GT','MT','AZI']}
|
|||
|
|
keys = keysList[workMode]
|
|||
|
|
|
|||
|
|
#新建数据文件
|
|||
|
|
if not proInfo['path']:
|
|||
|
|
raise ValueError(f"Please create project first.")
|
|||
|
|
if not os.path.exists(f"{proInfo['path']}/T150_SDIM/"):
|
|||
|
|
os.makedirs(f"{proInfo['path']}/T150_SDIM/")
|
|||
|
|
|
|||
|
|
#log(proInfo['path'])
|
|||
|
|
#判断模板文件是否存在
|
|||
|
|
if not os.path.exists(f"{os.path.dirname(os.path.dirname(proInfo['path']))}/Testcases/T150_SDIM/SDIM.FT.ReportTemplate.xlsx"):
|
|||
|
|
raise ValueError(f"Dosen't exist {os.path.dirname(os.path.dirname(proInfo['path']))}/Testcases/T150_SDIM/SDIM.FT.ReportTemplate.xlsx")
|
|||
|
|
TC_SDIM_Calib = _G.get('TC_SDIM_Calib')
|
|||
|
|
if not TC_SDIM_Calib:
|
|||
|
|
TC_SDIM_Calib = {'dataFile':f"{proInfo['path']}/T150_SDIM/{SN}静态标定_{datetime.now().strftime('%Y%m%d%H%M%S')}.csv"}
|
|||
|
|
_G.set('TC_SDIM_Calib', TC_SDIM_Calib)
|
|||
|
|
# _G.set('TcFlieCSV', TC_SDIM_Calib)
|
|||
|
|
fileName = TC_SDIM_Calib['dataFile']
|
|||
|
|
if os.path.exists(fileName):
|
|||
|
|
if isCreateFile:
|
|||
|
|
fileName = f"{proInfo['path']}/T150_SDIM/{SN}静态标定_{datetime.now().strftime('%Y%m%d%H%M%S')}.csv"
|
|||
|
|
TC_SDIM_Calib['dataFile'] = fileName
|
|||
|
|
_G.set('TC_SDIM_Calib', TC_SDIM_Calib)
|
|||
|
|
_G.set('TcFlieCSV', TC_SDIM_Calib)
|
|||
|
|
fo = open(fileName, 'w+', encoding='utf-8', newline='')
|
|||
|
|
fo.write(','.join(keys)+'\n')
|
|||
|
|
fo.flush()
|
|||
|
|
else:
|
|||
|
|
fo = open(fileName, 'a', encoding='utf-8', newline='')
|
|||
|
|
else:
|
|||
|
|
fo = open(fileName, 'w+', encoding='utf-8', newline='')
|
|||
|
|
fo.write(','.join(keys)+'\n')
|
|||
|
|
fo.flush()
|
|||
|
|
|
|||
|
|
deviceInfo['listenStatus'] = 1
|
|||
|
|
_G.set(devInfo['name'], deviceInfo)
|
|||
|
|
|
|||
|
|
timeDelta = None
|
|||
|
|
dataPoints = []
|
|||
|
|
sdimDatas = []
|
|||
|
|
|
|||
|
|
|
|||
|
|
startTime = time.time() #记录启动时间
|
|||
|
|
recvData = bytearray()
|
|||
|
|
|
|||
|
|
log(f"[{nowStr()}]: Start to listen response of T150-SDIM Board....")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
finish()
|
|||
|
|
log_e(f"Error in loop(): {str(e)}")
|
|||
|
|
|
|||
|
|
# 接收数据处理函数，当收到数据会调用此函数
|
|||
|
|
def recvDataHandler(data):
|
|||
|
|
global recvData
|
|||
|
|
recvData = recvData + data
|
|||
|
|
if len(recvData)>60:
|
|||
|
|
log(f"{wordData2HexStr(recvData)}")
|
|||
|
|
recvData = bytearray()
|
|||
|
|
|
|||
|
|
def dataParse():
|
|||
|
|
global recvData, rsp_len, remainingRetryTimes, deviceData, workMode, avePointsNum, deviceInfo, fo, dataPoints, sdimDatas
|
|||
|
|
global RECORDSNUM, recordNum, fileName, SN,istoxlsx
|
|||
|
|
try:
|
|||
|
|
temp = bytearray()
|
|||
|
|
for i in range(18):
|
|||
|
|
temp.extend(recvData[i*3+4:i*3+7])
|
|||
|
|
temp.append(0x00)
|
|||
|
|
rawData = list(struct.unpack('<18I', temp))
|
|||
|
|
sdimData = []
|
|||
|
|
if workMode == '0x0001':
|
|||
|
|
for i in range(17):
|
|||
|
|
sdimData.append(rawData[0+i])
|
|||
|
|
elif workMode == '0x0002':
|
|||
|
|
for i in range(3):
|
|||
|
|
sdimData.append(rawData[0+i]/10000.0-2)
|
|||
|
|
sdimData.append(rawData[3]/100.0)
|
|||
|
|
sdimData.append(rawData[4]/100.0-90)
|
|||
|
|
sdimData.append(rawData[5]/100.0)
|
|||
|
|
for i in range(3):
|
|||
|
|
sdimData.append(rawData[6+i]/10000.0-2)
|
|||
|
|
sdimData.append(rawData[9]/100.0)
|
|||
|
|
sdimData.append(rawData[10]/1.0)
|
|||
|
|
sdimData.append(rawData[11]/100.0)
|
|||
|
|
sdimData.append(rawData[12]/100.0)
|
|||
|
|
sdimData.append(rawData[13]/10000.0)
|
|||
|
|
sdimData.append(rawData[14])
|
|||
|
|
sdimData.append(rawData[15]/100.0)
|
|||
|
|
sdimData.append(rawData[16]/10000.0)
|
|||
|
|
|
|||
|
|
elif workMode == '0x0003' or workMode == '0x0004':
|
|||
|
|
log_w(f"{rawData}")
|
|||
|
|
for i in range(17):
|
|||
|
|
sdimData.append(rawData[i])
|
|||
|
|
|
|||
|
|
measData = dict(zip(keys, sdimData))
|
|||
|
|
#log_e(f"measData{measData}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
dataPoints.append({'measurement':f"SDIM.{modeName[workMode]}", 'time':datetime.utcfromtimestamp(time.time()), 'fields': measData})
|
|||
|
|
if len(dataPoints) >= avePointsNum:
|
|||
|
|
#写入influxDB
|
|||
|
|
tsdb.write_points(dataPoints)
|
|||
|
|
dataPoints = []
|
|||
|
|
#写入csv文件
|
|||
|
|
temp = np.average(np.array(sdimDatas), axis=0)
|
|||
|
|
temp_int = [int(num) for num in temp] #对平均后的数据取整
|
|||
|
|
|
|||
|
|
dataPoint = dict(zip(keys, temp_int)) #组合
|
|||
|
|
deviceInfo['data']['measData'] = dataPoint
|
|||
|
|
_G.set(devInfo['name'], deviceInfo)
|
|||
|
|
|
|||
|
|
log(f"AverageData{dataPoint}", level = 'INFO', color = '#228B22')
|
|||
|
|
fo.write(','.join([str(round(x,0)) for x in temp])+'\n')
|
|||
|
|
recordNum += 1
|
|||
|
|
if recordNum >= RECORDSNUM:
|
|||
|
|
finish()
|
|||
|
|
#根据模板文件和数据文件，新建报告文件
|
|||
|
|
fo.close()
|
|||
|
|
if istoxlsx == 1:
|
|||
|
|
shutil.copy(f"{os.path.dirname(os.path.dirname(proInfo['path']))}/Testcases/T150_SDIM/SDIM.FT.ReportTemplate.xlsx", f"{fileName[:-4]}.xlsx")
|
|||
|
|
copyDataFromCSVToXlsx(fileName, f"{fileName[:-4]}.xlsx",SN)
|
|||
|
|
log(f"[{nowStr()}]: Stop to listen response of T150-SDIM Board!")
|
|||
|
|
sdimDatas = []
|
|||
|
|
else:
|
|||
|
|
sdimDatas.append(sdimData)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
log_e(f"Error in dataParse. {str(e)}")
|
|||
|
|
finish()
|
|||
|
|
|
|||
|
|
# 此函数会被重复调用，间隔10毫秒，直到finish()
|
|||
|
|
def loop():
|
|||
|
|
global startTime, runMode, recvData, parseStr, timeDelta, dataPoints, deviceInfo, keys, workMode, modeName, avePointsNum
|
|||
|
|
global RECORDSNUM, recordNum
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
|
|||
|
|
if ((time.time() > startTime + int(cmdInfo['attr']['rsp_timeout'])) and runMode == 0):
|
|||
|
|
log_e(f"[{nowStr()}]: Timeout error. Stop to listen response of T150-SDIM Board!")
|
|||
|
|
fo.close()
|
|||
|
|
finish()
|
|||
|
|
else:
|
|||
|
|
if len(recvData) >= 60:
|
|||
|
|
if recvData[0:4] == bytearray().fromhex('90 EB 0C EA'):
|
|||
|
|
log_d(f"[{nowStr()}]: {wordData2HexStr(recvData[0:60])}")
|
|||
|
|
dataParse()
|
|||
|
|
recvData = recvData[60:]
|
|||
|
|
else:
|
|||
|
|
recvData = recvData[1:]
|
|||
|
|
except Exception as e:
|
|||
|
|
finish()
|
|||
|
|
log_e(f"Error in loop(): {str(e)}")
|
|||
|
|
|
|||
|
|
def copyDataFromCSVToXlsx(csvFile, xlsxFile, SN):
|
|||
|
|
try:
|
|||
|
|
pythoncom.CoInitialize()
|
|||
|
|
xl = xw._xlwindows.COMRetryObjectWrapper(Dispatch("ket.Application"))
|
|||
|
|
impl = xw._xlwindows.App(visible = False , add_book = False , xl = xl)
|
|||
|
|
app = xw.App(visible = False , add_book = False , impl = impl)
|
|||
|
|
workbook = app.books.open(xlsxFile)
|
|||
|
|
tempJsonData = csvFileToDict(csvFile)
|
|||
|
|
log(str(tempJsonData))
|
|||
|
|
fillData('', tempJsonData, workbook)
|
|||
|
|
#sheet = workbook.sheets['Sheet1']
|
|||
|
|
#sheet.name = f"{SN}#"
|
|||
|
|
workbook.save(xlsxFile)
|
|||
|
|
workbook.close()
|
|||
|
|
app.kill()
|
|||
|
|
except Exception as e:
|
|||
|
|
log_e(f"Error in copyDataFromCSVToXlsx. {str(e)}")
|
|||
|
|
finish()
|
|||
|
|
|
|||
|
|
def csvFileToDict(csvFilePath):
|
|||
|
|
info = ''
|
|||
|
|
try:
|
|||
|
|
if not os.path.exists(csvFilePath):
|
|||
|
|
raise Exception(f"{csvFilePath} file doesn't exist.")
|
|||
|
|
tempTitle = []
|
|||
|
|
tempData = []
|
|||
|
|
dfData = pd.DataFrame({})
|
|||
|
|
with open(csvFilePath, "r", encoding='utf-8') as f:
|
|||
|
|
reader = csv.reader(f)
|
|||
|
|
lines = [line for line in reader]
|
|||
|
|
# reversed_lines = lines[::-1]
|
|||
|
|
for line in lines:
|
|||
|
|
if line[0].startswith('ASM.GX.AD'):
|
|||
|
|
if len(tempData) > 0:
|
|||
|
|
tempDataFrame = pd.DataFrame(dict(zip(tempTitle, list(map(list, zip(*tempData))))))
|
|||
|
|
dfData = pd.concat([dfData, tempDataFrame])
|
|||
|
|
# tempDataFrame.to_csv(f"{csvFilePath}.{tempData[0][1]}.csv")
|
|||
|
|
tempData = []
|
|||
|
|
tempTitle = line
|
|||
|
|
else:
|
|||
|
|
tempData.append(line)
|
|||
|
|
tempDataFrame = pd.DataFrame(dict(zip(tempTitle, list(map(list, zip(*tempData))))))
|
|||
|
|
dfData = pd.concat([dfData, tempDataFrame])
|
|||
|
|
|
|||
|
|
# dfData = dfData.replace(np.nan, '0')
|
|||
|
|
dictData = dfData.to_dict(orient='list')
|
|||
|
|
return dictData
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"Error in csvFileToDict(): {str(e)}")
|
|||
|
|
|
|||
|
|
def fillData(baseName, data, workbook):
|
|||
|
|
try:
|
|||
|
|
info = ''
|
|||
|
|
if isinstance(data, dict):
|
|||
|
|
for key, value in data.items():
|
|||
|
|
if (not isinstance(value, dict)) or (not value):
|
|||
|
|
log(f"{{{baseName}.{key}}}".replace('{.', '{'))
|
|||
|
|
info += f"{{{baseName}.{key}}}"
|
|||
|
|
for worksheet in workbook.sheets:
|
|||
|
|
result = worksheet.used_range.api.find(f"{{{baseName}.{key.replace('{.','{')}}}".replace('{.', '{'))
|
|||
|
|
if result:
|
|||
|
|
if isinstance(value, list):
|
|||
|
|
address = f"{result.address}".split('$')
|
|||
|
|
worksheet.range(f"{result.address}:${address[1]}${int(address[2])+len(value)-1}").options(transpose=True).value = value
|
|||
|
|
else:
|
|||
|
|
worksheet.used_range.api.Replace(f"{{{baseName}.{key}}}", f"{value}")
|
|||
|
|
else:
|
|||
|
|
baseName += f".{key}"
|
|||
|
|
fillData(baseName, value, workbook)
|
|||
|
|
baseName = baseName[0:baseName.rfind('.')]
|
|||
|
|
else:
|
|||
|
|
info += f"{data}"
|
|||
|
|
raise Exception(f"Error in fillData(): Type of data is not dict.")
|
|||
|
|
return info
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"Error in fillData(): {str(e)}")
|