induced subnet

src/AlgebraicPetri.jl

@@ -5,7 +5,7 @@ module AlgebraicPetri
 export SchPetriNet, PetriNet, OpenPetriNetOb, AbstractPetriNet,
   ns, nt, ni, no, os, ot, is, it,
   add_species!, add_transition!, add_transitions!,
-  add_input!, add_inputs!, add_output!, add_outputs!, inputs, outputs,
+  add_input!, add_inputs!, add_output!, add_outputs!, inputs, outputs, induced_subnet,
   TransitionMatrices, vectorfield, vectorfield_expr, flatten_labels,
   SchLabelledPetriNet, LabelledPetriNet, AbstractLabelledPetriNet, sname, tname, snames, tnames,
   SchReactionNet, ReactionNet, AbstractReactionNet, concentration, concentrations, rate, rates,
@@ -272,6 +272,21 @@ inputs(p::AbstractPetriNet, t) = subpart(p, incident(p, t, :it), :is)
 """
 outputs(p::AbstractPetriNet, t) = subpart(p, incident(p, t, :ot), :os)
 
+""" induced_subnet(p::T, t)
+
+Return the induced subnet consisting of the transition(s) in `t`, all input
+places and input arcs, and all output places and output arcs. The returned
+object will be of the same type as `p`.
+"""
+function induced_subnet(p::T, t) where {T <: AbstractPetriNet}
+  induced_subnet(p, [t])
+end
+
+function induced_subnet(p::T, t::AbstractVector{Int}) where {T <: AbstractPetriNet}
+  subnet = Subobject(p, T=t)
+  return dom(hom(~(¬subnet)))
+end
+
 valueat(x::Number, u, t) = x
 valueat(f::Function, u, t) =
   try

test/algebraicpetri/types.jl

@@ -176,4 +176,57 @@ for p in [sir_proplpetri, sir_proplrxn]
   @test Open(p, [:S], [:I], [:R]) == Open(p)
 end
 
+# test subnet for each type of net
+for net in [sir_petri, sir_rxn]
+  si_net = induced_subnet(net, 1)
+
+  @test ns(si_net) == 2
+  @test nt(si_net) == 1
+  @test is(si_net) == [1,2]
+  @test os(si_net) == [2,2]
+
+  r_net = induced_subnet(net, 2)
+
+  @test ns(r_net) == 2
+  @test nt(r_net) == 1
+  @test is(r_net) == [1]
+  @test os(r_net) == [2]
+end
+
+for net in [sir_lpetri, sir_lrxn]
+  si_net = induced_subnet(sir_lpetri, 1)
+
+  @test ns(si_net) == 2
+  @test nt(si_net) == 1
+  @test is(si_net) == [1,2]
+  @test os(si_net) == [2,2]
+  @test only(si_net[:tname]) == :inf
+  @test si_net[:sname] == [:S,:I]
+
+  r_net = induced_subnet(sir_lpetri, 2)
+
+  @test ns(r_net) == 2
+  @test nt(r_net) == 1
+  @test is(r_net) == [1]
+  @test os(r_net) == [2]
+  @test only(r_net[:tname]) == :rec
+  @test r_net[:sname] == [:I,:R]  
+end
+
+sir_net = induced_subnet(sir_petri, [1,2])
+sir_net == sir_petri
+
+test_petri = PetriNet(
+  6,
+  1 => (3,4),
+  1 => 2,
+  5 => 6
+)
+
+test_net = induced_subnet(test_petri, [1,2])
+@test ns(test_net) == 4
+@test nt(test_net) == 2
+@test ni(test_net) == 2
+@test no(test_net) == 3
+
 end