Homec4science
Diffusion FENNECS Lite (3Dmpimode) Browse

fennecslite/matlab/helper_classes/splinequantity3D.m
eae9143bc19d3Dmpimode

Last Change
Raw File

splinequantity3D.m
View Options

classdef splinequantity3D < h5quantity3D
properties
knotsr
knotsz
femorder
index
postscale
rgrid
thetgrid
zgrid
invdr
invdz
rcenters
zcenters
fileversion
end
methods
function obj=splinequantity3D(filename, dataset, knotsr, thetgrid, knotsz, femorder, scale, postscale, index)
if nargin<7
scale=1;
end
obj=obj@h5quantity3D(filename, dataset, length(knotsr)-2*femorder(2), length(thetgrid), length(knotsz)-2*femorder(1), scale);
obj.thetgrid=thetgrid;
obj.knotsr=knotsr;
obj.knotsz=knotsz;
obj.femorder=double(femorder);
if nargin < 8
obj.postscale=ones(length(knotsr)-2*femorder(2), length(thetgrid), length(knotsz)-2*femorder(1));
else
obj.postscale=postscale;
end
if nargin < 9
obj.index=-1;
else
obj.index=index;
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},':')
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
temp=zeros((obj.nz+obj.femorder(1)-1)*(obj.nr+obj.femorder(2)-1),obj.nthet,length(t));
if obj.index ~= -1
for i=1:length(t)
temp(:,:,i) = h5read(obj.filename, obj.dataset,[obj.index 1 1 t(i)],[1 Inf Inf 1]);
end
else
for i=1:length(t)
temp(:,:,i) = h5read(obj.filename, obj.dataset,[1 1 t(i)],[Inf Inf 1]) ;
end
end
temp=reshape(temp,obj.nr+obj.femorder(1)-1,obj.nz+obj.femorder(2)-1,obj.nthet,[]);
temp=permute(temp,[1,3,2,4]);
quantity=temp(r,thet,z,:)*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},':')
thet=1:obj.nthet;
else
thet=indices{2};
end
if strcmp(indices{3},':')
z=1:obj.nz;
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});
quantity=zeros(length(r),length(thet),length(z),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:length(thet)
valued.Values=squeeze(temp(:,thet(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