revive-llvm-context 1.2.0

Shared front end code of the revive PolkaVM compilers
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

//! The LLVM function Yul data.

use std::collections::HashMap;

use num::BigUint;

/// The LLVM function Yul data.
/// Describes some data that is only relevant to Yul.
#[derive(Debug)]
pub struct YulData {
    /// The constants saved to variables. Used for peculiar cases like call simulation.
    /// It is a partial implementation of the constant propagation.
    constants: HashMap<String, BigUint>,
}

impl Default for YulData {
    fn default() -> Self {
        Self {
            constants: HashMap::with_capacity(Self::CONSTANTS_HASHMAP_INITIAL_CAPACITY),
        }
    }
}

impl YulData {
    /// The constants hashmap default capacity.
    const CONSTANTS_HASHMAP_INITIAL_CAPACITY: usize = 16;

    /// A shortcut constructor.
    pub fn new() -> Self {
        Self::default()
    }

    /// Returns a constant if it has been saved.
    pub fn get_constant(&self, name: &str) -> Option<BigUint> {
        self.constants.get(name).cloned()
    }

    /// Saves a constant detected with the partial constant propagation.
    pub fn insert_constant(&mut self, name: String, value: BigUint) {
        self.constants.insert(name, value);
    }
}