blob: 584f9ece04ee4de35630c5e06e75171902782a25 [file] [log] [blame]
Harald Welte77804892011-02-06 18:12:47 +01001% Osmocom Erlang utility functions
2
3% (C) 2011 by Harald Welte <laforge@gnumonks.org>
4%
5% All Rights Reserved
6%
7% This program is free software; you can redistribute it and/or modify
8% it under the terms of the GNU Affero General Public License as
9% published by the Free Software Foundation; either version 3 of the
10% License, or (at your option) any later version.
11%
12% This program is distributed in the hope that it will be useful,
13% but WITHOUT ANY WARRANTY; without even the implied warranty of
14% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15% GNU General Public License for more details.
16%
17% You should have received a copy of the GNU Affero General Public License
18% along with this program. If not, see <http://www.gnu.org/licenses/>.
19
20-module(osmo_util).
21-author('Harald Welte <laforge@gnumonks.org>').
22
23-export([digit_list2int/1, int2digit_list/1]).
24-export([reload_config/0]).
Harald Welte8a6823d2011-03-08 16:03:31 +010025-export([tuple_walk/3, tuple_walk_print_cb/3]).
Harald Welte0f2f5962011-04-04 15:59:49 +020026-export([make_prim/4, make_prim/3]).
Harald Welte8ff5df42011-04-04 21:52:52 +020027-export([pointcode2int/1, pointcode2int/2, pointcode_fmt/2]).
Harald Welte0f2f5962011-04-04 15:59:49 +020028
29-include("osmo_util.hrl").
30
31-compile({parse_transform, exprecs}).
32-export_records([primitive]).
Harald Welte77804892011-02-06 18:12:47 +010033
34% Convert a list of digits to an integer value
35digit_list2int(Int, []) ->
36 Int;
37digit_list2int(Int, [Digit|Tail]) ->
38 digit_list2int(Int*10 + Digit, Tail).
39digit_list2int(Digits) when is_list(Digits) ->
40 digit_list2int(0, Digits).
41
42% Convert an integer value into a list of decimal digits
43int2digit_list(0, Digits) when is_list(Digits) ->
44 Digits;
45int2digit_list(Int, Digits) when is_integer(Int), is_list(Digits) ->
46 Digit = Int rem 10,
47 int2digit_list(Int div 10, [Digit|Digits]).
48int2digit_list(Int) when is_integer(Int) ->
49 int2digit_list(Int, []).
50
51% reload configuration of an application
52reload_config() ->
53 case init:get_argument(config) of
54 {ok, [ Files ]} ->
55 ConfFiles = [begin
56 S = filename:basename(F,".config"),
57 filename:join(filename:dirname(F),
58 S ++ ".config")
59 end || F <- Files],
60 % Move sys.config to the head of the list
61 Config = lists:sort(fun("sys.config", _) -> true;
62 (_, _) -> false end, ConfFiles),
63
64 OldEnv = application_controller:prep_config_change(),
65
66 Apps = [{application, A, make_appl(A)}
67 || {A,_,_} <- application:which_applications()],
68 application_controller:change_application_data(Apps, Config),
69 application_controller:config_change(OldEnv);
70 _ ->
71 {ok, []}
72 end.
73
74make_appl(App) when is_atom(App) ->
75 AppList = element(2,application:get_all_key(App)),
76 FullName = code:where_is_file(atom_to_list(App) ++ ".app"),
77 case file:consult(FullName) of
78 {ok, [{application, _, Opts}]} ->
79 Env = proplists:get_value(env, Opts, []),
80 lists:keyreplace(env, 1, AppList, {env, Env});
81 {error, _Reason} ->
82 lists:keyreplace(env, 1, AppList, {env, []})
83 end.
Harald Weltecf436012011-03-07 12:54:15 +010084
85
86% Walk a named tuple and (recursively) all its fields, call user-supplied
87% callback for each of them
Harald Welte8a6823d2011-03-08 16:03:31 +010088tuple_walk(Tpl, TupleCb, Args) when is_tuple(Tpl), is_function(TupleCb),
89 is_list(Args) ->
90 tuple_walk([], Tpl, TupleCb, Args).
Harald Weltecf436012011-03-07 12:54:15 +010091
Harald Welte8a6823d2011-03-08 16:03:31 +010092tuple_walk(Path, Tpl, TupleCb, Args) when is_list(Path), is_tuple(Tpl),
93 is_list(Args) ->
Harald Weltecf436012011-03-07 12:54:15 +010094 % call Callback
Harald Welte8a6823d2011-03-08 16:03:31 +010095 NewTpl = TupleCb(Path, Tpl, Args),
Harald Weltecf436012011-03-07 12:54:15 +010096 [TplName|TplList] = tuple_to_list(NewTpl),
Harald Welte8a6823d2011-03-08 16:03:31 +010097 NewTplList = tuple_fieldlist_walk(Path, TplName, TplList, TupleCb, Args),
Harald Welte4a0afae2011-03-07 23:54:04 +010098 list_to_tuple([TplName|NewTplList]);
Harald Welte8a6823d2011-03-08 16:03:31 +010099tuple_walk(Path, TplL, TupleCb, Args) when is_list(Path), is_list(TplL),
100 is_list(Args) ->
101 tuple_walk_list(Path, TplL, TupleCb, Args, []).
Harald Welte4a0afae2011-03-07 23:54:04 +0100102
Harald Welte8a6823d2011-03-08 16:03:31 +0100103tuple_walk_list(_Path, [], _TupleCb, _Args, OutList) ->
Harald Welte4a0afae2011-03-07 23:54:04 +0100104 OutList;
Harald Welte8a6823d2011-03-08 16:03:31 +0100105tuple_walk_list(Path, [Head|Tail], TupleCb, Args, OutList) ->
Harald Welte4a0afae2011-03-07 23:54:04 +0100106 if
107 is_tuple(Head) ->
Harald Welte8a6823d2011-03-08 16:03:31 +0100108 NewHead = tuple_walk(Path, Head, TupleCb, Args);
Harald Welte4a0afae2011-03-07 23:54:04 +0100109 is_list(Head) ->
Harald Welte8a6823d2011-03-08 16:03:31 +0100110 NewHead = tuple_walk(Path, Head, TupleCb, Args);
Harald Welte4a0afae2011-03-07 23:54:04 +0100111 true ->
112 NewHead = Head
113 end,
Harald Welte8a6823d2011-03-08 16:03:31 +0100114 tuple_walk_list(Path, Tail, TupleCb, Args, OutList++[NewHead]).
Harald Welte4a0afae2011-03-07 23:54:04 +0100115
Harald Weltecf436012011-03-07 12:54:15 +0100116
Harald Welte8a6823d2011-03-08 16:03:31 +0100117tuple_fieldlist_walk(Path, TplName, FieldList, TupleCb, Args) ->
118 tuple_fieldlist_walk(Path, TplName, FieldList, TupleCb, Args, []).
Harald Weltecf436012011-03-07 12:54:15 +0100119
Harald Welte8a6823d2011-03-08 16:03:31 +0100120tuple_fieldlist_walk(_Path, _TplName, [], _TplCb, _Args, OutList) ->
Harald Weltecf436012011-03-07 12:54:15 +0100121 OutList;
Harald Welte8a6823d2011-03-08 16:03:31 +0100122tuple_fieldlist_walk(Path, TplName, [Head|List], TupleCb, Args, OutList) ->
Harald Weltecf436012011-03-07 12:54:15 +0100123 if
124 is_tuple(Head) ->
Harald Welte8a6823d2011-03-08 16:03:31 +0100125 NewHead = tuple_walk(Path++[TplName], Head, TupleCb, Args);
Harald Welte4a0afae2011-03-07 23:54:04 +0100126 is_list(Head) ->
Harald Welte8a6823d2011-03-08 16:03:31 +0100127 NewHead = tuple_walk(Path++[TplName], Head, TupleCb, Args);
Harald Weltecf436012011-03-07 12:54:15 +0100128 true ->
129 NewHead = Head
130 end,
Harald Welte8a6823d2011-03-08 16:03:31 +0100131 tuple_fieldlist_walk(Path, TplName, List, TupleCb, Args, OutList++[NewHead]).
Harald Weltecf436012011-03-07 12:54:15 +0100132
133
Harald Welte8a6823d2011-03-08 16:03:31 +0100134tuple_walk_print_cb(Path, Tpl, _Args) when is_list(Path), is_tuple(Tpl) ->
Harald Weltecf436012011-03-07 12:54:15 +0100135 io:format("~p:~p~n", [Path, Tpl]),
136 Tpl.
Harald Welte0f2f5962011-04-04 15:59:49 +0200137
138% helper function to create a #primitive record
139make_prim(Subsys, GenName, SpecName) ->
140 make_prim(Subsys, GenName, SpecName, []).
141make_prim(Subsys, GenName, SpecName, Param) ->
142 #primitive{subsystem = Subsys, gen_name = GenName,
143 spec_name = SpecName, parameters = Param}.
Harald Welte8ff5df42011-04-04 21:52:52 +0200144
145% parse a 3-tuple pointcode into a raw integer
Harald Welte7ad37a22011-12-08 11:50:00 +0100146pointcode2int(Int) when is_integer(Int) ->
147 Int;
Harald Welte0bd35282012-01-23 17:35:57 +0100148pointcode2int(undefined) ->
149 undefined;
Harald Welteba0ada72011-12-08 00:56:30 +0100150pointcode2int(#pointcode{repr=Type, value=Value}) ->
151 pointcode2int(Type, Value);
Harald Welte8ff5df42011-04-04 21:52:52 +0200152pointcode2int({Std, Param}) ->
153 pointcode2int(Std, Param).
154
155pointcode2int(itu, {A, B, C}) ->
156 <<PcInt:14/big>> = <<A:3, B:8, C:3>>,
157 PcInt;
158pointcode2int(ansi, {A, B, C}) ->
159 <<PcInt:24/big>> = <<A:8, B:8, C:8>>,
160 PcInt;
161pointcode2int(ttc, {A, B, C}) ->
162 <<PcInt:16/big>> = <<A:5, B:4, C:7>>,
163 PcInt.
164
165% format a point-code into a 3-tuple according to the standard used
166pointcode_fmt(Std, P) when is_binary(P) ->
167 <<PcInt/integer>> = P,
168 pointcode_fmt(Std, PcInt);
169pointcode_fmt(itu, PcInt) when is_integer(PcInt) ->
170 <<A:3, B:8, C:3>> = <<PcInt:14/big>>,
171 {pointcode, itu, {A, B, C}};
172pointcode_fmt(ansi, PcInt) ->
173 <<A:8, B:8, C:8>> = <<PcInt:24/big>>,
174 {pointcode, ansi, {A, B, C}};
175pointcode_fmt(ttc, PcInt) ->
176 <<A:5, B:4, C:7>> = <<PcInt:16/big>>,
177 {pointcode, ttc, {A, B, C}}.