Homec4science
Diffusion SPClibs (feature/fft-split) Browse

spclibs/matlab/two_grid.m
03540012fedefeature/fft-split

Last Change
Raw File

two_grid.m
View Options

%
% @file two_grid.m
%
% @brief
%
% @copyright
% Copyright (©) 2021 EPFL (Ecole Polytechnique Fédérale de Lausanne)
% SPC (Swiss Plasma Center)
%
% spclibs is free software: you can redistribute it and/or modify it under
% the terms of the GNU Lesser General Public License as published by the Free
% Software Foundation, either version 3 of the License, or (at your option)
% any later version.
%
% spclibs is distributed in the hope that it will be useful, but WITHOUT ANY
% WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
% FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
%
% You should have received a copy of the GNU Lesser General Public License
% along with this program. If not, see <https://www.gnu.org/licenses/>.
%
% @authors
% (in alphabetical order)
% @author Trach-Minh Tran <trach-minh.tran@epfl.ch>
%
file='two_grid.h5';
%
nx=h5readatt(file,'/','NX');
nidbas=h5readatt(file,'/','NIDBAS');
levels=h5readatt(file,'/','LEVELS');
title_str=sprintf('NX = %d, NIDBAS = %d, levels = %d', nx, nidbas, levels);
%
% Read matrices at coarset grid
%
for lev=2:levels
%
% FE mat at fine grid
mglevel=sprintf('/mglevels/level.%.2d', lev-1);
dset=strcat(mglevel,'/mata');
ku=h5readatt(file,dset,'KU'); kl=ku;
n=h5readatt(file,dset,'RANK');
gbmat=h5read(file,dset);
mata_f=zeros(n,n);
for i=1:n
jmin = max(1,i-kl);
jmax = min(n,i+ku);
for j=jmin:jmax
ib = kl+ku+i-j+1;
mata_f(i,j)=gbmat(ib,j);
end
end
dset=strcat(mglevel,'/f');
f_fine = h5read(file,dset);
dset=strcat(mglevel,'/v');
v_fine = h5read(file,dset);
%
% FE mat at coarse grid
mglevel=sprintf('/mglevels/level.%.2d', lev);
dset=strcat(mglevel,'/mata');
ku=h5readatt(file,dset,'KU'); kl=ku;
n=h5readatt(file,dset,'RANK');
gbmat=h5read(file,dset);
mata_c=zeros(n,n);
for i=1:n
jmin = max(1,i-kl);
jmax = min(n,i+ku);
for j=jmin:jmax
ib = kl+ku+i-j+1;
mata_c(i,j)=gbmat(ib,j);
end
end
dset=strcat(mglevel,'/f');
f_coarse = h5read(file,dset);
dset=strcat(mglevel,'/v');
v_coarse = h5read(file,dset);
%
% Prolong mat
dset=strcat(mglevel,'/matp');
matp=h5read(file,dset);
%
% Check
fprintf(1,'Level %d: ||A_coarse - P''*A_fine*P|| = %g\n', lev, norm(matp'*mata_f*matp ...
- mata_c))
end
%
v_prolong = h5read(file,'/v_prolong');
figure
plot(v_fine)
hold on
plot(v_prolong, 'r')
c4science · Help