improved dense search

Author: ChrisRackauckas

src/dense.jl

@@ -93,7 +93,7 @@ times ts (sorted), with values timeseries and derivatives ks
   end
   @inbounds for j in idx
     t = tvals[j]
-    i = findfirst((x)->tdir*x>=tdir*t,@view ts[i:end])+i-1 # It's in the interval ts[i-1] to ts[i]
+    i = searchsortedfirst(@view(ts[i:end]),t,rev=tdir<0)+i-1 # It's in the interval ts[i-1] to ts[i]
     if ts[i] == t
       if idxs == nothing
         vals[j] = timeseries[i]
@@ -129,7 +129,7 @@ times ts (sorted), with values timeseries and derivatives ks
 @inline function sde_interpolation(tval::Number,id,idxs,deriv)
   @unpack ts,timeseries = id
   tdir = sign(ts[end]-ts[1])
-  i = findfirst((x)->tdir*x>=tdir*tval,ts) # It's in the interval ts[i-1] to ts[i]
+  @inbounds i = searchsortedfirst(ts,tval,rev=tdir<0) # It's in the interval ts[i-1] to ts[i]
   @inbounds if ts[i] == tval
     if idxs == nothing
       val = timeseries[i]
@@ -158,7 +158,7 @@ end
 @inline function sde_interpolation!(out,tval::Number,id,idxs,deriv)
   @unpack ts,timeseries = id
   tdir = sign(ts[end]-ts[1])
-  @inbounds i = findfirst((x)->tdir*x>=tdir*tval,@view ts[notsaveat_idxs]) # It's in the interval ts[i-1] to ts[i]
+  @inbounds i = searchsortedfirst(ts,tval,rev=tdir<0) # It's in the interval ts[i-1] to ts[i]
   @inbounds if ts[i] == tval
     if idxs == nothing
       copy!(out,timeseries[i])
@@ -190,7 +190,7 @@ end
   i = 2 # Start the search thinking it's between ts[1] and ts[2]
   @inbounds for j in idx
     t = tvals[j]
-    i = findfirst((x)->tdir*x>=tdir*t,ts[i:end])+i-1 # It's in the interval ts[i-1] to ts[i]
+    i = searchsortedfirst(@view(ts[i:end]),t,rev=tdir<0)+i-1 # It's in the interval ts[i-1] to ts[i]
     if ts[i] == t
       if idxs == nothing
         vals[j] = timeseries[i]