Skip to content

Commit b48aeb3

Browse files
committed
add alias_noise and alias_jumps where appropriate
1 parent 30a9428 commit b48aeb3

File tree

3 files changed

+17
-12
lines changed

3 files changed

+17
-12
lines changed

src/problems/rode_problems.jl

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ When a keyword argument is `nothing`, the default behaviour of the solver is use
106106
* `alias_u0::Union{Bool, Nothing}`: alias the u0 array. Defaults to false .
107107
* `alias_du0::Union{Bool, Nothing}`: alias the du0 array for DAEs. Defaults to false.
108108
* `alias_tstops::Union{Bool, Nothing}`: alias the tstops array
109+
* `alias_noise::Union{Bool,Nothing}`: alias the noise process
109110
* `alias::Union{Bool, Nothing}`: sets all fields of the `RODEAliasSpecifier` to `alias`
110111
111112
"""
@@ -116,15 +117,17 @@ struct RODEAliasSpecifier <: AbstractAliasSpecifier
116117
alias_u0::Union{Bool, Nothing}
117118
alias_du0::Union{Bool, Nothing}
118119
alias_tstops::Union{Bool, Nothing}
120+
alias_noise::Union{Bool, Nothing}
121+
alias_jumps::Union{Bool, Nothing}
119122

120123
function RODEAliasSpecifier(; alias_p = nothing, alias_f = nothing, alias_u0 = nothing,
121-
alias_du0 = nothing, alias_tstops = nothing, alias = nothing)
124+
alias_du0 = nothing, alias_tstops = nothing, alias_noise = nothing, alias_jumps = nothing, alias = nothing)
122125
if alias == true
123-
new(true, true, true, true, true)
126+
new(true, true, true, true, true, true, true)
124127
elseif alias == false
125-
new(false, false, false, false, false)
128+
new(false, false, false, false, false, false, false)
126129
elseif isnothing(alias)
127-
new(alias_p, alias_f, alias_u0, alias_du0, alias_tstops)
130+
new(alias_p, alias_f, alias_u0, alias_du0, alias_tstops, alias_noise, alias_jumps)
128131
end
129132
end
130133
end

src/problems/sdde_problems.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -195,15 +195,16 @@ struct SDDEAliasSpecifier
195195
alias_f::Union{Bool, Nothing}
196196
alias_u0::Union{Bool, Nothing}
197197
alias_tstops::Union{Bool, Nothing}
198+
alias_jumps::Union{Bool, Nothing}
198199

199200
function SDDEAliasSpecifier(; alias_p = nothing, alias_f = nothing, alias_u0 = nothing,
200-
alias_du0 = nothing, alias_tstops = nothing, alias = nothing)
201+
alias_du0 = nothing, alias_tstops = nothing, alias_jumps = nothing, alias = nothing)
201202
if alias == true
202-
new(true, true, true, true)
203+
new(true, true, true, true, true)
203204
elseif alias == false
204-
new(false, false, false, false)
205+
new(false, false, false, false, false)
205206
elseif isnothing(alias)
206-
new(alias_p, alias_f, alias_u0, alias_tstops)
207+
new(alias_p, alias_f, alias_u0, alias_tstops, alias_jumps)
207208
end
208209
end
209210
end

src/problems/sde_problems.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -237,15 +237,16 @@ struct SDEAliasSpecifier <: AbstractAliasSpecifier
237237
alias_f::Union{Bool, Nothing}
238238
alias_u0::Union{Bool, Nothing}
239239
alias_tstops::Union{Bool, Nothing}
240+
alias_jumps::Union{Bool,Nothing}
240241

241242
function SDEAliasSpecifier(; alias_p = nothing, alias_f = nothing, alias_u0 = nothing,
242-
alias_du0 = nothing, alias_tstops = nothing, alias = nothing)
243+
alias_du0 = nothing, alias_tstops = nothing, alias_jumps = nothing, alias = nothing)
243244
if alias == true
244-
new(true, true, true, true)
245+
new(true, true, true, true, true)
245246
elseif alias == false
246-
new(false, false, false, false)
247+
new(false, false, false, false, false)
247248
elseif isnothing(alias)
248-
new(alias_p, alias_f, alias_u0, alias_tstops)
249+
new(alias_p, alias_f, alias_u0, alias_tstops, alias_jumps)
249250
end
250251
end
251252
end

0 commit comments

Comments
 (0)