xrcf 0.1.0

A compiler framework to enable the rapid development of programming language compilers
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use crate::convert::apply_rewrites;
use crate::convert::ChangedOp;
use crate::convert::Rewrite;
use crate::convert::RewriteResult;
use crate::dialect::arith;
use crate::dialect::func;
use crate::dialect::llvmir;
use crate::ir::Op;
use crate::Pass;
use anyhow::Result;
use std::sync::Arc;
use std::sync::RwLock;

struct FuncLowering;

impl Rewrite for FuncLowering {
    fn is_match(&self, op: Arc<RwLock<dyn Op>>) -> Result<bool> {
        Ok(op
            .try_read()
            .unwrap()
            .as_any()
            .downcast_ref::<func::FuncOp>()
            .is_some())
    }
    fn rewrite(&self, op: Arc<RwLock<dyn Op>>) -> Result<RewriteResult> {
        let op = op.try_read().unwrap();
        let op = op.as_any().downcast_ref::<func::FuncOp>().unwrap();
        {
            let operation = op.operation();
            let mut operation = operation.try_write().unwrap();
            let name = operation.name();
            assert!(name == func::FuncOp::operation_name());
            operation.set_name(llvmir::FuncOp::operation_name());

            let parent = operation.parent();
            assert!(
                parent.is_some(),
                "maybe func parent was not set during parsing?"
            );
        }

        let mut lowered = llvmir::FuncOp::from_operation(op.operation().clone())?;
        lowered.set_identifier(op.identifier().to_string());
        let new_op = Arc::new(RwLock::new(lowered));
        op.replace(new_op.clone());

        Ok(RewriteResult::Changed(ChangedOp::new(new_op)))
    }
}

struct ConstantOpLowering;

impl Rewrite for ConstantOpLowering {
    fn is_match(&self, op: Arc<RwLock<dyn Op>>) -> Result<bool> {
        Ok(op
            .try_read()
            .unwrap()
            .as_any()
            .downcast_ref::<arith::ConstantOp>()
            .is_some())
    }
    fn rewrite(&self, op: Arc<RwLock<dyn Op>>) -> Result<RewriteResult> {
        let op = op.try_read().unwrap();
        let op = op.as_any().downcast_ref::<arith::ConstantOp>().unwrap();
        let lowered = llvmir::ConstantOp::from_operation(op.operation().clone())?;
        let new_op = Arc::new(RwLock::new(lowered));
        op.replace(new_op.clone());

        Ok(RewriteResult::Changed(ChangedOp::new(new_op)))
    }
}

struct ReturnLowering;

impl Rewrite for ReturnLowering {
    fn is_match(&self, op: Arc<RwLock<dyn Op>>) -> Result<bool> {
        Ok(op
            .try_read()
            .unwrap()
            .as_any()
            .downcast_ref::<func::ReturnOp>()
            .is_some())
    }
    fn rewrite(&self, op: Arc<RwLock<dyn Op>>) -> Result<RewriteResult> {
        let op = op.try_read().unwrap();
        let op = op.as_any().downcast_ref::<func::ReturnOp>().unwrap();
        let lowered = llvmir::ReturnOp::from_operation(op.operation().clone())?;
        let new_op = Arc::new(RwLock::new(lowered));
        op.replace(new_op.clone());

        Ok(RewriteResult::Changed(ChangedOp::new(new_op)))
    }
}

pub struct ConvertFuncToLLVM;

impl Pass for ConvertFuncToLLVM {
    fn name() -> &'static str {
        "convert-func-to-llvm"
    }
    fn convert(op: Arc<RwLock<dyn Op>>) -> Result<RewriteResult> {
        let rewrites: Vec<&dyn Rewrite> = vec![&FuncLowering, &ConstantOpLowering, &ReturnLowering];
        apply_rewrites(op, &rewrites)
    }
}