knave 0.14.21

A Lisp programming language for humans and agents that infers types and ownership and compiles to readable Rust
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

use rustc_hash::FxHashSet;

use crate::core::Literal;
use crate::types::{GenericId, Type, TypedExpr, TypedExprKind, TypedLeaf};

#[derive(Default)]
pub struct LiteralGenerics {
    generics: FxHashSet<GenericId>,
}

impl LiteralGenerics {
    pub fn of(value: &TypedExpr) -> FxHashSet<GenericId> {
        let mut found = Self::default();
        found.expr(value);
        found.generics
    }

    fn expr(&mut self, expr: &TypedExpr) {
        if let TypedExprKind::Leaf(TypedLeaf::Literal(Literal::Number(_))) = &expr.kind {
            self.note(&expr.ty);
        }
        for child in expr.children() {
            self.expr(child);
        }
    }

    fn note(&mut self, literal_type: &Type) {
        if let Type::Generic(generic) = literal_type {
            self.generics.insert(*generic);
        }
    }
}