view find_elements.m @ 641:070d578997f6 feature/d1_staggered

Add helper function for primal and dual 1D staggered grid, and a test for that function.
author Martin Almquist <malmquist@stanford.edu>
date Sat, 11 Nov 2017 20:26:37 -0800
parents 48b6fb693025
children
line wrap: on
line source

% I = find_elements(a,b)
% Finds the index of elements a in b.
% a and b have to be in the same order.
function I = find_elements(a,b)
    I = [];

    j = 1;
    for i = 1:length(a)
        while b(j) ~= a(i)
            j = j + 1;
        end
        I(end+1) = j;
        j = j+1;
    end

    assert(length(I) == length(a),'Expected %d but got %d elements',length(a),length(I))
end