#!/bin/bash
# make sure you've ran create_toy_mc before
echo "This will take a while ... grab a coffee/tea/water"
mkdir -p logs
N_JOBS=4

i=0
k=3
options[0]="--offset_toy_mc=0 --offset_toy_data=0"
options[1]="--offset_toy_mc=35 --offset_toy_data=0"
options[2]="--offset_toy_mc=70 --offset_toy_data=0"
options[3]="--offset_toy_mc=0 --offset_toy_data=35" 
options[4]="--offset_toy_mc=0 --offset_toy_data=70" 
options[5]="--offset_toy_mc=35 --offset_toy_data=35" 
options[6]="--offset_toy_mc=70 --offset_toy_data=70" 
options[7]="--offset_toy_mc=35 --offset_toy_data=70" 
options[8]="--offset_toy_mc=70 --offset_toy_data=35"


echo 'Creating toy MC for all variables, 105x105, both electron and muon channels'
for var in MET HT ST WPT MT; do
echo "Doing unfolding pulls for: $var"
	case "$var" in
		MET)
			k=3
			;;
		HT)
			k=4
			;;
		ST)
			k=3
			;;
		MT)
			k=2
			;;
		WPT)
			k=3
			;;
		*)
			k=3
			;;
	esac
	
	for energy in 7 8; do
		for c in electron muon; do
			echo "Producing pull data for $var variable, $c channel, kv=$k, sqrt(s) = $energy TeV"
			for j in 1 2 3 4 5 6 7 8 9; do
				echo "Doing part $j out of 9"
				nohup time python dps/analysis/unfolding_tests/create_unfolding_pull_data.py -v $var -k $k -f data/toy_mc/toy_mc_${var}_N_300_${energy}TeV.root -n 35 ${options[$j-1]} -c $c -s $energy &> logs/pull_${var}_kv${k}_${c}_${j}_${energy}TeV.log &
				let i+=1
				if (( $i % N_JOBS == 0 ))
				then
					echo "Waiting on the above to finish."
    				wait;
				fi
			done
			nohup time python dps/analysis/unfolding_tests/make_unfolding_pull_plots.py -i data/pull_data/7TeV/$var/35_input_toy_mc/k_value_$k -c $c -v $var -k $k -s $energy &> logs/pull_${var}_kv${k}_${c}_plots_${energy}TeV.log &
			wait;
		done
	done
done
