DeviceControl.py 81 KB

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