matlab-incremented/structmapleaves.m at master · zjroth/matlab-incremented · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
% USAGE:
%    stctOut = structmapleaves(fcnMap, stctIn)
%
% DESCRIPTION:
%    Apply a function to every leaf of a structure. A leaf in a (nested)
%    structure is a descendant (a field, or a field of a field, etc.) that is
%    not itself a structure.
%
% ARGUMENTS:
%    fcnMap
%       The function to apply to every leaf of the provided structure
%    stctIn
%       The structure
%
% EXAMPLE:
%    >> stctIn.field1 = 4;
%    >> stctIn.field2.a = 5;
%    >> stctIn.field2.b = 6;
%    >> stctOut = structmapleaves(@(x) x^2, stctIn);
%
%    stctOut =
%
%        field1: 16
%        field2: [1x1 struct]
%
%    >> stctOut.field2
%
%    ans =
%
%        a: 25
%        b: 36
function stctOut = structmapleaves(fcnMap, stctIn, bFlatten)
    if nargin < 3
        bFlatten = false;
    end

    if bFlatten
        stctOut = [];
    else
        stctOut = stctIn;
    end

    if isa(stctIn, 'struct')
        cellFields = fieldnames(stctIn);

        for i = 1 : length(cellFields)
            strField = cellFields{i};

            if bFlatten
                stctOut = vertcat(stctOut, structmapleaves( ...
                    fcnMap, stctIn.(strField), true));
            else
                stctOut.(strField) = structmapleaves( ...
                    fcnMap, stctIn.(strField), bFlatten);
            end
        end
    else
        stctOut = fcnMap(stctIn);
    end
end