format

Author: ChrisRackauckas

lib/libsundials_api.jl

@@ -63,7 +63,6 @@ Base.convert(::Type{NVector}, nv::N_Vector) = NVector(nv)
 Base.convert(::Type{Vector{realtype}}, nv::NVector) = nv.v
 Base.convert(::Type{Vector}, nv::NVector) = nv.v
 
-
 """ 
     Base.cconvert(::Type{N_Vector}, v::Vector{realtype}) -> NVector
     Base.unsafe_convert(::Type{N_Vector}, nv::NVector) -> N_Vector
@@ -79,7 +78,6 @@ Conversion happens in two steps within ccall:
 Base.cconvert(::Type{N_Vector}, v::Vector{realtype}) = convert(NVector, v) # will just return v if v is an NVector
 Base.unsafe_convert(::Type{N_Vector}, nv::NVector) = nv.n_v
 
-
 Base.similar(nv::NVector) = NVector(similar(nv.v))
 
 nvlength(x::N_Vector) = unsafe_load(unsafe_load(convert(Ptr{Ptr{Clong}}, x)))

src/nvector_wrapper.jl

@@ -11,7 +11,8 @@ function f(t, y_nv, ydot_nv, user_data)
     return Sundials.ARK_SUCCESS
 end
 
-f_C = @cfunction(f, Cint, (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ptr{Cvoid}))
+f_C = @cfunction(f, Cint,
+                 (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ptr{Cvoid}))
 
 neq = 3
 

test/arkstep_Roberts_dns.jl

@@ -50,7 +50,6 @@ u0 = vec(init_brusselator_2d(xyd_brusselator))
 prob_ode_brusselator_2d = ODEProblem(brusselator_2d_vec,
                                      u0, (0.0, 11.5), p)
 
-
 # find Jacobian sparsity pattern
 u0_st = SparsityTracing.create_advec(u0)
 du_st = similar(u0_st)
@@ -60,19 +59,19 @@ const W = I - 1.0 * jaccache
 
 # setup sparse AD for Jacobian
 colors = SparseDiffTools.matrix_colors(jaccache)
-const jaccache_fc = SparseDiffTools.ForwardColorJacCache(
-    nothing, # don't use f to create unique Tag
-    u0, 
-    colorvec=colors,
-    sparsity=jaccache,
-)
+const jaccache_fc = SparseDiffTools.ForwardColorJacCache(nothing, # don't use f to create unique Tag
+                                                         u0,
+                                                         colorvec = colors,
+                                                         sparsity = jaccache)
 
 prectmp = ilu(W; τ = 50.0)
 const preccache = Ref(prectmp)
 
 function psetupilu(p, t, u, du, jok, jcurPtr, gamma)
     if jok
-        SparseDiffTools.forwarddiff_color_jacobian!(jaccache, (y, x) -> brusselator_2d_vec(y, x, p, t), u, jaccache_fc)
+        SparseDiffTools.forwarddiff_color_jacobian!(jaccache,
+                                                    (y, x) -> brusselator_2d_vec(y, x, p, t),
+                                                    u, jaccache_fc)
         jcurPtr[] = true
 
         # W = I - gamma*J
@@ -97,7 +96,9 @@ prectmp2 = AlgebraicMultigrid.aspreconditioner(AlgebraicMultigrid.ruge_stuben(W;
 const preccache2 = Ref(prectmp2)
 function psetupamg(p, t, u, du, jok, jcurPtr, gamma)
     if jok
-        SparseDiffTools.forwarddiff_color_jacobian!(jaccache, (y, x) -> brusselator_2d_vec(y, x, p, t), u, jaccache_fc)
+        SparseDiffTools.forwarddiff_color_jacobian!(jaccache,
+                                                    (y, x) -> brusselator_2d_vec(y, x, p, t),
+                                                    u, jaccache_fc)
         jcurPtr[] = true
 
         # W = I - gamma*J

test/common_interface/precs.jl

@@ -22,7 +22,7 @@ function g(t, y_nv, gout_ptr, user_data)
 end
 
 g_C = @cfunction(g, Cint,
-                (Sundials.realtype, Sundials.N_Vector, Ptr{Sundials.realtype}, Ptr{Cvoid}))
+                 (Sundials.realtype, Sundials.N_Vector, Ptr{Sundials.realtype}, Ptr{Cvoid}))
 
 ## Jacobian routine. Compute J(t,y) = df/dy.
 # broken -- needs a wrapper from Sundials._DlsMat to Matrix and Jac user function wrapper
@@ -64,7 +64,8 @@ function getcfunrob(userfun::T) where {T}
                (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ref{T}))
 end
 
-Sundials.@checkflag Sundials.CVodeInit(cvode_mem, getcfunrob(userfun), t1, convert(Sundials.NVector, y0))
+Sundials.@checkflag Sundials.CVodeInit(cvode_mem, getcfunrob(userfun), t1,
+                                       convert(Sundials.NVector, y0))
 Sundials.@checkflag Sundials.CVodeInit(cvode_mem, getcfunrob(userfun), t0, y0)
 Sundials.@checkflag Sundials.CVodeSVtolerances(cvode_mem, reltol, abstol)
 Sundials.@checkflag Sundials.CVodeRootInit(cvode_mem, 2, g_C)

test/cvode_Roberts_dns.jl

@@ -45,7 +45,8 @@ function f(t, y, ydot, user_data)
     return Sundials.ARK_SUCCESS
 end
 
-f_C = @cfunction(f, Cint, (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ptr{Cvoid}))
+f_C = @cfunction(f, Cint,
+                 (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ptr{Cvoid}))
 
 mem_ptr = Sundials.ERKStepCreate(f_C, t0, y0)
 erkStep_mem = Sundials.Handle(mem_ptr)

test/erkstep_nonlin.jl

@@ -45,7 +45,8 @@ function resrob(tres, yy_nv, yp_nv, rr_nv, user_data)
 end
 
 resrob_C = @cfunction(resrob, Cint,
-                (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Sundials.N_Vector, Ptr{Cvoid}))
+                      (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector,
+                       Sundials.N_Vector, Ptr{Cvoid}))
 
 ## Root function routine. Compute functions g_i(t,y) for i = 0,1.
 function grob(t, yy_nv, yp_nv, gout_ptr, user_data)
@@ -57,7 +58,8 @@ function grob(t, yy_nv, yp_nv, gout_ptr, user_data)
 end
 
 grob_C = @cfunction(grob, Cint,
-                (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector, Ptr{Sundials.realtype}, Ptr{Cvoid}))
+                    (Sundials.realtype, Sundials.N_Vector, Sundials.N_Vector,
+                     Ptr{Sundials.realtype}, Ptr{Cvoid}))
 
 ## Define the Jacobian function. BROKEN - JJ is wrong
 function jacrob(Neq, tt, cj, yy, yp, resvec, JJ, user_data, tempv1, tempv2, tempv3)