|
|
@ -8,10 +8,8 @@ from numba import jit, float64, int64, int32, int8, void
|
|
|
|
import numpy as np
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@jit(int64(int64[:], int8[:]))
|
|
|
|
@jit(void(int64, int64, int64, int64[:], int8[:], int64, int64))
|
|
|
|
def _findcross(ind, y):
|
|
|
|
def _find_first_cross(ix, start, dcross, ind, y, v, n):
|
|
|
|
ix, dcross, start, v = 0, 0, 0, 0
|
|
|
|
|
|
|
|
n = len(y)
|
|
|
|
|
|
|
|
if y[0] < v:
|
|
|
|
if y[0] < v:
|
|
|
|
dcross = -1 # first is a up-crossing
|
|
|
|
dcross = -1 # first is a up-crossing
|
|
|
|
elif y[0] > v:
|
|
|
|
elif y[0] > v:
|
|
|
@ -31,6 +29,13 @@ def _findcross(ind, y):
|
|
|
|
dcross = 1 # The next crossing is a down-crossing
|
|
|
|
dcross = 1 # The next crossing is a down-crossing
|
|
|
|
break
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@jit(int64(int64[:], int8[:]))
|
|
|
|
|
|
|
|
def _findcross(ind, y):
|
|
|
|
|
|
|
|
ix, dcross, start, v = 0, 0, 0, 0
|
|
|
|
|
|
|
|
n = len(y)
|
|
|
|
|
|
|
|
_find_first_cross(ix, start, dcross, ind, y, v, n)
|
|
|
|
|
|
|
|
|
|
|
|
for i in range(start, n - 1):
|
|
|
|
for i in range(start, n - 1):
|
|
|
|
if ((dcross == -1 and y[i] <= v and v < y[i + 1]) or
|
|
|
|
if ((dcross == -1 and y[i] <= v and v < y[i + 1]) or
|
|
|
|
(dcross == 1 and v <= y[i] and y[i + 1] < v)):
|
|
|
|
(dcross == 1 and v <= y[i] and y[i + 1] < v)):
|
|
|
|