Make introspecting dummy_derivative easier

Author: YingboMa

src/structural_transformation/bipartite_tearing/modia_tearing.jl

@@ -82,6 +82,7 @@ function tear_graph_modia(structure::SystemStructure, isder::F = nothing,
     var_eq_matching = complete(var_eq_matching,
         max(length(var_eq_matching),
             maximum(x -> x isa Int ? x : 0, var_eq_matching)))
+    full_var_eq_matching = copy(var_eq_matching)
     var_sccs::Vector{Union{Vector{Int}, Int}} = find_var_sccs(graph, var_eq_matching)
     vargraph = DiCMOBiGraph{true}(graph)
     ict = IncrementalCycleTracker(vargraph; dir = :in)
@@ -110,6 +111,5 @@ function tear_graph_modia(structure::SystemStructure, isder::F = nothing,
         empty!(ieqs)
         empty!(filtered_vars)
     end
-
-    return var_eq_matching
+    return var_eq_matching, full_var_eq_matching
 end

src/structural_transformation/partial_state_selection.jl

@@ -168,15 +168,15 @@ function partial_state_selection_graph!(structure::SystemStructure, var_eq_match
 end
 
 function dummy_derivative_graph!(state::TransformationState, jac = nothing;
-    state_priority = nothing, kwargs...)
+    state_priority = nothing, log = Val(false), kwargs...)
     state.structure.solvable_graph === nothing && find_solvables!(state; kwargs...)
     complete!(state.structure)
     var_eq_matching = complete(pantelides!(state))
-    dummy_derivative_graph!(state.structure, var_eq_matching, jac, state_priority)
+    dummy_derivative_graph!(state.structure, var_eq_matching, jac, state_priority, log)
 end
 
 function dummy_derivative_graph!(structure::SystemStructure, var_eq_matching, jac,
-    state_priority)
+    state_priority, ::Val{log} = Val(false)) where {log}
     @unpack eq_to_diff, var_to_diff, graph = structure
     diff_to_eq = invview(eq_to_diff)
     diff_to_var = invview(var_to_diff)
@@ -338,7 +338,7 @@ function dummy_derivative_graph!(structure::SystemStructure, var_eq_matching, ja
         end
     end
 
-    var_eq_matching = tear_graph_modia(structure, isdiffed,
+    var_eq_matching, full_var_eq_matching = tear_graph_modia(structure, isdiffed,
         Union{Unassigned, SelectedState};
         varfilter = can_eliminate)
     for v in eachindex(var_eq_matching)
@@ -348,5 +348,10 @@ function dummy_derivative_graph!(structure::SystemStructure, var_eq_matching, ja
         var_eq_matching[v] = SelectedState()
     end
 
-    return var_eq_matching
+    if log
+        candidates = can_eliminate.(1:ndsts(graph))
+        return var_eq_matching, full_var_eq_matching, candidates
+    else
+        return var_eq_matching
+    end
 end

src/structural_transformation/symbolics_tearing.jl

@@ -564,7 +564,7 @@ function tearing(state::TearingState; kwargs...)
     @unpack graph = state.structure
     algvars = BitSet(findall(v -> isalgvar(state.structure, v), 1:ndsts(graph)))
     aeqs = algeqs(state.structure)
-    var_eq_matching′ = tear_graph_modia(state.structure;
+    var_eq_matching′, = tear_graph_modia(state.structure;
         varfilter = var -> var in algvars,
         eqfilter = eq -> eq in aeqs)
     var_eq_matching = Matching{Union{Unassigned, SelectedState}}(var_eq_matching′)