PiClusterNode/node at main · doughadfield/PiClusterNode

executable file
395 lines (364 loc) · 10 KB
#!/bin/bash
# Access cluster nodes for serial I/O or power cycle
# This program is used with the Cluster controller board to control up to 8 Rpi cluster nodes
# Version V2 - for use with V2 board (also works with V1, but power monitor unavailable)
# Board V1 used 74HC138/74HC251 for serial I/O, and 74HC259 for power enables
# fixed serial speed issue and delivered power enable persistance
# Problem: no power state monitoring for nodes (is the node on or off?)
# Board V2 added a second 74HC251 for direct monitoring of power enable lines.
# A 74HC138 (for TX) and 74HC251 (for RX) (de)multiplexers for fanout of TX and RX UART signals
# from Pi1 to nodes 2-8 (pi2 has fixed ttyAMA2 - ttyAMA10 to pi1)
# One x 74HC259 addressable latch, driving the Enable lines of the DC/DC
# converters powering the Pi boards. CLK and DATA lines on GPIO 17 and 18
# GPIO pins 2, 3 and 4 are address lines 0, 1 and 2 for above signals
# Serial1 pins from pi1 via (de)mulitplexers, to serial0 of cluster nodes
# (Board V2) Addition of power monitor using a second 74HC251 connected to the power enable
# outputs from the 74HC259, so as to sample the current power status of nodes.
# Output of power monitor is GPIO 27 (was serial enable in previous board versions)
# GPIO 2,3,4 are address select pins for I/O multiplexers
# pin 2 = bit 0
# pin 3 = bit 1
# pin 4 = bit 2
# GPIO 18 is power pin -  active HIGH = POWEROFF!
# Power off signal is latched by GPIO 17 CLK pin pulsing low (active LOW)
# GPIO 17 is power latch Clk pin - high latches value, low enables Data to selected output
# GPIO 27 WAS serial enable (in V1), but now is INPUT from power monitor mpx.
serialport="/dev/ttyAMA0"				# outgoing UART into demultiplexer board
baudrate=115200						# Baud rate set for node serial ports
interactive=1						# Interactive flag for power commands
# Setup function configures GPIO pins to drive multiplexers
# Initialises GPIO pins to node 0 (pi1) selected, with power ON and Clk DISABLED
setup () {	# Set up GPIO pins if not already done
	if [[ $(pinctrl get 17) != *op* ]]
		pinctrl set 17 op dh			# set Power Clk pin to output and HIGH (disable)
		echo "Initialising GPIO pins"		# echo acts as time delay for enable pin settling
		pinctrl set 2,3,4,18 op dl		# set Address and PowerOn pins to output and LOW
							# default pins to 0, for address 000 and power 0 (ON)
		pinctrl set 27 ip pd			# GPIO27 is input (with pull-down)
		echo "GPIO pins initialised"
# Get selected node number from select pins and set global variable "selectednode"
nodenum() {
	selectednode=1					# start at count of 1
	if [[ $(pinctrl get 2) == *hi* ]]
		selectednode=$((selectednode + 1))
	if [[ $(pinctrl get 3) == *hi* ]]
		selectednode=$((selectednode + 2))
	if [[ $(pinctrl get 4) == *hi* ]]
		selectednode=$((selectednode + 4))
# Print out current state of GPIO pins
nodegpio () {
	if [[ $(pinctrl get 17) == *op* ]]
		echo "node select pins:"
		pinctrl get 2,3,4 | cut -d ' ' -f 2,7
		echo "Selected node =" $selectednode	# global variable retains selected node
		echo "Power Enable pins:"
		pinctrl get 17,18 | cut -d ' ' -f 1,6
		if [[ $(pinctrl get 17) == *hi* ]]
			echo "Power CLK High (Disabled)"
			echo "Power CLK Low (ENABLED!)"
		if [[ $(pinctrl get 18) == *hi* ]]
			echo "Power Data High (POWEROFF!)"
			echo "Power Data Low (PowerOn)"
		echo "GPIO pins not yet initialised"
		echo -n "Initialise GPIO pins [y/N]? "
		read reply
		if [[ $reply == "Y" || $reply == "y" ]]
			nodegpio
			echo "GPIO pins not initialised"
# Print currently selected node and power states for all nodes
nodestatus () {
	currentnode=$selectednode			# save current selection for restore below
	for i in 1 2 3 4 5 6 7 8
		nodeselect $i 
		echo -n "node $selectednode POWER is "
		if [[ $(pinctrl get 27) == *hi* ]]
			echo "OFF"
			echo "ON"
	nodeselect $currentnode				# restore previously selected node
	echo "Currently selected node =" $selectednode
# Set GPIO address pins of multiplexers to specified node
nodeselect () {
	pinctrl set 17 dh				# ensure power gate is DISABLED (high)
	pinctrl set 18 dl				# ensure power D pin is in ON state (low)
	case $1 in
	1)	pinctrl set 2,3,4 dl
	2)	pinctrl set 2 dh
		pinctrl set 3,4 dl
	3)	pinctrl set 2,4 dl
		pinctrl set 3 dh
	4)	pinctrl set 2,3 dh
		pinctrl set 4 dl
	5)	pinctrl set 2,3 dl
		pinctrl set 4 dh
	6)	pinctrl set 3 dl
		pinctrl set 2,4 dh
	7)	pinctrl set 2 dl
		pinctrl set 3,4 dh
	8)	pinctrl set 2,3,4 dh
	*)	echo -e "\nInvalid node: \"$1\""
	nodenum						# update global node number with new selection
# Print usage string
	echo -e "\nUsage: $scriptname <command> [<node number or range>]"
	echo "Commands:
	poweroff [n | n-n | n,n,n... | \"all\"]
	poweron [n | n-n | n,n,n... | \"all\"]
	powercycle [n | n-n | n,n,n... | \"all\"]"
	echo "n = Node number: 1 to 8"
	exit						# after printing usage we always exit
# Power off selected node
setpoweroff () {
	if [[ $(pinctrl get 27) == *hi* ]]		# test actual status of power enable line
		echo "Node $selectednode is already powered OFF"
		if [[ $interactive == 1 ]]
			echo -n "Press Return to power OFF node $selectednode "
			read input
		pinctrl set 17 dh			# ensure Clk disabled to start with
		pinctrl set 18 dh			# set power Data pin to POWER OFF
		pinctrl set 17 dl			# set power Clk pin to Enable
		pinctrl set 17 dh			# set power Clk pin to Latch
		pinctrl set 18 dl			# leave power Data pin in ON state
		echo "Node $selectednode powered OFF"
# Power on selected node
setpoweron () {
	if [[ $(pinctrl get 27) == *hi* ]]		# test actual status of power enable line
		if [[ $interactive == 1 ]]
			echo -n "Press Return to power ON node $selectednode and connect to serial "
			read input
			echo "Powering on node $selectednode"
		pinctrl set 18 dl			# set power pin to POWER ON
		pinctrl set 17 dl			# set power Clk pin to Enable
		pinctrl set 17 dh			# set power Clk pin to Latch
		if [[ $interactive == 1 ]]
			sleep 1				# needed for pi firmware baud rate stability
			minicom -D $serialport -b $baudrate
		echo "Node $selectednode is already powered ON"
# Execute command in first argument
execute() {
	case $1 in
	"gpio")						# Print current state of GPIO pins
	"status")					# Print status of mux GPU pins
		nodestatus
	"select")					# Set GPIO pins to select specified node
		echo Selected node: $selectednode
	"connect")					# connect to serial port
		minicom -D $serialport -b $baudrate
	"powercycle")					# Power cycle node 
		setpoweroff
		setpoweron
	"poweroff")					# Power off node and exit (power state latched persistently)
		setpoweroff
	"poweron")					# Power on specified node
		setpoweron
	*)						# Default action - print usage
		echo -e "\nUnknown command: \"$1\""
# Noderange: interpret range of nodes to execute on
noderange() {
	case ${#2} in					# interpret based on length of arg 2
		nodeselect $2				# select node from passed argument
		execute $1				# execute passed command on slected node
		exit					# all done, so don't drop to next code
		echo "invalid node range"
		usage					# just 2 chars is illegal value
	3) 						# 3 chars in arg means a range has been specified (n-n)
		range=$2				# use a named variable for range arg
		if [[ $range == "all" ]]		# "all" is a special case for arg 2
			range="1-8"			# convert to range notation
		readarray -d "-" -t array <<< "$range"	# test for node range specification
		if [[ ${#array[@]} >1 ]]		# a range has been specified
			arg1="${array[0]}"
			arg2="${array[1]}"
			if (( $arg1 < 1 || $arg1 > 8 || $arg2 < 1 || $arg2 > 8 || $arg1 >= $arg2 ))
				echo "\"$range\" - invalid range"
				interactive=0		# turn off interactive for power commands
				echo -n "really $1 nodes $range [y/N] "
				read input
				if [[ $input != "y" && $input != "Y" ]]
					echo aborting!
				for ((i=$arg1;i<=$arg2;i++))
					nodeselect $i
					execute $1 
			exit				# all done, so don't drop to next code
	# If we get here arg has 3 or more chars and must be a series (n,n,n...)
	readarray -d "," -t array <<< "$2"		# test for node series specification
	if [[ ${#array[@]} >1 ]]			# multiple nodes have been specified
		for arg in "${array[@]}"		# validate all numbers in the series
			if (( $arg < 1 || $arg > 8 ))
				echo invalid node numer in range
		interactive=0				# turn off interactive for power commands
		echo -n "really $1 nodes $2 [y/N] "
		read input
		if [[ $input != "y" && $input != "Y" ]]
			echo aborting!
		for arg in "${array[@]}"
			nodeselect $arg
			execute $1 
		exit					# all done, so don't drop down to next code
	echo illegal range or node number		# if we get here, no range or series specified
# Program proper starts here:
scriptname=`basename $0`				# needed for "usage" function
setup							# ensure GPIO pins are initialised
case $# in						# execute based on number of args passed
	0)						# no arguments passed
	1)						# just the command given
		nodenum					# set "selectednode" from current state of gpio pins
		execute $1				# now execute the command passed as arg 1
	2)						# node number given as argument
		case $1 in				# does a node number make sense for specified command
			gpio|status)
				echo "too many arguments for command \"$1\""
			select|connect)
				if (( ${#2} > 1 ))	# range makes no sense for connect command
					echo "specify only a single node for \"$1\" command"
		noderange $1 $2				# interpret range of nodes to execute on
		echo -e "\nToo many arguments"
Provide feedback

Saved searches

Use saved searches to filter your results more quickly

FilesExpand file tree

node

Latest commit

History

node

File metadata and controls
