#!/bin/sh
set -e

if [ $# -ne 4 ] ; then
  echo "Error in $0: invalid number of arguments $#"
  exit 1
fi

pfx=$1
inp1=$2
inp2=$3
out=$4

# Get CDO / TEMPORARY!
export PATH=/fs/ssm/hpco/exp/mib002/anaconda/anaconda-4.4.0/anaconda_4.4.0_ubuntu-14.04-amd64-64/envs/cdo-1.9.0/bin:$PATH

release nemo_mesh_mask
rm -f tmp*

access nemo_mesh_mask nemo_mesh_mask_rc3.nc

# deptho = e3t * tmask
ncap2 -O -s "deptho=e3t*tmask" nemo_mesh_mask tmp0
cdo -select,name=deptho tmp0 z.nc

# sum PFeN and PFeD
access pfen ${pfx}_${inp1}.nc
access pfed ${pfx}_${inp2}.nc
cdo add pfen pfed pbfe

# convert to mol m-3 s-1
cdo mulc,0.000001 pbfe tmp1

cdo mul tmp1 z.nc tmp2

# integral of -e3t*tmask*data along depth
cdo vertsum tmp2 ${pfx}_${out}

chmod uo+w ${pfx}_${out}

# Select nav_lon & nav_lat
ncks -O -C -a -v nav_lon,nav_lat pfen tmp3

# Append nav_lat & nav_lon
ncks -A tmp3 ${pfx}_${out}

release pfen pfed nemo_mesh_mask
rm -f tmp* z.nc pbfe