Mercurial > repos > public > sbplib
view +scheme/+bc/forcingSetup.m @ 873:dee5b5a57be6 bcSetupExperiment
Clean up closureSetup
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Thu, 01 Nov 2018 12:05:53 +0100 |
parents | fb91d12093f8 |
children | 93489ddb73e8 |
line wrap: on
line source
function S = forcingSetup(diffOp, penalties, bcs, S_sign) default_arg('S_sign', 1); assertType(bcs, 'cell'); assertIsMember(S_sign, [1, -1]); scheme.bc.verifyFormat(bcs, diffOp); % % Setup storage arrays % closure = spzeros(size(diffOp)); % gridData = {}; % symbolicData = {}; % Loop over bcs and collect data for i = 1:length(bcs) % [ok, isSym, data] = parseData(bcs{i}, penalties{i}, diffOp.grid) % if ~ok % % There was no data % continue % end % if isSym % gridData{end+1} = data; % else % symbolicData{end+1} = data; % end end % Setup penalty function O = spzeros(size(diffOp),1); function v = S_fun(t) v = O; for i = 1:length(gridData) v = v + gridData{i}.penalty*gridData{i}.func(t); end for i = 1:length(symbolicData) v = v + symbolicData{i}.penalty*symbolicData{i}.func(t, symbolicData{i}.coords{:}); end v = S_sign * v; end S = @S_fun; end function [ok, isSym, dataStruct] = parseData(bc, penalty, grid) if ~isfield(bc,'data') || isempty(bc.data) ok = false; return end ok = true; nArg = nargin(bc.data); if nArg > 1 % Symbolic data isSym = true; coord = grid.getBoundary(bc.boundary); dataStruct.penalty = penalty; dataStruct.func = bc.data; dataStruct.coords = num2cell(coord, 1); else % Grid data isSym = false; dataStruct.penalty = penalty; dataStruct.func = bcs{i}.data; end end