febc1725a0a370ae6510c9f915733a794ece0770
[tiramisu.git] / autolib.py
1 # Copyright (C) 2012 Team tiramisu (see README 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 DisabledOptionError, SpecialOwnersError
23 # ____________________________________________________________
24 # automatic Option object
25 special_owners = ['auto', 'fill']
26                   
27 def special_owner_factory(name, owner, value, 
28                           callback, callback_params=None, config=None):
29     # in case of an 'auto' and a 'fill' without a value, 
30     # we have to carry out a calculation
31     return calc_factory(name, callback, callback_params, config)
32
33 def calc_factory(name, callback, callback_params, config):
34     # FIXME we have to know the exact status of the config 
35     # not to disrupt it
36     # config.freeze()
37     if callback_params is None:
38         callback_params = {}
39     tcparams = {}
40     one_is_multi = False
41     len_multi = 0
42     for key, value in callback_params.items():
43         if type(value) == tuple:
44             path, check_disabled = value
45             try:
46                 opt_value = getattr(config, path)
47                 opt = config.unwrap_from_path(path)
48             except DisabledOptionError, e:
49                 if chek_disabled:
50                     continue
51                 raise DisabledOptionError(e)
52             is_multi = opt.is_multi()
53             if is_multi:
54                 if opt_value != None:
55                     len_value = len(opt_value)
56                     if len_multi != 0 and len_multi != len_value:
57                         raise SpecialOwnersError('unable to carry out a calculation, '
58                         'option values with multi types must have same length for: '
59                          + name)
60                     len_multi = len_value
61                 one_is_multi = True
62             tcparams[key] = (opt_value, is_multi)
63         else:
64             tcparams[key] = (value, False)
65
66     if one_is_multi:
67         ret = []
68         for incr in range(len_multi):
69             tcp = {}
70             for key, couple in tcparams.items():
71                 value, ismulti = couple
72                 if ismulti and value != None:
73                     tcp[key] = value[incr]
74                 else:
75                     tcp[key] = value
76             ret.append(calculate(name, callback, tcp))
77         return ret
78     else:
79         tcp = {}
80         for key, couple in tcparams.items():
81             tcp[key] = couple[0]
82         return calculate(name, callback, tcp)
83     
84 def calculate(name, callback, tcparams):
85     try:
86         # XXX not only creole...
87         from creole import eosfunc
88         return getattr(eosfunc, callback)(**tcparams) 
89     except AttributeError, err:
90         import traceback
91         traceback.print_exc()
92         raise SpecialOwnersError("callback: {0} return error {1} for "
93                        "option: {2}".format(callback, str(err), name))
94