use super::{Attribute, AttributeLike};
use crate::{
ir::{Type, TypeLike},
Error,
};
use mlir_sys::{
mlirIntegerAttrGet, mlirIntegerAttrGetValueInt, mlirIntegerAttrGetValueSInt,
mlirIntegerAttrGetValueUInt, MlirAttribute,
};
#[derive(Clone, Copy)]
pub struct IntegerAttribute<'c> {
attribute: Attribute<'c>,
}
impl<'c> IntegerAttribute<'c> {
pub fn new(r#type: Type<'c>, integer: i64) -> Self {
unsafe { Self::from_raw(mlirIntegerAttrGet(r#type.to_raw(), integer)) }
}
pub fn value(&self) -> i64 {
unsafe { mlirIntegerAttrGetValueInt(self.to_raw()) }
}
pub fn signed_value(&self) -> i64 {
unsafe { mlirIntegerAttrGetValueSInt(self.to_raw()) }
}
pub fn unsigned_value(&self) -> u64 {
unsafe { mlirIntegerAttrGetValueUInt(self.to_raw()) }
}
}
attribute_traits!(IntegerAttribute, is_integer, "integer");
#[cfg(test)]
mod tests {
use super::*;
use crate::{ir::r#type::IntegerType, test::create_test_context};
#[test]
fn value() {
let context = create_test_context();
assert_eq!(
IntegerAttribute::new(IntegerType::new(&context, 64).into(), 42).value(),
42
);
}
}