#!/bin/bash
export OPENBLAS_NUM_THREADS=1
export OMP_NUM_THREADS=1
export NPROC=96

export nb_lam=100
export nb_phi=59
export lambda_min=350
export lambda_max=800
export FLAG_TOTAL=0
export GRATING_CASE="half_ellipsoid"
export myDir="res_Matrix_nb_lam"$nb_lam"_nb_phi"$nb_phi"_total"$FLAG_TOTAL

rm -r $myDir
mkdir $myDir
gmsh grating3D.geo -3 -o grating3D.msh

myfunc() {
    local mysubDir=$myDir/run_lam$1_phi$2_psi$3
    mkdir $mysubDir
    cp grating3D.msh grating3D.pro grating3D_data_$GRATING_CASE.geo grating3D_data.geo grating3D_materials.pro $mysubDir
    local lam=$(echo "scale=10;400+400/($nb_lam-1)*$1" | bc )
    local phi=$(echo "scale=10;360/($nb_phi)*$2" | bc )
    local psi=$(echo "scale=10;90*$3" | bc )
    cd $mysubDir
    getdp grating3D.pro -pre helmholtz_vector -msh grating3D.msh -cal -pos postop_helmholtz_vector -petsc_prealloc 200 -setstring test_case $GRATING_CASE -setnumber lambda0 $lam -setnumber thetadeg 50 -setnumber phideg $phi -setnumber psideg $psi -setnumber FLAG_TOTAL $FLAG_TOTAL
    if [ $3 -eq 0 ]; then cp res3D/rs.txt ../r_pin_sout_lam$1_phi$2.out ; fi
    if [ $3 -eq 0 ]; then cp res3D/rp.txt ../r_pin_pout_lam$1_phi$2.out ; fi
    if [ $3 -eq 1 ]; then cp res3D/rs.txt ../r_sin_sout_lam$1_phi$2.out ; fi
    if [ $3 -eq 1 ]; then cp res3D/rp.txt ../r_sin_pout_lam$1_phi$2.out ; fi
    cp res3D/eff_t1.txt ../eff_t1_lam$1_phi$2_psi$3.out
    cp res3D/eff_r1.txt ../eff_r1_lam$1_phi$2_psi$3.out
    cp res3D/eff_t2.txt ../eff_t2_lam$1_phi$2_psi$3.out
    cp res3D/eff_r2.txt ../eff_r2_lam$1_phi$2_psi$3.out
    cp res3D/temp-Q_scat.txt ../Q_scat_lam$1_phi$2_psi$3.out
    cd ../..
    rm -r $mysubDir
}

export -f myfunc
parallel -j $NPROC myfunc ::: $(seq 0 $(($nb_lam-1))) ::: $(seq 0 $(($nb_phi-1))) ::: 0 1