Homec4science
Diffusion FENNECS: Finite Element Non-Neutral Electron Cloud Simulator (geom_test) Browse

fennecs/matlab/splinequantity.m
990eaab682cegeom_test

Last Change
Raw File

splinequantity.m
View Options

classdef splinequantity < h5quantity
properties
knotsr
knotsz
femorder
index
postscale
end
methods
function obj=splinequantity(filename, dataset, knotsr, knotsz, femorder, scale, postscale, index)
if nargin<6
scale=1;
end
if nargin<8
index=-1;
end
obj=obj@h5quantity(filename, dataset, length(knotsr)-2*femorder(2), length(knotsz)-2*femorder(1), scale);
obj.knotsr=knotsr;
obj.knotsz=knotsz;
obj.femorder=double(femorder);
if nargin < 7
obj.postscale=ones(length(knotsr)-2*femorder(2), length(knotsz)-2*femorder(1));
else
obj.postscale=postscale;
end
if nargin < 8
obj.index=-1;
else
obj.index=index;
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},':')
t=1:obj.nt;
else
t=indices{3};
end
temp=zeros((obj.nz+obj.femorder(1)-1)*(obj.nr+obj.femorder(2)-1),length(t));
if obj.index ~= -1
if(sum(diff(diff(t))) == 0 && length(t)>1)
stride=t(2)-t(1);
temp = h5read(obj.filename, obj.dataset,[obj.index 1 t(1)],[1 Inf length(t)],[1 1 stride]);
else
for i=1:length(t)
temp(:,i) = h5read(obj.filename, obj.dataset,[obj.index 1 t(i)],[1 Inf 1]);
end
end
else
if(sum(diff(diff(t))) == 0 && length(t)>1)
stride=t(2)-t(1);
temp = h5read(obj.filename, obj.dataset,[1 t(1)],[Inf length(t)],[1 stride]);
else
for i=1:length(t)
temp(:,i) = h5read(obj.filename, obj.dataset,[1 t(i)],[Inf 1]) ;
end
end
end
temp=reshape(temp,obj.nz+obj.femorder(1)-1,obj.nr+obj.femorder(2)-1,[]);
temp=permute(temp,[2,1,3]);
quantity=temp(r,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},':')
z=1:obj.nz;
else
z=indices{2};
end
if strcmp(indices{3},':')
t=1:obj.nt;
else
t=indices{3};
end
count=length(t);
temp=obj.coeffs({':',':',t});
quantity=zeros(length(r),length(z),count);
for i=1:size(temp,3)
quantity(:,:,i)=obj.postscale(r,z).*fnval(spmak({obj.knotsr,obj.knotsz},temp(:,:,i)),{obj.knotsr(r+obj.femorder(2)),obj.knotsz(z+obj.femorder(1))});
end
end
function quantity=posval(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
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)
quantity(:,:,i)=fnval(fnder(spmak({obj.knotsr,obj.knotsz},temp(:,:,i)), order),{obj.knotsr(r+obj.femorder(2)),obj.knotsz(z+obj.femorder(1))});
end
end
end
end
c4science · Help