#!/bin/bash

#IPFLOOT="151.217.15.90"
IPFLOOT="192.168.254.7"
FLOOTPORT="1337"
PPMFILE="images/$2.ppm"
HEXPPM="images/$2.hexppm"
PIXLIST="pixlists/$2.pixlist"
ALPHACOLOR="$3"
FNAME="$2"

test -z "$FLOOTFORKS" && FLOOTFORKS="4"

declare -a PIXMAP
declare -a LOL
declare -a LOLPID

# https://gist.github.com/nberlette/e3e303a81f2c41927bf4fe90fb89d97f
function hex() {
    printf "%02X%02X%02X" ${*//','/' '}
}



gen_pixmap() {
	y=0
	while read -r LINE
	do
		if [ "$y" -gt 2 ]
		then
			for x in $(seq 1 $x_ppm)
			do
				REDindex=$((x*3-2))
				GREENindex=$((x*3+1-2))
				BLUEindex=$((x*3+2-2))
				
				REDvalue="$(echo $LINE| cut -d ' ' -f${REDindex})"
				GREENvalue="$(echo $LINE| cut -d ' ' -f${GREENindex})"
				BLUEvalue="$(echo $LINE| cut -d ' ' -f${BLUEindex})"
				
				PIXELcolor="$(hex ${REDvalue} $GREENvalue $BLUEvalue)"
				PIXMAP[$y]="${PIXMAP[$y]} $PIXELcolor"
				
			done
			echo "please wait"
			echo ${PIXMAP[$y]} >> $HEXPPM
		elif [ "$y" -eq 1 ]
			then
			x_ppm="$(echo "$LINE"|cut -d ' ' -f1)"
		fi
		y=$((y+1)) 
		#~ for col in $LINE
		#~ do
			#~ case $count in
			#~ 0) RED="$(hex $col)"
			#~ ;;
			#~ 1) GREEN="$(hex $col)"
			#~ ;;
			#~ 2) BLUE="$(hex $col)"
			#~ ;;
			#~ esac
			#~ count=$((count+1))
			#~ test $count -ge 2 && count=0
		#~ done
	done < "$PPMFILE"
	
}

gen_field() {

test -z $W && W=640
test -z $H && H=480
test -z $COLOR && COLOR="666999"
echo "drawing $W x $H - $COLOR" >&2
for x in  $(seq 1 $W) 
	do
	for y in $(seq 1 $H)
	do
		echo "PX $x $y $COLOR"
	done
done

}

draw_pixmap() {
	y=1
	test -z $x_ppm && echo "x_ppm missing" && exit 1
	while read -r LINE
	do
		for x in $(seq 1 $x_ppm)
		do
			# when Color is FF00FE draw rainbow background
			if [[ "$(echo $LINE | cut -d ' ' -f$x)" != "FF00FE" ]]
			then
				echo "please wait"
				echo "PX $x $y $(echo $LINE | cut -d ' ' -f$x)" >> $PIXLIST

				# magnify double size
				#echo "PX $((x*2)) $((y*2)) $(echo $LINE | cut -d ' ' -f$x)" >> $PIXLIST
				#echo "PX $((x*2+1)) $((y*2)) $(echo $LINE | cut -d ' ' -f$x)" >> $PIXLIST
				#echo "PX $((x*2)) $((y*2+1)) $(echo $LINE | cut -d ' ' -f$x)" >> $PIXLIST
				#echo "PX $((x*2+1)) $((y*2+1)) $(echo $LINE | cut -d ' ' -f$x)" >> $PIXLIST			
			else
				
				if [ "$y" -lt 32 ]
				then
					RAINBOWCOLOR="$(hex 0 0 255)"
				elif [ "$y" -lt 64 ]
				then
					RAINBOWCOLOR="$(hex 0 255 255)"
				elif [ "$y" -lt 94 ]
				then
					RAINBOWCOLOR="$(hex 0 255 0)"
				elif [ "$y" -lt 126 ]
				then
					RAINBOWCOLOR="$(hex 255 255 0)"
				else
					RAINBOWCOLOR="$(hex 255 0 0)"
				
				fi
				
				 
								
				#~ if [ "$y" -lt 16 ]
				#~ then
					#~ RAINBOWCOLOR="$(hex 0 0 $(((y+1)*16)))"
				#~ elif [ "$y" -lt 32 ]
				#~ then
					#~ RAINBOWCOLOR="$(hex 0 $(((y-16)*16)) 255)"
				#~ elif [ "$y" -lt 48 ]
				#~ then
					#~ RAINBOWCOLOR="$(hex 0 $((y*16)) $((16-(y-32)*16)))"
				#~ elif [ "$y" -lt 64 ]
				#~ then
					#~ RAINBOWCOLOR="$(hex $((y*16)) $((y*16)) 0)"
				#~ else
					#~ RAINBOWCOLOR="$(hex $((y*16)) 0 0)"
				
				#~ fi
			    
			
				echo "please wait for rainbow"
				echo "PX $((x*2)) $((y*2)) $RAINBOWCOLOR" >> $PIXLIST
				echo "PX $((x*2+1)) $((y*2)) $RAINBOWCOLOR" >> $PIXLIST
				echo "PX $((x*2)) $((y*2+1)) $RAINBOWCOLOR" >> $PIXLIST
				echo "PX $((x*2+1)) $((y*2+1)) $RAINBOWCOLOR" >> $PIXLIST			
			fi
		done
		y=$((y+1)) 

	done < "$HEXPPM"
	
}

sx=0
sy=0
shuf_xy() {
  
	case $SHUFMODE in
	chaos) test -z $H && H=640
        test -z $W && W=480
        echo "OFFSET $(shuf -i 0-$W -n 1) $(shuf -i 0-$H -n 1)"
	;;
	
	shake) test -z $H && H=1
        test -z $W && W=1
        echo "OFFSET $(shuf -i $W-$(($W+10)) -n 1) $(shuf -i $H-$(($H+10)) -n 1)"
	;;
	
	static) echo "OFFSET $W $H"
	;;

	cursor) echo "OFFSET $(xdotool getmouselocation | tr ':' ' '|awk '{print $2 " " $4}')"
	;;
	
	*) echo "OFFSET 0 0" 
	;;
	esac
	#
	#
	#echo "OFFSET $(shuf -i 0-1760 -n 1) 919"
	
	#echo > /dev/null
	
	#~ echo "OFFSET $sx $sy"
	#~ sx=$((sx+1))
	#~ sy=$((sy+1))
	
	#~ test $sx -gt 1760 && sx=0
	#~ test $sy -gt 920 && sy=0
}

floot() {
	# small stupid animation, two alternating images
	if [ "$FNAME" == "winketuxS" ]
	then
		LOL[1]="$(cat ${FNAME}1.pixlist | shuf)"
		LOL[2]="$(cat ${FNAME}2.pixlist | shuf )"
		LOL[3]="$(cat ${FNAME}2.pixlist | shuf )"
		#LOL[3]="$(cat $FNAME-mc.pixlist.2 | shuf)"
	elif [ "$FNAME" == "fill" ]
	then
    LOL_org="$(gen_field)"
		for i in $(seq 1 $FLOOTFORKS)
		do
			LOL[$i]="$LOL_org"
		done
	else
		for i in $(seq 1 $FLOOTFORKS)
		do
		  #LOL[$i]="OFFSET 1 200"
		  #LOL[$i]="OFFSET $(shuf -i 0-1760 -n 1) $(shuf -i 0-920 -n 1)"
	#	  LOL[$i]="$(shuf_xy)"
		  LOL[$i]="$(cat $PIXLIST | shuf)"
		
		if [ -z "$ALPHACOLOR" ]
		then 
			LOL[$i]="$(cat $PIXLIST | shuf)"
		else
			LOL[$i]="$(grep -v $ALPHACOLOR $PIXLIST | shuf)"
		fi
		
		done
	fi
	

	
	while true
	do
		for i in $(seq $FLOOTFORKS) 
		do
			if [ -z ${LOLPID[$i]} ] || ! ps -p ${LOLPID[$i]} > /dev/null
			then
				echo "$(shuf_xy)
${LOL[$i]}" > /dev/tcp/$IPFLOOT/1337 &
				#echo "${LOL[$i]}" > /dev/tcp/127.0.0.1/1337 &
				LOLPID[$i]=$!
			fi
		done
	done
}

case $1 in

	draw_pixmap) draw_pixmap
	;;
	
	gen_pixmap) gen_pixmap
	;;

	convertimg)
		gen_pixmap
		draw_pixmap
	;;
		
	floot) if [ "$SHUFMODE" == "static" ] && ([ -z "$W" ] && [ -z "$H" ])
         then
           echo "please specify coords with e.g. 'W=420 H=420 SHUFMODE=static $0 floot $FNAME" >&2
           exit 1
         fi
         
         floot
	;;
	*)
		echo "$0 [convertimg, draw_pixmap, floot] [FILENAME] ([alpha color])"
		exit 1
		;;
	esac