forked from M-Labs/nac3
core/model: add ListModel
This commit is contained in:
parent
cd777dcb52
commit
86ed0140cb
|
@ -0,0 +1,42 @@
|
|||
use crate::codegen::{model::*, CodeGenContext};
|
||||
|
||||
/// Fields of [`List`]
|
||||
pub struct ListFields<'ctx, T: Model<'ctx>> {
|
||||
/// Length of the list
|
||||
pub size: Field<SizeTModel<'ctx>>,
|
||||
/// Base pointer of the list
|
||||
pub data: Field<PointerModel<T>>,
|
||||
}
|
||||
|
||||
/// nac3core's `List` definition
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||
pub struct List<'ctx, T: Model<'ctx>> {
|
||||
pub sizet: SizeTModel<'ctx>,
|
||||
pub element: T,
|
||||
}
|
||||
|
||||
impl<'ctx, T: Model<'ctx> + 'ctx> StructKind<'ctx> for List<'ctx, T> {
|
||||
type Fields = ListFields<'ctx, T>;
|
||||
|
||||
fn struct_name(&self) -> &'static str {
|
||||
"List"
|
||||
}
|
||||
|
||||
fn build_fields(&self, builder: &mut FieldBuilder<'ctx>) -> Self::Fields {
|
||||
Self::Fields {
|
||||
size: builder.add_field("size", self.sizet),
|
||||
data: builder.add_field("data", PointerModel(self.element)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<'ctx, T: Model<'ctx> + 'ctx> Pointer<'ctx, StructModel<List<'ctx, T>>> {
|
||||
pub fn as_slice(
|
||||
&self,
|
||||
ctx: &CodeGenContext<'ctx, '_>,
|
||||
) -> ArraySlice<'ctx, SizeTModel<'ctx>, T> {
|
||||
let num_elements = self.gep(ctx, |f| f.size).load(ctx, "num_elements");
|
||||
let pointer = self.gep(ctx, |f| f.data).load(ctx, "base");
|
||||
ArraySlice { num_elements, pointer }
|
||||
}
|
||||
}
|
|
@ -1,3 +1,4 @@
|
|||
pub mod cslice;
|
||||
pub mod exception;
|
||||
pub mod list;
|
||||
pub mod ndarray;
|
||||
|
|
Loading…
Reference in New Issue