@@ -64,20 +64,20 @@ cdef int c_jac(double t,double *y,double *dfdy,double *dfdt,void *params):
64
64
cdef int param_n
65
65
cdef PyFunctionWrapper wrapper
66
66
wrapper = < PyFunctionWrapper > params
67
- y_n= wrapper.y_n
68
- y_list= []
69
- for i from 0 <= i < y_n:
67
+ y_n = wrapper.y_n
68
+ y_list = []
69
+ for i in range ( y_n) :
70
70
y_list.append(y[i])
71
71
try :
72
72
if len (wrapper.the_parameters)== 0 :
73
73
jac_list= wrapper.the_jacobian(t,y_list)
74
74
else :
75
75
jac_list= wrapper.the_jacobian(t,y_list,wrapper.the_parameters)
76
- for i from 0 <= i < y_n:
77
- for j from 0 <= j < y_n:
76
+ for i in range ( y_n) :
77
+ for j in range ( y_n) :
78
78
dfdy[i* y_n+ j]= jac_list[i][j]
79
79
80
- for i from 0 <= i < y_n:
80
+ for i in range ( y_n) :
81
81
dfdt[i]= jac_list[y_n][i]
82
82
83
83
return GSL_SUCCESS
@@ -91,17 +91,17 @@ cdef int c_f(double t,double* y, double* dydt,void *params):
91
91
92
92
cdef PyFunctionWrapper wrapper
93
93
wrapper = < PyFunctionWrapper> params
94
- y_n= wrapper.y_n
95
- y_list= []
96
- for i from 0 <= i < y_n:
94
+ y_n = wrapper.y_n
95
+ y_list = []
96
+ for i in range ( y_n) :
97
97
y_list.append(y[i])
98
98
try :
99
99
if len (wrapper.the_parameters)!= 0 :
100
100
dydt_list= wrapper.the_function(t,y_list,wrapper.the_parameters)
101
101
else :
102
102
dydt_list= wrapper.the_function(t,y_list)
103
- for i from 0 <= i < y_n:
104
- dydt[i]= dydt_list[i]
103
+ for i in range ( y_n) :
104
+ dydt[i] = dydt_list[i]
105
105
return GSL_SUCCESS
106
106
except Exception :
107
107
return - 1
@@ -453,15 +453,15 @@ class ode_solver():
453
453
cdef double * scale_abs_array
454
454
scale_abs_array= NULL
455
455
456
- y= < double * > sig_malloc(sizeof(double )* (dim))
457
- if y== NULL :
456
+ y = < double * > sig_malloc(sizeof(double )* (dim))
457
+ if y == NULL :
458
458
raise MemoryError (" error allocating memory" )
459
- result= []
460
- v= [0 ]* dim
459
+ result = []
460
+ v = [0 ]* dim
461
461
cdef gsl_odeiv_step_type * T
462
462
463
- for i from 0 <= i < dim: # copy initial conditions into C array
464
- y[i]= self .y_0[i]
463
+ for i in range ( dim): # copy initial conditions into C array
464
+ y[i] = self .y_0[i]
465
465
466
466
if self .algorithm == " rkf45" :
467
467
T= gsl_odeiv_step_rkf45
@@ -502,9 +502,9 @@ class ode_solver():
502
502
if not self .scale_abs:
503
503
c = gsl_odeiv_control_standard_new(self .error_abs,self .error_rel,self .a,self .a_dydt)
504
504
elif hasattr (self .scale_abs,' __len__' ):
505
- if len (self .scale_abs)== dim:
505
+ if len (self .scale_abs) == dim:
506
506
scale_abs_array = < double * > sig_malloc(dim* sizeof(double ))
507
- for i from 0 <= i < dim:
507
+ for i in range ( dim) :
508
508
scale_abs_array[i]= self .scale_abs[i]
509
509
c = gsl_odeiv_control_scaled_new(self .error_abs,self .error_rel,self .a,self .a_dydt,scale_abs_array,dim)
510
510
@@ -551,11 +551,11 @@ class ode_solver():
551
551
sig_free(y)
552
552
sig_free(scale_abs_array)
553
553
raise TypeError (" numpoints must be integer" )
554
- result.append( (self .t_span[0 ],self .y_0))
554
+ result.append((self .t_span[0 ], self .y_0))
555
555
delta = (self .t_span[1 ]- self .t_span[0 ])/ (1.0 * num_points)
556
- t = self .t_span[0 ]
557
- t_end= self .t_span[0 ]+ delta
558
- for i from 0 < i <= n :
556
+ t = self .t_span[0 ]
557
+ t_end = self .t_span[0 ]+ delta
558
+ for i in range ( 1 , n + 1 ) :
559
559
while (t < t_end):
560
560
try :
561
561
sig_on()
@@ -571,17 +571,17 @@ class ode_solver():
571
571
sig_free(scale_abs_array)
572
572
raise ValueError (" error solving" )
573
573
574
- for j from 0 <= j < dim:
574
+ for j in range ( dim) :
575
575
v[j]= < double > y[j]
576
576
result.append( (t,copy.copy(v)) )
577
577
t = t_end
578
578
t_end= t+ delta
579
579
580
580
else :
581
581
n = len (self .t_span)
582
- result.append((self .t_span[0 ],self .y_0))
583
- t= self .t_span[0 ]
584
- for i from 0 < i < n :
582
+ result.append((self .t_span[0 ], self .y_0))
583
+ t = self .t_span[0 ]
584
+ for i in range ( 1 , n) :
585
585
t_end= self .t_span[i]
586
586
while (t < t_end):
587
587
try :
0 commit comments