runmat-mir 0.5.0

Mid-level IR and analysis scaffolding for RunMat
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use crate::{MirPlace, MirStmt};
use runmat_hir::{HirError, HirExpr, HirExprKind, HirPlace};

use super::{
    expr::{lower_indexing, lower_operand},
    MirLoweringContext,
};

pub(crate) fn lower_place(
    ctx: &MirLoweringContext,
    place: &HirPlace,
    temps: &mut Vec<MirStmt>,
) -> Result<MirPlace, HirError> {
    Ok(match place {
        HirPlace::Binding(binding) => MirPlace::Local(ctx.local_for_binding(*binding)?),
        HirPlace::Member(base, member) => MirPlace::Member(
            Box::new(lower_expr_place(ctx, base, temps)?),
            member.clone(),
        ),
        HirPlace::MemberDynamic(base, member) => MirPlace::DynamicMember(
            Box::new(lower_expr_place(ctx, base, temps)?),
            lower_operand(ctx, member, temps)?,
        ),
        HirPlace::Index(base, indexing) | HirPlace::IndexCell(base, indexing) => MirPlace::Index(
            Box::new(lower_expr_place(ctx, base, temps)?),
            lower_indexing(ctx, indexing, temps)?,
        ),
    })
}

fn lower_expr_place(
    ctx: &MirLoweringContext,
    expr: &HirExpr,
    temps: &mut Vec<MirStmt>,
) -> Result<MirPlace, HirError> {
    let lowered = match &expr.kind {
        HirExprKind::Binding(binding) => MirPlace::Local(ctx.local_for_binding(*binding)?),
        HirExprKind::Member(base, member) => MirPlace::Member(
            Box::new(lower_expr_place(ctx, base, temps)?),
            member.clone(),
        ),
        HirExprKind::MemberDynamic(base, member) => MirPlace::DynamicMember(
            Box::new(lower_expr_place(ctx, base, temps)?),
            lower_operand(ctx, member, temps)?,
        ),
        HirExprKind::Index(base, indexing) => MirPlace::Index(
            Box::new(lower_expr_place(ctx, base, temps)?),
            lower_indexing(ctx, indexing, temps)?,
        ),
        _ => {
            let operand = lower_operand(ctx, expr, temps)?;
            match operand {
                crate::MirOperand::Local(local) => MirPlace::Local(local),
                _ => return Err(HirError::new("expression is not a simple MIR place")),
            }
        }
    };
    Ok(lowered)
}