Ask Your Question

argentum2f's profile - activity

2015-01-13 14:13:22 -0500 received badge  Famous Question (source)
2014-10-07 01:17:20 -0500 received badge  Notable Question (source)
2014-07-25 03:51:07 -0500 received badge  Popular Question (source)
2014-06-27 14:36:35 -0500 commented answer Fastest way to call special function (elliptic integral) from cython code for gsl ode_solver()

I'm using - Is there any simple way of figure out what's available and how to get to it? I found that hitting tab auto completes, and after a lot of guesswork and navigating auto complete trees I did finally find the elliptic functions here: **sage.libs.mpmath.ext_main.global_context.ellipk** It has sped things up by about a factor of 5. Wouldn't it be faster if I cimport sage.mpmath or something (like I have to do with sage.gsl.ode)? I tried that and sage complains about their not being a pyx file or something. Do I have to write my own, and would it help at all?

2014-06-27 05:55:47 -0500 asked a question Fastest way to call special function (elliptic integral) from cython code for gsl ode_solver()

I am using sage.gsl.ode.ode_solver to solve an ode, and overloading the c_f() function as detailed here.

The code is working using an approximate c_f() function. For the exact one, I need to use the complete elliptic integrals. These are special functions that are not part of the libc.math library. They seem to be available through a couple packages on sage. I found I could call them by using sage.all.elliptic_ec(k) and sage.all.elliptic_kc(k)

This is causing the code to slow down by a factor of about 10^3. I know those special functions are going to be expensive, but timing the elliptic_ec() function in a separate cell returns 0 for the time, so these functions aren't that slow.

I'm wondering if the problem is mostly just that I'm calling a python function from cython and losing the speed up because of that? Is there a better way to do that? There are c libraries available with those functions - would it be better to import the files into my project, and call them from the cython code? (not sure how to do that...) Sage seems to have some gsl packages already - does it also have those special functions through gsl? Is there a different package with a faster form of those functions?

Disclaimer: I've played around with sage a bit in the past, but this is the first intensive numerical simulation I've attempted with it, so sage/python/cython are all relatively new to me. I'm more accustomed to Mathematica, Matlab/Octave and c++

Here are (I believe) the relevant bits of code:

from libc.math cimport pow
cimport sage.gsl.ode
import sage.gsl.ode
include 'gsl.pxi'

cdef class zeeman_acceleration(sage.gsl.ode.ode_system):
    ... other stuff ...
    cdef double coilFieldR(self, double rp, double zp, double coilR): # T
        cdef double B0, alpha, beta, Q, k
        ... some basic arithmetic ...
        if rp > coilR/10e4 :
            gamma = zp/rp
            EC = sage.all.elliptic_ec(k)
            EK = sage.all.elliptic_kc(k)
            Br = B0*gamma/(self.pi*sqrt(Q))*(EC*(1+alpha**2+beta**2)/(Q-4*alpha)-EK)
            return Br
        return 0
... other stuff (including c_f() function that calls coilField...