llvm-switch-env-official.sh

#!/bin/sh
if [ ! "$WORKING_PATH" = "" ] ; then
	ORIGINAL_SRC_ROOT=$WORKING_PATH/llvm
	ORIGINAL_BUILD_ROOT=$WORKING_PATH/build-llvm
	ORIGINAL_OBJ_ROOT=$WORKING_PATH/install-llvm

	SRC_ROOT=$WORKING_PATH/llvm-ykt
	BUILD_ROOT=$WORKING_PATH/build-llvm-ykt
	OBJ_ROOT=$WORKING_PATH/install-llvm-ykt

	export SRC_ROOT
	export OBJ_ROOT


	original_build_root="$ORIGINAL_BUILD_ROOT"
	original_obj_root="$ORIGINAL_OBJ_ROOT"
	build_root="$BUILD_ROOT"
	obj_root="$OBJ_ROOT"

	# http://stackoverflow.com/questions/918886/how-do-i-split-a-string-on-a-delimiter-in-bash

	# deal with CPATH
	ADDR=""
	IFS=':' read -ra ADDR <<< "$CPATH"
	#while IFS=':' read -ra ADDR; do
	for i in "${ADDR[@]}"; do
		if [[ ! ( "${i/$original_obj_root}" = "$i" ) ]]; then
			CPATH=${CPATH/$i}; 
		fi;
		if [[ ! ( "${i/$obj_root}" = "$i" ) ]]; then
			CPATH=${CPATH/$i}; 
		fi;
		if [[ ! ( "${i/$original_build_root}" = "$i" ) ]]; then
			CPATH=${CPATH/$i}; 
		fi;
		if [[ ! ( "${i/$build_root}" = "$i" ) ]]; then
			CPATH=${CPATH/$i}; 
		fi;

	done;
	#done <<< "$CPATH";
	ADDR=""
	tmp=""
	IFS=':' read -ra ADDR <<< "$CPATH"
	#while IFS=':' read -ra ADDR; do
	for i in "${ADDR[@]}"; do
		if [[ ! ( "$i" = "" ) ]] ; then
			if [ "$tmp" = "" ] ; then
				tmp=$i
			else
				tmp=$tmp:$i
			fi;
		fi;
	done
	#done <<< "$CPATH";
	if [ "$tmp" = "" ] ; then
		CPATH=$ORIGINAL_OBJ_ROOT/include:$ORIGINAL_OBJ_ROOT/lib:$ORIGINAL_OBJ_ROOT/bin
#		CPATH=$OBJ_ROOT/include:$tmp
	else
		CPATH=$ORIGINAL_OBJ_ROOT/include:$ORIGINAL_OBJ_ROOT/lib:$ORIGINAL_OBJ_ROOT/bin:$tmp
#		CPATH=$OBJ_ROOT/include:$tmp
	fi;

	export CPATH

	# deal with PATH
	ADDR=""
	IFS=':' read -ra ADDR <<< "$PATH"
	for i in "${ADDR[@]}"; do
		if [[ ! ( "${i/$original_obj_root}" = "$i" ) ]]; then
			PATH=${PATH/$i}; 
		fi;
		if [[ ! ( "${i/$obj_root}" = "$i" ) ]]; then
			PATH=${PATH/$i}; 
		fi;
		if [[ ! ( "${i/$original_build_root}" = "$i" ) ]]; then
			PATH=${PATH/$i}; 
		fi;
		if [[ ! ( "${i/$build_root}" = "$i" ) ]]; then
			PATH=${PATH/$i}; 
		fi;

	done;
	ADDR=""
	tmp=""
	IFS=':' read -ra ADDR <<< "$PATH"
	for i in "${ADDR[@]}"; do
		if [[ ! ( "$i" = "" ) ]] ; then
			if [ "$tmp" = "" ] ; then
				tmp=$i
			else
				tmp=$tmp:$i
			fi;
		fi;
	done
	if [ "$tmp" = "" ] ; then
		PATH=$ORIGINAL_OBJ_ROOT/include:$ORIGINAL_OBJ_ROOT/lib:$ORIGINAL_OBJ_ROOT/bin
	else
		PATH=$ORIGINAL_OBJ_ROOT/include:$ORIGINAL_OBJ_ROOT/lib:$ORIGINAL_OBJ_ROOT/bin:$tmp
	fi;

	export PATH


	cuda_device="$1"	# "/usr/lib/nvidia-cuda-toolkit/libdevice" 
	
	if ! [ "$cuda_device" = "" ]; then

		# deal with LD_LIBRARY_PATH
		ADDR=""
		IFS=':' read -ra ADDR <<< "$LD_LIBRARY_PATH"
		for i in "${ADDR[@]}"; do
			if [[ ! ( "${i/$original_obj_root}" = "$i" ) ]]; then
				LD_LIBRARY_PATH=${LD_LIBRARY_PATH/$i}; 
			fi;
			if [[ ! ( "${i/$obj_root}" = "$i" ) ]]; then
				LD_LIBRARY_PATH=${LD_LIBRARY_PATH/$i}; 
			fi;
			if [[ ! ( "${i/$original_build_root}" = "$i" ) ]]; then
				LD_LIBRARY_PATH=${LD_LIBRARY_PATH/$i}; 
			fi;
			if [[ ! ( "${i/$build_root}" = "$i" ) ]]; then
				LD_LIBRARY_PATH=${LD_LIBRARY_PATH/$i}; 
			fi;

			if [[ ! ( "${i/$cuda_device}" = "$i" ) ]]; then
				LD_LIBRARY_PATH=${LD_LIBRARY_PATH/$i}; 
			fi;
		done;
		ADDR=""
		tmp=""
		IFS=':' read -ra ADDR <<< "$LD_LIBRARY_PATH"
		for i in "${ADDR[@]}"; do
			if [[ ! ( "$i" = "" ) ]] ; then
				if [ "$tmp" = "" ] ; then
					tmp=$i
				else
					tmp=$tmp:$i
				fi;
			fi;
		done
		if [ "$tmp" = "" ] ; then
			LD_LIBRARY_PATH=$ORIGINAL_OBJ_ROOT/lib:$cuda_device
		else
			LD_LIBRARY_PATH=$ORIGINAL_OBJ_ROOT/lib:$cuda_device:$tmp
		fi;
		
		export LD_LIBRARY_PATH

		# deal with LIBRARY_PATH
		ADDR=""
		IFS=':' read -ra ADDR <<< "$LIBRARY_PATH"
		for i in "${ADDR[@]}"; do
			if ! [ "${i/$original_obj_root}" = "$i" ]; then
				LIBRARY_PATH=${LIBRARY_PATH/$i}; 
			fi;
			if ! [ "${i/$obj_root}" = "$i" ]; then
				LIBRARY_PATH=${LIBRARY_PATH/$i}; 
			fi;
			if ! [ "${i/$original_build_root}" = "$i" ]; then
				LIBRARY_PATH=${LIBRARY_PATH/$i}; 
			fi;
			if ! [ "${i/$build_root}" = "$i" ]; then
				LIBRARY_PATH=${LIBRARY_PATH/$i}; 
			fi;

			if ! [ "${i##$cuda_device}" = "$i" ]; then
				LIBRARY_PATH=${LIBRARY_PATH/$i}; 
			fi;
		done;
		ADDR=""
		tmp=""
		IFS=':' read -ra ADDR <<< "$LIBRARY_PATH"
		for i in "${ADDR[@]}"; do
			if ! [ "$i" = "" ] ; then
				if [ "$tmp" = "" ] ; then
					tmp=$i
				else
					tmp=$tmp:$i
				fi;
			fi;
		done
		if [ "$tmp" = "" ] ; then
			LIBRARY_PATH=$ORIGINAL_OBJ_ROOT/lib:$cuda_device
		else
			LIBRARY_PATH=$ORIGINAL_OBJ_ROOT/lib:$cuda_device:$tmp
		fi;
		
		export LIBRARY_PATH
	else
		echo "Cuda libdevice path has not provided."
	fi;
else
	echo "Export WORKING_PATH first."
fi;