mirror of
https://github.com/paparazzi/paparazzi.git
synced 2026-05-10 15:09:25 +08:00
Felix Ruess
9e0cf7f7d9
- enum is available since python3.4 or as backport (enum34 package) - otherwise just use normal member vars as fallback - cleanup some formatting/naming according to pep8 - SerialMessagesInterface should not depend on the messages_xml_map, as a workaround for now, only call parse_messages in main
111 lines
3.8 KiB
Python
111 lines
3.8 KiB
Python
from __future__ import absolute_import, division, print_function
|
|
|
|
import threading
|
|
import serial
|
|
import os
|
|
import sys
|
|
|
|
# if PAPARAZZI_SRC not set, then assume the tree containing this
|
|
# file is a reasonable substitute
|
|
PPRZ_SRC = os.getenv("PAPARAZZI_SRC", os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)),
|
|
'../../../../')))
|
|
sys.path.append(PPRZ_SRC + "/sw/lib/python")
|
|
|
|
from pprz_msg.message import PprzMessage
|
|
from pprz_msg.pprz_transport import PprzTransport
|
|
import pprz_msg.messages_xml_map
|
|
|
|
|
|
class SerialMessagesInterface(threading.Thread):
|
|
def __init__(self, callback, init=True, verbose=False, device='/dev/ttyUSB0', baudrate=115200,
|
|
msg_class='telemetry'):
|
|
threading.Thread.__init__(self)
|
|
self.callback = callback
|
|
self.verbose = verbose
|
|
self.msg_class = msg_class
|
|
self.running = True
|
|
try:
|
|
self.ser = serial.Serial(device, baudrate, timeout=1.0)
|
|
except serial.SerialException:
|
|
print("Error: unable to open serial port '%s'" % device)
|
|
exit(0)
|
|
self.trans = PprzTransport(msg_class)
|
|
|
|
def stop(self):
|
|
print("End thread and close serial link")
|
|
self.running = False
|
|
self.ser.close()
|
|
|
|
def shutdown(self):
|
|
self.stop()
|
|
|
|
def __init__del__(self):
|
|
try:
|
|
self.ser.close()
|
|
except:
|
|
pass
|
|
|
|
def send(self, msg, sender_id):
|
|
""" Send a message over a serial link"""
|
|
if isinstance(msg, PprzMessage):
|
|
data = self.trans.pack_pprz_msg(sender_id, msg)
|
|
self.ser.write(data)
|
|
self.ser.flush()
|
|
|
|
def run(self):
|
|
"""Thread running function"""
|
|
try:
|
|
while self.running:
|
|
# Parse incoming data
|
|
c = self.ser.read(1)
|
|
if len(c) == 1:
|
|
if self.trans.parse_byte(c):
|
|
(sender_id, msg) = self.trans.unpack()
|
|
if self.verbose:
|
|
print("New incoming message '%i' from %s" % (sender_id, msg.name))
|
|
# Callback function on new message
|
|
self.callback(sender_id, msg)
|
|
|
|
except StopIteration:
|
|
pass
|
|
|
|
# FIXME not working because of the thread ?
|
|
import signal
|
|
|
|
|
|
def signal_term_handler(signal, frame):
|
|
print("got SIGINT")
|
|
# sys.exit(0)
|
|
|
|
|
|
def test():
|
|
signal.signal(signal.SIGINT, signal_term_handler)
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-f", "--file", help="path to messages.xml file")
|
|
parser.add_argument("-c", "--class", help="message class", dest='msg_class', default='telemetry')
|
|
parser.add_argument("-d", "--device", help="device name", dest='dev', default='/dev/ttyUSB0')
|
|
parser.add_argument("-b", "--baudrate", help="baudrate", dest='baud', default=115200, type=int)
|
|
args = parser.parse_args()
|
|
pprz_msg.messages_xml_map.parse_messages(args.file)
|
|
serial_interface = SerialMessagesInterface(lambda s, m: print("new message from %i: %s" % (s, m)), device=args.dev,
|
|
baudrate=args.baud, msg_class=args.msg_class, verbose=True)
|
|
att_msg = PprzMessage('telemetry', 'ATTITUDE')
|
|
att_msg.set_value_by_name('phi', 0.1)
|
|
att_msg.set_value_by_name('theta', 0.2)
|
|
att_msg.set_value_by_name('psi', 0.3)
|
|
serial_interface.send(att_msg, 42)
|
|
to_msg = PprzMessage('telemetry', 'TAKEOFF')
|
|
to_msg.set_value_by_name('cpu_time', 10)
|
|
serial_interface.send(to_msg, 42)
|
|
print("Starting serial interface on %s at %i baud" % (args.dev, args.baud))
|
|
serial_interface.start()
|
|
signal.pause()
|
|
serial_interface.stop()
|
|
serial_interface.join()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test()
|