summaryrefslogtreecommitdiff
blob: 418d9ce675cab7f5d80059fcc77c7e55154d70e4 (plain)
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
-module(array_util).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% TYPES %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% EXPORTS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-export
(
   [
      any/2,
      none/2,
      all/2,

      mapiff/3
   ]
).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% LOCAL FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-spec any_internals
   (
      fun((any()) -> boolean()),
      array:array(any()),
      non_neg_integer()
   )
   -> boolean().
any_internals (_, _, 0) ->
   false;
any_internals (Fun, Array, PrevIX) ->
   IX = (PrevIX - 1),
   case Fun(array:get(IX, Array)) of
      true -> true;
      _ -> any_internals(Fun, Array, IX)
   end.

-spec all_internals
   (
      fun((any()) -> boolean()),
      array:array(any()),
      non_neg_integer()
   )
   -> boolean().
all_internals (_, _, 0) ->
   true;
all_internals (Fun, Array, PrevIX) ->
   IX = (PrevIX - 1),
   case Fun(array:get(IX, Array)) of
      true -> any_internals(Fun, Array, IX);
      _ -> false
   end.

-spec mapiff_internals
   (
      fun((any()) -> boolean()),
      fun((any()) -> any()),
      array:array(any()),
      list(non_neg_integer()),
      non_neg_integer()
   )
   -> {array:array(any()), list(non_neg_integer())}.
mapiff_internals (_, _, Array, IXList, 0) ->
   {Array, IXList};
mapiff_internals (Cond, Map, Array, IXList, PrevIX) ->
   IX = (PrevIX - 1),
   Elem = array:get(IX, Array),

   case Cond(Elem) of
      false -> mapiff_internals(Cond, Map, Array, IXList, IX);
      _ ->
         mapiff_internals
         (
            Cond,
            Map,
            array:set(IX, Map(Elem), Array),
            [IX|IXList],
            IX
         )
   end.

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-spec any (fun((any()) -> boolean()), array:array(any())) -> boolean().
any (Fun, Array) ->
   any_internals(Fun, Array, array:size(Array)).

-spec all (fun((any()) -> boolean()), array:array(any())) -> boolean().
all (Fun, Array) ->
   all_internals(Fun, Array, array:size(Array)).

-spec none (fun((any()) -> boolean()), array:array(any())) -> boolean().
none (Fun, Array) ->
   not any(Fun, Array).

-spec mapiff
   (
      fun((any()) -> boolean()),
      fun((any()) -> any()),
      array:array(any())
   )
   -> {array:array(any()), list(non_neg_integer())}.
mapiff (Cond, Map, Array) ->
   mapiff_internals(Cond, Map, Array, [], array:size(Array)).