lean4-htt/src/Lean/Util/PPExt.lean
2020-06-25 11:21:17 -07:00

37 lines
1.3 KiB
Text

/-
Copyright (c) 2020 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
-/
import Lean.Environment
import Lean.MetavarContext
namespace Lean
abbrev PPExprFn := Environment → MetavarContext → LocalContext → Options → Expr → Format
/- TODO: delete after we implement the new pretty printer in Lean -/
@[extern "lean_pp_expr"]
constant ppOld : Environment → MetavarContext → LocalContext → Options → Expr → Format := arbitrary _
def mkPPExprFnRef : IO (IO.Ref PPExprFn) := IO.mkRef ppOld
@[init mkPPExprFnRef] def PPExprFnRef : IO.Ref PPExprFn := arbitrary _
def mkPPExprFnExtension : IO (EnvExtension PPExprFn) :=
registerEnvExtension $ PPExprFnRef.get
@[init mkPPExprFnExtension]
constant ppExprExt : EnvExtension PPExprFn := arbitrary _
def ppExpr (env : Environment) (mctx : MetavarContext) (lctx : LocalContext) (opts : Options) (e : Expr) : Format :=
let e := (mctx.instantiateMVars e).1;
if opts.getBool `ppOld true then
(ppExprExt.getState env) env mctx lctx opts e
else
format (toString e)
-- TODO: remove after we remove ppOld
@[init] def ppOldOption : IO Unit :=
registerOption `ppOld { defValue := true, group := "", descr := "disable/enable old pretty printer" }
end Lean