mathic 0.2.0

A compiler with builtin support of symbolic operations, built with LLVM/MLIR
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

use crate::{lowering::ir::symbols::TypeIndex, parser::Span};

#[derive(Debug, Clone)]
pub enum Adt {
    Struct(StructAdt),
}

#[derive(Debug, Clone)]
pub struct StructAdt {
    pub name: String,
    pub fields: Vec<StructField>,
    pub _span: Span,
}

#[derive(Debug, Clone)]
pub struct StructField {
    pub name: String,
    pub ty: TypeIndex,
    pub _is_pub: bool,
}

impl Adt {
    pub fn fields_len(&self) -> usize {
        match self {
            Adt::Struct(s) => s.fields.len(),
        }
    }

    pub fn get_field_names(&self) -> Vec<String> {
        match self {
            Adt::Struct(s) => s.fields.iter().map(|f| f.name.clone()).collect(),
        }
    }

    pub fn get_field_index(&self, name: &str) -> Option<usize> {
        match self {
            Adt::Struct(s) => s.fields.iter().position(|f| f.name == name),
        }
    }

    pub fn get_field_ty(&self, name: &str) -> Option<TypeIndex> {
        match self {
            Adt::Struct(s) => s.fields.iter().find(|f| f.name == name).map(|f| f.ty),
        }
    }

    pub fn get_fields_tys(&self) -> Vec<TypeIndex> {
        match self {
            Adt::Struct(s) => s.fields.iter().map(|f| f.ty).collect(),
        }
    }
}