Add readable_code

Author: YingboMa

src/systems/abstractsystem.jl

@@ -506,7 +506,7 @@ function toexpr(sys::AbstractSystem)
     striplines(expr) # keeping the line numbers is never helpful
 end
 
-Base.write(io::IO, sys::AbstractSystem) = write(io, JuliaFormatter.format_text(string(toexpr(sys))))
+Base.write(io::IO, sys::AbstractSystem) = write(io, readable_code(toexpr(sys)))
 
 function Base.show(io::IO, ::MIME"text/plain", sys::AbstractSystem)
     eqs = equations(sys)

src/utils.jl

@@ -84,3 +84,17 @@ function indepvar2depvar(s::Sym, args...)
     ns = Sym{T}(nameof(s))(args...)
     @set! ns.metadata = s.metadata
 end
+
+function _readable_code(ex)
+    ex isa Expr || return ex
+    if ex.head === :call
+        f, args = ex.args[1], ex.args[2:end]
+        if f isa Function && (nf = nameof(f); Base.isoperator(nf))
+            return Expr(:call, nf, args...)
+        end
+    end
+    expr = Expr(ex.head)
+    expr.args = map(_readable_code, ex.args)
+    Base.remove_linenums!(expr)
+end
+readable_code(expr) = JuliaFormatter.format_text(string(_readable_code(expr)))