Mercurial > repos > public > sbplib

view gridDerivatives.m @ 1243:4e0b88f3def1 feature/dirac_discr

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Use rng(1) to set seed in diracDiscrTest because hex-speak crashed.
author Martin Almquist <malmquist@stanford.edu>
date Tue, 19 Nov 2019 17:11:33 -0800
parents 48b6fb693025
children
line wrap: on
line source

% Calculates derivatives in all directions of function.
% Uses multi-dim vector form.
%   gridDerivatives(F,Dx,Dy)
%   gridDerivatives(F,Dx,Dy,Dz)
function varargout = gridDerivatives(F, varargin)
    assert(length(varargin) == ndims(F));

    switch ndims(F)
        case 2
            varargout{1} = (varargin{1}*F')';
            varargout{2} = varargin{2}*F;
        otherwise
            error('Not implemented for ndims(F) = %d',ndims(F));
    end
end