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))