#!/bin/bash
#
# Build script for UHD+GnuRadio on Fedora and Ubuntu
#
#
#
function help {
	cat <<!EOF!
	
Usage: build-gnuradio [--help|-h] [-v|--verbose] [-jN] [-ja] 
                      [-l|--logfile logfile ] [-u|--users ulist] funcs

-v|--verbose   - turn on verbose logging to stdout

-jN            - have make use N concurrent jobs

-ja            - have make use N concurrent jobs with auto setting of N
                 (based on number of cpu cores on build system)
                 
-josh          - use Josh Blums enhanced Gnu Radio GIT repo

-u|--users ul  - add comma-separated users to 'usrp' group in addition
                 to calling user ( $USER )
                 
                
-l|--logfile lf - log messages to 'lf'
-a|--autotools  - do a traditional auto-tools build

-ut	<tag>       - set tag for UHD checkout to <tag>
-gt <tag>       - set tag for Gnu Radio checkout to <tag>

available funcs are:

all             - do all functions
prereqs         - install prerequisites
gitfetch        - use GIT to fetch Gnu Radio and UHD
uhd_build       - build only UHD
firmware        - fetch firmware/FPGA
gnuradio_build  - build only Gnu Radio
mod_groups      - modify the /etc/groups and add user to group 'usrp'
mod_udev        - add UDEV rule for USRP1
mod_sysctl      - modify SYSCTL for larger net buffers
!EOF!

}

if [ $USER = root -o $UID -eq 0 ]
then
	echo Please run this script as an ordinary user
	echo   it will acquire root privileges as it needs them via \"sudo\".
	exit
fi

VERBOSE=No
JFLAG=""
LOGDEV=/dev/null
USERSLIST=None
JOSHMODE=False
UTAG=None
GTAG=None
export LC_LANG=C

while : 
do
	case $1 in
		-ja)
			cnt=`grep 'processor.*:' /proc/cpuinfo|wc -l`
			cnt=`expr $cnt - 1`
			if [ $cnt -lt 1 ]
			then
				cnt=1
			fi
			JFLAG=-j$cnt
			shift
			;;
			
		-j[123456789])
			JFLAG=$1
			shift
			;;

		-josh)
			JOSHMODE=True
			shift
			;;
			
		-v|--verbose)
			LOGDEV=/dev/stdout
			shift
			;;
			
		-l|--logfile)
			case $2 in
				/*)
					LOGDEV=$2
				;;
				*)
					LOGDEV=`pwd`/$2
				;;
			esac
			shift
			shift
			rm -f $LOGDEV
			echo $LOGDEV Starts at: `date` >>$LOGDEV 2>&1
			;;
			
		-u|--users)
			USERSLIST=$2
			shift
			shift
			;;
			
		-a|--autotools)
			CMAKE_GNUBUILD=False
			shift
			;;
			
		-h|--help)
			help
			exit
			;;
			
		-ut)
			UTAG=$2
			shift
			shift
			;;
			
		-gt)
			GTAG=$2
			shift
			shift
			;;
			
		-*)
			echo Unrecognized option: $1
			echo
			help
			exit
			break
			;;
		*)
			break
			;;
	esac
done

CWD=`pwd`
SUDOASKED=n
SYSTYPE=unknown
if [ -f /etc/fedora-release -o -f /etc/lsb-release ]
then
	good_to_go=yes
else
	echo You appear to not be running on a Fedora or Ubuntu system.
	echo This script will only run correctly on a Fedora or Ubuntu system.
	echo Exiting.
	exit
fi

echo This script will install Gnu Radio from current GIT sources
echo You will require Internet access from the computer on which this
echo script runs.  You will also require SUDO access.  You will require
echo approximately 500MB of free disk space to perform the build.
echo " "
echo This script will, as a side-effect, remove any existing Gnu Radio
echo installation that was installed from your Linux distribution packages.
echo It must do this to prevent problems due to interference between
echo a linux-distribution-installed Gnu Radio/UHD and one installed from GIT source.
echo " "
echo The whole process may take up to two hours to complete, depending on the
echo capabilities of your system.
echo -n Proceed?

read ans
case $ans in
	y|yes|Y|YES|Yes)
		PROCEED=y
	;;
	*)
		exit
esac

SPACE=`df $HOME| grep -v blocks|grep '%'`
SPACE=`echo $SPACE | awk '/./ {n=NF-2; printf ("%d\n", $n/1.0e3)}'`


if [ $SPACE -lt 500 ]
then
	echo "You don't appear to have enough free disk space on $HOME"
	echo to complete this build/install
	echo exiting
	exit
fi

total=0
for file in uhd.20* gnuradio.20*
do
	if [ -d $file ]
	then
		sz=`du -s $file|awk '{print $1}'`
		total=`expr $total + $sz`
	fi
done
total=`expr $total '*' 1024`
total=`expr $total / 1000000`
if [ $total -gt 400 ]
then
	echo Your old 'uhd.*' and 'gnuradio.*' directories are using roughly $total MB
	echo of disk space:
	ls -ld uhd.20* gnuradio.20*
	echo " "
	echo -n Remove them'?'
	read ans
	
	case $ans in
		yes|YES)
			echo removing uhd.20* and gnuradio.20*
			rm -rf uhd.20* gnuradio.20*
			echo Done
			;;
	esac
fi

function my_echo {
	if [ $LOGDEV = /dev/stdout ]
	then
		echo $*
	else
		echo $*
		echo $* >>$LOGDEV 2>&1
	fi
}

function checkcmd {
	which $1 >tmp$$ 2>&1
	if grep -q no.${1}.in tmp$$
	then
		my_echo Failed to find command \'$1\' after pre-requisite installation.
		my_echo This very likely indicates that one or more parts of the
		my_echo pre-requisite installation has failed.
		rm -f tmp$$
		my_echo -n Show pre-requisites install log?
		read ans
		case $ans in
			y|Y|Yes|yes|YES)
				more prereq.log
			;;
		esac
		#rm -f prereq.log
		rm -f tmp$$
		my_echo exiting build
		exit
	fi
	rm -f tmp$$
}

function checklib {
	found=0
	my_echo -n Checking for library $1 ...
	for dir in /lib /usr/lib /usr/lib64 /lib64
	do
		ls -l $dir/${1}*.so* >tmp$$ 2>&1
		if grep -q "\\*.so\\*:" tmp$$
		then
			found=0
		else
			found=1
			if [ $2 -gt 0 ]
			then
				NL=`cat tmp$$|wc -l`
				if [ $NL -lt $2 ]
				then
					found=0
				fi
			fi
			break
		fi
		rm -f tmp$$
	done
	if [ $found -le 0 ]
	then
		my_echo "*******"
		my_echo Failed to find libraries with prefix \'$1\' after pre-requisite installation.
		my_echo This very likely indicates that the pre-requisite installation failed
		my_echo to install one or more critical pre-requisites for Gnu Radio/UHD
		my_echo -n Show pre-requisites install log?
		read ans
		case $ans in
			y|Y|yes|Yes|YES)
				more prereq.log
			;;
		esac
		#rm -rf prereq.log
		my_echo exiting build
		exit
	else
		my_echo Found library $1
	fi
}

function checkpkg {
	my_echo Checking for package $1
	if [ `apt-cache search $1 | wc -l` -eq 0 ]
	then
		my_echo Failed to find package \'$1\' in known packages
		my_echo Perhaps you need to add the Ubuntu universe or multiverse PPA?
		my_echo see https://help.ubuntu.com/community/Repositories/Ubuntu
		#rm -rf prereq.log
		my_echo exiting build
		exit
	fi
}
		
function prereqs {
	sudocheck
	my_echo -n Installing pre-prequisites...
	#
	# Check whether we're on a Fedora or Ubuntu system, and "do the right thing"
	#  for either system
	#
	# It's a Fedora system
	#
	if [ -f /etc/fedora-release ]
	then
		SYSTYPE=Fedora
		case `cat /etc/fedora-release` in
		*12*|*13*|*14*|*15*|*16*)
			sudo yum -y erase 'gnuradio*' >>$LOGDEV 2>&1
			sudo yum -y erase 'libgruel-*' >>$LOGDEV 2>&1
			sudo yum -y erase 'libgruel*' >>$LOGDEV 2>&1
			sudo yum -y groupinstall "Engineering and Scientific" "Development Tools" >>$LOGDEV 2>&1
			sudo yum -y install fftw-devel cppunit-devel wxPython-devel libusb-devel \
			  guile boost-devel alsa-lib-devel numpy gsl-devel python-devel pygsl \
			  python-cheetah python-lxml guile-devel PyOpenGL qt-devel qt qt4 qt4-devel \
			  PyQt4-devel qwt-devel qwtplot3d-qt4-devel libusb libusb-devel \
			  libusb1 libusb1-devel cmake git wget sdcc python-docutils \
			  PyQwt PyQwt-devel qwt-devel gtk2-engines >>$LOGDEV 2>&1
		;;
		*)
			my_echo Your Fedora system release must be at least release 12 to proceed
			exit
			;;
		esac
		
	#
	# It's a Ubuntu system
	#
	elif [ -f /etc/lsb-release ]
	then
		SYSTYPE=Ubuntu
		sudo apt-get -y purge 'gnuradio-*' >>$LOGDEV 2>&1
		sudo apt-get -y purge 'libgruel-*' >>$LOGDEV 2>&1
		sudo apt-get -y purge 'libgruel*' >>$LOGDEV 2>&1
		sudo apt-get -y purge 'libgruel0*' >>$LOGDEV 2>&1
		sudo apt-get -y purge 'libgnuradio*' >>$LOGDEV 2>&1
		sudo apt-get -y purge 'python-gnuradio*' >>$LOGDEV 2>&1
		case `grep DISTRIB_RELEASE /etc/lsb-release` in
		*11.*)
			PKGLIST="libfontconfig1-dev libxrender-dev libpulse-dev swig g++
			automake autoconf libtool python-dev libfftw3-dev
			libcppunit-dev libboost-all-dev libusb-dev libusb-1.0-0-dev fort77 sdcc sdcc-libraries
			libsdl1.2-dev python-wxgtk2.8 git-core guile-1.8-dev
			libqt4-dev python-numpy ccache python-opengl libgsl0-dev
			python-cheetah python-lxml doxygen qt4-dev-tools libusb-1.0-0-dev
			libqwt5-qt4-dev libqwtplot3d-qt4-dev pyqt4-dev-tools python-qwt5-qt4
			cmake git-core wget sdcc libxi-dev python-docutils gtk2-engines-pixbuf r-base-dev"
			;;
			
		*10.04*|*10.10*)
			PKGLIST="libfontconfig1-dev libxrender-dev libpulse-dev swig g++
			automake autoconf libtool python-dev libfftw3-dev
			libcppunit-dev libboost-all-dev libusb-dev libusb-1.0-0-dev fort77 sdcc sdcc-libraries
			libsdl1.2-dev python-wxgtk2.8 git-core guile-1.8-dev
			libqt4-dev python-numpy ccache python-opengl libgsl0-dev
			python-cheetah python-lxml doxygen qt4-dev-tools libusb-1.0-0-dev
			libqwt5-qt4-dev libqwtplot3d-qt4-dev pyqt4-dev-tools python-qwt5-qt4
			cmake git-core wget sdcc python-docutils gtk2-engines-pixbuf"
			;;
			
		*9.10*)
			PKGLIST="swig g++ automake autoconf libtool python-dev libfftw3-dev
			libcppunit-dev libboost1.38-dev libusb-dev libusb-1.0-0-dev fort77 sdcc sdcc-libraries
			libsdl1.2-dev python-wxgtk2.8 git-core guile-1.8-dev
			libqt4-dev python-numpy ccache python-opengl libgsl0-dev
			python-cheetah python-lxml doxygen qt4-dev-tools libusb-1.0-0-dev
			libqwt5-qt4-dev libqwtplot3d-qt4-dev pyqt4-dev-tools
			cmake git wget sdcc python-docutils"
			;;
			
		*9.04*)
			sudo apt-get purge python-wxgtk2.6
			PKGLIST="swig g++ automake1.9 autoconf libtool python-dev fftw3-dev 
			libcppunit-dev libboost1.35-dev sdcc-nf libusb-dev libusb-1.0-0-dev 
			libsdl1.2-dev python-wxgtk2.8 git-core guile-1.8-dev 
			libqt4-dev python-numpy ccache python-opengl libgsl0-dev 
			python-cheetah python-lxml doxygen qt4-dev-tools 
			libqwt5-qt4-dev libqwtplot3d-qt4-dev pyqt4-dev-tools 
			cmake git wget python-docutils"
			;;

			
		*)
			my_echo Your Ubuntu release must be at least 9.04 to proceed
			exit
			;;
		esac
		for pkg in $PKGLIST; do checkpkg $pkg; done
		sudo apt-get -y install $PKGLIST >>$LOGDEV 2>&1
	else
		my_echo This script supports only Ubuntu and Fedora systems
		my_echo Your Fedora system must be at least Fedora 12
		my_echo Your Ubuntu system must be at least Ubuntu 9.04
		exit
	fi
	PATH=$PATH:/usr/libexec/sdcc
	export PATH
	
	checkcmd sdcc
	checkcmd guile
	checkcmd git
	checkcmd cmake
	
	if [ $SYSTYPE = Fedora ]
	then
		checklib libusb-0 0
		checklib libusb-1 0
	else
		checklib libusb 2
	fi
	
	checklib libboost_ 5
	checklib libcppunit 0
	checklib libguile 5
	checklib libfftw 5
	checklib libgsl 0
	
	#rm -f prereq.log

	my_echo Done
}


function gitfetch {
	date=`date +%Y%m%d%H%M%S`

	#
	# Check for existing gnuradio directory
	#
	if [ $JOSHMODE = False ]
	then
		if [ -d gnuradio ]
		then
			mv gnuradio gnuradio.$date
		fi
	else
		if [ -d jblum ]
		then
			mv jblum jblum.$date
		fi
	fi
	
	#
	# Check for existing UHD directory
	#
	if [ -d uhd ]
	then
		mv uhd uhd.$date
	fi

	#
	# GIT the gnu radio source tree
	#
	my_echo -n Fetching Gnu Radio via GIT...
	if [ $JOSHMODE = False ]
	then
		git clone --progress http://gnuradio.org/git/gnuradio.git >>$LOGDEV 2>&1
		if [ ! -d gnuradio/gnuradio-core ]
		then
			my_echo GIT checkout of Gnu Radio failed!
			exit
		fi
		if [ $GTAG != None ]
		then
			cd gnuradio
			git checkout $GTAG >/dev/null 2>&1
			git status >tmp$$ 2>&1
			if grep -q "On branch $GTAG" tmp$$
			then
				whee=yes
				rm -f tmp$$
			else
				my_echo Could not fetch Gnu Radio tagged $GTAG from GIT
				rm -f tmp$$
				exit
			fi
			cd ..
		fi
	else
		git clone --progress git://gnuradio.org/jblum.git >>$LOGDEV 2>&1
		if [ $GTAG != "None" ]
		then
			git checkout $GTAG
		else
			git checkout next
		fi
		if [ ! -d jblum/gnuradio-core ]
		then
			my_echo GIT checkout of Gnu Radio from jblum.git failed!
			exit
		fi
	fi
	my_echo Done

	#
	# GIT the UHD source tree
	#
	rm -rf uhd
	my_echo -n Fetching UHD via GIT...
	git clone --progress git://code.ettus.com/ettus/uhd.git >>$LOGDEV 2>&1

	if [ ! -d uhd/host ]
	then
		my_echo GIT checkout of UHD using GIT protocol failed...trying
		my_echo    with http instead
		rm -rf UHD-Mirror
		git clone --progress http://github.com/EttusResearch/UHD-Mirror.git >>$LOGDEV 2>&1
		if [ ! -d UHD-Mirror/host ]
		then
			my_echo GIT check of UHD using http protocol failed--exiting
			exit
		fi
		mv UHD-Mirror uhd
	fi
	if [ $UTAG != None ]
	then
		cd uhd
		git checkout $UTAG >/dev/null 2>&1
		git status >tmp$$ 2>&1
		if grep -q "On branch $UTAG" tmp$$
		then
			whee=yes
			rm -f tmp$$
		else
			my_echo Could not fetch UHD tagged $UTAG from GIT
			rm -f tmp$$
		fi
		cd ..
	fi
	my_echo Done
}

function uhd_build {
	#
	# UHD build
	#
	sudocheck
	if [ ! -d uhd ]
	then
		my_echo you do not appear to have the \'uhd\' directory
		my_echo you should probably use $0 gitfetch to fetch the appropriate
		my_echo files using GIT
		exit
	fi
	if [ $UTAG != None ]
	then
		cd uhd
		git checkout $UTAG >/dev/null 2>&1
		cd ..
	fi
	my_echo -n Building UHD...
	cd uhd/host
	if [ ! -d build ]
	then
		mkdir build
	fi
	cd build
	make clean >/dev/null 2>&1
	cmake ../ >>$LOGDEV 2>&1
	make clean >>$LOGDEV 2>&1
	make $JFLAG >>$LOGDEV 2>&1
	if [ $? -ne 0  ]
	then
		my_echo UHD build apparently failed
		my_echo Exiting UHD build
		exit
	fi
	sudo make $JFLAG install >>$LOGDEV 2>&1
	
	if [ ! -f /usr/local/bin/uhd_find_devices ]
	then
		my_echo UHD build/install apparently failed
		my_echo Exiting UHD build
		exit
	fi
	sudo ldconfig >>$LOGDEV 2>&1
	my_echo Done building/installing UHD
}

function firmware {
	sudocheck
	#
	# Firmware/FPGA images...
	#
	my_echo Fetching and installing FPGA/Firmware images via wget...
	
	rm -rf wget-tmp*
	mkdir wget-tmp$$
	cd wget-tmp$$
	
	#
	# Fetch the tar file
	#
	UHD_IMG_PREFIX=http://files.ettus.com/uhd_releases/master_images
   	fn=NOTEXIST
   	
   	#
   	# Try for any .tar.gz files
   	#
	wget http://files.ettus.com/uhd_releases/master_images/auto/current.tar.gz >/dev/null 2>&1

	xx=`ls *.gz|sort|tail -1`
	if [ "@@" != @${xx}@ ]
	then
		fn=$xx
	fi
	
	if [ $fn = NOTEXIST ]
	then
		my_echo Failed to download any usable images file from: $UHD_IMG_PREFIX
		exit
	fi

    #
	# Check to make sure we got it
	#
	if [ ! -f $fn ]
	then
		my_echo Failed to fetch $fn: with latest UHD firmware/fpga images
		exit
	fi
	
	#
	# Now unpack the tar file
	#
	my_echo ...Installing from: $fn
	tar xf $fn
	
	if [ ! -d UHD*/share/uhd/images ]
	then
		my_echo There was a problem with the TAR file: $fn
		my_echo The appropriate directories were not unpacked from it
		my_echo In particular, UHD.../share/uhd/images does not exist
		exit
	fi
	
	#
	# There's now a directory structure from the tar file
	# Descend into it and copy the images to /usr/local/share/uhd/images
	#
	cd UHD*/share/uhd/images
	if [ ! -d /usr/local/share/uhd/images ]
	then
		sudo mkdir -p /usr/local/share/uhd/images
	fi
	my_echo ...Copying into /usr/local/share/uhd
	sudo cp *.bin /usr/local/share/uhd/images
	sudo cp *.ihx /usr/local/share/uhd/images
	sudo cp *.rbf /usr/local/share/uhd/images
	sudo cp *.tag /usr/local/share/uhd/images
	sudo chmod 644 /usr/local/share/uhd/images/*
	cd $CWD
	rm -rf wget-tmp$$
	my_echo Done
}

function gnuradio_build {
	sudocheck
	
	if [ $JOSHMODE = False ]
	then
		if [ ! -d gnuradio ]
		then
			my_echo you do not appear to have the \'gnuradio\' directory
			my_echo you should probably use $0 gitfetch to fetch the appropriate
			my_echo files using GIT
			exit
		fi
		if [ $GTAG != None ]
		then
			cd gnuradio
			git checkout $GTAG >/dev/null 2>&1
			cd ..
		fi
	else
		if [ ! -d jblum ]
		then
			my_echo you do not appear to have the \'jblum\' directory
			my_echo you should probably use $0 gitfetch to fetch the appropriate
			my_echo files using GIT
			exit
		fi
	fi
		
	#
	# LD stuff
	#
	my_echo /usr/local/lib >tmp$$
	my_echo /usr/local/lib64 >>tmp$$

	if grep -q /usr/local/lib /etc/ld.so.conf.d/*
	then
		my_echo /usr/local/lib already in ld.so.conf.d
	else
		sudo cp tmp$$ /etc/ld.so.conf.d/local.conf
	fi
	rm -f tmp$$
	my_echo Doing ldconfig...
	sudo ldconfig >/dev/null 2>&1

	PKG_CONFIG_PATH=/usr/local/lib/pkgconfig
	
	if [ -d /usr/local/lib64/pkgconfig ]
	then
		PKG_CONFIG_PATH=/usr/local/lib64/pkgconfig
	fi
	
	export PKG_CONFIG_PATH
	
	#
	# Build Gnuradio
	#
	if [ $JOSHMODE = False ]
	then
		cd gnuradio
	else
		cd jblum
		git checkout gr_basic >>$LOGDEV 2>&1
	fi
	if [ $CMAKE_GNUBUILD = True ]
	then
		my_echo Building Gnu Radio...
		my_echo ...Doing cmake
		if [ -d build ]
		then
			my_echo ...build directory already here
		else
			mkdir build
		fi
		cd build
		make clean >/dev/null 2>&1
		my_echo ...Cmaking
		cmake ../ >>$LOGDEV 2>&1
		my_echo ...Building
		make $JFLAG clean >>$LOGDEV 2>&1
		make $JFLAG >>$LOGDEV 2>&1
		if [ $? -ne 0 ]
		then
			my_echo make failed
			my_echo Exiting Gnu Radio build/install
			exit
		fi
	else
		my_echo ..Traditional autotools build
		my_echo ...Boostrapping
		./boostrap >>$LOGDEV 2>&1
		my_echo ...Configuring
		./configure >>$LOGDEV 2>&1
		my_echo ...Building
		make >>$LOGDEV 2>&1
	fi
	my_echo ...Installing
	sudo make $JFLAG install >>$LOGDEV 2>&1
	sudo ldconfig >>$LOGDEV 2>&1
	my_echo Done building and installing Gnu Radio
	my_echo -n GRC freedesktop icons install ...
	if [ -f /usr/local/libexec/gnuradio/grc_setup_freedesktop ]
	then
		sudo chmod 755 /usr/local/libexec/gnuradio/grc_setup_freedesktop
		sudo /usr/local/libexec/gnuradio/grc_setup_freedesktop install >>$LOGDEV 2>&1
	fi
	my_echo Done
}

function mod_groups {
	sudocheck
	#
	# Post install stuff
	#
	# USRP rules for UDEV and USRP group
	#
	#
	# Check for USRP group, and update if necessary
	if grep -q usrp /etc/group
	then
		my_echo Group \'usrp\' already in /etc/group
	else
		sudo /usr/sbin/groupadd usrp
	fi

	#
	# Check that our calling user is in the USRP group, update if necessary
	#
	if grep -q usrp.*${USER} /etc/group
	then
		my_echo User $USER already in group \'usrp\'
	else
		sudo /usr/sbin/usermod -a -G usrp $USER
cat <<"!EOF!"
********************************************************************************
This script has just modified /etc/group to place your userid '('$USER')' into group 'usrp'
In order for this change to take effect, you will need to log-out and log back
in again.  You will not be able to access your USRP1 device until you do this.

If you wish to allow others on your system to use the USRP1 device, you will need to use:

  sudo usermod -a -G usrp userid
  
For each userid you wish to allow access to the usrp

********************************************************************************

Further 
!EOF!
	fi
	if [ "$USERSLIST" = None ]
	then
		foo=bar
	else
		ul=`echo $USERSLIST|sed -e 's/,/ /g'`
		for u in $ul
		do
			sudo /usr/sbin/usermod -a -G usrp $u
			my_echo Added $u to group usrp
		done
	fi
}

function mod_udev {
	sudocheck
	#
	# Check for UDEV rules file, update if necessary
	#
	if [ ! -f /etc/udev/rules.d/10-usrp.rules ]
	then
		if [ -f $CWD/uhd/host/utils/uhd-usrp.rules ]
		then
			cp $CWD/uhd/host/utils/uhd-usrp.rules tmp$$
		else
		cat >>tmp$$ <<"!EOF!"
#
# Copyright 2011 Ettus Research LLC
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

SUBSYSTEMS=="usb", ATTRS{idVendor}=="fffe", ATTRS{idProduct}=="0002", MODE:="0666"
SUBSYSTEMS=="usb", ATTRS{idVendor}=="2500", ATTRS{idProduct}=="0002", MODE:="0666"
!EOF!
		fi
		chmod 644 tmp$$
		sudo mv tmp$$ /etc/udev/rules.d/10-usrp.rules
		sudo killall -HUP udevd
		sudo udevadm control --reload-rules
		rm -f tmp$$
	else
		my_echo USRP rules for UDEV already in place
	fi
}

function mod_sysctl {
	sudocheck
	#
	# Modify sysctl.conf as necessary
	#
	cat >tmp$$ <<!EOF!
# Updates for Gnu Radio
net.core.rmem_max = 1000000
net.core.wmem_max = 1000000
kernel.shmmax = 2147483648
!EOF!


	if grep -q 'Updates for Gnu Radio' /etc/sysctl.conf
	then
		my_echo Required updates to /etc/sysctl.conf already in place
	else
		my_echo Applying updates to /etc/sysctl.conf
		cat /etc/sysctl.conf tmp$$ >tmp2$$
		chmod 644 tmp2$$
		sudo mv tmp2$$ /etc/sysctl.conf
	fi

	sudo sysctl -w net.core.rmem_max=1000000 >/dev/null 2>&1
	sudo sysctl -w net.core.wmem_max=1000000 >/dev/null 2>&1
	sudo sysctl -w kernel.shmmax=2147483648  >/dev/null 2>&1
	 
	rm -f tmp$$
	rm -f tmp2$$
	
	if grep -q usrp /etc/security/limits.conf
	then
		my_echo usrp group already has real-time scheduling privilege
	else
		cat >tmp$$ <<!EOF!
@usrp  - rtprio 50
!EOF!
		cat /etc/security/limits.conf tmp$$ >tmp2$$
		sudo cp tmp2$$ /etc/security/limits.conf
		sudo chmod 644 /etc/security/limits.conf
		rm -f tmp$$ tmp2$$
		my_echo Group \'usrp\' now has real-time scheduling privileges
		my_echo You will need to log-out and back in again for this to
		my_echo take effect
	fi
}

function all {
	my_echo Starting all functions at: `date`
	cd $CWD
	prereqs
	touch -d "15 minutes ago" touch$$
	if [ -d uhd -a -d gnuradio ]
	then
		if [ uhd -ot touch$$ -o gnuradio -ot touch$$ ]
		then
			gitfetch
		else
			my_echo Skipping git fetch, since \'uhd\' and \'gnuradio\' are new enough
		fi
	else
		gitfetch
	fi
	rm -f touch$$
	for fcn in uhd_build firmware gnuradio_build mod_groups mod_udev mod_sysctl pythonpath
	do
		my_echo Starting function $fcn at: `date`
		cd $CWD
		$fcn
		my_echo Done function $fcn at: `date`
	done
	my_echo Done all functions at: `date`
}

function pythonpath {
	for PYVER in 2.6 2.7
	do
		for type in "" 64
		do
			if [ -d /usr/local/lib${type}/python${PYVER}/site-packages/gnuradio ]
			then
				PYTHONPATH=/usr/local/lib${type}/python${PYVER}/site-packages
			fi
			if [ -d /usr/local/lib/${type}/python${PYVER}/dist-packages/gnuradio ]
			then
				PYTHONPATH=/usr/local/lib${type}/python${PYVER}/dist-packages
			fi
		done
	done
	echo
	echo
	echo "************************************************************"
	echo You should probably set your PYTHONPATH to:
	echo " "
	echo "    " $PYTHONPATH
	echo " "
	echo in your .bashrc or equivalent file prior to attempting to run
	echo any Gnu Radio applications or Gnu Radio Companion.
	echo "*************************************************************"
}

function sudocheck {
	#
	# Check SUDO privileges
	#
	if [ $SUDOASKED = n ]
	then
		echo SUDO privileges are required 
		echo -n Do you have SUDO privileges'?'
		read ans
		case $ans in
			y|yes|YES|Y)
				echo Continuing with script
				SUDOASKED=y
				sudo grep timestamp_timeout /etc/sudoers >tmp$$
				timeout=`cat tmp$$|awk '/./ {print $4}'`
				rm -f tmp$$
				if [ "@@" = "@$timeout@" ]
				then
					sudo cp /etc/sudoers tmp$$
					sudo chown $USER tmp$$
					sudo chmod 644 tmp$$
					echo "Defaults  timestamp_timeout = 90" >>tmp$$
					sudo cp tmp$$ /etc/sudoers
					sudo chown root /etc/sudoers
					sudo chmod 440 /etc/sudoers
				elif [ $timeout -lt 90 ]
				then
					echo You need to have a timestamp_timout in /etc/sudoers of 90 or more
					echo Please ensure that your timestamp_timeout is 90 or more
					exit
				fi
				;;
			*)
				echo Exiting.  Please ensure that you have SUDO privileges on this system!
				exit
				;;
		esac
	fi
}


PATH=$PATH:/usr/libexec/sdcc
export PATH

CMAKE_GNUBUILD=True

case $# in
	0)
		all
		my_echo All Done
		exit
esac

for arg in $*
do
	cd $CWD
	$arg
done
my_echo All Done