Statistiques
| Révision :

root / pobysoPythonSage / src / pobyso.py @ 227

Historique | Voir | Annoter | Télécharger (77,31 ko)

1 5 storres
"""
2 209 storres
@file pobyso.py
3 5 storres
Actual functions to use in Sage
4 5 storres
ST 2012-11-13
5 5 storres

6 5 storres
Command line syntax:
7 5 storres
  use from Sage (via the "load" or the "attach" commands)
8 5 storres

9 38 storres
pobyso functions come in five flavors:
10 57 storres
- the _so_so (arguments and returned objects are pointers to Sollya objects,
11 57 storres
  includes the void function and the no arguments function that return a
12 57 storres
  pointer to a Sollya object);
13 38 storres
- the _so_sa (argument are pointers to Sollya objects, returned objects are
14 57 storres
  Sage/Python objects or, more generally, information is transfered from the
15 57 storres
  Sollya world to Sage/Python world; e.g. functions without arguments that
16 57 storres
  return a Sage/Python object);
17 38 storres
- the _sa_so (arguments are Sage/Python objects, returned objects are
18 38 storres
  pointers to Sollya objects);
19 38 storres
- the sa_sa (arguments and returned objects are all Sage/Python objects);
20 51 storres
- a catch all flavor, without any suffix, (e. g. functions that have no argument
21 51 storres
  nor return value).
22 57 storres
This classification is not always very strict. Conversion functions from Sollya
23 57 storres
to Sage/Python are sometimes decorated with Sage/Python arguments to set
24 57 storres
the precision. These functions remain in the so_sa category.
25 5 storres
NOTES:
26 5 storres
Reported errors in Eclipse come from the calls to
27 5 storres
the Sollya library
28 5 storres

29 10 storres
ToDo (among other things):
30 10 storres
 -memory management.
31 5 storres
"""
32 5 storres
from ctypes import *
33 37 storres
import re
34 37 storres
from sage.symbolic.expression_conversions import polynomial
35 59 storres
from sage.symbolic.expression_conversions import PolynomialConverter
36 38 storres
"""
37 38 storres
Create the equivalent to an enum for the Sollya function types.
38 38 storres
"""
39 5 storres
(SOLLYA_BASE_FUNC_ABS,
40 5 storres
SOLLYA_BASE_FUNC_ACOS,
41 5 storres
    SOLLYA_BASE_FUNC_ACOSH,
42 5 storres
    SOLLYA_BASE_FUNC_ADD,
43 5 storres
    SOLLYA_BASE_FUNC_ASIN,
44 5 storres
    SOLLYA_BASE_FUNC_ASINH,
45 5 storres
    SOLLYA_BASE_FUNC_ATAN,
46 5 storres
    SOLLYA_BASE_FUNC_ATANH,
47 5 storres
    SOLLYA_BASE_FUNC_CEIL,
48 5 storres
    SOLLYA_BASE_FUNC_CONSTANT,
49 5 storres
    SOLLYA_BASE_FUNC_COS,
50 5 storres
    SOLLYA_BASE_FUNC_COSH,
51 5 storres
    SOLLYA_BASE_FUNC_DIV,
52 5 storres
    SOLLYA_BASE_FUNC_DOUBLE,
53 5 storres
    SOLLYA_BASE_FUNC_DOUBLEDOUBLE,
54 5 storres
    SOLLYA_BASE_FUNC_DOUBLEEXTENDED,
55 5 storres
    SOLLYA_BASE_FUNC_ERF,
56 5 storres
    SOLLYA_BASE_FUNC_ERFC,
57 5 storres
    SOLLYA_BASE_FUNC_EXP,
58 5 storres
    SOLLYA_BASE_FUNC_EXP_M1,
59 5 storres
    SOLLYA_BASE_FUNC_FLOOR,
60 5 storres
    SOLLYA_BASE_FUNC_FREE_VARIABLE,
61 5 storres
    SOLLYA_BASE_FUNC_HALFPRECISION,
62 5 storres
    SOLLYA_BASE_FUNC_LIBRARYCONSTANT,
63 5 storres
    SOLLYA_BASE_FUNC_LIBRARYFUNCTION,
64 5 storres
    SOLLYA_BASE_FUNC_LOG,
65 5 storres
    SOLLYA_BASE_FUNC_LOG_10,
66 5 storres
    SOLLYA_BASE_FUNC_LOG_1P,
67 5 storres
    SOLLYA_BASE_FUNC_LOG_2,
68 5 storres
    SOLLYA_BASE_FUNC_MUL,
69 5 storres
    SOLLYA_BASE_FUNC_NEARESTINT,
70 5 storres
    SOLLYA_BASE_FUNC_NEG,
71 5 storres
    SOLLYA_BASE_FUNC_PI,
72 5 storres
    SOLLYA_BASE_FUNC_POW,
73 5 storres
    SOLLYA_BASE_FUNC_PROCEDUREFUNCTION,
74 5 storres
    SOLLYA_BASE_FUNC_QUAD,
75 5 storres
    SOLLYA_BASE_FUNC_SIN,
76 5 storres
    SOLLYA_BASE_FUNC_SINGLE,
77 5 storres
    SOLLYA_BASE_FUNC_SINH,
78 5 storres
    SOLLYA_BASE_FUNC_SQRT,
79 5 storres
    SOLLYA_BASE_FUNC_SUB,
80 5 storres
    SOLLYA_BASE_FUNC_TAN,
81 5 storres
    SOLLYA_BASE_FUNC_TANH,
82 5 storres
SOLLYA_BASE_FUNC_TRIPLEDOUBLE) = map(int,xrange(44))
83 56 storres
print "\nSuperficial pobyso check..."
84 5 storres
print "First constant - SOLLYA_BASE_FUNC_ABS: ", SOLLYA_BASE_FUNC_ABS
85 5 storres
print "Last constant  - SOLLYA_BASE_FUNC_TRIPLEDOUBLE: ", SOLLYA_BASE_FUNC_TRIPLEDOUBLE
86 5 storres
87 5 storres
pobyso_max_arity = 9
88 5 storres
89 85 storres
def pobyso_absolute_so_so():
90 85 storres
    return(sollya_lib_absolute(None))
91 85 storres
92 5 storres
def pobyso_autoprint(arg):
93 218 storres
    sollya_lib_autoprint(arg, None)
94 5 storres
95 38 storres
def pobyso_autoprint_so_so(arg):
96 38 storres
    sollya_lib_autoprint(arg,None)
97 54 storres
98 84 storres
def pobyso_bounds_to_range_sa_so(rnLowerBoundSa, rnUpperBoundSa, \
99 84 storres
                                 precisionSa=None):
100 84 storres
    """
101 84 storres
    Return a Sollya range from to 2 RealField Sage elements.
102 84 storres
    The Sollya range element has a sufficient precision to hold all
103 162 storres
    the digits of the widest of the Sage bounds.
104 84 storres
    """
105 84 storres
    # Sanity check.
106 84 storres
    if rnLowerBoundSa > rnUpperBoundSa:
107 84 storres
        return None
108 115 storres
    # Precision stuff.
109 85 storres
    if precisionSa is None:
110 84 storres
        # Check for the largest precision.
111 84 storres
        lbPrecSa = rnLowerBoundSa.parent().precision()
112 84 storres
        ubPrecSa = rnLowerBoundSa.parent().precision()
113 84 storres
        maxPrecSa = max(lbPrecSa, ubPrecSa)
114 84 storres
    else:
115 84 storres
        maxPrecSa = precisionSa
116 115 storres
    # From Sage to Sollya bounds.
117 162 storres
#    lowerBoundSo = sollya_lib_constant(get_rn_value(rnLowerBoundSa),
118 162 storres
#                                       maxPrecSa)
119 162 storres
    lowerBoundSo = pobyso_constant_sa_so(rnLowerBoundSa,
120 162 storres
                                         maxPrecSa)
121 162 storres
    upperBoundSo = pobyso_constant_sa_so(rnUpperBoundSa,
122 154 storres
                                       maxPrecSa)
123 162 storres
124 115 storres
    # From Sollya bounds to range.
125 84 storres
    rangeSo = sollya_lib_range(lowerBoundSo, upperBoundSo)
126 84 storres
    # Back to original precision.
127 84 storres
    # Clean up
128 84 storres
    sollya_lib_clear_obj(lowerBoundSo)
129 84 storres
    sollya_lib_clear_obj(upperBoundSo)
130 154 storres
    return rangeSo
131 84 storres
# End pobyso_bounds_to_range_sa_so
132 84 storres
133 215 storres
def pobyso_build_end_elliptic_list_so_so(*args):
134 215 storres
    """
135 215 storres
    From argumrny Sollya objects, create a Sollya end elliptic list.
136 215 storres
    Elements of the list are "eaten" (should not be cleared individualy,
137 215 storres
    are cleared when the list is cleared).
138 215 storres
    """
139 215 storres
    if len(args) == 0:
140 215 storres
        ## Called with an empty list produced "error".
141 215 storres
        return sollya_lib_build_end_elliptic_list(None)
142 215 storres
    index = 0
143 215 storres
    ## One can not append elements to an elliptic list, prepend only is
144 215 storres
    #  permitted.
145 215 storres
    for argument in reversed(args):
146 215 storres
        if index == 0:
147 215 storres
            listSo = sollya_lib_build_end_elliptic_list(argument, None)
148 215 storres
        else:
149 215 storres
            listSo = sollya_lib_prepend(argument, listSo)
150 215 storres
        index += 1
151 215 storres
    return listSo
152 215 storres
153 215 storres
# End pobyso_build_end_elliptic_list_so_so
154 215 storres
155 54 storres
def pobyso_build_function_sub_so_so(exp1So, exp2So):
156 54 storres
    return(sollya_lib_build_function_sub(exp1So, exp2So))
157 54 storres
158 85 storres
def pobyso_change_var_in_function_so_so(funcSo, chvarExpSo):
159 54 storres
    """
160 85 storres
    Variable change in a function.
161 85 storres
    """
162 85 storres
    return(sollya_lib_evaluate(funcSo,chvarExpSo))
163 85 storres
# End pobyso_change_var_in_function_so_so
164 85 storres
165 85 storres
def pobyso_chebyshevform_so_so(functionSo, degreeSo, intervalSo):
166 85 storres
    resultSo = sollya_lib_chebyshevform(functionSo, degreeSo, intervalSo)
167 85 storres
    return(resultSo)
168 85 storres
# End pobyso_chebyshevform_so_so.
169 85 storres
170 117 storres
def pobyso_clear_taylorform_sa_so(taylorFormSaSo):
171 117 storres
    """
172 117 storres
    This method is necessary to correctly clean up the memory from Taylor forms.
173 117 storres
    These are made of a Sollya object, a Sollya object list, a Sollya object.
174 117 storres
    For no clearly understood reason, sollya_lib_clear_object_list crashed
175 117 storres
    when applied to the object list.
176 117 storres
    Here, we decompose it into Sage list of Sollya objects references and we
177 117 storres
     clear them one by one.
178 117 storres
    """
179 117 storres
    sollya_lib_clear_obj(taylorFormSaSo[0])
180 117 storres
    (coefficientsErrorsListSaSo, numElementsSa, isEndEllipticSa) = \
181 117 storres
        pobyso_get_list_elements_so_so(taylorFormSaSo[1])
182 117 storres
    for element in coefficientsErrorsListSaSo:
183 117 storres
        sollya_lib_clear_obj(element)
184 117 storres
    sollya_lib_clear_obj(taylorFormSaSo[1])
185 117 storres
    sollya_lib_clear_obj(taylorFormSaSo[2])
186 117 storres
# End pobyso_clear_taylorform_sa_so
187 117 storres
188 85 storres
def pobyso_cmp(rnArgSa, cteSo):
189 85 storres
    """
190 54 storres
    Compare the MPFR value a RealNumber with that of a Sollya constant.
191 54 storres

192 54 storres
    Get the value of the Sollya constant into a RealNumber and compare
193 54 storres
    using MPFR. Could be optimized by working directly with a mpfr_t
194 54 storres
    for the intermediate number.
195 54 storres
    """
196 115 storres
    # Get the precision of the Sollya constant to build a Sage RealNumber
197 115 storres
    # with enough precision.to hold it.
198 5 storres
    precisionOfCte = c_int(0)
199 5 storres
    # From the Sollya constant, create a local Sage RealNumber.
200 85 storres
    sollya_lib_get_prec_of_constant(precisionOfCte, cteSo)
201 5 storres
    #print "Precision of constant: ", precisionOfCte
202 5 storres
    RRRR = RealField(precisionOfCte.value)
203 85 storres
    rnLocalSa = RRRR(0)
204 85 storres
    sollya_lib_get_constant(get_rn_value(rnLocalSa), cteSo)
205 115 storres
    #
206 115 storres
    ## Compare the Sage RealNumber version of the Sollya constant with rnArg.
207 85 storres
    return(cmp_rn_value(rnArgSa, rnLocal))
208 83 storres
# End pobyso_smp
209 5 storres
210 54 storres
def pobyso_compute_pos_function_abs_val_bounds_sa_sa(funcSa, lowerBoundSa, \
211 54 storres
                                                     upperBoundSa):
212 54 storres
    """
213 85 storres
    TODO: completely rework and test.
214 54 storres
    """
215 85 storres
    pobyso = pobyso_name_free_variable_sa_so(funcSa.variables()[0])
216 159 storres
    funcSo = pobyso_parse_string(funcSa._assume_str().replace('_SAGE_VAR_', ''))
217 54 storres
    rangeSo = pobyso_range_sa_so(lowerBoundSa, upperBoundSa)
218 54 storres
    infnormSo = pobyso_infnorm_so_so(funcSo,rangeSo)
219 83 storres
    # Sollya return the infnorm as an interval.
220 54 storres
    fMaxSa = pobyso_get_interval_from_range_so_sa(infnormSo)
221 54 storres
    # Get the top bound and compute the binade top limit.
222 54 storres
    fMaxUpperBoundSa = fMaxSa.upper()
223 54 storres
    binadeTopLimitSa = 2**ceil(fMaxUpperBoundSa.log2())
224 54 storres
    # Put up together the function to use to compute the lower bound.
225 54 storres
    funcAuxSo = pobyso_parse_string(str(binadeTopLimitSa) +  \
226 159 storres
                                    '-(' + f._assume_str().replace('_SAGE_VAR_', '') + ')')
227 54 storres
    pobyso_autoprint(funcAuxSo)
228 83 storres
    # Clear the Sollya range before a new call to infnorm and issue the call.
229 54 storres
    sollya_lib_clear_obj(infnormSo)
230 54 storres
    infnormSo = pobyso_infnorm_so_so(funcAuxSo,rangeSo)
231 54 storres
    fMinSa = pobyso_get_interval_from_range_so_sa(infnormSo)
232 54 storres
    sollya_lib_clear_obj(infnormSo)
233 85 storres
    fMinLowerBoundSa = binadeTopLimitSa - fMinSa.lower()
234 54 storres
    # Compute the maximum of the precisions of the different bounds.
235 54 storres
    maxPrecSa = max([fMinLowerBoundSa.parent().precision(), \
236 54 storres
                     fMaxUpperBoundSa.parent().precision()])
237 54 storres
    # Create a RealIntervalField and create an interval with the "good" bounds.
238 54 storres
    RRRI = RealIntervalField(maxPrecSa)
239 54 storres
    imageIntervalSa = RRRI(fMinLowerBoundSa, fMaxUpperBoundSa)
240 83 storres
    # Free the unneeded Sollya objects
241 54 storres
    sollya_lib_clear_obj(funcSo)
242 54 storres
    sollya_lib_clear_obj(funcAuxSo)
243 54 storres
    sollya_lib_clear_obj(rangeSo)
244 54 storres
    return(imageIntervalSa)
245 83 storres
# End pobyso_compute_pos_function_abs_val_bounds_sa_sa
246 54 storres
247 215 storres
def pobyso_compute_precision_decay_ratio_function_sa_so():
248 215 storres
    """
249 215 storres
    Compute the precision decay ratio function for polynomial
250 215 storres
    coefficient progressive trucation.
251 215 storres
    """
252 215 storres
    functionText = """
253 215 storres
    proc(deg, a, b, we, wq)
254 215 storres
    {
255 215 storres
      k = we * (exp(x/a)-1) + wq * (b*x)^2 + (1-we-wq) * x;
256 215 storres
      return k/k(d);
257 215 storres
    };
258 215 storres
    """
259 215 storres
    return pobyso_parse_string_sa_so(functionText)
260 215 storres
# End  pobyso_compute_precision_decay_ratio_function.
261 215 storres
262 215 storres
263 5 storres
def pobyso_constant(rnArg):
264 38 storres
    """ Legacy function. See pobyso_constant_sa_so. """
265 38 storres
    return(pobyso_constant_sa_so(rnArg))
266 5 storres
267 84 storres
def pobyso_constant_sa_so(rnArgSa, precisionSa=None):
268 52 storres
    """
269 115 storres
    Create a Sollya constant from a Sage RealNumber.
270 209 storres
    The sollya_lib_constant() function creates a constant
271 209 storres
    with the same precision as the source.
272 52 storres
    """
273 209 storres
    ## Precision stuff. If one wants to change precisions,
274 209 storres
    #  everything takes place in Sage. That only makes
275 209 storres
    #  sense if one wants to reduce the precision.
276 226 storres
    # TODO: revisit precision stuff with new technique.
277 209 storres
    if not precisionSa is None:
278 209 storres
        RRR = RealField(precisionSa)
279 209 storres
        rnArgSa = RRR(rnArgSa)
280 209 storres
    #print rnArgSa, rnArgSa.precision()
281 115 storres
    # Sollya constant creation takes place here.
282 209 storres
    return sollya_lib_constant(get_rn_value(rnArgSa))
283 115 storres
# End pobyso_constant_sa_so
284 115 storres
285 55 storres
def pobyso_constant_0_sa_so():
286 115 storres
    """
287 115 storres
    Obvious.
288 115 storres
    """
289 215 storres
    return pobyso_constant_from_int_sa_so(0)
290 55 storres
291 5 storres
def pobyso_constant_1():
292 115 storres
    """
293 115 storres
    Obvious.
294 115 storres
    Legacy function. See pobyso_constant_so_so.
295 115 storres
    """
296 215 storres
    return pobyso_constant_1_sa_so()
297 5 storres
298 52 storres
def pobyso_constant_1_sa_so():
299 115 storres
    """
300 115 storres
    Obvious.
301 115 storres
    """
302 38 storres
    return(pobyso_constant_from_int_sa_so(1))
303 38 storres
304 5 storres
def pobyso_constant_from_int(anInt):
305 38 storres
    """ Legacy function. See pobyso_constant_from_int_sa_so. """
306 215 storres
    return pobyso_constant_from_int_sa_so(anInt)
307 38 storres
308 38 storres
def pobyso_constant_from_int_sa_so(anInt):
309 115 storres
    """
310 115 storres
    Get a Sollya constant from a Sage int.
311 115 storres
    """
312 215 storres
    return sollya_lib_constant_from_int64(long(anInt))
313 5 storres
314 84 storres
def pobyso_constant_from_int_so_sa(constSo):
315 84 storres
    """
316 117 storres
    Get a Sage int from a Sollya int constant.
317 115 storres
    Usefull for precision or powers in polynomials.
318 84 storres
    """
319 200 storres
    constSa = c_long(0)
320 200 storres
    sollya_lib_get_constant_as_int64(byref(constSa), constSo)
321 215 storres
    return constSa.value
322 84 storres
# End pobyso_constant_from_int_so_sa
323 84 storres
324 209 storres
def pobyso_constant_from_mpq_sa_so(rationalSa):
325 200 storres
    """
326 200 storres
    Make a Sollya constant from Sage rational.
327 209 storres
    The Sollya constant is an unevaluated expression.
328 209 storres
    Hence no precision argument is needed.
329 209 storres
    It is better to leave this way since Sollya has its own
330 209 storres
    optimized evaluation mecanism that tries very hard to
331 209 storres
    return exact values or at least faithful ones.
332 200 storres
    """
333 200 storres
    ratExprSo = \
334 200 storres
        sollya_lib_constant_from_mpq(sgmp_get_rational_value(rationalSa))
335 209 storres
    return ratExprSo
336 200 storres
# End pobyso_constant_from_mpq_sa_so.
337 200 storres
338 209 storres
def pobyso_constant_sollya_prec_sa_so(rnArgSa):
339 209 storres
    """
340 209 storres
    Create a Sollya constant from a Sage RealNumber at the
341 209 storres
    current precision in Sollya.
342 209 storres
    """
343 209 storres
    currentSollyaPrecSa = pobyso_get_prec_so_sa()
344 209 storres
    return pobyso_constant_sa_so(rnArgSa, currentSollyaPrecSa)
345 209 storres
# End pobyso_constant_sollya_prec_sa_so
346 215 storres
347 215 storres
def pobyso_end_elliptic_list_so_sa_so(objectsListSo, intCountSa):
348 215 storres
    """
349 215 storres
    Create a Sollya end elliptic list made of the objectListSo[0] to
350 215 storres
     objectsListSo[intCountSa-1] objects.
351 215 storres
    """
352 215 storres
    return sollya_lib_end_elliptic_list(objectSo, int(intCountSa))
353 215 storres
354 155 storres
def pobyso_error_so():
355 155 storres
    return sollya_lib_error(None)
356 155 storres
# End pobyso_error().
357 155 storres
358 215 storres
def pobyso_evaluate_so_so(funcSo, argumentSo):
359 215 storres
    """
360 215 storres
    Evaluates funcSo for arguemntSo through sollya_lib_evaluate().
361 215 storres
    """
362 215 storres
    return sollya_lib_evaluate(funcSo, argumentSo)
363 215 storres
# End pobyso_evaluate_so_so.
364 215 storres
365 209 storres
def pobyso_float_poly_sa_so(polySa, precSa = None):
366 209 storres
    """
367 209 storres
    Create a Sollya polynomial from a Sage RealField polynomial.
368 209 storres
    """
369 209 storres
    ## TODO: filter arguments.
370 209 storres
    ## Precision. If a precision is given, convert the polynomial
371 209 storres
    #  into the right polynomial field. If not convert it straight
372 209 storres
    #  to Sollya.
373 218 storres
    sollyaPrecChanged = False
374 226 storres
    (initialSollyaPrecSo, initialSollyaPrecSa) = pobyso_get_prec_so_so_sa()
375 218 storres
    if precSa is None:
376 209 storres
        precSa = polySa.parent().base_ring().precision()
377 226 storres
    if (precSa > initialSollyaPrecSa):
378 226 storres
        if precSa <= 2:
379 226 storres
            print inspect.stack()[0][3], ": precision change <= 2 requested"
380 218 storres
        precSo = pobyso_constant_from_int(precSa)
381 218 storres
        pobyso_set_prec_so_so(precSo)
382 218 storres
        sollya_lib_clear_obj(precSo)
383 218 storres
        sollyaPrecChanged = True
384 209 storres
    ## Get exponents and coefficients.
385 218 storres
    exponentsSa     = polySa.exponents()
386 218 storres
    coefficientsSa  = polySa.coefficients()
387 209 storres
    ## Build the polynomial.
388 209 storres
    polySo = None
389 213 storres
    for coefficientSa, exponentSa in zip(coefficientsSa, exponentsSa):
390 209 storres
        #print coefficientSa.n(prec=precSa), exponentSa
391 209 storres
        coefficientSo = \
392 209 storres
            pobyso_constant_sa_so(coefficientSa)
393 209 storres
        #pobyso_autoprint(coefficientSo)
394 209 storres
        exponentSo = \
395 209 storres
            pobyso_constant_from_int_sa_so(exponentSa)
396 209 storres
        #pobyso_autoprint(exponentSo)
397 209 storres
        monomialSo = sollya_lib_build_function_pow(
398 209 storres
                       sollya_lib_build_function_free_variable(),
399 209 storres
                       exponentSo)
400 218 storres
        polyTermSo = sollya_lib_build_function_mul(coefficientSo,
401 218 storres
                                                       monomialSo)
402 209 storres
        if polySo is None:
403 218 storres
            polySo = polyTermSo
404 209 storres
        else:
405 209 storres
            polySo = sollya_lib_build_function_add(polySo, polyTermSo)
406 218 storres
    if sollyaPrecChanged:
407 226 storres
        pobyso_set_prec_so_so(initialSollyaPrecSo)
408 226 storres
        sollya_lib_clear_obj(initialSollyaPrecSo)
409 209 storres
    return polySo
410 209 storres
# End pobyso_float_poly_sa_so
411 209 storres
412 209 storres
def pobyso_float_poly_so_sa(polySo, realFieldSa=None):
413 209 storres
    """
414 209 storres
    Convert a Sollya polynomial into a Sage floating-point polynomial.
415 209 storres
    If no realField is given, a RealField corresponding to the maximum
416 209 storres
    precision of the coefficients is internally computed.
417 209 storres
    The real field is not returned but can be easily retrieved from
418 209 storres
    the polynomial itself.
419 209 storres
    ALGORITHM:
420 209 storres
    - (optional) compute the RealField of the coefficients;
421 209 storres
    - convert the Sollya expression into a Sage expression;
422 209 storres
    - convert the Sage expression into a Sage polynomial
423 209 storres
    """
424 209 storres
    if realFieldSa is None:
425 209 storres
        expressionPrecSa = pobyso_get_max_prec_of_exp_so_sa(polySo)
426 218 storres
        #print "Maximum precision of Sollya polynomial coefficients:", expressionPrecSa
427 226 storres
        if expressionPrecSa < 2 or expressionPrecSa > 2147483391:
428 226 storres
            print "Maximum degree of expression:", expressionPrecSa
429 209 storres
        realFieldSa      = RealField(expressionPrecSa)
430 209 storres
    #print "Sollya expression before...",
431 209 storres
    #pobyso_autoprint(polySo)
432 209 storres
433 209 storres
    expressionSa = pobyso_get_sage_exp_from_sollya_exp_so_sa(polySo,
434 209 storres
                                                             realFieldSa)
435 218 storres
    #print "...Sollya expression after."
436 209 storres
    #pobyso_autoprint(polySo)
437 209 storres
    polyVariableSa = expressionSa.variables()[0]
438 209 storres
    polyRingSa     = realFieldSa[str(polyVariableSa)]
439 209 storres
    #print polyRingSa
440 209 storres
    # Do not use the polynomial(expressionSa, ring=polyRingSa) form!
441 209 storres
    polynomialSa = polyRingSa(expressionSa)
442 215 storres
    polyCoeffsListSa = polynomialSa.coefficients()
443 215 storres
    #for coeff in polyCoeffsListSa:
444 215 storres
    #    print coeff.abs().n()
445 209 storres
    return polynomialSa
446 209 storres
# End pobyso_float_poly_so_sa
447 209 storres
448 215 storres
def pobyso_free_variable():
449 215 storres
    """
450 215 storres
    Ultra thin wrapper around the sollya_lib_function_build_free_variable function.
451 215 storres
    """
452 215 storres
    return sollya_lib_build_function_free_variable()
453 209 storres
454 5 storres
def pobyso_function_type_as_string(funcType):
455 38 storres
    """ Legacy function. See pobyso_function_type_as_string_so_sa. """
456 38 storres
    return(pobyso_function_type_as_string_so_sa(funcType))
457 38 storres
458 38 storres
def pobyso_function_type_as_string_so_sa(funcType):
459 38 storres
    """
460 38 storres
    Numeric Sollya function codes -> Sage mathematical function names.
461 38 storres
    Notice that pow -> ^ (a la Sage, not a la Python).
462 38 storres
    """
463 5 storres
    if funcType == SOLLYA_BASE_FUNC_ABS:
464 5 storres
        return "abs"
465 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ACOS:
466 5 storres
        return "arccos"
467 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ACOSH:
468 5 storres
        return "arccosh"
469 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ADD:
470 5 storres
        return "+"
471 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ASIN:
472 5 storres
        return "arcsin"
473 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ASINH:
474 5 storres
        return "arcsinh"
475 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ATAN:
476 5 storres
        return "arctan"
477 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ATANH:
478 5 storres
        return "arctanh"
479 5 storres
    elif funcType == SOLLYA_BASE_FUNC_CEIL:
480 5 storres
        return "ceil"
481 5 storres
    elif funcType == SOLLYA_BASE_FUNC_CONSTANT:
482 5 storres
        return "cte"
483 5 storres
    elif funcType == SOLLYA_BASE_FUNC_COS:
484 5 storres
        return "cos"
485 5 storres
    elif funcType == SOLLYA_BASE_FUNC_COSH:
486 5 storres
        return "cosh"
487 5 storres
    elif funcType == SOLLYA_BASE_FUNC_DIV:
488 5 storres
        return "/"
489 5 storres
    elif funcType == SOLLYA_BASE_FUNC_DOUBLE:
490 5 storres
        return "double"
491 5 storres
    elif funcType == SOLLYA_BASE_FUNC_DOUBLEDOUBLE:
492 5 storres
        return "doubleDouble"
493 5 storres
    elif funcType == SOLLYA_BASE_FUNC_DOUBLEEXTENDED:
494 5 storres
        return "doubleDxtended"
495 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ERF:
496 5 storres
        return "erf"
497 5 storres
    elif funcType == SOLLYA_BASE_FUNC_ERFC:
498 5 storres
        return "erfc"
499 5 storres
    elif funcType == SOLLYA_BASE_FUNC_EXP:
500 5 storres
        return "exp"
501 5 storres
    elif funcType == SOLLYA_BASE_FUNC_EXP_M1:
502 5 storres
        return "expm1"
503 5 storres
    elif funcType == SOLLYA_BASE_FUNC_FLOOR:
504 5 storres
        return "floor"
505 5 storres
    elif funcType == SOLLYA_BASE_FUNC_FREE_VARIABLE:
506 5 storres
        return "freeVariable"
507 5 storres
    elif funcType == SOLLYA_BASE_FUNC_HALFPRECISION:
508 5 storres
        return "halfPrecision"
509 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LIBRARYCONSTANT:
510 5 storres
        return "libraryConstant"
511 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LIBRARYFUNCTION:
512 5 storres
        return "libraryFunction"
513 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LOG:
514 5 storres
        return "log"
515 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LOG_10:
516 5 storres
        return "log10"
517 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LOG_1P:
518 5 storres
        return "log1p"
519 5 storres
    elif funcType == SOLLYA_BASE_FUNC_LOG_2:
520 5 storres
        return "log2"
521 5 storres
    elif funcType == SOLLYA_BASE_FUNC_MUL:
522 5 storres
        return "*"
523 5 storres
    elif funcType == SOLLYA_BASE_FUNC_NEARESTINT:
524 5 storres
        return "round"
525 5 storres
    elif funcType == SOLLYA_BASE_FUNC_NEG:
526 5 storres
        return "__neg__"
527 5 storres
    elif funcType == SOLLYA_BASE_FUNC_PI:
528 5 storres
        return "pi"
529 5 storres
    elif funcType == SOLLYA_BASE_FUNC_POW:
530 5 storres
        return "^"
531 5 storres
    elif funcType == SOLLYA_BASE_FUNC_PROCEDUREFUNCTION:
532 5 storres
        return "procedureFunction"
533 5 storres
    elif funcType == SOLLYA_BASE_FUNC_QUAD:
534 5 storres
        return "quad"
535 5 storres
    elif funcType == SOLLYA_BASE_FUNC_SIN:
536 5 storres
        return "sin"
537 5 storres
    elif funcType == SOLLYA_BASE_FUNC_SINGLE:
538 5 storres
        return "single"
539 5 storres
    elif funcType == SOLLYA_BASE_FUNC_SINH:
540 5 storres
        return "sinh"
541 5 storres
    elif funcType == SOLLYA_BASE_FUNC_SQRT:
542 5 storres
        return "sqrt"
543 5 storres
    elif funcType == SOLLYA_BASE_FUNC_SUB:
544 5 storres
        return "-"
545 5 storres
    elif funcType == SOLLYA_BASE_FUNC_TAN:
546 5 storres
        return "tan"
547 5 storres
    elif funcType == SOLLYA_BASE_FUNC_TANH:
548 5 storres
        return "tanh"
549 5 storres
    elif funcType == SOLLYA_BASE_FUNC_TRIPLEDOUBLE:
550 5 storres
        return "tripleDouble"
551 5 storres
    else:
552 5 storres
        return None
553 5 storres
554 85 storres
def pobyso_get_constant(rnArgSa, constSo):
555 38 storres
    """ Legacy function. See pobyso_get_constant_so_sa. """
556 209 storres
    return pobyso_get_constant_so_sa(rnArgSa, constSo)
557 209 storres
# End pobyso_get_constant
558 38 storres
559 84 storres
def pobyso_get_constant_so_sa(rnArgSa, constSo):
560 52 storres
    """
561 85 storres
    Set the value of rnArgSo to the value of constSo in MPFR_RNDN mode.
562 52 storres
    rnArg must already exist and belong to some RealField.
563 85 storres
    We assume that constSo points to a Sollya constant.
564 52 storres
    """
565 209 storres
    outcome = sollya_lib_get_constant(get_rn_value(rnArgSa), constSo)
566 209 storres
    if outcome == 0: # Failure because constSo is not a constant expression.
567 209 storres
        return None
568 209 storres
    else:
569 209 storres
        return outcome
570 209 storres
# End  pobyso_get_constant_so_sa
571 209 storres
572 57 storres
def pobyso_get_constant_as_rn(ctExpSo):
573 83 storres
    """
574 83 storres
    Legacy function. See pobyso_get_constant_as_rn_so_sa.
575 83 storres
    """
576 57 storres
    return(pobyso_get_constant_as_rn_so_sa(ctExpSo))
577 38 storres
578 56 storres
def pobyso_get_constant_as_rn_so_sa(constExpSo):
579 83 storres
    """
580 83 storres
    Get a Sollya constant as a Sage "real number".
581 83 storres
    The precision of the floating-point number returned is that of the Sollya
582 83 storres
    constant.
583 83 storres
    """
584 218 storres
    #print "Before computing precision of variable..."
585 218 storres
    #pobyso_autoprint(constExpSo)
586 209 storres
    precisionSa  = pobyso_get_prec_of_constant_so_sa(constExpSo)
587 218 storres
    #print "precisionSa:", precisionSa
588 209 storres
    ## If the expression can not be exactly converted, None is returned.
589 209 storres
    #  In this case opt for the Sollya current expression.
590 209 storres
    if precisionSa is None:
591 209 storres
        precisionSa = pobyso_get_prec_so_sa()
592 56 storres
    RRRR = RealField(precisionSa)
593 56 storres
    rnSa = RRRR(0)
594 209 storres
    outcome = sollya_lib_get_constant(get_rn_value(rnSa), constExpSo)
595 209 storres
    if outcome == 0:
596 209 storres
        return None
597 209 storres
    else:
598 209 storres
        return rnSa
599 83 storres
# End pobyso_get_constant_as_rn_so_sa
600 38 storres
601 38 storres
def pobyso_get_constant_as_rn_with_rf(ctExp, realField):
602 83 storres
    """
603 83 storres
    Legacy function. See pobyso_get_constant_as_rn_with_rf_so_sa.
604 83 storres
    """
605 209 storres
    return pobyso_get_constant_as_rn_with_rf_so_sa(ctExp, realField)
606 209 storres
# End pobyso_get_constant_as_rn_with_rf
607 5 storres
608 56 storres
def pobyso_get_constant_as_rn_with_rf_so_sa(ctExpSo, realFieldSa = None):
609 83 storres
    """
610 83 storres
    Get a Sollya constant as a Sage "real number".
611 83 storres
    If no real field is specified, the precision of the floating-point number
612 85 storres
    returned is that of the Sollya constant.
613 83 storres
    Otherwise is is that of the real field. Hence rounding may happen.
614 83 storres
    """
615 56 storres
    if realFieldSa is None:
616 209 storres
        return pobyso_get_constant_as_rn_so_sa(ctExpSo)
617 56 storres
    rnSa = realFieldSa(0)
618 209 storres
    outcome = sollya_lib_get_constant(get_rn_value(rnSa), ctExpSo)
619 209 storres
    if outcome == 0:
620 209 storres
        return None
621 209 storres
    else:
622 209 storres
        return rnSa
623 83 storres
# End pobyso_get_constant_as_rn_with_rf_so_sa
624 38 storres
625 5 storres
def pobyso_get_free_variable_name():
626 83 storres
    """
627 83 storres
    Legacy function. See pobyso_get_free_variable_name_so_sa.
628 83 storres
    """
629 38 storres
    return(pobyso_get_free_variable_name_so_sa())
630 38 storres
631 38 storres
def pobyso_get_free_variable_name_so_sa():
632 209 storres
    return sollya_lib_get_free_variable_name()
633 5 storres
634 38 storres
def pobyso_get_function_arity(expressionSo):
635 83 storres
    """
636 83 storres
    Legacy function. See pobyso_get_function_arity_so_sa.
637 83 storres
    """
638 38 storres
    return(pobyso_get_function_arity_so_sa(expressionSo))
639 38 storres
640 38 storres
def pobyso_get_function_arity_so_sa(expressionSo):
641 5 storres
    arity = c_int(0)
642 38 storres
    sollya_lib_get_function_arity(byref(arity),expressionSo)
643 209 storres
    return int(arity.value)
644 5 storres
645 38 storres
def pobyso_get_head_function(expressionSo):
646 83 storres
    """
647 83 storres
    Legacy function. See pobyso_get_head_function_so_sa.
648 83 storres
    """
649 38 storres
    return(pobyso_get_head_function_so_sa(expressionSo))
650 38 storres
651 38 storres
def pobyso_get_head_function_so_sa(expressionSo):
652 5 storres
    functionType = c_int(0)
653 218 storres
    sollya_lib_get_head_function(byref(functionType), expressionSo)
654 209 storres
    return int(functionType.value)
655 5 storres
656 56 storres
def pobyso_get_interval_from_range_so_sa(soRange, realIntervalFieldSa = None ):
657 53 storres
    """
658 53 storres
    Return the Sage interval corresponding to the Sollya range argument.
659 83 storres
    If no reaIntervalField is passed as an argument, the interval bounds are not
660 56 storres
    rounded: they are elements of RealIntervalField of the "right" precision
661 56 storres
    to hold all the digits.
662 53 storres
    """
663 53 storres
    prec = c_int(0)
664 56 storres
    if realIntervalFieldSa is None:
665 56 storres
        retval = sollya_lib_get_prec_of_range(byref(prec), soRange, None)
666 56 storres
        if retval == 0:
667 209 storres
            return None
668 56 storres
        realIntervalFieldSa = RealIntervalField(prec.value)
669 56 storres
    intervalSa = realIntervalFieldSa(0,0)
670 53 storres
    retval = \
671 53 storres
        sollya_lib_get_interval_from_range(get_interval_value(intervalSa),\
672 53 storres
                                           soRange)
673 53 storres
    if retval == 0:
674 209 storres
        return None
675 209 storres
    return intervalSa
676 56 storres
# End pobyso_get_interval_from_range_so_sa
677 56 storres
678 5 storres
def pobyso_get_list_elements(soObj):
679 38 storres
    """ Legacy function. See pobyso_get_list_elements_so_so. """
680 209 storres
    return pobyso_get_list_elements_so_so(soObj)
681 38 storres
682 117 storres
def pobyso_get_list_elements_so_so(objectListSo):
683 51 storres
    """
684 118 storres
    Get the Sollya list elements as a Sage/Python array of Sollya objects.
685 118 storres

686 118 storres
    INPUT:
687 118 storres
    - objectListSo: a Sollya list of Sollya objects.
688 118 storres

689 118 storres
    OUTPUT:
690 118 storres
    - a Sage/Python tuple made of:
691 118 storres
      - a Sage/Python list of Sollya objects,
692 118 storres
      - a Sage/Python int holding the number of elements,
693 118 storres
      - a Sage/Python int stating (!= 0) that the list is end-elliptic.
694 118 storres
    NOTE::
695 118 storres
        We recover the addresses of the Sollya object from the list of pointers
696 118 storres
        returned by sollya_lib_get_list_elements. The list itself is freed.
697 118 storres
    TODO::
698 118 storres
        Figure out what to do with numElements since the number of elements
699 118 storres
        can easily be recovered from the list itself.
700 118 storres
        Ditto for isEndElliptic.
701 51 storres
    """
702 5 storres
    listAddress = POINTER(c_longlong)()
703 5 storres
    numElements = c_int(0)
704 5 storres
    isEndElliptic = c_int(0)
705 117 storres
    listAsSageList = []
706 5 storres
    result = sollya_lib_get_list_elements(byref(listAddress),\
707 54 storres
                                          byref(numElements),\
708 54 storres
                                          byref(isEndElliptic),\
709 117 storres
                                          objectListSo)
710 5 storres
    if result == 0 :
711 5 storres
        return None
712 5 storres
    for i in xrange(0, numElements.value, 1):
713 118 storres
       #listAsSageList.append(sollya_lib_copy_obj(listAddress[i]))
714 118 storres
       listAsSageList.append(listAddress[i])
715 117 storres
       # Clear each of the elements returned by Sollya.
716 118 storres
       #sollya_lib_clear_obj(listAddress[i])
717 117 storres
    # Free the list itself.
718 117 storres
    sollya_lib_free(listAddress)
719 209 storres
    return (listAsSageList, numElements.value, isEndElliptic.value)
720 5 storres
721 38 storres
def pobyso_get_max_prec_of_exp(soExp):
722 38 storres
    """ Legacy function. See pobyso_get_max_prec_of_exp_so_sa. """
723 209 storres
    return pobyso_get_max_prec_of_exp_so_sa(soExp)
724 5 storres
725 85 storres
def pobyso_get_max_prec_of_exp_so_sa(expSo):
726 38 storres
    """
727 38 storres
    Get the maximum precision used for the numbers in a Sollya expression.
728 52 storres

729 52 storres
    Arguments:
730 52 storres
    soExp -- a Sollya expression pointer
731 52 storres
    Return value:
732 52 storres
    A Python integer
733 38 storres
    TODO:
734 38 storres
    - error management;
735 38 storres
    - correctly deal with numerical type such as DOUBLEEXTENDED.
736 38 storres
    """
737 226 storres
    if expSo is None:
738 226 storres
        print inspect.stack()[0][3], ": expSo is None."
739 226 storres
        return 0
740 5 storres
    maxPrecision = 0
741 52 storres
    minConstPrec = 0
742 52 storres
    currentConstPrec = 0
743 226 storres
    #pobyso_autoprint(expSo)
744 85 storres
    operator = pobyso_get_head_function_so_sa(expSo)
745 5 storres
    if (operator != SOLLYA_BASE_FUNC_CONSTANT) and \
746 5 storres
    (operator != SOLLYA_BASE_FUNC_FREE_VARIABLE):
747 85 storres
        (arity, subexpressions) = pobyso_get_subfunctions_so_sa(expSo)
748 5 storres
        for i in xrange(arity):
749 5 storres
            maxPrecisionCandidate = \
750 38 storres
                pobyso_get_max_prec_of_exp_so_sa(subexpressions[i])
751 5 storres
            if maxPrecisionCandidate > maxPrecision:
752 5 storres
                maxPrecision = maxPrecisionCandidate
753 209 storres
        return maxPrecision
754 5 storres
    elif operator == SOLLYA_BASE_FUNC_CONSTANT:
755 85 storres
        #minConstPrec = pobyso_get_min_prec_of_constant_so_sa(expSo)
756 52 storres
        #currentConstPrec = pobyso_get_min_prec_of_constant_so_sa(soExp)
757 52 storres
        #print minConstPrec, " - ", currentConstPrec
758 209 storres
        return pobyso_get_min_prec_of_constant_so_sa(expSo)
759 52 storres
760 5 storres
    elif operator == SOLLYA_BASE_FUNC_FREE_VARIABLE:
761 209 storres
        return 0
762 5 storres
    else:
763 38 storres
        print "pobyso_get_max_prec_of_exp_so_sa: unexepected operator."
764 209 storres
        return 0
765 5 storres
766 85 storres
def pobyso_get_min_prec_of_constant_so_sa(constExpSo):
767 52 storres
    """
768 52 storres
    Get the minimum precision necessary to represent the value of a Sollya
769 52 storres
    constant.
770 52 storres
    MPFR_MIN_PREC and powers of 2 are taken into account.
771 209 storres
    We assume that constExpSo is a pointer to a Sollay constant expression.
772 52 storres
    """
773 85 storres
    constExpAsRnSa = pobyso_get_constant_as_rn_so_sa(constExpSo)
774 85 storres
    return(min_mpfr_size(get_rn_value(constExpAsRnSa)))
775 52 storres
776 200 storres
def pobyso_get_poly_so_sa(polySo, realFieldSa=None):
777 200 storres
    """
778 200 storres
    Convert a Sollya polynomial into a Sage polynomial.
779 209 storres
    Legacy function. Use pobyso_float_poly_so_sa() instead.
780 200 storres
    """
781 213 storres
    return pobyso_float_poly_so_sa(polySo,realFieldSa)
782 200 storres
# End pobyso_get_poly_so_sa
783 200 storres
784 200 storres
def pobyso_get_prec():
785 200 storres
    """ Legacy function. See pobyso_get_prec_so_sa(). """
786 209 storres
    return pobyso_get_prec_so_sa()
787 200 storres
788 200 storres
def pobyso_get_prec_so():
789 200 storres
    """
790 200 storres
    Get the current default precision in Sollya.
791 200 storres
    The return value is a Sollya object.
792 200 storres
    Usefull when modifying the precision back and forth by avoiding
793 200 storres
    extra conversions.
794 200 storres
    """
795 209 storres
    return sollya_lib_get_prec(None)
796 200 storres
797 200 storres
def pobyso_get_prec_so_sa():
798 200 storres
    """
799 200 storres
    Get the current default precision in Sollya.
800 200 storres
    The return value is Sage/Python int.
801 200 storres
    """
802 227 storres
    precSo = sollya_lib_get_prec()
803 227 storres
    precSa = pobyso_constant_from_int_so_sa(precSo)
804 200 storres
    sollya_lib_clear_obj(precSo)
805 227 storres
    return precSa
806 200 storres
# End pobyso_get_prec_so_sa.
807 200 storres
808 209 storres
def pobyso_get_prec_so_so_sa():
809 209 storres
    """
810 209 storres
    Return the current precision both as a Sollya object and a
811 209 storres
    Sage integer as hybrid tuple.
812 209 storres
    To avoid multiple calls for precision manipulations.
813 209 storres
    """
814 227 storres
    precSo = sollya_lib_get_prec()
815 227 storres
    precSa = pobyso_constant_from_int_so_sa(precSo)
816 227 storres
    return (precSo, int(precSa))
817 200 storres
818 200 storres
def pobyso_get_prec_of_constant(ctExpSo):
819 200 storres
    """ Legacy function. See pobyso_get_prec_of_constant_so_sa. """
820 209 storres
    return pobyso_get_prec_of_constant_so_sa(ctExpSo)
821 200 storres
822 200 storres
def pobyso_get_prec_of_constant_so_sa(ctExpSo):
823 200 storres
    """
824 200 storres
    Tries to find a precision to represent ctExpSo without rounding.
825 200 storres
    If not possible, returns None.
826 200 storres
    """
827 218 storres
    #print "Entering pobyso_get_prec_of_constant_so_sa..."
828 200 storres
    prec = c_int(0)
829 200 storres
    retc = sollya_lib_get_prec_of_constant(byref(prec), ctExpSo, None)
830 200 storres
    if retc == 0:
831 218 storres
        #print "pobyso_get_prec_of_constant_so_sa failed."
832 209 storres
        return None
833 218 storres
    #print "...exiting pobyso_get_prec_of_constant_so_sa."
834 209 storres
    return int(prec.value)
835 200 storres
836 200 storres
def pobyso_get_prec_of_range_so_sa(rangeSo):
837 200 storres
    """
838 200 storres
    Returns the number of bits elements of a range are coded with.
839 200 storres
    """
840 200 storres
    prec = c_int(0)
841 200 storres
    retc = sollya_lib_get_prec_of_range(byref(prec), rangeSo, None)
842 200 storres
    if retc == 0:
843 200 storres
        return(None)
844 209 storres
    return int(prec.value)
845 200 storres
# End pobyso_get_prec_of_range_so_sa()
846 200 storres
847 85 storres
def pobyso_get_sage_exp_from_sollya_exp(sollyaExpSo, realField = RR):
848 38 storres
    """ Legacy function. See pobyso_get_sage_exp_from_sollya_exp_so_sa. """
849 209 storres
    return pobyso_get_sage_exp_from_sollya_exp_so_sa(sollyaExpSo,
850 209 storres
                                                     realField = RR)
851 38 storres
852 85 storres
def pobyso_get_sage_exp_from_sollya_exp_so_sa(sollyaExpSo, realFieldSa = RR):
853 5 storres
    """
854 38 storres
    Get a Sage expression from a Sollya expression.
855 38 storres
    Currently only tested with polynomials with floating-point coefficients.
856 5 storres
    Notice that, in the returned polynomial, the exponents are RealNumbers.
857 5 storres
    """
858 5 storres
    #pobyso_autoprint(sollyaExp)
859 85 storres
    operatorSa = pobyso_get_head_function_so_sa(sollyaExpSo)
860 83 storres
    sollyaLibFreeVariableName = sollya_lib_get_free_variable_name()
861 213 storres
    ## Get rid of the "_"'s in "_x_", if any.
862 213 storres
    sollyaLibFreeVariableName = re.sub('_', '', sollyaLibFreeVariableName)
863 5 storres
    # Constants and the free variable are special cases.
864 5 storres
    # All other operator are dealt with in the same way.
865 85 storres
    if (operatorSa != SOLLYA_BASE_FUNC_CONSTANT) and \
866 85 storres
       (operatorSa != SOLLYA_BASE_FUNC_FREE_VARIABLE):
867 85 storres
        (aritySa, subexpressionsSa) = pobyso_get_subfunctions_so_sa(sollyaExpSo)
868 85 storres
        if aritySa == 1:
869 85 storres
            sageExpSa = eval(pobyso_function_type_as_string_so_sa(operatorSa) + \
870 85 storres
            "(" + pobyso_get_sage_exp_from_sollya_exp_so_sa(subexpressionsSa[0], \
871 85 storres
            realFieldSa) + ")")
872 85 storres
        elif aritySa == 2:
873 63 storres
            # We do not get through the preprocessor.
874 63 storres
            # The "^" operator is then a special case.
875 85 storres
            if operatorSa == SOLLYA_BASE_FUNC_POW:
876 85 storres
                operatorAsStringSa = "**"
877 5 storres
            else:
878 85 storres
                operatorAsStringSa = \
879 85 storres
                    pobyso_function_type_as_string_so_sa(operatorSa)
880 85 storres
            sageExpSa = \
881 85 storres
              eval("pobyso_get_sage_exp_from_sollya_exp_so_sa(subexpressionsSa[0], realFieldSa)"\
882 85 storres
              + " " + operatorAsStringSa + " " + \
883 85 storres
                   "pobyso_get_sage_exp_from_sollya_exp_so_sa(subexpressionsSa[1], realFieldSa)")
884 63 storres
        # We do not know yet how to deal with arity >= 3
885 63 storres
        # (is there any in Sollya anyway?).
886 5 storres
        else:
887 85 storres
            sageExpSa = eval('None')
888 209 storres
        return sageExpSa
889 85 storres
    elif operatorSa == SOLLYA_BASE_FUNC_CONSTANT:
890 5 storres
        #print "This is a constant"
891 85 storres
        return pobyso_get_constant_as_rn_with_rf_so_sa(sollyaExpSo, realFieldSa)
892 85 storres
    elif operatorSa == SOLLYA_BASE_FUNC_FREE_VARIABLE:
893 218 storres
        #print "This is the free variable"
894 209 storres
        return eval(sollyaLibFreeVariableName)
895 5 storres
    else:
896 5 storres
        print "Unexpected"
897 5 storres
        return eval('None')
898 185 storres
# End pobyso_get_sage_exp_from_sollya_exp_so_sa
899 73 storres
900 185 storres
901 38 storres
def pobyso_get_subfunctions(expressionSo):
902 38 storres
    """ Legacy function. See pobyso_get_subfunctions_so_sa. """
903 209 storres
    return pobyso_get_subfunctions_so_sa(expressionSo)
904 200 storres
# End pobyso_get_subfunctions.
905 200 storres
906 38 storres
def pobyso_get_subfunctions_so_sa(expressionSo):
907 38 storres
    """
908 38 storres
    Get the subfunctions of an expression.
909 38 storres
    Return the number of subfunctions and the list of subfunctions addresses.
910 55 storres
    S.T.: Could not figure out another way than that ugly list of declarations
911 83 storres
    to recover the addresses of the subfunctions.
912 83 storres
    We limit ourselves to arity 8 functions.
913 38 storres
    """
914 5 storres
    subf0 = c_int(0)
915 5 storres
    subf1 = c_int(0)
916 5 storres
    subf2 = c_int(0)
917 5 storres
    subf3 = c_int(0)
918 5 storres
    subf4 = c_int(0)
919 5 storres
    subf5 = c_int(0)
920 5 storres
    subf6 = c_int(0)
921 5 storres
    subf7 = c_int(0)
922 5 storres
    subf8 = c_int(0)
923 5 storres
    arity = c_int(0)
924 5 storres
    nullPtr = POINTER(c_int)()
925 38 storres
    sollya_lib_get_subfunctions(expressionSo, byref(arity), \
926 83 storres
      byref(subf0), byref(subf1), byref(subf2), byref(subf3), \
927 83 storres
      byref(subf4), byref(subf5),\
928 83 storres
      byref(subf6), byref(subf7), byref(subf8), nullPtr, None)
929 83 storres
#    byref(cast(subfunctions[0], POINTER(c_int))), \
930 83 storres
#    byref(cast(subfunctions[0], POINTER(c_int))), \
931 83 storres
#    byref(cast(subfunctions[2], POINTER(c_int))), \
932 83 storres
#    byref(cast(subfunctions[3], POINTER(c_int))), \
933 83 storres
#    byref(cast(subfunctions[4], POINTER(c_int))), \
934 83 storres
#    byref(cast(subfunctions[5], POINTER(c_int))), \
935 83 storres
#    byref(cast(subfunctions[6], POINTER(c_int))), \
936 83 storres
#    byref(cast(subfunctions[7], POINTER(c_int))), \
937 5 storres
#    byref(cast(subfunctions[8], POINTER(c_int))), nullPtr)
938 83 storres
    subfunctions = [subf0, subf1, subf2, subf3, subf4, subf5, subf6, subf7, \
939 83 storres
                    subf8]
940 5 storres
    subs = []
941 5 storres
    if arity.value > pobyso_max_arity:
942 38 storres
        return(0,[])
943 5 storres
    for i in xrange(arity.value):
944 5 storres
        subs.append(int(subfunctions[i].value))
945 5 storres
        #print subs[i]
946 209 storres
    return (int(arity.value), subs)
947 200 storres
# End pobyso_get_subfunctions_so_sa
948 5 storres
949 155 storres
def pobyso_guess_degree_sa_sa(functionSa, intervalSa, approxErrorSa,
950 155 storres
                              weightSa=None, degreeBoundSa=None):
951 155 storres
    """
952 155 storres
    Sa_sa variant of the solly_guessdegree function.
953 155 storres
    Return 0 if something goes wrong.
954 155 storres
    """
955 159 storres
    functionAsStringSa = functionSa._assume_str().replace('_SAGE_VAR_', '')
956 154 storres
    functionSo = pobyso_parse_string_sa_so(functionAsStringSa)
957 155 storres
    if pobyso_is_error_so_sa(functionSo):
958 155 storres
        sollya_lib_clear_obj(functionSo)
959 155 storres
        return 0
960 154 storres
    rangeSo = pobyso_interval_to_range_sa_so(intervalSa)
961 155 storres
    # The approximation error is expected to be a floating point number.
962 155 storres
    if pobyso_is_floating_point_number_sa_sa(approxErrorSa):
963 155 storres
        approxErrorSo = pobyso_constant_sa_so(approxErrorSa)
964 155 storres
    else:
965 155 storres
        approxErrorSo = pobyso_constant_sa_so(RR(approxErrorSa))
966 154 storres
    if not weightSa is None:
967 159 storres
        weightAsStringSa = weightSa._assume_str().replace('_SAGE_VAR_', '')
968 154 storres
        weightSo = pobyso_parse_string_sa_so(weightAsStringSa)
969 166 storres
        if pobyso_is_error_so_sa(weightSo):
970 155 storres
            sollya_lib_clear_obj(functionSo)
971 155 storres
            sollya_lib_clear_obj(rangeSo)
972 155 storres
            sollya_lib_clear_obj(approxErrorSo)
973 155 storres
            sollya_lib_clear_obj(weightSo)
974 155 storres
            return 0
975 154 storres
    else:
976 154 storres
        weightSo = None
977 154 storres
    if not degreeBoundSa is None:
978 154 storres
        degreeBoundSo = pobyso_constant_from_int_sa_so(degreeBoundSa)
979 154 storres
    else:
980 154 storres
        degreeBoundSo = None
981 154 storres
    guessedDegreeSa = pobyso_guess_degree_so_sa(functionSo,
982 162 storres
                                                rangeSo,
983 162 storres
                                                approxErrorSo,
984 162 storres
                                                weightSo,
985 162 storres
                                                degreeBoundSo)
986 154 storres
    sollya_lib_clear_obj(functionSo)
987 154 storres
    sollya_lib_clear_obj(rangeSo)
988 155 storres
    sollya_lib_clear_obj(approxErrorSo)
989 154 storres
    if not weightSo is None:
990 154 storres
        sollya_lib_clear_obj(weightSo)
991 154 storres
    if not degreeBoundSo is None:
992 154 storres
        sollya_lib_clear_obj(degreeBoundSo)
993 154 storres
    return guessedDegreeSa
994 154 storres
# End poyso_guess_degree_sa_sa
995 154 storres
996 153 storres
def pobyso_guess_degree_so_sa(functionSo, rangeSo, errorSo, weightSo=None, \
997 154 storres
                              degreeBoundSo=None):
998 154 storres
    """
999 154 storres
    Thin wrapper around the guessdegree function.
1000 154 storres
    Nevertheless, some precision control stuff has been appended.
1001 154 storres
    """
1002 154 storres
    # Deal with Sollya internal precision issues: if it is too small,
1003 154 storres
    # compared with the error, increases it to about twice -log2(error).
1004 154 storres
    errorSa = pobyso_get_constant_as_rn_with_rf_so_sa(errorSo)
1005 154 storres
    log2ErrorSa = errorSa.log2()
1006 154 storres
    if log2ErrorSa < 0:
1007 154 storres
        neededPrecisionSa = int(2 * int(-log2ErrorSa) / 64) * 64
1008 154 storres
    else:
1009 154 storres
        neededPrecisionSa = int(2 * int(log2ErrorSa) / 64) * 64
1010 154 storres
    #print "Needed precision:", neededPrecisionSa
1011 226 storres
    sollyaPrecisionHasChanged = False
1012 226 storres
    (initialPrecSo, initialPrecSa) = pobyso_get_prec_so_so_sa()
1013 226 storres
    if neededPrecisionSa > initialPrecSa:
1014 226 storres
        if neededPrecisionSa <= 2:
1015 226 storres
            print inspect.stack()[0][3], ": precision change <= 2 requested."
1016 154 storres
        pobyso_set_prec_sa_so(neededPrecisionSa)
1017 226 storres
        sollyaPrecisionHasChanged = True
1018 166 storres
    #print "Guessing degree..."
1019 153 storres
    # weightSo and degreeBoundsSo are optional arguments.
1020 162 storres
    # As declared, sollya_lib_guessdegree must take 5 arguments.
1021 153 storres
    if weightSo is None:
1022 162 storres
        degreeRangeSo = sollya_lib_guessdegree(functionSo, rangeSo, errorSo,
1023 162 storres
                                               0, 0, None)
1024 154 storres
    elif degreeBoundSo is None:
1025 153 storres
        degreeRangeSo =  sollya_lib_guessdegree(functionSo, rangeSo, \
1026 162 storres
                                                errorSo, weightSo, 0, None)
1027 153 storres
    else:
1028 153 storres
        degreeRangeSo =  sollya_lib_guessdegree(functionSo, rangeSo, errorSo, \
1029 154 storres
                                                weightSo, degreeBoundSo, None)
1030 166 storres
    #print "...degree guess done."
1031 154 storres
    # Restore internal precision, if applicable.
1032 226 storres
    if sollyaPrecisionHasChanged:
1033 226 storres
        pobyso_set_prec_so_so(initialPrecSo)
1034 226 storres
        sollya_lib_clear_obj(initialPrecSo)
1035 154 storres
    degreeIntervalSa = pobyso_range_to_interval_so_sa(degreeRangeSo)
1036 154 storres
    sollya_lib_clear_obj(degreeRangeSo)
1037 154 storres
    # When ok, both bounds match.
1038 154 storres
    # When the degree bound is too low, the upper bound is the degree
1039 154 storres
    # for which the error can be honored.
1040 154 storres
    # When it really goes wrong, the upper bound is infinity.
1041 154 storres
    if degreeIntervalSa.lower() == degreeIntervalSa.upper():
1042 154 storres
        return int(degreeIntervalSa.lower())
1043 154 storres
    else:
1044 154 storres
        if degreeIntervalSa.upper().is_infinity():
1045 154 storres
            return None
1046 154 storres
        else:
1047 154 storres
            return int(degreeIntervalSa.upper())
1048 154 storres
    # End pobyso_guess_degree_so_sa
1049 153 storres
1050 215 storres
def pobyso_inf_so_so(intervalSo):
1051 215 storres
    """
1052 215 storres
    Very thin wrapper around sollya_lib_inf().
1053 215 storres
    """
1054 215 storres
    return sollya_lib_inf(intervalSo)
1055 215 storres
# End pobyso_inf_so_so.
1056 215 storres
1057 53 storres
def pobyso_infnorm_so_so(func, interval, file = None, intervalList = None):
1058 54 storres
    print "Do not use this function. User pobyso_supnorm_so_so instead."
1059 209 storres
    return None
1060 53 storres
1061 84 storres
def pobyso_interval_to_range_sa_so(intervalSa, precisionSa=None):
1062 84 storres
    if precisionSa is None:
1063 84 storres
        precisionSa = intervalSa.parent().precision()
1064 84 storres
    intervalSo = pobyso_bounds_to_range_sa_so(intervalSa.lower(),\
1065 84 storres
                                              intervalSa.upper(),\
1066 84 storres
                                              precisionSa)
1067 209 storres
    return intervalSo
1068 84 storres
# End pobyso_interval_to_range_sa_so
1069 84 storres
1070 155 storres
def pobyso_is_error_so_sa(objSo):
1071 155 storres
    """
1072 155 storres
    Thin wrapper around the sollya_lib_obj_is_error() function.
1073 155 storres
    """
1074 155 storres
    if sollya_lib_obj_is_error(objSo) != 0:
1075 155 storres
        return True
1076 155 storres
    else:
1077 155 storres
        return False
1078 155 storres
# End pobyso_is_error-so_sa
1079 155 storres
1080 155 storres
def pobyso_is_floating_point_number_sa_sa(numberSa):
1081 155 storres
    """
1082 209 storres
    Check whether a Sage number is floating point.
1083 209 storres
    Exception stuff added because numbers other than
1084 209 storres
    floating-point ones do not have the is_real() attribute.
1085 155 storres
    """
1086 209 storres
    try:
1087 209 storres
        return numberSa.is_real()
1088 209 storres
    except AttributeError:
1089 209 storres
        return False
1090 209 storres
# End pobyso_is_floating_piont_number_sa_sa
1091 155 storres
1092 226 storres
def pobyso_is_range_so_sa(rangeCandidateSo):
1093 226 storres
    """
1094 226 storres
    Thin wrapper over sollya_lib_is_range.
1095 226 storres
    """
1096 226 storres
    return sollya_lib_obj_is_range(rangeCandidateSo) != 0
1097 226 storres
1098 226 storres
# End pobyso_is_range_so_sa
1099 226 storres
1100 226 storres
1101 37 storres
def pobyso_lib_init():
1102 37 storres
    sollya_lib_init(None)
1103 116 storres
1104 116 storres
def pobyso_lib_close():
1105 116 storres
    sollya_lib_close(None)
1106 37 storres
1107 85 storres
def pobyso_name_free_variable(freeVariableNameSa):
1108 38 storres
    """ Legacy function. See pobyso_name_free_variable_sa_so. """
1109 85 storres
    pobyso_name_free_variable_sa_so(freeVariableNameSa)
1110 38 storres
1111 85 storres
def pobyso_name_free_variable_sa_so(freeVariableNameSa):
1112 83 storres
    """
1113 83 storres
    Set the free variable name in Sollya from a Sage string.
1114 83 storres
    """
1115 85 storres
    sollya_lib_name_free_variable(freeVariableNameSa)
1116 37 storres
1117 5 storres
def pobyso_parse_string(string):
1118 38 storres
    """ Legacy function. See pobyso_parse_string_sa_so. """
1119 209 storres
    return pobyso_parse_string_sa_so(string)
1120 38 storres
1121 38 storres
def pobyso_parse_string_sa_so(string):
1122 83 storres
    """
1123 155 storres
    Get the Sollya expression computed from a Sage string or
1124 155 storres
    a Sollya error object if parsing failed.
1125 83 storres
    """
1126 209 storres
    return sollya_lib_parse_string(string)
1127 5 storres
1128 200 storres
def pobyso_precision_so_sa(ctExpSo):
1129 209 storres
    """
1130 209 storres
    Computes the necessary precision to represent a number.
1131 209 storres
    If x is not zero, it can be uniquely written as x = m · 2e
1132 209 storres
    where m is an odd integer and e is an integer.
1133 209 storres
    precision(x) returns the number of bits necessary to write m
1134 209 storres
    in binary (i.e. ceil(log2(m))).
1135 209 storres
    """
1136 209 storres
    #TODO: take care of the special case: 0, @NaN@, @Inf@
1137 200 storres
    precisionSo = sollya_lib_precision(ctExpSo)
1138 200 storres
    precisionSa = pobyso_constant_from_int_so_sa(precisionSo)
1139 200 storres
    sollya_lib_clear_obj(precisionSo)
1140 200 storres
    return precisionSa
1141 200 storres
# End pobyso_precision_so_sa
1142 215 storres
1143 217 storres
def pobyso_polynomial_coefficients_progressive_round_so_so(polySo,
1144 217 storres
                                                           funcSo,
1145 217 storres
                                                           icSo,
1146 217 storres
                                                           intervalSo,
1147 217 storres
                                                           itpSo,
1148 217 storres
                                                           ftpSo,
1149 217 storres
                                                           maxPrecSo,
1150 219 storres
                                                           maxErrSo,
1151 219 storres
                                                           debug=False):
1152 219 storres
    if debug:
1153 219 storres
        print "Input arguments:"
1154 219 storres
        pobyso_autoprint(polySo)
1155 219 storres
        pobyso_autoprint(funcSo)
1156 219 storres
        pobyso_autoprint(icSo)
1157 219 storres
        pobyso_autoprint(intervalSo)
1158 219 storres
        pobyso_autoprint(itpSo)
1159 219 storres
        pobyso_autoprint(ftpSo)
1160 219 storres
        pobyso_autoprint(maxPrecSo)
1161 219 storres
        pobyso_autoprint(maxErrSo)
1162 219 storres
        print "________________"
1163 200 storres
1164 217 storres
    ## Higher order function see:
1165 217 storres
    #  http://effbot.org/pyfaq/how-do-you-make-a-higher-order-function-in-python.htm
1166 217 storres
    def precision_decay_ratio_function(degreeSa):
1167 217 storres
        def outer(x):
1168 217 storres
            def inner(x):
1169 217 storres
                we = 3/8
1170 217 storres
                wq = 2/8
1171 217 storres
                a  = 2.2
1172 217 storres
                b  = 2
1173 217 storres
                return we*(exp(x/a)-1) +  wq*((b*x)**2) + (1-we-wq)*x
1174 217 storres
            return  inner(x)/inner(degreeSa)
1175 217 storres
        return outer
1176 217 storres
1177 217 storres
    #
1178 217 storres
    degreeSa        = pobyso_polynomial_degree_so_sa(polySo)
1179 217 storres
    ratio           = precision_decay_ratio_function(degreeSa)
1180 217 storres
    itpSa           = pobyso_constant_from_int_so_sa(itpSo)
1181 217 storres
    ftpSa           = pobyso_constant_from_int_so_sa(ftpSo)
1182 217 storres
    maxPrecSa       = pobyso_constant_from_int_so_sa(maxPrecSo)
1183 217 storres
    maxErrSa        = pobyso_get_constant_as_rn_so_sa(maxErrSo)
1184 219 storres
    if debug:
1185 219 storres
        print "degreeSa:", degreeSa
1186 219 storres
        print "ratio:", ratio
1187 219 storres
        print "itpsSa:", itpSa
1188 219 storres
        print "ftpSa:", ftpSa
1189 219 storres
        print "maxPrecSa:", maxPrecSa
1190 219 storres
        print "maxErrSa:", maxErrSa
1191 217 storres
    lastResPolySo   = None
1192 218 storres
    lastInfNormSo   = None
1193 219 storres
    #print "About to enter the while loop..."
1194 217 storres
    while True:
1195 218 storres
        resPolySo   = pobyso_constant_0_sa_so()
1196 217 storres
        pDeltaSa    = ftpSa - itpSa
1197 217 storres
        for indexSa in reversed(xrange(0,degreeSa+1)):
1198 218 storres
            #print "Index:", indexSa
1199 217 storres
            indexSo = pobyso_constant_from_int_sa_so(indexSa)
1200 217 storres
            #pobyso_autoprint(indexSo)
1201 217 storres
            #print ratio(indexSa)
1202 217 storres
            ctpSa = floor(ftpSa - (pDeltaSa * ratio(indexSa)))
1203 217 storres
            ctpSo = pobyso_constant_from_int_sa_so(ctpSa)
1204 219 storres
            if debug:
1205 219 storres
                print "Index:", indexSa, " - Target precision:",
1206 219 storres
                pobyso_autoprint(ctpSo)
1207 217 storres
            cmonSo  = \
1208 217 storres
                sollya_lib_build_function_mul(sollya_lib_coeff(polySo, indexSo),
1209 217 storres
                                      sollya_lib_build_function_pow( \
1210 217 storres
                                          sollya_lib_build_function_free_variable(), \
1211 217 storres
                                          indexSo))
1212 217 storres
            #pobyso_autoprint(cmonSo)
1213 217 storres
            cmonrSo = pobyso_round_coefficients_single_so_so(cmonSo, ctpSo)
1214 217 storres
            sollya_lib_clear_obj(cmonSo)
1215 217 storres
            #pobyso_autoprint(cmonrSo)
1216 217 storres
            resPolySo = sollya_lib_build_function_add(resPolySo,
1217 217 storres
                                                      cmonrSo)
1218 218 storres
            #pobyso_autoprint(resPolySo)
1219 217 storres
        # End for index
1220 217 storres
        freeVarSo     = sollya_lib_build_function_free_variable()
1221 217 storres
        changeVarSo   = sollya_lib_sub(freeVarSo, icSo)
1222 217 storres
        resPolyCvSo   = sollya_lib_evaluate(resPolySo, changeVarSo)
1223 218 storres
        errFuncSo = sollya_lib_build_function_sub(sollya_lib_copy_obj(funcSo),
1224 218 storres
                                                  resPolyCvSo)
1225 218 storres
        infNormSo = sollya_lib_dirtyinfnorm(errFuncSo, intervalSo)
1226 217 storres
        cerrSa    = pobyso_get_constant_as_rn_so_sa(infNormSo)
1227 219 storres
        if debug:
1228 219 storres
            print "Infnorm (Sollya):",
1229 219 storres
            pobyso_autoprint(infNormSo)
1230 218 storres
        sollya_lib_clear_obj(errFuncSo)
1231 218 storres
        #print "Infnorm  (Sage):", cerrSa
1232 217 storres
        if (cerrSa > maxErrSa):
1233 219 storres
            if debug:
1234 219 storres
                print "Error is too large."
1235 218 storres
            if lastResPolySo is None:
1236 219 storres
                if debug:
1237 219 storres
                    print "Enlarging prec."
1238 217 storres
                ntpSa = floor(ftpSa + ftpSa/50)
1239 217 storres
                ## Can't enlarge (numerical)
1240 217 storres
                if ntpSa == ftpSa:
1241 217 storres
                    sollya_lib_clear_obj(resPolySo)
1242 217 storres
                    return None
1243 217 storres
                ## Can't enlarge (not enough precision left)
1244 217 storres
                if ntpSa > maxPrecSa:
1245 217 storres
                    sollya_lib_clear_obj(resPolySo)
1246 217 storres
                    return None
1247 217 storres
                ftpSa = ntpSa
1248 217 storres
                continue
1249 217 storres
            ## One enlargement took place.
1250 217 storres
            else:
1251 219 storres
                if debug:
1252 219 storres
                    print "Exit with the last before last polynomial."
1253 222 storres
                    print "Precision of highest degree monomial:", itpSa
1254 222 storres
                    print "Precision of constant term          :", ftpSa
1255 217 storres
                sollya_lib_clear_obj(resPolySo)
1256 218 storres
                sollya_lib_clear_obj(infNormSo)
1257 218 storres
                return (lastResPolySo, lastInfNormSo)
1258 218 storres
        # cerrSa <= maxErrSa: scrap more bits, possibly.
1259 217 storres
        else:
1260 219 storres
            if debug:
1261 219 storres
                print "Error is too small"
1262 219 storres
            if cerrSa <= (maxErrSa/2):
1263 219 storres
                if debug:
1264 219 storres
                    print "Shrinking prec."
1265 218 storres
                ntpSa = floor(ftpSa - ftpSa/50)
1266 218 storres
                ## Can't shrink (numerical)
1267 218 storres
                if ntpSa == ftpSa:
1268 218 storres
                    if not lastResPolySo is None:
1269 218 storres
                        sollya_lib_clear_obj(lastResPolySo)
1270 218 storres
                    if not lastInfNormSo is None:
1271 218 storres
                        sollya_lib_clear_obj(lastInfNormSo)
1272 222 storres
                    if debug:
1273 222 storres
                        print "Exit because can't shrink anymore (numerically)."
1274 222 storres
                        print "Precision of highest degree monomial:", itpSa
1275 222 storres
                        print "Precision of constant term          :", ftpSa
1276 218 storres
                    return (resPolySo, infNormSo)
1277 218 storres
                ## Can't shrink (not enough precision left)
1278 218 storres
                if ntpSa <= itpSa:
1279 218 storres
                    if not lastResPolySo is None:
1280 218 storres
                        sollya_lib_clear_obj(lastResPolySo)
1281 218 storres
                    if not lastInfNormSo is None:
1282 218 storres
                        sollya_lib_clear_obj(lastInfNormSo)
1283 222 storres
                        print "Exit because can't shrink anymore (no bits left)."
1284 222 storres
                        print "Precision of highest degree monomial:", itpSa
1285 222 storres
                        print "Precision of constant term          :", ftpSa
1286 218 storres
                    return (resPolySo, infNormSo)
1287 218 storres
                ftpSa = ntpSa
1288 217 storres
                if not lastResPolySo is None:
1289 217 storres
                    sollya_lib_clear_obj(lastResPolySo)
1290 218 storres
                if not lastInfNormSo is None:
1291 218 storres
                    sollya_lib_clear_obj(lastInfNormSo)
1292 218 storres
                lastResPolySo = resPolySo
1293 218 storres
                lastInfNormSo = infNormSo
1294 218 storres
                continue
1295 218 storres
            else: # Error is not that small, just return
1296 217 storres
                if not lastResPolySo is None:
1297 217 storres
                    sollya_lib_clear_obj(lastResPolySo)
1298 218 storres
                if not lastInfNormSo is None:
1299 218 storres
                    sollya_lib_clear_obj(lastInfNormSo)
1300 222 storres
                if debug:
1301 222 storres
                    print "Exit normally."
1302 222 storres
                    print "Precision of highest degree monomial:", itpSa
1303 222 storres
                    print "Precision of constant term          :", ftpSa
1304 218 storres
                return (resPolySo, infNormSo)
1305 217 storres
    # End wile True
1306 215 storres
    return None
1307 217 storres
# End pobyso_polynomial_coefficients_progressive_truncate_so_so.
1308 217 storres
1309 217 storres
def pobyso_polynomial_degree_so_sa(polySo):
1310 217 storres
    """
1311 217 storres
    Return the degree of a Sollya polynomial as a Sage int.
1312 217 storres
    """
1313 217 storres
    degreeSo = sollya_lib_degree(polySo)
1314 217 storres
    return pobyso_constant_from_int_so_sa(degreeSo)
1315 217 storres
# End pobyso_polynomial_degree_so_sa
1316 217 storres
1317 217 storres
def pobyso_polynomial_degree_so_so(polySo):
1318 217 storres
    """
1319 217 storres
    Thin wrapper around lib_sollya_degree().
1320 217 storres
    """
1321 217 storres
    return sollya_lib_degree(polySo)
1322 217 storres
# End pobyso_polynomial_degree_so_so
1323 217 storres
1324 5 storres
def pobyso_range(rnLowerBound, rnUpperBound):
1325 38 storres
    """ Legacy function. See pobyso_range_sa_so. """
1326 209 storres
    return pobyso_range_sa_so(rnLowerBound, rnUpperBound)
1327 38 storres
1328 226 storres
def pobyso_range_from_bounds_sa_so(rnLowerBound, rnUpperBound, precSa = None):
1329 226 storres
    """
1330 226 storres
    Create a Sollya range from 2 Sage real numbers as bounds
1331 226 storres
    """
1332 226 storres
    # TODO check precision stuff.
1333 226 storres
    sollyaPrecChanged = False
1334 226 storres
    (initialSollyaPrecSo, initialSollyaPrecSa) = \
1335 226 storres
        pobyso_get_prec_so_so_sa()
1336 226 storres
    if precSa is None:
1337 226 storres
        precSa = max(rnLowerBound.parent().prec(), rnUpperBound.parent().prec())
1338 226 storres
    if precSa > initialSollyaPrecSa:
1339 226 storres
        if precSa <= 2:
1340 226 storres
            print inspect.stack()[0][3], ": precision change <= 2 requested."
1341 226 storres
        pobyso_set_prec_sa_so(precSa)
1342 226 storres
        sollyaPrecChanged = True
1343 226 storres
    rangeSo = sollya_lib_range_from_bounds(get_rn_value(rnLowerBound),
1344 226 storres
                                           get_rn_value(rnUpperBound))
1345 226 storres
    if sollyaPrecChanged:
1346 226 storres
        pobyso_set_prec_so_so(initialSollyaPrecSo)
1347 226 storres
        pobyso_lib_clear_obj(initialSollyaPrecSo)
1348 226 storres
    return rangeSo
1349 226 storres
# End pobyso_range_from_bounds_sa_so
1350 5 storres
1351 226 storres
def pobyso_range_max_abs_so_so(rangeSo):
1352 226 storres
    """
1353 226 storres
    Return, as Sollya constant, the maximum absolute value of a Sollay range.
1354 226 storres
    """
1355 226 storres
    lowerBoundSo = sollya_lib_inf(rangeSo)
1356 226 storres
    upperBoundSo = sollya_lib_sup(rangeSo)
1357 226 storres
    #
1358 226 storres
    lowerBoundSo = sollya_lib_build_function_abs(lowerBoundSo)
1359 226 storres
    upperBoundSo = sollya_lib_build_function_abs(upperBoundSo)
1360 226 storres
    #pobyso_autoprint(lowerBoundSo)
1361 226 storres
    #pobyso_autoprint(upperBoundSo)
1362 226 storres
    #
1363 226 storres
    maxAbsSo = sollya_lib_max(lowerBoundSo, upperBoundSo, None)
1364 226 storres
    #sollya_lib_clear_obj(lowerBoundSo)
1365 226 storres
    #sollya_lib_clear_obj(upperBoundSo)
1366 226 storres
    return maxAbsSo
1367 226 storres
# End pobyso_range_max_abs_so_so
1368 226 storres
1369 85 storres
def pobyso_range_to_interval_so_sa(rangeSo, realIntervalFieldSa = None):
1370 83 storres
    """
1371 83 storres
    Get a Sage interval from a Sollya range.
1372 83 storres
    If no realIntervalField is given as a parameter, the Sage interval
1373 83 storres
    precision is that of the Sollya range.
1374 85 storres
    Otherwise, the precision is that of the realIntervalField. In this case
1375 85 storres
    rounding may happen.
1376 83 storres
    """
1377 85 storres
    if realIntervalFieldSa is None:
1378 56 storres
        precSa = pobyso_get_prec_of_range_so_sa(rangeSo)
1379 85 storres
        realIntervalFieldSa = RealIntervalField(precSa)
1380 56 storres
    intervalSa = \
1381 85 storres
        pobyso_get_interval_from_range_so_sa(rangeSo, realIntervalFieldSa)
1382 209 storres
    return intervalSa
1383 209 storres
# End pobyso_range_to_interval_so_sa
1384 56 storres
1385 209 storres
def pobyso_rat_poly_sa_so(polySa, precSa = None):
1386 209 storres
    """
1387 209 storres
    Create a Sollya polynomial from a Sage rational polynomial.
1388 209 storres
    """
1389 209 storres
    ## TODO: filter arguments.
1390 209 storres
    ## Precision. If no precision is given, use the current precision
1391 209 storres
    #  of Sollya.
1392 209 storres
    if precSa is None:
1393 209 storres
        precSa =  pobyso_get_prec_so_sa()
1394 209 storres
    #print "Precision:",  precSa
1395 209 storres
    RRR = RealField(precSa)
1396 209 storres
    ## Create a Sage polynomial in the "right" precision.
1397 209 storres
    P_RRR = RRR[polySa.variables()[0]]
1398 209 storres
    polyFloatSa = P_RRR(polySa)
1399 213 storres
    ## Make sure no precision is provided: pobyso_float_poly_sa_so will
1400 213 storres
    #  recover it all by itself and not make an extra conversion.
1401 209 storres
    return pobyso_float_poly_sa_so(polyFloatSa)
1402 209 storres
1403 209 storres
# End pobyso_rat_poly_sa_so
1404 209 storres
1405 52 storres
def pobyso_remez_canonical_sa_sa(func, \
1406 52 storres
                                 degree, \
1407 52 storres
                                 lowerBound, \
1408 52 storres
                                 upperBound, \
1409 52 storres
                                 weight = None, \
1410 52 storres
                                 quality = None):
1411 52 storres
    """
1412 52 storres
    All arguments are Sage/Python.
1413 52 storres
    The functions (func and weight) must be passed as expressions or strings.
1414 52 storres
    Otherwise the function fails.
1415 83 storres
    The return value is a Sage polynomial.
1416 52 storres
    """
1417 83 storres
    var('zorglub')    # Dummy variable name for type check only. Type of
1418 83 storres
    # zorglub is "symbolic expression".
1419 52 storres
    polySo = pobyso_remez_canonical_sa_so(func, \
1420 52 storres
                                 degree, \
1421 52 storres
                                 lowerBound, \
1422 52 storres
                                 upperBound, \
1423 85 storres
                                 weight, \
1424 85 storres
                                 quality)
1425 83 storres
    # String test
1426 52 storres
    if parent(func) == parent("string"):
1427 52 storres
        functionSa = eval(func)
1428 52 storres
    # Expression test.
1429 52 storres
    elif type(func) == type(zorglub):
1430 52 storres
        functionSa = func
1431 83 storres
    else:
1432 83 storres
        return None
1433 83 storres
    #
1434 52 storres
    maxPrecision = 0
1435 52 storres
    if polySo is None:
1436 52 storres
        return(None)
1437 52 storres
    maxPrecision = pobyso_get_max_prec_of_exp_so_sa(polySo)
1438 85 storres
    RRRRSa = RealField(maxPrecision)
1439 85 storres
    polynomialRingSa = RRRRSa[functionSa.variables()[0]]
1440 85 storres
    expSa = pobyso_get_sage_exp_from_sollya_exp_so_sa(polySo, RRRRSa)
1441 85 storres
    polySa = polynomial(expSa, polynomialRingSa)
1442 83 storres
    sollya_lib_clear_obj(polySo)
1443 52 storres
    return(polySa)
1444 85 storres
# End pobyso_remez_canonical_sa_sa
1445 52 storres
1446 38 storres
def pobyso_remez_canonical(func, \
1447 5 storres
                           degree, \
1448 5 storres
                           lowerBound, \
1449 5 storres
                           upperBound, \
1450 38 storres
                           weight = "1", \
1451 5 storres
                           quality = None):
1452 38 storres
    """ Legacy function. See pobyso_remez_canonical_sa_so. """
1453 51 storres
    return(pobyso_remez_canonical_sa_so(func, \
1454 51 storres
                                        degree, \
1455 51 storres
                                        lowerBound, \
1456 51 storres
                                        upperBound, \
1457 51 storres
                                        weight, \
1458 51 storres
                                        quality))
1459 200 storres
# End pobyso_remez_canonical.
1460 200 storres
1461 38 storres
def pobyso_remez_canonical_sa_so(func, \
1462 38 storres
                                 degree, \
1463 38 storres
                                 lowerBound, \
1464 38 storres
                                 upperBound, \
1465 52 storres
                                 weight = None, \
1466 38 storres
                                 quality = None):
1467 38 storres
    """
1468 38 storres
    All arguments are Sage/Python.
1469 51 storres
    The functions (func and weight) must be passed as expressions or strings.
1470 51 storres
    Otherwise the function fails.
1471 38 storres
    The return value is a pointer to a Sollya function.
1472 38 storres
    """
1473 83 storres
    var('zorglub')    # Dummy variable name for type check only. Type of
1474 83 storres
    # zorglub is "symbolic expression".
1475 85 storres
    currentVariableNameSa = None
1476 52 storres
    # The func argument can be of different types (string,
1477 52 storres
    # symbolic expression...)
1478 38 storres
    if parent(func) == parent("string"):
1479 85 storres
        localFuncSa = eval(func)
1480 85 storres
        if len(localFuncSa.variables()) > 0:
1481 85 storres
            currentVariableNameSa = localFuncSa.variables()[0]
1482 85 storres
            sollya_lib_name_free_variable(str(currentVariableNameSa))
1483 159 storres
            functionSo = \
1484 159 storres
              sollya_lib_parse_string(localFuncSa._assume_str().replace('_SAGE_VAR_', ''))
1485 51 storres
    # Expression test.
1486 52 storres
    elif type(func) == type(zorglub):
1487 52 storres
        # Until we are able to translate Sage expressions into Sollya
1488 52 storres
        # expressions : parse the string version.
1489 85 storres
        if len(func.variables()) > 0:
1490 85 storres
            currentVariableNameSa = func.variables()[0]
1491 85 storres
            sollya_lib_name_free_variable(str(currentVariableNameSa))
1492 159 storres
            functionSo = \
1493 159 storres
              sollya_lib_parse_string(func._assume_str().replace('_SAGE_VAR_', ''))
1494 38 storres
    else:
1495 38 storres
        return(None)
1496 85 storres
    if weight is None: # No weight given -> 1.
1497 52 storres
        weightSo = pobyso_constant_1_sa_so()
1498 85 storres
    elif parent(weight) == parent("string"): # Weight given as string: parse it.
1499 51 storres
        weightSo = sollya_lib_parse_string(func)
1500 85 storres
    elif type(weight) == type(zorglub): # Weight given as symbolice expression.
1501 159 storres
        functionSo = \
1502 159 storres
          sollya_lib_parse_string_sa_so(weight._assume_str().replace('_SAGE_VAR_', ''))
1503 51 storres
    else:
1504 51 storres
        return(None)
1505 5 storres
    degreeSo = pobyso_constant_from_int(degree)
1506 85 storres
    rangeSo = pobyso_bounds_to_range_sa_so(lowerBound, upperBound)
1507 38 storres
    if not quality is None:
1508 38 storres
        qualitySo= pobyso_constant_sa_so(quality)
1509 52 storres
    else:
1510 52 storres
        qualitySo = None
1511 83 storres
1512 83 storres
    remezPolySo = sollya_lib_remez(functionSo, \
1513 83 storres
                                   degreeSo, \
1514 83 storres
                                   rangeSo, \
1515 83 storres
                                   weightSo, \
1516 83 storres
                                   qualitySo, \
1517 83 storres
                                   None)
1518 83 storres
    sollya_lib_clear_obj(functionSo)
1519 83 storres
    sollya_lib_clear_obj(degreeSo)
1520 83 storres
    sollya_lib_clear_obj(rangeSo)
1521 83 storres
    sollya_lib_clear_obj(weightSo)
1522 83 storres
    if not qualitySo is None:
1523 85 storres
        sollya_lib_clear_obj(qualitySo)
1524 83 storres
    return(remezPolySo)
1525 83 storres
# End pobyso_remez_canonical_sa_so
1526 83 storres
1527 38 storres
def pobyso_remez_canonical_so_so(funcSo, \
1528 38 storres
                                 degreeSo, \
1529 38 storres
                                 rangeSo, \
1530 52 storres
                                 weightSo = pobyso_constant_1_sa_so(),\
1531 38 storres
                                 qualitySo = None):
1532 38 storres
    """
1533 38 storres
    All arguments are pointers to Sollya objects.
1534 38 storres
    The return value is a pointer to a Sollya function.
1535 38 storres
    """
1536 38 storres
    if not sollya_lib_obj_is_function(funcSo):
1537 38 storres
        return(None)
1538 38 storres
    return(sollya_lib_remez(funcSo, degreeSo, rangeSo, weightSo, qualitySo, None))
1539 200 storres
# End pobyso_remez_canonical_so_so.
1540 200 storres
1541 224 storres
def pobyso_round_coefficients_progressive_so_so(polySo,
1542 224 storres
                                                funcSo,
1543 224 storres
                                                precSo,
1544 224 storres
                                                intervalSo,
1545 224 storres
                                                icSo,
1546 224 storres
                                                currentApproxErrorSo,
1547 224 storres
                                                approxAccurSo,
1548 224 storres
                                                debug=False):
1549 223 storres
    if debug:
1550 223 storres
        print "Input arguments:"
1551 226 storres
        print "Polynomia:l", ; pobyso_autoprint(polySo)
1552 226 storres
        print "Function:", ; pobyso_autoprint(funcSo)
1553 226 storres
        print "Internal precision:", ; pobyso_autoprint(precSo)
1554 226 storres
        print "Interval:", ; pobyso_autoprint(intervalSo)
1555 226 storres
        print "Current approximation error:", ; pobyso_autoprint(currentApproxErrorSo)
1556 226 storres
        print "Requested approxiation error:", ; pobyso_autoprint(approxAccurSo)
1557 223 storres
        print "________________"
1558 223 storres
    approxAccurSa        = pobyso_get_constant_as_rn_so_sa(approxAccurSo)
1559 223 storres
    currentApproxErrorSa = pobyso_get_constant_as_rn_so_sa(currentApproxErrorSo)
1560 223 storres
    ## If the current approximation error is too close to the target, there is
1561 223 storres
    #  no possible gain.
1562 223 storres
    if currentApproxErrorSa >= approxAccurSa / 2:
1563 223 storres
        return (polySo, currentApproxErrorSo)
1564 223 storres
    degreeSa             = pobyso_polynomial_degree_so_sa(polySo)
1565 223 storres
    intervalSa           = pobyso_range_to_interval_so_sa(intervalSo)
1566 223 storres
1567 223 storres
    if debug:
1568 223 storres
        print "degreeSa  :", degreeSa
1569 223 storres
        print "intervalSa:", intervalSa.str(style='brackets')
1570 223 storres
        print "currentApproxErrorSa  :", currentApproxErrorSa
1571 223 storres
        print "approxAccurSa  :", approxAccurSa
1572 223 storres
    ### Start with a 0 value expression.
1573 223 storres
    radiusSa = intervalSa.absolute_diameter() / 2
1574 223 storres
    if debug:
1575 224 storres
        print "log2(radius):", RR(radiusSa).log2()
1576 224 storres
    iterIndex = 0
1577 224 storres
    while True:
1578 224 storres
        resPolySo = pobyso_constant_0_sa_so()
1579 224 storres
        roundedPolyApproxAccurSa = approxAccurSa / 2
1580 224 storres
        currentRadiusPowerSa = 1
1581 224 storres
        for degree in xrange(0,degreeSa + 1):
1582 224 storres
            #### At round 0, use the agressive formula. At round 1, run the
1583 224 storres
            #    proved formula.
1584 224 storres
            if iterIndex == 0:
1585 224 storres
                roundingPowerSa = \
1586 224 storres
                    floor(((currentRadiusPowerSa/roundedPolyApproxAccurSa)*(degree+1)).log2())
1587 224 storres
            else:
1588 224 storres
                roundingPowerSa = \
1589 224 storres
                    floor(((currentRadiusPowerSa/roundedPolyApproxAccurSa)*(degreeSa+1)).log2())
1590 224 storres
            if debug:
1591 224 storres
                print "roundedPolyApproxAccurSa", roundedPolyApproxAccurSa
1592 224 storres
                print "currentRadiusPowerSa", currentRadiusPowerSa
1593 224 storres
                print "Current rounding exponent:", roundingPowerSa
1594 224 storres
            currentRadiusPowerSa *= radiusSa
1595 224 storres
            index1So = pobyso_constant_from_int_sa_so(degree)
1596 224 storres
            index2So = pobyso_constant_from_int_sa_so(degree)
1597 224 storres
            ### Create a monomial with:
1598 224 storres
            #   - the coefficient in the initial monomial at the current degrree;
1599 224 storres
            #   - the current exponent;
1600 224 storres
            #   - the free variable.
1601 224 storres
            cmonSo  = \
1602 224 storres
                sollya_lib_build_function_mul(sollya_lib_coeff(polySo, index1So),
1603 224 storres
                                              sollya_lib_build_function_pow( \
1604 224 storres
                                                sollya_lib_build_function_free_variable(), \
1605 224 storres
                                                index2So))
1606 224 storres
            roundingPowerSo = pobyso_constant_from_int_sa_so(roundingPowerSa)
1607 224 storres
            cmonrSo = pobyso_round_coefficients_single_so_so(cmonSo, roundingPowerSo)
1608 224 storres
            sollya_lib_clear_obj(cmonSo)
1609 224 storres
            ### Add to the result polynomial.
1610 224 storres
            resPolySo = sollya_lib_build_function_add(resPolySo,
1611 224 storres
                                                      cmonrSo)
1612 224 storres
        # End for.
1613 224 storres
        ### Check the new polynomial.
1614 224 storres
        freeVarSo     = sollya_lib_build_function_free_variable()
1615 224 storres
        changeVarSo   = sollya_lib_sub(freeVarSo, icSo)
1616 224 storres
        resPolyCvSo   = sollya_lib_evaluate(resPolySo, changeVarSo)
1617 224 storres
        errFuncSo     = sollya_lib_build_function_sub(sollya_lib_copy_obj(funcSo),
1618 224 storres
                                                      resPolyCvSo)
1619 224 storres
        infNormSo = sollya_lib_dirtyinfnorm(errFuncSo, intervalSo)
1620 224 storres
        ### This also clears resPolyCvSo.
1621 224 storres
        sollya_lib_clear_obj(errFuncSo)
1622 224 storres
        cerrSa    = pobyso_get_constant_as_rn_so_sa(infNormSo)
1623 223 storres
        if debug:
1624 224 storres
            print "Error of the new polynomial:", cerrSa
1625 224 storres
        ### If at round 1, return the initial polynomial error. This should
1626 224 storres
        #   never happen since the rounding algorithm is proved. But some
1627 224 storres
        #   circumstances may break it (e.g. internal precision of tools).
1628 224 storres
        if cerrSa > approxAccurSa:
1629 224 storres
            if iterIndex > 0: # Round 1 and beyond.
1630 224 storres
                sollya_lib_clear_obj(resPolySo)
1631 224 storres
                sollya_lib_clear_obj(infNormSo)
1632 224 storres
                return (polySo, currentApproxErrorSo)
1633 224 storres
            else: # Round 0, got round 1
1634 224 storres
                sollya_lib_clear_obj(resPolySo)
1635 224 storres
                sollya_lib_clear_obj(infNormSo)
1636 224 storres
                iterIndex += 1
1637 224 storres
                continue
1638 224 storres
        ### If get here it is because cerrSa <= approxAccurSa
1639 224 storres
        ### Approximation error of the new polynomial is acceptable.
1640 224 storres
        return (resPolySo, infNormSo)
1641 224 storres
    # End while True
1642 224 storres
# End pobyso_round_coefficients_progressive_so_so
1643 223 storres
1644 215 storres
def pobyso_round_coefficients_single_so_so(polySo, precSo):
1645 215 storres
    """
1646 215 storres
    Create a rounded coefficients polynomial from polynomial argument to
1647 215 storres
    the number of bits in size argument.
1648 215 storres
    All coefficients are set to the same precision.
1649 215 storres
    """
1650 215 storres
    ## TODO: check arguments.
1651 215 storres
    endEllipListSo = pobyso_build_end_elliptic_list_so_so(precSo)
1652 215 storres
    polySo = sollya_lib_roundcoefficients(polySo, endEllipListSo, None)
1653 217 storres
    sollya_lib_clear_obj(endEllipListSo)
1654 215 storres
    #sollya_lib_clear_obj(endEllipListSo)
1655 215 storres
    return polySo
1656 215 storres
1657 215 storres
# End pobyso_round_coefficients_single_so_so
1658 215 storres
1659 5 storres
def pobyso_set_canonical_off():
1660 5 storres
    sollya_lib_set_canonical(sollya_lib_off())
1661 5 storres
1662 5 storres
def pobyso_set_canonical_on():
1663 5 storres
    sollya_lib_set_canonical(sollya_lib_on())
1664 5 storres
1665 5 storres
def pobyso_set_prec(p):
1666 38 storres
    """ Legacy function. See pobyso_set_prec_sa_so. """
1667 85 storres
    pobyso_set_prec_sa_so(p)
1668 38 storres
1669 38 storres
def pobyso_set_prec_sa_so(p):
1670 227 storres
    #a = c_int(p)
1671 226 storres
    #precSo = c_void_p(sollya_lib_constant_from_int(a))
1672 227 storres
    #precSo = sollya_lib_constant_from_int(a)
1673 227 storres
    precSo = pobyso_constant_from_int_sa_so(p)
1674 226 storres
    sollya_lib_set_prec(precSo)
1675 226 storres
    sollya_lib_clear_obj(precSo)
1676 215 storres
# End pobyso_set_prec_sa_so.
1677 5 storres
1678 85 storres
def pobyso_set_prec_so_so(newPrecSo):
1679 226 storres
    sollya_lib_set_prec(newPrecSo)
1680 215 storres
# End pobyso_set_prec_so_so.
1681 54 storres
1682 215 storres
def pobyso_inf_so_so(intervalSo):
1683 215 storres
    """
1684 215 storres
    Very thin wrapper around sollya_lib_inf().
1685 215 storres
    """
1686 215 storres
    return sollya_lib_inf(intervalSo)
1687 215 storres
# End pobyso_inf_so_so.
1688 215 storres
1689 85 storres
def pobyso_supnorm_so_so(polySo, funcSo, intervalSo, errorTypeSo = None,\
1690 85 storres
                         accuracySo = None):
1691 58 storres
    """
1692 85 storres
    Computes the supnorm of the approximation error between the given
1693 85 storres
    polynomial and function.
1694 85 storres
    errorTypeSo defaults to "absolute".
1695 85 storres
    accuracySo defaults to 2^(-40).
1696 85 storres
    """
1697 85 storres
    if errorTypeSo is None:
1698 85 storres
        errorTypeSo = sollya_lib_absolute(None)
1699 85 storres
        errorTypeIsNone = True
1700 85 storres
    else:
1701 85 storres
        errorTypeIsNone = False
1702 85 storres
    #
1703 85 storres
    if accuracySo is None:
1704 85 storres
        # Notice the **!
1705 85 storres
        accuracySo = pobyso_constant_sa_so(RR(2**(-40)))
1706 85 storres
        accuracyIsNone = True
1707 85 storres
    else:
1708 85 storres
        accuracyIsNone = False
1709 85 storres
    pobyso_autoprint(accuracySo)
1710 85 storres
    resultSo = \
1711 85 storres
        sollya_lib_supnorm(polySo, funcSo, intervalSo, errorTypeSo, \
1712 85 storres
                              accuracySo)
1713 85 storres
    if errorTypeIsNone:
1714 85 storres
        sollya_lib_clear_obj(errorTypeSo)
1715 85 storres
    if accuracyIsNone:
1716 85 storres
        sollya_lib_clear_obj(accuracySo)
1717 85 storres
    return resultSo
1718 85 storres
# End pobyso_supnorm_so_so
1719 85 storres
1720 162 storres
def pobyso_taylor_expansion_no_change_var_so_so(functionSo,
1721 162 storres
                                                degreeSo,
1722 162 storres
                                                rangeSo,
1723 162 storres
                                                errorTypeSo=None,
1724 162 storres
                                                sollyaPrecSo=None):
1725 85 storres
    """
1726 162 storres
    Compute the Taylor expansion without the variable change
1727 162 storres
    x -> x-intervalCenter.
1728 58 storres
    """
1729 226 storres
    # Change internal Sollya precision, if needed.
1730 227 storres
    (initialSollyaPrecSo, initialSollyaPrecSa) = pobyso_get_prec_so_so_sa()
1731 226 storres
    sollyaPrecChanged = False
1732 226 storres
    if sollyaPrecSo is None:
1733 226 storres
        pass
1734 226 storres
    else:
1735 58 storres
        sollya_lib_set_prec(sollyaPrecSo)
1736 226 storres
        sollyaPrecChanged = True
1737 85 storres
    # Error type stuff: default to absolute.
1738 85 storres
    if errorTypeSo is None:
1739 85 storres
        errorTypeIsNone = True
1740 85 storres
        errorTypeSo = sollya_lib_absolute(None)
1741 85 storres
    else:
1742 85 storres
        errorTypeIsNone = False
1743 162 storres
    intervalCenterSo = sollya_lib_mid(rangeSo, None)
1744 162 storres
    taylorFormSo = sollya_lib_taylorform(functionSo, degreeSo,
1745 162 storres
                                         intervalCenterSo,
1746 58 storres
                                         rangeSo, errorTypeSo, None)
1747 117 storres
    # taylorFormListSaSo is a Python list of Sollya objects references that
1748 117 storres
    # are copies of the elements of taylorFormSo.
1749 117 storres
    # pobyso_get_list_elements_so_so clears taylorFormSo.
1750 162 storres
    (taylorFormListSaSo, numElementsSa, isEndEllipticSa) = \
1751 58 storres
        pobyso_get_list_elements_so_so(taylorFormSo)
1752 162 storres
    polySo = sollya_lib_copy_obj(taylorFormListSaSo[0])
1753 162 storres
    #print "Num elements:", numElementsSa
1754 162 storres
    sollya_lib_clear_obj(taylorFormSo)
1755 162 storres
    #polySo = taylorFormListSaSo[0]
1756 162 storres
    #errorRangeSo = sollya_lib_copy_obj(taylorFormListSaSo[2])
1757 162 storres
    errorRangeSo = taylorFormListSaSo[2]
1758 181 storres
    # No copy_obj needed here: a new objects are created.
1759 181 storres
    maxErrorSo    = sollya_lib_sup(errorRangeSo)
1760 181 storres
    minErrorSo    = sollya_lib_inf(errorRangeSo)
1761 181 storres
    absMaxErrorSo = sollya_lib_abs(maxErrorSo)
1762 181 storres
    absMinErrorSo = sollya_lib_abs(minErrorSo)
1763 181 storres
    sollya_lib_clear_obj(maxErrorSo)
1764 181 storres
    sollya_lib_clear_obj(minErrorSo)
1765 181 storres
    absMaxErrorSa = pobyso_get_constant_as_rn_so_sa(absMaxErrorSo)
1766 181 storres
    absMinErrorSa = pobyso_get_constant_as_rn_so_sa(absMinErrorSo)
1767 58 storres
    # If changed, reset the Sollya working precision.
1768 226 storres
    if sollyaPrecChanged:
1769 226 storres
        sollya_lib_set_prec(initialSollyaPrecSo)
1770 226 storres
    sollya_lib_clear_obj(initialSollyaPrecSo)
1771 85 storres
    if errorTypeIsNone:
1772 85 storres
        sollya_lib_clear_obj(errorTypeSo)
1773 162 storres
    pobyso_clear_taylorform_sa_so(taylorFormListSaSo)
1774 181 storres
    if absMaxErrorSa > absMinErrorSa:
1775 181 storres
        sollya_lib_clear_obj(absMinErrorSo)
1776 227 storres
        return (polySo, intervalCenterSo, absMaxErrorSo)
1777 181 storres
    else:
1778 181 storres
        sollya_lib_clear_obj(absMaxErrorSo)
1779 227 storres
        return (polySo, intervalCenterSo, absMinErrorSo)
1780 162 storres
# end pobyso_taylor_expansion_no_change_var_so_so
1781 58 storres
1782 162 storres
def pobyso_taylor_expansion_with_change_var_so_so(functionSo, degreeSo, \
1783 162 storres
                                                  rangeSo, \
1784 162 storres
                                                  errorTypeSo=None, \
1785 162 storres
                                                  sollyaPrecSo=None):
1786 58 storres
    """
1787 162 storres
    Compute the Taylor expansion with the variable change
1788 162 storres
    x -> (x-intervalCenter) included.
1789 58 storres
    """
1790 226 storres
    # Change Sollya internal precision, if need.
1791 226 storres
    sollyaPrecChanged = False
1792 226 storres
    (initialSollyaPrecSo, initialSollyaPrecSa) = pobyso_get_prec_so_sos_sa()
1793 226 storres
    if sollyaPrecSo is None:
1794 226 storres
        pass
1795 226 storres
    else:
1796 56 storres
        sollya_lib_set_prec(sollyaPrecSo)
1797 226 storres
        sollyaPrecChanged = True
1798 162 storres
    #
1799 85 storres
    # Error type stuff: default to absolute.
1800 85 storres
    if errorTypeSo is None:
1801 85 storres
        errorTypeIsNone = True
1802 85 storres
        errorTypeSo = sollya_lib_absolute(None)
1803 85 storres
    else:
1804 85 storres
        errorTypeIsNone = False
1805 162 storres
    intervalCenterSo = sollya_lib_mid(rangeSo)
1806 162 storres
    taylorFormSo = sollya_lib_taylorform(functionSo, degreeSo, \
1807 162 storres
                                         intervalCenterSo, \
1808 56 storres
                                         rangeSo, errorTypeSo, None)
1809 116 storres
    # taylorFormListSaSo is a Python list of Sollya objects references that
1810 116 storres
    # are copies of the elements of taylorFormSo.
1811 116 storres
    # pobyso_get_list_elements_so_so clears taylorFormSo.
1812 162 storres
    (taylorFormListSo, numElements, isEndElliptic) = \
1813 56 storres
        pobyso_get_list_elements_so_so(taylorFormSo)
1814 162 storres
    polySo = taylorFormListSo[0]
1815 162 storres
    errorRangeSo = taylorFormListSo[2]
1816 181 storres
    maxErrorSo    = sollya_lib_sup(errorRangeSo)
1817 181 storres
    minErrorSo    = sollya_lib_inf(errorRangeSo)
1818 181 storres
    absMaxErrorSo = sollya_lib_abs(maxErrorSo)
1819 181 storres
    absMinErrorSo = sollya_lib_abs(minErrorSo)
1820 181 storres
    sollya_lib_clear_obj(maxErrorSo)
1821 181 storres
    sollya_lib_clear_obj(minErrorSo)
1822 181 storres
    absMaxErrorSa = pobyso_get_constant_as_rn_so_sa(absMaxErrorSo)
1823 181 storres
    absMinErrorSa = pobyso_get_constant_as_rn_so_sa(absMinErrorSo)
1824 162 storres
    changeVarExpSo = sollya_lib_build_function_sub(\
1825 162 storres
                       sollya_lib_build_function_free_variable(),\
1826 162 storres
                       sollya_lib_copy_obj(intervalCenterSo))
1827 181 storres
    polyVarChangedSo = sollya_lib_evaluate(polySo, changeVarExpSo)
1828 181 storres
    sollya_lib_clear_obj(polySo)
1829 162 storres
    sollya_lib_clear_obj(changeVarExpSo)
1830 56 storres
    # If changed, reset the Sollya working precision.
1831 226 storres
    if sollyaPrecChanged:
1832 226 storres
        sollya_lib_set_prec(initialSollyaPrecSo)
1833 226 storres
    sollya_lib_clear_obj(initialSollyaPrecSo)
1834 85 storres
    if errorTypeIsNone:
1835 85 storres
        sollya_lib_clear_obj(errorTypeSo)
1836 162 storres
    sollya_lib_clear_obj(taylorFormSo)
1837 162 storres
    # Do not clear maxErrorSo.
1838 181 storres
    if absMaxErrorSa > absMinErrorSa:
1839 181 storres
        sollya_lib_clear_obj(absMinErrorSo)
1840 181 storres
        return((polyVarChangedSo, intervalCenterSo, absMaxErrorSo))
1841 181 storres
    else:
1842 181 storres
        sollya_lib_clear_obj(absMaxErrorSo)
1843 181 storres
        return((polyVarChangedSo, intervalCenterSo, absMinErrorSo))
1844 162 storres
# end pobyso_taylor_expansion_with_change_var_so_so
1845 56 storres
1846 5 storres
def pobyso_taylor(function, degree, point):
1847 38 storres
    """ Legacy function. See pobysoTaylor_so_so. """
1848 38 storres
    return(pobyso_taylor_so_so(function, degree, point))
1849 38 storres
1850 56 storres
def pobyso_taylor_so_so(functionSo, degreeSo, pointSo):
1851 56 storres
    return(sollya_lib_taylor(functionSo, degreeSo, pointSo))
1852 5 storres
1853 85 storres
def pobyso_taylorform(function, degree, point = None,
1854 85 storres
                      interval = None, errorType=None):
1855 85 storres
    """ Legacy function. See pobyso_taylorform_sa_sa;"""
1856 38 storres
1857 38 storres
def pobyso_taylorform_sa_sa(functionSa, \
1858 84 storres
                            degreeSa, \
1859 84 storres
                            pointSa, \
1860 84 storres
                            intervalSa=None, \
1861 84 storres
                            errorTypeSa=None, \
1862 84 storres
                            precisionSa=None):
1863 37 storres
    """
1864 85 storres
    Compute the Taylor form of 'degreeSa' for 'functionSa' at 'pointSa'
1865 85 storres
    for 'intervalSa' with 'errorTypeSa' (a string) using 'precisionSa'.
1866 37 storres
    point: must be a Real or a Real interval.
1867 37 storres
    return the Taylor form as an array
1868 83 storres
    TODO: take care of the interval and of the point when it is an interval;
1869 38 storres
          when errorType is not None;
1870 83 storres
          take care of the other elements of the Taylor form (coefficients
1871 83 storres
          errors and delta.
1872 37 storres
    """
1873 37 storres
    # Absolute as the default error.
1874 84 storres
    if errorTypeSa is None:
1875 37 storres
        errorTypeSo = sollya_lib_absolute()
1876 84 storres
    elif errorTypeSa == "relative":
1877 84 storres
        errorTypeSo = sollya_lib_relative()
1878 84 storres
    elif errortypeSa == "absolute":
1879 84 storres
        errorTypeSo = sollya_lib_absolute()
1880 37 storres
    else:
1881 84 storres
        # No clean up needed.
1882 84 storres
        return None
1883 84 storres
    # Global precision stuff
1884 226 storres
    sollyaPrecisionChangedSa = False
1885 226 storres
    (initialSollyaPrecSo, initialSollyaPrecSa) = pobyso_get_prec_so_so_sa()
1886 226 storres
    if precisionSa is None:
1887 226 storres
        precSa = initialSollyaPrecSa
1888 226 storres
    else:
1889 226 storres
        if precSa > initialSollyaPrecSa:
1890 226 storres
            if precSa <= 2:
1891 226 storres
                print inspect.stack()[0][3], ":precision change <= 2 requested."
1892 226 storres
            pobyso_set_prec_sa_so(precSa)
1893 226 storres
            sollyaPrecisionChangedSa = True
1894 226 storres
    #
1895 85 storres
    if len(functionSa.variables()) > 0:
1896 85 storres
        varSa = functionSa.variables()[0]
1897 85 storres
        pobyso_name_free_variable_sa_so(str(varSa))
1898 84 storres
    # In any case (point or interval) the parent of pointSa has a precision
1899 84 storres
    # method.
1900 84 storres
    pointPrecSa = pointSa.parent().precision()
1901 226 storres
    if precSa > pointPrecSa:
1902 226 storres
        pointPrecSa = precSa
1903 84 storres
    # In any case (point or interval) pointSa has a base_ring() method.
1904 84 storres
    pointBaseRingString = str(pointSa.base_ring())
1905 84 storres
    if re.search('Interval', pointBaseRingString) is None: # Point
1906 84 storres
        pointSo = pobyso_constant_sa_so(pointSa, pointPrecSa)
1907 84 storres
    else: # Interval.
1908 84 storres
        pointSo = pobyso_interval_to_range_sa_so(pointSa, pointPrecSa)
1909 37 storres
    # Sollyafy the function.
1910 159 storres
    functionSo = pobyso_parse_string_sa_so(functionSa._assume_str().replace('_SAGE_VAR_', ''))
1911 37 storres
    if sollya_lib_obj_is_error(functionSo):
1912 37 storres
        print "pobyso_tailorform: function string can't be parsed!"
1913 37 storres
        return None
1914 37 storres
    # Sollyafy the degree
1915 84 storres
    degreeSo = sollya_lib_constant_from_int(int(degreeSa))
1916 37 storres
    # Sollyafy the point
1917 37 storres
    # Call Sollya
1918 83 storres
    taylorFormSo = \
1919 83 storres
        sollya_lib_taylorform(functionSo, degreeSo, pointSo, errorTypeSo,\
1920 37 storres
                                         None)
1921 85 storres
    sollya_lib_clear_obj(functionSo)
1922 85 storres
    sollya_lib_clear_obj(degreeSo)
1923 85 storres
    sollya_lib_clear_obj(pointSo)
1924 85 storres
    sollya_lib_clear_obj(errorTypeSo)
1925 38 storres
    (tfsAsList, numElements, isEndElliptic) = \
1926 38 storres
            pobyso_get_list_elements_so_so(taylorFormSo)
1927 37 storres
    polySo = tfsAsList[0]
1928 38 storres
    maxPrecision = pobyso_get_max_prec_of_exp_so_sa(polySo)
1929 37 storres
    polyRealField = RealField(maxPrecision)
1930 38 storres
    expSa = pobyso_get_sage_exp_from_sollya_exp_so_sa(polySo, polyRealField)
1931 226 storres
    if sollyaPrecisionChangedSa:
1932 226 storres
        sollya_lib_set_prec(initialSollyaPrecSo)
1933 226 storres
    sollya_lib_clear_obj(initialSollyaPrecSo)
1934 37 storres
    polynomialRing = polyRealField[str(varSa)]
1935 37 storres
    polySa = polynomial(expSa, polynomialRing)
1936 37 storres
    taylorFormSa = [polySa]
1937 85 storres
    # Final clean-up.
1938 85 storres
    sollya_lib_clear_obj(taylorFormSo)
1939 51 storres
    return(taylorFormSa)
1940 51 storres
# End pobyso_taylor_form_sa_sa
1941 54 storres
1942 54 storres
def pobyso_taylorform_so_so(functionSo, degreeSo, pointSo, intervalSo=None, \
1943 54 storres
                            errorTypeSo=None):
1944 54 storres
    createdErrorType = False
1945 51 storres
    if errorTypeSo is None:
1946 51 storres
        errorTypeSo = sollya_lib_absolute()
1947 54 storres
        createdErrorType = True
1948 51 storres
    else:
1949 51 storres
        #TODO: deal with the other case.
1950 51 storres
        pass
1951 51 storres
    if intervalSo is None:
1952 54 storres
        resultSo = sollya_lib_taylorform(functionSo, degreeSo, pointSo, \
1953 54 storres
                                         errorTypeSo, None)
1954 51 storres
    else:
1955 54 storres
        resultSo = sollya_lib_taylorform(functionSo, degreeSo, pointSo, \
1956 54 storres
                                         intervalSo, errorTypeSo, None)
1957 54 storres
    if createdErrorType:
1958 54 storres
        sollya_lib_clear_obj(errorTypeSo)
1959 215 storres
    return resultSo
1960 51 storres
1961 37 storres
1962 37 storres
def pobyso_univar_polynomial_print_reverse(polySa):
1963 51 storres
    """ Legacy function. See pobyso_univar_polynomial_print_reverse_sa_sa. """
1964 51 storres
    return(pobyso_univar_polynomial_print_reverse_sa_sa(polySa))
1965 38 storres
1966 51 storres
def pobyso_univar_polynomial_print_reverse_sa_sa(polySa):
1967 37 storres
    """
1968 37 storres
    Return the string representation of a univariate polynomial with
1969 38 storres
    monomials ordered in the x^0..x^n order of the monomials.
1970 37 storres
    Remember: Sage
1971 37 storres
    """
1972 37 storres
    polynomialRing = polySa.base_ring()
1973 37 storres
    # A very expensive solution:
1974 37 storres
    # -create a fake multivariate polynomial field with only one variable,
1975 37 storres
    #   specifying a negative lexicographical order;
1976 37 storres
    mpolynomialRing = PolynomialRing(polynomialRing.base(), \
1977 37 storres
                                     polynomialRing.variable_name(), \
1978 37 storres
                                     1, order='neglex')
1979 37 storres
    # - convert the univariate argument polynomial into a multivariate
1980 37 storres
    #   version;
1981 37 storres
    p = mpolynomialRing(polySa)
1982 37 storres
    # - return the string representation of the converted form.
1983 37 storres
    # There is no simple str() method defined for p's class.
1984 37 storres
    return(p.__str__())
1985 5 storres
#
1986 5 storres
print pobyso_get_prec()
1987 5 storres
pobyso_set_prec(165)
1988 5 storres
print pobyso_get_prec()
1989 5 storres
a=100
1990 5 storres
print type(a)
1991 5 storres
id(a)
1992 5 storres
print "Max arity: ", pobyso_max_arity
1993 5 storres
print "Function tripleDouble (43) as a string: ", pobyso_function_type_as_string(43)
1994 56 storres
print "Function None (44) as a string: ", pobyso_function_type_as_string(44)
1995 56 storres
print "...Pobyso check done"