370 lines
12 KiB
Bash
370 lines
12 KiB
Bash
#!/bin/bash
|
|
|
|
if [ $(id -u) -ne 0 ]; then
|
|
echo "Installer must be run as root."
|
|
echo "Try 'sudo bash $0'"
|
|
exit 1
|
|
fi
|
|
|
|
# FEATURE PROMPTS ----------------------------------------------------------
|
|
# Installation doesn't begin until after all user input is taken.
|
|
|
|
INSTALL_HALT=0
|
|
INSTALL_ADC=0
|
|
INSTALL_GADGET=0
|
|
|
|
# Given a list of strings representing options, display each option
|
|
# preceded by a number (1 to N), display a prompt, check input until
|
|
# a valid number within the selection range is entered.
|
|
selectN() {
|
|
for ((i=1; i<=$#; i++)); do
|
|
echo $i. ${!i}
|
|
done
|
|
echo
|
|
REPLY=""
|
|
while :
|
|
do
|
|
echo -n "SELECT 1-$#: "
|
|
read
|
|
if [[ $REPLY -ge 1 ]] && [[ $REPLY -le $# ]]; then
|
|
return $REPLY
|
|
fi
|
|
done
|
|
}
|
|
|
|
clear
|
|
|
|
echo "This script installs Adafruit Snake Eyes Bonnet"
|
|
echo "software for Raspberry Pi. It's best to dedicate"
|
|
echo "a card to this, not for systems in regular use."
|
|
echo
|
|
echo "Target system (may be different than current host):"
|
|
selectN "Raspberry Pi 4, Pi 400, or Compute Module 4" "All other models"
|
|
#if [[ $? -eq "1" ]]; then
|
|
if [ $? -eq "1" ]; then
|
|
IS_PI4=1
|
|
fi
|
|
|
|
if [ $IS_PI4 ]; then
|
|
# Verify this is Raspbian Desktop OS (X11 should exist)
|
|
if [ ! -d "/usr/bin/X11" ]; then
|
|
echo "Target system is Pi 4/400/CM4 but this appears to"
|
|
echo "be a \"Lite\" OS. \"Desktop\" OS is required."
|
|
echo -n "Continue anyway? [y/N]"
|
|
read
|
|
if [[ ! "$REPLY" =~ ^(yes|y|Y)$ ]]; then
|
|
echo "Canceled."
|
|
exit 0
|
|
fi
|
|
fi
|
|
else
|
|
# Verify this is Raspbian Lite OS (X11 should NOT exist)
|
|
if [ -d "/usr/bin/X11" ]; then
|
|
echo "Target system is not Pi 4/400/CM4 yet this appears to"
|
|
echo "be a \"Desktop\" OS. \"Lite\" OS (Legacy) is required."
|
|
echo -n "Continue anyway? [y/N]"
|
|
read
|
|
if [[ ! "$REPLY" =~ ^(yes|y|Y)$ ]]; then
|
|
echo "Canceled."
|
|
exit 0
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
SCREEN_VALUES=(-o -t -i)
|
|
SCREEN_NAMES=("OLED (128x128)" "TFT (128x128)" "IPS (240x240)" HDMI)
|
|
RADIUS_VALUES=(128 128 240)
|
|
OPTION_NAMES=(NO YES)
|
|
echo
|
|
echo "Select screen type:"
|
|
selectN "${SCREEN_NAMES[0]}" \
|
|
"${SCREEN_NAMES[1]}" \
|
|
"${SCREEN_NAMES[2]}" \
|
|
"${SCREEN_NAMES[3]}"
|
|
SCREEN_SELECT=$?
|
|
|
|
echo -n "Install GPIO-halt utility? [y/N] "
|
|
read
|
|
if [[ "$REPLY" =~ (yes|y|Y)$ ]]; then
|
|
INSTALL_HALT=1
|
|
echo -n "GPIO pin for halt: "
|
|
read
|
|
HALT_PIN=$REPLY
|
|
fi
|
|
|
|
echo -n "Install Bonnet ADC support? [y/N] "
|
|
read
|
|
if [[ "$REPLY" =~ (yes|y|Y)$ ]]; then
|
|
INSTALL_ADC=1
|
|
fi
|
|
|
|
echo -n "Install USB Ethernet gadget support? (Pi Zero) [y/N] "
|
|
read
|
|
if [[ "$REPLY" =~ (yes|y|Y)$ ]]; then
|
|
INSTALL_GADGET=1
|
|
fi
|
|
|
|
echo
|
|
echo "Screen type: ${SCREEN_NAMES[$SCREEN_SELECT-1]}"
|
|
if [ $INSTALL_HALT -eq 1 ]; then
|
|
echo "Install GPIO-halt: YES (GPIO$HALT_PIN)"
|
|
else
|
|
echo "Install GPIO-halt: NO"
|
|
fi
|
|
echo "Bonnet ADC support: ${OPTION_NAMES[$INSTALL_ADC]}"
|
|
echo "Ethernet USB gadget support: ${OPTION_NAMES[$INSTALL_GADGET]}"
|
|
if [ $SCREEN_SELECT -eq 3 ]; then
|
|
echo "Video resolution will be set to 1280x720, no overscan"
|
|
else
|
|
echo "Video resolution will be set to 640x480, no overscan"
|
|
fi
|
|
|
|
echo
|
|
echo "Installation steps include:"
|
|
echo "- Update package index files (apt-get update)"
|
|
echo "- Install Python libraries: numpy, pi3d, svg.path,"
|
|
echo " rpi-gpio, python3-dev, python3-pil"
|
|
echo "- Install Adafruit eye code and data in /boot"
|
|
echo "- Set HDMI resolution, GPU RAM, disable overscan"
|
|
if [ $IS_PI4 ]; then
|
|
echo "- Disable desktop, configure video driver"
|
|
fi
|
|
echo "- Enable SPI0 and SPI1 peripherals if needed"
|
|
echo
|
|
echo "THIS IS A ONE-WAY OPERATION, NO UNINSTALL PROVIDED."
|
|
echo "EXISTING INSTALLATION, IF ANY, WILL BE OVERWRITTEN."
|
|
echo "Run time ~10 minutes. Reboot required."
|
|
echo
|
|
echo -n "Do you understand and want to proceed? [y/N]"
|
|
read
|
|
if [[ ! "$REPLY" =~ ^(yes|y|Y)$ ]]; then
|
|
echo "Canceled."
|
|
exit 0
|
|
fi
|
|
|
|
# START INSTALL ------------------------------------------------------------
|
|
# All selections are validated at this point...
|
|
|
|
# Given a filename, a regex pattern to match and a replacement string,
|
|
# perform replacement if found, else append replacement to end of file.
|
|
# (# $1 = filename, $2 = pattern to match, $3 = replacement)
|
|
reconfig() {
|
|
grep $2 $1 >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# Pattern found; replace in file
|
|
sed -i "s/$2/$3/g" $1 >/dev/null
|
|
else
|
|
# Not found; append (silently)
|
|
echo $3 | sudo tee -a $1 >/dev/null
|
|
fi
|
|
}
|
|
|
|
# Same as above, but appends to same line rather than new line
|
|
reconfig2() {
|
|
grep $2 $1 >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# Pattern found; replace in file
|
|
sed -i "s/$2/$3/g" $1 >/dev/null
|
|
else
|
|
# Not found; append to line (silently)
|
|
sed -i "s/$/ $3/g" $1 >/dev/null
|
|
fi
|
|
}
|
|
|
|
echo
|
|
echo "Starting installation..."
|
|
echo "Updating package index files..."
|
|
apt-get update
|
|
|
|
echo "Installing Python libraries..."
|
|
# WAS: apt-get install -y --force-yes python-pip python-dev python-imaging python-smbus
|
|
apt-get install -y python3-pip python3-dev python3-pil python3-smbus libatlas-base-dev
|
|
# WAS: pip3 install numpy pi3d==2.34 svg.path rpi-gpio adafruit-ads1x15
|
|
pip3 install numpy pi3d svg.path rpi-gpio adafruit-blinka adafruit-circuitpython-ads1x15
|
|
# smbus and Blinka+ADC libs are installed regardless whether ADC is
|
|
# enabled; simplifies the Python code a little (no "uncomment this")
|
|
|
|
echo "Installing Adafruit code and data in /boot..."
|
|
cd /tmp
|
|
curl -LO https://github.com/adafruit/Pi_Eyes/archive/master.zip
|
|
unzip master.zip
|
|
# Moving between filesystems requires copy-and-delete:
|
|
cp -r Pi_Eyes-master /boot/Pi_Eyes
|
|
rm -rf master.zip Pi_Eyes-master
|
|
if [ $INSTALL_HALT -ne 0 ]; then
|
|
echo "Installing gpio-halt in /usr/local/bin..."
|
|
curl -LO https://github.com/adafruit/Adafruit-GPIO-Halt/archive/master.zip
|
|
unzip master.zip
|
|
cd Adafruit-GPIO-Halt-master
|
|
make
|
|
mv gpio-halt /usr/local/bin
|
|
cd ..
|
|
rm -rf Adafruit-GPIO-Halt-master
|
|
fi
|
|
|
|
# CONFIG -------------------------------------------------------------------
|
|
|
|
echo "Configuring system..."
|
|
|
|
if [ $IS_PI4 ]; then
|
|
# Make desktop system to boot to console (from raspi-config script):
|
|
systemctl set-default multi-user.target
|
|
ln -fs /lib/systemd/system/getty@.service /etc/systemd/system/getty.target.wants/getty@tty1.service
|
|
rm -f /etc/systemd/system/getty@tty1.service.d/autologin.conf
|
|
|
|
# Pi3D requires "fake" KMS overlay to work. Check /boot/config.txt for
|
|
# vc4-fkms-v3d overlay present and active. If so, nothing to do here,
|
|
# module's already configured.
|
|
grep '^dtoverlay=vc4-fkms-v3d' /boot/config.txt >/dev/null
|
|
if [ $? -ne 0 ]; then
|
|
# fkms overlay not present, or is commented out. Check if vc4-kms-v3d
|
|
# (no 'f') is present and active. That's normally the default.
|
|
grep '^dtoverlay=vc4-kms-v3d' /boot/config.txt >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# It IS present. Comment out that line for posterity, and insert the
|
|
# 'fkms' item on the next line.
|
|
sed -i "s/^dtoverlay=vc4-kms-v3d/#&\ndtoverlay=vc4-fkms-v3d/g" /boot/config.txt >/dev/null
|
|
else
|
|
# It's NOT present. Silently append 'fkms' overlay to end of file.
|
|
echo dtoverlay=vc4-fkms-v3d | sudo tee -a /boot/config.txt >/dev/null
|
|
fi
|
|
fi
|
|
|
|
# Disable screen blanking in X config
|
|
echo "Section \"ServerFlags\"
|
|
Option \"BlankTime\" \"0\"
|
|
Option \"StandbyTime\" \"0\"
|
|
Option \"SuspendTime\" \"0\"
|
|
Option \"OffTime\" \"0\"
|
|
Option \"dpms\" \"false\"
|
|
EndSection" > /etc/X11/xorg.conf
|
|
|
|
fi
|
|
|
|
# Disable overscan compensation (use full screen):
|
|
raspi-config nonint do_overscan 1
|
|
|
|
# Dedicate 128 MB to the GPU:
|
|
sudo raspi-config nonint do_memory_split 128
|
|
|
|
# HDMI settings for Pi eyes
|
|
reconfig /boot/config.txt "^.*hdmi_force_hotplug.*$" "hdmi_force_hotplug=1"
|
|
reconfig /boot/config.txt "^.*hdmi_group.*$" "hdmi_group=2"
|
|
reconfig /boot/config.txt "^.*hdmi_mode.*$" "hdmi_mode=87"
|
|
if [ $SCREEN_SELECT -eq 3 ]; then
|
|
# IPS display - set HDMI to 1280x720
|
|
reconfig /boot/config.txt "^.*hdmi_cvt.*$" "hdmi_cvt=1280 720 60 1 0 0 0"
|
|
else
|
|
# All others - set HDMI to 640x480
|
|
reconfig /boot/config.txt "^.*hdmi_cvt.*$" "hdmi_cvt=640 480 60 1 0 0 0"
|
|
fi
|
|
|
|
|
|
# Enable I2C for ADC
|
|
if [ $INSTALL_ADC -ne 0 ]; then
|
|
raspi-config nonint do_i2c 0
|
|
fi
|
|
|
|
if [ $INSTALL_HALT -ne 0 ]; then
|
|
# Add gpio-halt to /rc.local:
|
|
grep gpio-halt /etc/rc.local >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# gpio-halt already in rc.local, but make sure correct:
|
|
sed -i "s/^.*gpio-halt.*$/\/usr\/local\/bin\/gpio-halt $HALT_PIN \&/g" /etc/rc.local >/dev/null
|
|
else
|
|
# Insert gpio-halt into rc.local before final 'exit 0'
|
|
sed -i "s/^exit 0/\/usr\/local\/bin\/gpio-halt $HALT_PIN \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
fi
|
|
fi
|
|
|
|
# If using OLED, TFT or IPS, enable SPI and install fbx2 and eyes.py,
|
|
# else (HDMI) skip SPI, fbx2 and install cyclops.py (single eye)
|
|
if [ $SCREEN_SELECT -ne 4 ]; then
|
|
|
|
# Enable SPI0 using raspi-config
|
|
raspi-config nonint do_spi 0
|
|
|
|
# Enable SPI1 by adding overlay to /boot/config.txt
|
|
reconfig /boot/config.txt "^.*dtparam=spi1.*$" "dtparam=spi1=on"
|
|
reconfig /boot/config.txt "^.*dtoverlay=spi1.*$" "dtoverlay=spi1-3cs"
|
|
|
|
# Adjust spidev buffer size to 8K (default is 4K)
|
|
reconfig2 /boot/cmdline.txt "spidev\.bufsiz=.*" "spidev.bufsiz=8192"
|
|
|
|
SCREEN_OPT=${SCREEN_VALUES[($SCREEN_SELECT-1)]}
|
|
|
|
# Auto-start fbx2 on boot
|
|
grep fbx2 /etc/rc.local >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# fbx2 already in rc.local, but make sure correct:
|
|
sed -i "s/^.*fbx2.*$/\/boot\/Pi_Eyes\/fbx2 $SCREEN_OPT \&/g" /etc/rc.local >/dev/null
|
|
else
|
|
# Insert fbx2 into rc.local before final 'exit 0'
|
|
sed -i "s/^exit 0/\/boot\/Pi_Eyes\/fbx2 $SCREEN_OPT \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
fi
|
|
|
|
RADIUS=${RADIUS_VALUES[($SCREEN_SELECT-1)]}
|
|
# Auto-start eyes.py on boot
|
|
grep eyes.py /etc/rc.local >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# eyes.py already in rc.local, but make sure correct:
|
|
if [ $IS_PI4 ]; then
|
|
sed -i "s/^.*eyes.py.*$/cd \/boot\/Pi_Eyes;xinit \/usr\/bin\/python3 eyes.py --radius $RADIUS \:0 \&/g" /etc/rc.local >/dev/null
|
|
else
|
|
sed -i "s/^.*eyes.py.*$/cd \/boot\/Pi_Eyes;python3 eyes.py --radius $RADIUS \&/g" /etc/rc.local >/dev/null
|
|
fi
|
|
else
|
|
# Insert eyes.py into rc.local before final 'exit 0'
|
|
if [ $IS_PI4 ]; then
|
|
sed -i "s/^exit 0/cd \/boot\/Pi_Eyes;xinit \/usr\/bin\/python3 eyes.py --radius $RADIUS \:0 \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
else
|
|
sed -i "s/^exit 0/cd \/boot\/Pi_Eyes;python3 eyes.py --radius $RADIUS \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
fi
|
|
fi
|
|
|
|
else
|
|
|
|
# Auto-start cyclops.py on boot
|
|
grep cyclops.py /etc/rc.local >/dev/null
|
|
if [ $? -eq 0 ]; then
|
|
# cyclops.py already in rc.local, but make sure correct:
|
|
if [ $IS_PI4 ]; then
|
|
sed -i "s/^.*cyclops.py.*$/cd \/boot\/Pi_Eyes;xinit \/usr\/bin\/python3 cyclops.py \:0 \&/g" /etc/rc.local >/dev/null
|
|
else
|
|
sed -i "s/^.*cyclops.py.*$/cd \/boot\/Pi_Eyes;python3 cyclops.py \&/g" /etc/rc.local >/dev/null
|
|
fi
|
|
else
|
|
# Insert cyclops.py into rc.local before final 'exit 0'
|
|
if [ $IS_PI4 ]; then
|
|
sed -i "s/^exit 0/cd \/boot\/Pi_Eyes;xinit \/usr\/bin\/python3 cyclops.py \:0 \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
else
|
|
sed -i "s/^exit 0/cd \/boot\/Pi_Eyes;python3 cyclops.py \&\\nexit 0/g" /etc/rc.local >/dev/null
|
|
fi
|
|
fi
|
|
|
|
fi
|
|
|
|
if [ $INSTALL_GADGET -ne 0 ]; then
|
|
reconfig /boot/config.txt "^.*dtoverlay=dwc2.*$" "dtoverlay=dwc2"
|
|
grep "modules-load=dwc2,g_ether" /boot/cmdline.txt >/dev/null
|
|
if [ $? -ne 0 ]; then
|
|
# Insert ethernet gadget into config.txt after 'rootwait'
|
|
sed -i "s/rootwait/rootwait modules-load=dwc2,g_ether/g" /boot/cmdline.txt >/dev/null
|
|
fi
|
|
fi
|
|
|
|
# PROMPT FOR REBOOT --------------------------------------------------------
|
|
|
|
echo "Done."
|
|
echo
|
|
echo "Settings take effect on next boot."
|
|
echo
|
|
echo -n "REBOOT NOW? [y/N] "
|
|
read
|
|
if [[ ! "$REPLY" =~ ^(yes|y|Y)$ ]]; then
|
|
echo "Exiting without reboot."
|
|
exit 0
|
|
fi
|
|
echo "Reboot started..."
|
|
reboot
|
|
exit 0
|