Fix interpolation of SSH to transects

This merge fixes masking of the SSH to ignore cells where maxLevelCell=0
MPAS-Dev · Oct 28, 2023 · 525e376 · 525e376
1 parent 6e1ccbe
commit 525e376
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/conda_package/mpas_tools/ocean/transects.py b/conda_package/mpas_tools/ocean/transects.py
@@ -378,9 +378,14 @@ def _get_vertical_coordinate(dsTransect, layerThickness, bottomDepth,
     layerThicknessTransect = (layerThicknessTransect*interpCellWeights).sum(
         dim='nHorizWeights')
 
+    interpMask = maxLevelCell.isel(nCells=interpCellIndices) >= 0
+    interpHorizCellWeights = interpMask*dsTransect.interpHorizCellWeights
+    weightSum = interpHorizCellWeights.sum(dim='nHorizWeights')
+    interpHorizCellWeights = \
+        (interpHorizCellWeights/weightSum).where(interpMask)
+
     sshTransect = ssh.isel(nCells=interpCellIndices)
-    sshTransect = (sshTransect*dsTransect.interpHorizCellWeights).sum(
-        dim='nHorizWeights')
+    sshTransect = (sshTransect*interpHorizCellWeights).sum(dim='nHorizWeights')
 
     zBot = sshTransect - layerThicknessTransect.cumsum(dim='nVertLevels')
     zTop = zBot + layerThicknessTransect