Fix errors on SDEProblems

Author: albertomercurio

Project.toml

@@ -61,7 +61,7 @@ OrdinaryDiffEqTsit5 = "1"
 Pkg = "1"
 Random = "1"
 SciMLBase = "2"
-SciMLOperators = "0.3, 0.4, 1"
+SciMLOperators = "1"
 SparseArrays = "1"
 SpecialFunctions = "2"
 StaticArraysCore = "1"

src/QuantumToolbox.jl

@@ -16,6 +16,7 @@ import SciMLBase:
     u_modified!,
     NullParameters,
     ODEFunction,
+    SDEFunction,
     ODEProblem,
     SDEProblem,
     EnsembleProblem,

src/time_evolution/time_evolution.jl

@@ -395,37 +395,37 @@ end
 
 A struct to represent the diffusion operator. This is used to perform the diffusion process on N different Wiener processes.
 =#
-struct DiffusionOperator{T,OpType<:Tuple{Vararg{AbstractSciMLOperator}}} <: AbstractSciMLOperator{T}
+struct DiffusionOperator{T,OpType<:Tuple{Vararg{AbstractSciMLOperator}}}
     ops::OpType
     function DiffusionOperator(ops::OpType) where {OpType}
         T = mapreduce(eltype, promote_type, ops)
         return new{T,OpType}(ops)
     end
 end
 
-@generated function update_coefficients!(L::DiffusionOperator, u, p, t)
+@generated function (L::DiffusionOperator)(w, v, p, t)
     ops_types = L.parameters[2].parameters
     N = length(ops_types)
-    return quote
+    quote
+        M = length(v)
+        S = (size(w, 1), size(w, 2)) # This supports also `w` as a `Vector`
+        (S[1] == M && S[2] == $N) || throw(DimensionMismatch("The size of the output vector is incorrect."))
         Base.@nexprs $N i -> begin
-            update_coefficients!(L.ops[i], u, p, t)
+            # update_coefficients!(L.ops[i], v, p, t)
+            # mul!(@view(w[:, i]), L.ops[i], v)
+            op = L.ops[i]
+            op(@view(w[:, i]), v, v, p, t)
         end
-
-        nothing
+        return w
     end
 end
 
-@generated function LinearAlgebra.mul!(v::AbstractVecOrMat, L::DiffusionOperator, u::AbstractVecOrMat)
-    ops_types = L.parameters[2].parameters
-    N = length(ops_types)
-    quote
-        M = length(u)
-        S = (size(v, 1), size(v, 2)) # This supports also `v` as a `Vector`
-        (S[1] == M && S[2] == $N) || throw(DimensionMismatch("The size of the output vector is incorrect."))
-        Base.@nexprs $N i -> begin
-            mul!(@view(v[:, i]), L.ops[i], u)
-        end
-        return v
+#TODO: Remove when https://github.com/SciML/StochasticDiffEq.jl/issues/615 is fixed.
+function (f::SDEFunction)(du, u, p, t)
+    if f.f isa AbstractSciMLOperator
+        f.f(du, u, u, p, t)
+    else
+        f.f(du, u, p, t)
     end
 end
 

test/core-test/time_evolution.jl

@@ -484,7 +484,6 @@ end
     H = TESetup.H
     ψ0 = TESetup.ψ0
     tlist = TESetup.tlist
-    c_ops = TESetup.c_ops
     c_ops_sme = TESetup.c_ops_sme
     sc_ops_sme = TESetup.sc_ops_sme
     c_ops_sme2 = TESetup.c_ops_sme2