#!/bin/bash
# 包含GMT自带脚本文件 其中包含了一些有用的功能 比如获取网格文件的范围
. gmt_shell_functions.sh
# 包含dispOption脚本
. dispOptions.sh
. generic_func.sh
# 定义一个函数 执行输入的语句或者将其显示在屏幕上
RunOrEcho()
{
	first_str=`echo ${2//	/''}`
	if [[ x${3} != x ]]; then
		sec_str=`echo ${3//	/''}`
	fi

	if [[ ${1} == 1 ]]; then
		${first_str}
		if [[ x${3} != x ]]; then
			${sec_str}
		fi
	else
		if [[ x${3} != x ]]; then
			printf "%s\n%s\n" "${first_str}" "${sec_str}"
		else
			printf "%s\n" "${first_str}"
		fi
	fi
}
# GMT显示平面数据脚本，输入文件为网格文件，没有包含网格化语句因为网格化过程中的情况多样化，建议在其他脚本中个别添加再调用此脚本
# 准备几个常用的排版参数组 记录的参数包括 图片的宽度 默认坐标标示设定 色标位置 色标的大小 标示字体大小 
com_layouts=("1.5i,WesNZ,0.1i/-0.2i,1.3i/0.05i+h,10.5p" \
	"1.5i,WeSnZ,1.65i/0.05i,1.3i/0.05i,10.5p" \
	"1.2i,WesNZ,0.1i/-0.1i,1.0i/0.03i+h,7.5p" \
	"1.2i,WeSnZ,1.3i/0.05i,1.0i/0.05i,7.5p")
# 初始化参数
data='null'
outpsfile='null'
gridData='null'
unit='m'
color='rainbow'
range='null'
overwriteRange='null'
overwrite=0
labels=("x (m)" "y (m)")
plotgrad=0
axistick=("a" "a") #横纵坐标轴间隔 a表示自动
bartick="a" #色标轴标轴间隔 a表示自动
frameset="null"
plot_colorbar=1
annotation="null"
keep_open=0
from_open=">"
image_origin=("0.5" "0.5")
bar_origin=("0" "0")
polyfile='null'
lay_num=0
run_command=1
close_poly='-L'
# 从命令行获取参数
while getopts "hi:o:L:r:u:c:l:G:t:v:f:a:p:X:x:gnKOSP" arg
do
	case $arg in
		h)
			dispTitle "${0##*/}" "Image mapping of grid file(s) using the GMT software under the x-y plane. \
				This script accepts one or several grid (.nc or .grd) files as inputs and outputs \
				image files (.png and .eps files). The script could also output data specified shell scripts for \
				further modification. For detailed explanations of relevant arguments, please look for the GMT's manuscripts. \
				This software comes with ABSOLUTE NO WARRANTY. Use it at your own caution."
			dispAuthorInfo "Dr. Yi Zhang (yizhang-geo@zju.edu.cn). Contact me if you have questions or suggestions."
			dispUsage "${0##*/} -i<grid-data> [-o<img-file>] [-L<number>] [-r<xmin>/<xmax>/<ymin>/<ymax>] [-u<unit>] \
				[-c<cpt-file>] [-t<number>,<number>] [-v<number>] [-l<x-label>;<y-label>] [-g] [-G<grad-file>] \
				[-f<WesNZ>] [-p<poly-file>] [-n] [-a<x-coor>,<y-coor>,<annotation>] [-K] [-O] [-S] \
				[-X<x-offset>,<y-offset>] [-x<x-offset>,<y-offset>]"
			dispOptionShort "-i" "Input grid file. This is the only mandatory option that the script has. Just enter a filename \
				and you get an image."
			dispOptionShort "-o" "Output image files. Name extension is not needed. the input filename will be \
				used if this option is not set. NOTE: set this option for plotting multiple images."
			dispOptionShort "-L" "Select layout template that will be used for plotting. The script has four inbuilt templates \
				you can choose from (0 (default) - 3). Please try out to see their difference. You can also add more templates by \
				editing the script. Just make sure that you know what you are doing."
			dispOptionShort "-r" "Range of the input data. The script will detect the full range of the data automatically and \
				the use of this option will overwrite the argument."
			dispOptionShort "-u" "Data unit. The default is meter (m)."
			dispOptionShort "-c" "Filename of the color scale. The default is rainbow.cpt."
			dispOptionShort "-t" "Intervals of axis's ticks. the script will set the intervals automatically \
				if this option is not set."
			dispOptionShort "-v" "Intervals of color bar's labels. the script will set the intervals \
				automatically if this option is not set."
			dispOptionShort "-l" "Axis' labels separated by semicolons. The defaults are x (m) and y (m)."
			dispOptionShort "-g" "Plot a over layer of directional illumination. The default is false."
			dispOptionShort "-G" "Use a input grid data for applying directional illumination. \
				This should be used with the '-g' option at the same time."
			dispOptionShort "-f" "Set frames for plotting axises. The default is WesNZ for layout template 0."
			dispOptionShort "-n" "Don't plot color bar."
			dispOptionShort "-a" "Add an annotation on the map. you need to enter a string contains both \
				coordinates and text of the annotation, such as '10,10,(a)'. For more than one annotations, \
				please output a data specified script using the -S option, then edit the script accordingly."
			dispOptionShort "-p" "Take inputs from a table file and plot polygons. Each line of the file \
				represents a 2D point's coordinates."
			dispOptionShort "-P" "Keep the polygon plotted by the -p option open."
			dispOptionShort "-K" "Keep the .ps file open. This option must be set for plotting multiple images to a single output file."
			dispOptionShort "-O" "Continue from a previous .ps file. This option must be set for plotting multiple images to a single output file."
			dispOptionShort "-X" "Move the starting point for mapping the image. Use this option for plotting sub-figures."
			dispOptionShort "-x" "Move the starting point for mapping the color bar. Use this option for plotting sub-figures."
			dispOptionShort "-S" "Instead of running the script for actually plotting, print an executable shell script on screen. \
				This option is designed to help building data specified shell scripts for further modification."
			exit 0;;
		i)
			data=$OPTARG;;
		o)
			outpsfile=$OPTARG;;
		L)
			lay_num=$OPTARG;;
		u)
			unit=$OPTARG;;
		c)
			color=$OPTARG;;
		t)
			axistick=(${OPTARG//,/ });;
		X)
			image_origin=(${OPTARG//,/ });;
		x)
			bar_origin=(${OPTARG//,/ });;
		v)
			bartick=$OPTARG;;
		f)
			frameset=$OPTARG;;
		r)
			overwrite=1
			overwriteRange=$OPTARG;;
		l)
			OLD_IFS="${IFS}"
			IFS=";"
			labels=(${OPTARG})
			IFS="${OLD_IFS}";;
		g)
			plotgrad=1;;
		n)
			plot_colorbar=0;;
		G)
			gridData=$OPTARG;;
		a)
			annotation=$OPTARG;;
		p)
			polyfile=$OPTARG;;
		P)
			close_poly='';;
		K)
			keep_open=1;;
		O)
			from_open="-O >>";;
		S)
			run_command=0;;
		?)
			printf "error: unknow argument\nuse -h option to see help information\n"
			exit 1;;
	esac
done
# 进行必要的参数检查
if [[ $data == "null" ]]; then
	printf "error: no input file name\nuse -h option to see help information\n"
	exit 1
else
	# 初始化临时文件名
	cptfile=user.cpt
	if [[ ${outpsfile} == 'null' ]]; then
		psfile=${data%.*}.ps
		jpgfile=${data%.*}.png
	else
		psfile=${outpsfile}.ps
		jpgfile=${outpsfile}.png
	fi

	# 获取排版
	layouts=(${com_layouts[${lay_num}]//,/ })
	# 获取坐标轴设置
	if [[ ${frameset} == 'null' ]]; then
		frameset=${layouts[1]}
	fi

	# 获取网格范围
	if [[ $overwrite == 1 ]]; then
		range=${overwriteRange}
	else
		range=$(gmt_get_gridregion ${data})
	fi
	#根据横纵坐标的范围计算图片的长和高 我们默认图片宽度为1.5i
	range_coor=(${range//// })
	pic_height=`echo "scale=4; ${layouts[0]%i} * (${range_coor[3]} - ${range_coor[2]})/(${range_coor[1]} - ${range_coor[0]})"|bc`

	# 输出一个bash脚本的头部到屏幕
	if [[ ${run_command} == 0 && ${from_open} == '>' ]]; then
		echo "#!/bin/bash"
	fi

	# 设置绘图参数
	if [[ ${from_open} == '>' ]]; then
		RunOrEcho ${run_command} "gmt gmtset \
			FONT_ANNOT_PRIMARY=${layouts[4]},Times-Roman,black \
			MAP_FRAME_PEN=thinnest,black \
			MAP_GRID_PEN_PRIMARY=thinnest,black \
			MAP_TICK_PEN_PRIMARY=thinnest,black \
			MAP_TICK_LENGTH_PRIMARY=1p/0.5p \
			MAP_TITLE_OFFSET=7.5p \
			MAP_GRID_CROSS_SIZE_PRIMARY=2p \
			FONT_LABEL=${layouts[4]},Times-Roman,black \
			MAP_LABEL_OFFSET=2.5p \
			MAP_ANNOT_OFFSET_PRIMARY=2.5p"
	fi

	RunOrEcho ${run_command} "gmt gmtset MAP_FRAME_AXES=${frameset}"

	# 处理-B选项的参数
	if [[ ${axistick[0]} == 'a' ]]; then
		axistick[0]=${axistick[0]}g
		axistick[1]=${axistick[1]}g
	else
		axistick[0]=${axistick[0]}g${axistick[0]}
		axistick[1]=${axistick[1]}g${axistick[1]}
	fi

	RunOrEcho ${run_command} "gmt grd2cpt ${data} -C${color} -R${range} -Z -D > ${cptfile}"

	if [[ $plotgrad == 1 ]]; then
		gradfile=${data%.*}Grad.nc
		if [[ $gridData == 'null' ]]; then
			RunOrEcho ${run_command} "gmt grdgradient ${data} -G${gradfile} -Nt -A0/45"
		else
			RunOrEcho ${run_command} "gmt grdgradient ${gridData} -G${gradfile} -Nt -A0/45"
		fi

		if [[ ${run_command} == 0 ]]; then
			echo "gmt grdimage ${data} -R${range} -C${cptfile} -I${gradfile} \
-Bx${axistick[0]}+l\"${labels[0]}\" -By${axistick[1]}+l\"${labels[1]}\" \
-JX${layouts[0]}/${pic_height}i -X${image_origin[0]}i -Y${image_origin[1]}i -K -P ${from_open} $psfile"
		else
			gmt grdimage ${data} -R${range} -C${cptfile} -I${gradfile} \
			-Bx${axistick[0]}+l"${labels[0]}" -By${axistick[1]}+l"${labels[1]}" \
			-JX${layouts[0]}/${pic_height}i -X${image_origin[0]}i -Y${image_origin[1]}i -K -P ${from_open} $psfile
		fi
	else
		if [[ ${run_command} == 0 ]]; then
			echo "gmt grdimage ${data} -R${range} -C${cptfile} -Bx${axistick[0]}+l\"${labels[0]}\" \
-By${axistick[1]}+l\"${labels[1]}\" -JX${layouts[0]}/${pic_height}i \
-X${image_origin[0]}i -Y${image_origin[1]}i -K -P ${from_open} $psfile"
		else
			gmt grdimage ${data} -R${range} -C${cptfile} -Bx${axistick[0]}+l"${labels[0]}" \
			-By${axistick[1]}+l"${labels[1]}" -JX${layouts[0]}/${pic_height}i \
			-X${image_origin[0]}i -Y${image_origin[1]}i -K -P ${from_open} $psfile
		fi
	fi

	# 画多边形
	if [[ $polyfile != 'null' ]]; then
		RunOrEcho ${run_command} "gmt psxy ${polyfile} -JX${layouts[0]}/${pic_height}i -W0.25p,black,- -R${range} ${close_poly} -K -O >> $psfile"
	fi

	#添加号码
	if [[ ${annotation} != 'null' ]]; then
		if [[ ${run_command} == 0 ]]; then
			echo "gmt pstext -R${range} -JX${layouts[0]}/${pic_height}i -K -O <<- EOF >> $psfile"
			echo ${annotation}
			echo "EOF"
		else
			gmt pstext -R${range} -JX${layouts[0]}/${pic_height}i -K -O <<- EOF >> $psfile
			${annotation}
			EOF
		fi
	fi

	#-C${cptfile}+Uk 使用km（色标单位除1000
	#如果unit等于km则在cptfile后面添加+Uk
	if [[ ${plot_colorbar} == 1 ]]; then
		if [[ ${unit} == 'km+Uk' ]]; then
			RunOrEcho ${run_command} "gmt psscale -Dx${layouts[2]}+w${layouts[3]} -C${cptfile}+Uk -Bx${bartick} -By+l${unit} \
				-X${bar_origin[0]}i -Y${bar_origin[1]}i -O -K >> $psfile"
		else
			RunOrEcho ${run_command} "gmt psscale -Dx${layouts[2]}+w${layouts[3]} -C${cptfile} -Bx${bartick} -By+l${unit} \
				-X${bar_origin[0]}i -Y${bar_origin[1]}i -O -K >> $psfile"
		fi
	fi

	if [[ ${keep_open} == 0 ]]; then
		# 这一句什么都不做 唯一的作用就是关闭ps文件
		if [[ ${run_command} == 0 ]]; then
			echo "gmt pstext -R${range} -JX -O <<- EOF >> $psfile"
			echo "EOF"
		else
			gmt pstext -R${range} -JX -O <<- EOF >> $psfile
			EOF
		fi

		# 输出 eps 和 png 文件
		RunOrEcho ${run_command} "gmt psconvert $psfile -A -TEG -E300"

		# 删除临时文件 使用linux终端rm命令
		RunOrEcho ${run_command} "rm $cptfile $psfile gmt.history gmt.conf"
		if [[ $plotgrad == 1 ]]; then
			RunOrEcho ${run_command} "rm $gradfile"
		fi

		# 在终端显示图像 此命令需要imgcat.sh脚本和iTerm终端
		#imgcat $jpgfile
		# 打开图片文件 此命令使用MacOS终端open命令 和Linux下的xdg-open命令
		if [[ ${run_command} == 1 ]]; then
			view_file ${jpgfile}
		fi
	fi
fi