1. 程式人生 > >如何用Python控制LEGO Wedo 2.0的電機

如何用Python控制LEGO Wedo 2.0的電機

樂高官方提供的程式設計工具都是圖形化的,這篇文章分享下如何用Python來連線控制LEGO Wedo 2.0中提供的藍芽主控裝置和電機。

學習資源

網上有兩篇很有用的文章:

使用pygattlib可以通過Python和藍芽低功耗裝置通訊。這個庫只支援Linux

命令列工具和開發環境

通過測試發現,Windows的Linux子系統和VMWare虛擬機器是不支援的。我最終選了樹莓派做開發環境。 使用系統自帶的命令列工具可以直接測試藍芽連線。 掃描裝置獲取地址:

sudo hcitool -i hci0 lescan
LE Scan ...
98:07:2D:DD:98:56 (unknown)
98:07:2D:DD:98:56 LPF2 Smart Hub

連線裝置:

gatttool -I
[                 ][LE]> connect <ble address>

使用Python建立介面控制程式

安裝依賴庫以及gattlib:

sudo apt-get update
sudo apt-get install libbluetooth-dev bluez bluez-hcidump libboost-python-dev libboost-thread-dev libglib2.0-dev
sudo pip install gattlib

Tkinter建立介面:

import Tkinter as tk
from gattlib import DiscoveryService
from gattlib import GATTRequester
from time import sleep
 
def run():
    global button_run
    button_run.after(DELAY, motor_run)
 
def stop():
    global button_stop
    button_stop.after(DELAY, motor_stop)
 
def connect():
    global button_disconnect
    button_disconnect.after(DELAY, smart_hub_connect)
 
def disconnect():
    global button_disconnect
    button_disconnect.after(DELAY, smart_hub_disconnect)
 
def up():
    global button_up
    button_up.after(DELAY, motor_up)
 
def down():
    global button_down
    button_down.after(DELAY, motor_down)
 
 
root = tk.Tk()
root.title("Lego Wedo 2.0 Motor Control")
 
label = tk.Label(root, fg="dark green", text='N/A')
label.pack()
 
button_connect = tk.Button(root, text='Connect Smart Hub', width=BUTTON_WIDTH, command=connect)
button_connect.pack()
 
button_disconnect = tk.Button(root, text='Disconnect Smart Hub', width=BUTTON_WIDTH, command=disconnect, state='disabled')
button_disconnect.pack()
 
button_run = tk.Button(root, text='Run motor', width=BUTTON_WIDTH, command=run, state='disabled')
button_run.pack()
 
button_up = tk.Button(root, text='Speed up', width=BUTTON_WIDTH, command=up, state='disabled')
button_up.pack()
 
button_down = tk.Button(root, text='Speed down', width=BUTTON_WIDTH, command=down, state='disabled')
button_down.pack()
 
button_stop = tk.Button(root, text='Stop motor', width=BUTTON_WIDTH, command=stop, state='disabled')
button_stop.pack()
 
root.mainloop()

使用DiscoveryServiceGATTRequester查詢連線裝置:

def smart_hub_connect():
    service = DiscoveryService("hci0")
    devices = service.discover(2)
 
    for address, name in devices.items():
        if name != '' and 'Smart Hub' in name:
            label['text'] = address
 
            global button_run, button_stop, button_disconnect, req
            button_connect['state'] = 'disabled'
            button_run['state'] = 'normal'
            button_stop['state'] = 'normal'
            button_disconnect['state'] = 'normal'
            button_up['state'] = 'normal'
            button_down['state'] = 'normal'
 
            req = GATTRequester(address, True, "hci0")
            break

執行電機:

def motor_run():
    global req
    if req != None:
        req.write_by_handle(0x3d, str(bytearray([0x01, 0x01, 0x01, 0x64]))
)

第一個位元組代表埠號。主控裝置上有兩個埠,選擇對應的埠號。

在這裡插入圖片描述 通過gatttool也可以檢視埠變化:

[98:07:2D:DD:98:56][LE]> char-read-hnd 0015                                                                     
Characteristic value/descriptor: 01 01 00 01 01 00 00 00 01 00 00 00                                                    
[98:07:2D:DD:98:56][LE]> char-read-hnd 0015                                                                             
Characteristic value/descriptor: 02 01 01 01 01 00 00 00 01 00 00 00

最後一個位元組代表速度。可以通過程式碼動態調節:

MAX_SPEED = 100
MIN_SPEED = 1
SPEED_CHANGE = 4
 
current_speed = 100
req = None
 
def motor_up():
    global req, current_speed
    if req != None:
        if current_speed == MAX_SPEED:
            return
 
        current_speed += SPEED_CHANGE
        req.write_by_handle(HANDLE, str(bytearray([0x01, 0x01, 0x01, current_speed])))
        sleep(WEDO_DELAY)
 
def motor_down():
    global req, current_speed
    if req != None:
        if current_speed == MIN_SPEED:
            return
 
        current_speed -= SPEED_CHANGE
        req.write_by_handle(HANDLE, str(bytearray([0x01, 0x01, 0x01, current_speed])))
        sleep(WEDO_DELAY)

執行程式的時候,如果沒有root許可權,會出現錯誤。

在這裡插入圖片描述 執行介面程式需要用gksudo而不是sudo

在這裡插入圖片描述 在Windows中可以通過SmarTTY來顯示遠端的GUI。

在這裡插入圖片描述

原始碼

https://github.com/yushulx/lego-w