setup

#!/bin/bash

# setup for Raspberry PI 7" touchscreen display
# also supports HDMI display on HDMI 0

# Much is already installed and set up for this display
#  the touch screen calibration included in the image appears to work fine
#  even if the display is rotated 180 degrees
# what remains is the screensaver (blanking) and brightness control from Venus GUI
# and activating a local GUI
#
# Do NOT install this package if you do not want a local display (run Venus "headless")
#
# for HDMI displays, no brightness control is possible, only screensaver
# and some displays may not turn off the backlight when blanked, but a black screen will be displayed
#
# run manually initally, then called from reinstallMods after a Venus update
# to uninstall/deactivate, call this script with "uninstall" as first parameter
#
# /u-boot/config.txt modified to remove references to "framebuffer" which isn't needed, even for the RPI display
# and to insure overscan is not active
# and and hdmi_blanking is enabled so RPI display will power down during screen saver
# Note: the display will take longer to come out of screen saver if powered down
# but if this is not done, the display backlight will remain on and be visible as a glow
#
# add HDMI configuration to /u-boot/config.txt
# You may need to make additional changes/additions manaually to support the display in use
#
# Note, this script may reboot the system to comlete all operations
# Dimming the RPI display requires the rpi-backlight overlay
# A reboot is required to activate the overlay and the config.txt changes

headlessFlag=/etc/venus/headless
restartGui=false
overlayFile="/u-boot/overlays/rpi-backlight.dtbo"
configFile="/u-boot/config.txt"

blankDeviceFile=/etc/venus/blank_display_device
dimDeviceFile=/etc/venus/backlight_device
# no log file for this package
packageLogFile=""

setupForTsCalibrate=false


#### following line incorporates helper resources into this script
source "/data/SetupHelper/HelperResources/IncludeHelpers"
#### end of lines to include helper resources

#### running manually and OK to proceed - prompt for input
if [ $scriptAction == 'NONE' ] ; then
    # display innitial message
    echo
    echo "This package sets up the Raspberry PI for the offical RPI touchscreen"
    echo "  and for some other DSI or HDMI displays"
    echo "Some hand editing of config.txt may be needed for an HDMI display"
    echo
    echo "The screen saver will switch to a black screen"
    echo "  with the backlight still on for HDMI displays that do not support standby"
    echo "Display brighness controll is enabled only for the RPI touchscreen"
    echo
    echo "Uninstalling disables the local display (sets headless)"

    standardActionPrompt 'MORE_PROMPTS'
    # PROMPT is left over from a previous version of SetupHelper
    #  it is included here for backward compatibility
    if [ $scriptAction == 'NONE' ] || [ $scriptAction == 'PROMPT' ]; then
        echo
        echo "This script supports the following displays:"
        echo "    (R) Raspberry PI touchscreen - may also work with other DSI displays"
        echo "    (H) HDMI display connected to port 0 (closest to power connector)"
        echo "    (C) skip display configuration and run touch screen calibration only"
        echo
		customControl=false
		hdmiConfig=false
		touchScreenCalibrateOnly=false
        while true ; do
            read -p "Choose a display from the list above (R / H / C): " display
            case $display in
                [rR]*)
                    echo "RPI" > $setupOptionsDir/displayToControl
                    break
                    ;;
                [hH]*)
                    echo "HDMI" > $setupOptionsDir/displayToControl
                    hdmiConfig=true
                    break
                    ;;
                [cC]*)
					touchScreenCalibrateOnly=true
                    break
                    ;;
           esac
        done

		if $hdmiConfig ; then
			echo
			echo "HDMI displays require configuration lines to be added to /u-boot/config.txt" 
			echo "  and may also require toush screen calibraiton"
			echo
			echo "HDMI mode"
			echo "  some displays will automatically configure (A)"
			echo "  some will conform to predefined settings (P)"
			echo "  others will require custom configuration settings (C)"
			echo
			hdmiGroup="-"
			hdmiMode="-"
			horizontalResolution="-"
			verticalResolution="-"
			refreshRate="-"
			while true ; do
				read -p "Select the HDMI mode from the above list (A / P / C): " response
				case $response in
					[aA]*)	
						hdmiGroup=0
						break
						;;
					[pP]*)	
						echo
						echo "for group and mode numbers refer to"
						echo "  https://www.raspberrypi.com/documentation/computers/config_txt.html#hdmi-mode"
						while true ; do
							read -p "Enter the HDMI group number (1 = CEA, 2 = DMT): " response
							case $response in
								1 | [cC]*)
									hdmiGroup=1
									maxMode=107
									groupText="CEA"
									break
									;;
								2 | [dD]*)
									hdmiGroup=2
									maxMode=86
									groupText="DMT"
									break
									;;
								*)
									echo "invalid group $response"
									;;
							esac
						done
						while true ; do
							read -p "Enter the $groupText mode number (1 - $maxMode) : " hdmiMode
							if [ ! -z $hdmiMode ] && (( $hdmiMode > 0 )) && (( $hdmiMode <= $maxMode )); then
								break
							else
								echo "invalid mode $hdmiMode"
							fi
						done
						break
						;;
					[cC]*)
						hdmiGroup=2
						hdmiMode=87
						while true ; do
							read -p "Enter the horizontal resolution of the display: " horizontalResolution
							if [ ! -z $horizontalResolution ] && (( $horizontalResolution > 0 )) && (( $horizontalResolution < 9999 )); then
								break
							else
								echo "invalid H value $horizontalResolution"
							fi
						done
						while true ; do
							read -p "Enter the vertical resolution of the display: " verticalResolution
							if [ ! -z $verticalResolution ] && (( $verticalResolution > 0 )) && (( $verticalResolution < 9999 )); then
								break
							else
								echo "invalid V value $verticalResolution"
							fi
						done
						read -p "Enter the display refresh rate (cr for 60): " refreshRate
						if [ -z $refreshRate ]; then
							refreshRate=60
						fi
						echo "display aspect:"
						echo "  1 4:3"
						echo "  2 14:9"
						echo "  3 16:9"
						echo "  4 5:4"
						echo "  5 16:10"
						echo "  6 15:9"
						while true ; do
							read -p "Enter the display aspect ratio from the list above: " aspect
							if [ ! -z $aspect ] && (( $aspect >= 1 )) && (( $aspect <= 6 )); then
								break
							else
								echo "invalid aspect $aspect"
							fi
						done
						break
						;;
				esac
			done
			echo "$hdmiGroup $hdmiMode $horizontalResolution $verticalResolution $refreshRate $aspect" > "$setupOptionsDir/displayRates"
		fi


		if ! $touchScreenCalibrateOnly ; then
			yesNoPrompt "Rotate the display 180 degrees (y/n)?: "
			if $yesResponse ; then
				touch "$setupOptionsDir/rotateDisplay"

				if $hdmiConfig ; then
					touch "$setupOptionsDir/rotateDisplay"
					rm -f "$setupOptionsDir/altRotateDisplay"
				else
					echo "displays may respond to one of two rotate commands"
					echo "lcd_rotate rotates most DSI displays and also rotates the touch calibration"
					echo "if the display does not rotate run setup again and choose the alternate command here"
					yesNoPrompt "choose the alternate rotate command (display_lcd_rotate) (y/n)?: "
					if $yesResponse ; then
						touch "$setupOptionsDir/altRotateDisplay"
						rm -f "$setupOptionsDir/rotateDisplay"
						echo "the alternate rotate command does not rotate the touch screen"
						echo "you must run the touch screen calibration"
					else
						touch "$setupOptionsDir/rotateDisplay"
						rm -f "$setupOptionsDir/altRotateDisplay"
					fi
				fi
			else
				rm -f "$setupOptionsDir/rotateDisplay"
				rm -f "$setupOptionsDir/altRotateDisplay"
			fi

			yesNoPrompt "Enter custom blanking and dimming devices (y/n)?: "
			if $yesResponse ; then
				read -p "enter blanking device (/sys/class/...): " blankDevice
				if [ "$blankDevice" != "" ]; then
					echo $blankDevice > "$setupOptionsDir/blank_display_device"
				else
					rm -f "$setupOptionsDir/blank_display_device"
				fi
				read -p "enter dimming device (/sys/class/...): " dimDevice
				if [ "$dimDevice" != "" ]; then
					echo $dimDevice > "$setupOptionsDir/backlight_device"
				else
					rm -f "$setupOptionsDir/backlight_device"
				fi
			else
				rm -f "$setupOptionsDir/blank_display_device"
				rm -f "$setupOptionsDir/backlight_device"
			fi
			echo
			echo "Touch screen calibration requires the ts_calibtate utility from tsLib"
			echo "  and temporariliy turning off the local GUI. System must reboot twice"
			if [ "$displayToControl" == "RPI" ]; then
				echo "RPI display touch screen calibration is optional"
				echo "But other DSI displays may require it"
			fi
			echo
			yesNoPrompt "Do you want to calibrate the touch screen (y/n)?: "
		else
			echo "running touch screen calibtation - will reboot automatically"
			yesResponse=true
		fi

		if $yesResponse  ; then
			tsCalib=$(which ts_calibrate)
			if [ -z $tsCalib ]; then
				echo
				echo "ts_calibrate was not found. Installation requries an internet connection"
				yesNoPrompt "Download and install it now (y/n)?: "
				if $yesResponse ; then
					opkg update
					opkg install tslib-calibrate
				fi
			fi
			tsCalib=$(which ts_calibrate)
			if [ ! -z $tsCalib ]; then
				setupForTsCalibrate=true
			else
				logMessage "ts_calibrate was not found - can't run calibration"
				logMessage "installation will continue WITHOUT calibration !"
			fi
		fi
        scriptAction='INSTALL'
    fi
fi

#### installing
if [ $scriptAction == 'INSTALL' ]; then
	# the overlay is needed to automatically discover display type
	if [ ! -f "$overlayFile" ]; then
		logMessage "installing RPI Display Backlight overlay"
		cp "$scriptDir/rpi-backlight-overlay.dtb" "$overlayFile"
		rebootNeeded=true
		waitForOverlay=true
	else
		waitForOverlay=false
	fi

	# get selected display from NV storage
	if [ -f $setupOptionsDir/displayToControl ]; then
		displayToControl=$(cat "$setupOptionsDir/displayToControl")
	# options not yet set - use auto-discovery
	else
		if $waitForOverlay ; then
			logMessage "install options have not been set - installng overlay and rebooting for auto discovery"
			runAgain=true
			# stop setup now and come back later to finish
			endScript
		# with overlay in place auto-discover display type		
		else
			if [ -e /sys/class/backlight/rpi_backlight ]; then
				logMessage "auto-discovered display type RPI"
				displayToControl='RPI'
			else
				logMessage "RPI display not detected - HDMI requires manual configuration !!!"
				displayToControl=""
			fi
			echo $displayToControl > "$setupOptionsDir/displayToControl"
		fi
    fi

	if [ -z "$displayToControl" ]; then
		logMessage "++ ERROR: no display specified - can't continue - run install from command line"
		exit $EXIT_OPTIONS_NOT_SET
	fi

    case $displayToControl in
        RPI)
            blankDevice="/sys/class/backlight/rpi_backlight/bl_power"
			dimDevice="/sys/class/backlight/rpi_backlight"
            ;;
        HDMI)
            blankDevice="/sys/class/graphics/fb0/blank"
            dimDevice=""
            ;;
		default)
            blankDevice=""
            dimDevice=""
            ;;
    esac

    logMessage "++ Installing display config to /u-boot/config.txt for display: $displayToControl"

    #   remove lines with framebuffer - not needed even with RPI display
	sed -e '/framebuffer/d;' -e '/rotate/d' "$configFile" | awk '!NF || !seen[$0]++' > "$configFile.tmp"

    # remove any previouos lines added by this script
    sed -i -e "/#### RpiDisplaySetup begin/,/#### RpiDisplaySetup end/d" "$configFile.tmp"

	# add lines for this display
    echo "#### RpiDisplaySetup begin" >> "$configFile.tmp"
	echo "disable_overscan=1" >> "$configFile.tmp"
	if [ "$displayToControl" == "RPI" ]; then
		if [ -f "$setupOptionsDir/rotateDisplay" ]; then
			echo "lcd_rotate=2" >> "$configFile.tmp"
		elif [ -f "$setupOptionsDir/altRotateDisplay" ]; then
			echo "display_lcd_rotate=2" >> "$configFile.tmp"
		fi
	# HDMI display
	else
		read group mode horiz vert freq aspect <<< $(cat "$setupOptionsDir/displayRates")

		echo "# allow HDMI display to enter low powr mode" >> "$configFile.tmp"
		echo "hdmi_blanking=1" >> "$configFile.tmp"
		echo "hdmi_force_hotplug=1" >> "$configFile.tmp"
		echo "hdmi_group=$group" >> "$configFile.tmp"
		######## echo "ignore_lcd=1" >> "$configFile.tmp"
		if (( $group != 0 )); then
			echo "hdmi_mode=$mode" >> "$configFile.tmp"
			if (( $mode == 87 )); then
				echo "hdmi_cvt=$horiz $vert $freq $aspect 0 0 0" >> "$configFile.tmp"
			fi
		fi
		if [ -f "$setupOptionsDir/rotateDisplay" ]; then
			echo "display_hdmi_rotate=2" >> "$configFile.tmp"
		fi
	fi
    echo "#### RpiDisplaySetup end" >> "$configFile.tmp"    
    
    # detect changes
    cmp -s $configFile "$configFile.tmp" > /dev/null
    if (( $? == 1 )); then
		logMessage "updating config.txt"
        mv -f "$configFile.tmp" "$configFile"
		rebootNeeded=true
	else
		rm -f $configFile.tmp
    fi

	# use custom blank device file if provided
	if [ -f "$setupOptionsDir/blank_display_device" ]; then
		blankDevice=$(cat "$setupOptionsDir/blank_display_device")
		logMessage "using custom blank_display_device file: $blankDevice"
    fi
    # enable screen saver (blanking) from the Venus
	if [ "$blankDevice" != "" ]; then
		if [ ! -e "$blankDeviceFile" ] || [ $(grep -c $blankDevice "$blankDeviceFile") == 0 ]; then
			logMessage "setting display blanking to $blankDevice"
			echo $blankDevice > "$blankDeviceFile"
			restartGui=true
		fi
    elif [ -e "$blankDeviceFile" ]; then
        logMessage "disabling display blanking"
        rm -f "$blankDeviceFile"
        restartGui=true
    fi

    # use custom dimming file if provided
	if [ -f "$setupOptionsDir/backlight_device" ]; then
		dimDevice=$(cat "$setupOptionsDir/backlight_device")
		logMessage "using custom backlight_device file: $dimDevice"
	fi
	#enable display dimming if device is specified
	if [ "$dimDevice" != "" ]; then
        if [ ! -e $dimDeviceFile ] || [ $(grep -c $dimDevice "$dimDeviceFile") == 0 ]; then
            logMessage "setting up brightness control: $dimDevice"
            echo $dimDevice > $dimDeviceFile
			restartGui=true
		fi

		# setup other paremeters for dimming
		logMessage "disabling AutoBrightness"
		dbus -y com.victronenergy.settings /Settings/Gui/AutoBrightness SetValue 0 > /dev/null
        if [ ! -e $dimDeviceFile ] || [ $(grep -c $dimDevice "$dimDeviceFile") == 0 ]; then
            logMessage "setting up brightness control to $dimDevice"
            echo $dimDevice > "$dimDeviceFile"
            restartGui=true
        fi
    # disable display brightness
    elif [ -e "$dimDeviceFile" ]; then
        logMessage "disabling display brightness"
        rm -f "$dimDeviceFile"
        restartGui=true
    fi

	if $setupForTsCalibrate ; then
		echo
		echo "setting up for touch screen calibration"
		echo "system will reboot automatically now and again after calibration"
		touch "$setupOptionsDir/calibrate"
		# prevent PackageManager from running setup - until calibrate has finished
		if [ -f "$setupOptionsDir/DO_NOT_AUTO_INSTALL" ]; then
			touch "$setupOptionsDir/RESTORE_DO_NOT_AUTO_INSTALL"
		fi
		touch "$setupOptionsDir/DO_NOT_AUTO_INSTALL"
        touch $headlessFlag
		rebootNeeded=true
		runAgain=true
		# allow endScript to finish without additional prompts
		userInteraction=false
		# endScript won't reboot in this situation so do it here
		nohup sh -c 'sleep 1; reboot' > /dev/null &
		 
	# run ts_calibrate and restore local GUI
	elif [ -f "$setupOptionsDir/calibrate" ]; then
		logMessage "running touchscreen calibration"
		# ts_calibrate modifies /etc/pointercal internally
		#	make backup original first for uninstall
		backupActiveFile "/etc/pointercal"

		ts_calibrate

		# save calibration for reinstall after Venus OS update
		cp "/etc/pointercal" "$setupOptionsDir"

		rebootNeeded=true
		runAgain=true
		rm -f "$setupOptionsDir/calibrate"

		# restore do not auto install flag
		if [ -f "$setupOptionsDir/RESTORE_DO_NOT_AUTO_INSTALL" ]; then
			touch "$setupOptionsDir/DO_NOT_AUTO_INSTALL"
		else
			rm -f "$setupOptionsDir/DO_NOT_AUTO_INSTALL"
		fi
		rm "$setupOptionsDir/RESTORE_DO_NOT_AUTO_INSTALL"
	# restore pointercal from last calibrate pass if any
	else
		if [ -f "$setupOptionsDir/pointercal" ]; then
			updateActiveFile "$setupOptionsDir/pointercal" "/etc/pointercal"
		fi
	fi
    # remove the headless flag to allow a local display
    if [ -e $headlessFlag ] && [ ! -f "$setupOptionsDir/calibrate" ]; then
        logMessage "activating local GUI"
        rm -f $headlessFlag
		rebootNeeded=true
    fi
fi
#### uninstalling - check scriptAction again
# if an install step failed package needs to be removed
if [ $scriptAction == 'UNINSTALL' ] ; then
    logMessage "++ Uninstalling RPI display enhancements"
    if [ -e "$blankDeviceFile" ]; then
        logMessage "disabling display blanking"
        rm -f "$blankDeviceFile"
        restartGui=true
    fi
    if [ -e $dimDeviceFile ]; then
        logMessage "disabling display backlight control"
        restoreActiveFile "$guiStartFile"
        rm -f "$dimDeviceFile"
        restartGui=true
    fi
	# normally, we'd reboot after removing the overlay file
	#	but the overlay shouldn't have any affect on system behavior
	#	so waiting until the next reboot cycle should be fine
    if [ -f "$overlayFile" ]; then
		logMessage "removing $overlayFile"
		rm -f "$overlayFile"
	fi
	if [ -f "/u-boot/config.txt.orig" ] ; then
		sed -i -e "/#### rotate display 180 degrees/,/#### end rotate/d" "$configFile"
		logMessage "/u-boot/config.txt NOT fully restored - manual restore may be needed"
		logMessage "/u-boot.config.txt.orig may help with restore"
		rebootNeeded=true
	fi

	if [ ! -f "/etc/venus/headless" ]; then
		echo
		logMessage "DISABLING LOCAL GUI - reverts to console"
		echo
		touch "/etc/venus/headless"
		rebootNeeded=true
	fi
	logMessage "restoring previous touch screen calibration"
	restoreActiveFile "/etc/pointercal"
fi

#thats all folks - SCRIPT EXITS INSIDE THE FUNCTION
endScript