Homec4science
Diffusion FENNECS lite (master) Browse

fennecslite/matlab/helper_classes/modequantity.m
ed2b1d6a5204master

Last Change
Raw File

modequantity.m
View Options

classdef modequantity < h5quantity3D
properties
knotsr
knotsz
nmodes
femorder
index
postscale
rgrid
zgrid
invdr
invdz
rcenters
zcenters
fileversion
end
methods
function obj=modequantity(filename, dataset, knotsr, knotsz, nmodes, femorder, scale, postscale)
if nargin<7
scale=1;
end
obj=obj@h5quantity3D(filename, dataset, length(knotsr)-2*femorder(2), nmodes, length(knotsz)-2*femorder(1), scale);
obj.nmodes=nmodes;
obj.knotsr=knotsr;
obj.knotsz=knotsz;
obj.femorder=double(femorder);
if nargin < 8
obj.postscale=ones(length(knotsr)-2*femorder(2), length(knotsz)-2*femorder(1),nmodes);
else
obj.postscale=postscale;
end
try
obj.fileversion = h5readatt(filename,'/data','file_version');
catch
obj.fileversion=1;
end
obj.rgrid=obj.knotsr((1:obj.nr) +obj.femorder(2));
obj.zgrid=obj.knotsz((1:obj.nz) +obj.femorder(1));
[dz,dr]=meshgrid(obj.zgrid(3:end)-obj.zgrid(1:end-2),obj.rgrid(3:end)-obj.rgrid(1:end-2));
obj.invdr=1./dr;
obj.invdz=1./dz;
obj.rcenters=movmean(obj.knotsr,[femorder(2) 1]);
obj.zcenters=movmean(obj.knotsz,[femorder(1) 1]);
obj.rcenters=[obj.rgrid(1); obj.rcenters(femorder(2)+2:end-2); obj.rgrid(end)] ;
obj.zcenters=[obj.zgrid(1); obj.zcenters(femorder(2)+2:end-2); obj.zgrid(end)] ;
end
function quantity=coeffs(obj,indices)
if strcmp(indices{1},':')
r=1:obj.nr+obj.femorder(2)-1;
else
r=indices{1};
end
if strcmp(indices{2},':')
z=1:obj.nz+obj.femorder(1)-1;
else
z=indices{2};
end
if strcmp(indices{3},':')
modes=1:obj.nmodes;
else
modes=indices{3};
end
if strcmp(indices{4},':')
t=1:obj.nt;
else
t=indices{4};
end
temp=zeros((obj.nz+obj.femorder(1)-1)*(obj.nr+obj.femorder(2)-1),obj.nmodes,length(t));
for i=1:length(t)
temp(:,:,i) = h5read(obj.filename, obj.dataset,[1 1 t(i)],[Inf Inf 1]);
end
temp=reshape(temp,obj.nr+obj.femorder(1)-1,obj.nz+obj.femorder(2)-1,obj.nmodes,[]);
%temp=permute(temp,[1,3,2,4]);
quantity=temp(r,z,modes,:)*obj.scale;
end
function quantity=val(obj,indices)
if strcmp(indices{1},':')
r=1:obj.nr;
else
r=indices{1};
end
if strcmp(indices{2},':')
z=1:obj.nz;
else
z=indices{2};
end
if strcmp(indices{3},':')
modes=1:obj.nmodes;
else
modes=indices{3};
end
if strcmp(indices{4},':')
t=1:obj.nt;
else
t=indices{4};
end
count=length(t);
temp=obj.coeffs({':',':',':',t});
quantity=zeros(length(r),length(z),length(modes),count);
[Z,R]=ndgrid(obj.zcenters,obj.rcenters);
[zg,rg]=ndgrid(obj.zgrid,obj.rgrid);
valued=griddedInterpolant(Z,R,zeros(length(obj.rgrid)+obj.femorder(2)-1,length(obj.zgrid)+obj.femorder(1)-1)');
for i=1:size(temp,4)
for m=1:size(modes)
valued.Values=squeeze(temp(:,:,modes(m),i))';
vals=valued(zg,rg)';
quantity(:,:,m,i)=vals(r,z);
end
end
end
% function quantity=posval(obj,indices)
% if strcmp(indices{1},':')
% r=1:obj.nr+obj.femorder(2)-1;
% else
% r=indices{1};
% end
% if strcmp(indices{2},':')
% thet=1:obj.nthet;
% else
% thet=indices{2};
% end
% if strcmp(indices{3},':')
% z=1:obj.nz+obj.femorder(1)-1;
% else
% z=indices{3};
% end
% if strcmp(indices{4},':')
% t=1:obj.nt;
% else
% t=indices{4};
% end
% count=length(t);
% temp=obj.coeffs({':',':',t});
% if(length(r) ~= length(z))
% error("r and z array must be the same size")
% end
% quantity=zeros(min(length(r),length(z)),count);
% for i=1:size(temp,3)
% quantity(:,i)=fnval(spmak({obj.knotsr,obj.knotsz},temp(:,:,i)),[r(:)';z(:)']);
% end
%
% end
%
%
% function quantity=der(obj,indices)
% if strcmp(indices{1},':')
% r=1:obj.nr;
% else
% r=indices{1};
% end
% if strcmp(indices{2},':')
% z=1:obj.nz;
% else
% z=indices{2};
% end
% if strcmp(indices{3},':')
% t=1:obj.nt;
% else
% t=indices{3};
% end
% order=indices{4};
% count=length(t);
% temp=obj.coeffs({':',':',t});
% quantity=zeros(length(r),length(z),count);
% for i=1:size(temp,3)
% %f=spmak({obj.knotsr,obj.knotsz},temp(:,:,i));
% %preder=fnval(f,{obj.knotsr(r+obj.femorder(2)),obj.knotsz(z+obj.femorder(1))});
% preder=obj.val({r,z,t(i)});%
% if order(1)>0
% preder(2:end-1,2:end-1)=(preder(3:end,2:end-1)-preder(1:end-2,2:end-1)).*obj.invdr;
% end
% if order(2)>0
% preder(2:end-1,2:end-1)=(preder(2:end-1,3:end)-preder(2:end-1,1:end-2)).*obj.invdz;
% end
% quantity(2:end-1,2:end-1,i)=preder(2:end-1,2:end-1);
% end
% end
end
end
c4science · Help