12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- from UM.Job import Job
- from UM.Logger import Logger
- from .avr_isp import ispBase
- from .avr_isp.stk500v2 import Stk500v2
- from time import time, sleep
- from serial import Serial, SerialException
- class AutoDetectBaudJob(Job):
- def __init__(self, serial_port: int) -> None:
- super().__init__()
- self._serial_port = serial_port
- self._all_baud_rates = [115200, 250000, 500000, 230400, 76800, 57600, 38400, 19200, 9600]
- def run(self) -> None:
- Logger.debug(f"Auto detect baud rate started for {self._serial_port}")
- wait_response_timeouts = [3, 15, 30]
- wait_bootloader_times = [1.5, 5, 15]
- write_timeout = 3
- read_timeout = 3
- tries = 2
- programmer = Stk500v2()
- serial = None
- try:
- programmer.connect(self._serial_port)
- serial = programmer.leaveISP()
- except ispBase.IspError:
- programmer.close()
- for retry in range(tries):
- for baud_rate in self._all_baud_rates:
- if retry < len(wait_response_timeouts):
- wait_response_timeout = wait_response_timeouts[retry]
- else:
- wait_response_timeout = wait_response_timeouts[-1]
- if retry < len(wait_bootloader_times):
- wait_bootloader = wait_bootloader_times[retry]
- else:
- wait_bootloader = wait_bootloader_times[-1]
- Logger.debug(f"Checking {self._serial_port} if baud rate {baud_rate} works. Retry nr: {retry}. Wait timeout: {wait_response_timeout}")
- if serial is None:
- try:
- serial = Serial(str(self._serial_port), baud_rate, timeout = read_timeout, writeTimeout = write_timeout)
- except SerialException:
- Logger.warning(f"Unable to create serial connection to {serial} with baud rate {baud_rate}")
- continue
- else:
-
- try:
- serial.baudrate = baud_rate
- except ValueError:
- continue
-
- sleep(wait_bootloader)
- serial.write(b"\n")
- serial.write(b"M105\n")
- start_timeout_time = time()
- timeout_time = time() + wait_response_timeout
- while timeout_time > time():
-
-
-
- line = serial.read_until(size = 100)
- if b"ok" in line and b"T:" in line:
- self.setResult(baud_rate)
- Logger.log("d", "Detected baud rate {baud_rate} on serial {serial} on retry {retry} with after {time_elapsed:0.2f} seconds.".format(
- serial = self._serial_port, baud_rate = baud_rate, retry = retry, time_elapsed = time() - start_timeout_time))
- serial.close()
- return
- serial.write(b"M105\n")
- sleep(15)
- Logger.debug(f"Unable to find a working baudrate for {serial}")
- self.setResult(None)
|