source: branches/UKMO/dev_r5107_hadgem3_cplseq/NEMOGCM/TOOLS/MISCELLANEOUS/rewrite_nemo.sh @ 5477

#!/bin/bash
#
set -u
#set -xv
#
# for on each file containing a call to work alloc (exept BDY files that are too complicated...)
#for i in $( ack -il "^ *call *wrk_alloc *\(" | grep -v BDY )
for i in $( egrep -iRl "^ *call *wrk_alloc *\(" * | grep "90$" | grep -v BDY )
do
# create a temporary file that will be easier to process...
    sed -e "s/dimension/DIMENSION/" -e "s/pointer/POINTER/" $i > tmp$$
# replace "dimension(...), pointer" to "pointer, dimension(...)"
    sed -e "s/, *DIMENSION *(\([^)]*\)) *, *POINTER/, POINTER, DIMENSION(\1)/" tmp$$ > tmp2$$
    mv tmp2$$ tmp$$ 
# remove all comments in $i
    sed -e "s/\!.*//" tmp$$ > tmp2$$
    mv tmp2$$ tmp$$
# numbers of the lines defining the beginning and the end of each subroutine or functions
    n1=$( egrep -ic "(^ *subroutine *[0-9a-zA-Z_]|function *[0-9a-zA-Z_][0-9a-zA-Z_]* *\(.*\))" tmp$$ )
    n2=$( egrep -ic "(^ *end *subroutine|^ *end *function)" tmp$$ )
    if [ $n1 -ne $n2 ]
    then
   echo "error with the definition of subroutine/function blocks in $i" 
   exit
    fi
    linesbegin=$( egrep -in "(^ *subroutine *[0-9a-zA-Z_]|function *[0-9a-zA-Z_][0-9a-zA-Z_]* *\(.*\))" tmp$$ | sed -e "s/:.*//" )
    linesend=$( egrep -in "(^ *end *subroutine|^ *end *function)" tmp$$ | sed -e "s/:.*//" | sort -rn )
#
# number of the lines containing wrk_alloc
    cnt=$( grep -ci "^ *call *wrk_alloc *(" tmp$$ )
# for each of these lines
    ll=1
    while [ $ll -le $cnt ]
    do
# get the line with its number
   line=$( grep -in "^ *call *wrk_alloc *(" tmp$$ | sed -n ${ll}p | sed -e "s/\!.*//" )
# get its number
   lline=$( echo $line | sed -e "s/:.*//" )
# keep only the arument of wrk_alloc between ()
   line=$( echo $line | sed -e "s/[^(]*\((.*)\).*/\1/" | sed -e "s/, *k[ijkl]start *=[^,]*,/,/" | sed -e "s/, *k[ijkl]start *=.*)/ )/" )
# find in which subroutine or function is located this call to wrk_alloc: l1 beginning l2: end
   for lll in $linesbegin
   do
       [ $lline -gt $lll ] && l1=$lll
   done
   for lll in $linesend
   do
       [ $lline -lt $lll ] && l2=$lll
   done
   if [ $l2 -le $l1 ]
   then
       echo "error in $i in the indentification subroutine/function block containing line $lline"
       exit
   fi
# get the last argument of the call to wrk_alloc (between the last "," and ")"
   lw=$( echo $line | sed -e "s/.*,\(.*\)).*/\1/" | sed -e "s/ *//g" )
# get the line containing the definition of this argument and from it get the number of dimension of the argument
   tst=$( sed -n ${l1},${l2}p tmp$$  | grep -i "POINTER *, *DIMENSION *(.*) *::.*"${lw} | egrep -ic "[^0-9a-zA-Z_]("${lw}" *$|"${lw}" *,)" )
   if [ $tst -ne 1 ]
   then
       echo "error with ${lw} definition in $i" 
       exit
   fi
   dimsz=$( sed -n ${l1},${l2}p tmp$$  | grep -i "POINTER *, *DIMENSION *(.*) *::.*"${lw} | egrep -i "[^0-9a-zA-Z_]("${lw}" *$|"${lw}" *,)" | sed -e "s/.*POINTER *, *DIMENSION *(\(.*\)) *::.*/\1/" | sed -e "s/ //g" | wc -c )
   dimsz=$(( $dimsz / 2 ))
# get the number of variables concerned by the call to wrk_alloc (keep anly the , and count them)
   nvar=$( echo $line | sed -e "s/[^,]//g" | wc -c )
   nvar=$(( $nvar - $dimsz ))
# get the définition of the size of the dimensions from the call to wrk_alloc 
   case $dimsz in
       1) dim=$( echo $line | sed -e "s/( *\([^,]*\),.*)/\1/" | sed -e "s/ //g" ) ;; # keep between ( and the 1st ,
       2) dim=$( echo $line | sed -e "s/( *\([^,]*,[^,]*\),.*)/\1/" | sed -e "s/ //g" ) ;; # keep between ( and the 2nd ,
       3) dim=$( echo $line | sed -e "s/( *\([^,]*,[^,]*,[^,]*\),.*)/\1/" | sed -e "s/ //g" ) ;; # keep between ( and the 3rd ,
       4) dim=$( echo $line | sed -e "s/( *\([^,]*,[^,]*,[^,]*,[^,]*\),.*)/\1/" | sed -e "s/ //g" ) ;; # keep between ( and the 4th ,
       *) echo "error with the number of dimensions of $lw in $i"; exit ;;
   esac
# for each variable, change its definition...
   nn=1 
   var=$lw # start from the last argument
   while [ $nn -le $nvar ]
   do
# get the line where this variable is defined.  
       tst=$( sed -n ${l1},${l2}p tmp$$  | grep -ni "POINTER *, *DIMENSION *(.*) *::.*"${var} | egrep -ic "[^0-9a-zA-Z_]("${var}" *$|"${var}" *,)" )
       if [ $tst -ne 1 ]
       then
      echo "error with ${lw} definition in $i" 
      exit
       fi
# get the line number where the variable is defined
       lvar=$( sed -n ${l1},${l2}p tmp$$  | grep -ni "POINTER *, *DIMENSION *(.*) *::.*"${var} | egrep -i "[^0-9a-zA-Z_]("${var}" *$|"${var}" *,)" | sed -e "s/:.*//" )
       lvar=$(( $lvar + $l1 - 1 ))
# get the dimension definition and compare it to ${dim}
       dim2=$( sed -n -e "${lvar}p" $i | sed -e "s/dimension/DIMENSION/" -e "s/\!.*//" | sed -e "s/.*, *DIMENSION *(\(.*\)).*/\1/" | sed -e "s/ //g" )
       if [ $( echo $dim2 | wc -c ) -ne $(( $dimsz * 2 )) ] 
       then 
      if [ "$dim2" != "$dim" ] 
      then 
          echo "problem in $i in the defininition of variable $var: its size is $dim but we have $dim2"
          sed -n -e "${lvar}p" $i
          exit
      fi
       fi
# remove pointer
       sed -e "${lvar}s/pointer/POINTER/" $i | sed -e "${lvar}s/, *POINTER *,/,/" | sed -e "${lvar}s/, *POINTER */ /" > tmp2$$
       mv tmp2$$ $i
# redefine the dimensions 
       case $dimsz in
      1) sed -e "${lvar}s/\(DIMENSION *(\) *: *)/\1${dim})/" $i > tmp2$$ ;;
      2) sed -e "${lvar}s/\(DIMENSION *(\) *: *, *: *)/\1${dim})/" $i > tmp2$$ ;;
      3) sed -e "${lvar}s/\(DIMENSION *(\) *: *, *: *, *: *)/\1${dim})/" $i > tmp2$$ ;;
      4) sed -e "${lvar}s/\(DIMENSION *(\) *: *, *: *, *: *, *: *)/\1${dim})/" $i > tmp2$$ ;;
       esac
       mv tmp2$$ $i
       
# define the next variable (previous argument)
       var=$( echo $line | sed -e "s/.*,\(.*\), *[^0-9a-zA-Z_]*${var}[^0-9a-zA-Z_]*[,)].*/\1/" | sed -e "s/ *//g" )
       nn=$(( $nn + 1 ))
   done
   
   ll=$(( $ll + 1 ))
    done
#delete wrk_alloc line
    sed -e "/call *wrk_alloc *(/d" -e "/CALL *wrk_alloc *(/d" \
   -e "/call *wrk_dealloc *(/d" -e "/CALL *wrk_dealloc *(/d" \
   -e "/USE wrk_nemo/d"  -e "/use wrk_nemo/d" $i > tmp$$
    mv tmp$$ $i
    
    echo $i done
done
#
# some specific changes...
#
# OPA_SRC/SBC/albedo.F90
sed -e "s/DIMENSION(jpi,jpj,ijpl)/DIMENSION(jpi,jpj,SIZE(pt_ice,3))/" OPA_SRC/SBC/albedo.F90 > tmp$$
mv tmp$$ OPA_SRC/SBC/albedo.F90
# LIM_SRC_2/limrhg_2.F90
sed -e "s/DIMENSION(jpi,jpj+2)/DIMENSION(jpi,0:jpj+1)/" LIM_SRC_2/limrhg_2.F90 > tmp$$
mv tmp$$ LIM_SRC_2/limrhg_2.F90
# LIM_SRC_3/limitd_me.F90
sed -e "s/DIMENSION(jpi,jpj,jpl+2)/DIMENSION(jpi,jpj,-1:jpl)/" LIM_SRC_3/limitd_me.F90 > tmp$$
mv tmp$$ LIM_SRC_3/limitd_me.F90
# LIM_SRC_3/limitd_th.F90
sed -e "s/DIMENSION(jpi,jpj,jpl+1)/DIMENSION(jpi,jpj,0:jpl)/" LIM_SRC_3/limitd_th.F90 > tmp$$
mv tmp$$ LIM_SRC_3/limitd_th.F90
# LIM_SRC_3/limthd_dif.F90
sed -e "s/DIMENSION(kiut,nlay_i+1)/DIMENSION(kiut,0:nlay_i)/" \
    -e "s/DIMENSION(kiut,nlay_s+1)/DIMENSION(kiut,0:nlay_s)/" LIM_SRC_3/limthd_dif.F90 > tmp$$
mv tmp$$ LIM_SRC_3/limthd_dif.F90
# LIM_SRC_3/limthd_ent.F90
sed -e "s/DIMENSION(jpij,jkmax+4)/DIMENSION(jpij,0:jkmax+3)/" \
    -e "s/DIMENSION(jkmax+4,jkmax+4)/DIMENSION(0:jkmax+3,0:jkmax+3)/" LIM_SRC_3/limthd_ent.F90 > tmp$$
mv tmp$$ LIM_SRC_3/limthd_ent.F90
# OPA_SRC/DYN/divcur.F90
sed -e "s/DIMENSION(jpi+4,jpj)/DIMENSION(-1:jpi+2,jpj)/" OPA_SRC/DYN/divcur.F90 > tmp$$
mv tmp$$ OPA_SRC/DYN/divcur.F90
# OPA_SRC/LDF/ldfslp.F90
sed -e "s/DIMENSION(jpi,jpj,jpk,2)/DIMENSION(jpi,jpj,jpk,0:1)/" \
    -e "s/DIMENSION(jpi,jpj,2,2)/DIMENSION(jpi,jpj,0:1,0:1)/" OPA_SRC/LDF/ldfslp.F90 > tmp$$
mv tmp$$ OPA_SRC/LDF/ldfslp.F90
# OPA_SRC/ZDF/zdfkpp.F90
sed -e "s/DIMENSION(jpi,3) *::* zmoek/DIMENSION(jpi,0:2) ::   zmoek/" OPA_SRC/ZDF/zdfkpp.F90 > tmp$$
mv tmp$$ OPA_SRC/ZDF/zdfkpp.F90

# link for limrhg.F90...
cd LIM_SRC_2
ln -sf ../LIM_SRC_3/limrhg.F90 .