util.py

# calc dt and dt^2 (first and second derivative)
def calc_adv(cv, u, v):

    dxcv = cv.differentiate(coord='lon')  # optional: add edge_order=2. default is 1.
    dycv = cv.differentiate(coord='lat')  # optional: add edge_order=2. default is 1.

    adv = -( u *dxcv) - ( v *dycv)           # adv = u*d/dx(cv) + v*d/dy(cv)

    dxadv = adv.differentiate(coord='lon')
    dyadv = adv.differentiate(coord='lat')

    advadv = ( u *dxadv) + ( v *dyadv)  # to constrain adv to locations where there is a local min.
    return (adv, advadv)