WIP: Add method to calculate load on a bus

can/bus.py

@@ -450,6 +450,43 @@ def _detect_available_configs() -> List[can.typechecking.AutoDetectedConfig]:
     def fileno(self) -> int:
         raise NotImplementedError("fileno is not implemented using current CAN bus")
 
+    def calc_load(self, interval: float = 1.0) -> int:
+        """
+        Calculate the load of the interface in the last 1s or specified interval of time
+
+        :param interval:
+            Seconds to capture packets
+
+        :return: a int representing the load of the interface in bits
+        """
+        start_time = time()
+        count = 0
+        temp_buffer = []
+        while time() < start_time + interval:
+            temp_buffer.append(self.recv(interval))
+        for msg in temp_buffer:
+            if msg is not None and isinstance(msg, can.Message):
+                count += 8 * len(msg.data)
+                if msg.is_extended_id:
+                    count += 29
+                else:
+                    count += 11
+        return count
+
+    def calc_load_percentage(self, bitrate, interval: float = 1.0) -> float:
+        """
+        Calculate the load of the interface in the last 1s or specified interval of time
+
+        :param bitrate:
+            The fixed bitrate of the interface
+
+        :param interval:
+            Seconds to capture packets
+
+        :return: a float representing the load of the interface as a percentage
+        """
+        return self.calc_load(interval) / bitrate
+
 
 class _SelfRemovingCyclicTask(CyclicSendTaskABC, ABC):
     """Removes itself from a bus.