DeviceControl.py 84 KB

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