Add action to show system structure

Author: serenity4

ext/DAECompilerCthulhuExt.jl

@@ -1,7 +1,7 @@
 module DAECompilerCthulhuExt
 
 using Core.IR
-using DAECompiler: DAECompiler, DAEIPOResult, UncompilableIPOResult, Settings, ADAnalyzer, structural_analysis!, find_matching_ci, StructureCache, ir_to_src, get_method_instance, MappingInfo, AnalyzedSource
+using DAECompiler: DAECompiler, DAEIPOResult, UncompilableIPOResult, Settings, ADAnalyzer, structural_analysis!, find_matching_ci, matched_system_structure, StructureCache, ir_to_src, get_method_instance, MappingInfo, AnalyzedSource
 using Compiler: Compiler, InferenceResult, NativeInterpreter, SOURCE_MODE_GET_SOURCE, get_inference_world, typeinf_ext, Effects, get_ci_mi, NoCallInfo
 using Accessors: setproperties
 using Diffractor: FRuleCallInfo
@@ -105,9 +105,21 @@ function Cthulhu.menu_commands(provider::DAEProvider)
     commands = Cthulhu.default_menu_commands(provider)
     filter!(x -> !in(x.name, (:optimize, :dump_params, :llvm, :native)), commands)
     push!(commands, toggle_setting(provider, 'f', :force_inline_all, "force inline all"))
+    push!(commands, Cthulhu.perform_action(show_mss, 'm', :show_mss, :actions, "Show system structure"))
     return commands
 end
 
+function show_mss(state::CthulhuState)
+    result = state.ci.inferred::DAEIPOResult
+    terminal = state.terminal
+    io = terminal.out_stream::IO
+    mss = matched_system_structure(result, state.provider.settings.mode)
+    (_, width) = displaysize(terminal)
+    printstyled(io, '\n', '-'^((width - 26) ÷ 2), " Showing system structure ", '-'^((width - 26) ÷ 2), '\n'; color = :light_black)
+    show(io, MIME"text/plain"(), mss)
+    printstyled(io, '\n', '-'^width, "\n\n"; color = :light_black)
+end
+
 function toggle_setting(provider::DAEProvider, key::Char, name::Symbol, description::String = string(name))
   callback = state -> toggle_setting!(state, name)
   Command(callback, key, name, description, :toggles)

src/analysis/cache.jl

@@ -115,3 +115,20 @@ function make_structure_from_ipo(ipo::DAEIPOResult)
 
     structure = DAESystemStructure(StateSelection.complete(var_to_diff), StateSelection.complete(eq_to_diff), graph, solvable_graph)
 end
+
+function matched_system_structure(result::DAEIPOResult, mode)
+  structure = make_structure_from_ipo(result)
+
+  tstate = TransformationState(result, structure)
+  err = StateSelection.check_consistency(tstate, nothing)
+  err !== nothing && throw(err)
+
+  ret = top_level_state_selection!(tstate)
+  isa(ret, UncompilableIPOResult) && throw(ret.error)
+
+  (diff_key, init_key) = ret
+  key = in(mode, (DAE, DAENoInit, ODE, ODENoInit)) ? diff_key : init_key
+
+  var_eq_matching = matching_for_key(tstate, key)
+  return StateSelection.MatchedSystemStructure(result, structure, var_eq_matching)
+end

src/reflection.jl

@@ -32,22 +32,7 @@ function code_structure_by_type(@nospecialize(tt::Type); world::UInt = Base.tls_
   _result = structural_analysis!(ci, world, settings)
   isa(_result, UncompilableIPOResult) && throw(_result.error)
   !matched && return result ? _result : _result.ir
-  result = _result
-
-  structure = make_structure_from_ipo(result)
-
-  tstate = TransformationState(result, structure)
-  err = StateSelection.check_consistency(tstate, nothing)
-  err !== nothing && throw(err)
-
-  ret = top_level_state_selection!(tstate)
-  isa(ret, UncompilableIPOResult) && throw(ret.error)
-
-  (diff_key, init_key) = ret
-  key = in(mode, (DAE, DAENoInit, ODE, ODENoInit)) ? diff_key : init_key
-
-  var_eq_matching = matching_for_key(tstate, key)
-  return StateSelection.MatchedSystemStructure(result, structure, var_eq_matching)
+  return matched_system_structure(_result, mode)
 end
 
 """