LEGO SPIKEプライムとRaspberryPiをUARTで連携する

今回はLEGO SPIKEPrimeとRaspberryPiを連携して使用していく前段階としてUARTを用いて信号をやり取りする仕組みを作ってみた。
今後はこれを応用してRaspberryPi側でカメラなどを用いて機械学習をしその結果に応じてSPIKEのモーターなどを駆動させる連携をさせていくう予定。

使うもの

• LEGO SPIKE PRIME
• RaspberryPi4(3ではURATの設定が変わる可能性がある)
• Breadboard Connector Kit for SPIKE Prime Ultrasonic sensor cable(要はんだ付け)
• T6　トルクスドライバー (超音波センサーを分解するのに必要)
• RaspberryPiを環境構築するためのネットワーク環境
  (RaspberrypiOSを用いるため別途SDHCカードも必要)
  

RaspberryPiの環境構築

OS:RaspberryPi OSを利用する
PCからリモートアクセスする場合はVNCviewerを使うと便利である。

RaspberryPiOSの準備が完了したら必要なものをインストールしていく。

• 今回はPythonエディタMuを利用する。
  MuはPython3を簡単に実行できるエディタな他、SPIKEをUSB接続することでSPIKE内にあるMicroPythonを直接実行することができる。
  RaspberryPi上でターミナルを起動してコマンドを入力してMuをインストールしていく。

1.Raspberrypi OSのリポジトリから必要なものをインストールする。

sudo apt-get install python3-pyqt5 python3-pyqt5.qsci python3-pyqt5.qtserialport
python3-pyqt5.qtsvg python3-dev python3-gpiozero python3-pgzero libxmlsec1-dev libxml2
libxml2-dev

2.virtualenvの環境を作成する。

sudo pip3 install virtualenv

virtualenv -p /usr/bin/python3 --system-site-packages ~/mu-venv

3.作成したvirtualenvの環境をアクティブにする。
source ~/mu-venv/bin/activate

4.Muをgitからクローンする。
git clone https://github.com/mu-editor/mu.git ~/mu-source

5.virtualenvを有効にした状態で、pipでRaspberry Pi用のPythonパッケージをインストールする。

cd ~/mu-source

pip install -e ".[dev]"

6.Muのエディタを起動する。

mu-editor

これでMuの導入は完了。
次回以降Muを起動するときは
source ~/mu-venv/bin/activate
mu-editor
の順でターミナルでコマンドを実行すれば起動できる。(複数のターミナルで複数のMuを起動することもできる)

RaspberryPiのGPIOを有効にする(UART5を利用する)

SPIKEとの信号のやりとりに利用するGPIOを利用できるようにRaspberryPiの設定を行っていく。

1. RaspberryPi内の「/boot/config.txt」を編集する。

ファイル内の末尾に「dtoverlay=uart5」を追加で入力し保存する。
ファイルの編集はnanoエディターやviエディターを利用する。
以下はnanoエディターを利用して編集する際のコマンド。

sudo nano /boot/config.txt
↓
ファイル内編集、保存

2.新しいポートが開いてるかコマンドで確認する。

ls /dev/ttyA*
↓
表示されるリストの中にttyAMA1があればOK

3.RaspberrypiとBreadboard Connector Kitを用いてSPIKEと接続する。

4.SPIKEをRaspberryPiのUSBポートに接続する。

これで通信を行う準備は完了。

通信のやり取りを行ってみる。

以下の手順でプログラムを作成し実行することで通信を行ってみた。
1.RaspberryPiでMuエディタを2つ立ち上げる。
RaspberryPi上でターミナルを2つ立ち上げ、それぞれで
source ~/mu-venv/bin/activate
mu-editor
を実行することでMuのウィンドウが2つ立ち上がる。

2.2つ立ち上げたMuエディタのモードを変更する。
一方はPython3モード、一方はSPIKE MicroPythonモードに変更する。

3.RaspberryPi側、SPIKE側のPythonプログラムをそれぞれMuエディタにて作成する。
それぞれのPythonプログラムは以下

Raspberrypi側(Python3モード側)

import time
import serial,time


ser = serial.Serial('/dev/ttyAMA1',115200)

while True:
    if ser.in_waiting: #if data, read until \n
        x=ser.readline()
        print(x)
        ser.write(b’read: %s\n’ % x)

SPIKE側(SPIKE MicroPythonモード側)

import hub, time

serial = hub.port.C
serial.mode(hub.port.MODE_FULL_DUPLEX)
time.sleep(1)
serial.baud(115200)

serial.read(1000)
serial.write("\n testing \n\n")
serial.read(1000)

4.RaspberryPiとSPIKEの接続を行い、2つのMuエディタにてそれぞれのPythonプログラムを実行する。
接続は以下の画像をのようにしている。(Aポートのモーターは次の項目で使用する。)

プログラム内のwriteの記述部分が送信されている。
実行するとPython3側Muエディタのコンソール部分に受信したデータが出力される。(testing)

受信をフラグにしてSPIKEを制御する

信号のやり取りはできたので、それをSPIKEの制御に当てる。

同じようにそれぞれのMuエディタでプログラムを作成する。

Raspberrypi側(Python3モード側)

import time
import serial

# serial ini
ser = serial.Serial('/dev/ttyAMA1', 115200)

# start send
print('start')
while True:
    time.sleep(5.0)
    ser.write(b't')
    print('send test')

SPIKE側(SPIKE MicroPythonモード側)

import hub, time
from spike import Motor

# serial ini
serial = hub.port.C
serial.mode(hub.port.MODE_FULL_DUPLEX)
time.sleep(1)
serial.baud(115200)

# motor ini
motor = Motor('A')

# start recive
print('start')
while True:
    reply = serial.read(1000)
    if reply:
        response = reply.decode('utf-8')
        print(response)
        if response == 't':
            motor.run_for_degrees(360,speed=50)
            time.sleep(1.0)

プログラムを作成できたらSPIKEのAポートにMモーターを接続する。
プログラムを実行して動作を確認するが、今回は受信待機状態になるSPIKE側から先に実行し、
その後Raspberrypi側を実行する。

正常に動作していればRaspberrypi側から5秒おきにバイト型のtが送信され、
受け取ったSPIKE側がutf-8にデコードし、tを受け取っていることを判定できたらモーターを360度回すという処理が行われる。