Add get_model_nodes

Author: VEZY

src/PlantSimEngine.jl

@@ -58,6 +58,7 @@ include("dependencies/traversal.jl")
 include("dependencies/is_graph_cyclic.jl")
 include("dependencies/printing.jl")
 include("dependencies/dependencies.jl")
+include("dependencies/get_model_in_dependency_graph.jl")
 
 # MTG compatibility:
 include("mtg/GraphSimulation.jl")

src/dependencies/get_model_in_dependency_graph.jl

@@ -0,0 +1,31 @@
+"""
+    get_model_nodes(dep_graph::DependencyGraph, model)
+
+Get the nodes in the dependency graph implementing a type of model.
+
+# Arguments
+
+- `dep_graph::DependencyGraph`: the dependency graph.
+- `model`: the model type to look for.
+
+# Returns
+
+- An array of nodes implementing the model type.
+
+# Examples
+
+```julia
+PlantSimEngine.get_model_nodes(dependency_graph, Beer)
+```
+"""
+function get_model_nodes(dep_graph::DependencyGraph, model)
+    model_node = Union{SoftDependencyNode,HardDependencyNode}[]
+
+    traverse_dependency_graph!(dep_graph) do node
+        if isa(node.value, model)
+            push!(model_node, node)
+        end
+    end
+
+    return model_node
+end

src/dependencies/traversal.jl

@@ -47,12 +47,13 @@ function traverse_dependency_graph(
 end
 
 
-function traverse_dependency_graph!(
-    f::Function,
-    node::SoftDependencyNode;
-    visit_hard_dep=true
-)
+function traverse_dependency_graph!(f::Function, graph::DependencyGraph; visit_hard_dep=true)
+    for (p, root) in graph.roots
+        traverse_dependency_graph!(f, root, visit_hard_dep=visit_hard_dep)
+    end
+end
 
+function traverse_dependency_graph!(f::Function, node::SoftDependencyNode; visit_hard_dep=true)
     f(node)
     # Traverse the hard dependencies of the SoftDependencyNode if any:
     if visit_hard_dep && node isa SoftDependencyNode
@@ -67,12 +68,7 @@ function traverse_dependency_graph!(
     end
 end
 
-function traverse_dependency_graph!(
-    f::Function,
-    node::HardDependencyNode;
-    visit_hard_dep=true
-)
-
+function traverse_dependency_graph!(f::Function, node::HardDependencyNode; visit_hard_dep=true)
     f(node)
     # Traverse all hard dependencies:
     for child in node.children