melior 0.27.2

The rustic MLIR bindings in Rust
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

use crate::ir::{Block, BlockRef};
use mlir_sys::{
    MlirRegion, mlirRegionAppendOwnedBlock, mlirRegionGetFirstBlock,
    mlirRegionInsertOwnedBlockAfter, mlirRegionInsertOwnedBlockBefore,
};

/// A region-like trait.
pub trait RegionLike<'c, 'a>: 'a {
    /// Converts a region into a raw object.
    fn to_raw(&self) -> MlirRegion;

    /// Returns the first block in a region.
    fn first_block(&self) -> Option<BlockRef<'c, 'a>> {
        unsafe {
            let block = mlirRegionGetFirstBlock(self.to_raw());

            if block.ptr.is_null() {
                None
            } else {
                Some(BlockRef::from_raw(block))
            }
        }
    }

    /// Inserts a block after another block.
    fn insert_block_after(&self, one: BlockRef<'c, 'a>, other: Block<'c>) -> BlockRef<'c, 'a> {
        unsafe {
            let r#ref = BlockRef::from_raw(other.to_raw());

            mlirRegionInsertOwnedBlockAfter(self.to_raw(), one.to_raw(), other.into_raw());

            r#ref
        }
    }

    /// Inserts a block before another block.
    fn insert_block_before(&self, one: BlockRef<'c, 'a>, other: Block<'c>) -> BlockRef<'c, 'a> {
        unsafe {
            let r#ref = BlockRef::from_raw(other.to_raw());

            mlirRegionInsertOwnedBlockBefore(self.to_raw(), one.to_raw(), other.into_raw());

            r#ref
        }
    }

    /// Appends a block.
    fn append_block(&self, block: Block<'c>) -> BlockRef<'c, 'a> {
        unsafe {
            let r#ref = BlockRef::from_raw(block.to_raw());

            mlirRegionAppendOwnedBlock(self.to_raw(), block.into_raw());

            r#ref
        }
    }
}