UWB_Module_Firmware/config.py at master · repowered-electronics/UWB_Module_Firmware · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import serial
import repowered
import serial.tools.list_ports

# CMD_READ_CONFIG	= 0x11,
# CMD_READ_ANCHORS 	= 0x12,
# CMD_SET_CONFIG 	= 0x22,
# CMD_RANGE 		= 0x33,
# CMD_RESTART 		= 0x44,
# CMD_RESET 		= 0x55,
# CMD_SAVE_CONFIG 	= 0x66
cmd_com  = "com"
cmd_read = "read"
cmd_set  = "set"
cmd_save = "save"
cmd_range = "range"
cmd_quit = "quit"
cmd_help = "help"

opt_id_v   = "--id"
opt_mode_v = "--mode"

arg_tag    = "tag"
arg_anchor = "anchor"

options =  cmd_com + "   : Set comport\r\n"
options += cmd_read + "  : Read configuration\n\r"
options += cmd_set + "   : Set configuration\n\r"
options +=        "      :   " + opt_id_v + "=[N]\n\r"
options +=        "      :   " + opt_mode_v + "=["+ arg_tag + ", " + arg_anchor + "]\n\r"
options += cmd_range + " : trigger a ranging transaction\n\r"
options += cmd_save + "  : save configuration\n\r"
options += cmd_help + "  : display this menu\n\r"
options += cmd_quit + "  : quit\r\n"


ports = serial.tools.list_ports.comports()

ser = serial.Serial()
def main():
    print("Available Comports")
    for port in ports:
        print(port.name)

    print("-----")

    print(options)

    uwb = None


    done = False
    while not done:
        s = input("> ")
        words = s.split(' ')

        if words[0].lower() == cmd_com:
            uwb = repowered.UwbConfig(words[1])
            uwb.id = 0
            uwb.mode = 0
            uwb.channel = 2
            uwb.samples_per_range = 1
            uwb.number_of_anchors = 5

        elif words[0].lower() == cmd_read:
            print("reading...")
            print(uwb.read_config())

        elif words[0].lower() == cmd_set:
            print("setting...")
            print(uwb.set_config())
        elif words[0].lower() == cmd_save:
            print("saving...")
        elif words[0].lower() == cmd_help:
            print(options)
        elif words[0].lower() == cmd_quit:
            print("quitting...")
            if ser.is_open:
                ser.close()
            done = True


if __name__ == '__main__':
    main()