close all; clear all;

%% load bathymetry you want to simulate;

dx = 5;
dy = dx;

m = 128;
x =[0:1:m-1]*dx;
y = x;

nxe = ceil(length(x)*(sqrt(2)-1)/2);

xe = [-nxe:1:m-1+nxe]*dx;
ye = xe;

[XE,YE] = meshgrid(xe,ye);

initfiles = {'h_init','h_vegh','h_vegpos'};


for i = 1:length(initfiles);

    
    ve = zeros(size(XE));

    v = load([initfiles{i},'.dat']);

    ve(nxe+1:nxe+m,nxe+1:nxe+m) = v;

    ve(1:nxe,:) = repmat(ve(nxe+1,:),nxe,1);
    ve(m+nxe+1:m+2*nxe,:) = repmat(ve(nxe+m,:),nxe,1);
    ve(:,1:nxe) = repmat(ve(:,nxe+1),1,nxe);
    ve(:,m+1+nxe:m+2*nxe) = repmat(ve(:,nxe+m),1,nxe);

    figure; surf(XE,YE,ve); shading flat; colorbar;
    pause

    save([initfiles{i},'_ext.dat'],'ve','-ascii');
end


