hiro/jobmanager
1
0
Fork 0
mirror of https://github.com/vale981/jobmanager synced 2025-03-05 09:51:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

apply scale_function before extracting integration data

Browse source
This commit is contained in:
Richard Hartmann 2020-05-27 16:11:11 +02:00
parent 1469e9940a
commit 15ed8415ea
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
jobmanager/ode_wrapper.py
View file

@ -130,6 +130,9 @@ def integrate_cplx(c, t0, t1, N, f, args, x0, integrator, res_dim=None, x_to_res
_t = time()
r_y = copy.copy(r.y)
if scale_function:
r_y = scale_function(r_y)
if integrator == 'zvode':
# complex integration -> yields complex values
x[i] = x_to_res(r.t, r_y)
@ -149,7 +152,7 @@ def integrate_cplx(c, t0, t1, N, f, args, x0, integrator, res_dim=None, x_to_res
del r
r = ode(f__)
r.set_integrator(integrator, **kwargs)
r.set_initial_value(y=scale_function(copy.copy(r_y)), t=t[i])
r.set_initial_value(y=r_y, t=t[i])
i += 1