First content

modified:   Project.toml
modified:   src/MechGlueDiffEqBase.jl

Author: hustf

Project.toml

@@ -3,6 +3,10 @@ uuid = "2532746b-52b5-4539-9431-8bb183ab067f"
 authors = ["hustf <[email protected]> and contributors"]
 version = "0.1.0"
 
+[deps]
+DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
+MechanicalUnits = "e6be9192-89dc-11e9-36e6-5dbcb28f419e"
+
 [compat]
 julia = "1"
 

src/MechGlueDiffEqBase.jl

@@ -1,5 +1,25 @@
 module MechGlueDiffEqBase
 
-# Write your package code here.
+using MechanicalUnits
+import MechanicalUnits.Unitfu as Unitful
+import DiffEqBase
+import DiffEqBase: value, ODE_DEFAULT_NORM, UNITLESS_ABS2
+
+function __init__()
+    # Glue code copied from DiffEqBase.jl init.jl
+    # Here, we are using an alias. We are actually using types from Unitfu, but
+    # with the alias Unitful. Unitfu should only vary from Unitful in how it shows and parses quantities.
+    value(x::Type{Unitful.AbstractQuantity{T,D,U}}) where {T,D,U} = T
+    value(x::Unitful.AbstractQuantity) = x.val
+    @inline function ODE_DEFAULT_NORM(u::AbstractArray{<:Unitful.AbstractQuantity,N},t) where {N}
+        sqrt(sum(x->ODE_DEFAULT_NORM(x[1],x[2]),zip((value(x) for x in u),Iterators.repeated(t))) / length(u))
+    end
+    @inline function ODE_DEFAULT_NORM(u::Array{<:Unitful.AbstractQuantity,N},t) where {N}
+        sqrt(sum(x->ODE_DEFAULT_NORM(x[1],x[2]),zip((value(x) for x in u),Iterators.repeated(t))) / length(u))
+    end
+    @inline ODE_DEFAULT_NORM(u::Unitful.AbstractQuantity,t) = abs(value(u))
+    @inline UNITLESS_ABS2(x::Unitful.AbstractQuantity) = real(abs2(x)/oneunit(x)*oneunit(x))
+end
+
 
 end