tools/pyboard.py: Add fast raw-paste mode.

dpgeorge · dpgeorge · commit 33d7d9254978 · 2020-10-11T17:43:24.000+11:00
diff --git a/tools/pyboard.py b/tools/pyboard.py
@@ -370,6 +370,51 @@ def exec_raw_no_follow(self, command):
         if not data.endswith(b">"):
             raise PyboardError("could not enter raw repl")
 
+        t0 = time.time_ns()
+
+        if True:
+            # raw-paste mode
+            self.serial.write(b"\x05A\x01")
+            data = self.serial.read(2)
+            if data == b"R\x00":
+                # unsupported raw-paste mode
+                pass
+            elif data == b"R\x01":
+                # board supports raw-paste mode
+                data = self.serial.read(2)
+                window_size = data[0] | data[1] << 8
+                window_remain = window_size
+                i = 0
+                while i < len(command_bytes):
+                    while window_remain == 0 or self.serial.inWaiting():
+                        data = self.serial.read(1)
+                        if data == b"\x01":
+                            window_remain += window_size
+                        else:
+                            assert 0
+                    b = command_bytes[i : min(i + window_remain, len(command_bytes))]
+                    self.serial.write(b)
+                    window_remain -= len(b)
+                    i += len(b)
+                self.serial.write(b"\x04")
+
+                data = self.read_until(1, b"\x04")
+                if not data.endswith(b"\x04"):
+                    print(data)
+                    raise PyboardError("could not complete raw paste")
+
+                dt = (time.time_ns() - t0) // 1000000 + 1
+                print(
+                    "raw-paste took", dt, len(command_bytes), len(command_bytes) * 10 / (dt * 1e-3)
+                )
+                return
+            else:
+                # board doesn't support raw-paste
+                data = self.read_until(1, b"w REPL; CTRL-B to exit\r\n>")
+                if not data.endswith(b"w REPL; CTRL-B to exit\r\n>"):
+                    print(data)
+                    raise PyboardError("could not enter raw repl")
+
         # write command
         for i in range(0, len(command_bytes), 256):
             self.serial.write(command_bytes[i : min(i + 256, len(command_bytes))])
@@ -381,6 +426,9 @@ def exec_raw_no_follow(self, command):
         if data != b"OK":
             raise PyboardError("could not exec command (response: %r)" % data)
 
+        dt = (time.time_ns() - t0) // 1000000 + 1
+        print("raw took", dt, len(command_bytes), len(command_bytes) * 10 / (dt * 1e-3))
+
     def exec_raw(self, command, timeout=10, data_consumer=None):
         self.exec_raw_no_follow(command)
         return self.follow(timeout, data_consumer)
