Update converter.py

Author: Mattemn97

converter.py

@@ -62,34 +62,72 @@ def process_blf_file(db, blf_file, msg_list, sgn_list, output):
     return output
 
 
-def write_csv(output, dbc_file, blf_file):
-    """Write the output structure to a CSV file with a name based on the DBC and BLF filenames."""
+def process_trc_file(db, trc_file, msg_list, sgn_list, output):
+    """Process the TRC file and fill in the output structure with decoded signal values."""
+    can_log = can.CanalyzerLogReader(trc_file)
+    first = True
+
+    for msg in tqdm(can_log, desc=f"Reading signals in {os.path.basename(trc_file)}"):
+        try:
+            msg_name = db.get_message_by_frame_id(msg.arbitration_id).name
+            cur_frame = db.decode_message(msg.arbitration_id, msg.data, decode_choices=False)
+            if first:
+                first = False
+                start_abs = msg.timestamp
+                output[0].append("Timestamp")
+            else:
+                output[0].append(msg.timestamp - start_abs)
+
+            for i in range(len(sgn_list)):
+                if sgn_list[i] in cur_frame and msg_name == msg_list[i]:
+                    output[i + 1].append(cur_frame[sgn_list[i]])
+                else:
+                    output[i + 1].append('')
+
+        except KeyError:
+            pass
+
+    return output
+
+
+def write_csv(output, dbc_file, log_file):
+    """Write the output structure to a CSV file with a name based on the DBC and log filenames."""
     dbc_basename = os.path.splitext(os.path.basename(dbc_file))[0]
-    blf_basename = os.path.splitext(os.path.basename(blf_file))[0]
-    csv_filename = f"{dbc_basename}_{blf_basename}.csv"
+    log_basename = os.path.splitext(os.path.basename(log_file))[0]
+    csv_filename = f"{dbc_basename}_{log_basename}.csv"
 
     print(f"Creating {csv_filename}")
     with open(csv_filename, "w", newline='') as f:
         writer = csv.writer(f, delimiter=';')
         writer.writerows(zip(*output))
 
 
-def convert_blf_to_csv(dbc_files, blf_file):
-    """Main function to convert a BLF file to multiple CSV files using multiple DBC files."""
+def convert_log_to_csv(dbc_files, log_file):
+    """Main function to convert a log file (.blf or .trc) to multiple CSV files using multiple DBC files."""
+    file_ext = os.path.splitext(log_file)[1].lower()
+
     for dbc_file in dbc_files:
         db = load_dbc_file(dbc_file)
         msg_list, sgn_list = extract_signals_from_dbc(db, dbc_file)
         output = initialize_output_structure(sgn_list)
-        output = process_blf_file(db, blf_file, msg_list, sgn_list, output)
-        write_csv(output, dbc_file, blf_file)
+
+        if file_ext == ".blf":
+            output = process_blf_file(db, log_file, msg_list, sgn_list, output)
+        elif file_ext == ".trc":
+            output = process_trc_file(db, log_file, msg_list, sgn_list, output)
+        else:
+            print(f"Unsupported file format: {file_ext}")
+            return
+
+        write_csv(output, dbc_file, log_file)
 
 
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Convert BLF files to CSV using one or more DBC files.")
-    parser.add_argument("--blf", required=True, help="The path to the .blf file to convert.")
+    parser = argparse.ArgumentParser(description="Convert log files (.blf or .trc) to CSV using one or more DBC files.")
+    parser.add_argument("--log", required=True, help="The path to the .blf or .trc file to convert.")
     parser.add_argument("--dbc", required=True, nargs='+',
                         help="The path(s) to the .dbc file(s) used for interpreting CAN messages.")
 
     args = parser.parse_args()
 
-    convert_blf_to_csv(args.dbc, args.blf)
+    convert_log_to_csv(args.dbc, args.log)