bambu_mqtt.py 124 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700
  1. import json
  2. import ssl
  3. import asyncio
  4. import logging
  5. import time
  6. from collections import deque
  7. from datetime import datetime
  8. from typing import Callable
  9. from dataclasses import dataclass, field
  10. import paho.mqtt.client as mqtt
  11. logger = logging.getLogger(__name__)
  12. @dataclass
  13. class MQTTLogEntry:
  14. """Log entry for MQTT message debugging."""
  15. timestamp: str
  16. topic: str
  17. direction: str # "in" or "out"
  18. payload: dict
  19. @dataclass
  20. class HMSError:
  21. """Health Management System error from printer."""
  22. code: str
  23. attr: int # Attribute value for constructing wiki URL
  24. module: int
  25. severity: int # 1=fatal, 2=serious, 3=common, 4=info
  26. message: str = ""
  27. @dataclass
  28. class KProfile:
  29. """Pressure advance (K) calibration profile from printer."""
  30. slot_id: int
  31. extruder_id: int
  32. nozzle_id: str
  33. nozzle_diameter: str
  34. filament_id: str
  35. name: str
  36. k_value: str
  37. n_coef: str = "0.000000"
  38. ams_id: int = 0
  39. tray_id: int = -1
  40. setting_id: str | None = None
  41. @dataclass
  42. class NozzleInfo:
  43. """Nozzle hardware configuration."""
  44. nozzle_type: str = "" # "stainless_steel" or "hardened_steel"
  45. nozzle_diameter: str = "" # e.g., "0.4"
  46. @dataclass
  47. class PrintOptions:
  48. """AI detection and print options from xcam data."""
  49. # Core AI detectors
  50. spaghetti_detector: bool = False
  51. print_halt: bool = False
  52. halt_print_sensitivity: str = "medium" # Spaghetti sensitivity
  53. first_layer_inspector: bool = False
  54. printing_monitor: bool = False # AI print quality monitoring
  55. buildplate_marker_detector: bool = False
  56. allow_skip_parts: bool = False
  57. # Additional AI detectors - decoded from cfg bitmask
  58. nozzle_clumping_detector: bool = True
  59. nozzle_clumping_sensitivity: str = "medium"
  60. pileup_detector: bool = True
  61. pileup_sensitivity: str = "medium"
  62. airprint_detector: bool = True
  63. airprint_sensitivity: str = "medium"
  64. auto_recovery_step_loss: bool = True # Uses print.print_option command
  65. filament_tangle_detect: bool = False
  66. @dataclass
  67. class PrinterState:
  68. connected: bool = False
  69. state: str = "unknown"
  70. current_print: str | None = None
  71. subtask_name: str | None = None
  72. progress: float = 0.0
  73. remaining_time: int = 0
  74. layer_num: int = 0
  75. total_layers: int = 0
  76. temperatures: dict = field(default_factory=dict)
  77. raw_data: dict = field(default_factory=dict)
  78. gcode_file: str | None = None
  79. subtask_id: str | None = None
  80. hms_errors: list = field(default_factory=list) # List of HMSError
  81. kprofiles: list = field(default_factory=list) # List of KProfile
  82. sdcard: bool = False # SD card inserted
  83. store_to_sdcard: bool = False # Store sent files on SD card (home_flag bit 11)
  84. timelapse: bool = False # Timelapse recording active
  85. ipcam: bool = False # Live view / camera streaming enabled
  86. wifi_signal: int | None = None # WiFi signal strength in dBm
  87. # Nozzle hardware info (for dual nozzle printers, index 0 = left, 1 = right)
  88. nozzles: list = field(default_factory=lambda: [NozzleInfo(), NozzleInfo()])
  89. # AI detection and print options
  90. print_options: PrintOptions = field(default_factory=PrintOptions)
  91. # Calibration stage tracking (from stg_cur and stg fields)
  92. stg_cur: int = -1 # Current stage index (-1 = not calibrating)
  93. stg: list = field(default_factory=list) # List of stages to execute
  94. # Air conditioning mode (0=cooling, 1=heating)
  95. airduct_mode: int = 0
  96. # Print speed level (1=silent, 2=standard, 3=sport, 4=ludicrous)
  97. speed_level: int = 2
  98. # Chamber light on/off
  99. chamber_light: bool = False
  100. # Active extruder for dual nozzle (0=right, 1=left) - from device.extruder.info[X].hnow
  101. active_extruder: int = 0
  102. # Currently loaded tray (global ID): 254 = external spool, 255 = no filament
  103. tray_now: int = 255
  104. # Pending load target - used to track what tray we're loading for H2D disambiguation
  105. pending_tray_target: int | None = None
  106. # AMS status for filament change tracking (from print.ams.ams_status field)
  107. # ams_status is a combined value: lower 8 bits = sub status, bits 8-15 = main status
  108. # Main status: 0=idle, 1=filament_change, 2=rfid_identifying, 3=assist, 4=calibration, etc.
  109. ams_status: int = 0
  110. ams_status_main: int = 0 # (ams_status >> 8) & 0xFF
  111. ams_status_sub: int = 0 # ams_status & 0xFF
  112. # mc_print_sub_stage - filament change step indicator from print.mc_print_sub_stage
  113. # Used by OrcaSlicer/BambuStudio to track progress during filament load/unload
  114. mc_print_sub_stage: int = 0
  115. # AMS mapping for dual nozzle: which slot is active (from ams.ams_exist_bits/tray_exist_bits)
  116. ams_mapping: list = field(default_factory=list)
  117. # Per-AMS extruder map: {ams_id: extruder_id} where 0=right, 1=left
  118. ams_extruder_map: dict = field(default_factory=dict)
  119. # Timestamp of last AMS data update (for RFID refresh detection)
  120. last_ams_update: float = 0.0
  121. # Stage name mapping from BambuStudio DeviceManager.cpp
  122. STAGE_NAMES = {
  123. 0: "Printing",
  124. 1: "Auto bed leveling",
  125. 2: "Heatbed preheating",
  126. 3: "Vibration compensation",
  127. 4: "Changing filament",
  128. 5: "M400 pause",
  129. 6: "Paused (filament ran out)",
  130. 7: "Heating nozzle",
  131. 8: "Calibrating dynamic flow",
  132. 9: "Scanning bed surface",
  133. 10: "Inspecting first layer",
  134. 11: "Identifying build plate type",
  135. 12: "Calibrating Micro Lidar",
  136. 13: "Homing toolhead",
  137. 14: "Cleaning nozzle tip",
  138. 15: "Checking extruder temperature",
  139. 16: "Paused by the user",
  140. 17: "Pause (front cover fall off)",
  141. 18: "Calibrating the micro lidar",
  142. 19: "Calibrating flow ratio",
  143. 20: "Pause (nozzle temperature malfunction)",
  144. 21: "Pause (heatbed temperature malfunction)",
  145. 22: "Filament unloading",
  146. 23: "Pause (step loss)",
  147. 24: "Filament loading",
  148. 25: "Motor noise cancellation",
  149. 26: "Pause (AMS offline)",
  150. 27: "Pause (low speed of the heatbreak fan)",
  151. 28: "Pause (chamber temperature control problem)",
  152. 29: "Cooling chamber",
  153. 30: "Pause (Gcode inserted by user)",
  154. 31: "Motor noise showoff",
  155. 32: "Pause (nozzle clumping)",
  156. 33: "Pause (cutter error)",
  157. 34: "Pause (first layer error)",
  158. 35: "Pause (nozzle clog)",
  159. 36: "Measuring motion precision",
  160. 37: "Enhancing motion precision",
  161. 38: "Measure motion accuracy",
  162. 39: "Nozzle offset calibration",
  163. 40: "High temperature auto bed leveling",
  164. 41: "Auto Check: Quick Release Lever",
  165. 42: "Auto Check: Door and Upper Cover",
  166. 43: "Laser Calibration",
  167. 44: "Auto Check: Platform",
  168. 45: "Confirming BirdsEye Camera location",
  169. 46: "Calibrating BirdsEye Camera",
  170. 47: "Auto bed leveling - phase 1",
  171. 48: "Auto bed leveling - phase 2",
  172. 49: "Heating chamber",
  173. 50: "Cooling heatbed",
  174. 51: "Printing calibration lines",
  175. 52: "Auto Check: Material",
  176. 53: "Live View Camera Calibration",
  177. 54: "Waiting for heatbed temperature",
  178. 55: "Auto Check: Material Position",
  179. 56: "Cutting Module Offset Calibration",
  180. 57: "Measuring Surface",
  181. 58: "Thermal Preconditioning",
  182. 59: "Homing Blade Holder",
  183. 60: "Calibrating Camera Offset",
  184. 61: "Calibrating Blade Holder Position",
  185. 62: "Hotend Pick and Place Test",
  186. 63: "Waiting for Chamber temperature",
  187. 64: "Preparing Hotend",
  188. 65: "Calibrating nozzle clumping detection",
  189. 66: "Purifying the chamber air",
  190. }
  191. def get_stage_name(stage: int) -> str:
  192. """Get human-readable stage name from stage number."""
  193. return STAGE_NAMES.get(stage, f"Unknown stage ({stage})")
  194. class BambuMQTTClient:
  195. """MQTT client for Bambu Lab printer communication."""
  196. MQTT_PORT = 8883
  197. def __init__(
  198. self,
  199. ip_address: str,
  200. serial_number: str,
  201. access_code: str,
  202. on_state_change: Callable[[PrinterState], None] | None = None,
  203. on_print_start: Callable[[dict], None] | None = None,
  204. on_print_complete: Callable[[dict], None] | None = None,
  205. on_ams_change: Callable[[list], None] | None = None,
  206. ):
  207. self.ip_address = ip_address
  208. self.serial_number = serial_number
  209. self.access_code = access_code
  210. self.on_state_change = on_state_change
  211. self.on_print_start = on_print_start
  212. self.on_print_complete = on_print_complete
  213. self.on_ams_change = on_ams_change
  214. self.state = PrinterState()
  215. self._client: mqtt.Client | None = None
  216. self._loop: asyncio.AbstractEventLoop | None = None
  217. self._previous_gcode_state: str | None = None
  218. self._previous_gcode_file: str | None = None
  219. self._was_running: bool = False # Track if we've seen RUNNING state for current print
  220. self._completion_triggered: bool = False # Prevent duplicate completion triggers
  221. self._message_log: deque[MQTTLogEntry] = deque(maxlen=100)
  222. self._logging_enabled: bool = False
  223. self._last_message_time: float = 0.0 # Track when we last received a message
  224. self._previous_ams_hash: str | None = None # Track AMS changes
  225. # K-profile command tracking
  226. self._sequence_id: int = 0
  227. self._pending_kprofile_response: asyncio.Event | None = None
  228. self._kprofile_response_data: list | None = None
  229. # Xcam hold timers - OrcaSlicer pattern: ignore incoming data for 3 seconds after command
  230. # Key: module_name, Value: timestamp when command was sent
  231. self._xcam_hold_start: dict[str, float] = {}
  232. self._xcam_hold_time: float = 3.0 # Ignore incoming data for 3 seconds after command
  233. # Track last requested tray ID for H2D dual-nozzle printers
  234. # H2D only reports slot number (0-3) in tray_now, not global tray ID
  235. # We use our tracked value to resolve the correct global ID
  236. self._last_load_tray_id: int | None = None
  237. @property
  238. def topic_subscribe(self) -> str:
  239. return f"device/{self.serial_number}/report"
  240. @property
  241. def topic_publish(self) -> str:
  242. return f"device/{self.serial_number}/request"
  243. def _on_connect(self, client, userdata, flags, rc, properties=None):
  244. if rc == 0:
  245. self.state.connected = True
  246. client.subscribe(self.topic_subscribe)
  247. # Request full status update (includes nozzle info in push_status response)
  248. self._request_push_all()
  249. # Note: get_accessories returns stale nozzle data on H2D, so we don't use it.
  250. # The correct nozzle data comes from push_status.
  251. # Prime K-profile request (Bambu printers often ignore first request)
  252. self._prime_kprofile_request()
  253. # Immediately broadcast connection state change
  254. if self.on_state_change:
  255. self.on_state_change(self.state)
  256. else:
  257. self.state.connected = False
  258. def _on_disconnect(self, client, userdata, disconnect_flags=None, rc=None, properties=None):
  259. # Ignore spurious disconnect callbacks if we've received a message recently
  260. # Paho-mqtt sometimes fires disconnect callbacks while the connection is still active
  261. time_since_last_message = time.time() - self._last_message_time
  262. if time_since_last_message < 30.0 and self._last_message_time > 0:
  263. logger.debug(
  264. f"[{self.serial_number}] Ignoring spurious disconnect (last message {time_since_last_message:.1f}s ago)"
  265. )
  266. return
  267. logger.warning(f"[{self.serial_number}] MQTT disconnected: rc={rc}, flags={disconnect_flags}")
  268. self.state.connected = False
  269. if self.on_state_change:
  270. self.on_state_change(self.state)
  271. def _on_message(self, client, userdata, msg):
  272. try:
  273. payload = json.loads(msg.payload.decode())
  274. # Track last message time - receiving a message proves we're connected
  275. self._last_message_time = time.time()
  276. self.state.connected = True
  277. # TEMP: Dump full payload once to find extruder state field
  278. if not hasattr(self, '_payload_dumped'):
  279. self._payload_dumped = True
  280. logger.info(f"[{self.serial_number}] FULL MQTT PAYLOAD DUMP:\n{json.dumps(payload, indent=2)}")
  281. # Log message if logging is enabled
  282. if self._logging_enabled:
  283. self._message_log.append(MQTTLogEntry(
  284. timestamp=datetime.now().isoformat(),
  285. topic=msg.topic,
  286. direction="in",
  287. payload=payload,
  288. ))
  289. self._process_message(payload)
  290. except json.JSONDecodeError:
  291. pass
  292. def _process_message(self, payload: dict):
  293. """Process incoming MQTT message from printer."""
  294. # Handle top-level AMS data (comes outside of "print" key)
  295. # Wrap in try/except to prevent breaking the MQTT connection
  296. if "ams" in payload:
  297. try:
  298. self._handle_ams_data(payload["ams"])
  299. except Exception as e:
  300. logger.error(f"[{self.serial_number}] Error handling AMS data: {e}")
  301. # Handle xcam data (camera settings and AI detection) at top level
  302. if "xcam" in payload:
  303. xcam_data = payload["xcam"]
  304. logger.info(f"[{self.serial_number}] Received xcam data at top level: {xcam_data}")
  305. self._parse_xcam_data(xcam_data)
  306. # Fire state change callback for top-level xcam (not nested in "print")
  307. if "print" not in payload and self.on_state_change:
  308. self.on_state_change(self.state)
  309. # Handle system responses (accessories info, etc.)
  310. if "system" in payload:
  311. system_data = payload["system"]
  312. logger.info(f"[{self.serial_number}] Received system data: {system_data}")
  313. self._handle_system_response(system_data)
  314. # Parse WiFi signal at top level (some printers send it here)
  315. if "wifi_signal" in payload:
  316. wifi_signal = payload["wifi_signal"]
  317. if isinstance(wifi_signal, (int, float)):
  318. self.state.wifi_signal = int(wifi_signal)
  319. elif isinstance(wifi_signal, str):
  320. try:
  321. self.state.wifi_signal = int(wifi_signal.replace("dBm", "").strip())
  322. except ValueError:
  323. pass
  324. if "print" in payload:
  325. print_data = payload["print"]
  326. # Check if xcam is nested inside print data
  327. if "xcam" in print_data:
  328. logger.info(f"[{self.serial_number}] Found xcam inside print data: {print_data['xcam']}")
  329. self._parse_xcam_data(print_data["xcam"])
  330. # Log when we see gcode_state changes
  331. if "gcode_state" in print_data:
  332. logger.info(
  333. f"[{self.serial_number}] Received gcode_state: {print_data.get('gcode_state')}, "
  334. f"gcode_file: {print_data.get('gcode_file')}, subtask_name: {print_data.get('subtask_name')}"
  335. )
  336. # Handle AMS data that comes inside print key
  337. if "ams" in print_data:
  338. try:
  339. self._handle_ams_data(print_data["ams"])
  340. except Exception as e:
  341. logger.error(f"[{self.serial_number}] Error handling AMS data from print: {e}")
  342. # Handle vt_tray (virtual tray / external spool) data
  343. if "vt_tray" in print_data:
  344. vt_tray = print_data["vt_tray"]
  345. self.state.raw_data["vt_tray"] = vt_tray
  346. # Log vt_tray to investigate per-extruder data for H2D
  347. if not hasattr(self, '_vt_tray_logged') or not self._vt_tray_logged:
  348. logger.info(f"[{self.serial_number}] vt_tray data: {vt_tray}")
  349. self._vt_tray_logged = True
  350. # Parse ams_status directly from print data (NOT from print.ams)
  351. # ams_status is a combined value: lower 8 bits = sub status, bits 8-15 = main status
  352. # Main status: 0=idle, 1=filament_change, 2=rfid_identifying, 3=assist, 4=calibration
  353. # Sub status (when main=1): 2=heating, 3=AMS feeding, 4=retract, 6=push, 7=purge
  354. if "ams_status" in print_data:
  355. raw_ams_status = print_data["ams_status"]
  356. if isinstance(raw_ams_status, str):
  357. try:
  358. self.state.ams_status = int(raw_ams_status)
  359. except ValueError:
  360. self.state.ams_status = 0
  361. else:
  362. self.state.ams_status = raw_ams_status if raw_ams_status is not None else 0
  363. # Compute main and sub status
  364. self.state.ams_status_sub = self.state.ams_status & 0xFF
  365. self.state.ams_status_main = (self.state.ams_status >> 8) & 0xFF
  366. # Log when ams_status changes (for filament change tracking debug)
  367. logger.debug(
  368. f"[{self.serial_number}] ams_status: {self.state.ams_status} "
  369. f"(main={self.state.ams_status_main}, sub={self.state.ams_status_sub})"
  370. )
  371. # Check for K-profile response (extrusion_cali)
  372. if "command" in print_data:
  373. logger.debug(f"[{self.serial_number}] Received command response: {print_data.get('command')}")
  374. if "command" in print_data and print_data.get("command") == "extrusion_cali_get":
  375. self._handle_kprofile_response(print_data)
  376. self._update_state(print_data)
  377. def _handle_system_response(self, data: dict):
  378. """Handle system responses including accessories info.
  379. Note: get_accessories returns stale/incorrect nozzle_type data on H2D.
  380. The correct nozzle data comes from push_status, so we don't update
  381. nozzle type/diameter from get_accessories. We just log the response
  382. for debugging purposes.
  383. """
  384. command = data.get("command")
  385. if command == "get_accessories":
  386. # Log response for debugging - but DON'T use it to update nozzle data
  387. # because it returns stale values (e.g., 'stainless_steel' when the
  388. # actual nozzle is 'HH01' hardened steel high-flow)
  389. logger.info(f"[{self.serial_number}] Accessories response (not used for nozzle data): {data}")
  390. def _parse_xcam_data(self, xcam_data):
  391. """Parse xcam data for camera settings and AI detection options."""
  392. if not isinstance(xcam_data, dict):
  393. return
  394. current_time = time.time()
  395. # Helper to check if we should accept incoming value for a module
  396. # OrcaSlicer pattern: simple hold timer, ignore ALL data for 3 seconds after command
  397. def should_accept_value(module_name: str, incoming_value: bool) -> bool:
  398. """Check if we should accept an incoming xcam value.
  399. OrcaSlicer pattern: After sending a command, ignore incoming data
  400. for 3 seconds. After that, accept whatever the printer sends.
  401. """
  402. if module_name not in self._xcam_hold_start:
  403. return True # No hold timer, accept incoming
  404. hold_start = self._xcam_hold_start[module_name]
  405. elapsed = current_time - hold_start
  406. if elapsed > self._xcam_hold_time:
  407. # Hold timer expired - accept incoming and clear hold
  408. del self._xcam_hold_start[module_name]
  409. logger.debug(
  410. f"[{self.serial_number}] Hold expired for {module_name}, accepting {incoming_value}"
  411. )
  412. return True
  413. # Within hold period - ignore incoming data
  414. logger.debug(
  415. f"[{self.serial_number}] Ignoring {module_name}={incoming_value} "
  416. f"(hold active, {elapsed:.1f}s < {self._xcam_hold_time}s)"
  417. )
  418. return False
  419. # Log all xcam fields for debugging
  420. logger.debug(f"[{self.serial_number}] Parsing xcam data - all fields: {list(xcam_data.keys())}")
  421. # The cfg bitmask contains the ACTUAL detector states - the individual boolean
  422. # fields (spaghetti_detector, etc.) are often stale/cached.
  423. # CFG bitmask structure (each detector uses 3 bits: [sens_low, sens_high, enabled]):
  424. # - Bits 5-7: spaghetti_detector (sens in 5-6, enabled in 7)
  425. # - Bits 8-10: pileup_detector (sens in 8-9, enabled in 10)
  426. # - Bits 11-13: clump_detector/nozzle_clumping (sens in 11-12, enabled in 13)
  427. # - Bits 14-16: airprint_detector (sens in 14-15, enabled in 16)
  428. # Sensitivity values: 0=low, 1=medium, 2=high
  429. if "cfg" in xcam_data:
  430. cfg = xcam_data["cfg"]
  431. logger.debug(f"[{self.serial_number}] xcam cfg bitmask: {cfg} (binary: {bin(cfg)})")
  432. def decode_detector(start_bit):
  433. """Decode a detector from cfg: returns (enabled, sensitivity_str)"""
  434. sens_bits = (cfg >> start_bit) & 0x3
  435. enabled = bool((cfg >> (start_bit + 2)) & 1)
  436. sensitivity = {0: "low", 1: "medium", 2: "high"}.get(sens_bits, "medium")
  437. return enabled, sensitivity
  438. # Spaghetti detector (bits 5-7)
  439. cfg_spaghetti, cfg_sensitivity = decode_detector(5)
  440. if should_accept_value("spaghetti_detector", cfg_spaghetti):
  441. old_value = self.state.print_options.spaghetti_detector
  442. if cfg_spaghetti != old_value:
  443. logger.info(f"[{self.serial_number}] spaghetti_detector changed (from cfg): {old_value} -> {cfg_spaghetti}")
  444. self.state.print_options.spaghetti_detector = cfg_spaghetti
  445. # Check hold timer for sensitivity before accepting
  446. if "halt_print_sensitivity" not in self._xcam_hold_start:
  447. if cfg_sensitivity != self.state.print_options.halt_print_sensitivity:
  448. logger.info(
  449. f"[{self.serial_number}] Sensitivity changed (from cfg): "
  450. f"{self.state.print_options.halt_print_sensitivity} -> {cfg_sensitivity}"
  451. )
  452. self.state.print_options.halt_print_sensitivity = cfg_sensitivity
  453. else:
  454. hold_start = self._xcam_hold_start["halt_print_sensitivity"]
  455. elapsed = current_time - hold_start
  456. if elapsed <= self._xcam_hold_time:
  457. logger.debug(
  458. f"[{self.serial_number}] Ignoring cfg sensitivity={cfg_sensitivity} "
  459. f"(hold active, {elapsed:.1f}s < {self._xcam_hold_time}s)"
  460. )
  461. else:
  462. # Hold expired - accept from cfg
  463. if cfg_sensitivity != self.state.print_options.halt_print_sensitivity:
  464. logger.info(
  465. f"[{self.serial_number}] Sensitivity synced (from cfg after hold): "
  466. f"{self.state.print_options.halt_print_sensitivity} -> {cfg_sensitivity}"
  467. )
  468. self.state.print_options.halt_print_sensitivity = cfg_sensitivity
  469. del self._xcam_hold_start["halt_print_sensitivity"]
  470. # Pileup detector (bits 8-10)
  471. cfg_pileup, cfg_pileup_sens = decode_detector(8)
  472. if should_accept_value("pileup_detector", cfg_pileup):
  473. if cfg_pileup != self.state.print_options.pileup_detector:
  474. logger.info(f"[{self.serial_number}] pileup_detector changed (from cfg): {self.state.print_options.pileup_detector} -> {cfg_pileup}")
  475. self.state.print_options.pileup_detector = cfg_pileup
  476. # Pileup sensitivity with hold timer
  477. if "pileup_sensitivity" not in self._xcam_hold_start:
  478. if cfg_pileup_sens != self.state.print_options.pileup_sensitivity:
  479. logger.info(f"[{self.serial_number}] pileup_sensitivity changed (from cfg): {self.state.print_options.pileup_sensitivity} -> {cfg_pileup_sens}")
  480. self.state.print_options.pileup_sensitivity = cfg_pileup_sens
  481. else:
  482. hold_start = self._xcam_hold_start["pileup_sensitivity"]
  483. elapsed = current_time - hold_start
  484. if elapsed > self._xcam_hold_time:
  485. if cfg_pileup_sens != self.state.print_options.pileup_sensitivity:
  486. logger.info(f"[{self.serial_number}] pileup_sensitivity synced (from cfg after hold): {self.state.print_options.pileup_sensitivity} -> {cfg_pileup_sens}")
  487. self.state.print_options.pileup_sensitivity = cfg_pileup_sens
  488. del self._xcam_hold_start["pileup_sensitivity"]
  489. # Clump/nozzle clumping detector (bits 11-13)
  490. cfg_clump, cfg_clump_sens = decode_detector(11)
  491. if should_accept_value("clump_detector", cfg_clump):
  492. if cfg_clump != self.state.print_options.nozzle_clumping_detector:
  493. logger.info(f"[{self.serial_number}] nozzle_clumping_detector changed (from cfg): {self.state.print_options.nozzle_clumping_detector} -> {cfg_clump}")
  494. self.state.print_options.nozzle_clumping_detector = cfg_clump
  495. # Clump sensitivity with hold timer
  496. if "nozzle_clumping_sensitivity" not in self._xcam_hold_start:
  497. if cfg_clump_sens != self.state.print_options.nozzle_clumping_sensitivity:
  498. logger.info(f"[{self.serial_number}] nozzle_clumping_sensitivity changed (from cfg): {self.state.print_options.nozzle_clumping_sensitivity} -> {cfg_clump_sens}")
  499. self.state.print_options.nozzle_clumping_sensitivity = cfg_clump_sens
  500. else:
  501. hold_start = self._xcam_hold_start["nozzle_clumping_sensitivity"]
  502. elapsed = current_time - hold_start
  503. if elapsed > self._xcam_hold_time:
  504. if cfg_clump_sens != self.state.print_options.nozzle_clumping_sensitivity:
  505. logger.info(f"[{self.serial_number}] nozzle_clumping_sensitivity synced (from cfg after hold): {self.state.print_options.nozzle_clumping_sensitivity} -> {cfg_clump_sens}")
  506. self.state.print_options.nozzle_clumping_sensitivity = cfg_clump_sens
  507. del self._xcam_hold_start["nozzle_clumping_sensitivity"]
  508. # Airprint detector (bits 14-16)
  509. cfg_airprint, cfg_airprint_sens = decode_detector(14)
  510. if should_accept_value("airprint_detector", cfg_airprint):
  511. if cfg_airprint != self.state.print_options.airprint_detector:
  512. logger.info(f"[{self.serial_number}] airprint_detector changed (from cfg): {self.state.print_options.airprint_detector} -> {cfg_airprint}")
  513. self.state.print_options.airprint_detector = cfg_airprint
  514. # Airprint sensitivity with hold timer
  515. if "airprint_sensitivity" not in self._xcam_hold_start:
  516. if cfg_airprint_sens != self.state.print_options.airprint_sensitivity:
  517. logger.info(f"[{self.serial_number}] airprint_sensitivity changed (from cfg): {self.state.print_options.airprint_sensitivity} -> {cfg_airprint_sens}")
  518. self.state.print_options.airprint_sensitivity = cfg_airprint_sens
  519. else:
  520. hold_start = self._xcam_hold_start["airprint_sensitivity"]
  521. elapsed = current_time - hold_start
  522. if elapsed > self._xcam_hold_time:
  523. if cfg_airprint_sens != self.state.print_options.airprint_sensitivity:
  524. logger.info(f"[{self.serial_number}] airprint_sensitivity synced (from cfg after hold): {self.state.print_options.airprint_sensitivity} -> {cfg_airprint_sens}")
  525. self.state.print_options.airprint_sensitivity = cfg_airprint_sens
  526. del self._xcam_hold_start["airprint_sensitivity"]
  527. # Camera settings
  528. if "ipcam_record" in xcam_data:
  529. self.state.ipcam = xcam_data.get("ipcam_record") == "enable"
  530. if "timelapse" in xcam_data:
  531. self.state.timelapse = xcam_data.get("timelapse") == "enable"
  532. # Skip spaghetti_detector boolean field - we read from cfg bitmask above
  533. if "print_halt" in xcam_data:
  534. self.state.print_options.print_halt = bool(xcam_data.get("print_halt"))
  535. # Skip halt_print_sensitivity field - it's always stale ("medium")
  536. # We read the actual sensitivity from cfg bits 5-6 above
  537. if "first_layer_inspector" in xcam_data:
  538. new_value = bool(xcam_data.get("first_layer_inspector"))
  539. if should_accept_value("first_layer_inspector", new_value):
  540. self.state.print_options.first_layer_inspector = new_value
  541. if "printing_monitor" in xcam_data:
  542. new_value = bool(xcam_data.get("printing_monitor"))
  543. if should_accept_value("printing_monitor", new_value):
  544. self.state.print_options.printing_monitor = new_value
  545. if "buildplate_marker_detector" in xcam_data:
  546. new_value = bool(xcam_data.get("buildplate_marker_detector"))
  547. if should_accept_value("buildplate_marker_detector", new_value):
  548. self.state.print_options.buildplate_marker_detector = new_value
  549. if "allow_skip_parts" in xcam_data:
  550. new_value = bool(xcam_data.get("allow_skip_parts"))
  551. if should_accept_value("allow_skip_parts", new_value):
  552. self.state.print_options.allow_skip_parts = new_value
  553. # Additional AI detectors - these are decoded from cfg bitmask above, not from
  554. # individual boolean fields (which are not sent by the printer)
  555. # pileup_detector, nozzle_clumping_detector, airprint_detector - from cfg
  556. # auto_recovery_step_loss and filament_tangle_detect - tracked locally only
  557. if "auto_recovery_step_loss" in xcam_data:
  558. self.state.print_options.auto_recovery_step_loss = bool(xcam_data.get("auto_recovery_step_loss"))
  559. if "filament_tangle_detect" in xcam_data:
  560. self.state.print_options.filament_tangle_detect = bool(xcam_data.get("filament_tangle_detect"))
  561. def _handle_ams_data(self, ams_data):
  562. """Handle AMS data changes for Spoolman integration.
  563. This is called when we receive top-level AMS data in MQTT messages.
  564. It detects changes and triggers the callback for Spoolman sync.
  565. """
  566. import hashlib
  567. # Handle nested ams structure: {"ams": {"ams": [...]}} or {"ams": [...]}
  568. if isinstance(ams_data, dict) and "ams" in ams_data:
  569. ams_list = ams_data["ams"]
  570. # Log all AMS dict fields to debug tray_now for H2D dual-nozzle
  571. non_list_fields = {k: v for k, v in ams_data.items() if k != "ams"}
  572. if non_list_fields:
  573. logger.info(f"[{self.serial_number}] AMS dict fields: {non_list_fields}")
  574. # IMPORTANT: Parse ams_status FIRST before tray_now, so we have fresh status
  575. # when checking if we're in filament change mode for tray_now disambiguation
  576. if "ams_status" in ams_data:
  577. raw_ams_status = ams_data["ams_status"]
  578. if isinstance(raw_ams_status, str):
  579. try:
  580. self.state.ams_status = int(raw_ams_status)
  581. except ValueError:
  582. self.state.ams_status = 0
  583. else:
  584. self.state.ams_status = raw_ams_status if raw_ams_status is not None else 0
  585. # Compute main and sub status
  586. self.state.ams_status_sub = self.state.ams_status & 0xFF
  587. self.state.ams_status_main = (self.state.ams_status >> 8) & 0xFF
  588. logger.debug(
  589. f"[{self.serial_number}] ams_status: {self.state.ams_status} "
  590. f"(main={self.state.ams_status_main}, sub={self.state.ams_status_sub})"
  591. )
  592. # Parse tray_now from AMS dict - this is the currently loaded tray global ID
  593. if "tray_now" in ams_data:
  594. raw_tray_now = ams_data["tray_now"]
  595. # Convert string to int if needed
  596. if isinstance(raw_tray_now, str):
  597. try:
  598. parsed_tray_now = int(raw_tray_now)
  599. except ValueError:
  600. parsed_tray_now = 255
  601. else:
  602. parsed_tray_now = raw_tray_now if raw_tray_now is not None else 255
  603. # H2D dual-nozzle printers report only slot number (0-3), not global tray ID
  604. # Use pending_tray_target from our load command tracking for disambiguation
  605. if parsed_tray_now >= 0 and parsed_tray_now <= 3:
  606. # Check if we have a pending target that matches this slot
  607. pending_target = self.state.pending_tray_target
  608. if pending_target is not None:
  609. pending_slot = pending_target % 4
  610. if pending_slot == parsed_tray_now:
  611. # Slot matches our pending target - use the full global ID
  612. logger.info(
  613. f"[{self.serial_number}] H2D tray_now disambiguation: "
  614. f"slot {parsed_tray_now} matches pending_tray_target {pending_target} -> using global ID {pending_target}"
  615. )
  616. self.state.tray_now = pending_target
  617. # Clear pending target now that load is confirmed
  618. self.state.pending_tray_target = None
  619. else:
  620. # Slot doesn't match our pending target - something changed, use slot as-is
  621. logger.warning(
  622. f"[{self.serial_number}] H2D tray_now: slot {parsed_tray_now} doesn't match "
  623. f"pending_tray_target {pending_target} (slot {pending_slot}) - using slot as global ID"
  624. )
  625. self.state.tray_now = parsed_tray_now
  626. # Clear pending target since it's stale
  627. self.state.pending_tray_target = None
  628. else:
  629. # No pending target - check if we already have a resolved global ID
  630. # that matches this slot (from a previous successful disambiguation)
  631. current_tray = self.state.tray_now
  632. if current_tray > 3 and current_tray != 255 and (current_tray % 4) == parsed_tray_now:
  633. # Current tray_now is already a valid global ID that matches this slot
  634. # Keep it (don't overwrite with raw slot number)
  635. logger.debug(
  636. f"[{self.serial_number}] H2D tray_now: keeping existing global ID {current_tray} "
  637. f"(matches incoming slot {parsed_tray_now})"
  638. )
  639. else:
  640. # No pending target and no valid existing global ID
  641. # For H2D with multiple AMS units, we can't reliably determine which AMS
  642. # the slot belongs to without a pending_tray_target from our load command.
  643. # Use slot number as-is - this may be incorrect for multi-AMS setups,
  644. # but it's better than guessing wrong based on unreliable heuristics.
  645. # The user can load filament via our API to get correct tracking.
  646. logger.warning(
  647. f"[{self.serial_number}] H2D tray_now: no pending target, "
  648. f"using slot {parsed_tray_now} as global ID (may be incorrect for multi-AMS)"
  649. )
  650. self.state.tray_now = parsed_tray_now
  651. else:
  652. # tray_now > 3 means it's already a global ID, or 255 means unloaded
  653. # Note: Do NOT clear pending_tray_target on tray_now=255 here.
  654. # During filament change, the printer sends 255 first (unload), then the slot.
  655. # We only clear pending_tray_target explicitly in ams_unload_filament().
  656. # Trust the printer's reported value.
  657. self.state.tray_now = parsed_tray_now
  658. logger.debug(f"[{self.serial_number}] tray_now updated: {self.state.tray_now}")
  659. # NOTE: ams_status is parsed BEFORE tray_now (see above) to ensure correct
  660. # state when checking filament change mode for H2D disambiguation
  661. elif isinstance(ams_data, list):
  662. ams_list = ams_data
  663. else:
  664. logger.warning(f"[{self.serial_number}] Unexpected AMS data format: {type(ams_data)}")
  665. return
  666. # Store AMS data in raw_data so it's accessible via API
  667. self.state.raw_data["ams"] = ams_list
  668. # Update timestamp for RFID refresh detection (frontend can detect "new data arrived")
  669. self.state.last_ams_update = time.time()
  670. logger.debug(f"[{self.serial_number}] Stored AMS data with {len(ams_list)} units")
  671. # Extract ams_extruder_map from each AMS unit's info field
  672. # According to OpenBambuAPI: info field bit 8 indicates which extruder (0=right, 1=left)
  673. # Log AMS unit fields once to discover available fields
  674. if not hasattr(self, '_ams_fields_logged') and ams_list:
  675. first_unit = ams_list[0]
  676. logger.info(f"[{self.serial_number}] AMS unit fields: {sorted(first_unit.keys())}")
  677. for ams_unit in ams_list:
  678. ams_id = ams_unit.get("id")
  679. unit_info = {k: v for k, v in ams_unit.items() if k != "tray"}
  680. logger.info(f"[{self.serial_number}] AMS {ams_id} data: {unit_info}")
  681. self._ams_fields_logged = True
  682. ams_extruder_map = {}
  683. for ams_unit in ams_list:
  684. ams_id = ams_unit.get("id")
  685. info = ams_unit.get("info")
  686. if ams_id is not None and info is not None:
  687. try:
  688. info_val = int(info) if isinstance(info, str) else info
  689. # Extract bit 8 for extruder assignment
  690. # Bit 8 = 0 means LEFT extruder (id 1), bit 8 = 1 means RIGHT extruder (id 0)
  691. # So we invert: extruder_id = 1 - bit8
  692. bit8 = (info_val >> 8) & 0x1
  693. extruder_id = 1 - bit8 # 0=right, 1=left
  694. ams_extruder_map[str(ams_id)] = extruder_id
  695. logger.debug(f"[{self.serial_number}] AMS {ams_id} info={info_val} (bit8={bit8}) -> extruder {extruder_id}")
  696. except (ValueError, TypeError):
  697. pass
  698. if ams_extruder_map:
  699. self.state.raw_data["ams_extruder_map"] = ams_extruder_map
  700. self.state.ams_extruder_map = ams_extruder_map # Also set on state for inference logic
  701. logger.debug(f"[{self.serial_number}] ams_extruder_map: {ams_extruder_map}")
  702. # Create a hash of relevant AMS data to detect changes
  703. ams_hash_data = []
  704. for ams_unit in ams_list:
  705. for tray in ams_unit.get("tray", []):
  706. # Include fields that matter for filament tracking
  707. ams_hash_data.append(
  708. f"{ams_unit.get('id')}:{tray.get('id')}:"
  709. f"{tray.get('tray_type')}:{tray.get('tag_uid')}:{tray.get('remain')}"
  710. )
  711. ams_hash = hashlib.md5(":".join(ams_hash_data).encode()).hexdigest()
  712. # Only trigger callback if AMS data actually changed
  713. if ams_hash != self._previous_ams_hash:
  714. self._previous_ams_hash = ams_hash
  715. if self.on_ams_change:
  716. logger.info(f"[{self.serial_number}] AMS data changed, triggering sync callback")
  717. self.on_ams_change(ams_list)
  718. def _update_state(self, data: dict):
  719. """Update printer state from message data."""
  720. previous_state = self.state.state
  721. # Update state fields
  722. if "gcode_state" in data:
  723. self.state.state = data["gcode_state"]
  724. if "gcode_file" in data:
  725. self.state.gcode_file = data["gcode_file"]
  726. self.state.current_print = data["gcode_file"]
  727. if "subtask_name" in data:
  728. self.state.subtask_name = data["subtask_name"]
  729. # Prefer subtask_name as current_print if available
  730. if data["subtask_name"]:
  731. self.state.current_print = data["subtask_name"]
  732. if "subtask_id" in data:
  733. self.state.subtask_id = data["subtask_id"]
  734. if "mc_percent" in data:
  735. self.state.progress = float(data["mc_percent"])
  736. if "mc_remaining_time" in data:
  737. self.state.remaining_time = int(data["mc_remaining_time"])
  738. if "mc_print_sub_stage" in data:
  739. new_sub_stage = int(data["mc_print_sub_stage"])
  740. if new_sub_stage != self.state.mc_print_sub_stage:
  741. logger.debug(
  742. f"[{self.serial_number}] mc_print_sub_stage changed: "
  743. f"{self.state.mc_print_sub_stage} -> {new_sub_stage}"
  744. )
  745. self.state.mc_print_sub_stage = new_sub_stage
  746. if "layer_num" in data:
  747. self.state.layer_num = int(data["layer_num"])
  748. if "total_layer_num" in data:
  749. self.state.total_layers = int(data["total_layer_num"])
  750. # Calibration stage tracking
  751. if "stg_cur" in data:
  752. new_stg = data["stg_cur"]
  753. # Always log ANY stg_cur change for debugging filament operations
  754. if new_stg != self.state.stg_cur:
  755. logger.info(
  756. f"[{self.serial_number}] stg_cur changed: {self.state.stg_cur} -> {new_stg} ({get_stage_name(new_stg)})"
  757. )
  758. self.state.stg_cur = new_stg
  759. if "stg" in data:
  760. self.state.stg = data["stg"] if isinstance(data["stg"], list) else []
  761. # Temperature data
  762. temps = {}
  763. # Log all fields for debugging dual-nozzle temperature discovery (only once)
  764. if "bed_temper" in data and not hasattr(self, '_temp_fields_logged'):
  765. temp_fields = {k: v for k, v in data.items() if 'temp' in k.lower() or 'chamber' in k.lower()}
  766. logger.info(f"[{self.serial_number}] Temperature-related fields: {temp_fields}")
  767. # Log ALL keys in print data for H2D temperature discovery
  768. all_keys = sorted(data.keys())
  769. logger.info(f"[{self.serial_number}] ALL print data keys ({len(all_keys)}): {all_keys}")
  770. self._temp_fields_logged = True
  771. # Log vir_slot data (once) - this may contain per-extruder slot mapping for H2D
  772. if "vir_slot" in data and not hasattr(self, '_vir_slot_logged'):
  773. logger.info(f"[{self.serial_number}] vir_slot data: {data['vir_slot']}")
  774. self._vir_slot_logged = True
  775. # Log nozzle hardware info fields (once)
  776. nozzle_fields = {k: v for k, v in data.items() if 'nozzle' in k.lower() or 'hw' in k.lower() or 'extruder' in k.lower() or 'upgrade' in k.lower()}
  777. if nozzle_fields and not hasattr(self, '_nozzle_fields_logged'):
  778. logger.info(f"[{self.serial_number}] Nozzle/hardware fields in MQTT data: {nozzle_fields}")
  779. self._nozzle_fields_logged = True
  780. # Parse active extruder from device.extruder.state bit 8
  781. # bit 8 = 0 → RIGHT extruder (active_extruder=0)
  782. # bit 8 = 1 → LEFT extruder (active_extruder=1)
  783. if "device" in data and isinstance(data.get("device"), dict):
  784. device = data["device"]
  785. if "extruder" in device and "state" in device["extruder"]:
  786. state_val = device["extruder"]["state"]
  787. # Extract bit 8 for extruder position
  788. new_extruder = (state_val >> 8) & 0x1
  789. if new_extruder != self.state.active_extruder:
  790. logger.info(f"[{self.serial_number}] ACTIVE EXTRUDER CHANGED (state bit 8): {self.state.active_extruder} -> {new_extruder} (0=right, 1=left) [state={state_val}]")
  791. self.state.active_extruder = new_extruder
  792. # Log device.extruder structure for active extruder
  793. if "device" in data and isinstance(data.get("device"), dict):
  794. device = data["device"]
  795. if "extruder" in device:
  796. ext_data = device["extruder"]
  797. # Log 'state' field - OrcaSlicer uses bits 12-14 for switch state
  798. if "state" in ext_data:
  799. state_val = ext_data["state"]
  800. # Extract bits 12-14 (3 bits) for switch state
  801. switch_state = (state_val >> 12) & 0x7
  802. logger.info(f"[{self.serial_number}] device.extruder.state={state_val} (switch_state bits 12-14: {switch_state})")
  803. # Log 'cur' field if present (might indicate current/active extruder)
  804. if "cur" in ext_data:
  805. logger.info(f"[{self.serial_number}] device.extruder.cur: {ext_data['cur']}")
  806. if "bed_temper" in data:
  807. temps["bed"] = float(data["bed_temper"])
  808. if "bed_target_temper" in data:
  809. temps["bed_target"] = float(data["bed_target_temper"])
  810. # Check if this is H2D (has device.extruder.info with 2 extruders)
  811. has_h2d_extruder_info = (
  812. "device" in data and
  813. isinstance(data.get("device"), dict) and
  814. "extruder" in data["device"] and
  815. isinstance(data["device"]["extruder"].get("info"), list) and
  816. len(data["device"]["extruder"]["info"]) >= 2
  817. )
  818. # Standard nozzle fields: these are for the RIGHT/default nozzle on H2D
  819. # For H2D, we use these for nozzle_2 (RIGHT), for others use as nozzle (primary)
  820. if "nozzle_temper" in data:
  821. if has_h2d_extruder_info:
  822. temps["nozzle_2"] = float(data["nozzle_temper"]) # RIGHT nozzle on H2D
  823. else:
  824. temps["nozzle"] = float(data["nozzle_temper"])
  825. if "nozzle_target_temper" in data:
  826. if has_h2d_extruder_info:
  827. temps["nozzle_2_target"] = float(data["nozzle_target_temper"]) # RIGHT target on H2D
  828. else:
  829. temps["nozzle_target"] = float(data["nozzle_target_temper"])
  830. # Second nozzle for dual-extruder printers - skip for H2D (uses device.extruder.info instead)
  831. if not has_h2d_extruder_info:
  832. # Try multiple possible field names used by different firmware versions
  833. if "nozzle_temper_2" in data:
  834. val = float(data["nozzle_temper_2"])
  835. if -50 < val < 500: # Valid temp range
  836. temps["nozzle_2"] = val
  837. else:
  838. logger.debug(f"[{self.serial_number}] nozzle_temper_2={val} out of range")
  839. elif "right_nozzle_temper" in data:
  840. val = float(data["right_nozzle_temper"])
  841. if -50 < val < 500: # Valid temp range
  842. temps["nozzle_2"] = val
  843. else:
  844. logger.debug(f"[{self.serial_number}] right_nozzle_temper={val} out of range")
  845. if "nozzle_target_temper_2" in data:
  846. val = float(data["nozzle_target_temper_2"])
  847. if 0 <= val < 500: # Valid temp range
  848. temps["nozzle_2_target"] = val
  849. else:
  850. logger.debug(f"[{self.serial_number}] nozzle_target_temper_2={val} out of range")
  851. elif "right_nozzle_target_temper" in data:
  852. val = float(data["right_nozzle_target_temper"])
  853. if 0 <= val < 500: # Valid temp range
  854. temps["nozzle_2_target"] = val
  855. else:
  856. logger.debug(f"[{self.serial_number}] right_nozzle_target_temper={val} out of range")
  857. # Also check for left nozzle as primary (some H2 models)
  858. if "left_nozzle_temper" in data and "nozzle" not in temps:
  859. temps["nozzle"] = float(data["left_nozzle_temper"])
  860. if "left_nozzle_target_temper" in data and "nozzle_target" not in temps:
  861. temps["nozzle_target"] = float(data["left_nozzle_target_temper"])
  862. if "chamber_temper" in data:
  863. chamber_val = float(data["chamber_temper"])
  864. logger.debug(f"[{self.serial_number}] chamber_temper raw value: {chamber_val}")
  865. # Check if we recently set the target locally (within 5 seconds)
  866. local_set_time = self.state.temperatures.get("_chamber_target_set_time", 0)
  867. respect_local = (time.time() - local_set_time) < 5.0
  868. # H2D protocol: chamber_temper encoding indicates heater state
  869. # - When > 500: encoded as (target * 65536 + current) - heater is ON
  870. # - When < 500: direct Celsius current temp only - heater is OFF
  871. if -50 < chamber_val < 100:
  872. # Direct value = heater is OFF
  873. temps["chamber"] = chamber_val
  874. if not respect_local:
  875. temps["chamber_target"] = 0.0 # Heater off means target = 0
  876. logger.debug(f"[{self.serial_number}] chamber_temper direct value: {chamber_val}°C (heater OFF)")
  877. else:
  878. logger.debug(f"[{self.serial_number}] chamber_temper {chamber_val} out of direct range")
  879. # Try to decode if it looks like an encoded value
  880. if chamber_val > 500:
  881. mqtt_target = int(chamber_val) // 65536
  882. current = int(chamber_val) % 65536
  883. logger.debug(f"[{self.serial_number}] chamber_temper decoded: mqtt_target={mqtt_target}, current={current}, respect_local={respect_local}")
  884. if -50 < current < 100:
  885. temps["chamber"] = float(current)
  886. # Store decoded target for later use, but DON'T set chamber_heating here!
  887. # Heating state will be calculated later after parsing ctc.info.target (explicit target)
  888. # which is the authoritative source the slicer uses.
  889. if not respect_local:
  890. if 0 <= mqtt_target <= 60:
  891. # Store as "decoded" target - may be overridden by explicit target fields
  892. temps["_chamber_decoded_target"] = float(mqtt_target)
  893. # Chamber target temperature (set by print file or display)
  894. if "mc_target_cham" in data:
  895. mc_target = float(data["mc_target_cham"])
  896. logger.debug(f"[{self.serial_number}] mc_target_cham raw value: {mc_target}")
  897. # Filter out encoded/invalid values - valid chamber target is 0-60°C
  898. if 0 <= mc_target <= 60:
  899. temps["chamber_target"] = mc_target
  900. # H2D series: Chamber temp is in info.temp (may be encoded or direct °C)
  901. # NOTE: Don't set chamber_heating here - let ctc.info.target or fallback logic handle it
  902. # The encoded target in info.temp may be stale (slicer uses ctc.info.target as source of truth)
  903. try:
  904. if "info" in data and isinstance(data["info"], dict):
  905. info_temp = data["info"].get("temp")
  906. if info_temp is not None and "chamber" not in temps:
  907. # Check for encoded value (target * 65536 + current)
  908. if info_temp > 500:
  909. # Decode: extract current temperature and target
  910. target = info_temp // 65536
  911. current = info_temp % 65536
  912. temps["chamber"] = float(current)
  913. # Store decoded target as fallback (may be overridden by ctc.info.target)
  914. if "_chamber_decoded_target" not in temps:
  915. temps["_chamber_decoded_target"] = float(target)
  916. logger.debug(f"[{self.serial_number}] info.temp encoded: {info_temp} -> current={current}, decoded_target={target}")
  917. elif -50 < info_temp < 100:
  918. # Valid direct temperature - heater is OFF
  919. temps["chamber"] = float(info_temp)
  920. temps["chamber_target"] = 0.0 # Direct value means heater off
  921. logger.debug(f"[{self.serial_number}] info.temp direct: {info_temp}°C (heater OFF)")
  922. # H2D series: Dual extruder temps are in device.extruder.info array
  923. # Temperature values are encoded as fixed-point (value / 65536 = °C)
  924. if "device" in data and isinstance(data["device"], dict):
  925. device = data["device"]
  926. # Parse dual extruder temperatures
  927. extruder_data = device.get("extruder", {})
  928. extruder_info = extruder_data.get("info", [])
  929. if isinstance(extruder_info, list) and len(extruder_info) >= 1:
  930. # H2D nozzle mapping: id=0 is RIGHT nozzle (default), id=1 is LEFT nozzle
  931. # RIGHT nozzle uses standard nozzle_temper/nozzle_target_temper fields (already parsed above)
  932. # LEFT nozzle uses extruder_info[1] - no standard fields available
  933. # Note: hnow/htar flags are unreliable (static values, not actual heating state)
  934. # Real heating indicator: temp > 500 means encoded (target*65536+current)
  935. # Heating = target > 0 AND current < target
  936. # Right nozzle (extruder 0)
  937. if len(extruder_info) >= 1 and "temp" in extruder_info[0]:
  938. temp_val = extruder_info[0]["temp"]
  939. if temp_val > 500:
  940. target = temp_val // 65536
  941. current = temp_val % 65536
  942. temps["nozzle_2_heating"] = target > 0 and current < target
  943. else:
  944. temps["nozzle_2_heating"] = False
  945. # Left nozzle (extruder 1)
  946. # H2D protocol: temp field encoding depends on value
  947. # - When > 500: encoded as (target * 65536 + current) - heater is ON
  948. # - When < 500: direct Celsius current temp only - heater is OFF
  949. if len(extruder_info) >= 2 and "temp" in extruder_info[1]:
  950. ext1 = extruder_info[1]
  951. temp_val = ext1["temp"]
  952. # Check if we recently set the target locally (within 5 seconds)
  953. # If so, don't let MQTT data overwrite it
  954. local_set_time = self.state.temperatures.get("_nozzle_target_set_time", 0)
  955. respect_local_target = (time.time() - local_set_time) < 5.0
  956. if temp_val > 500:
  957. # Encoded format: temp = target * 65536 + current
  958. target = temp_val // 65536
  959. current = temp_val % 65536
  960. if 0 < target < 500 and not respect_local_target:
  961. temps["nozzle_target"] = float(target)
  962. if -50 < current < 500:
  963. temps["nozzle"] = float(current)
  964. # Heating = encoded AND we're using the MQTT target (not local override)
  965. # If local target is being respected, use local target to determine heating
  966. if respect_local_target:
  967. local_target = self.state.temperatures.get("nozzle_target", 0)
  968. temps["nozzle_heating"] = local_target > 0 and current < local_target
  969. else:
  970. temps["nozzle_heating"] = target > 0 and current < target
  971. elif -50 < temp_val < 500:
  972. # Direct Celsius = heater is OFF (or at target with heater off)
  973. temps["nozzle"] = float(temp_val)
  974. if not respect_local_target:
  975. temps["nozzle_target"] = 0.0
  976. temps["nozzle_heating"] = False # Direct = not heating
  977. # Parse bed heating state from device.bed.info.temp encoding
  978. # temp > 500 means encoded (target*65536+current), heating = target > 0 AND current < target
  979. bed_data = device.get("bed", {})
  980. bed_info = bed_data.get("info", {})
  981. if "temp" in bed_info:
  982. temp_val = bed_info["temp"]
  983. if temp_val > 500:
  984. target = temp_val // 65536
  985. current = temp_val % 65536
  986. temps["bed_heating"] = target > 0 and current < target
  987. else:
  988. temps["bed_heating"] = False
  989. # Parse chamber temp from device.ctc.info.temp if not already set
  990. ctc_data = device.get("ctc", {})
  991. ctc_info = ctc_data.get("info", {})
  992. # Parse airduct mode (0=cooling, 1=heating)
  993. airduct_data = device.get("airduct", {})
  994. if "modeCur" in airduct_data:
  995. new_mode = airduct_data["modeCur"]
  996. if new_mode != self.state.airduct_mode:
  997. logger.info(f"[{self.serial_number}] airduct_mode changed: {self.state.airduct_mode} -> {new_mode}")
  998. self.state.airduct_mode = new_mode
  999. # Parse chamber temp - may be encoded as (target*65536+current) when > 500
  1000. # Check if we recently set the target locally (within 5 seconds)
  1001. local_set_time = self.state.temperatures.get("_chamber_target_set_time", 0)
  1002. respect_local_target = (time.time() - local_set_time) < 5.0
  1003. # Log ctc_info contents for debugging
  1004. if ctc_info:
  1005. logger.debug(f"[{self.serial_number}] ctc_info keys: {list(ctc_info.keys())}")
  1006. # FIRST: Parse explicit ctc.info.target if available - this is the authoritative target
  1007. # (what the slicer shows). This OVERRIDES any previously decoded target.
  1008. explicit_target = None
  1009. if "target" in ctc_info:
  1010. target_val = ctc_info["target"]
  1011. logger.debug(f"[{self.serial_number}] ctc_info.target explicit value: {target_val}, respect_local={respect_local_target}")
  1012. # Filter out invalid values (valid chamber target is 0-60°C)
  1013. if 0 <= target_val <= 60 and not respect_local_target:
  1014. explicit_target = float(target_val)
  1015. temps["chamber_target"] = explicit_target # Override any previous value
  1016. logger.debug(f"[{self.serial_number}] Setting chamber_target from ctc_info.target: {explicit_target}")
  1017. # Parse chamber temp from ctc.info.temp - may be encoded
  1018. if "temp" in ctc_info and "chamber" not in temps:
  1019. temp_val = ctc_info["temp"]
  1020. logger.debug(f"[{self.serial_number}] ctc_info.temp raw value: {temp_val}")
  1021. if temp_val > 500:
  1022. # Encoded value: decode target and current
  1023. decoded_target = temp_val // 65536
  1024. current = temp_val % 65536
  1025. temps["chamber"] = float(current)
  1026. logger.debug(f"[{self.serial_number}] ctc_info.temp decoded: target={decoded_target}, current={current}, explicit_target={explicit_target}")
  1027. # Determine which target to use for heating state:
  1028. # Priority: local target > explicit target > decoded target
  1029. if respect_local_target:
  1030. local_target = self.state.temperatures.get("chamber_target", 0)
  1031. temps["chamber_heating"] = local_target > 0 and current < local_target
  1032. elif explicit_target is not None:
  1033. # Use explicit ctc.info.target - this is what slicer sees
  1034. temps["chamber_heating"] = explicit_target > 0 and current < explicit_target
  1035. else:
  1036. # Fallback to decoded target only if no explicit target available
  1037. if not respect_local_target and "chamber_target" not in temps:
  1038. temps["chamber_target"] = float(decoded_target)
  1039. temps["chamber_heating"] = decoded_target > 0 and current < decoded_target
  1040. else:
  1041. # Direct value (not encoded) - heater is OFF
  1042. temps["chamber"] = float(temp_val)
  1043. temps["chamber_heating"] = False
  1044. except Exception as e:
  1045. logger.warning(f"[{self.serial_number}] Error parsing H2D temperatures: {e}")
  1046. if temps:
  1047. # Handle chamber_target: prefer explicit over decoded
  1048. if "_chamber_decoded_target" in temps and "chamber_target" not in temps:
  1049. # No explicit target available, use decoded target from chamber_temper
  1050. temps["chamber_target"] = temps["_chamber_decoded_target"]
  1051. # Remove internal temp key before merging
  1052. temps.pop("_chamber_decoded_target", None)
  1053. # Merge new temps into existing, preserving valid values when new ones are filtered out
  1054. for key, value in temps.items():
  1055. self.state.temperatures[key] = value
  1056. # Calculate chamber_heating after all targets are known
  1057. # Priority: local target (if recent) > explicit target (chamber_target) > 0
  1058. if "chamber" in temps and "chamber_heating" not in temps:
  1059. current = self.state.temperatures.get("chamber", 0)
  1060. local_set_time = self.state.temperatures.get("_chamber_target_set_time", 0)
  1061. respect_local = (time.time() - local_set_time) < 5.0
  1062. if respect_local:
  1063. # Use locally-set target
  1064. target = self.state.temperatures.get("chamber_target", 0)
  1065. else:
  1066. # Use explicit/decoded target from MQTT
  1067. target = self.state.temperatures.get("chamber_target", 0)
  1068. self.state.temperatures["chamber_heating"] = target > 0 and current < target
  1069. logger.debug(f"[{self.serial_number}] Chamber heating calculated: target={target}, current={current}, heating={self.state.temperatures['chamber_heating']}, respect_local={respect_local}")
  1070. # Debug: log chamber value if it was updated
  1071. if "chamber" in temps:
  1072. logger.debug(f"[{self.serial_number}] Chamber temp updated to: {self.state.temperatures.get('chamber')}, target: {self.state.temperatures.get('chamber_target')}, heating: {self.state.temperatures.get('chamber_heating')}")
  1073. # Parse HMS (Health Management System) errors
  1074. if "hms" in data:
  1075. hms_list = data["hms"]
  1076. self.state.hms_errors = []
  1077. if isinstance(hms_list, list):
  1078. for hms in hms_list:
  1079. if isinstance(hms, dict):
  1080. # HMS format: {"attr": attribute_code, "code": error_code}
  1081. # attr contains module/severity info, code contains error number
  1082. # Both are needed to construct the wiki URL
  1083. attr = hms.get("attr", 0)
  1084. code = hms.get("code", 0)
  1085. if isinstance(attr, str):
  1086. attr = int(attr.replace("0x", ""), 16) if attr else 0
  1087. if isinstance(code, str):
  1088. code = int(code.replace("0x", ""), 16) if code else 0
  1089. # Severity is in attr byte 1 (bits 8-15)
  1090. severity = (attr >> 8) & 0xF
  1091. # Module is in attr byte 3 (bits 24-31)
  1092. module = (attr >> 24) & 0xFF
  1093. self.state.hms_errors.append(HMSError(
  1094. code=f"0x{code:x}" if code else "0x0",
  1095. attr=attr,
  1096. module=module,
  1097. severity=severity if severity > 0 else 2,
  1098. ))
  1099. # Parse SD card status
  1100. if "sdcard" in data:
  1101. self.state.sdcard = data["sdcard"] is True
  1102. # Parse home_flag for "Store Sent Files on External Storage" setting (bit 11)
  1103. if "home_flag" in data:
  1104. home_flag = data["home_flag"]
  1105. # Bit 11 controls "Store Sent Files on External Storage"
  1106. # Convert to unsigned 32-bit if negative
  1107. if home_flag < 0:
  1108. home_flag = home_flag & 0xFFFFFFFF
  1109. store_to_sdcard = bool((home_flag >> 11) & 1)
  1110. if store_to_sdcard != self.state.store_to_sdcard:
  1111. logger.info(f"[{self.serial_number}] store_to_sdcard changed: {self.state.store_to_sdcard} -> {store_to_sdcard}")
  1112. self.state.store_to_sdcard = store_to_sdcard
  1113. # Parse timelapse status (recording active during print)
  1114. if "timelapse" in data:
  1115. logger.debug(f"[{self.serial_number}] timelapse field: {data['timelapse']}")
  1116. self.state.timelapse = data["timelapse"] is True
  1117. # Parse ipcam/live view status
  1118. if "ipcam" in data:
  1119. ipcam_data = data["ipcam"]
  1120. logger.debug(f"[{self.serial_number}] ipcam field: {ipcam_data}")
  1121. if isinstance(ipcam_data, dict):
  1122. # Check ipcam_record field for live view status
  1123. self.state.ipcam = ipcam_data.get("ipcam_record") == "enable"
  1124. else:
  1125. self.state.ipcam = ipcam_data is True
  1126. # Parse WiFi signal strength (dBm)
  1127. if "wifi_signal" in data:
  1128. wifi_signal = data["wifi_signal"]
  1129. logger.info(f"[{self.serial_number}] wifi_signal received: {wifi_signal}")
  1130. if isinstance(wifi_signal, (int, float)):
  1131. self.state.wifi_signal = int(wifi_signal)
  1132. elif isinstance(wifi_signal, str):
  1133. # Handle string format like "-52dBm"
  1134. try:
  1135. self.state.wifi_signal = int(wifi_signal.replace("dBm", "").strip())
  1136. except ValueError:
  1137. pass
  1138. # Parse print speed level (1=silent, 2=standard, 3=sport, 4=ludicrous)
  1139. if "spd_lvl" in data:
  1140. new_speed = data["spd_lvl"]
  1141. if new_speed != self.state.speed_level:
  1142. logger.info(f"[{self.serial_number}] speed_level changed: {self.state.speed_level} -> {new_speed}")
  1143. self.state.speed_level = new_speed
  1144. # Parse chamber light status from lights_report
  1145. if "lights_report" in data:
  1146. lights = data["lights_report"]
  1147. logger.debug(f"[{self.serial_number}] lights_report: {lights}")
  1148. if isinstance(lights, list):
  1149. for light in lights:
  1150. if isinstance(light, dict) and light.get("node") == "chamber_light":
  1151. new_light_state = light.get("mode") == "on"
  1152. if new_light_state != self.state.chamber_light:
  1153. logger.info(f"[{self.serial_number}] chamber_light changed: {self.state.chamber_light} -> {new_light_state}")
  1154. self.state.chamber_light = new_light_state
  1155. break
  1156. # Parse nozzle hardware info (single nozzle printers)
  1157. if "nozzle_type" in data:
  1158. self.state.nozzles[0].nozzle_type = str(data["nozzle_type"])
  1159. if "nozzle_diameter" in data:
  1160. self.state.nozzles[0].nozzle_diameter = str(data["nozzle_diameter"])
  1161. # Parse nozzle hardware info (dual nozzle printers - H2D series)
  1162. # Left nozzle
  1163. if "left_nozzle_type" in data:
  1164. self.state.nozzles[0].nozzle_type = str(data["left_nozzle_type"])
  1165. if "left_nozzle_diameter" in data:
  1166. self.state.nozzles[0].nozzle_diameter = str(data["left_nozzle_diameter"])
  1167. # Right nozzle
  1168. if "right_nozzle_type" in data:
  1169. self.state.nozzles[1].nozzle_type = str(data["right_nozzle_type"])
  1170. if "right_nozzle_diameter" in data:
  1171. self.state.nozzles[1].nozzle_diameter = str(data["right_nozzle_diameter"])
  1172. # Alternative format for dual nozzle (nozzle_type_2, etc.)
  1173. if "nozzle_type_2" in data:
  1174. self.state.nozzles[1].nozzle_type = str(data["nozzle_type_2"])
  1175. if "nozzle_diameter_2" in data:
  1176. self.state.nozzles[1].nozzle_diameter = str(data["nozzle_diameter_2"])
  1177. # H2D series: Nozzle hardware info is in device.nozzle.info array
  1178. if "device" in data and isinstance(data["device"], dict):
  1179. device = data["device"]
  1180. nozzle_data = device.get("nozzle", {})
  1181. nozzle_info = nozzle_data.get("info", [])
  1182. if isinstance(nozzle_info, list):
  1183. for nozzle in nozzle_info:
  1184. idx = nozzle.get("id", 0)
  1185. if idx < len(self.state.nozzles):
  1186. if "type" in nozzle and nozzle["type"]:
  1187. self.state.nozzles[idx].nozzle_type = str(nozzle["type"])
  1188. if "diameter" in nozzle:
  1189. self.state.nozzles[idx].nozzle_diameter = str(nozzle["diameter"])
  1190. # Preserve AMS, vt_tray, and ams_extruder_map data when updating raw_data
  1191. ams_data = self.state.raw_data.get("ams")
  1192. vt_tray_data = self.state.raw_data.get("vt_tray")
  1193. ams_extruder_map_data = self.state.raw_data.get("ams_extruder_map")
  1194. self.state.raw_data = data
  1195. if ams_data is not None:
  1196. self.state.raw_data["ams"] = ams_data
  1197. if vt_tray_data is not None:
  1198. self.state.raw_data["vt_tray"] = vt_tray_data
  1199. if ams_extruder_map_data is not None:
  1200. self.state.raw_data["ams_extruder_map"] = ams_extruder_map_data
  1201. # Log state transitions for debugging
  1202. if "gcode_state" in data:
  1203. logger.debug(
  1204. f"[{self.serial_number}] gcode_state: {self._previous_gcode_state} -> {self.state.state}, "
  1205. f"file: {self.state.gcode_file}, subtask: {self.state.subtask_name}"
  1206. )
  1207. # Detect print start (state changes TO RUNNING with a file)
  1208. current_file = self.state.gcode_file or self.state.current_print
  1209. is_new_print = (
  1210. self.state.state == "RUNNING"
  1211. and self._previous_gcode_state != "RUNNING"
  1212. and current_file
  1213. )
  1214. # Also detect if file changed while running (new print started)
  1215. is_file_change = (
  1216. self.state.state == "RUNNING"
  1217. and current_file
  1218. and current_file != self._previous_gcode_file
  1219. and self._previous_gcode_file is not None
  1220. )
  1221. # Track RUNNING state for more robust completion detection
  1222. if self.state.state == "RUNNING" and current_file:
  1223. if not self._was_running:
  1224. logger.info(f"[{self.serial_number}] Now tracking RUNNING state for {current_file}")
  1225. self._was_running = True
  1226. self._completion_triggered = False
  1227. if is_new_print or is_file_change:
  1228. # Clear any old HMS errors when a new print starts
  1229. self.state.hms_errors = []
  1230. # Reset completion tracking for new print
  1231. self._was_running = True
  1232. self._completion_triggered = False
  1233. if (is_new_print or is_file_change) and self.on_print_start:
  1234. logger.info(
  1235. f"[{self.serial_number}] PRINT START detected - file: {current_file}, "
  1236. f"subtask: {self.state.subtask_name}, is_new: {is_new_print}, is_file_change: {is_file_change}"
  1237. )
  1238. self.on_print_start({
  1239. "filename": current_file,
  1240. "subtask_name": self.state.subtask_name,
  1241. "raw_data": data,
  1242. })
  1243. # Detect print completion (FINISH = success, FAILED = error, IDLE = aborted)
  1244. # Use _was_running flag in addition to _previous_gcode_state for more robust detection
  1245. # This handles cases where server restarts during a print
  1246. should_trigger_completion = (
  1247. self.state.state in ("FINISH", "FAILED")
  1248. and not self._completion_triggered
  1249. and self.on_print_complete
  1250. and (
  1251. self._previous_gcode_state == "RUNNING" # Normal transition
  1252. or (self._was_running and self._previous_gcode_state != self.state.state) # After server restart
  1253. )
  1254. )
  1255. # For IDLE, only trigger if we just came from RUNNING (explicit abort/cancel)
  1256. if (
  1257. self.state.state == "IDLE"
  1258. and self._previous_gcode_state == "RUNNING"
  1259. and not self._completion_triggered
  1260. and self.on_print_complete
  1261. ):
  1262. should_trigger_completion = True
  1263. if should_trigger_completion:
  1264. if self.state.state == "FINISH":
  1265. status = "completed"
  1266. elif self.state.state == "FAILED":
  1267. status = "failed"
  1268. else:
  1269. status = "aborted"
  1270. logger.info(
  1271. f"[{self.serial_number}] PRINT COMPLETE detected - state: {self.state.state}, "
  1272. f"status: {status}, file: {self._previous_gcode_file or current_file}, "
  1273. f"subtask: {self.state.subtask_name}, was_running: {self._was_running}"
  1274. )
  1275. self._completion_triggered = True
  1276. self._was_running = False
  1277. self.on_print_complete({
  1278. "status": status,
  1279. "filename": self._previous_gcode_file or current_file,
  1280. "subtask_name": self.state.subtask_name,
  1281. "raw_data": data,
  1282. })
  1283. self._previous_gcode_state = self.state.state
  1284. if current_file:
  1285. self._previous_gcode_file = current_file
  1286. if self.on_state_change:
  1287. self.on_state_change(self.state)
  1288. def _request_push_all(self):
  1289. """Request full status update from printer."""
  1290. if self._client:
  1291. message = {"pushing": {"command": "pushall"}}
  1292. self._client.publish(self.topic_publish, json.dumps(message))
  1293. def request_status_update(self) -> bool:
  1294. """Request a full status update from the printer (public API).
  1295. Sends both pushall and get_accessories commands to refresh all data
  1296. including nozzle hardware info.
  1297. Returns:
  1298. True if the request was sent, False if not connected.
  1299. """
  1300. if not self._client or not self.state.connected:
  1301. return False
  1302. self._request_push_all()
  1303. # Note: get_accessories returns stale nozzle data on H2D.
  1304. # The correct nozzle data comes from push_status response.
  1305. return True
  1306. def _request_accessories(self):
  1307. """Request accessories info (nozzle type, etc.) from printer."""
  1308. if self._client:
  1309. self._sequence_id += 1
  1310. message = {
  1311. "system": {
  1312. "sequence_id": str(self._sequence_id),
  1313. "command": "get_accessories",
  1314. "accessory_type": "none"
  1315. }
  1316. }
  1317. logger.debug(f"[{self.serial_number}] Requesting accessories info")
  1318. self._client.publish(self.topic_publish, json.dumps(message))
  1319. def _prime_kprofile_request(self):
  1320. """Send a priming K-profile request on connect.
  1321. Bambu printers often ignore the first K-profile request after connection,
  1322. so we send a dummy request on connect to 'prime' the system.
  1323. """
  1324. if self._client:
  1325. self._sequence_id += 1
  1326. command = {
  1327. "print": {
  1328. "command": "extrusion_cali_get",
  1329. "filament_id": "",
  1330. "nozzle_diameter": "0.4",
  1331. "sequence_id": str(self._sequence_id),
  1332. }
  1333. }
  1334. logger.debug(f"[{self.serial_number}] Sending K-profile priming request")
  1335. self._client.publish(self.topic_publish, json.dumps(command))
  1336. def connect(self, loop: asyncio.AbstractEventLoop | None = None):
  1337. """Connect to the printer MQTT broker.
  1338. Args:
  1339. loop: The asyncio event loop to use for thread-safe callbacks.
  1340. If not provided, will try to get the running loop.
  1341. """
  1342. self._loop = loop
  1343. self._client = mqtt.Client(
  1344. callback_api_version=mqtt.CallbackAPIVersion.VERSION2,
  1345. client_id=f"bambutrack_{self.serial_number}",
  1346. protocol=mqtt.MQTTv311,
  1347. )
  1348. self._client.username_pw_set("bblp", self.access_code)
  1349. self._client.on_connect = self._on_connect
  1350. self._client.on_disconnect = self._on_disconnect
  1351. self._client.on_message = self._on_message
  1352. # TLS setup - Bambu uses self-signed certs
  1353. ssl_context = ssl.create_default_context()
  1354. ssl_context.check_hostname = False
  1355. ssl_context.verify_mode = ssl.CERT_NONE
  1356. self._client.tls_set_context(ssl_context)
  1357. # Use shorter keepalive (15s) for faster disconnect detection
  1358. # Paho considers connection lost after 1.5x keepalive with no response
  1359. self._client.connect_async(self.ip_address, self.MQTT_PORT, keepalive=15)
  1360. self._client.loop_start()
  1361. def start_print(self, filename: str, plate_id: int = 1):
  1362. """Start a print job on the printer.
  1363. The file should already be uploaded to /cache/ on the printer via FTP.
  1364. """
  1365. if self._client and self.state.connected:
  1366. # Bambu print command format
  1367. # Based on: https://github.com/darkorb/bambu-ftp-and-print
  1368. command = {
  1369. "print": {
  1370. "sequence_id": 0,
  1371. "command": "project_file",
  1372. "param": f"Metadata/plate_{plate_id}.gcode",
  1373. "subtask_name": filename,
  1374. "url": f"ftp://{filename}",
  1375. "timelapse": False,
  1376. "bed_leveling": True,
  1377. "flow_cali": True,
  1378. "vibration_cali": True,
  1379. "layer_inspect": False,
  1380. "use_ams": True,
  1381. }
  1382. }
  1383. logger.info(f"[{self.serial_number}] Sending print command: {json.dumps(command)}")
  1384. self._client.publish(self.topic_publish, json.dumps(command))
  1385. return True
  1386. return False
  1387. def stop_print(self) -> bool:
  1388. """Stop the current print job."""
  1389. if self._client and self.state.connected:
  1390. command = {
  1391. "print": {
  1392. "command": "stop",
  1393. "sequence_id": "0"
  1394. }
  1395. }
  1396. self._client.publish(self.topic_publish, json.dumps(command))
  1397. logger.info(f"[{self.serial_number}] Sent stop print command")
  1398. return True
  1399. return False
  1400. def set_xcam_option(
  1401. self,
  1402. module_name: str,
  1403. enabled: bool,
  1404. print_halt: bool = True,
  1405. sensitivity: str = "medium"
  1406. ) -> bool:
  1407. """Set an xcam (AI detection) option on the printer.
  1408. Args:
  1409. module_name: The xcam module to control (e.g., "spaghetti_detector",
  1410. "first_layer_inspector", "printing_monitor", "buildplate_marker_detector")
  1411. enabled: Whether to enable or disable the feature
  1412. print_halt: Whether to halt print on detection (only applies to some detectors)
  1413. sensitivity: Sensitivity level ("low", "medium", "high", or "never_halt")
  1414. Returns:
  1415. True if command was sent, False if not connected
  1416. """
  1417. if not self._client or not self.state.connected:
  1418. return False
  1419. # auto_recovery_step_loss uses a different command format (print.print_option)
  1420. if module_name == "auto_recovery_step_loss":
  1421. return self._set_print_option("auto_recovery", enabled)
  1422. self._sequence_id += 1
  1423. # Build the xcam control command (exact OrcaSlicer format)
  1424. # Key findings from OrcaSlicer source:
  1425. # - Uses "xcam" wrapper (not "print")
  1426. # - print_halt is ALWAYS true (legacy protocol requirement)
  1427. # - Both "control" and "enable" are set to the same value
  1428. # - halt_print_sensitivity controls actual halt behavior
  1429. command = {
  1430. "xcam": {
  1431. "command": "xcam_control_set",
  1432. "sequence_id": str(self._sequence_id),
  1433. "module_name": module_name,
  1434. "control": enabled,
  1435. "enable": enabled, # old protocol compatibility
  1436. "print_halt": True, # ALWAYS true per OrcaSlicer
  1437. }
  1438. }
  1439. # Only add sensitivity if not "never_halt"
  1440. # OrcaSlicer uses halt_print_sensitivity for ALL detectors
  1441. # The module_name field determines which detector's sensitivity is being set
  1442. if sensitivity and sensitivity != "never_halt":
  1443. command["xcam"]["halt_print_sensitivity"] = sensitivity
  1444. command_json = json.dumps(command)
  1445. self._client.publish(self.topic_publish, command_json, qos=1)
  1446. logger.info(f"[{self.serial_number}] Set xcam option: {module_name}={enabled}, sensitivity={sensitivity}")
  1447. logger.debug(f"[{self.serial_number}] MQTT command sent: {command_json}")
  1448. # OrcaSlicer pattern: Set hold timer to ignore incoming data for 3 seconds
  1449. # This prevents stale MQTT data from immediately overwriting our change
  1450. self._xcam_hold_start[module_name] = time.time()
  1451. # Update local state immediately for responsive UI
  1452. # NOTE: Spaghetti and Pileup sensitivities are linked in firmware
  1453. # When spaghetti_detector sensitivity is changed, pileup also changes
  1454. if module_name == "spaghetti_detector":
  1455. self.state.print_options.spaghetti_detector = enabled
  1456. self.state.print_options.print_halt = print_halt
  1457. if sensitivity and sensitivity != "never_halt":
  1458. # spaghetti_detector controls BOTH spaghetti and pileup sensitivities
  1459. self.state.print_options.halt_print_sensitivity = sensitivity
  1460. self.state.print_options.pileup_sensitivity = sensitivity
  1461. self._xcam_hold_start["halt_print_sensitivity"] = time.time()
  1462. self._xcam_hold_start["pileup_sensitivity"] = time.time()
  1463. elif module_name == "first_layer_inspector":
  1464. self.state.print_options.first_layer_inspector = enabled
  1465. elif module_name == "printing_monitor":
  1466. self.state.print_options.printing_monitor = enabled
  1467. elif module_name == "buildplate_marker_detector":
  1468. self.state.print_options.buildplate_marker_detector = enabled
  1469. elif module_name == "allow_skip_parts":
  1470. self.state.print_options.allow_skip_parts = enabled
  1471. elif module_name == "pileup_detector":
  1472. self.state.print_options.pileup_detector = enabled
  1473. # Pileup sensitivity is linked to spaghetti - both are set via spaghetti_detector
  1474. elif module_name == "clump_detector":
  1475. self.state.print_options.nozzle_clumping_detector = enabled
  1476. if sensitivity and sensitivity != "never_halt":
  1477. self.state.print_options.nozzle_clumping_sensitivity = sensitivity
  1478. self._xcam_hold_start["nozzle_clumping_sensitivity"] = time.time()
  1479. elif module_name == "airprint_detector":
  1480. self.state.print_options.airprint_detector = enabled
  1481. if sensitivity and sensitivity != "never_halt":
  1482. self.state.print_options.airprint_sensitivity = sensitivity
  1483. self._xcam_hold_start["airprint_sensitivity"] = time.time()
  1484. elif module_name == "auto_recovery_step_loss":
  1485. self.state.print_options.auto_recovery_step_loss = enabled
  1486. return True
  1487. def _set_print_option(self, option_name: str, enabled: bool) -> bool:
  1488. """Set a print option using the print.print_option command.
  1489. This is different from xcam_control_set and is used for options like:
  1490. - auto_recovery
  1491. - air_print_detect
  1492. - filament_tangle_detect
  1493. - nozzle_blob_detect
  1494. - sound_enable
  1495. Args:
  1496. option_name: The option to control (e.g., "auto_recovery")
  1497. enabled: Whether to enable or disable the option
  1498. Returns:
  1499. True if command was sent, False if not connected
  1500. """
  1501. if not self._client or not self.state.connected:
  1502. return False
  1503. self._sequence_id += 1
  1504. command = {
  1505. "print": {
  1506. "command": "print_option",
  1507. "sequence_id": str(self._sequence_id),
  1508. option_name: enabled,
  1509. }
  1510. }
  1511. command_json = json.dumps(command)
  1512. self._client.publish(self.topic_publish, command_json, qos=1)
  1513. logger.info(f"[{self.serial_number}] Set print option: {option_name}={enabled}")
  1514. # Set hold timer
  1515. hold_key = f"print_option_{option_name}"
  1516. self._xcam_hold_start[hold_key] = time.time()
  1517. # Update local state immediately
  1518. if option_name == "auto_recovery":
  1519. self.state.print_options.auto_recovery_step_loss = enabled
  1520. return True
  1521. def start_calibration(
  1522. self,
  1523. bed_leveling: bool = False,
  1524. vibration: bool = False,
  1525. motor_noise: bool = False,
  1526. nozzle_offset: bool = False,
  1527. high_temp_heatbed: bool = False,
  1528. ) -> bool:
  1529. """Start printer calibration with selected options.
  1530. Args:
  1531. bed_leveling: Run bed leveling calibration
  1532. vibration: Run vibration compensation calibration
  1533. motor_noise: Run motor noise cancellation calibration
  1534. nozzle_offset: Run nozzle offset calibration (dual nozzle printers)
  1535. high_temp_heatbed: Run high-temperature heatbed calibration
  1536. Returns:
  1537. True if command was sent, False if not connected
  1538. """
  1539. if not self._client or not self.state.connected:
  1540. return False
  1541. # Build calibration bitmask based on OrcaSlicer DeviceManager.cpp
  1542. # Bit 0: xcam_cali (not exposed in UI)
  1543. # Bit 1: bed_leveling
  1544. # Bit 2: vibration
  1545. # Bit 3: motor_noise
  1546. # Bit 4: nozzle_cali
  1547. # Bit 5: bed_cali (high-temp heatbed)
  1548. # Bit 6: clumppos_cali (not exposed in UI)
  1549. option = 0
  1550. if bed_leveling:
  1551. option |= 1 << 1
  1552. if vibration:
  1553. option |= 1 << 2
  1554. if motor_noise:
  1555. option |= 1 << 3
  1556. if nozzle_offset:
  1557. option |= 1 << 4
  1558. if high_temp_heatbed:
  1559. option |= 1 << 5
  1560. if option == 0:
  1561. logger.warning(f"[{self.serial_number}] No calibration options selected")
  1562. return False
  1563. self._sequence_id += 1
  1564. command = {
  1565. "print": {
  1566. "command": "calibration",
  1567. "sequence_id": str(self._sequence_id),
  1568. "option": option,
  1569. }
  1570. }
  1571. command_json = json.dumps(command)
  1572. self._client.publish(self.topic_publish, command_json, qos=1)
  1573. logger.info(
  1574. f"[{self.serial_number}] Starting calibration: "
  1575. f"bed_leveling={bed_leveling}, vibration={vibration}, "
  1576. f"motor_noise={motor_noise}, nozzle_offset={nozzle_offset}, "
  1577. f"high_temp_heatbed={high_temp_heatbed} (option={option})"
  1578. )
  1579. return True
  1580. def disconnect(self):
  1581. """Disconnect from the printer."""
  1582. if self._client:
  1583. self._client.loop_stop()
  1584. self._client.disconnect()
  1585. self._client = None
  1586. self.state.connected = False
  1587. def send_command(self, command: dict):
  1588. """Send a command to the printer."""
  1589. if self._client and self.state.connected:
  1590. # Log outgoing message if logging is enabled
  1591. if self._logging_enabled:
  1592. self._message_log.append(MQTTLogEntry(
  1593. timestamp=datetime.now().isoformat(),
  1594. topic=self.topic_publish,
  1595. direction="out",
  1596. payload=command,
  1597. ))
  1598. self._client.publish(self.topic_publish, json.dumps(command))
  1599. def enable_logging(self, enabled: bool = True):
  1600. """Enable or disable MQTT message logging."""
  1601. self._logging_enabled = enabled
  1602. # Don't clear logs when stopping - user can manually clear with clear_logs()
  1603. def get_logs(self) -> list[MQTTLogEntry]:
  1604. """Get all logged MQTT messages."""
  1605. return list(self._message_log)
  1606. def clear_logs(self):
  1607. """Clear the message log."""
  1608. self._message_log.clear()
  1609. @property
  1610. def logging_enabled(self) -> bool:
  1611. """Check if logging is enabled."""
  1612. return self._logging_enabled
  1613. def _handle_kprofile_response(self, data: dict):
  1614. """Handle K-profile response from printer."""
  1615. filaments = data.get("filaments", [])
  1616. profiles = []
  1617. # Log first profile to see what fields the printer returns
  1618. if filaments and isinstance(filaments[0], dict):
  1619. logger.debug(f"[{self.serial_number}] Raw K-profile fields: {list(filaments[0].keys())}")
  1620. logger.debug(f"[{self.serial_number}] First K-profile: {filaments[0]}")
  1621. for i, f in enumerate(filaments):
  1622. if isinstance(f, dict):
  1623. try:
  1624. # cali_idx is the actual slot/calibration index from the printer
  1625. cali_idx = f.get("cali_idx", i)
  1626. profiles.append(KProfile(
  1627. slot_id=cali_idx,
  1628. extruder_id=int(f.get("extruder_id", 0)),
  1629. nozzle_id=str(f.get("nozzle_id", "")),
  1630. nozzle_diameter=str(f.get("nozzle_diameter", "0.4")),
  1631. filament_id=str(f.get("filament_id", "")),
  1632. name=str(f.get("name", "")),
  1633. k_value=str(f.get("k_value", "0.000000")),
  1634. n_coef=str(f.get("n_coef", "0.000000")),
  1635. ams_id=int(f.get("ams_id", 0)),
  1636. tray_id=int(f.get("tray_id", -1)),
  1637. setting_id=f.get("setting_id"),
  1638. ))
  1639. except (ValueError, TypeError) as e:
  1640. logger.warning(f"Failed to parse K-profile: {e}")
  1641. self.state.kprofiles = profiles
  1642. self._kprofile_response_data = profiles
  1643. # Signal that we received the response
  1644. # Use thread-safe method since MQTT callbacks run in a different thread
  1645. if self._pending_kprofile_response:
  1646. if self._loop and self._loop.is_running():
  1647. self._loop.call_soon_threadsafe(self._pending_kprofile_response.set)
  1648. else:
  1649. # Fallback for when loop is not available
  1650. self._pending_kprofile_response.set()
  1651. logger.info(f"[{self.serial_number}] Received {len(profiles)} K-profiles")
  1652. async def get_kprofiles(self, nozzle_diameter: str = "0.4", timeout: float = 5.0, max_retries: int = 3) -> list[KProfile]:
  1653. """Request K-profiles from the printer with retry logic.
  1654. Bambu printers sometimes ignore the first K-profile request, so we
  1655. implement retry logic to ensure reliable retrieval.
  1656. Args:
  1657. nozzle_diameter: Filter by nozzle diameter (e.g., "0.4")
  1658. timeout: Timeout in seconds to wait for each response attempt
  1659. max_retries: Maximum number of retry attempts
  1660. Returns:
  1661. List of KProfile objects
  1662. """
  1663. if not self._client or not self.state.connected:
  1664. logger.warning(f"[{self.serial_number}] Cannot get K-profiles: not connected")
  1665. return []
  1666. # Capture current event loop for thread-safe callback
  1667. try:
  1668. self._loop = asyncio.get_running_loop()
  1669. except RuntimeError:
  1670. logger.warning(f"[{self.serial_number}] No running event loop")
  1671. return []
  1672. for attempt in range(max_retries):
  1673. # Set up response event for this attempt
  1674. self._sequence_id += 1
  1675. self._pending_kprofile_response = asyncio.Event()
  1676. self._kprofile_response_data = None
  1677. # Send the command
  1678. command = {
  1679. "print": {
  1680. "command": "extrusion_cali_get",
  1681. "filament_id": "",
  1682. "nozzle_diameter": nozzle_diameter,
  1683. "sequence_id": str(self._sequence_id),
  1684. }
  1685. }
  1686. logger.info(f"[{self.serial_number}] Requesting K-profiles for nozzle {nozzle_diameter} (attempt {attempt + 1}/{max_retries})")
  1687. self._client.publish(self.topic_publish, json.dumps(command))
  1688. # Wait for response
  1689. try:
  1690. await asyncio.wait_for(self._pending_kprofile_response.wait(), timeout=timeout)
  1691. profiles = self._kprofile_response_data or []
  1692. logger.info(f"[{self.serial_number}] Got {len(profiles)} K-profiles on attempt {attempt + 1}")
  1693. return profiles
  1694. except asyncio.TimeoutError:
  1695. logger.warning(f"[{self.serial_number}] Timeout on K-profiles request attempt {attempt + 1}/{max_retries}")
  1696. if attempt < max_retries - 1:
  1697. # Brief delay before retry
  1698. await asyncio.sleep(0.5)
  1699. finally:
  1700. self._pending_kprofile_response = None
  1701. logger.error(f"[{self.serial_number}] Failed to get K-profiles after {max_retries} attempts")
  1702. return []
  1703. def set_kprofile(
  1704. self,
  1705. filament_id: str,
  1706. name: str,
  1707. k_value: str,
  1708. nozzle_diameter: str = "0.4",
  1709. nozzle_id: str = "HS00-0.4",
  1710. extruder_id: int = 0,
  1711. setting_id: str | None = None,
  1712. slot_id: int = 0,
  1713. cali_idx: int | None = None,
  1714. ) -> bool:
  1715. """Set/update a K-profile on the printer.
  1716. Args:
  1717. filament_id: Bambu filament identifier
  1718. name: Profile name
  1719. k_value: Pressure advance value (e.g., "0.020000")
  1720. nozzle_diameter: Nozzle diameter (e.g., "0.4")
  1721. nozzle_id: Nozzle identifier (e.g., "HS00-0.4")
  1722. extruder_id: Extruder ID (0 or 1 for dual nozzle)
  1723. setting_id: Existing setting ID for updates, None for new
  1724. slot_id: Calibration index (cali_idx) for the profile
  1725. cali_idx: For H2D edits, the existing slot being edited (enables in-place edit)
  1726. Returns:
  1727. True if command was sent, False otherwise
  1728. """
  1729. if not self._client or not self.state.connected:
  1730. logger.warning(f"[{self.serial_number}] Cannot set K-profile: not connected")
  1731. return False
  1732. self._sequence_id += 1
  1733. # Detect printer type by serial number prefix
  1734. # X1C/P1/A1 series (single nozzle): serial starts with "00M", "00W", "01P", "01S", "03W", etc.
  1735. # H2D series (dual nozzle): serial starts with "094"
  1736. is_dual_nozzle = self.serial_number.startswith("094")
  1737. # For H2D edits, use empty setting_id per OrcaSlicer sniff
  1738. # For new profiles, generate a setting_id
  1739. import secrets
  1740. if cali_idx is not None:
  1741. # Edit mode - use empty setting_id per OrcaSlicer sniff
  1742. setting_id = ""
  1743. elif not setting_id and slot_id == 0:
  1744. # New profile - generate setting_id
  1745. setting_id = f"PFUS{secrets.token_hex(7)}" # 7 bytes = 14 hex chars
  1746. if is_dual_nozzle:
  1747. # H2D format - exact OrcaSlicer format (captured via MQTT sniffing)
  1748. # For edits: include cali_idx (existing slot), slot_id=0, setting_id=""
  1749. # For new profiles: no cali_idx, slot_id=0, setting_id=generated
  1750. filament_entry = {
  1751. "ams_id": 0,
  1752. "extruder_id": extruder_id,
  1753. "filament_id": filament_id,
  1754. "k_value": k_value,
  1755. "n_coef": "0.000000",
  1756. "name": name,
  1757. "nozzle_diameter": nozzle_diameter,
  1758. "nozzle_id": nozzle_id,
  1759. "setting_id": setting_id if setting_id else "",
  1760. "slot_id": slot_id,
  1761. "tray_id": -1,
  1762. }
  1763. # For edits, add cali_idx field (position matters - alphabetical order)
  1764. if cali_idx is not None:
  1765. # Insert cali_idx in alphabetical position (after ams_id, before extruder_id)
  1766. # n_coef must be "0.000000" for H2D edits (matches OrcaSlicer sniff)
  1767. filament_entry = {
  1768. "ams_id": 0,
  1769. "cali_idx": cali_idx,
  1770. "extruder_id": extruder_id,
  1771. "filament_id": filament_id,
  1772. "k_value": k_value,
  1773. "n_coef": "0.000000",
  1774. "name": name,
  1775. "nozzle_diameter": nozzle_diameter,
  1776. "nozzle_id": nozzle_id,
  1777. "setting_id": "",
  1778. "slot_id": 0,
  1779. "tray_id": -1,
  1780. }
  1781. command = {
  1782. "print": {
  1783. "command": "extrusion_cali_set",
  1784. "filaments": [filament_entry],
  1785. "nozzle_diameter": nozzle_diameter,
  1786. "sequence_id": str(self._sequence_id),
  1787. }
  1788. }
  1789. else:
  1790. # X1C/P1/A1 format - based on actual X1C profile data:
  1791. # - n_coef: "1.000000" (NOT 0.000000 like H2D)
  1792. # - nozzle_id: "" (empty string, NOT the nozzle type)
  1793. # - tray_id: -1 (NOT 0)
  1794. filament_entry = {
  1795. "ams_id": 0,
  1796. "extruder_id": 0, # X1C is single nozzle
  1797. "filament_id": filament_id,
  1798. "k_value": k_value,
  1799. "n_coef": "1.000000", # X1C uses 1.0, not 0.0
  1800. "name": name,
  1801. "nozzle_diameter": nozzle_diameter,
  1802. "nozzle_id": "", # X1C uses empty string
  1803. "setting_id": setting_id,
  1804. "slot_id": slot_id,
  1805. "tray_id": -1, # X1C uses -1
  1806. }
  1807. command = {
  1808. "print": {
  1809. "command": "extrusion_cali_set",
  1810. "filaments": [filament_entry],
  1811. "nozzle_diameter": nozzle_diameter,
  1812. "sequence_id": str(self._sequence_id),
  1813. }
  1814. }
  1815. command_json = json.dumps(command)
  1816. logger.info(f"[{self.serial_number}] Setting K-profile: {name} = {k_value} (cali_idx={cali_idx}, new={slot_id==0}, dual={is_dual_nozzle})")
  1817. logger.info(f"[{self.serial_number}] K-profile SET command: {command_json}")
  1818. # Use QoS 1 for reliable delivery (at least once)
  1819. self._client.publish(self.topic_publish, command_json, qos=1)
  1820. return True
  1821. def delete_kprofile(
  1822. self,
  1823. cali_idx: int,
  1824. filament_id: str,
  1825. nozzle_id: str,
  1826. nozzle_diameter: str = "0.4",
  1827. extruder_id: int = 0,
  1828. setting_id: str | None = None,
  1829. ) -> bool:
  1830. """Delete a K-profile from the printer.
  1831. Args:
  1832. cali_idx: The calibration index (slot_id) of the profile to delete
  1833. filament_id: Bambu filament identifier
  1834. nozzle_id: Nozzle identifier (e.g., "HH00-0.4")
  1835. nozzle_diameter: Nozzle diameter (e.g., "0.4")
  1836. extruder_id: Extruder ID (0 or 1 for dual nozzle)
  1837. setting_id: Unique setting identifier (for X1C series)
  1838. Returns:
  1839. True if command was sent, False otherwise
  1840. """
  1841. if not self._client or not self.state.connected:
  1842. logger.warning(f"[{self.serial_number}] Cannot delete K-profile: not connected")
  1843. return False
  1844. self._sequence_id += 1
  1845. # Detect printer type by serial number prefix
  1846. # H2D series (dual nozzle): serial starts with "094"
  1847. is_dual_nozzle = self.serial_number.startswith("094")
  1848. if is_dual_nozzle:
  1849. # H2D format: uses extruder_id, nozzle_id, nozzle_diameter
  1850. command = {
  1851. "print": {
  1852. "command": "extrusion_cali_del",
  1853. "sequence_id": str(self._sequence_id),
  1854. "extruder_id": extruder_id,
  1855. "nozzle_id": nozzle_id,
  1856. "filament_id": filament_id,
  1857. "cali_idx": cali_idx,
  1858. "nozzle_diameter": nozzle_diameter,
  1859. }
  1860. }
  1861. else:
  1862. # X1C/P1/A1 format: uses setting_id, nozzle_diameter, no extruder/nozzle_id fields
  1863. command = {
  1864. "print": {
  1865. "command": "extrusion_cali_del",
  1866. "sequence_id": str(self._sequence_id),
  1867. "filament_id": filament_id,
  1868. "cali_idx": cali_idx,
  1869. "setting_id": setting_id,
  1870. "nozzle_diameter": nozzle_diameter,
  1871. }
  1872. }
  1873. command_json = json.dumps(command)
  1874. logger.info(f"[{self.serial_number}] Deleting K-profile: cali_idx={cali_idx}, filament={filament_id}, dual={is_dual_nozzle}")
  1875. logger.info(f"[{self.serial_number}] K-profile DELETE command: {command_json}")
  1876. # Use QoS 1 for reliable delivery (at least once)
  1877. self._client.publish(self.topic_publish, command_json, qos=1)
  1878. return True
  1879. # =========================================================================
  1880. # Printer Control Commands
  1881. # =========================================================================
  1882. def pause_print(self) -> bool:
  1883. """Pause the current print job."""
  1884. if not self._client or not self.state.connected:
  1885. logger.warning(f"[{self.serial_number}] Cannot pause print: not connected")
  1886. return False
  1887. command = {
  1888. "print": {
  1889. "command": "pause",
  1890. "sequence_id": "0"
  1891. }
  1892. }
  1893. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  1894. logger.info(f"[{self.serial_number}] Sent pause print command")
  1895. return True
  1896. def resume_print(self) -> bool:
  1897. """Resume a paused print job."""
  1898. if not self._client or not self.state.connected:
  1899. logger.warning(f"[{self.serial_number}] Cannot resume print: not connected")
  1900. return False
  1901. command = {
  1902. "print": {
  1903. "command": "resume",
  1904. "sequence_id": "0"
  1905. }
  1906. }
  1907. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  1908. logger.info(f"[{self.serial_number}] Sent resume print command")
  1909. return True
  1910. def send_gcode(self, gcode: str) -> bool:
  1911. """Send G-code command(s) to the printer.
  1912. Multiple commands can be separated by newlines.
  1913. Args:
  1914. gcode: G-code command(s) to send
  1915. Returns:
  1916. True if command was sent, False otherwise
  1917. """
  1918. if not self._client or not self.state.connected:
  1919. logger.warning(f"[{self.serial_number}] Cannot send G-code: not connected")
  1920. return False
  1921. self._sequence_id += 1
  1922. command = {
  1923. "print": {
  1924. "command": "gcode_line",
  1925. "param": gcode,
  1926. "sequence_id": str(self._sequence_id)
  1927. }
  1928. }
  1929. # Use QoS 1 for reliable delivery (at least once)
  1930. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  1931. logger.debug(f"[{self.serial_number}] Sent G-code: {gcode[:50]}...")
  1932. return True
  1933. def set_bed_temperature(self, target: int) -> bool:
  1934. """Set the bed target temperature.
  1935. Args:
  1936. target: Target temperature in Celsius (0 to turn off)
  1937. Returns:
  1938. True if command was sent, False otherwise
  1939. """
  1940. return self.send_gcode(f"M140 S{target}")
  1941. def set_nozzle_temperature(self, target: int, nozzle: int = 0) -> bool:
  1942. """Set the nozzle target temperature.
  1943. Args:
  1944. target: Target temperature in Celsius (0 to turn off)
  1945. nozzle: Nozzle index (0 for right/default, 1 for left on H2D)
  1946. Returns:
  1947. True if command was sent, False otherwise
  1948. """
  1949. # Use M104 for non-blocking
  1950. # Always use T parameter for H2D compatibility
  1951. result = self.send_gcode(f"M104 T{nozzle} S{target}")
  1952. # H2D quirk: left nozzle (nozzle=1) target isn't reported in MQTT
  1953. # Track it locally so we can display it correctly
  1954. if result and nozzle == 1:
  1955. self.state.temperatures["nozzle_target"] = float(target)
  1956. self.state.temperatures["_nozzle_target_set_time"] = time.time()
  1957. logger.info(f"[{self.serial_number}] Tracking LEFT nozzle target locally: {target}°C")
  1958. return result
  1959. def set_chamber_temperature(self, target: int) -> bool:
  1960. """Set the chamber target temperature.
  1961. Args:
  1962. target: Target temperature in Celsius (0 to turn off heating)
  1963. Returns:
  1964. True if command was sent, False otherwise
  1965. """
  1966. # M141 sets chamber temperature
  1967. result = self.send_gcode(f"M141 S{target}")
  1968. # Track chamber target locally (MQTT reports encoded values that need filtering)
  1969. if result:
  1970. self.state.temperatures["chamber_target"] = float(target)
  1971. self.state.temperatures["_chamber_target_set_time"] = time.time()
  1972. # Update heating state immediately based on new target
  1973. current_temp = self.state.temperatures.get("chamber", 0)
  1974. self.state.temperatures["chamber_heating"] = target > 0 and current_temp < target
  1975. logger.info(f"[{self.serial_number}] Tracking chamber target locally: {target}°C (heating={self.state.temperatures['chamber_heating']})")
  1976. return result
  1977. def set_print_speed(self, mode: int) -> bool:
  1978. """Set the print speed mode.
  1979. Args:
  1980. mode: Speed mode (1=silent, 2=standard, 3=sport, 4=ludicrous)
  1981. Returns:
  1982. True if command was sent, False otherwise
  1983. """
  1984. if not self._client or not self.state.connected:
  1985. logger.warning(f"[{self.serial_number}] Cannot set print speed: not connected")
  1986. return False
  1987. if mode not in (1, 2, 3, 4):
  1988. logger.warning(f"[{self.serial_number}] Invalid speed mode: {mode}")
  1989. return False
  1990. command = {
  1991. "print": {
  1992. "command": "print_speed",
  1993. "param": str(mode),
  1994. "sequence_id": "0"
  1995. }
  1996. }
  1997. self._client.publish(self.topic_publish, json.dumps(command))
  1998. logger.info(f"[{self.serial_number}] Set print speed mode to {mode}")
  1999. return True
  2000. def set_fan_speed(self, fan: int, speed: int) -> bool:
  2001. """Set fan speed.
  2002. Args:
  2003. fan: Fan index (1=part cooling, 2=auxiliary, 3=chamber)
  2004. speed: Speed 0-255 (0=off, 255=full)
  2005. Returns:
  2006. True if command was sent, False otherwise
  2007. """
  2008. if fan not in (1, 2, 3):
  2009. logger.warning(f"[{self.serial_number}] Invalid fan index: {fan}")
  2010. return False
  2011. speed = max(0, min(255, speed)) # Clamp to 0-255
  2012. return self.send_gcode(f"M106 P{fan} S{speed}")
  2013. def set_part_fan(self, speed: int) -> bool:
  2014. """Set part cooling fan speed (0-255)."""
  2015. return self.set_fan_speed(1, speed)
  2016. def set_aux_fan(self, speed: int) -> bool:
  2017. """Set auxiliary fan speed (0-255)."""
  2018. return self.set_fan_speed(2, speed)
  2019. def set_chamber_fan(self, speed: int) -> bool:
  2020. """Set chamber fan speed (0-255)."""
  2021. return self.set_fan_speed(3, speed)
  2022. def set_airduct_mode(self, mode: str) -> bool:
  2023. """Set air conditioning mode (cooling or heating).
  2024. Args:
  2025. mode: "cooling" (modeId=0) or "heating" (modeId=1)
  2026. - Cooling: Suitable for PLA/PETG/TPU, filters and cools chamber air
  2027. - Heating: Suitable for ABS/ASA/PC/PA, circulates and heats chamber air,
  2028. closes top exhaust flap
  2029. Returns:
  2030. True if command was sent, False otherwise
  2031. """
  2032. if not self._client or not self.state.connected:
  2033. logger.warning(f"[{self.serial_number}] Cannot set airduct mode: not connected")
  2034. return False
  2035. self._sequence_id += 1
  2036. mode_id = 0 if mode == "cooling" else 1
  2037. command = {
  2038. "print": {
  2039. "command": "set_airduct",
  2040. "modeId": mode_id,
  2041. "sequence_id": str(self._sequence_id),
  2042. "submode": -1
  2043. }
  2044. }
  2045. # Use QoS 1 for reliable delivery
  2046. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  2047. logger.info(f"[{self.serial_number}] Set airduct mode to {mode} (modeId={mode_id}, seq={self._sequence_id})")
  2048. return True
  2049. def set_chamber_light(self, on: bool) -> bool:
  2050. """Turn chamber light on or off.
  2051. Args:
  2052. on: True to turn on, False to turn off
  2053. Returns:
  2054. True if command was sent, False otherwise
  2055. """
  2056. if not self._client or not self.state.connected:
  2057. logger.warning(f"[{self.serial_number}] Cannot set chamber light: not connected")
  2058. return False
  2059. mode = "on" if on else "off"
  2060. # Control both chamber lights (some printers like H2D have two)
  2061. for led_node in ["chamber_light", "chamber_light2"]:
  2062. self._sequence_id += 1
  2063. command = {
  2064. "system": {
  2065. "command": "ledctrl",
  2066. "led_node": led_node,
  2067. "led_mode": mode,
  2068. "led_on_time": 500,
  2069. "led_off_time": 500,
  2070. "loop_times": 0,
  2071. "interval_time": 0,
  2072. "sequence_id": str(self._sequence_id)
  2073. }
  2074. }
  2075. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  2076. logger.info(f"[{self.serial_number}] Set chamber lights {'on' if on else 'off'} (seq={self._sequence_id})")
  2077. return True
  2078. def select_extruder(self, extruder: int) -> bool:
  2079. """Select the active extruder for dual-nozzle printers (H2D).
  2080. Args:
  2081. extruder: Extruder index (0=right, 1=left for H2D)
  2082. Returns:
  2083. True if command was sent, False otherwise
  2084. """
  2085. if extruder not in (0, 1):
  2086. logger.warning(f"[{self.serial_number}] Invalid extruder: {extruder}")
  2087. return False
  2088. if not self._client or not self.state.connected:
  2089. logger.warning(f"[{self.serial_number}] Cannot switch extruder: not connected")
  2090. return False
  2091. # H2D extruder switching via select_extruder command
  2092. # Command format captured from OrcaSlicer:
  2093. # {"print": {"command": "select_extruder", "extruder_index": 0, "sequence_id": "..."}}
  2094. # extruder_index: 0 = RIGHT, 1 = LEFT
  2095. self._sequence_id += 1
  2096. command = {
  2097. "print": {
  2098. "command": "select_extruder",
  2099. "extruder_index": extruder,
  2100. "sequence_id": str(self._sequence_id)
  2101. }
  2102. }
  2103. self._client.publish(self.topic_publish, json.dumps(command), qos=1)
  2104. logger.info(f"[{self.serial_number}] Sent select_extruder command: extruder_index={extruder} (0=right, 1=left)")
  2105. return True
  2106. def home_axes(self, axes: str = "XYZ") -> bool:
  2107. """Home the specified axes.
  2108. Args:
  2109. axes: Axes to home (e.g., "XYZ", "X", "XY", "Z")
  2110. Returns:
  2111. True if command was sent, False otherwise
  2112. """
  2113. # G28 homes all axes, G28 X Y Z homes specific axes
  2114. axes_param = " ".join(axes.upper())
  2115. return self.send_gcode(f"G28 {axes_param}")
  2116. def move_axis(self, axis: str, distance: float, speed: int = 3000) -> bool:
  2117. """Move an axis by a relative distance.
  2118. Args:
  2119. axis: Axis to move ("X", "Y", or "Z")
  2120. distance: Distance to move in mm (positive or negative)
  2121. speed: Movement speed in mm/min
  2122. Returns:
  2123. True if command was sent, False otherwise
  2124. """
  2125. axis = axis.upper()
  2126. if axis not in ("X", "Y", "Z"):
  2127. logger.warning(f"[{self.serial_number}] Invalid axis: {axis}")
  2128. return False
  2129. # G91 = relative mode, G0 = rapid move, G90 = back to absolute
  2130. gcode = f"G91\nG0 {axis}{distance:.2f} F{speed}\nG90"
  2131. return self.send_gcode(gcode)
  2132. def disable_motors(self) -> bool:
  2133. """Disable all stepper motors.
  2134. Warning: This will cause the printer to lose its position.
  2135. A homing operation will be required before printing.
  2136. Returns:
  2137. True if command was sent, False otherwise
  2138. """
  2139. return self.send_gcode("M18")
  2140. def enable_motors(self) -> bool:
  2141. """Enable all stepper motors.
  2142. Returns:
  2143. True if command was sent, False otherwise
  2144. """
  2145. return self.send_gcode("M17")
  2146. def ams_load_filament(self, tray_id: int, extruder_id: int | None = None) -> bool:
  2147. """Load filament from a specific AMS tray.
  2148. Args:
  2149. tray_id: Global tray ID (0-15 for AMS slots, or 254 for external spool)
  2150. extruder_id: Unused - kept for API compatibility
  2151. Returns:
  2152. True if command was sent, False otherwise
  2153. """
  2154. if not self._client or not self.state.connected:
  2155. logger.warning(f"[{self.serial_number}] Cannot load filament: not connected")
  2156. return False
  2157. # Calculate ams_id and slot_id for logging
  2158. if tray_id == 254:
  2159. ams_id = 255 # External spool
  2160. slot_id = 254
  2161. else:
  2162. ams_id = tray_id // 4 # AMS unit (0, 1, 2, 3...)
  2163. slot_id = tray_id % 4 # Slot within AMS (0, 1, 2, 3)
  2164. # Command format from BambuStudio traffic capture:
  2165. # - No extruder_id field
  2166. # - curr_temp and tar_temp are -1 (not 0)
  2167. self._sequence_id += 1
  2168. command = {
  2169. "print": {
  2170. "command": "ams_change_filament",
  2171. "sequence_id": str(self._sequence_id),
  2172. "ams_id": ams_id,
  2173. "slot_id": slot_id,
  2174. "target": tray_id,
  2175. "curr_temp": -1,
  2176. "tar_temp": -1
  2177. }
  2178. }
  2179. command_json = json.dumps(command)
  2180. logger.info(f"[{self.serial_number}] Publishing ams_change_filament command: {command_json}")
  2181. self._client.publish(self.topic_publish, command_json)
  2182. logger.info(f"[{self.serial_number}] Loading filament from tray {tray_id} (AMS {ams_id} slot {slot_id})")
  2183. # Track this load request for H2D dual-nozzle disambiguation
  2184. # H2D reports only slot number (0-3) in tray_now, so we use our tracked value
  2185. self._last_load_tray_id = tray_id
  2186. self.state.pending_tray_target = tray_id
  2187. logger.info(f"[{self.serial_number}] Set pending_tray_target={tray_id} for H2D disambiguation")
  2188. return True
  2189. def ams_unload_filament(self) -> bool:
  2190. """Unload the currently loaded filament.
  2191. Returns:
  2192. True if command was sent, False otherwise
  2193. """
  2194. if not self._client or not self.state.connected:
  2195. logger.warning(f"[{self.serial_number}] Cannot unload filament: not connected")
  2196. return False
  2197. # Get the currently loaded tray info
  2198. tray_now = self.state.tray_now
  2199. logger.info(f"[{self.serial_number}] Unload requested, tray_now={tray_now}")
  2200. # Determine source ams_id for the unload command
  2201. if tray_now == 255 or tray_now == 254:
  2202. ams_id = 255 # No filament or external spool
  2203. else:
  2204. ams_id = tray_now // 4 # Source AMS
  2205. # Command format from BambuStudio traffic capture:
  2206. # - No extruder_id field
  2207. # - For UNLOAD: curr_temp and tar_temp are the actual nozzle temp (e.g., 210)
  2208. # - slot_id=255 and target=255 for unload
  2209. # Get current nozzle temperature for the unload command
  2210. nozzle_temp = int(self.state.temperatures.get("nozzle", 210))
  2211. if nozzle_temp < 180:
  2212. nozzle_temp = 210 # Default to PLA temp if nozzle is cold
  2213. self._sequence_id += 1
  2214. command = {
  2215. "print": {
  2216. "command": "ams_change_filament",
  2217. "sequence_id": str(self._sequence_id),
  2218. "ams_id": ams_id,
  2219. "slot_id": 255, # 255 = unload marker
  2220. "target": 255, # 255 = unload destination
  2221. "curr_temp": nozzle_temp,
  2222. "tar_temp": nozzle_temp
  2223. }
  2224. }
  2225. command_json = json.dumps(command)
  2226. logger.info(f"[{self.serial_number}] Publishing ams_change_filament (unload) command: {command_json}")
  2227. self._client.publish(self.topic_publish, command_json)
  2228. logger.info(f"[{self.serial_number}] Unloading filament (tray_now was {tray_now})")
  2229. # Clear tracked load request since we're unloading
  2230. self._last_load_tray_id = None
  2231. self.state.pending_tray_target = None
  2232. logger.info(f"[{self.serial_number}] Cleared pending_tray_target (unload)")
  2233. return True
  2234. def ams_control(self, action: str) -> bool:
  2235. """Control AMS operations.
  2236. Args:
  2237. action: "resume", "reset", or "pause"
  2238. Returns:
  2239. True if command was sent, False otherwise
  2240. """
  2241. if not self._client or not self.state.connected:
  2242. logger.warning(f"[{self.serial_number}] Cannot control AMS: not connected")
  2243. return False
  2244. if action not in ("resume", "reset", "pause"):
  2245. logger.warning(f"[{self.serial_number}] Invalid AMS action: {action}")
  2246. return False
  2247. command = {
  2248. "print": {
  2249. "command": "ams_control",
  2250. "param": action,
  2251. "sequence_id": "0"
  2252. }
  2253. }
  2254. self._client.publish(self.topic_publish, json.dumps(command))
  2255. logger.info(f"[{self.serial_number}] AMS control: {action}")
  2256. return True
  2257. def ams_refresh_tray(self, ams_id: int, tray_id: int) -> tuple[bool, str]:
  2258. """Trigger RFID re-read for a specific AMS tray.
  2259. Args:
  2260. ams_id: AMS unit ID (0-3, or 128 for H2D external tray)
  2261. tray_id: Tray ID within the AMS (0-3)
  2262. Returns:
  2263. Tuple of (success, message)
  2264. """
  2265. if not self._client or not self.state.connected:
  2266. logger.warning(f"[{self.serial_number}] Cannot refresh AMS tray: not connected")
  2267. return False, "Printer not connected"
  2268. # Check if filament is currently loaded (tray_now != 255)
  2269. # RFID refresh requires the AMS to move filament, which can't happen if one is loaded
  2270. tray_now = self.state.tray_now
  2271. if tray_now != 255:
  2272. # Decode which tray is loaded for the message
  2273. if tray_now == 254:
  2274. loaded_tray = "external spool"
  2275. elif tray_now >= 0 and tray_now < 128:
  2276. loaded_ams = tray_now // 4
  2277. loaded_slot = tray_now % 4
  2278. loaded_tray = f"AMS {loaded_ams + 1} slot {loaded_slot + 1}"
  2279. else:
  2280. loaded_tray = f"tray {tray_now}"
  2281. logger.warning(f"[{self.serial_number}] Cannot refresh AMS tray: filament loaded from {loaded_tray}")
  2282. return False, f"Please unload filament first. Currently loaded: {loaded_tray}"
  2283. # Use ams_get_rfid command to trigger RFID re-read
  2284. # This command is used by Bambu Studio to re-read the RFID tag
  2285. command = {
  2286. "print": {
  2287. "command": "ams_get_rfid",
  2288. "ams_id": ams_id,
  2289. "slot_id": tray_id,
  2290. "sequence_id": "0"
  2291. }
  2292. }
  2293. self._client.publish(self.topic_publish, json.dumps(command))
  2294. logger.info(f"[{self.serial_number}] Triggering RFID re-read: AMS {ams_id}, slot {tray_id}")
  2295. return True, f"Refreshing AMS {ams_id} tray {tray_id}"
  2296. def ams_set_filament_setting(
  2297. self,
  2298. ams_id: int,
  2299. tray_id: int,
  2300. tray_info_idx: str,
  2301. tray_type: str,
  2302. tray_sub_brands: str,
  2303. tray_color: str,
  2304. nozzle_temp_min: int,
  2305. nozzle_temp_max: int,
  2306. k: float,
  2307. ) -> bool:
  2308. """Set AMS tray filament settings including K (pressure advance) value.
  2309. Args:
  2310. ams_id: AMS unit ID (0-3)
  2311. tray_id: Tray ID within the AMS (0-3)
  2312. tray_info_idx: Filament preset ID (e.g., "GFA00")
  2313. tray_type: Filament type (e.g., "PLA", "PETG")
  2314. tray_sub_brands: Sub-brand name (e.g., "PLA Basic", "PETG HF")
  2315. tray_color: Color in RRGGBBAA hex format (e.g., "FFFF00FF")
  2316. nozzle_temp_min: Minimum nozzle temperature
  2317. nozzle_temp_max: Maximum nozzle temperature
  2318. k: Pressure advance (K) value (e.g., 0.020)
  2319. Returns:
  2320. True if command was sent, False otherwise
  2321. """
  2322. if not self._client or not self.state.connected:
  2323. logger.warning(f"[{self.serial_number}] Cannot set AMS filament setting: not connected")
  2324. return False
  2325. command = {
  2326. "print": {
  2327. "command": "ams_filament_setting",
  2328. "ams_id": ams_id,
  2329. "tray_id": tray_id,
  2330. "tray_info_idx": tray_info_idx,
  2331. "tray_type": tray_type,
  2332. "tray_sub_brands": tray_sub_brands,
  2333. "tray_color": tray_color,
  2334. "nozzle_temp_min": nozzle_temp_min,
  2335. "nozzle_temp_max": nozzle_temp_max,
  2336. "k": k,
  2337. "sequence_id": "0"
  2338. }
  2339. }
  2340. command_json = json.dumps(command)
  2341. logger.info(f"[{self.serial_number}] Publishing ams_filament_setting: AMS {ams_id}, tray {tray_id}, k={k}")
  2342. logger.debug(f"[{self.serial_number}] ams_filament_setting command: {command_json}")
  2343. self._client.publish(self.topic_publish, command_json)
  2344. return True
  2345. def set_timelapse(self, enable: bool) -> bool:
  2346. """Enable or disable timelapse recording.
  2347. Args:
  2348. enable: True to enable, False to disable
  2349. Returns:
  2350. True if command was sent, False otherwise
  2351. """
  2352. if not self._client or not self.state.connected:
  2353. logger.warning(f"[{self.serial_number}] Cannot set timelapse: not connected")
  2354. return False
  2355. command = {
  2356. "pushing": {
  2357. "command": "pushall",
  2358. "sequence_id": "0"
  2359. }
  2360. }
  2361. # First send the timelapse setting
  2362. timelapse_cmd = {
  2363. "print": {
  2364. "command": "gcode_line",
  2365. "param": f"M981 S{1 if enable else 0} P20000",
  2366. "sequence_id": "0"
  2367. }
  2368. }
  2369. self._client.publish(self.topic_publish, json.dumps(timelapse_cmd))
  2370. # Request status update
  2371. self._client.publish(self.topic_publish, json.dumps(command))
  2372. logger.info(f"[{self.serial_number}] Set timelapse {'enabled' if enable else 'disabled'}")
  2373. return True
  2374. def set_liveview(self, enable: bool) -> bool:
  2375. """Enable or disable live view / camera streaming.
  2376. Args:
  2377. enable: True to enable, False to disable
  2378. Returns:
  2379. True if command was sent, False otherwise
  2380. """
  2381. if not self._client or not self.state.connected:
  2382. logger.warning(f"[{self.serial_number}] Cannot set liveview: not connected")
  2383. return False
  2384. command = {
  2385. "xcam": {
  2386. "command": "ipcam_record_set",
  2387. "control": "enable" if enable else "disable",
  2388. "sequence_id": "0"
  2389. }
  2390. }
  2391. self._client.publish(self.topic_publish, json.dumps(command))
  2392. # Request status update
  2393. pushall = {
  2394. "pushing": {
  2395. "command": "pushall",
  2396. "sequence_id": "0"
  2397. }
  2398. }
  2399. self._client.publish(self.topic_publish, json.dumps(pushall))
  2400. logger.info(f"[{self.serial_number}] Set liveview {'enabled' if enable else 'disabled'}")
  2401. return True