Merge pull request #83 from AayushSabharwal/as/symbolic-save-idxs

feat: add support for symbolic `save_idxs`

Author: ChrisRackauckas

Project.toml

@@ -15,6 +15,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 RandomNumbers = "e6cf234a-135c-5ec9-84dd-332b85af5143"
 RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
+SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 StochasticDiffEq = "789caeaf-c7a9-5a7d-9973-96adeb23e2a0"
@@ -32,6 +33,7 @@ Random = "1.9"
 RandomNumbers = "1"
 RecursiveArrayTools = "1,2,3"
 Reexport = "1.0"
+SciMLBase = "2.59.2"
 SparseArrays = "1.9"
 StaticArrays = "1.0"
 StochasticDiffEq = "6.19"

src/StochasticDelayDiffEq.jl

@@ -14,6 +14,7 @@ using UnPack, DataStructures
 using Logging
 using RecursiveArrayTools
 import FastPower
+import SciMLBase
 using DiffEqBase: AbstractSDDEProblem, AbstractSDDEAlgorithm, AbstractRODESolution,
                   AbstractRODEFunction, AbstractSDEIntegrator, AbstractSDDEIntegrator,
                   DEIntegrator, DEAlgorithm, AbstractRODEAlgorithm, AbstractSDEAlgorithm

src/solve.jl

@@ -316,6 +316,7 @@ function DiffEqBase.__init(prob::AbstractSDDEProblem,# TODO DiffEqBasee.Abstract
         end
     end
 
+    save_idxs, saved_subsystem = SciMLBase.get_save_idxs_and_saved_subsystem(prob, save_idxs)
     ts, timeseries, saveiter = solution_arrays(u, tspan, rate_prototype,
                                                timeseries_init = timeseries_init,
                                                ts_init = ts_init, save_idxs = save_idxs,
@@ -414,15 +415,15 @@ function DiffEqBase.__init(prob::AbstractSDDEProblem,# TODO DiffEqBasee.Abstract
         # TODO: DISCONNECT!!!!
         sol = DiffEqBase.build_solution(prob, alg, sde_integrator.sol.t,
                                         sde_integrator.sol.u, W = W,
-                                        stats = stats,
+                                        stats = stats, saved_subsystem = saved_subsystem,
                                         calculate_error = false, alg_choice = alg_choice,
                                         interp = id, dense = dense, seed = _seed)
         # separate statistics of the integrator and the history
     else
         # TODO: DISCONNECT!!!!
         sol = DiffEqBase.build_solution(prob, alg, sde_integrator.sol.t,
                                         sde_integrator.sol.u, W = W,
-                                        stats = stats,
+                                        stats = stats, saved_subsystem = saved_subsystem,
                                         calculate_error = false,
                                         interp = id, dense = dense, seed = _seed)
         # separate statistics of the integrator and the history