DeviceControl.py 69 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705
  1. import asyncio
  2. import datetime
  3. import serial.tools.list_ports
  4. import time, json
  5. from .SerialIns import SerialIns
  6. from utils.SingletonType import SingletonType
  7. from .BaseClass import BaseClass
  8. from sockets import ConnectionManager
  9. from collections import defaultdict
  10. import threading
  11. import settings
  12. from .ProgramItem import ProgramItem
  13. from .capture.module_digicam import DigiCam
  14. from databases import insert_photo_records
  15. from .McuDeviationSet import McuDeviationSet
  16. from .OtherSet import OtherSet
  17. from .DebugUart import DebugUart
  18. from .LineControl import LineControl
  19. import copy
  20. from mcu.capture.smart_shooter_class import SmartShooter
  21. # mcu命令
  22. class DeviceControl(BaseClass, metaclass=SingletonType):
  23. lock = threading.Lock()
  24. def __init__(
  25. self, websocket_manager: ConnectionManager, smart_shooter: SmartShooter = None
  26. ):
  27. super().__init__(websocket_manager=websocket_manager,smart_shooter=smart_shooter)
  28. self.msg_type = "mcu"
  29. self.mcu_deviation_set = McuDeviationSet(self)
  30. self.mcu_other_set = OtherSet(self)
  31. self.debug_uart = DebugUart(self)
  32. self.line_control = LineControl(websocket_manager)
  33. self.m_t = 1
  34. # 0未开始 1进行中 2已结束 99异常
  35. self.action_state = 2
  36. self._mcu_move_state = 0
  37. self.state_camera_motor = 3
  38. self.state_camera_steering = 3
  39. self.state_turntable_steering = 3
  40. self.state_overturn_steering = 3
  41. self.state_move_turntable_steering = 3
  42. self.last_from_mcu_move_respond_data = None
  43. self.camera_motor_speed = 0
  44. self.camera_motor_value = 0
  45. self.init_state = False
  46. self.port_name = ""
  47. self.mcu_exit = False
  48. self.t_n = 0
  49. self.serial_ins = None
  50. self.connected_ports_dict = {} # 已连接的ports
  51. self.p_list = []
  52. self.temp_ports_dict = {}
  53. self.is_running = False
  54. self.is_runn_action = False
  55. self.is_stop_action = False
  56. self.connect_state = False
  57. self.is_init_while = False
  58. self.device_name_dict = {
  59. "camera_steering": 0,
  60. "camera_high_motor": 1,
  61. "turntable_steering": 2,
  62. "overturn_steering": 3,
  63. "laser_position": 4,
  64. "buzzer": 5,
  65. "split": 6,
  66. "turntable_position_motor": 7,
  67. "mp3_player": 8,
  68. "mcu": 99,
  69. }
  70. # 最近的mcu基础信息,用于获取数据状态检查
  71. self.last_mcu_info_data = {
  72. "num": 0,
  73. "time": time.time(),
  74. "data": None,
  75. }
  76. # 最近的mcu的其他配置
  77. self.last_mcu_other_info_data = {
  78. "num": 0,
  79. "time": time.time(),
  80. "data": {},
  81. }
  82. self.command = {
  83. "to_device_move": 1, # 设备运动
  84. "to_init_device": 2, # 初始化设备
  85. "to_deal_other_device": 3, # 处理其他设备
  86. "get_all_info": 29, # 获取所有信息
  87. "set_deviation": 40, # 设置偏移量
  88. "get_deviation": 41, # 读取偏移量
  89. "signal_forwarding": 91, # 信号转发处理
  90. "signal_forwarding_return": 92, # 信号转发返回
  91. "get_other_info": 44, # 获取其他信息
  92. "open_rgb_led": 43, ## RGB灯的处理与通讯
  93. "set_other_info": 45, # 设置其他信息
  94. "query_remote_control_battery": 47, # 查询遥控器电量
  95. "set_turntable_mode": 48, # 设置转盘通讯方式 1、串口、2、无线、3 混合
  96. "stop_mcu": 93, # 停止运行mcu
  97. }
  98. # self.window = window
  99. self.last_push_time = defaultdict(float)
  100. self.is_running = False
  101. self.is_wait_connect = False # 等待链接
  102. self.send_data_queue = [] # 发送队列
  103. # self.lock = Lock()
  104. # 是否是刚进行完初始化;首次初始化,需要运动到指定第一个指定位置
  105. self.is_just_init_time = False
  106. # self.init()
  107. # ===========注册命令函数============
  108. self.deal_code_func_dict = {
  109. 29: self.get_from_mcu_base_info, # 获取基本情况
  110. 32: self.get_from_mcu_button, # 获取按键信息
  111. 42: self.get_from_mcu_deviation_info, # 获取偏移量信息
  112. 44: self.get_from_mcu_other_info, # 获取其他配置参数
  113. 90: self.get_from_mcu_connect_info, # 获取链接电脑信号
  114. 92: self.get_from_mcu_move_respond_data, # 获取MCU响应
  115. 100: self.print_mcu_error_data, # 打印下位机的错误内容
  116. }
  117. async def sendCommand(self,command):
  118. await asyncio.sleep(0.01)
  119. loop = asyncio.get_event_loop()
  120. loop.create_task(self.debug_uart.set(command), name="sendCommand2")
  121. async def getMcuOtherInfo(self):
  122. await asyncio.sleep(0.01)
  123. self.get_other_info()
  124. async def setMcuOtherInfo(self,data):
  125. await asyncio.sleep(0.01)
  126. for k, v in data.items():
  127. print("k:{},v:{}".format(k, v))
  128. data[k] = self.setOtherMaxMinValue(k, v)
  129. self.set_to_mcu_other_info(data)
  130. self.msg_type = "set_other_mcu_info"
  131. self.sendSocketMessage(
  132. code=0,
  133. msg="设置mcu其他配置信息完成",
  134. device_status=2,
  135. )
  136. self.msg_type = "mcu"
  137. def setOtherMaxMinValue(self,item, value):
  138. value = int(value)
  139. parameter_limits = {
  140. "is_auto_send_base_info": (0, 10000),
  141. "is_move_retry": (0, 10000),
  142. "is_data_response": (0, 10000),
  143. "low_speed": (0, 10000), # 假设低速范围是0到1000
  144. "is_test": (0, 10000),
  145. "to_init_mode": (0, 10000),
  146. "turntable_move_to_init_mode": (0, 10000), # 假设转盘通讯方式范围是0到3
  147. "led_count": (0, 10000), # 假设LED数量范围是0到10
  148. "turntable_steering_angle_ratio": (0, 10000), # 假设角度比例范围是0到100
  149. "is_manual_check": (0, 10000),
  150. "camera_steering_angle_ratio": (0, 100000), # 假设角度比例范围是0到100
  151. "is_auto_motor_to_disable": (0, 10000),
  152. "diff_dir": (0, 10000),
  153. "is_auto_send_pos_info": (0, 10000),
  154. "is_dog": (0, 10000),
  155. "has_been_set_motor_config": (0, 10000),
  156. }
  157. # 获取参数的最小值和最大值
  158. min_value, max_value = parameter_limits.get(item, (0, 10000))
  159. value = max(min_value, min(max_value, value))
  160. return value
  161. async def initDevice(self,is_force=False):
  162. if not self.is_running:
  163. self.sendSocketMessage(
  164. code=1, msg="mcu设备未连接,请先连接设备", device_status=0
  165. )
  166. return False
  167. if self.init_state ==True:
  168. print("已经初始化过,请勿重复初始化")
  169. self.sendSocketMessage(msg="设备初始化完成", device_status=2)
  170. return False
  171. self.serial_ins.clearn_flush()
  172. self.to_init_device_origin_point(device_name="mcu",is_force=is_force)
  173. print("MCU 开始循环~")
  174. while 1:
  175. await asyncio.sleep(0.01)
  176. if not self.serial_ins or not self.connect_state:
  177. break
  178. try:
  179. # print("mcu send_cmd")
  180. self.send_cmd()
  181. # time.sleep(0.01)
  182. if not self.get_basic_info_mcu():
  183. pass
  184. # self.close_other_window()
  185. except BaseException as e:
  186. print("121231298908", e)
  187. break
  188. self.is_running = False
  189. self.connect_state = False
  190. print("MCU 循环退出~")
  191. print("串口未连接,请检查")
  192. self.sendSocketMessage(
  193. code=1, msg="MCU串口未连接,请检查", device_status=-1
  194. )
  195. message = {"_type": "show_info", "plugins_mode": "mcu", "data": "MCU 连接失败"}
  196. self.sendSocketMessage(code=1, msg="MCU 连接失败", data=message,device_status=-1)
  197. self.close_connect()
  198. async def initControlLine(self,serial_handle):
  199. '''实例化有线控制设备'''
  200. self.line_control.connect_state = True
  201. self.line_control.serial_ins = serial_handle
  202. await self.line_control.run()
  203. # if self.init_state == True:
  204. # print("已经初始化过,请勿重复初始化")
  205. # self.sendSocketMessage(msg="设备初始化完成", device_status=2)
  206. # return False
  207. # self.serial_ins.clearn_flush()
  208. # self.to_init_device_origin_point(device_name="mcu", is_force=is_force)
  209. # print("MCU 开始循环~")
  210. # while 1:
  211. # await asyncio.sleep(0.01)
  212. # if not self.serial_ins or not self.connect_state:
  213. # break
  214. # try:
  215. # # print("mcu send_cmd")
  216. # self.send_cmd()
  217. # # time.sleep(0.01)
  218. # self.get_basic_info_mcu()
  219. # # self.close_other_window()
  220. # except BaseException as e:
  221. # print("121231298908", e)
  222. # break
  223. # self.is_running = False
  224. # self.connect_state = False
  225. # print("MCU 循环退出~")
  226. # # self.sign_data.emit(
  227. # # {"_type": "show_info", "plugins_mode": "mcu", "data": "MCU 连接失败"}
  228. # # )
  229. # message = {"_type": "show_info", "plugins_mode": "mcu", "data": "MCU 连接失败"}
  230. # self.sendSocketMessage(
  231. # code=1, msg="MCU 连接失败", data=message, device_status=-1
  232. # )
  233. # self.close_connect()
  234. def stop_mcu(self):
  235. buf = [self.command["stop_mcu"]]
  236. buf.extend(self.encapsulation_data(data=1, len_data=1))
  237. self.add_send_data_queue(buf)
  238. # 设置转盘通讯方式 1、串口、2、无线、3 混合
  239. def to_set_turntable_mode(self, mode=1):
  240. buf = [self.command["set_turntable_mode"]]
  241. buf.extend(self.encapsulation_data(data=mode, len_data=1))
  242. self.add_send_data_queue(buf)
  243. def query_remote_control_battery(self):
  244. """查询遥控器电量"""
  245. buf = [self.command["query_remote_control_battery"]]
  246. buf.extend(self.encapsulation_data(data=1, len_data=1))
  247. buf.extend(self.encapsulation_data(data=1, len_data=1))
  248. buf.extend(self.encapsulation_data(data=1, len_data=1))
  249. self.add_send_data_queue(buf)
  250. def encapsulation_data(self, data, len_data, data_magnification=1):
  251. # data_magnification 数据放大倍数,或缩小倍数,默认为1
  252. data = int(data * data_magnification)
  253. if len_data == 1:
  254. return [0xFF & data]
  255. elif len_data == 2:
  256. return [0xFF & data >> 8, 0xFF & data]
  257. elif len_data == 4:
  258. return [0xFF & data >> 24, 0xFF & data >> 16, 0xFF & data >> 8, 0xFF & data]
  259. def open_rgb_led(
  260. self,
  261. color_name,
  262. led_command=1,
  263. brightness=80,
  264. enable=True,
  265. mode="loop",
  266. times=2,
  267. interval=0.1,
  268. ):
  269. color_name_value = {
  270. "红色": (156, 6, 3),
  271. "黄色": (255, 206, 25),
  272. "绿色": (0, 128, 0),
  273. "蓝色": (0, 25, 255),
  274. "白色": (255, 250, 227),
  275. }
  276. if color_name in color_name_value:
  277. buf = [self.command["open_rgb_led"]]
  278. buf.append(1 if enable else 0)
  279. buf.append(led_command)
  280. buf.extend(color_name_value[color_name])
  281. buf.extend(
  282. [
  283. brightness,
  284. 1 if mode == "loop" else 2,
  285. times,
  286. int(interval * 10),
  287. ]
  288. )
  289. self.add_send_data_queue(buf)
  290. async def getDeviationInfo(self):
  291. await asyncio.sleep(0.01)
  292. try:
  293. # 发送获取偏移量
  294. data = [self.command["get_deviation"], 1]
  295. self.add_send_data_queue(data)
  296. # if self.serial_ins:
  297. # self.serial_ins.write_cmd(data)
  298. print("发送获取偏移量")
  299. except Exception as e:
  300. print(e)
  301. print("getDeviationInfo", "暂未获取到self.command")
  302. def set_deviation(self, device_name, _type=0, deviation=0):
  303. # turntable----0 angle_ratio 1 turntable_steering_deviation
  304. # overturn ----0 middle 1 high
  305. if device_name == "camera_high_motor":
  306. deviation = deviation / 10 # deviation 原单位为mm
  307. if device_name == "turntable_position_motor":
  308. deviation = deviation / 10 # deviation 原单位为mm
  309. if device_name == "camera_steering":
  310. pass
  311. if device_name == "turntable_steering":
  312. pass
  313. if device_name == "overturn_steering":
  314. pass
  315. device_id = self.device_name_dict[device_name]
  316. _dir = 1 if deviation >= 0 else 0
  317. deviation = int(abs(deviation * 10))
  318. data = [
  319. self.command["set_deviation"],
  320. device_id,
  321. _type,
  322. _dir,
  323. 0xFF & deviation >> 8,
  324. 0xFF & deviation,
  325. ]
  326. self.add_send_data_queue(data)
  327. # self.controlDevice(device_name, deviation)
  328. async def set_deviation_cmd(self,value,action_name,type):
  329. await asyncio.sleep(0.01)
  330. name_sets = [
  331. "相机电机", # min 0 max 400,步长1
  332. "相机舵机", # min -40 max 40,步长0.1
  333. "转盘舵机", # min -720 max 720,步长1
  334. "转盘前后电机", # min 0 max 950,步长1
  335. "翻板舵机中位", # min 0 max 180,步长0.5
  336. "翻板舵机高位", # min 0 max 180,步长0.5
  337. "翻板舵机上升速度", # min 1 max 10,步长1
  338. "翻板舵机下降速度", # min 1 max 10,步长1
  339. ]
  340. if action_name not in name_sets:
  341. self.msg_type = f"{type}_deviation"
  342. self.sendSocketMessage(msg="设置参数有误,请检查", device_status=0,code=1)
  343. self.msg_type = "mcu"
  344. return
  345. # 发送获取偏移量
  346. if type == "move":
  347. self.mcu_deviation_set.change_value(value, action_name)
  348. elif type == "set":
  349. self.mcu_deviation_set.set_deviation(action_name)
  350. self.msg_type = f"{type}_deviation"
  351. self.sendSocketMessage(msg=f"{action_name} 设置成功", device_status=2,code=0)
  352. self.msg_type = "mcu"
  353. def get_other_info(self):
  354. # 发送获取偏移量
  355. data = [self.command["get_other_info"], 1]
  356. self.add_send_data_queue(data)
  357. print("发送获取其他信息")
  358. def add_send_data_queue(self, data):
  359. self.lock.acquire()
  360. if self.serial_ins:
  361. print("send_data_queue append :{}".format(data))
  362. self.send_data_queue.append(data)
  363. self.lock.release()
  364. async def send_all_cmd(self):
  365. await asyncio.sleep(0.001)
  366. while True:
  367. if self.send_data_queue:
  368. self.sendSocketMessage(msg="正在发送命令", device_status=1)
  369. data = self.send_data_queue.pop(0)
  370. self.serial_ins.write_cmd(data)
  371. self.sendSocketMessage(msg="命令发送完成", device_status=2)
  372. else:
  373. break
  374. def send_cmd(self):
  375. self.lock.acquire()
  376. if self.send_data_queue:
  377. self.sendSocketMessage(msg="正在发送命令", device_status=1)
  378. data = self.send_data_queue.pop(0)
  379. self.serial_ins.write_cmd(data)
  380. self.sendSocketMessage(msg="命令发送完成", device_status=2)
  381. # else:
  382. # self.t_n += 1
  383. # # 加大发送获取基础数据的时间间隔
  384. # # 默认为0.01秒一个循环,每隔1.5秒发送数据
  385. # if self.t_n > 150:
  386. # self.t_n = 0
  387. # data = [self.command["get_all_info"], 1]
  388. # self.serial_ins.write_cmd(data)
  389. self.lock.release()
  390. def print_mcu_error_data(self, receive_data):
  391. # 扫码数据
  392. try:
  393. data = receive_data[1:].decode()
  394. if "设备初始化完成" in data:
  395. self.init_state = True
  396. self.sendSocketMessage(msg=data, device_status=2)
  397. print("115 print_mcu_error_data:", data)
  398. except BaseException as e:
  399. print("117 error {}".format(e))
  400. return
  401. def get_from_mcu_move_respond_data(self, receive_data):
  402. self.last_from_mcu_move_respond_data = receive_data
  403. def get_from_mcu_connect_info(self, receive_data):
  404. connect_flag = receive_data[1]
  405. device_id = receive_data[2]
  406. try:
  407. mcu_has_been_set = receive_data[6] # 设备是否有初始化 ,1 表示已初始化
  408. except:
  409. mcu_has_been_set = 99 # 未知状态
  410. # self.self_sign.emit({"type": "connect_sign", "data": connect_flag})
  411. message = {"type": "connect_sign", "data": connect_flag}
  412. self.sendSocketMessage(msg="接收链接信息", data=message)
  413. print("接收链接信息")
  414. return
  415. def to_init_device_origin_point(self, device_name, is_force=False):
  416. device_id = self.device_name_dict[device_name]
  417. cmd = 2
  418. data = [cmd, device_id, 0 if is_force is False else 1]
  419. self.open_rgb_led(color_name="红色")
  420. self.add_send_data_queue(data)
  421. if device_name == "mcu":
  422. # 重置初始化标记为 从未初始化
  423. self.is_just_init_time = False
  424. return True
  425. # 设置其他信息
  426. def set_to_mcu_other_info(self, data):
  427. is_auto_send_base_info = data["is_auto_send_base_info"]
  428. is_move_retry = data["is_move_retry"]
  429. is_data_response = data["is_data_response"]
  430. low_speed = data["low_speed"]
  431. is_test = data["is_test"]
  432. to_init_mode = data["to_init_mode"]
  433. turntable_move_to_init_mode = data["turntable_move_to_init_mode"]
  434. led_count = data["led_count"]
  435. turntable_steering_angle_ratio = data["turntable_steering_angle_ratio"]
  436. is_manual_check = data["is_manual_check"]
  437. camera_steering_angle_ratio = data["camera_steering_angle_ratio"]
  438. is_auto_motor_to_disable = data["is_auto_motor_to_disable"]
  439. diff_dir = data["diff_dir"]
  440. is_auto_send_pos_info = data["is_auto_send_pos_info"]
  441. is_dog = data["is_dog"]
  442. has_been_set_motor_config = data["has_been_set_motor_config"]
  443. buf = [self.command["set_other_info"]]
  444. buf.extend(self.encapsulation_data(data=is_auto_send_base_info, len_data=1))
  445. buf.extend(self.encapsulation_data(data=is_move_retry, len_data=1))
  446. buf.extend(self.encapsulation_data(data=is_data_response, len_data=1))
  447. buf.extend(self.encapsulation_data(data=low_speed, len_data=2))
  448. buf.extend(self.encapsulation_data(data=is_test, len_data=1))
  449. buf.extend(self.encapsulation_data(data=to_init_mode, len_data=1))
  450. buf.extend(
  451. self.encapsulation_data(data=turntable_move_to_init_mode, len_data=1)
  452. )
  453. buf.extend(self.encapsulation_data(data=led_count, len_data=2))
  454. buf.extend(
  455. self.encapsulation_data(data=turntable_steering_angle_ratio, len_data=2)
  456. )
  457. buf.extend(self.encapsulation_data(data=is_manual_check, len_data=1))
  458. buf.extend(
  459. self.encapsulation_data(data=camera_steering_angle_ratio, len_data=4)
  460. )
  461. buf.extend(self.encapsulation_data(data=is_auto_motor_to_disable, len_data=1))
  462. buf.extend(self.encapsulation_data(data=diff_dir, len_data=1))
  463. buf.extend(self.encapsulation_data(data=is_auto_send_pos_info, len_data=1))
  464. buf.extend(self.encapsulation_data(data=is_dog, len_data=1))
  465. buf.extend(self.encapsulation_data(data=has_been_set_motor_config, len_data=1))
  466. self.add_send_data_queue(buf)
  467. async def cleanAllReceiveData(self):
  468. while True:
  469. await asyncio.sleep(0.001)
  470. receive_data = self.serial_ins.read_cmd(out_time=1)
  471. if not receive_data:
  472. break
  473. def get_basic_info_mcu(self):
  474. receive_data = self.serial_ins.read_cmd(out_time=1)
  475. if receive_data is False:
  476. print("------------------------------------------------4657564654")
  477. self.connect_state = False
  478. return False
  479. if not receive_data:
  480. return False
  481. # print("receive_data", receive_data)
  482. # 数据 结构 command,按命令解析
  483. # command 0(9) 相机高度1-2 相机角度3-4 转盘角度5-6 灯光状态7 激光指示器状态8,运行状态9
  484. command = receive_data[0]
  485. # receive_data_temp = receive_data[2:]
  486. # receive_data_temp_text = " ".join([hex(x) for x in receive_data_temp])
  487. # print("command",command)
  488. # print("receive_data", receive_data_temp_text)
  489. if command in self.deal_code_func_dict:
  490. self.deal_code_func_dict[command](receive_data)
  491. def get_from_mcu_button(self, receive_data):
  492. button_name = receive_data[1]
  493. self.deal_mcu_button(button_name)
  494. def deal_mcu_button(self, button_name):
  495. # 防止重复点击
  496. s = time.time()
  497. if s - self.last_push_time[button_name] > 0.1:
  498. self.last_push_time[button_name] = s
  499. # print("button_name", button_name)
  500. else:
  501. self.last_push_time[button_name] = s
  502. return
  503. if button_name == 1:
  504. # 自动执行全部
  505. # self.sign_data.emit(
  506. # {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_1"}
  507. # )
  508. message = {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_1"}
  509. self.sendSocketMessage(code=0, msg="", data=message)
  510. if button_name == 2:
  511. # self.sign_data.emit(
  512. # {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_2"}
  513. # )
  514. message = {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_2"}
  515. self.sendSocketMessage(code=0, msg="", data=message)
  516. if button_name == 3:
  517. # self.sign_data.emit(
  518. # {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_3"}
  519. # )
  520. message = {"_type": "mcu_button", "plugins_mode": "mcu", "data": "todo_3"}
  521. self.sendSocketMessage(code=0, msg="", data=message)
  522. # 获取偏移量信息
  523. def get_from_mcu_deviation_info(self, receive_data):
  524. if len(receive_data) == 18:
  525. camera_high_motor_deviation_dir = receive_data[1]
  526. camera_high_motor_deviation = receive_data[2] << 8 | receive_data[3]
  527. camera_high_motor_deviation = (
  528. camera_high_motor_deviation * -1
  529. if camera_high_motor_deviation_dir == 0
  530. else camera_high_motor_deviation
  531. )
  532. camera_steering_deviation_dir = receive_data[4]
  533. camera_steering_deviation = (receive_data[5] << 8 | receive_data[6]) * 0.1
  534. camera_steering_deviation = (
  535. camera_steering_deviation * -1
  536. if camera_steering_deviation_dir == 0
  537. else camera_steering_deviation
  538. )
  539. turntable_steering_deviation_dir = receive_data[7]
  540. turntable_steering_deviation = (
  541. receive_data[8] << 8 | receive_data[9]
  542. ) * 0.1
  543. turntable_steering_deviation = (
  544. turntable_steering_deviation * -1
  545. if turntable_steering_deviation_dir == 0
  546. else turntable_steering_deviation
  547. )
  548. overturn_steering_middle_dir = receive_data[10]
  549. overturn_steering_middle = (receive_data[11] << 8 | receive_data[12]) * 0.1
  550. overturn_steering_middle = (
  551. overturn_steering_middle * -1
  552. if overturn_steering_middle_dir == 0
  553. else overturn_steering_middle
  554. )
  555. overturn_steering_high_dir = receive_data[13]
  556. overturn_steering_high = (receive_data[14] << 8 | receive_data[15]) * 0.1
  557. overturn_steering_high = (
  558. overturn_steering_middle * -1
  559. if overturn_steering_high_dir == 0
  560. else overturn_steering_high
  561. )
  562. overturn_steering_up_speed = receive_data[16]
  563. overturn_steering_down_speed = receive_data[17]
  564. # self.sign_data.emit(
  565. # {
  566. # "_type": "get_deviation_data",
  567. # "plugins_mode": "mcu",
  568. # "data": {
  569. # "camera_high_motor_deviation": camera_high_motor_deviation,
  570. # "camera_steering_deviation": camera_steering_deviation,
  571. # "turntable_steering_deviation": turntable_steering_deviation,
  572. # "overturn_steering_middle": overturn_steering_middle,
  573. # "overturn_steering_high": overturn_steering_high,
  574. # "overturn_steering_up_speed": overturn_steering_up_speed,
  575. # "overturn_steering_down_speed": overturn_steering_down_speed,
  576. # },
  577. # }
  578. # )
  579. # message = {
  580. # "_type": "get_deviation_data",
  581. # "plugins_mode": "mcu",
  582. # "data": {
  583. # "camera_high_motor_deviation": camera_high_motor_deviation,
  584. # "camera_steering_deviation": camera_steering_deviation,
  585. # "turntable_steering_deviation": turntable_steering_deviation,
  586. # "overturn_steering_middle": overturn_steering_middle,
  587. # "overturn_steering_high": overturn_steering_high,
  588. # "overturn_steering_up_speed": overturn_steering_up_speed,
  589. # "overturn_steering_down_speed": overturn_steering_down_speed,
  590. # },
  591. # }
  592. get_deviation_data = {
  593. "camera_high_motor_deviation": camera_high_motor_deviation,
  594. "camera_steering_deviation": camera_steering_deviation,
  595. "turntable_steering_deviation": turntable_steering_deviation,
  596. "turntable_front_end_deviation": 300,
  597. "overturn_steering_middle": overturn_steering_middle,
  598. "overturn_steering_high": overturn_steering_high,
  599. "overturn_steering_up_speed": overturn_steering_up_speed,
  600. "overturn_steering_down_speed": overturn_steering_down_speed,
  601. }
  602. self.initDeviationInfo(get_deviation_data)
  603. self.msg_type = "get_deviation_data"
  604. self.sendSocketMessage(msg="接收偏移量信息", data=get_deviation_data)
  605. self.msg_type = "mcu"
  606. print("接收偏移量信息")
  607. return
  608. def initDeviationInfo(self, get_deviation_data):
  609. # 初始化偏移量信息
  610. camera_high_motor_deviation = get_deviation_data["camera_high_motor_deviation"]
  611. camera_steering_deviation = get_deviation_data["camera_steering_deviation"]
  612. turntable_steering_deviation = get_deviation_data["turntable_steering_deviation"]
  613. overturn_steering_middle=get_deviation_data["overturn_steering_middle"]
  614. overturn_steering_high = get_deviation_data["overturn_steering_high"]
  615. overturn_steering_up_speed = get_deviation_data["overturn_steering_up_speed"]
  616. overturn_steering_down_speed = get_deviation_data["overturn_steering_down_speed"]
  617. self.mcu_deviation_set.last_value["相机电机"] = camera_high_motor_deviation
  618. self.mcu_deviation_set.last_value["相机舵机"] = camera_steering_deviation
  619. self.mcu_deviation_set.last_value["转盘舵机"] = turntable_steering_deviation
  620. self.mcu_deviation_set.last_value["翻板舵机中位"] = overturn_steering_middle
  621. self.mcu_deviation_set.last_value["翻板舵机高位"] = overturn_steering_high
  622. self.mcu_deviation_set.last_value["翻板舵机上升速度"] = (
  623. overturn_steering_up_speed
  624. )
  625. self.mcu_deviation_set.last_value["翻板舵机下降速度"] = (
  626. overturn_steering_down_speed
  627. )
  628. # 获取其他信息
  629. def get_from_mcu_other_info(self, receive_data):
  630. # TODO 已完成设置
  631. is_auto_send_base_info = self.get_data_from_receive_data(
  632. receive_data=receive_data, start=1, len_data=1
  633. )
  634. is_move_retry = self.get_data_from_receive_data(
  635. receive_data=receive_data, start=2, len_data=1
  636. )
  637. is_data_response = self.get_data_from_receive_data(
  638. receive_data=receive_data, start=3, len_data=1
  639. )
  640. low_speed = self.get_data_from_receive_data(
  641. receive_data=receive_data, start=4, len_data=2
  642. )
  643. is_test = self.get_data_from_receive_data(
  644. receive_data=receive_data, start=6, len_data=1
  645. )
  646. to_init_mode = self.get_data_from_receive_data(
  647. receive_data=receive_data, start=7, len_data=1
  648. )
  649. turntable_move_to_init_mode = self.get_data_from_receive_data(
  650. receive_data=receive_data, start=8, len_data=1
  651. )
  652. led_count = self.get_data_from_receive_data(
  653. receive_data=receive_data, start=9, len_data=2
  654. )
  655. turntable_steering_angle_ratio = self.get_data_from_receive_data(
  656. receive_data=receive_data, start=11, len_data=2
  657. )
  658. is_manual_check = self.get_data_from_receive_data(
  659. receive_data=receive_data, start=13, len_data=1
  660. )
  661. camera_steering_angle_ratio = self.get_data_from_receive_data(
  662. receive_data=receive_data, start=14, len_data=4
  663. )
  664. is_auto_motor_to_disable = self.get_data_from_receive_data(
  665. receive_data=receive_data, start=18, len_data=1
  666. )
  667. diff_dir = self.get_data_from_receive_data(
  668. receive_data=receive_data, start=19, len_data=1
  669. )
  670. is_auto_send_pos_info = self.get_data_from_receive_data(
  671. receive_data=receive_data, start=20, len_data=1
  672. )
  673. is_dog = self.get_data_from_receive_data(
  674. receive_data=receive_data, start=21, len_data=1
  675. )
  676. has_been_set_motor_config = self.get_data_from_receive_data(
  677. receive_data=receive_data, start=22, len_data=1
  678. )
  679. self.last_mcu_other_info_data["data"] = {
  680. "is_auto_send_base_info": is_auto_send_base_info,
  681. "is_move_retry": is_move_retry,
  682. "is_data_response": is_data_response,
  683. "low_speed": low_speed,
  684. "is_test": is_test,
  685. "to_init_mode": to_init_mode,
  686. "turntable_move_to_init_mode": turntable_move_to_init_mode,
  687. "led_count": led_count,
  688. "turntable_steering_angle_ratio": turntable_steering_angle_ratio,
  689. "is_manual_check": is_manual_check,
  690. "camera_steering_angle_ratio": camera_steering_angle_ratio,
  691. "is_auto_motor_to_disable": is_auto_motor_to_disable,
  692. "diff_dir": diff_dir,
  693. "is_auto_send_pos_info": is_auto_send_pos_info,
  694. "is_dog": is_dog,
  695. "has_been_set_motor_config": has_been_set_motor_config,
  696. }
  697. self.last_mcu_other_info_data["time"] = time.time()
  698. self.last_mcu_other_info_data["num"] += 1
  699. for k, v in self.last_mcu_other_info_data["data"].items():
  700. print("k:{},v:{}".format(k, v))
  701. self.msg_type = "get_mcu_other_info"
  702. self.sendSocketMessage(
  703. code=0,
  704. msg="获取mcu其他配置信息",
  705. device_status=2,
  706. data=self.last_mcu_other_info_data["data"],
  707. )
  708. self.msg_type = "mcu"
  709. def get_data_from_receive_data(
  710. self, receive_data, start, len_data, data_magnification=1
  711. ):
  712. # data_magnification 数据放大倍数,或缩小倍数,默认为1
  713. try:
  714. if len_data == 1:
  715. data = receive_data[start]
  716. return data * data_magnification
  717. elif len_data == 2:
  718. data = receive_data[start] << 8 | receive_data[start + 1]
  719. return data * data_magnification
  720. elif len_data == 4:
  721. data = (
  722. receive_data[start] << 24
  723. | receive_data[start + 1] << 16
  724. | receive_data[start + 2] << 8
  725. | receive_data[start + 3]
  726. )
  727. return data * data_magnification
  728. return None
  729. except:
  730. return None
  731. def get_from_mcu_base_info(self, receive_data):
  732. # 数据缓存
  733. self.last_mcu_info_data["time"] = time.time()
  734. self.last_mcu_info_data["num"] += 1
  735. # print("last_mcu_info_data:{}".format(self.last_mcu_info_data["time"]))
  736. self.state_camera_motor = 3
  737. self.state_camera_steering = 3
  738. self.state_turntable_steering = 3
  739. self.state_overturn_steering = 3
  740. self.state_move_turntable_steering = 3
  741. if len(receive_data) == 7:
  742. self.m_t = 1
  743. laser_state = receive_data[1]
  744. self.state_camera_motor = receive_data[2]
  745. self.state_camera_steering = receive_data[3]
  746. self.state_turntable_steering = receive_data[4]
  747. self.state_overturn_steering = receive_data[5]
  748. flag = receive_data[6]
  749. message = {
  750. "_type": "show_mcu_info",
  751. "plugins_mode": "mcu",
  752. "data": "激光状态;{laser_state},相机高度状态:{state_camera_motor},相机角度状态:{state_camera_steering},转盘状态:{state_turntable_steering},翻板状态:{state_overturn_steering},flag:{flag}".format(
  753. laser_state=laser_state,
  754. state_camera_motor=self.state_camera_motor,
  755. state_camera_steering=self.state_camera_steering,
  756. state_turntable_steering=self.state_turntable_steering,
  757. state_overturn_steering=self.state_overturn_steering,
  758. flag=flag,
  759. ),
  760. "data_state": {
  761. "state_camera_motor": self.state_camera_motor,
  762. "state_camera_steering": self.state_camera_steering,
  763. "state_turntable_steering": self.state_turntable_steering,
  764. "state_overturn_steering": self.state_overturn_steering,
  765. },
  766. }
  767. self.sendSocketMessage(msg="获取mcu设备运行状态信息", data=message)
  768. # print("转盘:{},时间:{}".format(self.state_turntable_steering, time.time()))
  769. if len(receive_data) == 8:
  770. self.m_t = 2
  771. laser_state = receive_data[1]
  772. self.state_camera_motor = receive_data[2]
  773. self.state_camera_steering = receive_data[3]
  774. self.state_turntable_steering = receive_data[4]
  775. self.state_overturn_steering = receive_data[5]
  776. self.state_move_turntable_steering = receive_data[6]
  777. flag = receive_data[7]
  778. message = {
  779. "_type": "show_mcu_info",
  780. "plugins_mode": "mcu",
  781. "data": "激光状态;{laser_state},相机高度状态:{state_camera_motor},相机角度状态:{state_camera_steering},转盘状态:{state_turntable_steering},转盘前后移动状态:{state_move_turntable_steering},翻板状态:{state_overturn_steering},flag:{flag}".format(
  782. laser_state=laser_state,
  783. state_camera_motor=self.state_camera_motor,
  784. state_camera_steering=self.state_camera_steering,
  785. state_turntable_steering=self.state_turntable_steering,
  786. state_overturn_steering=self.state_overturn_steering,
  787. state_move_turntable_steering=self.state_move_turntable_steering,
  788. flag=flag,
  789. ),
  790. "data_state": {
  791. "state_camera_motor": self.state_camera_motor,
  792. "state_camera_steering": self.state_camera_steering,
  793. "state_turntable_steering": self.state_turntable_steering,
  794. "state_overturn_steering": self.state_overturn_steering,
  795. "state_move_turntable_steering": self.state_move_turntable_steering,
  796. },
  797. }
  798. # if self.state_camera_motor
  799. if all(value == 2 for value in [self.state_camera_motor, self.state_camera_steering, self.state_turntable_steering, self.state_overturn_steering]):
  800. self.init_state = True
  801. self.sendSocketMessage(msg="设备初始化完成", device_status=2)
  802. self.sendSocketMessage(msg="获取mcu设备运行状态信息", data=message)
  803. # 检查是否成功初始化
  804. if self.is_just_init_time is False:
  805. if self.mcu_move_state == 2:
  806. self.is_just_init_time = True
  807. print("is_just_init_time")
  808. message = {
  809. "_type": "is_just_init_time",
  810. "plugins_mode": "mcu",
  811. "data": "",
  812. }
  813. self.sendSocketMessage(msg="检查设备初始化", data=message)
  814. return
  815. # def init(self, sign_data):
  816. # if sign_data["type"] == "connect_sign":
  817. # if sign_data["data"] == 0:
  818. # self.is_running = False
  819. # else:
  820. # self.is_running = True
  821. def scan_serial_port(self) -> dict:
  822. # 获取所有可用串口列表
  823. ports_dict = {}
  824. ports = serial.tools.list_ports.comports()
  825. # 遍历所有端口并打印信息
  826. for port in ports:
  827. if "CH340" in port.description:
  828. ports_dict[port.name] = {
  829. "name": port.name,
  830. "device": port.device,
  831. "description": port.description,
  832. "hwid": port.hwid,
  833. "manufacturer": port.manufacturer,
  834. "product": port.product,
  835. "serial_number": port.serial_number,
  836. }
  837. if len(ports_dict) <= 0:
  838. return {}
  839. return ports_dict
  840. def remove_port(self, port_name):
  841. """移除串口"""
  842. print("remove", port_name)
  843. data = {
  844. "_type": "remove_port",
  845. "plugins_mode": "auto_select_com",
  846. "data": {"port_name": port_name},
  847. }
  848. self.sendSocketMessage(1, "串口被移除", data)
  849. async def add_port_by_linkage(self, port_name):
  850. # port_value :串口基础信息
  851. # todo 根据prot_value 信息自动进行连接
  852. print("add", port_name)
  853. # 对没有连接的设备进行尝试连接
  854. message_data = {
  855. "_type": "show_info",
  856. "plugins_mode": "auto_select_com",
  857. "data": {"text": "开始识别接口:{}".format(port_name)},
  858. }
  859. self.sendSocketMessage(
  860. msg="开始识别接口:{}".format(port_name), data=message_data, device_status=1
  861. )
  862. await asyncio.sleep(1)
  863. """
  864. 步骤:
  865. 1、进行临时连接,并发送命令,成功后,自动连接对应设备
  866. """
  867. try:
  868. # 尝试使用115200波特率链接
  869. serial_handle = serial.Serial(port=port_name, baudrate=115200, timeout=0.5)
  870. except Exception as e:
  871. message_data = {
  872. "_type": "show_info",
  873. "plugins_mode": "auto_select_com",
  874. "data": {"text": "串口:{} 被占用,或无法识别".format(port_name)},
  875. }
  876. self.sendSocketMessage(
  877. 1,
  878. msg="串口:{} 被占用,或无法识别".format(port_name).format(port_name),
  879. data=message_data,
  880. device_status=-1,
  881. )
  882. print("串口:{} 被占用".format(port_name))
  883. return False
  884. await asyncio.sleep(2)
  885. print("开始发送命令")
  886. data = [90, 1]
  887. try:
  888. serial_handle.flushInput() # 尝试重置输入缓冲区
  889. except serial.SerialTimeoutException:
  890. print("超时错误:无法在规定时间内重置输入缓冲区。")
  891. self.sendSocketMessage(
  892. 1,
  893. msg="超时错误:无法在规定时间内重置输入缓冲区。",
  894. data=None,
  895. )
  896. serial_handle.close()
  897. return False
  898. print("尝试写入数据")
  899. buf = bytearray(b"")
  900. buf.extend([0x55, 0x55, (0xFF & len(data))])
  901. buf.extend(data)
  902. buf.extend([0xFF & ~sum(data)])
  903. try:
  904. self.receive_data = b""
  905. serial_handle.write(buf)
  906. except serial.SerialTimeoutException:
  907. print("写入数据错误")
  908. serial_handle.close()
  909. return False
  910. await asyncio.sleep(0.3)
  911. receive_data = self.read_cmd(serial_handle)
  912. print("尝试接收命令", receive_data)
  913. device_id = 0
  914. if receive_data:
  915. print("receive_data", receive_data)
  916. if receive_data[0] == 90:
  917. connect_flag = receive_data[1]
  918. device_id = receive_data[2]
  919. print("关闭串口:{}".format(port_name))
  920. serial_handle.close()
  921. loop = asyncio.get_event_loop()
  922. print("device_id============>>>", device_id)
  923. if device_id > 0:
  924. if device_id == 1:
  925. await self.to_connect_com(port_name)
  926. message_data = {
  927. "_type": "show_info",
  928. "plugins_mode": "auto_select_com",
  929. "data": {"text": "MCU开始连接"},
  930. }
  931. self.sendSocketMessage(
  932. msg="MCU开始连接", data=message_data, device_status=1
  933. )
  934. self.connected_ports_dict[port_name] = "MCU"
  935. message_data = {
  936. "_type": "select_port_name",
  937. "plugins_mode": "auto_select_com",
  938. "data": {
  939. "device_name": "mcu" if device_id == 1 else "remote_control",
  940. "port_name": port_name,
  941. },
  942. }
  943. self.sendSocketMessage(
  944. msg="MCU连接成功", data=message_data, device_status=2
  945. )
  946. await asyncio.sleep(2)
  947. # 异步处理
  948. loop.create_task(self.initDevice(),name="initDevice")
  949. elif device_id == 2:
  950. # 有线接收器
  951. self.connected_ports_dict[port_name] = "Line_Control"
  952. print("device_id", device_id)
  953. loop.create_task(
  954. self.line_control.to_connect_com(port_name),
  955. name="line_control_to_connect_com",
  956. )
  957. # message_data = {
  958. # "_type": "show_info",
  959. # "plugins_mode": "auto_select_com",
  960. # "data": {"text": "有线控制器开始连接"},
  961. # }
  962. # self.sendSocketMessage(
  963. # msg="有线控制器开始连接", data=message_data, device_status=1
  964. # )
  965. # self.connected_ports_dict[port_name] = "MCU"
  966. # message_data = {
  967. # "_type": "select_port_name",
  968. # "plugins_mode": "auto_select_com",
  969. # "data": {
  970. # "device_name": "mcu" if device_id == 1 else "remote_control",
  971. # "port_name": port_name,
  972. # },
  973. # }
  974. # self.sendSocketMessage(
  975. # msg="有线控制器开始连接", data=message_data, device_status=2
  976. # )
  977. # await asyncio.sleep(2)
  978. # await self.initControlLine(serial_handle)
  979. # async def getBaseInfo():
  980. # while True:
  981. # await asyncio.sleep(1)
  982. # # 异步循环获取设备信息
  983. # self.to_get_mcu_base_info()
  984. # asyncio.gather(getBaseInfo())
  985. return True
  986. else:
  987. print("串口无法识别")
  988. self.sendSocketMessage(
  989. code=1,
  990. msg="串口无法识别,请重新插拔拍照机USB", data=message_data, device_status=-1
  991. )
  992. return False
  993. # 走其他途径处理
  994. # 检查当前MCU链接是否正常
  995. # 正常跳过;记录为其他列表
  996. # 不正常进行尝试连接
  997. # 连接不上,记录为其他列表
  998. def clearMyInstance(self):
  999. SingletonType.clear_instance()
  1000. async def to_connect_com(self, port_name):
  1001. # 关闭串口
  1002. print("to_connect_com", port_name)
  1003. self.close_connect()
  1004. await asyncio.sleep(0.3)
  1005. self.connect_state = False
  1006. try:
  1007. self.serial_ins = SerialIns(port_name=port_name, baud=115200, timeout=0.1)
  1008. if not self.serial_ins.serial_handle:
  1009. message_data = {
  1010. "_type": "show_info",
  1011. "plugins_mode": "mcu",
  1012. "data": "MCU 打开串口失败",
  1013. }
  1014. self.sendSocketMessage(
  1015. msg="MCU 打开串口失败", data=message_data, device_status=-1
  1016. )
  1017. self.serial_ins = None
  1018. self.connect_state = False
  1019. return False,None
  1020. except:
  1021. message_data = {
  1022. "_type": "show_info",
  1023. "plugins_mode": "mcu",
  1024. "data": "MCU 打开串口失败",
  1025. }
  1026. self.sendSocketMessage(
  1027. msg="MCU 打开串口失败", data=message_data, device_status=-1
  1028. )
  1029. self.serial_ins = None
  1030. self.connect_state = False
  1031. return False, None
  1032. message_data = {
  1033. "_type": "show_info",
  1034. "plugins_mode": "mcu",
  1035. "data": "MCU 开始连接",
  1036. }
  1037. self.sendSocketMessage(msg="MCU 开始连接", data=message_data, device_status=1)
  1038. # =======================发送连接请求=================================
  1039. cmd = 90
  1040. data = [cmd, 1]
  1041. print("405 发送 连接请求 -----------------------------------------")
  1042. print(self.serial_ins)
  1043. # self.serial_ins.clearn_flush()
  1044. self.serial_ins.write_cmd(data)
  1045. # 延迟接收数据
  1046. await asyncio.sleep(0.3)
  1047. receive_data = self.serial_ins.read_cmd(out_time=1)
  1048. if receive_data:
  1049. print(
  1050. "409 receive_data--90:{}".format(self.change_hex_to_int(receive_data))
  1051. )
  1052. if receive_data:
  1053. # receive_data[2]=1 表示为MCU设备编号
  1054. if receive_data[0] == 90 and receive_data[2] == 1:
  1055. connect_flag = receive_data[1]
  1056. # 是否有初始化
  1057. try:
  1058. mcu_has_been_set = receive_data[
  1059. 6
  1060. ] # 设备是否有初始化 ,1 表示已初始化
  1061. except:
  1062. mcu_has_been_set = 99 # 未知状态
  1063. print("MCU初始化信息{}".format(mcu_has_been_set))
  1064. message_data = {
  1065. "_type": "show_info",
  1066. "plugins_mode": "mcu",
  1067. "data": "MCU 已连接",
  1068. }
  1069. self.sendSocketMessage(
  1070. msg="MCU 已连接", data=message_data, device_status=1
  1071. )
  1072. self.connect_state = True
  1073. self.is_running = True
  1074. print("MCU 已连接")
  1075. self.port_name = port_name
  1076. return
  1077. print("MCU 连接失败")
  1078. message_data = {
  1079. "_type": "show_info",
  1080. "plugins_mode": "mcu",
  1081. "data": "MCU 连接失败",
  1082. }
  1083. self.sendSocketMessage(msg="MCU 连接失败", data=message_data, device_status=-1)
  1084. self.close_connect()
  1085. def close_connect(self):
  1086. self.port_name = ""
  1087. if self.serial_ins:
  1088. self.serial_ins.close_serial_port()
  1089. self.is_running = False
  1090. self.connect_state = False
  1091. self.connected_ports_dict = {} # 已连接的ports
  1092. # self.p_list = []
  1093. # self.temp_ports_dict = {}
  1094. self.init_state = False
  1095. self.is_init_while = False
  1096. print("关闭MCU")
  1097. @property
  1098. def mcu_move_state(self):
  1099. if self.m_t == 1:
  1100. if (
  1101. self.state_camera_motor == 2
  1102. and self.state_camera_steering == 2
  1103. and self.state_turntable_steering == 2
  1104. and self.state_overturn_steering == 2
  1105. ):
  1106. self._mcu_move_state = 2
  1107. else:
  1108. self._mcu_move_state = 1
  1109. else:
  1110. if (
  1111. self.state_camera_motor == 2
  1112. and self.state_camera_steering == 2
  1113. and self.state_turntable_steering == 2
  1114. and self.state_overturn_steering == 2
  1115. and self.state_move_turntable_steering == 2
  1116. ):
  1117. self._mcu_move_state = 2
  1118. else:
  1119. self._mcu_move_state = 1
  1120. # self._mcu_move_state = 2
  1121. return self._mcu_move_state
  1122. def to_deal_device(self, device_name, value=1, _type=0, times=1, delay=0):
  1123. """
  1124. value 激光0关 1开
  1125. mp3_player value 表示0表示关,1表示开,_type 表示歌曲切换到指定歌曲
  1126. delay:延迟处理,单位为0.1秒,即delay=100时,表示延迟10秒
  1127. """
  1128. device_id = self.device_name_dict[device_name]
  1129. if device_name == "buzzer":
  1130. value = int(value)
  1131. cmd = 3
  1132. data = [
  1133. cmd,
  1134. device_id,
  1135. value,
  1136. _type,
  1137. times,
  1138. delay,
  1139. ]
  1140. self.add_send_data_queue(data)
  1141. # if self.serial_ins:
  1142. # self.serial_ins.write_cmd(data)
  1143. return True
  1144. def to_device_move(
  1145. self,
  1146. device_name,
  1147. value=0,
  1148. max_speed=None,
  1149. up_speed=None,
  1150. down_speed=None,
  1151. _is_debug=0,
  1152. is_relative=0,
  1153. is_deviation=1,
  1154. ):
  1155. """
  1156. 此处输入单位为 毫米,以及度 需要先缩小,再放大
  1157. """
  1158. print("移动",time.time())
  1159. speed = settings.moveSpeed()
  1160. cmd = 1
  1161. device_id = self.device_name_dict[device_name]
  1162. # if device_id != 1:
  1163. # print("F非MCU设备,禁止处理")
  1164. # return
  1165. print("正在执行",device_name)
  1166. match device_name:
  1167. case "camera_high_motor":
  1168. # value 单位毫米
  1169. # max_speed = 10000 if max_speed is None else max_speed
  1170. # up_speed = 800 if up_speed is None else up_speed
  1171. # down_speed = 700 if down_speed is None else down_speed
  1172. max_speed = (
  1173. speed[device_name]["max_speed"] if max_speed is None else max_speed
  1174. )
  1175. up_speed = (
  1176. speed[device_name]["up_speed"] if up_speed is None else up_speed
  1177. )
  1178. down_speed = (
  1179. speed[device_name]["down_speed"]
  1180. if down_speed is None
  1181. else down_speed
  1182. )
  1183. value = value / 10 # value 单位毫米
  1184. assert 0 <= value <= 40
  1185. assert 0 <= max_speed <= 10000
  1186. case "camera_steering":
  1187. # 角度为度 未放大 精确到0.1度
  1188. max_speed = 6000 if max_speed is None else max_speed
  1189. up_speed = 500 if up_speed is None else up_speed
  1190. down_speed = 500 if down_speed is None else down_speed
  1191. assert -360 <= value <= 360
  1192. case "turntable_steering":
  1193. # 角度为度 未放大 精确到0.1度
  1194. # max_speed = 6000 if max_speed is None else max_speed
  1195. # up_speed = 500 if up_speed is None else up_speed
  1196. # down_speed = 400 if down_speed is None else down_speed
  1197. max_speed = (
  1198. speed[device_name]["max_speed"] if max_speed is None else max_speed
  1199. )
  1200. up_speed = (
  1201. speed[device_name]["up_speed"] if up_speed is None else up_speed
  1202. )
  1203. down_speed = (
  1204. speed[device_name]["down_speed"]
  1205. if down_speed is None
  1206. else down_speed
  1207. )
  1208. assert -720 <= value <= 720
  1209. case "overturn_steering":
  1210. # 角度为度 未放大 精确到0.1度
  1211. max_speed = 2 if max_speed is None else max_speed
  1212. up_speed = 1 if up_speed is None else up_speed
  1213. down_speed = 1 if down_speed is None else down_speed
  1214. assert 0 <= value <= 360
  1215. case "turntable_position_motor":
  1216. # value 单位毫米
  1217. max_speed = 11000 if max_speed is None else max_speed
  1218. up_speed = 900 if up_speed is None else up_speed
  1219. down_speed = 900 if down_speed is None else down_speed
  1220. value = value / 10 # value 单位毫米
  1221. assert 0 <= value <= 900
  1222. assert 0 <= max_speed <= 15000
  1223. _dir = True if value >= 0 else False
  1224. value = int(abs(value * 10)) # 此处value赋值后,单位为mm以及0.1度
  1225. print("准备执行",device_name, value)
  1226. data = [
  1227. cmd,
  1228. device_id,
  1229. 1 if _dir else 0,
  1230. 0xFF & value >> 8,
  1231. 0xFF & value,
  1232. 0xFF & max_speed >> 8,
  1233. 0xFF & max_speed,
  1234. 0xFF & up_speed >> 8,
  1235. 0xFF & up_speed,
  1236. 0xFF & down_speed >> 8,
  1237. 0xFF & down_speed,
  1238. _is_debug,
  1239. is_deviation,
  1240. is_relative,
  1241. ]
  1242. self.add_send_data_queue(data)
  1243. def to_get_mcu_base_info(self):
  1244. if self.connect_state:
  1245. self.lock.acquire()
  1246. # print('==========================>1111')
  1247. # print("-------------------to_get_mcu_base_info--------------------------")
  1248. data = [self.command["get_all_info"], 1]
  1249. f = True
  1250. try:
  1251. self.serial_ins.write_cmd(data)
  1252. except:
  1253. f = False
  1254. pass
  1255. self.lock.release()
  1256. if not f:
  1257. self.connect_state = False
  1258. return False
  1259. else:
  1260. return True
  1261. def check_before_action(self):
  1262. if self.state != 2:
  1263. print("check_before_action 设备正在运行中~")
  1264. self.sendSocketMessage(
  1265. code=1, msg="设备正在运行中", device_status=1
  1266. )
  1267. return False
  1268. if self.mcu_move_state != 2:
  1269. if settings.IS_LIN_SHI_TEST:
  1270. return True
  1271. # self.show_info("mcu 非停止状态")
  1272. self.sendSocketMessage(code=1, msg="mcu 非停止状态", device_status=1)
  1273. return True
  1274. async def controlDevice(self, device_name, value):
  1275. '''控制设备移动等'''
  1276. await asyncio.sleep(0.01)
  1277. if not self.is_running:
  1278. self.sendSocketMessage(
  1279. code=1, msg="mcu设备未连接,请先连接设备", device_status=0
  1280. )
  1281. return False
  1282. if not self.init_state:
  1283. self.sendSocketMessage(
  1284. code=1, msg="mcu设备未初始化", device_status=4
  1285. )
  1286. return False
  1287. _is_debug = 1 if settings.IS_DEBUG == "true" else 0
  1288. is_deviation = 0 if settings.IS_DEBUG == "true" else 1
  1289. print("控制设备==>_is_debug", _is_debug)
  1290. match device_name:
  1291. case "camera_high_motor":
  1292. # 相机电机
  1293. print(device_name, value)
  1294. self.to_device_move(
  1295. device_name=device_name,
  1296. value=float(value),
  1297. # max_speed=1400,
  1298. # up_speed=400,
  1299. # down_speed=100,
  1300. # _is_debug=_is_debug,
  1301. # is_deviation=is_deviation,
  1302. )
  1303. case "camera_steering":
  1304. print(device_name, value)
  1305. # 相机舵机
  1306. self.to_device_move(
  1307. device_name=device_name,
  1308. value=float(value),
  1309. # _is_debug=_is_debug,
  1310. # is_deviation=0,
  1311. )
  1312. case "turntable_steering":
  1313. # 转盘舵机
  1314. self.to_device_move(
  1315. device_name=device_name,
  1316. value=float(value),
  1317. # _is_debug=_is_debug,
  1318. # is_deviation=0,
  1319. )
  1320. case "turntable_position_motor":
  1321. # 转盘舵机
  1322. self.to_device_move(
  1323. device_name=device_name,
  1324. value=float(value),
  1325. # max_speed=1400,
  1326. # up_speed=400,
  1327. # down_speed=100,
  1328. # _is_debug=_is_debug,
  1329. # is_deviation=is_deviation,
  1330. )
  1331. case "overturn_steering":
  1332. # 翻板舵机中位
  1333. self.to_deal_device(
  1334. device_name="overturn_steering",
  1335. )
  1336. case "laser_position":
  1337. self.to_deal_device(
  1338. device_name="laser_position", value=0 if int(value) <= 0 else 1
  1339. )
  1340. case "take_picture":
  1341. # loop = asyncio.get_event_loop()
  1342. # if value>0:
  1343. # # 指定自动对焦
  1344. # loop.create_task(
  1345. # self.smart_shooter.CameraAutofocus(),
  1346. # name="CameraAutofocus",
  1347. # )
  1348. # 拍照
  1349. await self.smart_shooter.CameraShooter(msg_type="run_mcu")
  1350. case "to_deal_device":
  1351. self.to_deal_device(device_name, value=value, _type=0, times=1)
  1352. case _:
  1353. pass
  1354. # case "photograph":
  1355. # self.photograph(goods_art_no=None)
  1356. def checkDevice(self):
  1357. print("检查设备是否运行中")
  1358. if not self.is_running:
  1359. self.sendSocketMessage(
  1360. code=1, msg="mcu设备未连接,请先连接设备", device_status=0
  1361. )
  1362. return False
  1363. if self.init_state is not True:
  1364. self.sendSocketMessage(code=1, msg="mcu设备未初始化", device_status=4)
  1365. return False
  1366. if self.action_state != 2:
  1367. self.sendSocketMessage(
  1368. code=1,
  1369. msg="当前有未完成的任务,请稍后再试",
  1370. device_status=0,
  1371. )
  1372. return False
  1373. async def run_mcu_config(
  1374. self, config_list, goods_art_no, action_info, smart_shooter
  1375. ):
  1376. if self.checkDevice() == False:
  1377. return
  1378. image_counts = 0
  1379. if config_list:
  1380. action_names = []
  1381. record_ids = []
  1382. if len(config_list) > 1:
  1383. if config_list[-1]["take_picture"] is True:
  1384. new_init_config = copy.copy(config_list[0])
  1385. new_init_config["action_name"] = "移动到初始位"
  1386. new_init_config["number_focus"] = 0
  1387. new_init_config["take_picture"] = False
  1388. new_init_config["shoe_upturn"] = False
  1389. new_init_config["pre_delay"] = 0.0
  1390. new_init_config["after_delay"] = 0.0
  1391. new_init_config["led_switch"] = True
  1392. new_init_config["turntable_angle"] = 0.0
  1393. new_init_config["is_wait"] = False
  1394. new_init_config["is_need_confirm"] = False
  1395. config_list.append(new_init_config)
  1396. for idx, item in enumerate(config_list):
  1397. is_take_picture = item["take_picture"]
  1398. action_id = item["id"]
  1399. record_id = -1
  1400. if is_take_picture:
  1401. image_counts += 1
  1402. # 批量插入
  1403. image_deal_mode = 0 if action_info == "执行左脚程序" else 1
  1404. state,record_id = await insert_photo_records(
  1405. image_deal_mode=image_deal_mode,
  1406. goods_art_no=goods_art_no,
  1407. image_index=idx,
  1408. action_id=action_id,
  1409. )
  1410. config_list[idx]["record_id"] = record_id
  1411. total_len = len(config_list)
  1412. self.action_state = 1
  1413. self.msg_type = "image_process"
  1414. self.sendSocketMessage(
  1415. code=0,
  1416. msg="MCU 命令已发送完成",
  1417. device_status=2,
  1418. data={
  1419. "goods_art_no": goods_art_no,
  1420. "image_counts": image_counts,
  1421. "action_names": action_names,
  1422. "current_time": datetime.datetime.now(settings.TIME_ZONE).strftime(
  1423. "%Y-%m-%d %H:%M:%S"
  1424. ),
  1425. },
  1426. )
  1427. await self.controlDevice("laser_position", 0)
  1428. self.msg_type = "mcu"
  1429. self.is_runn_action = True
  1430. for index, action in enumerate(config_list):
  1431. # await asyncio.sleep(0.01)
  1432. if self.is_stop_action == True:
  1433. print("停止命令接收,立即终止")
  1434. break
  1435. # action_is_take_picture = action["take_picture"]
  1436. record_id = action["record_id"]
  1437. image_index = -1
  1438. if record_id == -1:
  1439. image_index = -1
  1440. else:
  1441. image_index = index
  1442. program_item = ProgramItem(
  1443. websocket_manager=self.websocket_manager,
  1444. action_data=action,
  1445. mcu=self,
  1446. goods_art_no=goods_art_no,
  1447. image_index=image_index,
  1448. smart_shooter=smart_shooter,
  1449. record_id=record_id,
  1450. )
  1451. if self.action_state != 1:
  1452. # 异常终止
  1453. print("action异常终止")
  1454. break
  1455. self.msg_type = "photo_take"
  1456. if not await program_item.run(total_len):
  1457. self.sendSocketMessage(
  1458. code=1,
  1459. msg="{} 执行失败~".format(program_item.action_name),
  1460. data={"goods_art_no": goods_art_no, "id": action_id},
  1461. device_status=0,
  1462. )
  1463. self.to_deal_device(device_name="buzzer", times=3)
  1464. break
  1465. else:
  1466. if program_item.action_name == "移动到初始位":
  1467. self.msg_type = "mcu"
  1468. break
  1469. self.sendSocketMessage(
  1470. code=0,
  1471. msg="{} 执行完成~".format(program_item.action_name),
  1472. data={"goods_art_no": goods_art_no, "id": record_id},
  1473. device_status=2,
  1474. )
  1475. self.msg_type = "mcu"
  1476. # 在第三张图时检查是否有对应图片生成
  1477. # if index == 3:
  1478. # # if not self.image_process_data.check_photo_is_get():
  1479. # self.sendSocketMessage(
  1480. # code=1,
  1481. # msg="未获取到图片数据",
  1482. # device_status=0,
  1483. # )
  1484. # self.action_state = 2
  1485. # return
  1486. if index == total_len - 1:
  1487. # 最后一个初始化处理
  1488. pass
  1489. # self.action_state = 2
  1490. if self.is_stop_action==True:
  1491. self.msg_type = "run_mcu_stop"
  1492. self.sendSocketMessage(
  1493. code=0,
  1494. msg=f"货号:{goods_art_no},执行终止",
  1495. device_status=2,
  1496. )
  1497. self.is_stop_action = False
  1498. self.action_state = 2
  1499. self.is_runn_action = False
  1500. self.msg_type = "photo_take_finish"
  1501. self.sendSocketMessage(
  1502. code=0,
  1503. msg=f"货号:{goods_art_no},执行完成",
  1504. device_status=2,
  1505. )
  1506. self.msg_type = "mcu"
  1507. await self.controlDevice("laser_position", 1)
  1508. async def run_mcu_config_single(self, config_info, goods_art_no,msg_type="run_mcu_single_finish",image_index=-1,smart_shooter=None,action_id=-1):
  1509. '''独立拍照 仅作测试用'''
  1510. await asyncio.sleep(0.01)
  1511. if self.checkDevice() == False:
  1512. return
  1513. print("检查完成", config_info)
  1514. if config_info:
  1515. self.action_state = 1
  1516. self.msg_type = "mcu"
  1517. program_item = ProgramItem(
  1518. websocket_manager=self.websocket_manager,
  1519. action_data=config_info,
  1520. mcu=self,
  1521. goods_art_no=goods_art_no,
  1522. image_index=image_index,
  1523. smart_shooter=smart_shooter,
  1524. record_id=action_id,
  1525. )
  1526. if self.action_state != 1:
  1527. # 异常终止
  1528. print("action异常终止")
  1529. return
  1530. self.msg_type = "photo_take"
  1531. program_item.smart_shooter = smart_shooter
  1532. await program_item.run(3)
  1533. self.sendSocketMessage(
  1534. code=0,
  1535. msg="{} 执行完成~".format(program_item.action_name),
  1536. data={"goods_art_no": goods_art_no, "id": action_id},
  1537. device_status=2,
  1538. )
  1539. self.msg_type = "mcu"
  1540. self.action_state = 2
  1541. self.msg_type = msg_type
  1542. self.sendSocketMessage(
  1543. code=0,
  1544. msg=f"执行完成",
  1545. device_status=2,
  1546. )
  1547. self.msg_type = "mcu"
  1548. else:
  1549. self.sendSocketMessage(
  1550. code=1, msg="未查询到重拍记录得配置信息,请确认", device_status=0
  1551. )
  1552. async def only_take_photo(self, goods_art_no, image_index, record_id):
  1553. await asyncio.sleep(0.1)
  1554. print("only_take_photo=====>",goods_art_no, image_index, record_id)
  1555. if goods_art_no == "":
  1556. print("only_take_photo 数据查询异常")
  1557. return
  1558. # 关闭led
  1559. self.controlDevice("laser_position", 0)
  1560. program_item = ProgramItem(
  1561. websocket_manager=self.websocket_manager,
  1562. action_data={},
  1563. mcu=self,
  1564. goods_art_no=goods_art_no,
  1565. image_index=image_index,
  1566. record_id=record_id,
  1567. )
  1568. program_item.digicam_take_picture()
  1569. # 打开led
  1570. self.controlDevice("laser_position", 1)
  1571. async def checkMcuConnection(device_ctrl: DeviceControl):
  1572. if device_ctrl.is_running == True:
  1573. message = {
  1574. "_type": "select_port_name",
  1575. "plugins_mode": "auto_select_com",
  1576. "data": device_ctrl.temp_ports_dict,
  1577. }
  1578. device_ctrl.device_status = 2
  1579. device_ctrl.sendSocketMessage(code=0, msg="MCU连接成功", data=message)
  1580. return
  1581. if device_ctrl.is_init_while:
  1582. return
  1583. device_ctrl.is_init_while = True
  1584. """实时检测串口是否连接"""
  1585. while True:
  1586. await asyncio.sleep(1)
  1587. if device_ctrl.mcu_exit:
  1588. break
  1589. ports_dict = device_ctrl.scan_serial_port()
  1590. device_ctrl.temp_ports_dict = ports_dict
  1591. print("device_ctrl.p_list", device_ctrl.p_list)
  1592. if not ports_dict:
  1593. # 全部清空 移除所有串口
  1594. if device_ctrl.p_list:
  1595. _p = device_ctrl.p_list.pop()
  1596. device_ctrl.remove_port(_p)
  1597. continue
  1598. if ports_dict:
  1599. for index, _i in enumerate(device_ctrl.p_list):
  1600. if _i not in ports_dict:
  1601. _p = device_ctrl.p_list.pop(index)
  1602. device_ctrl.remove_port(_p)
  1603. for _port_name, _port_value in ports_dict.items():
  1604. if _port_name not in device_ctrl.p_list:
  1605. try:
  1606. flag = await device_ctrl.add_port_by_linkage(_port_name)
  1607. if flag == False:
  1608. continue
  1609. device_ctrl.p_list.append(_port_name)
  1610. except BaseException as e:
  1611. print("串口不存在{} {}".format(_port_name, e))
  1612. # if not device_ctrl.connect_state:
  1613. # print("串口未连接,请检查")
  1614. # device_ctrl.sendSocketMessage(
  1615. # code=1, msg="串口未连接,请检查", device_status=-1
  1616. # )
  1617. device_ctrl.is_init_while = False
  1618. print("MCU断开连接,已释放")