Mercurial > repos > public > sbplib
annotate +grid/evalOn.m @ 275:3ea2ae2a3d15 feature/beams
Improvments to eval on grids. Added function to extrac components of a vector gf.
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Tue, 06 Sep 2016 17:28:11 +0200 |
| parents | ac8e00883986 |
| children | 30321dc180e1 |
| rev | line source |
|---|---|
| 157 | 1 % Takes a funciton and evaluates it on a grid to return a grid function in the |
| 2 % form of a (n*k)x1 vector, where n is the number of grid points and k is the | |
| 3 % number of components of the function. | |
| 4 % g -- Grid to evaluate on. | |
| 5 % func -- Function to evaluate. May be a function handle or a constant. If | |
| 6 % it is a vector value it has to be provided as a column vector, | |
| 7 function gf = evalOn(g, func) | |
| 8 if ~isa(func, 'function_handle') | |
| 9 % We should have a constant. | |
| 10 if size(func,2) ~= 1 | |
| 11 error('grid:evalOn:VectorValuedWrongDim', 'A vector valued function must be given as a column vector') | |
| 12 end | |
| 13 | |
| 14 gf = repmat(func,[g.N, 1]); | |
| 15 return | |
| 16 end | |
| 17 % func should now be a function_handle | |
| 18 | |
| 19 % Get coordinates and convert to cell array for easier use as a parameter | |
|
275
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
20 x = num2cell(g.points()); |
| 157 | 21 |
| 22 % Find the number of components | |
|
275
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
23 x0 = x(1,:); |
| 157 | 24 f0 = func(x0{:}); |
| 25 k = length(f0); | |
| 26 | |
| 27 if size(f0,2) ~= 1 | |
| 28 error('grid:evalOn:VectorValuedWrongDim', 'A vector valued function must be given as a column vector') | |
| 29 end | |
| 30 | |
|
275
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
31 gf = zeros(g.N*k, 1); |
|
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
32 % keyboard |
|
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
33 for i = 1:g.N |
|
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
34 % (1 + (i-1)*k):(i*k) |
|
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
35 % func(x{i,:}) |
|
3ea2ae2a3d15
Improvments to eval on grids. Added function to extrac components of a vector gf.
Jonatan Werpers <jonatan@werpers.com>
parents:
274
diff
changeset
|
36 gf((1 + (i-1)*k):(i*k)) = func(x{i,:}); |
|
171
0e56192f6459
Made evalOn a bit clearer.
Jonatan Werpers <jonatan@werpers.com>
parents:
157
diff
changeset
|
37 end |
| 157 | 38 end |