cairo_lang_sierra/extensions/modules/int/
unsigned512.rs

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
use super::unsigned128::{U128MulGuaranteeType, Uint128Type};
use crate::define_libfunc_hierarchy;
use crate::extensions::lib_func::{
    LibfuncSignature, OutputVarInfo, ParamSignature, SierraApChange, SignatureSpecializationContext,
};
use crate::extensions::modules::get_u256_type;
use crate::extensions::non_zero::nonzero_ty;
use crate::extensions::range_check::RangeCheckType;
use crate::extensions::structure::StructType;
use crate::extensions::{
    NamedType, NoGenericArgsGenericLibfunc, OutputVarReferenceInfo, SpecializationError,
};
use crate::ids::{ConcreteTypeId, UserTypeId};
use crate::program::GenericArg;

define_libfunc_hierarchy! {
    pub enum Uint512Libfunc {
        DivModU256(Uint512DivmodU256Libfunc),
    }, Uint512Concrete
}

// Divmod.
#[derive(Default)]
pub struct Uint512DivmodU256Libfunc;
impl NoGenericArgsGenericLibfunc for Uint512DivmodU256Libfunc {
    const STR_ID: &'static str = "u512_safe_divmod_by_u256";

    fn specialize_signature(
        &self,
        context: &dyn SignatureSpecializationContext,
    ) -> Result<LibfuncSignature, SpecializationError> {
        let u256_ty = get_u256_type(context)?;
        let u512_ty = get_u512_type(context)?;
        let range_check_type = context.get_concrete_type(RangeCheckType::id(), &[])?;
        let guarantee_ty = context.get_concrete_type(U128MulGuaranteeType::id(), &[])?;
        let guarantee_output_info =
            OutputVarInfo { ty: guarantee_ty, ref_info: OutputVarReferenceInfo::SimpleDerefs };
        Ok(LibfuncSignature::new_non_branch_ex(
            vec![
                ParamSignature::new(range_check_type.clone()).with_allow_add_const(),
                ParamSignature::new(u512_ty.clone()),
                ParamSignature::new(nonzero_ty(context, &u256_ty)?),
            ],
            vec![
                OutputVarInfo::new_builtin(range_check_type, 0),
                OutputVarInfo { ty: u512_ty, ref_info: OutputVarReferenceInfo::SimpleDerefs },
                OutputVarInfo { ty: u256_ty, ref_info: OutputVarReferenceInfo::SimpleDerefs },
                guarantee_output_info.clone(),
                guarantee_output_info.clone(),
                guarantee_output_info.clone(),
                guarantee_output_info.clone(),
                guarantee_output_info,
            ],
            SierraApChange::Known { new_vars_only: false },
        ))
    }
}

/// Helper for u512 type def.
fn get_u512_type(
    context: &dyn SignatureSpecializationContext,
) -> Result<ConcreteTypeId, SpecializationError> {
    let u128_ty = context.get_concrete_type(Uint128Type::id(), &[])?;
    context.get_concrete_type(StructType::id(), &[
        GenericArg::UserType(UserTypeId::from_string("core::integer::u512")),
        GenericArg::Type(u128_ty.clone()),
        GenericArg::Type(u128_ty.clone()),
        GenericArg::Type(u128_ty.clone()),
        GenericArg::Type(u128_ty),
    ])
}