Mercurial > repos > public > sbplib
diff +scheme/+bc/bcForcingSetup.m @ 869:d356f1a22d4f bcSetupExperiment
Start organizing the code
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Fri, 07 Sep 2018 09:19:03 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/+scheme/+bc/bcForcingSetup.m Fri Sep 07 09:19:03 2018 +0200 @@ -0,0 +1,70 @@ +function S = bcForcingSetup(diffOp, penalties, bcs, S_sign) %% SETUP in the name!! + default_arg('S_sign', 1); + + assertType(bcs, 'cell'); + assertIsMember(S_sign, [1, -1]); + + verifyBcFormat(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