modified state dep graph 1

Author: isaacsas

src/systems/dependency_graphs.jl

@@ -6,7 +6,7 @@ function equation_dependencies(sys::AbstractSystem; variables=states(sys))
     depeqs_to_vars = Vector{Vector{Variable}}(undef,length(eqs))
 
     for (i,eq) in enumerate(eqs)      
-        depeqs_to_vars[i] = collect(get_variables!(deps, eq, variables))
+        depeqs_to_vars[i] = collect(equation_dependencies!(deps, eq, variables))
         empty!(deps)
     end
 

src/systems/jumps/jumpsystem.jl

@@ -128,12 +128,12 @@ end
 
 
 ### Functions to determine which states a jump depends on
-function get_variables!(dep, jump::Union{ConstantRateJump,VariableRateJump}, variables)
+function equation_dependencies!(dep, jump::Union{ConstantRateJump,VariableRateJump}, variables)
     foreach(var -> (var in variables) && push!(dep, var), vars(jump.rate))
     dep
 end
 
-function get_variables!(dep, jump::MassActionJump, variables)
+function equation_dependencies!(dep, jump::MassActionJump, variables)
     jsr = jump.scaled_rates
 
     if jsr isa Variable
@@ -150,6 +150,17 @@ function get_variables!(dep, jump::MassActionJump, variables)
     dep
 end
 
-
 ### Functions to determine which states are modified by a given jump
+function modified_states!(mstates, jump::Union{ConstantRateJump,VariableRateJump}, sts)
+    for eq in jump.affect!
+        st = convert(Variable, eq.lhs)
+        (st in sts) && push!(mstates, st)
+    end
+end
 
+function modified_states!(mstates, jump::MassActionJump, sts)
+    for (state,stoich) in jump.net_stoch
+        st = convert(Variable, state)
+        (st in sts) && push!(mstates, st)
+    end
+end