Minor fixes and cleanup in SPARK integrators.

JuliaGNI · Dec 12, 2024 · ab01cd4 · ab01cd4
1 parent 53b7444
commit ab01cd4
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 9 deletions.
diff --git a/src/spark/integrators_hpark.jl b/src/spark/integrators_hpark.jl
@@ -9,7 +9,7 @@ end
 
 tableau(method::HPARK) = method.tableau
 
-solversize(problem::Union{PDAEProblem,HDAEProblem}, method::HPARK) =
+solversize(problem::AbstractProblemPDAE, method::HPARK) =
     2 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_hspark.jl b/src/spark/integrators_hspark.jl
@@ -9,7 +9,7 @@ HSPARK(method::SPARKMethod) = HSPARK(tableau(method))
 
 tableau(method::HSPARK) = method.tableau
 
-solversize(problem::Union{PDAEProblem,HDAEProblem}, method::Union{HSPARK,SPARKMethod}) =
+solversize(problem::AbstractProblemPDAE, method::Union{HSPARK,SPARKMethod}) =
     2 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_hspark_primary.jl b/src/spark/integrators_hspark_primary.jl
@@ -7,7 +7,7 @@ end
 
 tableau(method::HSPARKprimary) = method.tableau
 
-solversize(problem::Union{PDAEProblem,HDAEProblem}, method::HSPARKprimary) =
+solversize(problem::AbstractProblemPDAE, method::HSPARKprimary) =
     2 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_hspark_secondary.jl b/src/spark/integrators_hspark_secondary.jl
@@ -42,7 +42,7 @@ pstages(method::HSPARKsecondary) = method.r
 hasnullvector(method::HSPARKsecondary{DT,Nothing}) where {DT} = false
 hasnullvector(method::HSPARKsecondary{DT,<:AbstractVector}) where {DT} = true
 
-solversize(problem::HDAEProblem, method::HSPARKsecondary) =
+solversize(problem::AbstractProblemPDAE, method::HSPARKsecondary) =
     2 * ndims(problem) * nstages(method) + 4 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_spark.jl b/src/spark/integrators_spark.jl
@@ -27,7 +27,7 @@ end
 
 tableau(method::SPARKMethod) = method.tableau
 
-solversize(problem::Union{IDAEProblem,LDAEProblem}, method::SPARKMethod) =
+solversize(problem::AbstractProblemIDAE, method::SPARKMethod) =
     2 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_vpark.jl b/src/spark/integrators_vpark.jl
@@ -8,7 +8,7 @@ end
 
 tableau(method::VPARK) = method.tableau
 
-solversize(problem::Union{IDAEProblem,LDAEProblem}, method::VPARK) =
+solversize(problem::AbstractProblemIDAE, method::VPARK) =
     3 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_vspark.jl b/src/spark/integrators_vspark.jl
@@ -9,7 +9,7 @@ VSPARK(method::SPARKMethod) = VSPARK(tableau(method))
 
 tableau(method::VSPARK) = method.tableau
 
-solversize(problem::Union{IDAEProblem,LDAEProblem}, method::VSPARK) =
+solversize(problem::AbstractProblemIDAE, method::VSPARK) =
     3 * ndims(problem) * nstages(method) + 3 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_vspark_primary.jl b/src/spark/integrators_vspark_primary.jl
@@ -78,7 +78,7 @@ end
 
 tableau(method::VSPARKprimary) = method.tableau
 
-solversize(problem::Union{IDAEProblem,LDAEProblem}, method::VSPARKprimary) =
+solversize(problem::AbstractProblemIDAE, method::VSPARKprimary) =
     2 * ndims(problem) * nstages(method) + 2 * ndims(problem) * pstages(method)
 
 

diff --git a/src/spark/integrators_vspark_secondary.jl b/src/spark/integrators_vspark_secondary.jl
@@ -42,7 +42,7 @@ pstages(method::VSPARKsecondary) = method.r
 hasnullvector(method::VSPARKsecondary{DT,Nothing}) where {DT} = false
 hasnullvector(method::VSPARKsecondary{DT,<:AbstractVector}) where {DT} = true
 
-solversize(problem::Union{IDAEProblem,LDAEProblem}, method::VSPARKsecondary) =
+solversize(problem::AbstractProblemIDAE, method::VSPARKsecondary) =
     4 * ndims(problem) * pstages(method)