2363fdf544
The result type of a join point is always equal to the function return type. Moreover, the extra bookkeeping introduces extra work, and doesn't really help.
107 lines
5.1 KiB
Text
107 lines
5.1 KiB
Text
/-
|
||
Copyright (c) 2019 Microsoft Corporation. All rights reserved.
|
||
Released under Apache 2.0 license as described in the file LICENSE.
|
||
Authors: Leonardo de Moura
|
||
-/
|
||
prelude
|
||
import init.lean.compiler.ir.basic init.lean.format
|
||
|
||
namespace Lean
|
||
namespace IR
|
||
|
||
private def formatArg : Arg → Format
|
||
| (Arg.var id) := format id
|
||
| Arg.irrelevant := "◾"
|
||
|
||
instance argHasFormat : HasFormat Arg := ⟨formatArg⟩
|
||
|
||
private def formatArray {α : Type} [HasFormat α] (args : Array α) : Format :=
|
||
args.foldl (λ r a, r ++ " " ++ format a) Format.nil
|
||
|
||
private def formatLitVal : LitVal → Format
|
||
| (LitVal.num v) := format v
|
||
| (LitVal.str v) := format (repr v)
|
||
|
||
instance litValHasFormat : HasFormat LitVal := ⟨formatLitVal⟩
|
||
|
||
private def formatCtorInfo : CtorInfo → Format
|
||
| { name := name, cidx := cidx, usize := usize, ssize := ssize, .. } :=
|
||
let r := format "ctor_" ++ format cidx in
|
||
let r := if usize > 0 || ssize > 0 then r ++ "." ++ format usize ++ "." ++ format ssize else r in
|
||
let r := if name != Name.anonymous then r ++ "[" ++ format name ++ "]" else r in
|
||
r
|
||
|
||
instance ctorInfoHasFormat : HasFormat CtorInfo := ⟨formatCtorInfo⟩
|
||
|
||
private def formatExpr : Expr → Format
|
||
| (Expr.ctor i ys) := format i ++ formatArray ys
|
||
| (Expr.reset x) := "reset " ++ format x
|
||
| (Expr.reuse x i u ys) := "reuse" ++ (if u then "!" else "") ++ " " ++ format x ++ " in " ++ format i ++ formatArray ys
|
||
| (Expr.proj i x) := "proj_" ++ format i ++ " " ++ format x
|
||
| (Expr.uproj i x) := "uproj_" ++ format i ++ " " ++ format x
|
||
| (Expr.sproj n o x) := "sproj_" ++ format n ++ "_" ++ format o ++ " " ++ format x
|
||
| (Expr.fap c ys) := format c ++ formatArray ys
|
||
| (Expr.pap c ys) := "pap " ++ format c ++ formatArray ys
|
||
| (Expr.ap x ys) := "app " ++ format x ++ formatArray ys
|
||
| (Expr.box _ x) := "box " ++ format x
|
||
| (Expr.unbox x) := "unbox " ++ format x
|
||
| (Expr.lit v) := format v
|
||
| (Expr.isShared x) := "isShared " ++ format x
|
||
| (Expr.isTaggedPtr x) := "isTaggedPtr " ++ format x
|
||
|
||
instance exprHasFormat : HasFormat Expr := ⟨formatExpr⟩
|
||
|
||
private def formatIRType : IRType → Format
|
||
| IRType.float := "float"
|
||
| IRType.uint8 := "u8"
|
||
| IRType.uint16 := "u16"
|
||
| IRType.uint32 := "u32"
|
||
| IRType.uint64 := "u64"
|
||
| IRType.usize := "usize"
|
||
| IRType.irrelevant := "◾"
|
||
| IRType.object := "obj"
|
||
| IRType.tobject := "tobj"
|
||
|
||
instance typeHasFormat : HasFormat IRType := ⟨formatIRType⟩
|
||
|
||
private def formatParam : Param → Format
|
||
| { x := name, borrowed := b, ty := ty } := "(" ++ format name ++ " : " ++ (if b then "@^ " else "") ++ format ty ++ ")"
|
||
|
||
instance paramHasFormat : HasFormat Param := ⟨formatParam⟩
|
||
|
||
def formatAlt (fmt : FnBody → Format) (indent : Nat) : Alt → Format
|
||
| (Alt.ctor i b) := format i.name ++ " →" ++ Format.nest indent (Format.line ++ fmt b)
|
||
| (Alt.default b) := "default →" ++ Format.nest indent (Format.line ++ fmt b)
|
||
|
||
partial def formatFnBody (indent : Nat := 2) : FnBody → Format
|
||
| (FnBody.vdecl x ty e b) := "let " ++ format x ++ " : " ++ format ty ++ " := " ++ format e ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.jdecl j xs v b) := format j ++ formatArray xs ++ " :=" ++ Format.nest indent (Format.line ++ formatFnBody v) ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.set x i y b) := "set " ++ format x ++ "[" ++ format i ++ "] := " ++ format y ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.uset x i y b) := "uset " ++ format x ++ "[" ++ format i ++ "] := " ++ format y ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.sset x i o y ty b) := "sset " ++ format x ++ "[" ++ format i ++ ", " ++ format o ++ "] : " ++ format ty ++ " := " ++ format y ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.release x i b) := "release " ++ format x ++ "[" ++ format i ++ "];" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.inc x n c b) := "inc " ++ format x ++ (if n != 1 then format n else "") ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.dec x n c b) := "dec " ++ format x ++ (if n != 1 then format n else "") ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.mdata d b) := "mdata " ++ format d ++ ";" ++ Format.line ++ formatFnBody b
|
||
| (FnBody.case tid x cs) := "case " ++ format x ++ " of" ++ cs.foldl (λ r c, r ++ Format.line ++ formatAlt formatFnBody indent c) Format.nil
|
||
| (FnBody.jmp j ys) := "jmp " ++ format j ++ formatArray ys
|
||
| (FnBody.ret x) := "ret " ++ format x
|
||
| FnBody.unreachable := "⊥"
|
||
|
||
instance fnBodyHasFormat : HasFormat FnBody := ⟨formatFnBody⟩
|
||
instance fnBodyHasToString : HasToString FnBody := ⟨λ b, (format b).pretty⟩
|
||
|
||
def formatDecl (indent : Nat := 2) : Decl → Format
|
||
| (Decl.fdecl f xs ty b) := "def " ++ format f ++ formatArray xs ++ format " : " ++ format ty ++ " :=" ++ Format.nest indent (Format.line ++ formatFnBody indent b)
|
||
| (Decl.extern f xs ty) := "extern " ++ format f ++ formatArray xs ++ format " : " ++ format ty
|
||
|
||
instance declHasFormat : HasFormat Decl := ⟨formatDecl⟩
|
||
|
||
@[export lean.ir.decl_to_string_core]
|
||
def declToString (d : Decl) : String :=
|
||
(format d).pretty
|
||
|
||
instance declHasToString : HasToString Decl := ⟨declToString⟩
|
||
|
||
end IR
|
||
end Lean
|