Add HT, HU,... to get_grid

[matt-long]

HT = z_w(KMT)
HU = z_w(KMU)
KMU[i, j] = min(KMT[i, j] , KMT[i−1,j] , KMT[i, j−1], KMT[i−1,j−1])

[klindsay28]

Note that the above equations will generate an IndexError where KMT==km,
and are incorrect if partial_bottom_cells=.true..
To handle pbc grids, I suggest adding bottom_cell_file to pop_grid_definitions.yaml.

[dcherian]

From @klindsay28 in #44 (comment)

The following (untested) code replaces nested j,i loops with a single k loop.
It seems like it would be more cache friendly.

for k in range(VOL.shape[0]):
    VOL[k, :, :] = np.where(k < KMT, VOL[k, :, :], 0.0)

We'll need this to specify VOL as an xgcm metric.