cf5d40ca5b046d5c1d2a94f8c053b95e7eeb3020
[tiramisu.git] / tiramisu / autolib.py
1 # Copyright (C) 2012-2013 Team tiramisu (see AUTHORS for all contributors)
2 #
3 # This program is free software; you can redistribute it and/or modify
4 # it under the terms of the GNU General Public License as published by
5 # the Free Software Foundation; either version 2 of the License, or
6 # (at your option) any later version.
7 #
8 # This program is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 # GNU General Public License for more details.
12 #
13 # You should have received a copy of the GNU General Public License
14 # along with this program; if not, write to the Free Software
15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16 #
17 # The original `Config` design model is unproudly borrowed from
18 # the rough gus of pypy: pypy: http://codespeak.net/svn/pypy/dist/pypy/config/
19 # the whole pypy projet is under MIT licence
20 # ____________________________________________________________
21 "enables us to carry out a calculation and return an option's value"
22 from tiramisu.error import PropertiesOptionError, ConflictConfigError
23 # ____________________________________________________________
24 # automatic Option object
25 #def special_owner_factory(name, owner, value,
26 #                          callback, callback_params=None, config=None):
27 #    # in case of an 'auto' and a 'fill' without a value,
28 #    # we have to carry out a calculation
29 #    return calc_factory(name, callback, callback_params, config)
30
31
32 def carry_out_calculation(name, config, callback, callback_params):
33     # FIXME we have to know the exact status of the config
34     # not to disrupt it
35     # config.freeze()
36     #callback, callback_params = option.getcallback()
37     #if callback_params is None:
38     #    callback_params = {}
39     tcparams = {}
40     one_is_multi = False
41     len_multi = 0
42
43     for key, values in callback_params.items():
44         for value in values:
45             if type(value) == tuple:
46                 path, check_disabled = value
47                 try:
48                     opt_value = config._getattr(path, force_permissive=True)
49                     opt = config.unwrap_from_path(path)
50                 except PropertiesOptionError, err:
51                     if check_disabled:
52                         continue
53                     raise PropertiesOptionError(err, err.proptype)
54                 is_multi = opt.is_multi()
55                 if is_multi:
56                     if opt_value is not None:
57                         len_value = len(opt_value)
58                         if len_multi != 0 and len_multi != len_value:
59                             raise ConflictConfigError('unable to carry out '
60                                                       'a calculation, option '
61                                                       'values with multi types'
62                                                       ' must have same length '
63                                                       'for: ' + name)
64                         len_multi = len_value
65                     one_is_multi = True
66                 tcparams.setdefault(key, []).append((opt_value, is_multi))
67             else:
68                 tcparams.setdefault(key, []).append((value, False))
69
70     if one_is_multi:
71         ret = []
72         for incr in range(len_multi):
73             tcp = {}
74             params = []
75             for key, couples in tcparams.items():
76                 for couple in couples:
77                     value, ismulti = couple
78                     if ismulti and value is not None:
79                         if key == '':
80                             params.append(value[incr])
81                         else:
82                             if len(value) > incr:
83                                 tcp[key] = value[incr]
84                             else:
85                                 tcp[key] = ''
86                     else:
87                         if key == '':
88                             params.append(value)
89                         else:
90                             tcp[key] = value
91             calc = calculate(name, callback, params, tcp)
92             if isinstance(calc, list):
93                 ret.extend(calc)
94             else:
95                 ret.append(calc)
96
97         return ret
98     else:
99         tcp = {}
100         params = []
101         for key, couples in tcparams.items():
102             for couple in couples:
103                 if key == '':
104                     value = couple[0]
105                     params.append(value)
106                 else:
107                     tcp[key] = couple[0]
108         return calculate(name, callback, params, tcp)
109
110
111 def calculate(name, callback, params, tcparams):
112     try:
113         # XXX not only creole...
114         from creole import eosfunc
115         return getattr(eosfunc, callback)(*params, **tcparams)
116     except AttributeError, err:
117         import traceback
118         traceback.print_exc()
119         raise ConflictConfigError("callback: {0} return error {1} for "
120                                   "option: {2}".format(callback, str(err),
121                                                        name))