Skip to content

Commit 6340428

Browse files
committed
REF: Use compute complex
Use compute complex for DRY pruposes
1 parent 8c97648 commit 6340428

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

randomstate/randomstate.pyx

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1849,12 +1849,8 @@ cdef class RandomState:
18491849
random_gauss_fill(&self.rng_state, 1, &f_real)
18501850
random_gauss_fill(&self.rng_state, 1, &f_imag)
18511851

1852-
f_imag *= sqrt(1 - f_rho * f_rho)
1853-
f_imag += f_rho * f_real
1854-
f_real *= sqrt(0.5 * f_v_real)
1855-
f_imag *= sqrt(0.5 * f_v_imag)
1856-
1857-
return PyComplex_FromDoubles(floc_r + f_real, floc_i + f_imag)
1852+
compute_complex(&f_real, &f_imag, floc_r, floc_i, f_v_real, f_v_imag, f_rho)
1853+
return PyComplex_FromDoubles(f_real, f_imag)
18581854

18591855
randoms = <np.ndarray>np.empty(size, np.complex128)
18601856
randoms_data = <double *>np.PyArray_DATA(randoms)
@@ -1880,7 +1876,6 @@ cdef class RandomState:
18801876
randoms_data[j] = floc_r + r_scale * f_real
18811877
j += 2
18821878

1883-
18841879
return randoms
18851880

18861881
gpc = ogamma + orelation

0 commit comments

Comments
 (0)