implement trc writer for v1.1

Author: hiteshhedwig

can/io/trc.py

@@ -260,12 +260,15 @@ class TRCWriter(TextIOMessageWriter):
     FORMAT_MESSAGE = (
         "{msgnr:>7} {time:13.3f} DT {channel:>2} {id:>8} {dir:>2} -  {dlc:<4} {data}"
     )
+    FORMAT_MESSAGE_V1_1 = "{msgnr:>6}){time:12.3f}  Rx     {id:>8}  {dlc:<1}  {data}"
+
     FORMAT_MESSAGE_V1_0 = "{msgnr:>6}) {time:7.0f} {id:>8} {dlc:<1} {data}"
 
     def __init__(
         self,
         file: Union[StringPathLike, TextIO],
         channel: int = 1,
+        fileversion : int = TRCFileVersion.V1_1,
         **kwargs: Any,
     ) -> None:
         """
@@ -287,7 +290,7 @@ def __init__(
         self.header_written = False
         self.msgnr = 0
         self.first_timestamp = None
-        self.file_version = TRCFileVersion.V2_1
+        self.file_version = fileversion
         self._msg_fmt_string = self.FORMAT_MESSAGE_V1_0
         self._format_message = self._format_message_init
 
@@ -312,6 +315,33 @@ def _write_header_v1_0(self, start_time: datetime) -> None:
         ]
         self.file.writelines(line + "\n" for line in lines)
 
+    def _write_header_v1_1(self, start_time: datetime) -> None:
+        header_time = start_time - datetime(year=1899, month=12, day=30)      
+        lines = [
+            ";$FILEVERSION=1.1",
+            f";$STARTTIME={header_time/timedelta(days=1)}",
+            ";$COLUMNS=N,O,T,B,I,d,R,L,D",
+            ";",
+            f";   {self.filepath}",
+            ";",
+            f";   Start time: {start_time}",
+            ";   Generated by python-can TRCWriter",
+            ";-------------------------------------------------------------------------------",
+            ";   Bus   Name            Connection               Protocol",
+            ";   N/A   N/A             N/A                      N/A",
+            ";-------------------------------------------------------------------------------",
+            ";   Message Number",
+            ";   |         Time Offset (ms)",
+            ";   |         |        Type",
+            ";   |         |        |        ID (hex)",
+            ";   |         |        |        |     Data Length",
+            ";   |         |        |        |     |   Data Bytes (hex) ...",
+            ";   |         |        |        |     |   |",
+            ";---+--   ----+----  --+--  ----+---  +  -+ -- -- -- -- -- -- --" ,
+        ]
+        self.file.writelines(line + "\n" for line in lines)
+
+
     def _write_header_v2_1(self, start_time: datetime) -> None:
         header_time = start_time - datetime(year=1899, month=12, day=30)
         lines = [
@@ -362,6 +392,9 @@ def _format_message_init(self, msg, channel):
         elif self.file_version == TRCFileVersion.V2_1:
             self._format_message = self._format_message_by_format
             self._msg_fmt_string = self.FORMAT_MESSAGE
+        elif self.file_version == TRCFileVersion.V1_1:
+            self._format_message = self._format_message_by_format
+            self._msg_fmt_string = self.FORMAT_MESSAGE_V1_1
         else:
             raise NotImplementedError("File format is not supported")
 
@@ -373,6 +406,8 @@ def write_header(self, timestamp: float) -> None:
 
         if self.file_version == TRCFileVersion.V1_0:
             self._write_header_v1_0(start_time)
+        elif self.file_version == TRCFileVersion.V1_1 :
+            self._write_header_v1_1(start_time)
         elif self.file_version == TRCFileVersion.V2_1:
             self._write_header_v2_1(start_time)
         else: