From 80cd2027b36af5866a7a82005afc367859e73b47 Mon Sep 17 00:00:00 2001
From: tao_z <tzj0429@163.com>
Date: Sat, 18 Jun 2022 17:16:17 +0800
Subject: [PATCH] 更新DTC测试功能

---
 widgets/ShifterTool.py |  180 ++++++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 151 insertions(+), 29 deletions(-)

diff --git a/widgets/ShifterTool.py b/widgets/ShifterTool.py
index 3ea4ac8..361fa4e 100644
--- a/widgets/ShifterTool.py
+++ b/widgets/ShifterTool.py
@@ -38,7 +38,7 @@
 fileHandler = logging.FileHandler('./log/ShiftTool.log',
                                   mode='a',
                                   encoding='UTF-8')
-fileHandler.setLevel(logging.NOTSET)
+fileHandler.setLevel(logging.DEBUG)
 
 consoleHandler.setFormatter(formatter)
 fileHandler.setFormatter(formatter)
@@ -335,10 +335,18 @@
             seed[3])
         if level == 0x01:
             output_key_temp = 0x20AA097B
+            output_key = struct.pack('BBBB', (output_key_temp >> 24) & 0xFF,
+                                     (output_key_temp >> 16) & 0xFF,
+                                     (output_key_temp >> 8) & 0xFF,
+                                     output_key_temp & 0xFF)
             # output_key_temp = ((((temp_key >> 4) ^ temp_key)
             #                     << 3) ^ temp_key) & 0xFFFFFFFF
         elif level == 0x09:
             output_key_temp = 0x70237577
+            output_key = struct.pack('BBBB', (output_key_temp >> 24) & 0xFF,
+                                     (output_key_temp >> 16) & 0xFF,
+                                     (output_key_temp >> 8) & 0xFF,
+                                     output_key_temp & 0xFF)
             # _temp_y = ((temp_key << 24) & 0xFF000000) + ((temp_key << 8) &
             #                                              0xFF0000) + ((temp_key >> 8) & 0xFF00) + ((temp_key >> 24) & 0xFF)
             # _temp_z = 0
@@ -354,13 +362,21 @@
             #     _temp_z = _temp_z & 0xFFFFFFFF
             # output_key_temp = (((_temp_z << 24) & 0xFF000000) | ((_temp_z << 8) & 0xFF0000) | (
             #     (_temp_z >> 8) & 0xFF00) | ((_temp_z >> 24) & 0xFF))
+        elif level == 0x7D:
+            output_key_temp = 0xAB2F9F36099D81F3
+            output_key = struct.pack(
+                'BBBBBBBB', (output_key_temp >> 56) & 0xFF,
+                (output_key_temp >> 48) & 0xFF, (output_key_temp >> 40) & 0xFF,
+                (output_key_temp >> 32) & 0xFF, (output_key_temp >> 24) & 0xFF,
+                (output_key_temp >> 16) & 0xFF, (output_key_temp >> 8) & 0xFF,
+                output_key_temp & 0xFF)
         else:
             output_key_temp = temp_key
 
-        output_key = struct.pack('BBBB', (output_key_temp >> 24) & 0xFF,
-                                 (output_key_temp >> 16) & 0xFF,
-                                 (output_key_temp >> 8) & 0xFF,
-                                 output_key_temp & 0xFF)
+            output_key = struct.pack('BBBB', (output_key_temp >> 24) & 0xFF,
+                                     (output_key_temp >> 16) & 0xFF,
+                                     (output_key_temp >> 8) & 0xFF,
+                                     output_key_temp & 0xFF)
 
         return output_key
 
@@ -430,14 +446,17 @@
         self.UI.comboBox_6.addItems(list(DID_dic.keys()))
         self.DID_display()
         self.UI.comboBox_6.activated.connect(self.DID_display)
+        self.UI.comboBox_7.addItems(list(DTCGroup_dic.keys()))
+        self.UI.comboBox_7.activated.connect(self.ClearDTC)
 
         self.UI.pushButton_31.clicked.connect(self.SecurityUnlockLevel_1)
-        self.UI.pushButton_32.clicked.connect(self.SecurityUnlockLevel_3)
+        self.UI.pushButton_32.clicked.connect(self.SecurityUnlockLevel_2)
 
         self.UI.pushButton_12.clicked.connect(self.ReadSupplyID)
         self.UI.pushButton_4.clicked.connect(self.ReadVIN)
         self.UI.pushButton_6.clicked.connect(self.ReadMfgDate)
         self.UI.pushButton_14.clicked.connect(self.ReadDataByID)
+        self.UI.pushButton_3.clicked.connect(self.WriteDataByID)
         self.UI.pushButton_9.clicked.connect(self.StartCalibraiton)
         self.UI.pushButton_11.clicked.connect(self.Calibraiton_Z)
         self.UI.pushButton_13.clicked.connect(self.Calibraiton_M)
@@ -454,6 +473,15 @@
         self.UI.pushButton_48.clicked.connect(self.start_programming)
         self.UI.pushButton_46.clicked.connect(self.loadAppfile)
         self.UI.radioButton_2.toggled.connect(self.SetFromBootFlag)
+
+        self.UI.comboBox_11.addItems(list(ReportBymask_DTC.keys()))
+        self.UI.comboBox_12.addItems(list(DTC_Dic.keys()))
+        self.UI.comboBox_13.addItems(list(DTC_Control_dic.keys()))
+        self.UI.comboBox_13.activated.connect(self.DTC_Control)
+
+        self.UI.pushButton_49.clicked.connect(self.ReportSupportDTC)
+        self.UI.comboBox_11.activated.connect(self.ReportByMask)
+        self.UI.comboBox_12.activated.connect(self.ReportSnapshotByDTC)
 
     def _formatMsgData(self, index, item, received):
         '''msg data to list
@@ -487,7 +515,7 @@
             data.append('标准帧')
 
         data.append(' '.join(
-            ['0x' + '{:0<2x}'.format(a).upper() for a in list(item.data)]))
+            ['0x' + '{:0>2x}'.format(a).upper() for a in list(item.data)]))
         return data
 
     def can_thread_stop(self):
@@ -624,7 +652,7 @@
         for i in range(_dataSize):
             receiveNum += 1
             msg = self.msgQueue.get()
-            self.shifter.FramUnpack(msg.arbitration_id, msg.data)
+            # self.shifter.FramUnpack(msg.arbitration_id, msg.data)
             formateddata.append(self._formatMsgData(
                 receiveNum, msg, True))  # return a data list
         self._insertDataSmooth(data=formateddata, datasize=_dataSize)
@@ -660,16 +688,20 @@
 
     def sessioncontrol_req(self):
         sesson = self.UI.comboBox.currentIndex() + 1
-        t = threading.Thread(target=self.udsclient.change_session,
-                             args=(sesson, ))
-        t.start()
-        t.join()
-        # self.udsclient.change_session(sesson)
+        self.udsclient.change_session(sesson)
 
     # self.UI.comboBox_8.addItems(['0x01硬件复位', '0x03软件复位'])
     def MCUReset_req(self):
-        req = 0x03 if self.UI.comboBox_8.currentData() == '0x01硬件复位' else 0x01
+        req = 0x03 if self.UI.comboBox_8.currentText() == '0x01硬件复位' else 0x01
         self.udsclient.ecu_reset(req)
+
+    def ClearDTC(self):
+        self.udsclient.clear_dtc(
+            group=DTCGroup_dic[self.UI.comboBox_7.currentText()])
+
+    def DTC_Control(self):
+        self.udsclient.control_dtc_setting(
+            setting_type=DTC_Control_dic[self.UI.comboBox_13.currentText()])
 
     def Security_req(self):
         self.udsclient.unlock_security_access(1)
@@ -681,11 +713,18 @@
             self.UI.pushButton_31.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
-    def SecurityUnlockLevel_3(self):
+    def SecurityUnlockLevel_2(self):
         data = self.udsclient.unlock_security_access(9)
         if data.positive:
             self.UI.pushButton_32.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
+
+    def SecurityUnlockLevel_3(self):
+        data = self.udsclient.unlock_security_access(0x7d)
+        if data.positive:
+            self.UI.pushButton_32.setStyleSheet(
+                "background-color:rgb(0, 128, 0)")
+        return data
 
     def ReadSupplyID(self):
         data = self.ReadByDID(0xF18A)
@@ -722,7 +761,19 @@
         self.UI.lineEdit_2.clear()
         data = self.ReadByDID(tempdid)
         if data is not None and len(str(data[tempdid])):
-            self.UI.lineEdit_2.setText(str(data[tempdid]) + ' ')
+            # print(data[tempdid])
+            self.UI.lineEdit_2.setText(data[tempdid])
+
+    def WriteDataByID(self):
+        tempdid = DID_dic[self.UI.comboBox_6.currentText()]
+        writedata = self.UI.lineEdit_2.text()
+        try:
+            response = self.udsclient.write_data_by_identifier(
+                tempdid, writedata)
+            if response.positive:
+                values = response.service_data.values
+        except Exception as e:
+            g_signal.sig_Disp_str.emit(e)
 
     def communicationControl_req(self):
         req = 0x00
@@ -733,14 +784,23 @@
             req = 1
         else:
             req = 3
-        self.udsclient.communication_control(control_type=0,
-                                             communication_type=req)
+        try:
+            self.udsclient.communication_control(control_type=req,
+                                                 communication_type=1)
+        except Exception as e:
+            self.disp_string('%s' % e)
 
     def StartCalibraiton(self):
-        startcmd = bytearray([0x0, 0x0, 0x0, 0x0])
-        if self.RequestRoutControl(0xFFAA, bytes(startcmd)):
-            self.UI.pushButton_9.setStyleSheet(
-                "background-color:rgb(255, 85, 127)")
+        sesson = 3
+        self.UI.comboBox.setCurrentIndex(sesson - 1)
+        response = self.udsclient.change_session(sesson)
+        if response.positive:
+            response = self.SecurityUnlockLevel_3()
+        if response.positive:
+            startcmd = bytearray([0x8, 0x0, 0x0, 0x0])
+            if self.RequestRoutControl(0xFFAA, bytes(startcmd)):
+                self.UI.pushButton_9.setStyleSheet(
+                    "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_Z(self):
         calcmd = bytearray([0x1, 0x0, 0x0, 0x0])
@@ -749,43 +809,43 @@
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_M(self):
-        calcmd = bytearray([0x2, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x0, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_13.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_MP(self):
-        calcmd = bytearray([0x3, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x2, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_18.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_MN(self):
-        calcmd = bytearray([0x4, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x3, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_22.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_X2(self):
-        calcmd = bytearray([0x5, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x4, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_24.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_X1(self):
-        calcmd = bytearray([0x6, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x5, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_26.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_Y1(self):
-        calcmd = bytearray([0x7, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x6, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_27.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
 
     def Calibraiton_Y2(self):
-        calcmd = bytearray([0x8, 0x0, 0x0, 0x0])
+        calcmd = bytearray([0x7, 0x0, 0x0, 0x0])
         if self.RequestRoutControl(0xFFAA, bytes(calcmd)):
             self.UI.pushButton_28.setStyleSheet(
                 "background-color:rgb(255, 85, 127)")
@@ -800,6 +860,68 @@
         resp_1 = self.udsclient.start_routine(routine_id=routine_id, data=data)
         return resp_1.positive
 
+    def ReportSupportDTC(self):
+        resp_1 = self.udsclient.get_supported_dtc()
+        if resp_1.positive:
+            for i in range(len(resp_1.data) // 4):
+                a = i * 4 + 2
+                b = i * 4 + 3
+                c = i * 4 + 4
+                d = i * 4 + 5
+                dtc_hex = resp_1.data[a] << 16 | resp_1.data[
+                    b] << 8 | resp_1.data[c] << 0
+                log = 'DTC:0x%.6x Status:0x%.2x ' % (
+                    dtc_hex, resp_1.data[d]) + DTC_DescriptionDic[dtc_hex]
+                for j in range(8):
+                    if resp_1.data[d] >> j & 0x1:
+                        log += ' bit%d' % j
+                self.disp_string(log)
+
+    def ReportByMask(self):
+        mask = ReportBymask_DTC[self.UI.comboBox_11.currentText()]
+        resp_1 = self.udsclient.get_dtc_by_status_mask(status_mask=mask)
+        if resp_1.positive:
+            for i in range(len(resp_1.data) // 4):
+                a = i * 4 + 2
+                b = i * 4 + 3
+                c = i * 4 + 4
+                d = i * 4 + 5
+                dtc_hex = resp_1.data[a] << 16 | resp_1.data[
+                    b] << 8 | resp_1.data[c] << 0
+                log = 'DTC:0x%.2x%.2x%.2x Status:0x%.2x %s ' % (
+                    resp_1.data[a], resp_1.data[b], resp_1.data[c],
+                    resp_1.data[d], DTC_DescriptionDic[dtc_hex])
+                for j in range(8):
+                    if resp_1.data[d] >> j & 0x1:
+                        log += 'bit%d ' % j
+                self.disp_string(log)
+
+    def ReportSnapshotByDTC(self):
+        dtc = DTC_Dic[self.UI.comboBox_12.currentText()][
+            self.UI.comboBox_12.currentIndex()]
+        try:
+            resp = self.udsclient.get_dtc_snapshot_by_dtc_number(
+                dtc=dtc, record_number=1)
+            if resp.positive:
+                number_of_did = resp.data[6]
+                status = resp.data[4]
+                print(status)
+                dtc = resp.service_data.dtcs[0]
+                snapshots = dtc.snapshots
+                # print(dtc.id)
+                # print(dtc.status.get_byte())
+                log = "DTC:0x%.6x Status:0x%.2x" % (dtc.id, status) + ' '
+                # print(type(snapshots[0].did))
+                # print(type(snapshots[0].data))
+                for i in range(number_of_did):
+                    log += (DTC_SanpshotDescriptionDic[snapshots[i].did] +
+                            ':' + snapshots[i].data + ' ')
+                self.disp_string(log)
+
+                # print(snapshots[i].did, snapshots[i].data)
+        except Exception as e:
+            self.disp_string('%s' % e)
+
     def SetFromBootFlag(self):
         self.startfromboot = self.UI.radioButton_2.isChecked()
 

--
Gitblit v1.8.0