WIP: use HdlOption[the_type_var] or UInt[123 + n] for creating types
	
		
			
	
		
	
	
		
	
		
			Some checks failed
		
		
	
	
		
			
				
	
				/ test (push) Has been cancelled
				
			
		
		
	
	
				
					
				
			
		
			Some checks failed
		
		
	
	/ test (push) Has been cancelled
				
			This commit is contained in:
		
							parent
							
								
									cd99dbc849
								
							
						
					
					
						commit
						7ff1cafb74
					
				
					 62 changed files with 13241 additions and 12608 deletions
				
			
		| 
						 | 
				
			
			@ -8,7 +8,7 @@ jobs:
 | 
			
		|||
        with:
 | 
			
		||||
          fetch-depth: 0
 | 
			
		||||
      - run: |
 | 
			
		||||
          curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --default-toolchain 1.79.0
 | 
			
		||||
          curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --default-toolchain 1.80.1
 | 
			
		||||
          source "$HOME/.cargo/env"
 | 
			
		||||
          echo "$PATH" >> "$GITHUB_PATH"
 | 
			
		||||
      - uses: https://github.com/Swatinem/rust-cache@v2
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -233,16 +233,20 @@ forward_fold!(syn::WherePredicate => fold_where_predicate);
 | 
			
		|||
no_op_fold!(syn::parse::Nothing);
 | 
			
		||||
no_op_fold!(syn::token::Brace);
 | 
			
		||||
no_op_fold!(syn::token::Bracket);
 | 
			
		||||
no_op_fold!(syn::token::Group);
 | 
			
		||||
no_op_fold!(syn::token::Paren);
 | 
			
		||||
no_op_fold!(syn::Token![_]);
 | 
			
		||||
no_op_fold!(syn::Token![,]);
 | 
			
		||||
no_op_fold!(syn::Token![;]);
 | 
			
		||||
no_op_fold!(syn::Token![:]);
 | 
			
		||||
no_op_fold!(syn::Token![::]);
 | 
			
		||||
no_op_fold!(syn::Token![..]);
 | 
			
		||||
no_op_fold!(syn::Token![.]);
 | 
			
		||||
no_op_fold!(syn::Token![#]);
 | 
			
		||||
no_op_fold!(syn::Token![<]);
 | 
			
		||||
no_op_fold!(syn::Token![=]);
 | 
			
		||||
no_op_fold!(syn::Token![=>]);
 | 
			
		||||
no_op_fold!(syn::Token![>]);
 | 
			
		||||
no_op_fold!(syn::Token![|]);
 | 
			
		||||
no_op_fold!(syn::Token![enum]);
 | 
			
		||||
no_op_fold!(syn::Token![extern]);
 | 
			
		||||
| 
						 | 
				
			
			@ -251,3 +255,4 @@ no_op_fold!(syn::Token![mut]);
 | 
			
		|||
no_op_fold!(syn::Token![static]);
 | 
			
		||||
no_op_fold!(syn::Token![struct]);
 | 
			
		||||
no_op_fold!(syn::Token![where]);
 | 
			
		||||
no_op_fold!(usize);
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										857
									
								
								crates/fayalite-proc-macros-impl/src/hdl_bundle.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										857
									
								
								crates/fayalite-proc-macros-impl/src/hdl_bundle.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,857 @@
 | 
			
		|||
use crate::{
 | 
			
		||||
    hdl_type_common::{
 | 
			
		||||
        common_derives, get_target, ItemOptions, MakeHdlTypeExpr, MaybeParsed, ParsedField,
 | 
			
		||||
        ParsedFieldsNamed, ParsedGenerics, SplitForImpl, TypesParser, WrappedInConst,
 | 
			
		||||
    },
 | 
			
		||||
    kw, Errors, HdlAttr, PairsIterExt,
 | 
			
		||||
};
 | 
			
		||||
use proc_macro2::TokenStream;
 | 
			
		||||
use quote::{format_ident, quote_spanned, ToTokens};
 | 
			
		||||
use syn::{
 | 
			
		||||
    parse_quote, parse_quote_spanned,
 | 
			
		||||
    punctuated::{Pair, Punctuated},
 | 
			
		||||
    spanned::Spanned,
 | 
			
		||||
    token::Brace,
 | 
			
		||||
    AngleBracketedGenericArguments, Attribute, Field, FieldMutability, Fields, FieldsNamed,
 | 
			
		||||
    GenericParam, Generics, Ident, ItemStruct, Path, Token, Type, Visibility,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct ParsedBundle {
 | 
			
		||||
    pub(crate) attrs: Vec<Attribute>,
 | 
			
		||||
    pub(crate) options: HdlAttr<ItemOptions>,
 | 
			
		||||
    pub(crate) vis: Visibility,
 | 
			
		||||
    pub(crate) struct_token: Token![struct],
 | 
			
		||||
    pub(crate) ident: Ident,
 | 
			
		||||
    pub(crate) generics: MaybeParsed<ParsedGenerics, Generics>,
 | 
			
		||||
    pub(crate) fields: MaybeParsed<ParsedFieldsNamed, FieldsNamed>,
 | 
			
		||||
    pub(crate) field_flips: Vec<Option<HdlAttr<kw::flip>>>,
 | 
			
		||||
    pub(crate) mask_type_ident: Ident,
 | 
			
		||||
    pub(crate) mask_type_match_variant_ident: Ident,
 | 
			
		||||
    pub(crate) match_variant_ident: Ident,
 | 
			
		||||
    pub(crate) builder_ident: Ident,
 | 
			
		||||
    pub(crate) mask_type_builder_ident: Ident,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedBundle {
 | 
			
		||||
    fn parse_field(
 | 
			
		||||
        errors: &mut Errors,
 | 
			
		||||
        field: &mut Field,
 | 
			
		||||
        index: usize,
 | 
			
		||||
    ) -> Option<HdlAttr<kw::flip>> {
 | 
			
		||||
        let Field {
 | 
			
		||||
            attrs,
 | 
			
		||||
            vis: _,
 | 
			
		||||
            mutability,
 | 
			
		||||
            ident,
 | 
			
		||||
            colon_token,
 | 
			
		||||
            ty,
 | 
			
		||||
        } = field;
 | 
			
		||||
        let ident = ident.get_or_insert_with(|| format_ident!("_{}", index, span = ty.span()));
 | 
			
		||||
        if !matches!(mutability, FieldMutability::None) {
 | 
			
		||||
            // FIXME: use mutability as the spanned tokens,
 | 
			
		||||
            // blocked on https://github.com/dtolnay/syn/issues/1717
 | 
			
		||||
            errors.error(&ident, "field mutability is not supported");
 | 
			
		||||
            *mutability = FieldMutability::None;
 | 
			
		||||
        }
 | 
			
		||||
        *mutability = FieldMutability::None;
 | 
			
		||||
        colon_token.get_or_insert(Token));
 | 
			
		||||
        let options = errors.unwrap_or_default(HdlAttr::parse_and_take_attr(attrs));
 | 
			
		||||
        options
 | 
			
		||||
    }
 | 
			
		||||
    fn parse(item: ItemStruct) -> syn::Result<Self> {
 | 
			
		||||
        let ItemStruct {
 | 
			
		||||
            mut attrs,
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            mut generics,
 | 
			
		||||
            fields,
 | 
			
		||||
            semi_token,
 | 
			
		||||
        } = item;
 | 
			
		||||
        let mut errors = Errors::new();
 | 
			
		||||
        let mut options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::<ItemOptions>::parse_and_take_attr(&mut attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        errors.ok(options.body.validate());
 | 
			
		||||
        let ItemOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            target: _,
 | 
			
		||||
            custom_bounds,
 | 
			
		||||
            no_static: _,
 | 
			
		||||
            no_runtime_generics: _,
 | 
			
		||||
        } = options.body;
 | 
			
		||||
        let mut fields = match fields {
 | 
			
		||||
            syn::Fields::Named(fields) => fields,
 | 
			
		||||
            syn::Fields::Unnamed(fields) => {
 | 
			
		||||
                errors.error(&fields, "#[hdl] struct must use curly braces: {}");
 | 
			
		||||
                FieldsNamed {
 | 
			
		||||
                    brace_token: Brace(fields.paren_token.span),
 | 
			
		||||
                    named: fields.unnamed,
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            syn::Fields::Unit => {
 | 
			
		||||
                errors.error(&fields, "#[hdl] struct must use curly braces: {}");
 | 
			
		||||
                FieldsNamed {
 | 
			
		||||
                    brace_token: Brace(semi_token.unwrap_or_default().span),
 | 
			
		||||
                    named: Punctuated::default(),
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        let mut field_flips = Vec::with_capacity(fields.named.len());
 | 
			
		||||
        for (index, field) in fields.named.iter_mut().enumerate() {
 | 
			
		||||
            field_flips.push(Self::parse_field(&mut errors, field, index));
 | 
			
		||||
        }
 | 
			
		||||
        let generics = if custom_bounds.is_some() {
 | 
			
		||||
            MaybeParsed::Unrecognized(generics)
 | 
			
		||||
        } else if let Some(generics) = errors.ok(ParsedGenerics::parse(&mut generics)) {
 | 
			
		||||
            MaybeParsed::Parsed(generics)
 | 
			
		||||
        } else {
 | 
			
		||||
            MaybeParsed::Unrecognized(generics)
 | 
			
		||||
        };
 | 
			
		||||
        let fields = TypesParser::maybe_run(generics.as_ref(), fields, &mut errors);
 | 
			
		||||
        errors.finish()?;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            generics,
 | 
			
		||||
            fields,
 | 
			
		||||
            field_flips,
 | 
			
		||||
            mask_type_ident: format_ident!("__{}__MaskType", ident),
 | 
			
		||||
            mask_type_match_variant_ident: format_ident!("__{}__MaskType__MatchVariant", ident),
 | 
			
		||||
            match_variant_ident: format_ident!("__{}__MatchVariant", ident),
 | 
			
		||||
            mask_type_builder_ident: format_ident!("__{}__MaskType__Builder", ident),
 | 
			
		||||
            builder_ident: format_ident!("__{}__Builder", ident),
 | 
			
		||||
            ident,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
struct Builder {
 | 
			
		||||
    vis: Visibility,
 | 
			
		||||
    struct_token: Token![struct],
 | 
			
		||||
    ident: Ident,
 | 
			
		||||
    target: Path,
 | 
			
		||||
    generics: Generics,
 | 
			
		||||
    fields: FieldsNamed,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Debug)]
 | 
			
		||||
enum BuilderFieldState {
 | 
			
		||||
    Unfilled,
 | 
			
		||||
    Generic,
 | 
			
		||||
    Filled,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Builder {
 | 
			
		||||
    fn phantom_field_name(&self) -> Ident {
 | 
			
		||||
        format_ident!("__phantom", span = self.ident.span())
 | 
			
		||||
    }
 | 
			
		||||
    fn phantom_field(&self) -> Field {
 | 
			
		||||
        let target = &self.target;
 | 
			
		||||
        let type_generics = self.generics.split_for_impl().1;
 | 
			
		||||
        Field {
 | 
			
		||||
            attrs: vec![],
 | 
			
		||||
            vis: Visibility::Inherited,
 | 
			
		||||
            mutability: FieldMutability::None,
 | 
			
		||||
            ident: Some(self.phantom_field_name()),
 | 
			
		||||
            colon_token: Some(Token)),
 | 
			
		||||
            ty: parse_quote_spanned! {self.ident.span()=>
 | 
			
		||||
                ::fayalite::__std::marker::PhantomData<#target #type_generics>
 | 
			
		||||
            },
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    fn builder_struct_generics(
 | 
			
		||||
        &self,
 | 
			
		||||
        mut get_field_state: impl FnMut(usize) -> BuilderFieldState,
 | 
			
		||||
    ) -> Generics {
 | 
			
		||||
        let mut retval = self.generics.clone();
 | 
			
		||||
        for param in retval.params.iter_mut() {
 | 
			
		||||
            match param {
 | 
			
		||||
                GenericParam::Lifetime(_) => {}
 | 
			
		||||
                GenericParam::Type(param) => param.default = None,
 | 
			
		||||
                GenericParam::Const(param) => param.default = None,
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        for (field_index, field) in self.fields.named.iter().enumerate() {
 | 
			
		||||
            match get_field_state(field_index) {
 | 
			
		||||
                BuilderFieldState::Unfilled | BuilderFieldState::Filled => continue,
 | 
			
		||||
                BuilderFieldState::Generic => {}
 | 
			
		||||
            }
 | 
			
		||||
            if !retval.params.empty_or_trailing() {
 | 
			
		||||
                retval.params.push_punct(Token));
 | 
			
		||||
            }
 | 
			
		||||
            retval.params.push_value(GenericParam::Type(
 | 
			
		||||
                type_var_for_field_name(field.ident.as_ref().unwrap()).into(),
 | 
			
		||||
            ));
 | 
			
		||||
        }
 | 
			
		||||
        retval
 | 
			
		||||
    }
 | 
			
		||||
    fn builder_struct_ty(
 | 
			
		||||
        &self,
 | 
			
		||||
        mut get_field_state: impl FnMut(usize) -> BuilderFieldState,
 | 
			
		||||
    ) -> Type {
 | 
			
		||||
        let mut ty_arguments: AngleBracketedGenericArguments = if self.generics.params.is_empty() {
 | 
			
		||||
            parse_quote_spanned! {self.ident.span()=>
 | 
			
		||||
                <>
 | 
			
		||||
            }
 | 
			
		||||
        } else {
 | 
			
		||||
            let builder_type_generics = self.generics.split_for_impl().1;
 | 
			
		||||
            parse_quote! { #builder_type_generics }
 | 
			
		||||
        };
 | 
			
		||||
        for (field_index, Field { ident, ty, .. }) in self.fields.named.iter().enumerate() {
 | 
			
		||||
            let ident = ident.as_ref().unwrap();
 | 
			
		||||
            if !ty_arguments.args.empty_or_trailing() {
 | 
			
		||||
                ty_arguments.args.push_punct(Token));
 | 
			
		||||
            }
 | 
			
		||||
            ty_arguments
 | 
			
		||||
                .args
 | 
			
		||||
                .push_value(match get_field_state(field_index) {
 | 
			
		||||
                    BuilderFieldState::Unfilled => parse_quote_spanned! {self.ident.span()=>
 | 
			
		||||
                        ::fayalite::bundle::Unfilled<#ty>
 | 
			
		||||
                    },
 | 
			
		||||
                    BuilderFieldState::Generic => {
 | 
			
		||||
                        let type_var = type_var_for_field_name(ident);
 | 
			
		||||
                        parse_quote_spanned! {self.ident.span()=>
 | 
			
		||||
                            #type_var
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    BuilderFieldState::Filled => parse_quote_spanned! {self.ident.span()=>
 | 
			
		||||
                        ::fayalite::expr::Expr<#ty>
 | 
			
		||||
                    },
 | 
			
		||||
                });
 | 
			
		||||
        }
 | 
			
		||||
        let ident = &self.ident;
 | 
			
		||||
        parse_quote_spanned! {ident.span()=>
 | 
			
		||||
            #ident #ty_arguments
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn type_var_for_field_name(ident: &Ident) -> Ident {
 | 
			
		||||
    format_ident!("__T_{}", ident)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn field_fn_for_field_name(ident: &Ident) -> Ident {
 | 
			
		||||
    format_ident!("field_{}", ident)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for Builder {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            target,
 | 
			
		||||
            generics: _,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let phantom_field_name = self.phantom_field_name();
 | 
			
		||||
        let builder_struct = ItemStruct {
 | 
			
		||||
            attrs: vec![parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
            }],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: ident.clone(),
 | 
			
		||||
            generics: self.builder_struct_generics(|_| BuilderFieldState::Generic),
 | 
			
		||||
            fields: Fields::Named(FieldsNamed {
 | 
			
		||||
                brace_token: fields.brace_token,
 | 
			
		||||
                named: Punctuated::from_iter(
 | 
			
		||||
                    [Pair::Punctuated(
 | 
			
		||||
                        self.phantom_field(),
 | 
			
		||||
                        Token),
 | 
			
		||||
                    )]
 | 
			
		||||
                    .into_iter()
 | 
			
		||||
                    .chain(fields.named.pairs().map_pair_value_ref(|field| {
 | 
			
		||||
                        let ident = field.ident.as_ref().unwrap();
 | 
			
		||||
                        let type_var = type_var_for_field_name(ident);
 | 
			
		||||
                        Field {
 | 
			
		||||
                            vis: Visibility::Inherited,
 | 
			
		||||
                            ty: parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                                #type_var
 | 
			
		||||
                            },
 | 
			
		||||
                            ..field.clone()
 | 
			
		||||
                        }
 | 
			
		||||
                    })),
 | 
			
		||||
                ),
 | 
			
		||||
            }),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        };
 | 
			
		||||
        builder_struct.to_tokens(tokens);
 | 
			
		||||
        let field_idents = Vec::from_iter(
 | 
			
		||||
            self.fields
 | 
			
		||||
                .named
 | 
			
		||||
                .iter()
 | 
			
		||||
                .map(|field| field.ident.as_ref().unwrap()),
 | 
			
		||||
        );
 | 
			
		||||
        for (
 | 
			
		||||
            field_index,
 | 
			
		||||
            Field {
 | 
			
		||||
                vis,
 | 
			
		||||
                ident: field_ident,
 | 
			
		||||
                ty,
 | 
			
		||||
                ..
 | 
			
		||||
            },
 | 
			
		||||
        ) in self.fields.named.iter().enumerate()
 | 
			
		||||
        {
 | 
			
		||||
            let field_ident = field_ident.as_ref().unwrap();
 | 
			
		||||
            let fn_ident = field_fn_for_field_name(field_ident);
 | 
			
		||||
            let fn_generics = self.builder_struct_generics(|i| {
 | 
			
		||||
                if i == field_index {
 | 
			
		||||
                    BuilderFieldState::Unfilled
 | 
			
		||||
                } else {
 | 
			
		||||
                    BuilderFieldState::Generic
 | 
			
		||||
                }
 | 
			
		||||
            });
 | 
			
		||||
            let (impl_generics, _, where_clause) = fn_generics.split_for_impl();
 | 
			
		||||
            let unfilled_ty = self.builder_struct_ty(|i| {
 | 
			
		||||
                if i == field_index {
 | 
			
		||||
                    BuilderFieldState::Unfilled
 | 
			
		||||
                } else {
 | 
			
		||||
                    BuilderFieldState::Generic
 | 
			
		||||
                }
 | 
			
		||||
            });
 | 
			
		||||
            let filled_ty = self.builder_struct_ty(|i| {
 | 
			
		||||
                if i == field_index {
 | 
			
		||||
                    BuilderFieldState::Filled
 | 
			
		||||
                } else {
 | 
			
		||||
                    BuilderFieldState::Generic
 | 
			
		||||
                }
 | 
			
		||||
            });
 | 
			
		||||
            let pat_fields =
 | 
			
		||||
                Vec::from_iter(self.fields.named.iter().enumerate().map(|(i, field)| {
 | 
			
		||||
                    let field_ident = field.ident.as_ref().unwrap();
 | 
			
		||||
                    if field_index == i {
 | 
			
		||||
                        quote_spanned! {self.ident.span()=>
 | 
			
		||||
                            #field_ident: _,
 | 
			
		||||
                        }
 | 
			
		||||
                    } else {
 | 
			
		||||
                        quote_spanned! {self.ident.span()=>
 | 
			
		||||
                            #field_ident,
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }));
 | 
			
		||||
            quote_spanned! {self.ident.span()=>
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
                impl #impl_generics #unfilled_ty
 | 
			
		||||
                #where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    #vis fn #fn_ident(
 | 
			
		||||
                        self,
 | 
			
		||||
                        #field_ident: impl ::fayalite::expr::ToExpr<Type = #ty>,
 | 
			
		||||
                    ) -> #filled_ty {
 | 
			
		||||
                        let Self {
 | 
			
		||||
                            #phantom_field_name: _,
 | 
			
		||||
                            #(#pat_fields)*
 | 
			
		||||
                        } = self;
 | 
			
		||||
                        let #field_ident = ::fayalite::expr::ToExpr::to_expr(&#field_ident);
 | 
			
		||||
                        #ident {
 | 
			
		||||
                            #phantom_field_name: ::fayalite::__std::marker::PhantomData,
 | 
			
		||||
                            #(#field_idents,)*
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let unfilled_generics = self.builder_struct_generics(|_| BuilderFieldState::Unfilled);
 | 
			
		||||
        let unfilled_ty = self.builder_struct_ty(|_| BuilderFieldState::Unfilled);
 | 
			
		||||
        let (unfilled_impl_generics, _, unfilled_where_clause) = unfilled_generics.split_for_impl();
 | 
			
		||||
        quote_spanned! {self.ident.span()=>
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
            impl #unfilled_impl_generics ::fayalite::__std::default::Default for #unfilled_ty
 | 
			
		||||
            #unfilled_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn default() -> Self {
 | 
			
		||||
                    #ident {
 | 
			
		||||
                        #phantom_field_name: ::fayalite::__std::marker::PhantomData,
 | 
			
		||||
                        #(#field_idents: ::fayalite::__std::default::Default::default(),)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let filled_generics = self.builder_struct_generics(|_| BuilderFieldState::Filled);
 | 
			
		||||
        let filled_ty = self.builder_struct_ty(|_| BuilderFieldState::Filled);
 | 
			
		||||
        let (filled_impl_generics, _, filled_where_clause) = filled_generics.split_for_impl();
 | 
			
		||||
        let type_generics = self.generics.split_for_impl().1;
 | 
			
		||||
        quote_spanned! {self.ident.span()=>
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
            impl #filled_impl_generics ::fayalite::expr::ToExpr for #filled_ty
 | 
			
		||||
            #filled_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Type = #target #type_generics;
 | 
			
		||||
                fn to_expr(
 | 
			
		||||
                    &self,
 | 
			
		||||
                ) -> ::fayalite::expr::Expr<<Self as ::fayalite::expr::ToExpr>::Type> {
 | 
			
		||||
                    let __ty = #target {
 | 
			
		||||
                        #(#field_idents: ::fayalite::expr::Expr::ty(self.#field_idents),)*
 | 
			
		||||
                    };
 | 
			
		||||
                    let __field_values = [
 | 
			
		||||
                        #(::fayalite::expr::Expr::canonical(self.#field_idents),)*
 | 
			
		||||
                    ];
 | 
			
		||||
                    ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                        &::fayalite::expr::ops::BundleLiteral::new(
 | 
			
		||||
                            __ty,
 | 
			
		||||
                            ::fayalite::intern::Intern::intern(&__field_values[..]),
 | 
			
		||||
                        ),
 | 
			
		||||
                    )
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for ParsedBundle {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            generics,
 | 
			
		||||
            fields,
 | 
			
		||||
            field_flips,
 | 
			
		||||
            mask_type_ident,
 | 
			
		||||
            mask_type_match_variant_ident,
 | 
			
		||||
            match_variant_ident,
 | 
			
		||||
            builder_ident,
 | 
			
		||||
            mask_type_builder_ident,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let ItemOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            target,
 | 
			
		||||
            custom_bounds: _,
 | 
			
		||||
            no_static,
 | 
			
		||||
            no_runtime_generics,
 | 
			
		||||
        } = &options.body;
 | 
			
		||||
        let target = get_target(target, ident);
 | 
			
		||||
        let mut item_attrs = attrs.clone();
 | 
			
		||||
        item_attrs.push(common_derives(ident.span()));
 | 
			
		||||
        ItemStruct {
 | 
			
		||||
            attrs: item_attrs,
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: Fields::Named(fields.clone().into()),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let (impl_generics, type_generics, where_clause) = generics.split_for_impl();
 | 
			
		||||
        if let (MaybeParsed::Parsed(generics), MaybeParsed::Parsed(fields), None) =
 | 
			
		||||
            (generics, fields, no_runtime_generics)
 | 
			
		||||
        {
 | 
			
		||||
            generics.make_runtime_generics(tokens, vis, ident, &target, |context| {
 | 
			
		||||
                let fields: Vec<_> = fields
 | 
			
		||||
                    .named
 | 
			
		||||
                    .iter()
 | 
			
		||||
                    .map(|ParsedField { ident, ty, .. }| {
 | 
			
		||||
                        let ident = ident.as_ref().unwrap();
 | 
			
		||||
                        let expr = ty.make_hdl_type_expr(context);
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            #ident: #expr,
 | 
			
		||||
                        }
 | 
			
		||||
                    })
 | 
			
		||||
                    .collect();
 | 
			
		||||
                parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                    #target {
 | 
			
		||||
                        #(#fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            })
 | 
			
		||||
        }
 | 
			
		||||
        let mut wrapped_in_const = WrappedInConst::new(tokens, ident.span());
 | 
			
		||||
        let tokens = wrapped_in_const.inner();
 | 
			
		||||
        let builder = Builder {
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: builder_ident.clone(),
 | 
			
		||||
            target: target.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: fields.clone().into(),
 | 
			
		||||
        };
 | 
			
		||||
        builder.to_tokens(tokens);
 | 
			
		||||
        let unfilled_builder_ty = builder.builder_struct_ty(|_| BuilderFieldState::Unfilled);
 | 
			
		||||
        let filled_builder_ty = builder.builder_struct_ty(|_| BuilderFieldState::Filled);
 | 
			
		||||
        let mut mask_type_fields = FieldsNamed::from(fields.clone());
 | 
			
		||||
        for Field { ident, ty, .. } in &mut mask_type_fields.named {
 | 
			
		||||
            let ident = ident.as_ref().unwrap();
 | 
			
		||||
            *ty = parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                <#ty as ::fayalite::ty::Type>::MaskType
 | 
			
		||||
            };
 | 
			
		||||
        }
 | 
			
		||||
        let mask_type_builder = Builder {
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: mask_type_builder_ident.clone(),
 | 
			
		||||
            target: mask_type_ident.clone().into(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: mask_type_fields.clone(),
 | 
			
		||||
        };
 | 
			
		||||
        mask_type_builder.to_tokens(tokens);
 | 
			
		||||
        let unfilled_mask_type_builder_ty =
 | 
			
		||||
            mask_type_builder.builder_struct_ty(|_| BuilderFieldState::Unfilled);
 | 
			
		||||
        let filled_mask_type_builder_ty =
 | 
			
		||||
            mask_type_builder.builder_struct_ty(|_| BuilderFieldState::Filled);
 | 
			
		||||
        ItemStruct {
 | 
			
		||||
            attrs: vec![
 | 
			
		||||
                common_derives(ident.span()),
 | 
			
		||||
                parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                    #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
                },
 | 
			
		||||
            ],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: mask_type_ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: Fields::Named(mask_type_fields.clone()),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let mut mask_type_match_variant_fields = mask_type_fields;
 | 
			
		||||
        for Field { ident, ty, .. } in &mut mask_type_match_variant_fields.named {
 | 
			
		||||
            let ident = ident.as_ref().unwrap();
 | 
			
		||||
            *ty = parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                ::fayalite::expr::Expr<#ty>
 | 
			
		||||
            };
 | 
			
		||||
        }
 | 
			
		||||
        ItemStruct {
 | 
			
		||||
            attrs: vec![
 | 
			
		||||
                common_derives(ident.span()),
 | 
			
		||||
                parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                    #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
                },
 | 
			
		||||
            ],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: mask_type_match_variant_ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: Fields::Named(mask_type_match_variant_fields),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let mut match_variant_fields = FieldsNamed::from(fields.clone());
 | 
			
		||||
        for Field { ident, ty, .. } in &mut match_variant_fields.named {
 | 
			
		||||
            let ident = ident.as_ref().unwrap();
 | 
			
		||||
            *ty = parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                ::fayalite::expr::Expr<#ty>
 | 
			
		||||
            };
 | 
			
		||||
        }
 | 
			
		||||
        ItemStruct {
 | 
			
		||||
            attrs: vec![
 | 
			
		||||
                common_derives(ident.span()),
 | 
			
		||||
                parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                    #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
                },
 | 
			
		||||
            ],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: match_variant_ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: Fields::Named(match_variant_fields),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let match_variant_body_fields = Vec::from_iter(fields.named().into_iter().map(|field| {
 | 
			
		||||
            let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
            let ident_str = ident.to_string();
 | 
			
		||||
            quote_spanned! {ident.span()=>
 | 
			
		||||
                #ident: ::fayalite::expr::Expr::field(__this, #ident_str),
 | 
			
		||||
            }
 | 
			
		||||
        }));
 | 
			
		||||
        let mask_type_body_fields = Vec::from_iter(fields.named().into_iter().map(|field| {
 | 
			
		||||
            let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
            quote_spanned! {ident.span()=>
 | 
			
		||||
                #ident: ::fayalite::ty::Type::mask_type(&self.#ident),
 | 
			
		||||
            }
 | 
			
		||||
        }));
 | 
			
		||||
        let from_canonical_body_fields =
 | 
			
		||||
            Vec::from_iter(fields.named().into_iter().enumerate().zip(field_flips).map(
 | 
			
		||||
                |((index, field), flip)| {
 | 
			
		||||
                    let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                    let ident_str = ident.to_string();
 | 
			
		||||
                    let flipped = flip.is_some();
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        #ident: {
 | 
			
		||||
                            let ::fayalite::bundle::BundleField {
 | 
			
		||||
                                name: __name,
 | 
			
		||||
                                flipped: __flipped,
 | 
			
		||||
                                ty: __ty,
 | 
			
		||||
                            } = __fields[#index];
 | 
			
		||||
                            ::fayalite::__std::assert_eq!(&*__name, #ident_str);
 | 
			
		||||
                            ::fayalite::__std::assert_eq!(__flipped, #flipped);
 | 
			
		||||
                            ::fayalite::ty::Type::from_canonical(__ty)
 | 
			
		||||
                        },
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
            ));
 | 
			
		||||
        let fields_body_fields = Vec::from_iter(fields.named().into_iter().zip(field_flips).map(
 | 
			
		||||
            |(field, flip)| {
 | 
			
		||||
                let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                let ident_str = ident.to_string();
 | 
			
		||||
                let flipped = flip.is_some();
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    ::fayalite::bundle::BundleField {
 | 
			
		||||
                        name: ::fayalite::intern::Intern::intern(#ident_str),
 | 
			
		||||
                        flipped: #flipped,
 | 
			
		||||
                        ty: ::fayalite::ty::Type::canonical(&self.#ident),
 | 
			
		||||
                    },
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        let fields_len = fields.named().into_iter().len();
 | 
			
		||||
        quote_spanned! {ident.span()=>
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::Type for #mask_type_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type BaseType = ::fayalite::bundle::Bundle;
 | 
			
		||||
                type MaskType = #mask_type_ident #type_generics;
 | 
			
		||||
                type MatchVariant = #mask_type_match_variant_ident #type_generics;
 | 
			
		||||
                type MatchActiveScope = ();
 | 
			
		||||
                type MatchVariantAndInactiveScope = ::fayalite::ty::MatchVariantWithoutScope<
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariant,
 | 
			
		||||
                >;
 | 
			
		||||
                type MatchVariantsIter = ::fayalite::__std::iter::Once<
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariantAndInactiveScope,
 | 
			
		||||
                >;
 | 
			
		||||
                fn match_variants(
 | 
			
		||||
                    __this: ::fayalite::expr::Expr<Self>,
 | 
			
		||||
                    __source_location: ::fayalite::source_location::SourceLocation,
 | 
			
		||||
                ) -> <Self as ::fayalite::ty::Type>::MatchVariantsIter {
 | 
			
		||||
                    let __retval = #mask_type_match_variant_ident {
 | 
			
		||||
                        #(#match_variant_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    ::fayalite::__std::iter::once(::fayalite::ty::MatchVariantWithoutScope(__retval))
 | 
			
		||||
                }
 | 
			
		||||
                fn mask_type(&self) -> <Self as ::fayalite::ty::Type>::MaskType {
 | 
			
		||||
                    *self
 | 
			
		||||
                }
 | 
			
		||||
                fn canonical(&self) -> ::fayalite::ty::CanonicalType {
 | 
			
		||||
                    ::fayalite::ty::Type::canonical(&::fayalite::bundle::Bundle::new(::fayalite::bundle::BundleType::fields(self)))
 | 
			
		||||
                }
 | 
			
		||||
                #[track_caller]
 | 
			
		||||
                fn from_canonical(__canonical_type: ::fayalite::ty::CanonicalType) -> Self {
 | 
			
		||||
                    let ::fayalite::ty::CanonicalType::Bundle(__bundle) = __canonical_type else {
 | 
			
		||||
                        ::fayalite::__std::panic!("expected bundle");
 | 
			
		||||
                    };
 | 
			
		||||
                    let __fields = ::fayalite::bundle::BundleType::fields(&__bundle);
 | 
			
		||||
                    ::fayalite::__std::assert_eq!(__fields.len(), #fields_len, "bundle has wrong number of fields");
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#from_canonical_body_fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                fn source_location() -> ::fayalite::source_location::SourceLocation {
 | 
			
		||||
                    ::fayalite::source_location::SourceLocation::caller()
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::bundle::BundleType for #mask_type_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Builder = #unfilled_mask_type_builder_ty;
 | 
			
		||||
                type FilledBuilder = #filled_mask_type_builder_ty;
 | 
			
		||||
                fn fields(&self) -> ::fayalite::intern::Interned<[::fayalite::bundle::BundleField]> {
 | 
			
		||||
                    ::fayalite::intern::Intern::intern(&[#(#fields_body_fields)*][..])
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            impl #impl_generics #mask_type_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                #vis fn __bundle_builder() -> #unfilled_mask_type_builder_ty {
 | 
			
		||||
                    ::fayalite::__std::default::Default::default()
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::TypeWithDeref for #mask_type_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn expr_deref(__this: &::fayalite::expr::Expr<Self>) -> &<Self as ::fayalite::ty::Type>::MatchVariant {
 | 
			
		||||
                    let __this = *__this;
 | 
			
		||||
                    let __retval = #mask_type_match_variant_ident {
 | 
			
		||||
                        #(#match_variant_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    ::fayalite::intern::Interned::<_>::into_inner(::fayalite::intern::Intern::intern_sized(__retval))
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::Type for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type BaseType = ::fayalite::bundle::Bundle;
 | 
			
		||||
                type MaskType = #mask_type_ident #type_generics;
 | 
			
		||||
                type MatchVariant = #match_variant_ident #type_generics;
 | 
			
		||||
                type MatchActiveScope = ();
 | 
			
		||||
                type MatchVariantAndInactiveScope = ::fayalite::ty::MatchVariantWithoutScope<
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariant,
 | 
			
		||||
                >;
 | 
			
		||||
                type MatchVariantsIter = ::fayalite::__std::iter::Once<
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariantAndInactiveScope,
 | 
			
		||||
                >;
 | 
			
		||||
                fn match_variants(
 | 
			
		||||
                    __this: ::fayalite::expr::Expr<Self>,
 | 
			
		||||
                    __source_location: ::fayalite::source_location::SourceLocation,
 | 
			
		||||
                ) -> <Self as ::fayalite::ty::Type>::MatchVariantsIter {
 | 
			
		||||
                    let __retval = #match_variant_ident {
 | 
			
		||||
                        #(#match_variant_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    ::fayalite::__std::iter::once(::fayalite::ty::MatchVariantWithoutScope(__retval))
 | 
			
		||||
                }
 | 
			
		||||
                fn mask_type(&self) -> <Self as ::fayalite::ty::Type>::MaskType {
 | 
			
		||||
                    #mask_type_ident {
 | 
			
		||||
                        #(#mask_type_body_fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                fn canonical(&self) -> ::fayalite::ty::CanonicalType {
 | 
			
		||||
                    ::fayalite::ty::Type::canonical(&::fayalite::bundle::Bundle::new(::fayalite::bundle::BundleType::fields(self)))
 | 
			
		||||
                }
 | 
			
		||||
                #[track_caller]
 | 
			
		||||
                fn from_canonical(__canonical_type: ::fayalite::ty::CanonicalType) -> Self {
 | 
			
		||||
                    let ::fayalite::ty::CanonicalType::Bundle(__bundle) = __canonical_type else {
 | 
			
		||||
                        ::fayalite::__std::panic!("expected bundle");
 | 
			
		||||
                    };
 | 
			
		||||
                    let __fields = ::fayalite::bundle::BundleType::fields(&__bundle);
 | 
			
		||||
                    ::fayalite::__std::assert_eq!(__fields.len(), #fields_len, "bundle has wrong number of fields");
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#from_canonical_body_fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                fn source_location() -> ::fayalite::source_location::SourceLocation {
 | 
			
		||||
                    ::fayalite::source_location::SourceLocation::caller()
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::bundle::BundleType for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Builder = #unfilled_builder_ty;
 | 
			
		||||
                type FilledBuilder = #filled_builder_ty;
 | 
			
		||||
                fn fields(&self) -> ::fayalite::intern::Interned<[::fayalite::bundle::BundleField]> {
 | 
			
		||||
                    ::fayalite::intern::Intern::intern(&[#(#fields_body_fields)*][..])
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            impl #impl_generics #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                #vis fn __bundle_builder() -> #unfilled_builder_ty {
 | 
			
		||||
                    ::fayalite::__std::default::Default::default()
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::TypeWithDeref for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn expr_deref(__this: &::fayalite::expr::Expr<Self>) -> &<Self as ::fayalite::ty::Type>::MatchVariant {
 | 
			
		||||
                    let __this = *__this;
 | 
			
		||||
                    let __retval = #match_variant_ident {
 | 
			
		||||
                        #(#match_variant_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    ::fayalite::intern::Interned::<_>::into_inner(::fayalite::intern::Intern::intern_sized(__retval))
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        if let (None, MaybeParsed::Parsed(generics)) = (no_static, &self.generics) {
 | 
			
		||||
            let static_generics = generics.clone().for_static_type();
 | 
			
		||||
            let (static_impl_generics, static_type_generics, static_where_clause) =
 | 
			
		||||
                static_generics.split_for_impl();
 | 
			
		||||
            let static_type_body_fields = Vec::from_iter(fields.named().into_iter().map(|field| {
 | 
			
		||||
                let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                let ty = field.ty();
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    #ident: <#ty as ::fayalite::ty::StaticType>::TYPE,
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
            let static_mask_type_body_fields =
 | 
			
		||||
                Vec::from_iter(fields.named().into_iter().map(|field| {
 | 
			
		||||
                    let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                    let ty = field.ty();
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        #ident: <#ty as ::fayalite::ty::StaticType>::MASK_TYPE,
 | 
			
		||||
                    }
 | 
			
		||||
                }));
 | 
			
		||||
            let type_properties = format_ident!("__type_properties", span = ident.span());
 | 
			
		||||
            let type_properties_fields = Vec::from_iter(fields.named().into_iter().zip(field_flips).map(|(field, field_flip)| {
 | 
			
		||||
                let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                let flipped = field_flip.is_some();
 | 
			
		||||
                let ty = field.ty();
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    let #type_properties = #type_properties.field(#flipped, <#ty as ::fayalite::ty::StaticType>::TYPE_PROPERTIES);
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
            let type_properties_mask_fields = Vec::from_iter(fields.named().into_iter().zip(field_flips).map(|(field, field_flip)| {
 | 
			
		||||
                let ident: &Ident = field.ident().as_ref().unwrap();
 | 
			
		||||
                let flipped = field_flip.is_some();
 | 
			
		||||
                let ty = field.ty();
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    let #type_properties = #type_properties.field(#flipped, <#ty as ::fayalite::ty::StaticType>::MASK_TYPE_PROPERTIES);
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
            quote_spanned! {ident.span()=>
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #static_impl_generics ::fayalite::ty::StaticType for #mask_type_ident #static_type_generics
 | 
			
		||||
                #static_where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    const TYPE: Self = Self {
 | 
			
		||||
                        #(#static_mask_type_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE: <Self as ::fayalite::ty::Type>::MaskType = Self {
 | 
			
		||||
                        #(#static_mask_type_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    const TYPE_PROPERTIES: ::fayalite::ty::TypeProperties = {
 | 
			
		||||
                        let #type_properties = ::fayalite::bundle::BundleTypePropertiesBuilder::new();
 | 
			
		||||
                        #(#type_properties_mask_fields)*
 | 
			
		||||
                        #type_properties.finish()
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE_PROPERTIES: ::fayalite::ty::TypeProperties = {
 | 
			
		||||
                        let #type_properties = ::fayalite::bundle::BundleTypePropertiesBuilder::new();
 | 
			
		||||
                        #(#type_properties_mask_fields)*
 | 
			
		||||
                        #type_properties.finish()
 | 
			
		||||
                    };
 | 
			
		||||
                }
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #static_impl_generics ::fayalite::ty::StaticType for #target #static_type_generics
 | 
			
		||||
                #static_where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    const TYPE: Self = Self {
 | 
			
		||||
                        #(#static_type_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE: <Self as ::fayalite::ty::Type>::MaskType = #mask_type_ident {
 | 
			
		||||
                        #(#static_mask_type_body_fields)*
 | 
			
		||||
                    };
 | 
			
		||||
                    const TYPE_PROPERTIES: ::fayalite::ty::TypeProperties = {
 | 
			
		||||
                        let #type_properties = ::fayalite::bundle::BundleTypePropertiesBuilder::new();
 | 
			
		||||
                        #(#type_properties_fields)*
 | 
			
		||||
                        #type_properties.finish()
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE_PROPERTIES: ::fayalite::ty::TypeProperties = {
 | 
			
		||||
                        let #type_properties = ::fayalite::bundle::BundleTypePropertiesBuilder::new();
 | 
			
		||||
                        #(#type_properties_mask_fields)*
 | 
			
		||||
                        #type_properties.finish()
 | 
			
		||||
                    };
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn hdl_bundle(item: ItemStruct) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = ParsedBundle::parse(item)?;
 | 
			
		||||
    let outline_generated = item.options.body.outline_generated;
 | 
			
		||||
    let mut contents = item.to_token_stream();
 | 
			
		||||
    if outline_generated.is_some() {
 | 
			
		||||
        contents = crate::outline_generated(contents, "hdl-bundle-");
 | 
			
		||||
    }
 | 
			
		||||
    Ok(contents)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										656
									
								
								crates/fayalite-proc-macros-impl/src/hdl_enum.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										656
									
								
								crates/fayalite-proc-macros-impl/src/hdl_enum.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,656 @@
 | 
			
		|||
use crate::{
 | 
			
		||||
    hdl_type_common::{
 | 
			
		||||
        common_derives, get_target, ItemOptions, MakeHdlTypeExpr, MaybeParsed, ParsedGenerics,
 | 
			
		||||
        ParsedType, SplitForImpl, TypesParser, WrappedInConst,
 | 
			
		||||
    },
 | 
			
		||||
    Errors, HdlAttr, PairsIterExt,
 | 
			
		||||
};
 | 
			
		||||
use proc_macro2::TokenStream;
 | 
			
		||||
use quote::{format_ident, quote_spanned, ToTokens};
 | 
			
		||||
use syn::{
 | 
			
		||||
    parse_quote_spanned,
 | 
			
		||||
    punctuated::{Pair, Punctuated},
 | 
			
		||||
    token::{Brace, Paren},
 | 
			
		||||
    Attribute, Field, FieldMutability, Fields, FieldsNamed, FieldsUnnamed, Generics, Ident,
 | 
			
		||||
    ItemEnum, ItemStruct, Token, Type, Variant, Visibility,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = VariantOptions]
 | 
			
		||||
    pub(crate) enum VariantOption {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = FieldOptions]
 | 
			
		||||
    pub(crate) enum FieldOption {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct ParsedVariantField {
 | 
			
		||||
    pub(crate) paren_token: Paren,
 | 
			
		||||
    pub(crate) attrs: Vec<Attribute>,
 | 
			
		||||
    pub(crate) options: HdlAttr<FieldOptions>,
 | 
			
		||||
    pub(crate) ty: MaybeParsed<ParsedType, Type>,
 | 
			
		||||
    pub(crate) comma_token: Option<Token![,]>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct ParsedVariant {
 | 
			
		||||
    pub(crate) attrs: Vec<Attribute>,
 | 
			
		||||
    pub(crate) options: HdlAttr<VariantOptions>,
 | 
			
		||||
    pub(crate) ident: Ident,
 | 
			
		||||
    pub(crate) field: Option<ParsedVariantField>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedVariant {
 | 
			
		||||
    fn parse(
 | 
			
		||||
        errors: &mut Errors,
 | 
			
		||||
        variant: Variant,
 | 
			
		||||
        generics: &MaybeParsed<ParsedGenerics, Generics>,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        let Variant {
 | 
			
		||||
            mut attrs,
 | 
			
		||||
            ident,
 | 
			
		||||
            fields,
 | 
			
		||||
            discriminant,
 | 
			
		||||
        } = variant;
 | 
			
		||||
        let options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::parse_and_take_attr(&mut attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        let field = match fields {
 | 
			
		||||
            Fields::Unnamed(FieldsUnnamed {
 | 
			
		||||
                paren_token,
 | 
			
		||||
                unnamed,
 | 
			
		||||
            }) if unnamed.len() == 1 => {
 | 
			
		||||
                let (field, comma_token) = unnamed.into_pairs().next().unwrap().into_tuple();
 | 
			
		||||
                let Field {
 | 
			
		||||
                    mut attrs,
 | 
			
		||||
                    vis,
 | 
			
		||||
                    mutability,
 | 
			
		||||
                    ident: _,
 | 
			
		||||
                    colon_token: _,
 | 
			
		||||
                    ty,
 | 
			
		||||
                } = field;
 | 
			
		||||
                let options = errors
 | 
			
		||||
                    .unwrap_or_default(HdlAttr::parse_and_take_attr(&mut attrs))
 | 
			
		||||
                    .unwrap_or_default();
 | 
			
		||||
                if !matches!(vis, Visibility::Inherited) {
 | 
			
		||||
                    errors.error(
 | 
			
		||||
                        &vis,
 | 
			
		||||
                        "enum variant fields must not have a visibility specifier",
 | 
			
		||||
                    );
 | 
			
		||||
                }
 | 
			
		||||
                if !matches!(mutability, FieldMutability::None) {
 | 
			
		||||
                    // FIXME: use mutability as the spanned tokens,
 | 
			
		||||
                    // blocked on https://github.com/dtolnay/syn/issues/1717
 | 
			
		||||
                    errors.error(&ty, "field mutability is not supported");
 | 
			
		||||
                }
 | 
			
		||||
                Some(ParsedVariantField {
 | 
			
		||||
                    paren_token,
 | 
			
		||||
                    attrs,
 | 
			
		||||
                    options,
 | 
			
		||||
                    ty: TypesParser::maybe_run(generics.as_ref(), ty, errors),
 | 
			
		||||
                    comma_token,
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
            Fields::Unit => None,
 | 
			
		||||
            Fields::Unnamed(fields) if fields.unnamed.is_empty() => None,
 | 
			
		||||
            Fields::Named(fields) if fields.named.is_empty() => None,
 | 
			
		||||
            Fields::Unnamed(_) | Fields::Named(_) => {
 | 
			
		||||
                errors.error(
 | 
			
		||||
                    fields,
 | 
			
		||||
                    "enum variant must either have no fields or a single parenthesized field",
 | 
			
		||||
                );
 | 
			
		||||
                None
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        if let Some((eq, _)) = discriminant {
 | 
			
		||||
            errors.error(eq, "custom enum discriminants are not allowed");
 | 
			
		||||
        }
 | 
			
		||||
        Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
            options,
 | 
			
		||||
            ident,
 | 
			
		||||
            field,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct ParsedEnum {
 | 
			
		||||
    pub(crate) attrs: Vec<Attribute>,
 | 
			
		||||
    pub(crate) options: HdlAttr<ItemOptions>,
 | 
			
		||||
    pub(crate) vis: Visibility,
 | 
			
		||||
    pub(crate) enum_token: Token![enum],
 | 
			
		||||
    pub(crate) ident: Ident,
 | 
			
		||||
    pub(crate) generics: MaybeParsed<ParsedGenerics, Generics>,
 | 
			
		||||
    pub(crate) brace_token: Brace,
 | 
			
		||||
    pub(crate) variants: Punctuated<ParsedVariant, Token![,]>,
 | 
			
		||||
    pub(crate) match_variant_ident: Ident,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedEnum {
 | 
			
		||||
    fn parse(item: ItemEnum) -> syn::Result<Self> {
 | 
			
		||||
        let ItemEnum {
 | 
			
		||||
            mut attrs,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            mut generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
        } = item;
 | 
			
		||||
        let mut errors = Errors::new();
 | 
			
		||||
        let mut options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::<ItemOptions>::parse_and_take_attr(&mut attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        errors.ok(options.body.validate());
 | 
			
		||||
        let ItemOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            target: _,
 | 
			
		||||
            custom_bounds,
 | 
			
		||||
            no_static: _,
 | 
			
		||||
            no_runtime_generics: _,
 | 
			
		||||
        } = options.body;
 | 
			
		||||
        attrs.retain(|attr| {
 | 
			
		||||
            if attr.path().is_ident("repr") {
 | 
			
		||||
                errors.error(attr, "#[repr] is not supported on #[hdl] enums");
 | 
			
		||||
                false
 | 
			
		||||
            } else {
 | 
			
		||||
                true
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
        let generics = if custom_bounds.is_some() {
 | 
			
		||||
            MaybeParsed::Unrecognized(generics)
 | 
			
		||||
        } else if let Some(generics) = errors.ok(ParsedGenerics::parse(&mut generics)) {
 | 
			
		||||
            MaybeParsed::Parsed(generics)
 | 
			
		||||
        } else {
 | 
			
		||||
            MaybeParsed::Unrecognized(generics)
 | 
			
		||||
        };
 | 
			
		||||
        let variants = Punctuated::from_iter(
 | 
			
		||||
            variants
 | 
			
		||||
                .into_pairs()
 | 
			
		||||
                .map_pair_value(|v| ParsedVariant::parse(&mut errors, v, &generics)),
 | 
			
		||||
        );
 | 
			
		||||
        errors.finish()?;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
            match_variant_ident: format_ident!("__{}__MatchVariant", ident),
 | 
			
		||||
            ident,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for ParsedEnum {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
            match_variant_ident,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let ItemOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            target,
 | 
			
		||||
            custom_bounds: _,
 | 
			
		||||
            no_static,
 | 
			
		||||
            no_runtime_generics,
 | 
			
		||||
        } = &options.body;
 | 
			
		||||
        let target = get_target(target, ident);
 | 
			
		||||
        let mut struct_attrs = attrs.clone();
 | 
			
		||||
        struct_attrs.push(common_derives(ident.span()));
 | 
			
		||||
        struct_attrs.push(parse_quote_spanned! {ident.span()=>
 | 
			
		||||
            #[allow(non_snake_case)]
 | 
			
		||||
        });
 | 
			
		||||
        let struct_fields = Punctuated::from_iter(variants.pairs().map_pair_value_ref(
 | 
			
		||||
            |ParsedVariant {
 | 
			
		||||
                 attrs: _,
 | 
			
		||||
                 options,
 | 
			
		||||
                 ident,
 | 
			
		||||
                 field,
 | 
			
		||||
             }| {
 | 
			
		||||
                let VariantOptions {} = options.body;
 | 
			
		||||
                let colon_token;
 | 
			
		||||
                let ty = if let Some(ParsedVariantField {
 | 
			
		||||
                    paren_token,
 | 
			
		||||
                    attrs: _,
 | 
			
		||||
                    options,
 | 
			
		||||
                    ty,
 | 
			
		||||
                    comma_token: _,
 | 
			
		||||
                }) = field
 | 
			
		||||
                {
 | 
			
		||||
                    let FieldOptions {} = options.body;
 | 
			
		||||
                    colon_token = Token);
 | 
			
		||||
                    ty.clone().into()
 | 
			
		||||
                } else {
 | 
			
		||||
                    colon_token = Token);
 | 
			
		||||
                    parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                        ()
 | 
			
		||||
                    }
 | 
			
		||||
                };
 | 
			
		||||
                Field {
 | 
			
		||||
                    attrs: vec![],
 | 
			
		||||
                    vis: vis.clone(),
 | 
			
		||||
                    mutability: FieldMutability::None,
 | 
			
		||||
                    ident: Some(ident.clone()),
 | 
			
		||||
                    colon_token: Some(colon_token),
 | 
			
		||||
                    ty,
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        ItemStruct {
 | 
			
		||||
            attrs: struct_attrs,
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: Token,
 | 
			
		||||
            ident: ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            fields: if struct_fields.is_empty() {
 | 
			
		||||
                Fields::Unit
 | 
			
		||||
            } else {
 | 
			
		||||
                Fields::Named(FieldsNamed {
 | 
			
		||||
                    brace_token: *brace_token,
 | 
			
		||||
                    named: struct_fields,
 | 
			
		||||
                })
 | 
			
		||||
            },
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let (impl_generics, type_generics, where_clause) = generics.split_for_impl();
 | 
			
		||||
        if let (MaybeParsed::Parsed(generics), None) = (generics, no_runtime_generics) {
 | 
			
		||||
            generics.make_runtime_generics(tokens, vis, ident, &target, |context| {
 | 
			
		||||
                let fields: Vec<_> = variants
 | 
			
		||||
                    .iter()
 | 
			
		||||
                    .map(|ParsedVariant { ident, field, .. }| {
 | 
			
		||||
                        if let Some(ParsedVariantField {
 | 
			
		||||
                            ty: MaybeParsed::Parsed(ty),
 | 
			
		||||
                            ..
 | 
			
		||||
                        }) = field
 | 
			
		||||
                        {
 | 
			
		||||
                            let expr = ty.make_hdl_type_expr(context);
 | 
			
		||||
                            quote_spanned! {ident.span()=>
 | 
			
		||||
                                #ident: #expr,
 | 
			
		||||
                            }
 | 
			
		||||
                        } else {
 | 
			
		||||
                            quote_spanned! {ident.span()=>
 | 
			
		||||
                                #ident: (),
 | 
			
		||||
                            }
 | 
			
		||||
                        }
 | 
			
		||||
                    })
 | 
			
		||||
                    .collect();
 | 
			
		||||
                parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                    #target {
 | 
			
		||||
                        #(#fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            })
 | 
			
		||||
        }
 | 
			
		||||
        let mut wrapped_in_const = WrappedInConst::new(tokens, ident.span());
 | 
			
		||||
        let tokens = wrapped_in_const.inner();
 | 
			
		||||
        {
 | 
			
		||||
            let mut wrapped_in_const = WrappedInConst::new(tokens, ident.span());
 | 
			
		||||
            let tokens = wrapped_in_const.inner();
 | 
			
		||||
            let mut enum_attrs = attrs.clone();
 | 
			
		||||
            enum_attrs.push(parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                #[allow(dead_code)]
 | 
			
		||||
            });
 | 
			
		||||
            ItemEnum {
 | 
			
		||||
                attrs: enum_attrs,
 | 
			
		||||
                vis: vis.clone(),
 | 
			
		||||
                enum_token: *enum_token,
 | 
			
		||||
                ident: ident.clone(),
 | 
			
		||||
                generics: generics.into(),
 | 
			
		||||
                brace_token: *brace_token,
 | 
			
		||||
                variants: Punctuated::from_iter(variants.pairs().map_pair_value_ref(
 | 
			
		||||
                    |ParsedVariant {
 | 
			
		||||
                         attrs,
 | 
			
		||||
                         options: _,
 | 
			
		||||
                         ident,
 | 
			
		||||
                         field,
 | 
			
		||||
                     }| Variant {
 | 
			
		||||
                        attrs: attrs.clone(),
 | 
			
		||||
                        ident: ident.clone(),
 | 
			
		||||
                        fields: match field {
 | 
			
		||||
                            Some(ParsedVariantField {
 | 
			
		||||
                                paren_token,
 | 
			
		||||
                                attrs,
 | 
			
		||||
                                options: _,
 | 
			
		||||
                                ty,
 | 
			
		||||
                                comma_token,
 | 
			
		||||
                            }) => Fields::Unnamed(FieldsUnnamed {
 | 
			
		||||
                                paren_token: *paren_token,
 | 
			
		||||
                                unnamed: Punctuated::from_iter([Pair::new(
 | 
			
		||||
                                    Field {
 | 
			
		||||
                                        attrs: attrs.clone(),
 | 
			
		||||
                                        vis: Visibility::Inherited,
 | 
			
		||||
                                        mutability: FieldMutability::None,
 | 
			
		||||
                                        ident: None,
 | 
			
		||||
                                        colon_token: None,
 | 
			
		||||
                                        ty: ty.clone().into(),
 | 
			
		||||
                                    },
 | 
			
		||||
                                    *comma_token,
 | 
			
		||||
                                )]),
 | 
			
		||||
                            }),
 | 
			
		||||
                            None => Fields::Unit,
 | 
			
		||||
                        },
 | 
			
		||||
                        discriminant: None,
 | 
			
		||||
                    },
 | 
			
		||||
                )),
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let mut enum_attrs = attrs.clone();
 | 
			
		||||
        enum_attrs.push(parse_quote_spanned! {ident.span()=>
 | 
			
		||||
            #[allow(dead_code, non_camel_case_types)]
 | 
			
		||||
        });
 | 
			
		||||
        ItemEnum {
 | 
			
		||||
            attrs: enum_attrs,
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            enum_token: *enum_token,
 | 
			
		||||
            ident: match_variant_ident.clone(),
 | 
			
		||||
            generics: generics.into(),
 | 
			
		||||
            brace_token: *brace_token,
 | 
			
		||||
            variants: Punctuated::from_iter(variants.pairs().map_pair_value_ref(
 | 
			
		||||
                |ParsedVariant {
 | 
			
		||||
                     attrs,
 | 
			
		||||
                     options: _,
 | 
			
		||||
                     ident,
 | 
			
		||||
                     field,
 | 
			
		||||
                 }| Variant {
 | 
			
		||||
                    attrs: attrs.clone(),
 | 
			
		||||
                    ident: ident.clone(),
 | 
			
		||||
                    fields: match field {
 | 
			
		||||
                        Some(ParsedVariantField {
 | 
			
		||||
                            paren_token,
 | 
			
		||||
                            attrs,
 | 
			
		||||
                            options: _,
 | 
			
		||||
                            ty,
 | 
			
		||||
                            comma_token,
 | 
			
		||||
                        }) => Fields::Unnamed(FieldsUnnamed {
 | 
			
		||||
                            paren_token: *paren_token,
 | 
			
		||||
                            unnamed: Punctuated::from_iter([Pair::new(
 | 
			
		||||
                                Field {
 | 
			
		||||
                                    attrs: attrs.clone(),
 | 
			
		||||
                                    vis: Visibility::Inherited,
 | 
			
		||||
                                    mutability: FieldMutability::None,
 | 
			
		||||
                                    ident: None,
 | 
			
		||||
                                    colon_token: None,
 | 
			
		||||
                                    ty: parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                                        ::fayalite::expr::Expr<#ty>
 | 
			
		||||
                                    },
 | 
			
		||||
                                },
 | 
			
		||||
                                *comma_token,
 | 
			
		||||
                            )]),
 | 
			
		||||
                        }),
 | 
			
		||||
                        None => Fields::Unit,
 | 
			
		||||
                    },
 | 
			
		||||
                    discriminant: None,
 | 
			
		||||
                },
 | 
			
		||||
            )),
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        for (index, ParsedVariant { ident, field, .. }) in variants.iter().enumerate() {
 | 
			
		||||
            if let Some(ParsedVariantField { ty, .. }) = field {
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    #[automatically_derived]
 | 
			
		||||
                    impl #impl_generics #target #type_generics
 | 
			
		||||
                    #where_clause
 | 
			
		||||
                    {
 | 
			
		||||
                        #[allow(non_snake_case, dead_code)]
 | 
			
		||||
                        #vis fn #ident<__V: ::fayalite::expr::ToExpr<Type = #ty>>(
 | 
			
		||||
                            self,
 | 
			
		||||
                            v: __V,
 | 
			
		||||
                        ) -> ::fayalite::expr::Expr<Self> {
 | 
			
		||||
                            ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                                &::fayalite::expr::ops::EnumLiteral::new_by_index(
 | 
			
		||||
                                    self,
 | 
			
		||||
                                    #index,
 | 
			
		||||
                                    ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                        ::fayalite::expr::Expr::canonical(
 | 
			
		||||
                                            ::fayalite::expr::ToExpr::to_expr(&v),
 | 
			
		||||
                                        ),
 | 
			
		||||
                                    ),
 | 
			
		||||
                                ),
 | 
			
		||||
                            )
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            } else {
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    #[automatically_derived]
 | 
			
		||||
                    impl #impl_generics #target #type_generics
 | 
			
		||||
                    #where_clause
 | 
			
		||||
                    {
 | 
			
		||||
                        #[allow(non_snake_case, dead_code)]
 | 
			
		||||
                        #vis fn #ident(self) -> ::fayalite::expr::Expr<Self> {
 | 
			
		||||
                            ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                                &::fayalite::expr::ops::EnumLiteral::new_by_index(
 | 
			
		||||
                                    self,
 | 
			
		||||
                                    #index,
 | 
			
		||||
                                    ::fayalite::__std::option::Option::None,
 | 
			
		||||
                                ),
 | 
			
		||||
                            )
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let from_canonical_body_fields = Vec::from_iter(variants.iter().enumerate().map(
 | 
			
		||||
            |(index, ParsedVariant { ident, field, .. })| {
 | 
			
		||||
                let ident_str = ident.to_string();
 | 
			
		||||
                let val = if field.is_some() {
 | 
			
		||||
                    let missing_value_msg = format!("expected variant {ident} to have a field");
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        ::fayalite::ty::Type::from_canonical(ty.expect(#missing_value_msg))
 | 
			
		||||
                    }
 | 
			
		||||
                } else {
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        ::fayalite::__std::assert!(ty.is_none());
 | 
			
		||||
                    }
 | 
			
		||||
                };
 | 
			
		||||
                quote_spanned! {ident.span()=>
 | 
			
		||||
                    #ident: {
 | 
			
		||||
                        let ::fayalite::enum_::EnumVariant {
 | 
			
		||||
                            name,
 | 
			
		||||
                            ty,
 | 
			
		||||
                        } = variants[#index];
 | 
			
		||||
                        ::fayalite::__std::assert_eq!(&*name, #ident_str);
 | 
			
		||||
                        #val
 | 
			
		||||
                    },
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        let match_active_scope_match_arms = Vec::from_iter(variants.iter().enumerate().map(
 | 
			
		||||
            |(index, ParsedVariant { ident, field, .. })| {
 | 
			
		||||
                if field.is_some() {
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        #index => #match_variant_ident::#ident(
 | 
			
		||||
                            ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                                &::fayalite::expr::ops::VariantAccess::new_by_index(
 | 
			
		||||
                                    variant_access.base(),
 | 
			
		||||
                                    variant_access.variant_index(),
 | 
			
		||||
                                ),
 | 
			
		||||
                            ),
 | 
			
		||||
                        ),
 | 
			
		||||
                    }
 | 
			
		||||
                } else {
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        #index => #match_variant_ident::#ident,
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        let variants_body_variants = Vec::from_iter(variants.iter().map(
 | 
			
		||||
            |ParsedVariant {
 | 
			
		||||
                 attrs: _,
 | 
			
		||||
                 options,
 | 
			
		||||
                 ident,
 | 
			
		||||
                 field,
 | 
			
		||||
             }| {
 | 
			
		||||
                let VariantOptions {} = options.body;
 | 
			
		||||
                let ident_str = ident.to_string();
 | 
			
		||||
                match field {
 | 
			
		||||
                    Some(ParsedVariantField { options, .. }) => {
 | 
			
		||||
                        let FieldOptions {} = options.body;
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            ::fayalite::enum_::EnumVariant {
 | 
			
		||||
                                name: ::fayalite::intern::Intern::intern(#ident_str),
 | 
			
		||||
                                ty: ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                    ::fayalite::ty::Type::canonical(&self.#ident),
 | 
			
		||||
                                ),
 | 
			
		||||
                            },
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    None => quote_spanned! {ident.span()=>
 | 
			
		||||
                        ::fayalite::enum_::EnumVariant {
 | 
			
		||||
                            name: ::fayalite::intern::Intern::intern(#ident_str),
 | 
			
		||||
                            ty: ::fayalite::__std::option::Option::None,
 | 
			
		||||
                        },
 | 
			
		||||
                    },
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        let variants_len = variants.len();
 | 
			
		||||
        quote_spanned! {ident.span()=>
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::Type for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type BaseType = ::fayalite::enum_::Enum;
 | 
			
		||||
                type MaskType = ::fayalite::int::Bool;
 | 
			
		||||
                type MatchVariant = #match_variant_ident #type_generics;
 | 
			
		||||
                type MatchActiveScope = ::fayalite::module::Scope;
 | 
			
		||||
                type MatchVariantAndInactiveScope = ::fayalite::enum_::EnumMatchVariantAndInactiveScope<Self>;
 | 
			
		||||
                type MatchVariantsIter = ::fayalite::enum_::EnumMatchVariantsIter<Self>;
 | 
			
		||||
 | 
			
		||||
                fn match_variants(
 | 
			
		||||
                    this: ::fayalite::expr::Expr<Self>,
 | 
			
		||||
                    source_location: ::fayalite::source_location::SourceLocation,
 | 
			
		||||
                ) -> <Self as ::fayalite::ty::Type>::MatchVariantsIter {
 | 
			
		||||
                    ::fayalite::module::enum_match_variants_helper(this, source_location)
 | 
			
		||||
                }
 | 
			
		||||
                fn mask_type(&self) -> <Self as ::fayalite::ty::Type>::MaskType {
 | 
			
		||||
                    ::fayalite::int::Bool
 | 
			
		||||
                }
 | 
			
		||||
                fn canonical(&self) -> ::fayalite::ty::CanonicalType {
 | 
			
		||||
                    ::fayalite::ty::CanonicalType::Enum(::fayalite::enum_::Enum::new(::fayalite::enum_::EnumType::variants(self)))
 | 
			
		||||
                }
 | 
			
		||||
                #[track_caller]
 | 
			
		||||
                #[allow(non_snake_case)]
 | 
			
		||||
                fn from_canonical(canonical_type: ::fayalite::ty::CanonicalType) -> Self {
 | 
			
		||||
                    let ::fayalite::ty::CanonicalType::Enum(enum_) = canonical_type else {
 | 
			
		||||
                        ::fayalite::__std::panic!("expected enum");
 | 
			
		||||
                    };
 | 
			
		||||
                    let variants = ::fayalite::enum_::EnumType::variants(&enum_);
 | 
			
		||||
                    ::fayalite::__std::assert_eq!(variants.len(), #variants_len, "enum has wrong number of variants");
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#from_canonical_body_fields)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                fn source_location() -> ::fayalite::source_location::SourceLocation {
 | 
			
		||||
                    ::fayalite::source_location::SourceLocation::caller()
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::enum_::EnumType for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn match_activate_scope(
 | 
			
		||||
                    v: <Self as ::fayalite::ty::Type>::MatchVariantAndInactiveScope,
 | 
			
		||||
                ) -> (<Self as ::fayalite::ty::Type>::MatchVariant, <Self as ::fayalite::ty::Type>::MatchActiveScope) {
 | 
			
		||||
                    let (variant_access, scope) = v.activate();
 | 
			
		||||
                    (
 | 
			
		||||
                        match variant_access.variant_index() {
 | 
			
		||||
                            #(#match_active_scope_match_arms)*
 | 
			
		||||
                            #variants_len.. => ::fayalite::__std::panic!("invalid variant index"),
 | 
			
		||||
                        },
 | 
			
		||||
                        scope,
 | 
			
		||||
                    )
 | 
			
		||||
                }
 | 
			
		||||
                fn variants(&self) -> ::fayalite::intern::Interned<[::fayalite::enum_::EnumVariant]> {
 | 
			
		||||
                    ::fayalite::intern::Intern::intern(&[
 | 
			
		||||
                        #(#variants_body_variants)*
 | 
			
		||||
                    ][..])
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        if let (None, MaybeParsed::Parsed(generics)) = (no_static, &self.generics) {
 | 
			
		||||
            let static_generics = generics.clone().for_static_type();
 | 
			
		||||
            let (static_impl_generics, static_type_generics, static_where_clause) =
 | 
			
		||||
                static_generics.split_for_impl();
 | 
			
		||||
            let static_type_body_variants =
 | 
			
		||||
                Vec::from_iter(variants.iter().map(|ParsedVariant { ident, field, .. }| {
 | 
			
		||||
                    if let Some(_) = field {
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            #ident: ::fayalite::ty::StaticType::TYPE,
 | 
			
		||||
                        }
 | 
			
		||||
                    } else {
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            #ident: (),
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }));
 | 
			
		||||
            let type_properties = format_ident!("__type_properties", span = ident.span());
 | 
			
		||||
            let type_properties_variants =
 | 
			
		||||
                Vec::from_iter(variants.iter().map(|ParsedVariant { ident, field, .. }| {
 | 
			
		||||
                    let variant = if let Some(ParsedVariantField { ty, .. }) = field {
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                <#ty as ::fayalite::ty::StaticType>::TYPE_PROPERTIES,
 | 
			
		||||
                            )
 | 
			
		||||
                        }
 | 
			
		||||
                    } else {
 | 
			
		||||
                        quote_spanned! {ident.span()=>
 | 
			
		||||
                            ::fayalite::__std::option::Option::None
 | 
			
		||||
                        }
 | 
			
		||||
                    };
 | 
			
		||||
                    quote_spanned! {ident.span()=>
 | 
			
		||||
                        let #type_properties = #type_properties.variant(#variant);
 | 
			
		||||
                    }
 | 
			
		||||
                }));
 | 
			
		||||
            quote_spanned! {ident.span()=>
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #static_impl_generics ::fayalite::ty::StaticType
 | 
			
		||||
                for #target #static_type_generics
 | 
			
		||||
                #static_where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    const TYPE: Self = Self {
 | 
			
		||||
                        #(#static_type_body_variants)*
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE: <Self as ::fayalite::ty::Type>::MaskType =
 | 
			
		||||
                        ::fayalite::int::Bool;
 | 
			
		||||
                    const TYPE_PROPERTIES: ::fayalite::ty::TypeProperties = {
 | 
			
		||||
                        let #type_properties = ::fayalite::enum_::EnumTypePropertiesBuilder::new();
 | 
			
		||||
                        #(#type_properties_variants)*
 | 
			
		||||
                        #type_properties.finish()
 | 
			
		||||
                    };
 | 
			
		||||
                    const MASK_TYPE_PROPERTIES: ::fayalite::ty::TypeProperties =
 | 
			
		||||
                        <::fayalite::int::Bool as ::fayalite::ty::StaticType>::TYPE_PROPERTIES;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn hdl_enum(item: ItemEnum) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = ParsedEnum::parse(item)?;
 | 
			
		||||
    let outline_generated = item.options.body.outline_generated;
 | 
			
		||||
    let mut contents = item.to_token_stream();
 | 
			
		||||
    if outline_generated.is_some() {
 | 
			
		||||
        contents = crate::outline_generated(contents, "hdl-enum-");
 | 
			
		||||
    }
 | 
			
		||||
    Ok(contents)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										3210
									
								
								crates/fayalite-proc-macros-impl/src/hdl_type_common.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										3210
									
								
								crates/fayalite-proc-macros-impl/src/hdl_type_common.rs
									
										
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -13,15 +13,15 @@ use syn::{
 | 
			
		|||
};
 | 
			
		||||
 | 
			
		||||
mod fold;
 | 
			
		||||
mod hdl_bundle;
 | 
			
		||||
mod hdl_enum;
 | 
			
		||||
mod hdl_type_common;
 | 
			
		||||
mod module;
 | 
			
		||||
mod value_derive_common;
 | 
			
		||||
mod value_derive_enum;
 | 
			
		||||
mod value_derive_struct;
 | 
			
		||||
//mod value_derive_common;
 | 
			
		||||
//mod value_derive_struct;
 | 
			
		||||
 | 
			
		||||
mod kw {
 | 
			
		||||
    pub(crate) use syn::token::{
 | 
			
		||||
        Enum as enum_, Extern as extern_, Static as static_, Struct as struct_, Where as where_,
 | 
			
		||||
    };
 | 
			
		||||
    pub(crate) use syn::token::Extern as extern_;
 | 
			
		||||
 | 
			
		||||
    macro_rules! custom_keyword {
 | 
			
		||||
        ($kw:ident) => {
 | 
			
		||||
| 
						 | 
				
			
			@ -43,6 +43,7 @@ mod kw {
 | 
			
		|||
 | 
			
		||||
    custom_keyword!(clock_domain);
 | 
			
		||||
    custom_keyword!(connect_inexact);
 | 
			
		||||
    custom_keyword!(custom_bounds);
 | 
			
		||||
    custom_keyword!(flip);
 | 
			
		||||
    custom_keyword!(hdl);
 | 
			
		||||
    custom_keyword!(input);
 | 
			
		||||
| 
						 | 
				
			
			@ -52,6 +53,8 @@ mod kw {
 | 
			
		|||
    custom_keyword!(memory_array);
 | 
			
		||||
    custom_keyword!(memory_with_init);
 | 
			
		||||
    custom_keyword!(no_reset);
 | 
			
		||||
    custom_keyword!(no_runtime_generics);
 | 
			
		||||
    custom_keyword!(no_static);
 | 
			
		||||
    custom_keyword!(outline_generated);
 | 
			
		||||
    custom_keyword!(output);
 | 
			
		||||
    custom_keyword!(reg_builder);
 | 
			
		||||
| 
						 | 
				
			
			@ -519,6 +522,26 @@ macro_rules! impl_extra_traits_for_options {
 | 
			
		|||
    ) => {
 | 
			
		||||
        impl Copy for $option_enum_name {}
 | 
			
		||||
 | 
			
		||||
        impl PartialEq for $option_enum_name {
 | 
			
		||||
            fn eq(&self, other: &Self) -> bool {
 | 
			
		||||
                self.cmp(other).is_eq()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl Eq for $option_enum_name {}
 | 
			
		||||
 | 
			
		||||
        impl PartialOrd for $option_enum_name {
 | 
			
		||||
            fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
 | 
			
		||||
                Some(self.cmp(other))
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl Ord for $option_enum_name {
 | 
			
		||||
            fn cmp(&self, other: &Self) -> std::cmp::Ordering {
 | 
			
		||||
                self.variant().cmp(&other.variant())
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl quote::IdentFragment for $option_enum_name {
 | 
			
		||||
            fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
 | 
			
		||||
                let _ = f;
 | 
			
		||||
| 
						 | 
				
			
			@ -554,6 +577,66 @@ pub(crate) use impl_extra_traits_for_options;
 | 
			
		|||
macro_rules! options {
 | 
			
		||||
    (
 | 
			
		||||
        #[options = $options_name:ident]
 | 
			
		||||
        $($tt:tt)*
 | 
			
		||||
    ) => {
 | 
			
		||||
        crate::options! {
 | 
			
		||||
            #[options = $options_name, punct = syn::Token![,], allow_duplicates = false]
 | 
			
		||||
            $($tt)*
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
    (
 | 
			
		||||
        #[options = $options_name:ident, punct = $Punct:ty, allow_duplicates = true]
 | 
			
		||||
        $(#[$($enum_meta:tt)*])*
 | 
			
		||||
        $enum_vis:vis enum $option_enum_name:ident {
 | 
			
		||||
            $($Variant:ident($key:ident $(, $value:ty)?),)*
 | 
			
		||||
        }
 | 
			
		||||
    ) => {
 | 
			
		||||
        crate::options! {
 | 
			
		||||
            #[options = $options_name, punct = $Punct, allow_duplicates = (true)]
 | 
			
		||||
            $(#[$($enum_meta)*])*
 | 
			
		||||
            $enum_vis enum $option_enum_name {
 | 
			
		||||
                $($Variant($key $(, $value)?),)*
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl Extend<$option_enum_name> for $options_name {
 | 
			
		||||
            fn extend<T: IntoIterator<Item = $option_enum_name>>(&mut self, iter: T) {
 | 
			
		||||
                iter.into_iter().for_each(|v| match v {
 | 
			
		||||
                    $($option_enum_name::$Variant(v) => {
 | 
			
		||||
                        self.$key = Some(v);
 | 
			
		||||
                    })*
 | 
			
		||||
                });
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl FromIterator<$option_enum_name> for $options_name {
 | 
			
		||||
            fn from_iter<T: IntoIterator<Item = $option_enum_name>>(iter: T) -> Self {
 | 
			
		||||
                let mut retval = Self::default();
 | 
			
		||||
                retval.extend(iter);
 | 
			
		||||
                retval
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl Extend<$options_name> for $options_name {
 | 
			
		||||
            fn extend<T: IntoIterator<Item = $options_name>>(&mut self, iter: T) {
 | 
			
		||||
                iter.into_iter().for_each(|v| {
 | 
			
		||||
                    $(if let Some(v) = v.$key {
 | 
			
		||||
                        self.$key = Some(v);
 | 
			
		||||
                    })*
 | 
			
		||||
                });
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl FromIterator<$options_name> for $options_name {
 | 
			
		||||
            fn from_iter<T: IntoIterator<Item = $options_name>>(iter: T) -> Self {
 | 
			
		||||
                let mut retval = Self::default();
 | 
			
		||||
                retval.extend(iter);
 | 
			
		||||
                retval
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
    (
 | 
			
		||||
        #[options = $options_name:ident, punct = $Punct:ty, allow_duplicates = $allow_duplicates:expr]
 | 
			
		||||
        $(#[$($enum_meta:tt)*])*
 | 
			
		||||
        $enum_vis:vis enum $option_enum_name:ident {
 | 
			
		||||
            $($Variant:ident($key:ident $(, $value:ty)?),)*
 | 
			
		||||
| 
						 | 
				
			
			@ -567,8 +650,11 @@ macro_rules! options {
 | 
			
		|||
        }
 | 
			
		||||
 | 
			
		||||
        #[derive(Clone, Debug, Default)]
 | 
			
		||||
        #[allow(non_snake_case)]
 | 
			
		||||
        $enum_vis struct $options_name {
 | 
			
		||||
            $($enum_vis $key: Option<(crate::kw::$key, $(syn::token::Paren, $value)?)>,)*
 | 
			
		||||
            $(
 | 
			
		||||
                $enum_vis $key: Option<(crate::kw::$key, $(syn::token::Paren, $value)?)>,
 | 
			
		||||
            )*
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        crate::fold::impl_fold! {
 | 
			
		||||
| 
						 | 
				
			
			@ -577,6 +663,43 @@ macro_rules! options {
 | 
			
		|||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const _: () = {
 | 
			
		||||
            #[derive(Clone, Debug)]
 | 
			
		||||
            $enum_vis struct Iter($enum_vis $options_name);
 | 
			
		||||
 | 
			
		||||
            impl IntoIterator for $options_name {
 | 
			
		||||
                type Item = $option_enum_name;
 | 
			
		||||
                type IntoIter = Iter;
 | 
			
		||||
 | 
			
		||||
                fn into_iter(self) -> Self::IntoIter {
 | 
			
		||||
                    Iter(self)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            impl Iterator for Iter {
 | 
			
		||||
                type Item = $option_enum_name;
 | 
			
		||||
 | 
			
		||||
                fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
                    $(
 | 
			
		||||
                        if let Some(value) = self.0.$key.take() {
 | 
			
		||||
                            return Some($option_enum_name::$Variant(value));
 | 
			
		||||
                        }
 | 
			
		||||
                    )*
 | 
			
		||||
                    None
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                #[allow(unused_mut, unused_variables)]
 | 
			
		||||
                fn fold<B, F: FnMut(B, Self::Item) -> B>(mut self, mut init: B, mut f: F) -> B {
 | 
			
		||||
                    $(
 | 
			
		||||
                        if let Some(value) = self.0.$key.take() {
 | 
			
		||||
                            init = f(init, $option_enum_name::$Variant(value));
 | 
			
		||||
                        }
 | 
			
		||||
                    )*
 | 
			
		||||
                    init
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        impl syn::parse::Parse for $options_name {
 | 
			
		||||
            fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
 | 
			
		||||
                #![allow(unused_mut, unused_variables, unreachable_code)]
 | 
			
		||||
| 
						 | 
				
			
			@ -585,7 +708,7 @@ macro_rules! options {
 | 
			
		|||
                    let old_input = input.fork();
 | 
			
		||||
                    match input.parse::<$option_enum_name>()? {
 | 
			
		||||
                        $($option_enum_name::$Variant(v) => {
 | 
			
		||||
                            if retval.$key.replace(v).is_some() {
 | 
			
		||||
                            if retval.$key.replace(v).is_some() && !$allow_duplicates {
 | 
			
		||||
                                return Err(old_input.error(concat!("duplicate ", stringify!($key), " option")));
 | 
			
		||||
                            }
 | 
			
		||||
                        })*
 | 
			
		||||
| 
						 | 
				
			
			@ -593,7 +716,7 @@ macro_rules! options {
 | 
			
		|||
                    if input.is_empty() {
 | 
			
		||||
                        break;
 | 
			
		||||
                    }
 | 
			
		||||
                    input.parse::<syn::Token![,]>()?;
 | 
			
		||||
                    input.parse::<$Punct>()?;
 | 
			
		||||
                }
 | 
			
		||||
                Ok(retval)
 | 
			
		||||
            }
 | 
			
		||||
| 
						 | 
				
			
			@ -602,7 +725,7 @@ macro_rules! options {
 | 
			
		|||
        impl quote::ToTokens for $options_name {
 | 
			
		||||
            #[allow(unused_mut, unused_variables, unused_assignments)]
 | 
			
		||||
            fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
 | 
			
		||||
                let mut separator: Option<syn::Token![,]> = None;
 | 
			
		||||
                let mut separator: Option<$Punct> = None;
 | 
			
		||||
                $(if let Some(v) = &self.$key {
 | 
			
		||||
                    separator.to_tokens(tokens);
 | 
			
		||||
                    separator = Some(Default::default());
 | 
			
		||||
| 
						 | 
				
			
			@ -673,6 +796,20 @@ macro_rules! options {
 | 
			
		|||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl $option_enum_name {
 | 
			
		||||
            #[allow(dead_code)]
 | 
			
		||||
            fn variant(&self) -> usize {
 | 
			
		||||
                #[repr(usize)]
 | 
			
		||||
                enum Variant {
 | 
			
		||||
                    $($Variant,)*
 | 
			
		||||
                    __Last, // so it doesn't complain about zero-variant enums
 | 
			
		||||
                }
 | 
			
		||||
                match *self {
 | 
			
		||||
                    $(Self::$Variant(..) => Variant::$Variant as usize,)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -686,6 +823,15 @@ pub(crate) fn outline_generated(contents: TokenStream, prefix: &str) -> TokenStr
 | 
			
		|||
        .suffix(".tmp.rs")
 | 
			
		||||
        .tempfile_in(out_dir)
 | 
			
		||||
        .unwrap();
 | 
			
		||||
    struct PrintOnPanic<'a>(&'a TokenStream);
 | 
			
		||||
    impl Drop for PrintOnPanic<'_> {
 | 
			
		||||
        fn drop(&mut self) {
 | 
			
		||||
            if std::thread::panicking() {
 | 
			
		||||
                println!("{}", self.0);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    let _print_on_panic = PrintOnPanic(&contents);
 | 
			
		||||
    let contents = prettyplease::unparse(&parse_quote! { #contents });
 | 
			
		||||
    let hash = <sha2::Sha256 as sha2::Digest>::digest(&contents);
 | 
			
		||||
    let hash = base16ct::HexDisplay(&hash[..5]);
 | 
			
		||||
| 
						 | 
				
			
			@ -706,7 +852,7 @@ pub(crate) fn outline_generated(contents: TokenStream, prefix: &str) -> TokenStr
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn module(attr: TokenStream, item: TokenStream) -> syn::Result<TokenStream> {
 | 
			
		||||
pub fn hdl_module(attr: TokenStream, item: TokenStream) -> syn::Result<TokenStream> {
 | 
			
		||||
    let options = syn::parse2::<module::ConfigOptions>(attr)?;
 | 
			
		||||
    let options = HdlAttr::from(options);
 | 
			
		||||
    let func = syn::parse2::<module::ModuleFn>(quote! { #options #item })?;
 | 
			
		||||
| 
						 | 
				
			
			@ -717,14 +863,14 @@ pub fn module(attr: TokenStream, item: TokenStream) -> syn::Result<TokenStream>
 | 
			
		|||
    Ok(contents)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn value_derive(item: TokenStream) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = syn::parse2::<Item>(item)?;
 | 
			
		||||
pub fn hdl_attr(attr: TokenStream, item: TokenStream) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = syn::parse2::<Item>(quote! { #[hdl(#attr)] #item })?;
 | 
			
		||||
    match item {
 | 
			
		||||
        Item::Enum(item) => value_derive_enum::value_derive_enum(item),
 | 
			
		||||
        Item::Struct(item) => value_derive_struct::value_derive_struct(item),
 | 
			
		||||
        Item::Enum(item) => hdl_enum::hdl_enum(item),
 | 
			
		||||
        Item::Struct(item) => hdl_bundle::hdl_bundle(item),
 | 
			
		||||
        _ => Err(syn::Error::new(
 | 
			
		||||
            Span::call_site(),
 | 
			
		||||
            "derive(Value) can only be used on structs or enums",
 | 
			
		||||
            "top-level #[hdl] can only be used on structs or enums",
 | 
			
		||||
        )),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,7 +1,7 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    is_hdl_attr,
 | 
			
		||||
    hdl_type_common::{ParsedGenerics, SplitForImpl},
 | 
			
		||||
    module::transform_body::{HdlLet, HdlLetKindIO},
 | 
			
		||||
    options, Errors, HdlAttr, PairsIterExt,
 | 
			
		||||
};
 | 
			
		||||
| 
						 | 
				
			
			@ -57,13 +57,6 @@ impl Visit<'_> for CheckNameConflictsWithModuleBuilderVisitor<'_> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn retain_struct_attrs<F: FnMut(&Attribute) -> bool>(item: &mut ItemStruct, mut f: F) {
 | 
			
		||||
    item.attrs.retain(&mut f);
 | 
			
		||||
    for field in item.fields.iter_mut() {
 | 
			
		||||
        field.attrs.retain(&mut f);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) type ModuleIO = HdlLet<HdlLetKindIO>;
 | 
			
		||||
 | 
			
		||||
pub(crate) struct ModuleFn {
 | 
			
		||||
| 
						 | 
				
			
			@ -74,7 +67,7 @@ pub(crate) struct ModuleFn {
 | 
			
		|||
    sig: Signature,
 | 
			
		||||
    block: Box<Block>,
 | 
			
		||||
    io: Vec<ModuleIO>,
 | 
			
		||||
    struct_generics: Generics,
 | 
			
		||||
    struct_generics: ParsedGenerics,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Copy, PartialEq, Eq, Debug, Hash)]
 | 
			
		||||
| 
						 | 
				
			
			@ -220,8 +213,16 @@ impl Parse for ModuleFn {
 | 
			
		|||
                "return type not allowed here",
 | 
			
		||||
            ));
 | 
			
		||||
        }
 | 
			
		||||
        let body_results = errors.ok(transform_body::transform_body(module_kind, block));
 | 
			
		||||
        let struct_generics = errors.ok(ParsedGenerics::parse(&mut { struct_generics }));
 | 
			
		||||
        let body_results = struct_generics.as_ref().and_then(|struct_generics| {
 | 
			
		||||
            errors.ok(transform_body::transform_body(
 | 
			
		||||
                module_kind,
 | 
			
		||||
                block,
 | 
			
		||||
                struct_generics,
 | 
			
		||||
            ))
 | 
			
		||||
        });
 | 
			
		||||
        errors.finish()?;
 | 
			
		||||
        let struct_generics = struct_generics.unwrap();
 | 
			
		||||
        let (block, io) = body_results.unwrap();
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            attrs,
 | 
			
		||||
| 
						 | 
				
			
			@ -273,19 +274,18 @@ impl ModuleFn {
 | 
			
		|||
                });
 | 
			
		||||
                name
 | 
			
		||||
            }));
 | 
			
		||||
        let module_kind_ty = match module_kind {
 | 
			
		||||
            ModuleKind::Extern => quote! { ::fayalite::module::ExternModule },
 | 
			
		||||
            ModuleKind::Normal => quote! { ::fayalite::module::NormalModule },
 | 
			
		||||
        let module_kind_value = match module_kind {
 | 
			
		||||
            ModuleKind::Extern => quote! { ::fayalite::module::ModuleKind::Extern },
 | 
			
		||||
            ModuleKind::Normal => quote! { ::fayalite::module::ModuleKind::Normal },
 | 
			
		||||
        };
 | 
			
		||||
        let fn_name = &outer_sig.ident;
 | 
			
		||||
        let (_struct_impl_generics, struct_type_generics, struct_where_clause) =
 | 
			
		||||
            struct_generics.split_for_impl();
 | 
			
		||||
        let struct_ty = quote! {#fn_name #struct_type_generics};
 | 
			
		||||
        body_sig.ident = parse_quote! {__body};
 | 
			
		||||
        body_sig.inputs.insert(
 | 
			
		||||
            0,
 | 
			
		||||
            parse_quote! {m: &mut ::fayalite::module::ModuleBuilder<#struct_ty, #module_kind_ty>},
 | 
			
		||||
        );
 | 
			
		||||
        body_sig
 | 
			
		||||
            .inputs
 | 
			
		||||
            .insert(0, parse_quote! { m: &::fayalite::module::ModuleBuilder });
 | 
			
		||||
        let body_fn = ItemFn {
 | 
			
		||||
            attrs: vec![],
 | 
			
		||||
            vis: Visibility::Inherited,
 | 
			
		||||
| 
						 | 
				
			
			@ -310,34 +310,21 @@ impl ModuleFn {
 | 
			
		|||
        let body_turbofish_type_generics = body_type_generics.as_turbofish();
 | 
			
		||||
        let block = parse_quote! {{
 | 
			
		||||
            #body_fn
 | 
			
		||||
            ::fayalite::module::ModuleBuilder::run(#fn_name_str, |m| __body #body_turbofish_type_generics(m, #(#param_names,)*))
 | 
			
		||||
            ::fayalite::module::ModuleBuilder::run(
 | 
			
		||||
                #fn_name_str,
 | 
			
		||||
                #module_kind_value,
 | 
			
		||||
                |m| __body #body_turbofish_type_generics(m, #(#param_names,)*),
 | 
			
		||||
            )
 | 
			
		||||
        }};
 | 
			
		||||
        let static_type = io.iter().all(|io| io.kind.ty_expr.is_none());
 | 
			
		||||
        let struct_options = if static_type {
 | 
			
		||||
            quote! { #[hdl(static)] }
 | 
			
		||||
        } else {
 | 
			
		||||
            quote! {}
 | 
			
		||||
        };
 | 
			
		||||
        let the_struct: ItemStruct = parse_quote! {
 | 
			
		||||
            #[derive(::fayalite::__std::clone::Clone,
 | 
			
		||||
                ::fayalite::__std::hash::Hash,
 | 
			
		||||
                ::fayalite::__std::cmp::PartialEq,
 | 
			
		||||
                ::fayalite::__std::cmp::Eq,
 | 
			
		||||
                ::fayalite::__std::fmt::Debug)]
 | 
			
		||||
            #[allow(non_camel_case_types)]
 | 
			
		||||
            #struct_options
 | 
			
		||||
            #[hdl(no_runtime_generics, no_static)]
 | 
			
		||||
            #vis struct #fn_name #struct_generics #struct_where_clause {
 | 
			
		||||
                #(
 | 
			
		||||
                #io_flips
 | 
			
		||||
                #vis #io_names: #io_types,)*
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        let mut struct_without_hdl_attrs = the_struct.clone();
 | 
			
		||||
        let mut struct_without_derives = the_struct;
 | 
			
		||||
        retain_struct_attrs(&mut struct_without_hdl_attrs, |attr| !is_hdl_attr(attr));
 | 
			
		||||
        retain_struct_attrs(&mut struct_without_derives, |attr| {
 | 
			
		||||
            !attr.path().is_ident("derive")
 | 
			
		||||
        });
 | 
			
		||||
        let outer_fn = ItemFn {
 | 
			
		||||
            attrs,
 | 
			
		||||
            vis,
 | 
			
		||||
| 
						 | 
				
			
			@ -345,10 +332,7 @@ impl ModuleFn {
 | 
			
		|||
            block,
 | 
			
		||||
        };
 | 
			
		||||
        let mut retval = outer_fn.into_token_stream();
 | 
			
		||||
        struct_without_hdl_attrs.to_tokens(&mut retval);
 | 
			
		||||
        retval.extend(
 | 
			
		||||
            crate::value_derive_struct::value_derive_struct(struct_without_derives).unwrap(),
 | 
			
		||||
        );
 | 
			
		||||
        retval.extend(crate::hdl_bundle::hdl_bundle(the_struct).unwrap());
 | 
			
		||||
        retval
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,11 +2,14 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    fold::{impl_fold, DoFold},
 | 
			
		||||
    hdl_type_common::{
 | 
			
		||||
        known_items, ParseFailed, ParseTypes, ParsedGenerics, ParsedType, TypesParser,
 | 
			
		||||
    },
 | 
			
		||||
    is_hdl_attr, kw,
 | 
			
		||||
    module::{check_name_conflicts_with_module_builder, ModuleIO, ModuleIOKind, ModuleKind},
 | 
			
		||||
    options, Errors, HdlAttr,
 | 
			
		||||
};
 | 
			
		||||
use num_bigint::{BigInt, Sign};
 | 
			
		||||
use num_bigint::BigInt;
 | 
			
		||||
use proc_macro2::{Span, TokenStream};
 | 
			
		||||
use quote::{quote, quote_spanned, ToTokens};
 | 
			
		||||
use std::{borrow::Borrow, convert::Infallible};
 | 
			
		||||
| 
						 | 
				
			
			@ -87,9 +90,9 @@ macro_rules! with_debug_clone_and_fold {
 | 
			
		|||
pub(crate) use with_debug_clone_and_fold;
 | 
			
		||||
 | 
			
		||||
with_debug_clone_and_fold! {
 | 
			
		||||
    pub(crate) struct HdlLetKindIO<Kind = ModuleIOKind> {
 | 
			
		||||
    pub(crate) struct HdlLetKindIO<Kind = ModuleIOKind, Ty = ParsedType> {
 | 
			
		||||
        pub(crate) colon_token: Token![:],
 | 
			
		||||
        pub(crate) ty: Box<Type>,
 | 
			
		||||
        pub(crate) ty: Box<Ty>,
 | 
			
		||||
        pub(crate) m: kw::m,
 | 
			
		||||
        pub(crate) dot_token: Token![.],
 | 
			
		||||
        pub(crate) kind: Kind,
 | 
			
		||||
| 
						 | 
				
			
			@ -98,6 +101,32 @@ with_debug_clone_and_fold! {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<Kind: Clone, T: ParseTypes<I>, I> ParseTypes<HdlLetKindIO<Kind, I>> for HdlLetKindIO<Kind, T> {
 | 
			
		||||
    fn parse_types(
 | 
			
		||||
        input: &mut HdlLetKindIO<Kind, I>,
 | 
			
		||||
        parser: &mut TypesParser<'_>,
 | 
			
		||||
    ) -> Result<Self, ParseFailed> {
 | 
			
		||||
        let HdlLetKindIO {
 | 
			
		||||
            colon_token,
 | 
			
		||||
            ty,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            kind,
 | 
			
		||||
            paren,
 | 
			
		||||
            ty_expr,
 | 
			
		||||
        } = input;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            colon_token: *colon_token,
 | 
			
		||||
            ty: ParseTypes::parse_types(ty, parser)?,
 | 
			
		||||
            m: *m,
 | 
			
		||||
            dot_token: *dot_token,
 | 
			
		||||
            kind: kind.clone(),
 | 
			
		||||
            paren: *paren,
 | 
			
		||||
            ty_expr: ty_expr.clone(),
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn parse_single_fn_arg(input: ParseStream) -> syn::Result<Box<Expr>> {
 | 
			
		||||
    let retval = input.parse()?;
 | 
			
		||||
    let _: Option<Token![,]> = input.parse()?;
 | 
			
		||||
| 
						 | 
				
			
			@ -145,17 +174,19 @@ impl<Kind: ToTokens> HdlLetKindToTokens for HdlLetKindIO<Kind> {
 | 
			
		|||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct HdlLetKindInstance {
 | 
			
		||||
    pub(crate) m: kw::m,
 | 
			
		||||
    pub(crate) dot_token: Token![.],
 | 
			
		||||
    pub(crate) instance: kw::instance,
 | 
			
		||||
    pub(crate) paren: Paren,
 | 
			
		||||
    pub(crate) module: Box<Expr>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParseTypes<Self> for HdlLetKindInstance {
 | 
			
		||||
    fn parse_types(input: &mut Self, _parser: &mut TypesParser<'_>) -> Result<Self, ParseFailed> {
 | 
			
		||||
        Ok(input.clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    struct HdlLetKindInstance<> {
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        instance: kw::instance,
 | 
			
		||||
        paren: Paren,
 | 
			
		||||
        module: Box<Expr>,
 | 
			
		||||
| 
						 | 
				
			
			@ -167,14 +198,10 @@ impl HdlLetKindToTokens for HdlLetKindInstance {
 | 
			
		|||
 | 
			
		||||
    fn expr_to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            instance,
 | 
			
		||||
            paren,
 | 
			
		||||
            module,
 | 
			
		||||
        } = self;
 | 
			
		||||
        m.to_tokens(tokens);
 | 
			
		||||
        dot_token.to_tokens(tokens);
 | 
			
		||||
        instance.to_tokens(tokens);
 | 
			
		||||
        paren.surround(tokens, |tokens| module.to_tokens(tokens));
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -381,19 +408,21 @@ make_builder_method_enum! {
 | 
			
		|||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct HdlLetKindRegBuilder {
 | 
			
		||||
    pub(crate) ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
    pub(crate) m: kw::m,
 | 
			
		||||
    pub(crate) dot_token: Token![.],
 | 
			
		||||
    pub(crate) reg_builder: kw::reg_builder,
 | 
			
		||||
    pub(crate) reg_builder_paren: Paren,
 | 
			
		||||
    pub(crate) clock_domain: Option<RegBuilderClockDomain>,
 | 
			
		||||
    pub(crate) reset: RegBuilderReset,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParseTypes<Self> for HdlLetKindRegBuilder {
 | 
			
		||||
    fn parse_types(input: &mut Self, _parser: &mut TypesParser<'_>) -> Result<Self, ParseFailed> {
 | 
			
		||||
        Ok(input.clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    struct HdlLetKindRegBuilder<> {
 | 
			
		||||
        ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        reg_builder: kw::reg_builder,
 | 
			
		||||
        reg_builder_paren: Paren,
 | 
			
		||||
        clock_domain: Option<RegBuilderClockDomain>,
 | 
			
		||||
| 
						 | 
				
			
			@ -406,10 +435,10 @@ impl HdlLetKindRegBuilder {
 | 
			
		|||
        input: ParseStream,
 | 
			
		||||
        parsed_ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        _after_ty: Token![=],
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        m_dot: Option<(kw::m, Token![.])>,
 | 
			
		||||
        reg_builder: kw::reg_builder,
 | 
			
		||||
    ) -> syn::Result<Self> {
 | 
			
		||||
        check_empty_m_dot(m_dot, reg_builder)?;
 | 
			
		||||
        let _reg_builder_paren_inner;
 | 
			
		||||
        let reg_builder_paren = parenthesized!(_reg_builder_paren_inner in input);
 | 
			
		||||
        let mut clock_domain = None;
 | 
			
		||||
| 
						 | 
				
			
			@ -430,8 +459,6 @@ impl HdlLetKindRegBuilder {
 | 
			
		|||
        }
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            ty: parsed_ty,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            reg_builder,
 | 
			
		||||
            reg_builder_paren,
 | 
			
		||||
            clock_domain,
 | 
			
		||||
| 
						 | 
				
			
			@ -451,15 +478,11 @@ impl HdlLetKindToTokens for HdlLetKindRegBuilder {
 | 
			
		|||
    fn expr_to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            ty: _,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            reg_builder,
 | 
			
		||||
            reg_builder_paren,
 | 
			
		||||
            clock_domain,
 | 
			
		||||
            reset,
 | 
			
		||||
        } = self;
 | 
			
		||||
        m.to_tokens(tokens);
 | 
			
		||||
        dot_token.to_tokens(tokens);
 | 
			
		||||
        reg_builder.to_tokens(tokens);
 | 
			
		||||
        reg_builder_paren.surround(tokens, |_tokens| {});
 | 
			
		||||
        clock_domain.to_tokens(tokens);
 | 
			
		||||
| 
						 | 
				
			
			@ -470,18 +493,20 @@ impl HdlLetKindToTokens for HdlLetKindRegBuilder {
 | 
			
		|||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct HdlLetKindWire {
 | 
			
		||||
    pub(crate) ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
    pub(crate) m: kw::m,
 | 
			
		||||
    pub(crate) dot_token: Token![.],
 | 
			
		||||
    pub(crate) wire: kw::wire,
 | 
			
		||||
    pub(crate) paren: Paren,
 | 
			
		||||
    pub(crate) ty_expr: Option<Box<Expr>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParseTypes<Self> for HdlLetKindWire {
 | 
			
		||||
    fn parse_types(input: &mut Self, _parser: &mut TypesParser<'_>) -> Result<Self, ParseFailed> {
 | 
			
		||||
        Ok(input.clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    struct HdlLetKindWire<> {
 | 
			
		||||
        ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        wire: kw::wire,
 | 
			
		||||
        paren: Paren,
 | 
			
		||||
        ty_expr: Option<Box<Expr>>,
 | 
			
		||||
| 
						 | 
				
			
			@ -499,14 +524,10 @@ impl HdlLetKindToTokens for HdlLetKindWire {
 | 
			
		|||
    fn expr_to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            ty: _,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            wire,
 | 
			
		||||
            paren,
 | 
			
		||||
            ty_expr,
 | 
			
		||||
        } = self;
 | 
			
		||||
        m.to_tokens(tokens);
 | 
			
		||||
        dot_token.to_tokens(tokens);
 | 
			
		||||
        wire.to_tokens(tokens);
 | 
			
		||||
        paren.surround(tokens, |tokens| ty_expr.to_tokens(tokens));
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -627,16 +648,18 @@ impl ToTokens for MemoryFn {
 | 
			
		|||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct HdlLetKindMemory {
 | 
			
		||||
    pub(crate) ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
    pub(crate) m: kw::m,
 | 
			
		||||
    pub(crate) dot_token: Token![.],
 | 
			
		||||
    pub(crate) memory_fn: MemoryFn,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParseTypes<Self> for HdlLetKindMemory {
 | 
			
		||||
    fn parse_types(input: &mut Self, _parser: &mut TypesParser<'_>) -> Result<Self, ParseFailed> {
 | 
			
		||||
        Ok(input.clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    struct HdlLetKindMemory<> {
 | 
			
		||||
        ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        memory_fn: MemoryFn,
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -650,14 +673,7 @@ impl HdlLetKindToTokens for HdlLetKindMemory {
 | 
			
		|||
    }
 | 
			
		||||
 | 
			
		||||
    fn expr_to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            ty: _,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            memory_fn,
 | 
			
		||||
        } = self;
 | 
			
		||||
        m.to_tokens(tokens);
 | 
			
		||||
        dot_token.to_tokens(tokens);
 | 
			
		||||
        let Self { ty: _, memory_fn } = self;
 | 
			
		||||
        memory_fn.to_tokens(tokens);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -667,22 +683,20 @@ impl HdlLetKindMemory {
 | 
			
		|||
        input: ParseStream,
 | 
			
		||||
        parsed_ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        _after_ty: Token![=],
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        m_dot: Option<(kw::m, Token![.])>,
 | 
			
		||||
        memory_fn_name: MemoryFnName,
 | 
			
		||||
    ) -> syn::Result<Self> {
 | 
			
		||||
        check_empty_m_dot(m_dot, memory_fn_name)?;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            ty: parsed_ty,
 | 
			
		||||
            m,
 | 
			
		||||
            dot_token,
 | 
			
		||||
            memory_fn: MemoryFn::parse_rest(input, memory_fn_name)?,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) enum HdlLetKind {
 | 
			
		||||
    IO(HdlLetKindIO),
 | 
			
		||||
pub(crate) enum HdlLetKind<IOType = ParsedType> {
 | 
			
		||||
    IO(HdlLetKindIO<ModuleIOKind, IOType>),
 | 
			
		||||
    Instance(HdlLetKindInstance),
 | 
			
		||||
    RegBuilder(HdlLetKindRegBuilder),
 | 
			
		||||
    Wire(HdlLetKindWire),
 | 
			
		||||
| 
						 | 
				
			
			@ -690,8 +704,8 @@ pub(crate) enum HdlLetKind {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    enum HdlLetKind<> {
 | 
			
		||||
        IO(HdlLetKindIO),
 | 
			
		||||
    enum HdlLetKind<IOType,> {
 | 
			
		||||
        IO(HdlLetKindIO<ModuleIOKind, IOType>),
 | 
			
		||||
        Instance(HdlLetKindInstance),
 | 
			
		||||
        RegBuilder(HdlLetKindRegBuilder),
 | 
			
		||||
        Wire(HdlLetKindWire),
 | 
			
		||||
| 
						 | 
				
			
			@ -699,6 +713,27 @@ impl_fold! {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ParseTypes<I>, I> ParseTypes<HdlLetKind<I>> for HdlLetKind<T> {
 | 
			
		||||
    fn parse_types(
 | 
			
		||||
        input: &mut HdlLetKind<I>,
 | 
			
		||||
        parser: &mut TypesParser<'_>,
 | 
			
		||||
    ) -> Result<Self, ParseFailed> {
 | 
			
		||||
        match input {
 | 
			
		||||
            HdlLetKind::IO(input) => ParseTypes::parse_types(input, parser).map(HdlLetKind::IO),
 | 
			
		||||
            HdlLetKind::Instance(input) => {
 | 
			
		||||
                ParseTypes::parse_types(input, parser).map(HdlLetKind::Instance)
 | 
			
		||||
            }
 | 
			
		||||
            HdlLetKind::RegBuilder(input) => {
 | 
			
		||||
                ParseTypes::parse_types(input, parser).map(HdlLetKind::RegBuilder)
 | 
			
		||||
            }
 | 
			
		||||
            HdlLetKind::Wire(input) => ParseTypes::parse_types(input, parser).map(HdlLetKind::Wire),
 | 
			
		||||
            HdlLetKind::Memory(input) => {
 | 
			
		||||
                ParseTypes::parse_types(input, parser).map(HdlLetKind::Memory)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn parsed_ty_or_err(
 | 
			
		||||
    parsed_ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
    after_ty: Token![=],
 | 
			
		||||
| 
						 | 
				
			
			@ -710,15 +745,15 @@ fn parsed_ty_or_err(
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl HdlLetKindIO {
 | 
			
		||||
impl HdlLetKindIO<ModuleIOKind, Type> {
 | 
			
		||||
    fn rest_of_parse(
 | 
			
		||||
        input: ParseStream,
 | 
			
		||||
        parsed_ty: Option<(Token![:], Box<Type>)>,
 | 
			
		||||
        after_ty: Token![=],
 | 
			
		||||
        m: kw::m,
 | 
			
		||||
        dot_token: Token![.],
 | 
			
		||||
        m_dot: Option<(kw::m, Token![.])>,
 | 
			
		||||
        kind: ModuleIOKind,
 | 
			
		||||
    ) -> syn::Result<Self> {
 | 
			
		||||
        let (m, dot_token) = unwrap_m_dot(m_dot, kind)?;
 | 
			
		||||
        let (colon_token, ty) = parsed_ty_or_err(parsed_ty, after_ty)?;
 | 
			
		||||
        let paren_contents;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
| 
						 | 
				
			
			@ -733,7 +768,36 @@ impl HdlLetKindIO {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl HdlLetKindParse for HdlLetKind {
 | 
			
		||||
fn check_empty_m_dot(m_dot: Option<(kw::m, Token![.])>, kind: impl ToTokens) -> syn::Result<()> {
 | 
			
		||||
    if let Some((m, dot_token)) = m_dot {
 | 
			
		||||
        Err(Error::new_spanned(
 | 
			
		||||
            quote! {#m #dot_token #kind},
 | 
			
		||||
            format_args!(
 | 
			
		||||
                "{} is a free function, not a method of ModuleBuilder: try removing the `m.`",
 | 
			
		||||
                kind.to_token_stream()
 | 
			
		||||
            ),
 | 
			
		||||
        ))
 | 
			
		||||
    } else {
 | 
			
		||||
        Ok(())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn unwrap_m_dot(
 | 
			
		||||
    m_dot: Option<(kw::m, Token![.])>,
 | 
			
		||||
    kind: impl ToTokens,
 | 
			
		||||
) -> syn::Result<(kw::m, Token![.])> {
 | 
			
		||||
    m_dot.ok_or_else(|| {
 | 
			
		||||
        Error::new_spanned(
 | 
			
		||||
            &kind,
 | 
			
		||||
            format_args!(
 | 
			
		||||
                "{} is a ModuleBuilder method, not a free function: try prefixing it with `m.`",
 | 
			
		||||
                kind.to_token_stream()
 | 
			
		||||
            ),
 | 
			
		||||
        )
 | 
			
		||||
    })
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl HdlLetKindParse for HdlLetKind<Type> {
 | 
			
		||||
    type ParsedTy = Option<(Token![:], Box<Type>)>;
 | 
			
		||||
 | 
			
		||||
    fn parse_ty(input: ParseStream) -> syn::Result<Self::ParsedTy> {
 | 
			
		||||
| 
						 | 
				
			
			@ -753,16 +817,20 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
        after_ty: Token![=],
 | 
			
		||||
        input: ParseStream,
 | 
			
		||||
    ) -> syn::Result<Self> {
 | 
			
		||||
        let m = input.parse()?;
 | 
			
		||||
        let dot_token = input.parse()?;
 | 
			
		||||
        let m_dot = if input.peek(kw::m) && input.peek2(Token![.]) {
 | 
			
		||||
            let m = input.parse()?;
 | 
			
		||||
            let dot_token = input.parse()?;
 | 
			
		||||
            Some((m, dot_token))
 | 
			
		||||
        } else {
 | 
			
		||||
            None
 | 
			
		||||
        };
 | 
			
		||||
        let kind: LetFnKind = input.parse()?;
 | 
			
		||||
        match kind {
 | 
			
		||||
            LetFnKind::Input(input_token) => HdlLetKindIO::rest_of_parse(
 | 
			
		||||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                m_dot,
 | 
			
		||||
                ModuleIOKind::Input(input_token),
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::IO),
 | 
			
		||||
| 
						 | 
				
			
			@ -770,8 +838,7 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                m_dot,
 | 
			
		||||
                ModuleIOKind::Output(output),
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::IO),
 | 
			
		||||
| 
						 | 
				
			
			@ -782,30 +849,23 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
                        "type annotation not allowed for instance",
 | 
			
		||||
                    ));
 | 
			
		||||
                }
 | 
			
		||||
                check_empty_m_dot(m_dot, kind)?;
 | 
			
		||||
                let paren_contents;
 | 
			
		||||
                Ok(Self::Instance(HdlLetKindInstance {
 | 
			
		||||
                    m,
 | 
			
		||||
                    dot_token,
 | 
			
		||||
                    instance,
 | 
			
		||||
                    paren: parenthesized!(paren_contents in input),
 | 
			
		||||
                    module: paren_contents.call(parse_single_fn_arg)?,
 | 
			
		||||
                }))
 | 
			
		||||
            }
 | 
			
		||||
            LetFnKind::RegBuilder((reg_builder,)) => HdlLetKindRegBuilder::rest_of_parse(
 | 
			
		||||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                reg_builder,
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::RegBuilder),
 | 
			
		||||
            LetFnKind::RegBuilder((reg_builder,)) => {
 | 
			
		||||
                HdlLetKindRegBuilder::rest_of_parse(input, parsed_ty, after_ty, m_dot, reg_builder)
 | 
			
		||||
                    .map(Self::RegBuilder)
 | 
			
		||||
            }
 | 
			
		||||
            LetFnKind::Wire((wire,)) => {
 | 
			
		||||
                check_empty_m_dot(m_dot, wire)?;
 | 
			
		||||
                let paren_contents;
 | 
			
		||||
                Ok(Self::Wire(HdlLetKindWire {
 | 
			
		||||
                    ty: parsed_ty,
 | 
			
		||||
                    m,
 | 
			
		||||
                    dot_token,
 | 
			
		||||
                    wire,
 | 
			
		||||
                    paren: parenthesized!(paren_contents in input),
 | 
			
		||||
                    ty_expr: paren_contents.call(parse_optional_fn_arg)?,
 | 
			
		||||
| 
						 | 
				
			
			@ -815,8 +875,7 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                m_dot,
 | 
			
		||||
                MemoryFnName::Memory(fn_name),
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::Memory),
 | 
			
		||||
| 
						 | 
				
			
			@ -824,8 +883,7 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                m_dot,
 | 
			
		||||
                MemoryFnName::MemoryArray(fn_name),
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::Memory),
 | 
			
		||||
| 
						 | 
				
			
			@ -833,8 +891,7 @@ impl HdlLetKindParse for HdlLetKind {
 | 
			
		|||
                input,
 | 
			
		||||
                parsed_ty,
 | 
			
		||||
                after_ty,
 | 
			
		||||
                m,
 | 
			
		||||
                dot_token,
 | 
			
		||||
                m_dot,
 | 
			
		||||
                MemoryFnName::MemoryWithInit(fn_name),
 | 
			
		||||
            )
 | 
			
		||||
            .map(Self::Memory),
 | 
			
		||||
| 
						 | 
				
			
			@ -878,6 +935,34 @@ with_debug_clone_and_fold! {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ParseTypes<I>, I> ParseTypes<HdlLet<I>> for HdlLet<T> {
 | 
			
		||||
    fn parse_types(
 | 
			
		||||
        input: &mut HdlLet<I>,
 | 
			
		||||
        parser: &mut TypesParser<'_>,
 | 
			
		||||
    ) -> Result<Self, ParseFailed> {
 | 
			
		||||
        let HdlLet {
 | 
			
		||||
            attrs,
 | 
			
		||||
            hdl_attr,
 | 
			
		||||
            let_token,
 | 
			
		||||
            mut_token,
 | 
			
		||||
            name,
 | 
			
		||||
            eq_token,
 | 
			
		||||
            kind,
 | 
			
		||||
            semi_token,
 | 
			
		||||
        } = input;
 | 
			
		||||
        Ok(Self {
 | 
			
		||||
            attrs: attrs.clone(),
 | 
			
		||||
            hdl_attr: hdl_attr.clone(),
 | 
			
		||||
            let_token: *let_token,
 | 
			
		||||
            mut_token: *mut_token,
 | 
			
		||||
            name: name.clone(),
 | 
			
		||||
            eq_token: *eq_token,
 | 
			
		||||
            kind: T::parse_types(kind, parser)?,
 | 
			
		||||
            semi_token: *semi_token,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<Kind> HdlLet<Kind> {
 | 
			
		||||
    pub(crate) fn try_map<Kind2, E>(
 | 
			
		||||
        self,
 | 
			
		||||
| 
						 | 
				
			
			@ -1006,7 +1091,7 @@ fn wrap_ty_with_expr(ty: impl ToTokens) -> Type {
 | 
			
		|||
fn unwrap_or_static_type(expr: Option<impl ToTokens>, span: Span) -> TokenStream {
 | 
			
		||||
    expr.map(ToTokens::into_token_stream).unwrap_or_else(|| {
 | 
			
		||||
        quote_spanned! {span=>
 | 
			
		||||
            ::fayalite::ty::StaticType::static_type()
 | 
			
		||||
            ::fayalite::ty::StaticType::TYPE
 | 
			
		||||
        }
 | 
			
		||||
    })
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -1026,14 +1111,15 @@ impl<T: ToString> ToTokens for ImplicitName<T> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
struct Visitor {
 | 
			
		||||
struct Visitor<'a> {
 | 
			
		||||
    module_kind: ModuleKind,
 | 
			
		||||
    errors: Errors,
 | 
			
		||||
    io: Vec<ModuleIO>,
 | 
			
		||||
    block_depth: usize,
 | 
			
		||||
    parsed_generics: &'a ParsedGenerics,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Visitor {
 | 
			
		||||
impl Visitor<'_> {
 | 
			
		||||
    fn take_hdl_attr<T: Parse>(&mut self, attrs: &mut Vec<Attribute>) -> Option<HdlAttr<T>> {
 | 
			
		||||
        self.errors.unwrap_or(
 | 
			
		||||
            HdlAttr::parse_and_take_attr(attrs),
 | 
			
		||||
| 
						 | 
				
			
			@ -1086,7 +1172,7 @@ impl Visitor {
 | 
			
		|||
            parse_quote_spanned! {if_token.span=>
 | 
			
		||||
                #(#attrs)*
 | 
			
		||||
                {
 | 
			
		||||
                    let mut __scope = m.if_(#cond);
 | 
			
		||||
                    let mut __scope = ::fayalite::module::if_(#cond);
 | 
			
		||||
                    let _: () = #then_branch;
 | 
			
		||||
                    let mut __scope = __scope.else_();
 | 
			
		||||
                    let _: () = #else_expr;
 | 
			
		||||
| 
						 | 
				
			
			@ -1096,7 +1182,7 @@ impl Visitor {
 | 
			
		|||
            parse_quote_spanned! {if_token.span=>
 | 
			
		||||
                #(#attrs)*
 | 
			
		||||
                {
 | 
			
		||||
                    let mut __scope = m.if_(#cond);
 | 
			
		||||
                    let mut __scope = ::fayalite::module::if_(#cond);
 | 
			
		||||
                    let _: () = #then_branch;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
| 
						 | 
				
			
			@ -1157,8 +1243,6 @@ impl Visitor {
 | 
			
		|||
            eq_token,
 | 
			
		||||
            kind:
 | 
			
		||||
                HdlLetKindInstance {
 | 
			
		||||
                    m,
 | 
			
		||||
                    dot_token,
 | 
			
		||||
                    instance,
 | 
			
		||||
                    paren,
 | 
			
		||||
                    module,
 | 
			
		||||
| 
						 | 
				
			
			@ -1166,7 +1250,7 @@ impl Visitor {
 | 
			
		|||
            semi_token,
 | 
			
		||||
        } = hdl_let;
 | 
			
		||||
        self.require_normal_module(instance);
 | 
			
		||||
        let mut expr = quote! {#m #dot_token #instance};
 | 
			
		||||
        let mut expr = instance.to_token_stream();
 | 
			
		||||
        paren.surround(&mut expr, |expr| {
 | 
			
		||||
            let name_str = ImplicitName {
 | 
			
		||||
                name: &name,
 | 
			
		||||
| 
						 | 
				
			
			@ -1191,11 +1275,9 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    fn process_hdl_let_reg_builder(&mut self, hdl_let: HdlLet<HdlLetKindRegBuilder>) -> Local {
 | 
			
		||||
        let name = &hdl_let.name;
 | 
			
		||||
        let m = hdl_let.kind.m;
 | 
			
		||||
        let dot = hdl_let.kind.dot_token;
 | 
			
		||||
        let reg_builder = hdl_let.kind.reg_builder;
 | 
			
		||||
        self.require_normal_module(reg_builder);
 | 
			
		||||
        let mut expr = quote! {#m #dot #reg_builder};
 | 
			
		||||
        let mut expr = reg_builder.to_token_stream();
 | 
			
		||||
        hdl_let.kind.reg_builder_paren.surround(&mut expr, |expr| {
 | 
			
		||||
            let name_str = ImplicitName {
 | 
			
		||||
                name,
 | 
			
		||||
| 
						 | 
				
			
			@ -1244,12 +1326,10 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    fn process_hdl_let_wire(&mut self, hdl_let: HdlLet<HdlLetKindWire>) -> Local {
 | 
			
		||||
        let name = &hdl_let.name;
 | 
			
		||||
        let m = hdl_let.kind.m;
 | 
			
		||||
        let dot = hdl_let.kind.dot_token;
 | 
			
		||||
        let wire = hdl_let.kind.wire;
 | 
			
		||||
        self.require_normal_module(wire);
 | 
			
		||||
        let ty_expr = unwrap_or_static_type(hdl_let.kind.ty_expr.as_ref(), wire.span());
 | 
			
		||||
        let mut expr = quote! {#m #dot #wire};
 | 
			
		||||
        let mut expr = wire.to_token_stream();
 | 
			
		||||
        hdl_let.kind.paren.surround(&mut expr, |expr| {
 | 
			
		||||
            let name_str = ImplicitName {
 | 
			
		||||
                name,
 | 
			
		||||
| 
						 | 
				
			
			@ -1279,18 +1359,19 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    fn process_hdl_let_memory(&mut self, hdl_let: HdlLet<HdlLetKindMemory>) -> Local {
 | 
			
		||||
        let name = &hdl_let.name;
 | 
			
		||||
        let m = hdl_let.kind.m;
 | 
			
		||||
        let dot = hdl_let.kind.dot_token;
 | 
			
		||||
        let memory_fn = hdl_let.kind.memory_fn;
 | 
			
		||||
        let memory_fn_name = memory_fn.name();
 | 
			
		||||
        self.require_normal_module(memory_fn_name);
 | 
			
		||||
        let mut expr = quote! {#m #dot #memory_fn_name};
 | 
			
		||||
        let mut expr = memory_fn_name.to_token_stream();
 | 
			
		||||
        let (paren, arg) = match memory_fn {
 | 
			
		||||
            MemoryFn::Memory {
 | 
			
		||||
                memory,
 | 
			
		||||
                paren,
 | 
			
		||||
                ty_expr,
 | 
			
		||||
            } => (paren, unwrap_or_static_type(ty_expr.as_ref(), memory.span())),
 | 
			
		||||
            } => (
 | 
			
		||||
                paren,
 | 
			
		||||
                unwrap_or_static_type(ty_expr.as_ref(), memory.span()),
 | 
			
		||||
            ),
 | 
			
		||||
            MemoryFn::MemoryArray {
 | 
			
		||||
                memory_array,
 | 
			
		||||
                paren,
 | 
			
		||||
| 
						 | 
				
			
			@ -1377,16 +1458,17 @@ impl Visitor {
 | 
			
		|||
        let value: BigInt = self
 | 
			
		||||
            .errors
 | 
			
		||||
            .ok(base10_digits.parse().map_err(|e| Error::new(span, e)))?;
 | 
			
		||||
        let (negative, bytes) = match value.sign() {
 | 
			
		||||
            Sign::Minus => (true, value.magnitude().to_bytes_le()),
 | 
			
		||||
            Sign::NoSign => (false, vec![]),
 | 
			
		||||
            Sign::Plus => (false, value.magnitude().to_bytes_le()),
 | 
			
		||||
        let bytes = value.to_signed_bytes_le();
 | 
			
		||||
        let path = if signed {
 | 
			
		||||
            known_items::SInt(span).path
 | 
			
		||||
        } else {
 | 
			
		||||
            known_items::UInt(span).path
 | 
			
		||||
        };
 | 
			
		||||
        Some(parse_quote_spanned! {span=>
 | 
			
		||||
            ::fayalite::int::make_int_literal::<#signed, #width>(
 | 
			
		||||
                #negative,
 | 
			
		||||
            <#path<#width> as ::fayalite::int::BoolOrIntType>::le_bytes_to_expr_wrapping(
 | 
			
		||||
                &[#(#bytes,)*],
 | 
			
		||||
            ).to_int_expr()
 | 
			
		||||
                #width,
 | 
			
		||||
            )
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
    fn process_literal(&mut self, literal: ExprLit) -> Expr {
 | 
			
		||||
| 
						 | 
				
			
			@ -1449,7 +1531,7 @@ fn empty_let() -> Local {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Fold for Visitor {
 | 
			
		||||
impl Fold for Visitor<'_> {
 | 
			
		||||
    fn fold_item(&mut self, item: Item) -> Item {
 | 
			
		||||
        // don't process item interiors
 | 
			
		||||
        item
 | 
			
		||||
| 
						 | 
				
			
			@ -1536,11 +1618,16 @@ impl Fold for Visitor {
 | 
			
		|||
            Some(None) => return fold_local(self, let_stmt),
 | 
			
		||||
            Some(Some(HdlAttr { .. })) => {}
 | 
			
		||||
        };
 | 
			
		||||
        let hdl_let = syn::parse2::<HdlLet>(let_stmt.into_token_stream());
 | 
			
		||||
        let hdl_let = syn::parse2::<HdlLet<HdlLetKind<Type>>>(let_stmt.into_token_stream());
 | 
			
		||||
        let Some(hdl_let) = self.errors.ok(hdl_let) else {
 | 
			
		||||
            return empty_let();
 | 
			
		||||
        };
 | 
			
		||||
        let hdl_let = hdl_let.do_fold(self);
 | 
			
		||||
        let mut hdl_let = hdl_let.do_fold(self);
 | 
			
		||||
        let Ok(hdl_let) =
 | 
			
		||||
            TypesParser::run_with_errors(self.parsed_generics, &mut hdl_let, &mut self.errors)
 | 
			
		||||
        else {
 | 
			
		||||
            return empty_let();
 | 
			
		||||
        };
 | 
			
		||||
        self.process_hdl_let(hdl_let)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1563,12 +1650,14 @@ impl Fold for Visitor {
 | 
			
		|||
pub(crate) fn transform_body(
 | 
			
		||||
    module_kind: ModuleKind,
 | 
			
		||||
    mut body: Box<Block>,
 | 
			
		||||
    parsed_generics: &ParsedGenerics,
 | 
			
		||||
) -> syn::Result<(Box<Block>, Vec<ModuleIO>)> {
 | 
			
		||||
    let mut visitor = Visitor {
 | 
			
		||||
        module_kind,
 | 
			
		||||
        errors: Errors::new(),
 | 
			
		||||
        io: vec![],
 | 
			
		||||
        block_depth: 0,
 | 
			
		||||
        parsed_generics,
 | 
			
		||||
    };
 | 
			
		||||
    *body = syn::fold::fold_block(&mut visitor, *body);
 | 
			
		||||
    visitor.errors.finish()?;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,6 +1,6 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{module::transform_body::Visitor, options, Errors, HdlAttr, PairsIterExt};
 | 
			
		||||
use crate::{module::transform_body::Visitor, Errors, HdlAttr, PairsIterExt};
 | 
			
		||||
use proc_macro2::{Span, TokenStream};
 | 
			
		||||
use quote::{format_ident, quote_spanned, ToTokens, TokenStreamExt};
 | 
			
		||||
use syn::{
 | 
			
		||||
| 
						 | 
				
			
			@ -13,15 +13,6 @@ use syn::{
 | 
			
		|||
    FieldValue, Ident, Index, Member, Path, PathArguments, PathSegment, Token, TypePath,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
options! {
 | 
			
		||||
    #[options = AggregateLiteralOptions]
 | 
			
		||||
    #[no_ident_fragment]
 | 
			
		||||
    pub(crate) enum AggregateLiteralOption {
 | 
			
		||||
        Struct(struct_),
 | 
			
		||||
        Enum(enum_),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct StructOrEnumPath {
 | 
			
		||||
    pub(crate) ty: TypePath,
 | 
			
		||||
| 
						 | 
				
			
			@ -79,11 +70,7 @@ impl StructOrEnumPath {
 | 
			
		|||
            },
 | 
			
		||||
        ]
 | 
			
		||||
    };
 | 
			
		||||
    pub(crate) fn new(
 | 
			
		||||
        errors: &mut Errors,
 | 
			
		||||
        path: TypePath,
 | 
			
		||||
        options: &AggregateLiteralOptions,
 | 
			
		||||
    ) -> Result<Self, ()> {
 | 
			
		||||
    pub(crate) fn new(errors: &mut Errors, path: TypePath) -> Result<Self, ()> {
 | 
			
		||||
        let Path {
 | 
			
		||||
            leading_colon,
 | 
			
		||||
            segments,
 | 
			
		||||
| 
						 | 
				
			
			@ -95,18 +82,6 @@ impl StructOrEnumPath {
 | 
			
		|||
            None
 | 
			
		||||
        };
 | 
			
		||||
        let enum_type = 'guess_enum_type: {
 | 
			
		||||
            if options.enum_.is_some() {
 | 
			
		||||
                if let Some((struct_,)) = options.struct_ {
 | 
			
		||||
                    errors.error(
 | 
			
		||||
                        struct_,
 | 
			
		||||
                        "can't specify both #[hdl(enum)] and #[hdl(struct)]",
 | 
			
		||||
                    );
 | 
			
		||||
                }
 | 
			
		||||
                break 'guess_enum_type Some(None);
 | 
			
		||||
            }
 | 
			
		||||
            if options.struct_.is_some() {
 | 
			
		||||
                break 'guess_enum_type None;
 | 
			
		||||
            }
 | 
			
		||||
            if path.qself.is_none() && leading_colon.is_none() && segments.len() == 1 {
 | 
			
		||||
                let PathSegment { ident, arguments } = &segments[0];
 | 
			
		||||
                for &SingleSegmentVariant {
 | 
			
		||||
| 
						 | 
				
			
			@ -330,7 +305,7 @@ impl ToTokens for StructOrEnumLiteral {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Visitor {
 | 
			
		||||
impl Visitor<'_> {
 | 
			
		||||
    pub(crate) fn process_hdl_array(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        hdl_attr: HdlAttr<Nothing>,
 | 
			
		||||
| 
						 | 
				
			
			@ -358,7 +333,7 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    pub(crate) fn process_struct_enum(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        hdl_attr: HdlAttr<AggregateLiteralOptions>,
 | 
			
		||||
        hdl_attr: HdlAttr<Nothing>,
 | 
			
		||||
        mut literal: StructOrEnumLiteral,
 | 
			
		||||
    ) -> Expr {
 | 
			
		||||
        let span = hdl_attr.hdl.span;
 | 
			
		||||
| 
						 | 
				
			
			@ -384,7 +359,7 @@ impl Visitor {
 | 
			
		|||
            parse_quote! { #field_var }
 | 
			
		||||
        });
 | 
			
		||||
        let Ok(StructOrEnumPath { ty, variant }) =
 | 
			
		||||
            StructOrEnumPath::new(&mut self.errors, literal.path.clone(), &hdl_attr.body)
 | 
			
		||||
            StructOrEnumPath::new(&mut self.errors, literal.path.clone())
 | 
			
		||||
        else {
 | 
			
		||||
            return parse_quote_spanned! {span=>
 | 
			
		||||
                {}
 | 
			
		||||
| 
						 | 
				
			
			@ -445,11 +420,59 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    pub(crate) fn process_hdl_struct(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        hdl_attr: HdlAttr<AggregateLiteralOptions>,
 | 
			
		||||
        hdl_attr: HdlAttr<Nothing>,
 | 
			
		||||
        expr_struct: ExprStruct,
 | 
			
		||||
    ) -> Expr {
 | 
			
		||||
        self.require_normal_module(&hdl_attr);
 | 
			
		||||
        self.process_struct_enum(hdl_attr, expr_struct.into())
 | 
			
		||||
        let name_span = expr_struct.path.segments.last().unwrap().ident.span();
 | 
			
		||||
        let builder_ident = format_ident!("__builder", span = name_span);
 | 
			
		||||
        let empty_builder = if expr_struct.qself.is_some()
 | 
			
		||||
            || expr_struct
 | 
			
		||||
                .path
 | 
			
		||||
                .segments
 | 
			
		||||
                .iter()
 | 
			
		||||
                .any(|seg| !seg.arguments.is_none())
 | 
			
		||||
        {
 | 
			
		||||
            let ty = TypePath {
 | 
			
		||||
                qself: expr_struct.qself,
 | 
			
		||||
                path: expr_struct.path,
 | 
			
		||||
            };
 | 
			
		||||
            let builder_ty = quote_spanned! {name_span=>
 | 
			
		||||
                <#ty as ::fayalite::bundle::BundleType>::Builder
 | 
			
		||||
            };
 | 
			
		||||
            quote_spanned! {name_span=>
 | 
			
		||||
                <#builder_ty as ::fayalite::__std::default::Default>::default()
 | 
			
		||||
            }
 | 
			
		||||
        } else {
 | 
			
		||||
            let path = ExprPath {
 | 
			
		||||
                attrs: vec![],
 | 
			
		||||
                qself: expr_struct.qself,
 | 
			
		||||
                path: expr_struct.path,
 | 
			
		||||
            };
 | 
			
		||||
            quote_spanned! {name_span=>
 | 
			
		||||
                #path::__bundle_builder()
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        let field_calls = Vec::from_iter(expr_struct.fields.iter().map(
 | 
			
		||||
            |FieldValue {
 | 
			
		||||
                 attrs: _,
 | 
			
		||||
                 member,
 | 
			
		||||
                 colon_token: _,
 | 
			
		||||
                 expr,
 | 
			
		||||
             }| {
 | 
			
		||||
                let field_fn = format_ident!("field_{}", member);
 | 
			
		||||
                quote_spanned! {member.span()=>
 | 
			
		||||
                    let #builder_ident = #builder_ident.#field_fn(#expr);
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
        parse_quote_spanned! {name_span=>
 | 
			
		||||
            {
 | 
			
		||||
                let #builder_ident = #empty_builder;
 | 
			
		||||
                #(#field_calls)*
 | 
			
		||||
                ::fayalite::expr::ToExpr::to_expr(&#builder_ident)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn process_hdl_tuple(
 | 
			
		||||
        &mut self,
 | 
			
		||||
| 
						 | 
				
			
			@ -473,7 +496,7 @@ impl Visitor {
 | 
			
		|||
    }
 | 
			
		||||
    pub(crate) fn process_hdl_call(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        hdl_attr: HdlAttr<AggregateLiteralOptions>,
 | 
			
		||||
        hdl_attr: HdlAttr<Nothing>,
 | 
			
		||||
        expr_call: ExprCall,
 | 
			
		||||
    ) -> Expr {
 | 
			
		||||
        self.require_normal_module(&hdl_attr);
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -3,8 +3,7 @@
 | 
			
		|||
use crate::{
 | 
			
		||||
    fold::impl_fold,
 | 
			
		||||
    module::transform_body::{
 | 
			
		||||
        expand_aggregate_literals::{AggregateLiteralOptions, StructOrEnumPath},
 | 
			
		||||
        with_debug_clone_and_fold, Visitor,
 | 
			
		||||
        expand_aggregate_literals::StructOrEnumPath, with_debug_clone_and_fold, Visitor,
 | 
			
		||||
    },
 | 
			
		||||
    Errors, HdlAttr, PairsIterExt,
 | 
			
		||||
};
 | 
			
		||||
| 
						 | 
				
			
			@ -561,8 +560,7 @@ struct HdlMatchParseState<'a> {
 | 
			
		|||
 | 
			
		||||
impl HdlMatchParseState<'_> {
 | 
			
		||||
    fn resolve_enum_struct_path(&mut self, path: TypePath) -> Result<Path, ()> {
 | 
			
		||||
        let StructOrEnumPath { ty, variant } =
 | 
			
		||||
            StructOrEnumPath::new(self.errors, path, &AggregateLiteralOptions::default())?;
 | 
			
		||||
        let StructOrEnumPath { ty, variant } = StructOrEnumPath::new(self.errors, path)?;
 | 
			
		||||
        Ok(if let Some((_variant_path, variant_name)) = variant {
 | 
			
		||||
            parse_quote_spanned! {self.span=>
 | 
			
		||||
                __MatchTy::<#ty>::#variant_name
 | 
			
		||||
| 
						 | 
				
			
			@ -575,7 +573,7 @@ impl HdlMatchParseState<'_> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Visitor {
 | 
			
		||||
impl Visitor<'_> {
 | 
			
		||||
    pub(crate) fn process_hdl_match(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        _hdl_attr: HdlAttr<Nothing>,
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,761 +0,0 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{fold::impl_fold, kw, Errors, HdlAttr};
 | 
			
		||||
use proc_macro2::{Span, TokenStream};
 | 
			
		||||
use quote::{format_ident, quote, quote_spanned, ToTokens};
 | 
			
		||||
use std::collections::{BTreeMap, HashMap, HashSet};
 | 
			
		||||
use syn::{
 | 
			
		||||
    fold::{fold_generics, Fold},
 | 
			
		||||
    parse::{Parse, ParseStream},
 | 
			
		||||
    parse_quote, parse_quote_spanned,
 | 
			
		||||
    punctuated::Punctuated,
 | 
			
		||||
    spanned::Spanned,
 | 
			
		||||
    token::{Brace, Paren, Where},
 | 
			
		||||
    Block, ConstParam, Expr, Field, Fields, FieldsNamed, FieldsUnnamed, GenericParam, Generics,
 | 
			
		||||
    Ident, Index, ItemImpl, Lifetime, LifetimeParam, Member, Path, Token, Type, TypeParam,
 | 
			
		||||
    TypePath, Visibility, WhereClause, WherePredicate,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct Bounds(pub(crate) Punctuated<WherePredicate, Token![,]>);
 | 
			
		||||
 | 
			
		||||
impl_fold! {
 | 
			
		||||
    struct Bounds<>(Punctuated<WherePredicate, Token![,]>);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Parse for Bounds {
 | 
			
		||||
    fn parse(input: ParseStream) -> syn::Result<Self> {
 | 
			
		||||
        Ok(Bounds(Punctuated::parse_terminated(input)?))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<Option<WhereClause>> for Bounds {
 | 
			
		||||
    fn from(value: Option<WhereClause>) -> Self {
 | 
			
		||||
        Self(value.map_or_else(Punctuated::new, |v| v.predicates))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for Bounds {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        self.0.to_tokens(tokens)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone)]
 | 
			
		||||
pub(crate) struct ParsedField<O> {
 | 
			
		||||
    pub(crate) options: HdlAttr<O>,
 | 
			
		||||
    pub(crate) vis: Visibility,
 | 
			
		||||
    pub(crate) name: Member,
 | 
			
		||||
    pub(crate) ty: Type,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<O> ParsedField<O> {
 | 
			
		||||
    pub(crate) fn var_name(&self) -> Ident {
 | 
			
		||||
        format_ident!("__v_{}", self.name)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn get_field_name(
 | 
			
		||||
    index: usize,
 | 
			
		||||
    name: Option<Ident>,
 | 
			
		||||
    ty_span: impl FnOnce() -> Span,
 | 
			
		||||
) -> Member {
 | 
			
		||||
    match name {
 | 
			
		||||
        Some(name) => Member::Named(name),
 | 
			
		||||
        None => Member::Unnamed(Index {
 | 
			
		||||
            index: index as _,
 | 
			
		||||
            span: ty_span(),
 | 
			
		||||
        }),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn get_field_names(fields: &Fields) -> impl Iterator<Item = Member> + '_ {
 | 
			
		||||
    fields
 | 
			
		||||
        .iter()
 | 
			
		||||
        .enumerate()
 | 
			
		||||
        .map(|(index, field)| get_field_name(index, field.ident.clone(), || field.ty.span()))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<O: Parse + Default> ParsedField<O> {
 | 
			
		||||
    pub(crate) fn parse_fields(
 | 
			
		||||
        errors: &mut Errors,
 | 
			
		||||
        fields: &mut Fields,
 | 
			
		||||
        in_enum: bool,
 | 
			
		||||
    ) -> (FieldsKind, Vec<ParsedField<O>>) {
 | 
			
		||||
        let mut unit_fields = Punctuated::new();
 | 
			
		||||
        let (fields_kind, fields) = match fields {
 | 
			
		||||
            Fields::Named(fields) => (FieldsKind::Named(fields.brace_token), &mut fields.named),
 | 
			
		||||
            Fields::Unnamed(fields) => {
 | 
			
		||||
                (FieldsKind::Unnamed(fields.paren_token), &mut fields.unnamed)
 | 
			
		||||
            }
 | 
			
		||||
            Fields::Unit => (FieldsKind::Unit, &mut unit_fields),
 | 
			
		||||
        };
 | 
			
		||||
        let fields = fields
 | 
			
		||||
            .iter_mut()
 | 
			
		||||
            .enumerate()
 | 
			
		||||
            .map(|(index, field)| {
 | 
			
		||||
                let options = errors
 | 
			
		||||
                    .unwrap_or_default(HdlAttr::parse_and_take_attr(&mut field.attrs))
 | 
			
		||||
                    .unwrap_or_default();
 | 
			
		||||
                let name = get_field_name(index, field.ident.clone(), || field.ty.span());
 | 
			
		||||
                if in_enum && !matches!(field.vis, Visibility::Inherited) {
 | 
			
		||||
                    errors.error(&field.vis, "field visibility not allowed in enums");
 | 
			
		||||
                }
 | 
			
		||||
                ParsedField {
 | 
			
		||||
                    options,
 | 
			
		||||
                    vis: field.vis.clone(),
 | 
			
		||||
                    name,
 | 
			
		||||
                    ty: field.ty.clone(),
 | 
			
		||||
                }
 | 
			
		||||
            })
 | 
			
		||||
            .collect();
 | 
			
		||||
        (fields_kind, fields)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Debug)]
 | 
			
		||||
pub(crate) enum FieldsKind {
 | 
			
		||||
    Unit,
 | 
			
		||||
    Named(Brace),
 | 
			
		||||
    Unnamed(Paren),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl FieldsKind {
 | 
			
		||||
    pub(crate) fn into_fields_named(
 | 
			
		||||
        brace_token: Brace,
 | 
			
		||||
        fields: impl IntoIterator<Item = syn::Field>,
 | 
			
		||||
    ) -> Fields {
 | 
			
		||||
        Fields::Named(FieldsNamed {
 | 
			
		||||
            brace_token,
 | 
			
		||||
            named: Punctuated::from_iter(fields),
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn into_fields_unnamed(
 | 
			
		||||
        paren_token: Paren,
 | 
			
		||||
        fields: impl IntoIterator<Item = syn::Field>,
 | 
			
		||||
    ) -> Fields {
 | 
			
		||||
        Fields::Unnamed(FieldsUnnamed {
 | 
			
		||||
            paren_token,
 | 
			
		||||
            unnamed: Punctuated::from_iter(fields),
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn into_fields(self, fields: impl IntoIterator<Item = syn::Field>) -> Fields {
 | 
			
		||||
        match self {
 | 
			
		||||
            FieldsKind::Unit => {
 | 
			
		||||
                let mut fields = fields.into_iter().peekable();
 | 
			
		||||
                let Some(first_field) = fields.peek() else {
 | 
			
		||||
                    return Fields::Unit;
 | 
			
		||||
                };
 | 
			
		||||
                if first_field.ident.is_some() {
 | 
			
		||||
                    Self::into_fields_named(Default::default(), fields)
 | 
			
		||||
                } else {
 | 
			
		||||
                    Self::into_fields_unnamed(Default::default(), fields)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            FieldsKind::Named(brace_token) => Self::into_fields_named(brace_token, fields),
 | 
			
		||||
            FieldsKind::Unnamed(paren_token) => Self::into_fields_unnamed(paren_token, fields),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn get_target(target: &Option<(kw::target, Paren, Path)>, item_ident: &Ident) -> Path {
 | 
			
		||||
    match target {
 | 
			
		||||
        Some((_, _, target)) => target.clone(),
 | 
			
		||||
        None => item_ident.clone().into(),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) struct ValueDeriveGenerics {
 | 
			
		||||
    pub(crate) generics: Generics,
 | 
			
		||||
    pub(crate) static_type_generics: Generics,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ValueDeriveGenerics {
 | 
			
		||||
    pub(crate) fn get(mut generics: Generics, where_: &Option<(Where, Paren, Bounds)>) -> Self {
 | 
			
		||||
        let mut static_type_generics = generics.clone();
 | 
			
		||||
        if let Some((_, _, bounds)) = where_ {
 | 
			
		||||
            generics
 | 
			
		||||
                .make_where_clause()
 | 
			
		||||
                .predicates
 | 
			
		||||
                .extend(bounds.0.iter().cloned());
 | 
			
		||||
            static_type_generics
 | 
			
		||||
                .where_clause
 | 
			
		||||
                .clone_from(&generics.where_clause);
 | 
			
		||||
        } else {
 | 
			
		||||
            let type_params = Vec::from_iter(generics.type_params().map(|v| v.ident.clone()));
 | 
			
		||||
            let predicates = &mut generics.make_where_clause().predicates;
 | 
			
		||||
            let static_type_predicates = &mut static_type_generics.make_where_clause().predicates;
 | 
			
		||||
            for type_param in type_params {
 | 
			
		||||
                predicates.push(parse_quote! {#type_param: ::fayalite::ty::Value<Type: ::fayalite::ty::Type<Value = #type_param>>});
 | 
			
		||||
                static_type_predicates
 | 
			
		||||
                    .push(parse_quote! {#type_param: ::fayalite::ty::StaticValue});
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        Self {
 | 
			
		||||
            generics,
 | 
			
		||||
            static_type_generics,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn derive_clone_hash_eq_partialeq_for_struct<Name: ToTokens>(
 | 
			
		||||
    the_struct_ident: &Ident,
 | 
			
		||||
    generics: &Generics,
 | 
			
		||||
    field_names: &[Name],
 | 
			
		||||
) -> TokenStream {
 | 
			
		||||
    let (impl_generics, type_generics, where_clause) = generics.split_for_impl();
 | 
			
		||||
    quote! {
 | 
			
		||||
        #[automatically_derived]
 | 
			
		||||
        impl #impl_generics ::fayalite::__std::clone::Clone for #the_struct_ident #type_generics
 | 
			
		||||
        #where_clause
 | 
			
		||||
        {
 | 
			
		||||
            fn clone(&self) -> Self {
 | 
			
		||||
                Self {
 | 
			
		||||
                    #(#field_names: ::fayalite::__std::clone::Clone::clone(&self.#field_names),)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        #[automatically_derived]
 | 
			
		||||
        impl #impl_generics ::fayalite::__std::hash::Hash for #the_struct_ident #type_generics
 | 
			
		||||
        #where_clause
 | 
			
		||||
        {
 | 
			
		||||
            #[allow(unused_variables)]
 | 
			
		||||
            fn hash<__H: ::fayalite::__std::hash::Hasher>(&self, hasher: &mut __H) {
 | 
			
		||||
                #(::fayalite::__std::hash::Hash::hash(&self.#field_names, hasher);)*
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        #[automatically_derived]
 | 
			
		||||
        impl #impl_generics ::fayalite::__std::cmp::Eq for #the_struct_ident #type_generics
 | 
			
		||||
        #where_clause
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        #[automatically_derived]
 | 
			
		||||
        impl #impl_generics ::fayalite::__std::cmp::PartialEq for #the_struct_ident #type_generics
 | 
			
		||||
        #where_clause
 | 
			
		||||
        {
 | 
			
		||||
            #[allow(unused_variables)]
 | 
			
		||||
            #[allow(clippy::nonminimal_bool)]
 | 
			
		||||
            fn eq(&self, other: &Self) -> ::fayalite::__std::primitive::bool {
 | 
			
		||||
                true #(&& ::fayalite::__std::cmp::PartialEq::eq(
 | 
			
		||||
                    &self.#field_names,
 | 
			
		||||
                    &other.#field_names,
 | 
			
		||||
                ))*
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn append_field(fields: &mut Fields, mut field: Field) -> Member {
 | 
			
		||||
    let ident = field.ident.clone().expect("ident is supplied");
 | 
			
		||||
    match fields {
 | 
			
		||||
        Fields::Named(FieldsNamed { named, .. }) => {
 | 
			
		||||
            named.push(field);
 | 
			
		||||
            Member::Named(ident)
 | 
			
		||||
        }
 | 
			
		||||
        Fields::Unnamed(FieldsUnnamed { unnamed, .. }) => {
 | 
			
		||||
            field.ident = None;
 | 
			
		||||
            field.colon_token = None;
 | 
			
		||||
            let index = unnamed.len();
 | 
			
		||||
            unnamed.push(field);
 | 
			
		||||
            Member::Unnamed(index.into())
 | 
			
		||||
        }
 | 
			
		||||
        Fields::Unit => {
 | 
			
		||||
            *fields = Fields::Named(FieldsNamed {
 | 
			
		||||
                brace_token: Default::default(),
 | 
			
		||||
                named: Punctuated::from_iter([field]),
 | 
			
		||||
            });
 | 
			
		||||
            Member::Named(ident)
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub(crate) struct BuilderField {
 | 
			
		||||
    pub(crate) names: HashSet<Member>,
 | 
			
		||||
    pub(crate) mapped_value: Expr,
 | 
			
		||||
    pub(crate) mapped_type: Type,
 | 
			
		||||
    pub(crate) where_clause: Option<WhereClause>,
 | 
			
		||||
    pub(crate) builder_field_name: Ident,
 | 
			
		||||
    pub(crate) type_param: Ident,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug)]
 | 
			
		||||
pub(crate) struct Builder {
 | 
			
		||||
    struct_name: Ident,
 | 
			
		||||
    vis: Visibility,
 | 
			
		||||
    fields: BTreeMap<String, BuilderField>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug)]
 | 
			
		||||
pub(crate) struct BuilderWithFields {
 | 
			
		||||
    struct_name: Ident,
 | 
			
		||||
    vis: Visibility,
 | 
			
		||||
    phantom_type_param: Ident,
 | 
			
		||||
    phantom_type_field: Ident,
 | 
			
		||||
    fields: Vec<(String, BuilderField)>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Builder {
 | 
			
		||||
    pub(crate) fn new(struct_name: Ident, vis: Visibility) -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis,
 | 
			
		||||
            fields: BTreeMap::new(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn insert_field(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        name: Member,
 | 
			
		||||
        map_value: impl FnOnce(&Ident) -> Expr,
 | 
			
		||||
        map_type: impl FnOnce(&Ident) -> Type,
 | 
			
		||||
        where_clause: impl FnOnce(&Ident) -> Option<WhereClause>,
 | 
			
		||||
    ) {
 | 
			
		||||
        self.fields
 | 
			
		||||
            .entry(name.to_token_stream().to_string())
 | 
			
		||||
            .or_insert_with_key(|name| {
 | 
			
		||||
                let builder_field_name =
 | 
			
		||||
                    format_ident!("field_{}", name, span = self.struct_name.span());
 | 
			
		||||
                let type_param = format_ident!("__T_{}", name, span = self.struct_name.span());
 | 
			
		||||
                BuilderField {
 | 
			
		||||
                    names: HashSet::new(),
 | 
			
		||||
                    mapped_value: map_value(&builder_field_name),
 | 
			
		||||
                    mapped_type: map_type(&type_param),
 | 
			
		||||
                    where_clause: where_clause(&type_param),
 | 
			
		||||
                    builder_field_name,
 | 
			
		||||
                    type_param,
 | 
			
		||||
                }
 | 
			
		||||
            })
 | 
			
		||||
            .names
 | 
			
		||||
            .insert(name);
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn finish_filling_in_fields(self) -> BuilderWithFields {
 | 
			
		||||
        let Self {
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let fields = Vec::from_iter(fields);
 | 
			
		||||
        BuilderWithFields {
 | 
			
		||||
            phantom_type_param: Ident::new("__Phantom", struct_name.span()),
 | 
			
		||||
            phantom_type_field: Ident::new("__phantom", struct_name.span()),
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis,
 | 
			
		||||
            fields,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl BuilderWithFields {
 | 
			
		||||
    pub(crate) fn get_field(&self, name: &Member) -> Option<(usize, &BuilderField)> {
 | 
			
		||||
        let index = self
 | 
			
		||||
            .fields
 | 
			
		||||
            .binary_search_by_key(&&*name.to_token_stream().to_string(), |v| &*v.0)
 | 
			
		||||
            .ok()?;
 | 
			
		||||
        Some((index, &self.fields[index].1))
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn ty(
 | 
			
		||||
        &self,
 | 
			
		||||
        specified_fields: impl IntoIterator<Item = (Member, Type)>,
 | 
			
		||||
        phantom_type: Option<&Type>,
 | 
			
		||||
        other_fields_are_any_type: bool,
 | 
			
		||||
    ) -> TypePath {
 | 
			
		||||
        let Self {
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis: _,
 | 
			
		||||
            phantom_type_param,
 | 
			
		||||
            phantom_type_field: _,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let span = struct_name.span();
 | 
			
		||||
        let mut arguments =
 | 
			
		||||
            Vec::from_iter(fields.iter().map(|(_, BuilderField { type_param, .. })| {
 | 
			
		||||
                if other_fields_are_any_type {
 | 
			
		||||
                    parse_quote_spanned! {span=>
 | 
			
		||||
                        #type_param
 | 
			
		||||
                    }
 | 
			
		||||
                } else {
 | 
			
		||||
                    parse_quote_spanned! {span=>
 | 
			
		||||
                        ()
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
        for (name, ty) in specified_fields {
 | 
			
		||||
            let Some((index, _)) = self.get_field(&name) else {
 | 
			
		||||
                panic!("field not found: {}", name.to_token_stream());
 | 
			
		||||
            };
 | 
			
		||||
            arguments[index] = ty;
 | 
			
		||||
        }
 | 
			
		||||
        let phantom_type_param = phantom_type.is_none().then_some(phantom_type_param);
 | 
			
		||||
        parse_quote_spanned! {span=>
 | 
			
		||||
            #struct_name::<#phantom_type_param #phantom_type #(, #arguments)*>
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn append_generics(
 | 
			
		||||
        &self,
 | 
			
		||||
        specified_fields: impl IntoIterator<Item = Member>,
 | 
			
		||||
        has_phantom_type_param: bool,
 | 
			
		||||
        other_fields_are_any_type: bool,
 | 
			
		||||
        generics: &mut Generics,
 | 
			
		||||
    ) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            struct_name: _,
 | 
			
		||||
            vis: _,
 | 
			
		||||
            phantom_type_param,
 | 
			
		||||
            phantom_type_field: _,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        if has_phantom_type_param {
 | 
			
		||||
            generics.params.push(GenericParam::from(TypeParam::from(
 | 
			
		||||
                phantom_type_param.clone(),
 | 
			
		||||
            )));
 | 
			
		||||
        }
 | 
			
		||||
        if !other_fields_are_any_type {
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
        let mut type_params = Vec::from_iter(
 | 
			
		||||
            fields
 | 
			
		||||
                .iter()
 | 
			
		||||
                .map(|(_, BuilderField { type_param, .. })| Some(type_param)),
 | 
			
		||||
        );
 | 
			
		||||
        for name in specified_fields {
 | 
			
		||||
            let Some((index, _)) = self.get_field(&name) else {
 | 
			
		||||
                panic!("field not found: {}", name.to_token_stream());
 | 
			
		||||
            };
 | 
			
		||||
            type_params[index] = None;
 | 
			
		||||
        }
 | 
			
		||||
        generics.params.extend(
 | 
			
		||||
            type_params
 | 
			
		||||
                .into_iter()
 | 
			
		||||
                .filter_map(|v| Some(GenericParam::from(TypeParam::from(v?.clone())))),
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn make_build_method(
 | 
			
		||||
        &self,
 | 
			
		||||
        build_fn_name: &Ident,
 | 
			
		||||
        specified_fields: impl IntoIterator<Item = (Member, Type)>,
 | 
			
		||||
        generics: &Generics,
 | 
			
		||||
        phantom_type: &Type,
 | 
			
		||||
        return_ty: &Type,
 | 
			
		||||
        mut body: Block,
 | 
			
		||||
    ) -> ItemImpl {
 | 
			
		||||
        let Self {
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis,
 | 
			
		||||
            phantom_type_param: _,
 | 
			
		||||
            phantom_type_field,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let span = struct_name.span();
 | 
			
		||||
        let field_names = Vec::from_iter(fields.iter().map(|v| &v.1.builder_field_name));
 | 
			
		||||
        let (impl_generics, _type_generics, where_clause) = generics.split_for_impl();
 | 
			
		||||
        let empty_arg = parse_quote_spanned! {span=>
 | 
			
		||||
            ()
 | 
			
		||||
        };
 | 
			
		||||
        let mut ty_arguments = vec![empty_arg; fields.len()];
 | 
			
		||||
        let empty_field_pat = quote_spanned! {span=>
 | 
			
		||||
            : _
 | 
			
		||||
        };
 | 
			
		||||
        let mut field_pats = vec![Some(empty_field_pat); fields.len()];
 | 
			
		||||
        for (name, ty) in specified_fields {
 | 
			
		||||
            let Some((index, _)) = self.get_field(&name) else {
 | 
			
		||||
                panic!("field not found: {}", name.to_token_stream());
 | 
			
		||||
            };
 | 
			
		||||
            ty_arguments[index] = ty;
 | 
			
		||||
            field_pats[index] = None;
 | 
			
		||||
        }
 | 
			
		||||
        body.stmts.insert(
 | 
			
		||||
            0,
 | 
			
		||||
            parse_quote_spanned! {span=>
 | 
			
		||||
                let Self {
 | 
			
		||||
                    #(#field_names #field_pats,)*
 | 
			
		||||
                    #phantom_type_field: _,
 | 
			
		||||
                } = self;
 | 
			
		||||
            },
 | 
			
		||||
        );
 | 
			
		||||
        parse_quote_spanned! {span=>
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics #struct_name<#phantom_type #(, #ty_arguments)*>
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                #[allow(non_snake_case, dead_code)]
 | 
			
		||||
                #vis fn #build_fn_name(self) -> #return_ty
 | 
			
		||||
                #body
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for BuilderWithFields {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            struct_name,
 | 
			
		||||
            vis,
 | 
			
		||||
            phantom_type_param,
 | 
			
		||||
            phantom_type_field,
 | 
			
		||||
            fields,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let span = struct_name.span();
 | 
			
		||||
        let mut any_generics = Generics::default();
 | 
			
		||||
        self.append_generics([], true, true, &mut any_generics);
 | 
			
		||||
        let empty_ty = self.ty([], None, false);
 | 
			
		||||
        let field_names = Vec::from_iter(fields.iter().map(|v| &v.1.builder_field_name));
 | 
			
		||||
        let field_type_params = Vec::from_iter(fields.iter().map(|v| &v.1.type_param));
 | 
			
		||||
        quote_spanned! {span=>
 | 
			
		||||
            #[allow(non_camel_case_types)]
 | 
			
		||||
            #[non_exhaustive]
 | 
			
		||||
            #vis struct #struct_name #any_generics {
 | 
			
		||||
                #(#field_names: #field_type_params,)*
 | 
			
		||||
                #phantom_type_field: ::fayalite::__std::marker::PhantomData<#phantom_type_param>,
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl<#phantom_type_param> #empty_ty {
 | 
			
		||||
                fn new() -> Self {
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#field_names: (),)*
 | 
			
		||||
                        #phantom_type_field: ::fayalite::__std::marker::PhantomData,
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        for (field_index, (_, field)) in self.fields.iter().enumerate() {
 | 
			
		||||
            let initial_fields = &fields[..field_index];
 | 
			
		||||
            let final_fields = &fields[field_index..][1..];
 | 
			
		||||
            let initial_type_params =
 | 
			
		||||
                Vec::from_iter(initial_fields.iter().map(|v| &v.1.type_param));
 | 
			
		||||
            let final_type_params = Vec::from_iter(final_fields.iter().map(|v| &v.1.type_param));
 | 
			
		||||
            let initial_field_names =
 | 
			
		||||
                Vec::from_iter(initial_fields.iter().map(|v| &v.1.builder_field_name));
 | 
			
		||||
            let final_field_names =
 | 
			
		||||
                Vec::from_iter(final_fields.iter().map(|v| &v.1.builder_field_name));
 | 
			
		||||
            let BuilderField {
 | 
			
		||||
                names: _,
 | 
			
		||||
                mapped_value,
 | 
			
		||||
                mapped_type,
 | 
			
		||||
                where_clause,
 | 
			
		||||
                builder_field_name,
 | 
			
		||||
                type_param,
 | 
			
		||||
            } = field;
 | 
			
		||||
            quote_spanned! {span=>
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                #[allow(non_camel_case_types, dead_code)]
 | 
			
		||||
                impl<#phantom_type_param #(, #initial_type_params)* #(, #final_type_params)*>
 | 
			
		||||
                    #struct_name<
 | 
			
		||||
                        #phantom_type_param,
 | 
			
		||||
                        #(#initial_type_params,)*
 | 
			
		||||
                        (), #(#final_type_params,)*
 | 
			
		||||
                    >
 | 
			
		||||
                {
 | 
			
		||||
                    #vis fn #builder_field_name<#type_param>(
 | 
			
		||||
                        self,
 | 
			
		||||
                        #builder_field_name: #type_param,
 | 
			
		||||
                    ) -> #struct_name<
 | 
			
		||||
                            #phantom_type_param,
 | 
			
		||||
                            #(#initial_type_params,)*
 | 
			
		||||
                            #mapped_type,
 | 
			
		||||
                            #(#final_type_params,)*
 | 
			
		||||
                        >
 | 
			
		||||
                    #where_clause
 | 
			
		||||
                    {
 | 
			
		||||
                        let Self {
 | 
			
		||||
                            #(#initial_field_names,)*
 | 
			
		||||
                            #builder_field_name: (),
 | 
			
		||||
                            #(#final_field_names,)*
 | 
			
		||||
                            #phantom_type_field: _,
 | 
			
		||||
                        } = self;
 | 
			
		||||
                        let #builder_field_name = #mapped_value;
 | 
			
		||||
                        #struct_name {
 | 
			
		||||
                            #(#field_names,)*
 | 
			
		||||
                            #phantom_type_field: ::fayalite::__std::marker::PhantomData,
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) struct MapIdents {
 | 
			
		||||
    pub(crate) map: HashMap<Ident, Ident>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Fold for &MapIdents {
 | 
			
		||||
    fn fold_ident(&mut self, i: Ident) -> Ident {
 | 
			
		||||
        self.map.get(&i).cloned().unwrap_or(i)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) struct DupGenerics<M> {
 | 
			
		||||
    pub(crate) combined: Generics,
 | 
			
		||||
    pub(crate) maps: M,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn merge_punctuated<T, P: Default>(
 | 
			
		||||
    target: &mut Punctuated<T, P>,
 | 
			
		||||
    source: Punctuated<T, P>,
 | 
			
		||||
    make_punct: impl FnOnce() -> P,
 | 
			
		||||
) {
 | 
			
		||||
    if source.is_empty() {
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
    if target.is_empty() {
 | 
			
		||||
        *target = source;
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
    if !target.trailing_punct() {
 | 
			
		||||
        target.push_punct(make_punct());
 | 
			
		||||
    }
 | 
			
		||||
    target.extend(source.into_pairs());
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn merge_generics(target: &mut Generics, source: Generics) {
 | 
			
		||||
    let Generics {
 | 
			
		||||
        lt_token,
 | 
			
		||||
        params,
 | 
			
		||||
        gt_token,
 | 
			
		||||
        where_clause,
 | 
			
		||||
    } = source;
 | 
			
		||||
    let span = lt_token.map(|v| v.span).unwrap_or_else(|| params.span());
 | 
			
		||||
    target.lt_token = target.lt_token.or(lt_token);
 | 
			
		||||
    merge_punctuated(&mut target.params, params, || Token);
 | 
			
		||||
    target.gt_token = target.gt_token.or(gt_token);
 | 
			
		||||
    if let Some(where_clause) = where_clause {
 | 
			
		||||
        if let Some(target_where_clause) = &mut target.where_clause {
 | 
			
		||||
            let WhereClause {
 | 
			
		||||
                where_token,
 | 
			
		||||
                predicates,
 | 
			
		||||
            } = where_clause;
 | 
			
		||||
            let span = where_token.span;
 | 
			
		||||
            target_where_clause.where_token = where_token;
 | 
			
		||||
            merge_punctuated(&mut target_where_clause.predicates, predicates, || {
 | 
			
		||||
                Token
 | 
			
		||||
            });
 | 
			
		||||
        } else {
 | 
			
		||||
            target.where_clause = Some(where_clause);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl DupGenerics<Vec<MapIdents>> {
 | 
			
		||||
    pub(crate) fn new_dyn(generics: &Generics, count: usize) -> Self {
 | 
			
		||||
        let mut maps = Vec::from_iter((0..count).map(|_| MapIdents {
 | 
			
		||||
            map: HashMap::new(),
 | 
			
		||||
        }));
 | 
			
		||||
        for param in &generics.params {
 | 
			
		||||
            let (GenericParam::Lifetime(LifetimeParam {
 | 
			
		||||
                lifetime: Lifetime { ident, .. },
 | 
			
		||||
                ..
 | 
			
		||||
            })
 | 
			
		||||
            | GenericParam::Type(TypeParam { ident, .. })
 | 
			
		||||
            | GenericParam::Const(ConstParam { ident, .. })) = param;
 | 
			
		||||
            for (i, map_idents) in maps.iter_mut().enumerate() {
 | 
			
		||||
                map_idents
 | 
			
		||||
                    .map
 | 
			
		||||
                    .insert(ident.clone(), format_ident!("__{}_{}", ident, i));
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        let mut combined = Generics::default();
 | 
			
		||||
        for map_idents in maps.iter() {
 | 
			
		||||
            merge_generics(
 | 
			
		||||
                &mut combined,
 | 
			
		||||
                fold_generics(&mut { map_idents }, generics.clone()),
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
        Self { combined, maps }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<const COUNT: usize> DupGenerics<[MapIdents; COUNT]> {
 | 
			
		||||
    pub(crate) fn new(generics: &Generics) -> Self {
 | 
			
		||||
        let DupGenerics { combined, maps } = DupGenerics::new_dyn(generics, COUNT);
 | 
			
		||||
        Self {
 | 
			
		||||
            combined,
 | 
			
		||||
            maps: maps.try_into().ok().unwrap(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn add_where_predicate(
 | 
			
		||||
    target: &mut Generics,
 | 
			
		||||
    span: Span,
 | 
			
		||||
    where_predicate: WherePredicate,
 | 
			
		||||
) {
 | 
			
		||||
    let WhereClause {
 | 
			
		||||
        where_token: _,
 | 
			
		||||
        predicates,
 | 
			
		||||
    } = target.where_clause.get_or_insert_with(|| WhereClause {
 | 
			
		||||
        where_token: Token,
 | 
			
		||||
        predicates: Punctuated::new(),
 | 
			
		||||
    });
 | 
			
		||||
    if !predicates.empty_or_trailing() {
 | 
			
		||||
        predicates.push_punct(Token);
 | 
			
		||||
    }
 | 
			
		||||
    predicates.push_value(where_predicate);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn make_connect_impl(
 | 
			
		||||
    connect_inexact: Option<(crate::kw::connect_inexact,)>,
 | 
			
		||||
    generics: &Generics,
 | 
			
		||||
    ty_ident: &Ident,
 | 
			
		||||
    field_types: impl IntoIterator<Item = Type>,
 | 
			
		||||
) -> TokenStream {
 | 
			
		||||
    let span = ty_ident.span();
 | 
			
		||||
    let impl_generics;
 | 
			
		||||
    let combined_generics;
 | 
			
		||||
    let where_clause;
 | 
			
		||||
    let lhs_generics;
 | 
			
		||||
    let lhs_type_generics;
 | 
			
		||||
    let rhs_generics;
 | 
			
		||||
    let rhs_type_generics;
 | 
			
		||||
    if connect_inexact.is_some() {
 | 
			
		||||
        let DupGenerics {
 | 
			
		||||
            mut combined,
 | 
			
		||||
            maps: [lhs_map, rhs_map],
 | 
			
		||||
        } = DupGenerics::new(generics);
 | 
			
		||||
        for field_type in field_types {
 | 
			
		||||
            let lhs_type = (&lhs_map).fold_type(field_type.clone());
 | 
			
		||||
            let rhs_type = (&rhs_map).fold_type(field_type);
 | 
			
		||||
            add_where_predicate(
 | 
			
		||||
                &mut combined,
 | 
			
		||||
                span,
 | 
			
		||||
                parse_quote_spanned! {span=>
 | 
			
		||||
                    #lhs_type: ::fayalite::ty::Connect<#rhs_type>
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
        combined_generics = combined;
 | 
			
		||||
        (impl_generics, _, where_clause) = combined_generics.split_for_impl();
 | 
			
		||||
        lhs_generics = (&lhs_map).fold_generics(generics.clone());
 | 
			
		||||
        (_, lhs_type_generics, _) = lhs_generics.split_for_impl();
 | 
			
		||||
        rhs_generics = (&rhs_map).fold_generics(generics.clone());
 | 
			
		||||
        (_, rhs_type_generics, _) = rhs_generics.split_for_impl();
 | 
			
		||||
    } else {
 | 
			
		||||
        let mut generics = generics.clone();
 | 
			
		||||
        for field_type in field_types {
 | 
			
		||||
            add_where_predicate(
 | 
			
		||||
                &mut generics,
 | 
			
		||||
                span,
 | 
			
		||||
                parse_quote_spanned! {span=>
 | 
			
		||||
                    #field_type: ::fayalite::ty::Connect<#field_type>
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
        combined_generics = generics;
 | 
			
		||||
        (impl_generics, lhs_type_generics, where_clause) = combined_generics.split_for_impl();
 | 
			
		||||
        rhs_type_generics = lhs_type_generics.clone();
 | 
			
		||||
    }
 | 
			
		||||
    quote_spanned! {span=>
 | 
			
		||||
        #[automatically_derived]
 | 
			
		||||
        #[allow(non_camel_case_types)]
 | 
			
		||||
        impl #impl_generics ::fayalite::ty::Connect<#ty_ident #rhs_type_generics>
 | 
			
		||||
            for #ty_ident #lhs_type_generics
 | 
			
		||||
        #where_clause
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -1,969 +0,0 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    value_derive_common::{
 | 
			
		||||
        append_field, derive_clone_hash_eq_partialeq_for_struct, get_field_names, get_target,
 | 
			
		||||
        make_connect_impl, Bounds, Builder, FieldsKind, ParsedField, ValueDeriveGenerics,
 | 
			
		||||
    },
 | 
			
		||||
    value_derive_struct::{self, ParsedStruct, ParsedStructNames, StructOptions},
 | 
			
		||||
    Errors, HdlAttr,
 | 
			
		||||
};
 | 
			
		||||
use proc_macro2::TokenStream;
 | 
			
		||||
use quote::{format_ident, quote, quote_spanned, ToTokens};
 | 
			
		||||
use syn::{
 | 
			
		||||
    parse_quote, parse_quote_spanned, punctuated::Punctuated, spanned::Spanned, token::Brace,
 | 
			
		||||
    Field, FieldMutability, Fields, FieldsNamed, Generics, Ident, Index, ItemEnum, ItemStruct,
 | 
			
		||||
    Member, Path, Token, Type, Variant, Visibility,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = EnumOptions]
 | 
			
		||||
    enum EnumOption {
 | 
			
		||||
        OutlineGenerated(outline_generated),
 | 
			
		||||
        ConnectInexact(connect_inexact),
 | 
			
		||||
        Bounds(where_, Bounds),
 | 
			
		||||
        Target(target, Path),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = VariantOptions]
 | 
			
		||||
    enum VariantOption {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = FieldOptions]
 | 
			
		||||
    enum FieldOption {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
enum VariantValue {
 | 
			
		||||
    None,
 | 
			
		||||
    Direct {
 | 
			
		||||
        value_type: Type,
 | 
			
		||||
    },
 | 
			
		||||
    Struct {
 | 
			
		||||
        value_struct: ItemStruct,
 | 
			
		||||
        parsed_struct: ParsedStruct,
 | 
			
		||||
    },
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl VariantValue {
 | 
			
		||||
    fn is_none(&self) -> bool {
 | 
			
		||||
        matches!(self, Self::None)
 | 
			
		||||
    }
 | 
			
		||||
    fn value_ty(&self) -> Option<Type> {
 | 
			
		||||
        match self {
 | 
			
		||||
            VariantValue::None => None,
 | 
			
		||||
            VariantValue::Direct { value_type } => Some(value_type.clone()),
 | 
			
		||||
            VariantValue::Struct { value_struct, .. } => {
 | 
			
		||||
                let (_, type_generics, _) = value_struct.generics.split_for_impl();
 | 
			
		||||
                let ident = &value_struct.ident;
 | 
			
		||||
                Some(parse_quote! { #ident #type_generics })
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
struct ParsedVariant {
 | 
			
		||||
    options: HdlAttr<VariantOptions>,
 | 
			
		||||
    ident: Ident,
 | 
			
		||||
    fields_kind: FieldsKind,
 | 
			
		||||
    fields: Vec<ParsedField<FieldOptions>>,
 | 
			
		||||
    value: VariantValue,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedVariant {
 | 
			
		||||
    fn parse(
 | 
			
		||||
        errors: &mut Errors,
 | 
			
		||||
        variant: Variant,
 | 
			
		||||
        enum_options: &EnumOptions,
 | 
			
		||||
        enum_vis: &Visibility,
 | 
			
		||||
        enum_ident: &Ident,
 | 
			
		||||
        enum_generics: &Generics,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        let target = get_target(&enum_options.target, enum_ident);
 | 
			
		||||
        let Variant {
 | 
			
		||||
            mut attrs,
 | 
			
		||||
            ident,
 | 
			
		||||
            fields,
 | 
			
		||||
            discriminant,
 | 
			
		||||
        } = variant;
 | 
			
		||||
        if let Some((eq, _)) = discriminant {
 | 
			
		||||
            errors.error(eq, "#[derive(Value)]: discriminants not allowed");
 | 
			
		||||
        }
 | 
			
		||||
        let variant_options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::parse_and_take_attr(&mut attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        let (fields_kind, parsed_fields) =
 | 
			
		||||
            ParsedField::parse_fields(errors, &mut fields.clone(), true);
 | 
			
		||||
        let value = match (&fields_kind, &*parsed_fields) {
 | 
			
		||||
            (FieldsKind::Unit, _) => VariantValue::None,
 | 
			
		||||
            (
 | 
			
		||||
                FieldsKind::Unnamed(_),
 | 
			
		||||
                [ParsedField {
 | 
			
		||||
                    options,
 | 
			
		||||
                    vis: _,
 | 
			
		||||
                    name: Member::Unnamed(Index { index: 0, span: _ }),
 | 
			
		||||
                    ty,
 | 
			
		||||
                }],
 | 
			
		||||
            ) => {
 | 
			
		||||
                let FieldOptions {} = options.body;
 | 
			
		||||
                VariantValue::Direct {
 | 
			
		||||
                    value_type: ty.clone(),
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            _ => {
 | 
			
		||||
                let variant_value_struct_ident =
 | 
			
		||||
                    format_ident!("__{}__{}", enum_ident, ident, span = ident.span());
 | 
			
		||||
                let variant_type_struct_ident =
 | 
			
		||||
                    format_ident!("__{}__{}__Type", enum_ident, ident, span = ident.span());
 | 
			
		||||
                let mut value_struct_fields = fields.clone();
 | 
			
		||||
                let (_, type_generics, _) = enum_generics.split_for_impl();
 | 
			
		||||
                append_field(
 | 
			
		||||
                    &mut value_struct_fields,
 | 
			
		||||
                    Field {
 | 
			
		||||
                        attrs: vec![HdlAttr::from(value_derive_struct::FieldOptions {
 | 
			
		||||
                            flip: None,
 | 
			
		||||
                            skip: Some(Default::default()),
 | 
			
		||||
                        })
 | 
			
		||||
                        .to_attr()],
 | 
			
		||||
                        vis: enum_vis.clone(),
 | 
			
		||||
                        mutability: FieldMutability::None,
 | 
			
		||||
                        ident: Some(Ident::new("__phantom", ident.span())),
 | 
			
		||||
                        colon_token: None,
 | 
			
		||||
                        ty: parse_quote_spanned! {ident.span()=>
 | 
			
		||||
                            ::fayalite::__std::marker::PhantomData<#target #type_generics>
 | 
			
		||||
                        },
 | 
			
		||||
                    },
 | 
			
		||||
                );
 | 
			
		||||
                let (value_struct_fields_kind, value_struct_parsed_fields) =
 | 
			
		||||
                    ParsedField::parse_fields(errors, &mut value_struct_fields, false);
 | 
			
		||||
                let value_struct = ItemStruct {
 | 
			
		||||
                    attrs: vec![parse_quote! { #[allow(non_camel_case_types)] }],
 | 
			
		||||
                    vis: enum_vis.clone(),
 | 
			
		||||
                    struct_token: Token),
 | 
			
		||||
                    ident: variant_value_struct_ident.clone(),
 | 
			
		||||
                    generics: enum_generics.clone(),
 | 
			
		||||
                    fields: value_struct_fields,
 | 
			
		||||
                    semi_token: None,
 | 
			
		||||
                };
 | 
			
		||||
                VariantValue::Struct {
 | 
			
		||||
                    value_struct,
 | 
			
		||||
                    parsed_struct: ParsedStruct {
 | 
			
		||||
                        options: StructOptions {
 | 
			
		||||
                            outline_generated: None,
 | 
			
		||||
                            static_: Some(Default::default()),
 | 
			
		||||
                            where_: Some((
 | 
			
		||||
                                Default::default(),
 | 
			
		||||
                                Default::default(),
 | 
			
		||||
                                ValueDeriveGenerics::get(
 | 
			
		||||
                                    enum_generics.clone(),
 | 
			
		||||
                                    &enum_options.where_,
 | 
			
		||||
                                )
 | 
			
		||||
                                .static_type_generics
 | 
			
		||||
                                .where_clause
 | 
			
		||||
                                .into(),
 | 
			
		||||
                            )),
 | 
			
		||||
                            target: None,
 | 
			
		||||
                            connect_inexact: enum_options.connect_inexact,
 | 
			
		||||
                        }
 | 
			
		||||
                        .into(),
 | 
			
		||||
                        vis: enum_vis.clone(),
 | 
			
		||||
                        struct_token: Default::default(),
 | 
			
		||||
                        generics: enum_generics.clone(),
 | 
			
		||||
                        fields_kind: value_struct_fields_kind,
 | 
			
		||||
                        fields: value_struct_parsed_fields,
 | 
			
		||||
                        semi_token: None, // it will fill in the semicolon if needed
 | 
			
		||||
                        skip_check_fields: true,
 | 
			
		||||
                        names: ParsedStructNames {
 | 
			
		||||
                            ident: variant_value_struct_ident.clone(),
 | 
			
		||||
                            type_struct_debug_ident: Some(format!("{enum_ident}::{ident}::Type")),
 | 
			
		||||
                            type_struct_ident: variant_type_struct_ident,
 | 
			
		||||
                            match_variant_ident: None,
 | 
			
		||||
                            builder_struct_ident: None,
 | 
			
		||||
                            mask_match_variant_ident: None,
 | 
			
		||||
                            mask_type_ident: None,
 | 
			
		||||
                            mask_type_debug_ident: Some(format!(
 | 
			
		||||
                                "AsMask<{enum_ident}::{ident}>::Type"
 | 
			
		||||
                            )),
 | 
			
		||||
                            mask_value_ident: None,
 | 
			
		||||
                            mask_value_debug_ident: Some(format!("AsMask<{enum_ident}::{ident}>")),
 | 
			
		||||
                            mask_builder_struct_ident: None,
 | 
			
		||||
                        },
 | 
			
		||||
                    },
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        ParsedVariant {
 | 
			
		||||
            options: variant_options,
 | 
			
		||||
            ident,
 | 
			
		||||
            fields_kind,
 | 
			
		||||
            fields: parsed_fields,
 | 
			
		||||
            value,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
struct ParsedEnum {
 | 
			
		||||
    options: HdlAttr<EnumOptions>,
 | 
			
		||||
    vis: Visibility,
 | 
			
		||||
    enum_token: Token![enum],
 | 
			
		||||
    ident: Ident,
 | 
			
		||||
    generics: Generics,
 | 
			
		||||
    brace_token: Brace,
 | 
			
		||||
    variants: Vec<ParsedVariant>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedEnum {
 | 
			
		||||
    fn parse(item: ItemEnum) -> syn::Result<Self> {
 | 
			
		||||
        let ItemEnum {
 | 
			
		||||
            mut attrs,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
        } = item;
 | 
			
		||||
        let mut errors = Errors::new();
 | 
			
		||||
        let enum_options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::parse_and_take_attr(&mut attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        let variants = variants
 | 
			
		||||
            .into_iter()
 | 
			
		||||
            .map(|variant| {
 | 
			
		||||
                ParsedVariant::parse(
 | 
			
		||||
                    &mut errors,
 | 
			
		||||
                    variant,
 | 
			
		||||
                    &enum_options.body,
 | 
			
		||||
                    &vis,
 | 
			
		||||
                    &ident,
 | 
			
		||||
                    &generics,
 | 
			
		||||
                )
 | 
			
		||||
            })
 | 
			
		||||
            .collect();
 | 
			
		||||
        errors.finish()?;
 | 
			
		||||
        Ok(ParsedEnum {
 | 
			
		||||
            options: enum_options,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for ParsedEnum {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            enum_token,
 | 
			
		||||
            ident: enum_ident,
 | 
			
		||||
            generics: enum_generics,
 | 
			
		||||
            brace_token,
 | 
			
		||||
            variants,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let EnumOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            connect_inexact,
 | 
			
		||||
            where_,
 | 
			
		||||
            target,
 | 
			
		||||
        } = &options.body;
 | 
			
		||||
        let target = get_target(target, enum_ident);
 | 
			
		||||
        let ValueDeriveGenerics {
 | 
			
		||||
            generics: _,
 | 
			
		||||
            static_type_generics,
 | 
			
		||||
        } = ValueDeriveGenerics::get(enum_generics.clone(), where_);
 | 
			
		||||
        let (static_type_impl_generics, static_type_type_generics, static_type_where_clause) =
 | 
			
		||||
            static_type_generics.split_for_impl();
 | 
			
		||||
        let type_struct_ident = format_ident!("__{}__Type", enum_ident);
 | 
			
		||||
        let mut field_checks = vec![];
 | 
			
		||||
        let mut make_type_struct_variant_type = |variant: &ParsedVariant| {
 | 
			
		||||
            let VariantOptions {} = variant.options.body;
 | 
			
		||||
            let (value_struct, parsed_struct) = match &variant.value {
 | 
			
		||||
                VariantValue::None => {
 | 
			
		||||
                    return None;
 | 
			
		||||
                }
 | 
			
		||||
                VariantValue::Direct { value_type } => {
 | 
			
		||||
                    field_checks.push(quote_spanned! {value_type.span()=>
 | 
			
		||||
                        __check_field::<#value_type>();
 | 
			
		||||
                    });
 | 
			
		||||
                    return Some(parse_quote! { <#value_type as ::fayalite::expr::ToExpr>::Type });
 | 
			
		||||
                }
 | 
			
		||||
                VariantValue::Struct {
 | 
			
		||||
                    value_struct,
 | 
			
		||||
                    parsed_struct,
 | 
			
		||||
                } => (value_struct, parsed_struct),
 | 
			
		||||
            };
 | 
			
		||||
            value_struct.to_tokens(tokens);
 | 
			
		||||
            parsed_struct.to_tokens(tokens);
 | 
			
		||||
            let mut field_names = Vec::from_iter(get_field_names(&value_struct.fields));
 | 
			
		||||
            derive_clone_hash_eq_partialeq_for_struct(
 | 
			
		||||
                &value_struct.ident,
 | 
			
		||||
                &static_type_generics,
 | 
			
		||||
                &field_names,
 | 
			
		||||
            )
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
            field_names = Vec::from_iter(
 | 
			
		||||
                field_names
 | 
			
		||||
                    .into_iter()
 | 
			
		||||
                    .zip(parsed_struct.fields.iter())
 | 
			
		||||
                    .filter_map(|(member, field)| {
 | 
			
		||||
                        field.options.body.skip.is_none().then_some(member)
 | 
			
		||||
                    }),
 | 
			
		||||
            );
 | 
			
		||||
            let field_name_strs =
 | 
			
		||||
                Vec::from_iter(field_names.iter().map(|v| v.to_token_stream().to_string()));
 | 
			
		||||
            let debug_ident = format!("{enum_ident}::{}", variant.ident);
 | 
			
		||||
            let debug_body = match variant.fields_kind {
 | 
			
		||||
                FieldsKind::Unit => quote! {
 | 
			
		||||
                    f.debug_struct(#debug_ident).finish()
 | 
			
		||||
                },
 | 
			
		||||
                FieldsKind::Named(_) => quote! {
 | 
			
		||||
                    f.debug_struct(#debug_ident)
 | 
			
		||||
                        #(.field(#field_name_strs, &self.#field_names))*
 | 
			
		||||
                        .finish()
 | 
			
		||||
                },
 | 
			
		||||
                FieldsKind::Unnamed(_) => quote! {
 | 
			
		||||
                    f.debug_tuple(#debug_ident)#(.field(&self.#field_names))*.finish()
 | 
			
		||||
                },
 | 
			
		||||
            };
 | 
			
		||||
            let value_struct_ident = &value_struct.ident;
 | 
			
		||||
            quote! {
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #static_type_impl_generics ::fayalite::__std::fmt::Debug
 | 
			
		||||
                    for #value_struct_ident #static_type_type_generics
 | 
			
		||||
                #static_type_where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    fn fmt(
 | 
			
		||||
                        &self,
 | 
			
		||||
                        f: &mut ::fayalite::__std::fmt::Formatter<'_>,
 | 
			
		||||
                    ) -> ::fayalite::__std::fmt::Result {
 | 
			
		||||
                        #debug_body
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
            Some(parse_quote! {
 | 
			
		||||
                <
 | 
			
		||||
                    #value_struct_ident #static_type_type_generics
 | 
			
		||||
                    as ::fayalite::expr::ToExpr
 | 
			
		||||
                >::Type
 | 
			
		||||
            })
 | 
			
		||||
        };
 | 
			
		||||
        let type_struct_variants = Punctuated::from_iter(variants.iter().filter_map(|variant| {
 | 
			
		||||
            let VariantOptions {} = variant.options.body;
 | 
			
		||||
            Some(Field {
 | 
			
		||||
                attrs: vec![],
 | 
			
		||||
                vis: vis.clone(),
 | 
			
		||||
                mutability: FieldMutability::None,
 | 
			
		||||
                ident: Some(variant.ident.clone()),
 | 
			
		||||
                colon_token: None, // it will fill in the colon if needed
 | 
			
		||||
                ty: make_type_struct_variant_type(variant)?,
 | 
			
		||||
            })
 | 
			
		||||
        }));
 | 
			
		||||
        let type_struct = ItemStruct {
 | 
			
		||||
            attrs: vec![
 | 
			
		||||
                parse_quote! {#[allow(non_camel_case_types)]},
 | 
			
		||||
                parse_quote! {#[allow(non_snake_case)]},
 | 
			
		||||
            ],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: Token,
 | 
			
		||||
            ident: type_struct_ident,
 | 
			
		||||
            generics: static_type_generics.clone(),
 | 
			
		||||
            fields: Fields::Named(FieldsNamed {
 | 
			
		||||
                brace_token: *brace_token,
 | 
			
		||||
                named: type_struct_variants,
 | 
			
		||||
            }),
 | 
			
		||||
            semi_token: None,
 | 
			
		||||
        };
 | 
			
		||||
        let type_struct_ident = &type_struct.ident;
 | 
			
		||||
        let type_struct_debug_ident = format!("{enum_ident}::Type");
 | 
			
		||||
        type_struct.to_tokens(tokens);
 | 
			
		||||
        let non_empty_variant_names = Vec::from_iter(
 | 
			
		||||
            variants
 | 
			
		||||
                .iter()
 | 
			
		||||
                .filter(|v| !v.value.is_none())
 | 
			
		||||
                .map(|v| v.ident.clone()),
 | 
			
		||||
        );
 | 
			
		||||
        let non_empty_variant_name_strs =
 | 
			
		||||
            Vec::from_iter(non_empty_variant_names.iter().map(|v| v.to_string()));
 | 
			
		||||
        let debug_type_body = quote! {
 | 
			
		||||
            f.debug_struct(#type_struct_debug_ident)
 | 
			
		||||
                #(.field(#non_empty_variant_name_strs, &self.#non_empty_variant_names))*
 | 
			
		||||
                .finish()
 | 
			
		||||
        };
 | 
			
		||||
        derive_clone_hash_eq_partialeq_for_struct(
 | 
			
		||||
            type_struct_ident,
 | 
			
		||||
            &static_type_generics,
 | 
			
		||||
            &non_empty_variant_names,
 | 
			
		||||
        )
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let variant_names = Vec::from_iter(variants.iter().map(|v| &v.ident));
 | 
			
		||||
        let variant_name_strs = Vec::from_iter(variant_names.iter().map(|v| v.to_string()));
 | 
			
		||||
        let (variant_field_pats, variant_to_canonical_values): (Vec<_>, Vec<_>) = variants
 | 
			
		||||
            .iter()
 | 
			
		||||
            .map(|v| {
 | 
			
		||||
                let field_names: Vec<_> = v.fields.iter().map(|field| &field.name).collect();
 | 
			
		||||
                let var_names: Vec<_> = v.fields.iter().map(|field| field.var_name()).collect();
 | 
			
		||||
                let field_pats = quote! {
 | 
			
		||||
                    #(#field_names: #var_names,)*
 | 
			
		||||
                };
 | 
			
		||||
                let to_canonical_value = match &v.value {
 | 
			
		||||
                    VariantValue::None => quote! { ::fayalite::__std::option::Option::None },
 | 
			
		||||
                    VariantValue::Direct { .. } => {
 | 
			
		||||
                        debug_assert_eq!(var_names.len(), 1);
 | 
			
		||||
                        quote! {
 | 
			
		||||
                            ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                ::fayalite::ty::DynValueTrait::to_canonical_dyn(#(#var_names)*),
 | 
			
		||||
                            )
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    VariantValue::Struct {
 | 
			
		||||
                        value_struct,
 | 
			
		||||
                        parsed_struct,
 | 
			
		||||
                    } => {
 | 
			
		||||
                        let value_struct_ident = &value_struct.ident;
 | 
			
		||||
                        let phantom_field_name = &parsed_struct
 | 
			
		||||
                            .fields
 | 
			
		||||
                            .last()
 | 
			
		||||
                            .expect("missing phantom field")
 | 
			
		||||
                            .name;
 | 
			
		||||
                        let type_generics = static_type_type_generics.as_turbofish();
 | 
			
		||||
                        quote! {
 | 
			
		||||
                            ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                ::fayalite::ty::DynValueTrait::to_canonical_dyn(
 | 
			
		||||
                                    &#value_struct_ident #type_generics {
 | 
			
		||||
                                        #(#field_names:
 | 
			
		||||
                                            ::fayalite::__std::clone::Clone::clone(#var_names),)*
 | 
			
		||||
                                        #phantom_field_name: ::fayalite::__std::marker::PhantomData,
 | 
			
		||||
                                    },
 | 
			
		||||
                                ),
 | 
			
		||||
                            )
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                };
 | 
			
		||||
                (field_pats, to_canonical_value)
 | 
			
		||||
            })
 | 
			
		||||
            .unzip();
 | 
			
		||||
        let mut match_enum_variants = Punctuated::new();
 | 
			
		||||
        let mut match_enum_debug_arms = vec![];
 | 
			
		||||
        let mut match_enum_arms = vec![];
 | 
			
		||||
        let mut variant_vars = vec![];
 | 
			
		||||
        let mut from_canonical_type_variant_lets = vec![];
 | 
			
		||||
        let mut non_empty_variant_vars = vec![];
 | 
			
		||||
        let mut enum_type_variants = vec![];
 | 
			
		||||
        let mut enum_type_variants_hint = vec![];
 | 
			
		||||
        let match_enum_ident = format_ident!("__{}__MatchEnum", enum_ident);
 | 
			
		||||
        let mut builder = Builder::new(format_ident!("__{}__Builder", enum_ident), vis.clone());
 | 
			
		||||
        for variant in variants.iter() {
 | 
			
		||||
            for field in variant.fields.iter() {
 | 
			
		||||
                builder.insert_field(
 | 
			
		||||
                    field.name.clone(),
 | 
			
		||||
                    |v| {
 | 
			
		||||
                        parse_quote_spanned! {v.span()=>
 | 
			
		||||
                            ::fayalite::expr::ToExpr::to_expr(&#v)
 | 
			
		||||
                        }
 | 
			
		||||
                    },
 | 
			
		||||
                    |t| {
 | 
			
		||||
                        parse_quote_spanned! {t.span()=>
 | 
			
		||||
                            ::fayalite::expr::Expr<<
 | 
			
		||||
                                <#t as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                                as ::fayalite::ty::Type
 | 
			
		||||
                            >::Value>
 | 
			
		||||
                        }
 | 
			
		||||
                    },
 | 
			
		||||
                    |t| {
 | 
			
		||||
                        parse_quote_spanned! {t.span()=>
 | 
			
		||||
                            where
 | 
			
		||||
                                #t: ::fayalite::expr::ToExpr,
 | 
			
		||||
                        }
 | 
			
		||||
                    },
 | 
			
		||||
                );
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        let builder = builder.finish_filling_in_fields();
 | 
			
		||||
        builder.to_tokens(tokens);
 | 
			
		||||
        for (variant_index, variant) in variants.iter().enumerate() {
 | 
			
		||||
            let variant_var = format_ident!("__v_{}", variant.ident);
 | 
			
		||||
            let variant_name = &variant.ident;
 | 
			
		||||
            let variant_name_str = variant.ident.to_string();
 | 
			
		||||
            match_enum_variants.push(Variant {
 | 
			
		||||
                attrs: vec![],
 | 
			
		||||
                ident: variant.ident.clone(),
 | 
			
		||||
                fields: variant.fields_kind.into_fields(variant.fields.iter().map(
 | 
			
		||||
                    |ParsedField {
 | 
			
		||||
                         options,
 | 
			
		||||
                         vis,
 | 
			
		||||
                         name,
 | 
			
		||||
                         ty,
 | 
			
		||||
                     }| {
 | 
			
		||||
                        let FieldOptions {} = options.body;
 | 
			
		||||
                        Field {
 | 
			
		||||
                            attrs: vec![],
 | 
			
		||||
                            vis: vis.clone(),
 | 
			
		||||
                            mutability: FieldMutability::None,
 | 
			
		||||
                            ident: if let Member::Named(name) = name {
 | 
			
		||||
                                Some(name.clone())
 | 
			
		||||
                            } else {
 | 
			
		||||
                                None
 | 
			
		||||
                            },
 | 
			
		||||
                            colon_token: None,
 | 
			
		||||
                            ty: parse_quote! { ::fayalite::expr::Expr<#ty> },
 | 
			
		||||
                        }
 | 
			
		||||
                    },
 | 
			
		||||
                )),
 | 
			
		||||
                discriminant: None,
 | 
			
		||||
            });
 | 
			
		||||
            let match_enum_field_names = Vec::from_iter(variant.fields.iter().map(
 | 
			
		||||
                |ParsedField {
 | 
			
		||||
                     options,
 | 
			
		||||
                     vis: _,
 | 
			
		||||
                     name,
 | 
			
		||||
                     ty: _,
 | 
			
		||||
                 }| {
 | 
			
		||||
                    let FieldOptions {} = options.body;
 | 
			
		||||
                    name
 | 
			
		||||
                },
 | 
			
		||||
            ));
 | 
			
		||||
            let match_enum_field_name_strs = Vec::from_iter(variant.fields.iter().map(
 | 
			
		||||
                |ParsedField {
 | 
			
		||||
                     options,
 | 
			
		||||
                     vis: _,
 | 
			
		||||
                     name,
 | 
			
		||||
                     ty: _,
 | 
			
		||||
                 }| {
 | 
			
		||||
                    let FieldOptions {} = options.body;
 | 
			
		||||
                    name.to_token_stream().to_string()
 | 
			
		||||
                },
 | 
			
		||||
            ));
 | 
			
		||||
            let match_enum_debug_vars = Vec::from_iter(variant.fields.iter().map(
 | 
			
		||||
                |ParsedField {
 | 
			
		||||
                     options,
 | 
			
		||||
                     vis: _,
 | 
			
		||||
                     name,
 | 
			
		||||
                     ty: _,
 | 
			
		||||
                 }| {
 | 
			
		||||
                    let FieldOptions {} = options.body;
 | 
			
		||||
                    format_ident!("__v_{}", name)
 | 
			
		||||
                },
 | 
			
		||||
            ));
 | 
			
		||||
            match_enum_debug_arms.push(match variant.fields_kind {
 | 
			
		||||
                FieldsKind::Unit | FieldsKind::Named(_) => quote! {
 | 
			
		||||
                    Self::#variant_name {
 | 
			
		||||
                        #(#match_enum_field_names: ref #match_enum_debug_vars,)*
 | 
			
		||||
                    } => f.debug_struct(#variant_name_str)
 | 
			
		||||
                        #(.field(#match_enum_field_name_strs, #match_enum_debug_vars))*
 | 
			
		||||
                        .finish(),
 | 
			
		||||
                },
 | 
			
		||||
                FieldsKind::Unnamed(_) => quote! {
 | 
			
		||||
                    Self::#variant_name(
 | 
			
		||||
                        #(ref #match_enum_debug_vars,)*
 | 
			
		||||
                    ) => f.debug_tuple(#variant_name_str)
 | 
			
		||||
                        #(.field(#match_enum_debug_vars))*
 | 
			
		||||
                        .finish(),
 | 
			
		||||
                },
 | 
			
		||||
            });
 | 
			
		||||
            if let Some(value_ty) = variant.value.value_ty() {
 | 
			
		||||
                from_canonical_type_variant_lets.push(quote! {
 | 
			
		||||
                    let #variant_var =
 | 
			
		||||
                        #variant_var.from_canonical_type_helper_has_value(#variant_name_str);
 | 
			
		||||
                });
 | 
			
		||||
                non_empty_variant_vars.push(variant_var.clone());
 | 
			
		||||
                enum_type_variants.push(quote! {
 | 
			
		||||
                    ::fayalite::enum_::VariantType {
 | 
			
		||||
                        name: ::fayalite::intern::Intern::intern(#variant_name_str),
 | 
			
		||||
                        ty: ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                            ::fayalite::ty::DynType::canonical_dyn(&self.#variant_name),
 | 
			
		||||
                        ),
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
                enum_type_variants_hint.push(quote! {
 | 
			
		||||
                    ::fayalite::enum_::VariantType {
 | 
			
		||||
                        name: ::fayalite::intern::Intern::intern(#variant_name_str),
 | 
			
		||||
                        ty: ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                            ::fayalite::bundle::TypeHint::<
 | 
			
		||||
                                <#value_ty as ::fayalite::expr::ToExpr>::Type,
 | 
			
		||||
                            >::intern_dyn(),
 | 
			
		||||
                        ),
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
            } else {
 | 
			
		||||
                from_canonical_type_variant_lets.push(quote! {
 | 
			
		||||
                    #variant_var.from_canonical_type_helper_no_value(#variant_name_str);
 | 
			
		||||
                });
 | 
			
		||||
                enum_type_variants.push(quote! {
 | 
			
		||||
                    ::fayalite::enum_::VariantType {
 | 
			
		||||
                        name: ::fayalite::intern::Intern::intern(#variant_name_str),
 | 
			
		||||
                        ty: ::fayalite::__std::option::Option::None,
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
                enum_type_variants_hint.push(quote! {
 | 
			
		||||
                    ::fayalite::enum_::VariantType {
 | 
			
		||||
                        name: ::fayalite::intern::Intern::intern(#variant_name_str),
 | 
			
		||||
                        ty: ::fayalite::__std::option::Option::None,
 | 
			
		||||
                    }
 | 
			
		||||
                });
 | 
			
		||||
            }
 | 
			
		||||
            variant_vars.push(variant_var);
 | 
			
		||||
            match_enum_arms.push(match &variant.value {
 | 
			
		||||
                VariantValue::None => quote! {
 | 
			
		||||
                    #variant_index => #match_enum_ident::#variant_name,
 | 
			
		||||
                },
 | 
			
		||||
                VariantValue::Direct { value_type } => quote! {
 | 
			
		||||
                    #variant_index => #match_enum_ident::#variant_name {
 | 
			
		||||
                        #(#match_enum_field_names)*: ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                            &__variant_access.downcast_unchecked::<
 | 
			
		||||
                                <#value_type as ::fayalite::expr::ToExpr>::Type>(),
 | 
			
		||||
                        ),
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
                VariantValue::Struct {
 | 
			
		||||
                    value_struct: ItemStruct { ident, .. },
 | 
			
		||||
                    ..
 | 
			
		||||
                } => quote! {
 | 
			
		||||
                    #variant_index => {
 | 
			
		||||
                        let __variant_access = ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                            &__variant_access.downcast_unchecked::<<
 | 
			
		||||
                                #ident #static_type_type_generics
 | 
			
		||||
                                as ::fayalite::expr::ToExpr
 | 
			
		||||
                            >::Type>(),
 | 
			
		||||
                        );
 | 
			
		||||
                        #match_enum_ident::#variant_name {
 | 
			
		||||
                            #(#match_enum_field_names:
 | 
			
		||||
                                (*__variant_access).#match_enum_field_names,)*
 | 
			
		||||
                        }
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
            });
 | 
			
		||||
            let builder_field_and_types = Vec::from_iter(variant.fields.iter().map(
 | 
			
		||||
                |ParsedField {
 | 
			
		||||
                     options,
 | 
			
		||||
                     vis: _,
 | 
			
		||||
                     name,
 | 
			
		||||
                     ty,
 | 
			
		||||
                 }| {
 | 
			
		||||
                    let FieldOptions {} = options.body;
 | 
			
		||||
                    (name, ty)
 | 
			
		||||
                },
 | 
			
		||||
            ));
 | 
			
		||||
            let builder_field_vars = Vec::from_iter(
 | 
			
		||||
                builder_field_and_types
 | 
			
		||||
                    .iter()
 | 
			
		||||
                    .map(|(name, _)| &builder.get_field(name).unwrap().1.builder_field_name),
 | 
			
		||||
            );
 | 
			
		||||
            let build_body = match &variant.value {
 | 
			
		||||
                VariantValue::None => parse_quote! {
 | 
			
		||||
                    {
 | 
			
		||||
                        ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                            &::fayalite::expr::ops::EnumLiteral::<
 | 
			
		||||
                                #type_struct_ident #static_type_type_generics
 | 
			
		||||
                            >::new_unchecked(
 | 
			
		||||
                                ::fayalite::__std::option::Option::None,
 | 
			
		||||
                                #variant_index,
 | 
			
		||||
                                ::fayalite::ty::StaticType::static_type(),
 | 
			
		||||
                            ),
 | 
			
		||||
                        )
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
                VariantValue::Direct { value_type: _ } => parse_quote! {
 | 
			
		||||
                    {
 | 
			
		||||
                        ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                            &::fayalite::expr::ops::EnumLiteral::<
 | 
			
		||||
                                #type_struct_ident #static_type_type_generics
 | 
			
		||||
                            >::new_unchecked(
 | 
			
		||||
                                ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                    #(#builder_field_vars)*.to_canonical_dyn(),
 | 
			
		||||
                                ),
 | 
			
		||||
                                #variant_index,
 | 
			
		||||
                                ::fayalite::ty::StaticType::static_type(),
 | 
			
		||||
                            ),
 | 
			
		||||
                        )
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
                VariantValue::Struct {
 | 
			
		||||
                    parsed_struct:
 | 
			
		||||
                        ParsedStruct {
 | 
			
		||||
                            names:
 | 
			
		||||
                                ParsedStructNames {
 | 
			
		||||
                                    type_struct_ident: field_type_struct_ident,
 | 
			
		||||
                                    ..
 | 
			
		||||
                                },
 | 
			
		||||
                            ..
 | 
			
		||||
                        },
 | 
			
		||||
                    ..
 | 
			
		||||
                } => parse_quote! {
 | 
			
		||||
                    {
 | 
			
		||||
                        let __builder = <
 | 
			
		||||
                            #field_type_struct_ident #static_type_type_generics
 | 
			
		||||
                            as ::fayalite::bundle::BundleType
 | 
			
		||||
                        >::builder();
 | 
			
		||||
                        #(let __builder = __builder.#builder_field_vars(#builder_field_vars);)*
 | 
			
		||||
                        ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                            &::fayalite::expr::ops::EnumLiteral::<
 | 
			
		||||
                                #type_struct_ident #static_type_type_generics
 | 
			
		||||
                            >::new_unchecked(
 | 
			
		||||
                                ::fayalite::__std::option::Option::Some(
 | 
			
		||||
                                    __builder.build().to_canonical_dyn(),
 | 
			
		||||
                                ),
 | 
			
		||||
                                #variant_index,
 | 
			
		||||
                                ::fayalite::ty::StaticType::static_type(),
 | 
			
		||||
                            ),
 | 
			
		||||
                        )
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
            };
 | 
			
		||||
            builder
 | 
			
		||||
                .make_build_method(
 | 
			
		||||
                    &format_ident!("variant_{}", variant_name),
 | 
			
		||||
                    variant.fields.iter().map(
 | 
			
		||||
                        |ParsedField {
 | 
			
		||||
                             options,
 | 
			
		||||
                             vis: _,
 | 
			
		||||
                             name,
 | 
			
		||||
                             ty,
 | 
			
		||||
                         }| {
 | 
			
		||||
                            let FieldOptions {} = options.body;
 | 
			
		||||
                            (name.clone(), parse_quote! { ::fayalite::expr::Expr<#ty> })
 | 
			
		||||
                        },
 | 
			
		||||
                    ),
 | 
			
		||||
                    &static_type_generics,
 | 
			
		||||
                    &parse_quote! {#type_struct_ident #static_type_type_generics},
 | 
			
		||||
                    &parse_quote! { ::fayalite::expr::Expr<#target #static_type_type_generics> },
 | 
			
		||||
                    build_body,
 | 
			
		||||
                )
 | 
			
		||||
                .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let match_enum = ItemEnum {
 | 
			
		||||
            attrs: vec![parse_quote! {#[allow(non_camel_case_types)]}],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            enum_token: *enum_token,
 | 
			
		||||
            ident: match_enum_ident,
 | 
			
		||||
            generics: static_type_generics.clone(),
 | 
			
		||||
            brace_token: *brace_token,
 | 
			
		||||
            variants: match_enum_variants,
 | 
			
		||||
        };
 | 
			
		||||
        let match_enum_ident = &match_enum.ident;
 | 
			
		||||
        match_enum.to_tokens(tokens);
 | 
			
		||||
        make_connect_impl(
 | 
			
		||||
            *connect_inexact,
 | 
			
		||||
            &static_type_generics,
 | 
			
		||||
            type_struct_ident,
 | 
			
		||||
            variants.iter().flat_map(|variant| {
 | 
			
		||||
                variant.fields.iter().map(|field| {
 | 
			
		||||
                    let ty = &field.ty;
 | 
			
		||||
                    parse_quote_spanned! {field.name.span()=>
 | 
			
		||||
                        <#ty as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                    }
 | 
			
		||||
                })
 | 
			
		||||
            }),
 | 
			
		||||
        )
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let variant_count = variants.len();
 | 
			
		||||
        let empty_builder_ty = builder.ty([], Some(&parse_quote! { Self }), false);
 | 
			
		||||
        quote! {
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::__std::fmt::Debug
 | 
			
		||||
                for #match_enum_ident #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn fmt(
 | 
			
		||||
                    &self,
 | 
			
		||||
                    f: &mut ::fayalite::__std::fmt::Formatter<'_>,
 | 
			
		||||
                ) -> ::fayalite::__std::fmt::Result {
 | 
			
		||||
                    match *self {
 | 
			
		||||
                        #(#match_enum_debug_arms)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::ty::StaticType
 | 
			
		||||
                for #type_struct_ident #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn static_type() -> Self {
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#non_empty_variant_names: ::fayalite::ty::StaticType::static_type(),)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn __check_field<T: ::fayalite::ty::Value>()
 | 
			
		||||
            where
 | 
			
		||||
                <T as ::fayalite::expr::ToExpr>::Type: ::fayalite::ty::Type<Value = T>,
 | 
			
		||||
            {}
 | 
			
		||||
            fn __check_fields #static_type_impl_generics(_: #target #static_type_type_generics)
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                #(#field_checks)*
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::__std::fmt::Debug
 | 
			
		||||
                for #type_struct_ident #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn fmt(
 | 
			
		||||
                    &self,
 | 
			
		||||
                    f: &mut ::fayalite::__std::fmt::Formatter<'_>,
 | 
			
		||||
                ) -> ::fayalite::__std::fmt::Result {
 | 
			
		||||
                    #debug_type_body
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::ty::Type
 | 
			
		||||
                for #type_struct_ident #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type CanonicalType = ::fayalite::enum_::DynEnumType;
 | 
			
		||||
                type Value = #target #static_type_type_generics;
 | 
			
		||||
                type CanonicalValue = ::fayalite::enum_::DynEnum;
 | 
			
		||||
                type MaskType = ::fayalite::int::UIntType<1>;
 | 
			
		||||
                type MaskValue = ::fayalite::int::UInt<1>;
 | 
			
		||||
                type MatchVariant = #match_enum_ident #static_type_type_generics;
 | 
			
		||||
                type MatchActiveScope = ::fayalite::module::Scope;
 | 
			
		||||
                type MatchVariantAndInactiveScope =
 | 
			
		||||
                    ::fayalite::enum_::EnumMatchVariantAndInactiveScope<Self>;
 | 
			
		||||
                type MatchVariantsIter = ::fayalite::enum_::EnumMatchVariantsIter<Self>;
 | 
			
		||||
                fn match_variants<IO: ::fayalite::bundle::BundleValue>(
 | 
			
		||||
                    this: ::fayalite::expr::Expr<<Self as ::fayalite::ty::Type>::Value>,
 | 
			
		||||
                    module_builder: &mut ::fayalite::module::ModuleBuilder<
 | 
			
		||||
                        IO,
 | 
			
		||||
                        ::fayalite::module::NormalModule,
 | 
			
		||||
                    >,
 | 
			
		||||
                    source_location: ::fayalite::source_location::SourceLocation,
 | 
			
		||||
                ) -> <Self as ::fayalite::ty::Type>::MatchVariantsIter
 | 
			
		||||
                where
 | 
			
		||||
                    <IO as ::fayalite::expr::ToExpr>::Type:
 | 
			
		||||
                        ::fayalite::bundle::BundleType<Value = IO>,
 | 
			
		||||
                {
 | 
			
		||||
                    module_builder.enum_match_variants_helper(this, source_location)
 | 
			
		||||
                }
 | 
			
		||||
                fn mask_type(&self) -> <Self as ::fayalite::ty::Type>::MaskType {
 | 
			
		||||
                    ::fayalite::int::UIntType::new()
 | 
			
		||||
                }
 | 
			
		||||
                fn canonical(&self) -> <Self as ::fayalite::ty::Type>::CanonicalType {
 | 
			
		||||
                    let variants = ::fayalite::enum_::EnumType::variants(self);
 | 
			
		||||
                    ::fayalite::enum_::DynEnumType::new(variants)
 | 
			
		||||
                }
 | 
			
		||||
                fn source_location(&self) -> ::fayalite::source_location::SourceLocation {
 | 
			
		||||
                    ::fayalite::source_location::SourceLocation::caller()
 | 
			
		||||
                }
 | 
			
		||||
                fn type_enum(&self) -> ::fayalite::ty::TypeEnum {
 | 
			
		||||
                    ::fayalite::ty::TypeEnum::EnumType(::fayalite::ty::Type::canonical(self))
 | 
			
		||||
                }
 | 
			
		||||
                #[allow(non_snake_case)]
 | 
			
		||||
                fn from_canonical_type(t: <Self as ::fayalite::ty::Type>::CanonicalType) -> Self {
 | 
			
		||||
                    let [#(#variant_vars),*] = *::fayalite::enum_::EnumType::variants(&t) else {
 | 
			
		||||
                        ::fayalite::__std::panic!("wrong number of variants");
 | 
			
		||||
                    };
 | 
			
		||||
                    #(#from_canonical_type_variant_lets)*
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#non_empty_variant_names: #non_empty_variant_vars,)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            #[allow(clippy::init_numbered_fields)]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::enum_::EnumType
 | 
			
		||||
                for #type_struct_ident #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Builder = #empty_builder_ty;
 | 
			
		||||
                fn match_activate_scope(
 | 
			
		||||
                    v: <Self as ::fayalite::ty::Type>::MatchVariantAndInactiveScope,
 | 
			
		||||
                ) -> (
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariant,
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchActiveScope,
 | 
			
		||||
                ) {
 | 
			
		||||
                    let (__variant_access, __scope) = v.activate();
 | 
			
		||||
                    (
 | 
			
		||||
                        match ::fayalite::expr::ops::VariantAccess::variant_index(
 | 
			
		||||
                            &*__variant_access,
 | 
			
		||||
                        ) {
 | 
			
		||||
                            #(#match_enum_arms)*
 | 
			
		||||
                            #variant_count.. => ::fayalite::__std::panic!("invalid variant index"),
 | 
			
		||||
                        },
 | 
			
		||||
                        __scope,
 | 
			
		||||
                    )
 | 
			
		||||
                }
 | 
			
		||||
                fn builder() -> <Self as ::fayalite::enum_::EnumType>::Builder {
 | 
			
		||||
                    #empty_builder_ty::new()
 | 
			
		||||
                }
 | 
			
		||||
                fn variants(&self) -> ::fayalite::intern::Interned<[::fayalite::enum_::VariantType<
 | 
			
		||||
                    ::fayalite::intern::Interned<dyn ::fayalite::ty::DynCanonicalType>,
 | 
			
		||||
                >]> {
 | 
			
		||||
                    ::fayalite::intern::Intern::intern(&[#(#enum_type_variants,)*][..])
 | 
			
		||||
                }
 | 
			
		||||
                fn variants_hint() -> ::fayalite::enum_::VariantsHint {
 | 
			
		||||
                    ::fayalite::enum_::VariantsHint::new([#(#enum_type_variants_hint,)*], false)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::expr::ToExpr
 | 
			
		||||
                for #target #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Type = #type_struct_ident #static_type_type_generics;
 | 
			
		||||
                fn ty(&self) -> <Self as ::fayalite::expr::ToExpr>::Type {
 | 
			
		||||
                    ::fayalite::ty::StaticType::static_type()
 | 
			
		||||
                }
 | 
			
		||||
                fn to_expr(&self) -> ::fayalite::expr::Expr<Self> {
 | 
			
		||||
                    ::fayalite::expr::Expr::from_value(self)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::ty::Value
 | 
			
		||||
                for #target #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn to_canonical(&self) -> <
 | 
			
		||||
                        <Self as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                        as ::fayalite::ty::Type
 | 
			
		||||
                    >::CanonicalValue
 | 
			
		||||
                {
 | 
			
		||||
                    let __ty = ::fayalite::ty::Type::canonical(&::fayalite::expr::ToExpr::ty(self));
 | 
			
		||||
                    match self {
 | 
			
		||||
                        #(Self::#variant_names { #variant_field_pats } => {
 | 
			
		||||
                            ::fayalite::enum_::DynEnum::new_by_name(
 | 
			
		||||
                                __ty,
 | 
			
		||||
                                ::fayalite::intern::Intern::intern(#variant_name_strs),
 | 
			
		||||
                                #variant_to_canonical_values,
 | 
			
		||||
                            )
 | 
			
		||||
                        })*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #static_type_impl_generics ::fayalite::enum_::EnumValue
 | 
			
		||||
                for #target #static_type_type_generics
 | 
			
		||||
            #static_type_where_clause
 | 
			
		||||
            {
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn value_derive_enum(item: ItemEnum) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = ParsedEnum::parse(item)?;
 | 
			
		||||
    let outline_generated = item.options.body.outline_generated;
 | 
			
		||||
    let mut contents = quote! {
 | 
			
		||||
        const _: () = {
 | 
			
		||||
            #item
 | 
			
		||||
        };
 | 
			
		||||
    };
 | 
			
		||||
    if outline_generated.is_some() {
 | 
			
		||||
        contents = crate::outline_generated(contents, "value-enum-");
 | 
			
		||||
    }
 | 
			
		||||
    Ok(contents)
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -1,765 +0,0 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    value_derive_common::{
 | 
			
		||||
        append_field, derive_clone_hash_eq_partialeq_for_struct, get_target, make_connect_impl,
 | 
			
		||||
        Bounds, Builder, FieldsKind, ParsedField, ValueDeriveGenerics,
 | 
			
		||||
    },
 | 
			
		||||
    Errors, HdlAttr,
 | 
			
		||||
};
 | 
			
		||||
use proc_macro2::TokenStream;
 | 
			
		||||
use quote::{format_ident, quote, quote_spanned, ToTokens};
 | 
			
		||||
use syn::{
 | 
			
		||||
    parse_quote, parse_quote_spanned, spanned::Spanned, FieldMutability, Generics, Ident,
 | 
			
		||||
    ItemStruct, Member, Path, Token, Visibility,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = StructOptions]
 | 
			
		||||
    pub(crate) enum StructOption {
 | 
			
		||||
        OutlineGenerated(outline_generated),
 | 
			
		||||
        Static(static_),
 | 
			
		||||
        ConnectInexact(connect_inexact),
 | 
			
		||||
        Bounds(where_, Bounds),
 | 
			
		||||
        Target(target, Path),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
crate::options! {
 | 
			
		||||
    #[options = FieldOptions]
 | 
			
		||||
    pub(crate) enum FieldOption {
 | 
			
		||||
        Flip(flip),
 | 
			
		||||
        Skip(skip),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) struct ParsedStructNames<I, S> {
 | 
			
		||||
    pub(crate) ident: Ident,
 | 
			
		||||
    pub(crate) type_struct_debug_ident: S,
 | 
			
		||||
    pub(crate) type_struct_ident: Ident,
 | 
			
		||||
    pub(crate) match_variant_ident: I,
 | 
			
		||||
    pub(crate) builder_struct_ident: I,
 | 
			
		||||
    pub(crate) mask_match_variant_ident: I,
 | 
			
		||||
    pub(crate) mask_type_ident: I,
 | 
			
		||||
    pub(crate) mask_type_debug_ident: S,
 | 
			
		||||
    pub(crate) mask_value_ident: I,
 | 
			
		||||
    pub(crate) mask_value_debug_ident: S,
 | 
			
		||||
    pub(crate) mask_builder_struct_ident: I,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) struct ParsedStruct {
 | 
			
		||||
    pub(crate) options: HdlAttr<StructOptions>,
 | 
			
		||||
    pub(crate) vis: Visibility,
 | 
			
		||||
    pub(crate) struct_token: Token![struct],
 | 
			
		||||
    pub(crate) generics: Generics,
 | 
			
		||||
    pub(crate) fields_kind: FieldsKind,
 | 
			
		||||
    pub(crate) fields: Vec<ParsedField<FieldOptions>>,
 | 
			
		||||
    pub(crate) semi_token: Option<Token![;]>,
 | 
			
		||||
    pub(crate) skip_check_fields: bool,
 | 
			
		||||
    pub(crate) names: ParsedStructNames<Option<Ident>, Option<String>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ParsedStruct {
 | 
			
		||||
    pub(crate) fn parse(item: &mut ItemStruct) -> syn::Result<Self> {
 | 
			
		||||
        let ItemStruct {
 | 
			
		||||
            attrs,
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            ident,
 | 
			
		||||
            generics,
 | 
			
		||||
            fields,
 | 
			
		||||
            semi_token,
 | 
			
		||||
        } = item;
 | 
			
		||||
        let mut errors = Errors::new();
 | 
			
		||||
        let struct_options = errors
 | 
			
		||||
            .unwrap_or_default(HdlAttr::parse_and_take_attr(attrs))
 | 
			
		||||
            .unwrap_or_default();
 | 
			
		||||
        let (fields_kind, fields) = ParsedField::parse_fields(&mut errors, fields, false);
 | 
			
		||||
        errors.finish()?;
 | 
			
		||||
        Ok(ParsedStruct {
 | 
			
		||||
            options: struct_options,
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            generics: generics.clone(),
 | 
			
		||||
            fields_kind,
 | 
			
		||||
            fields,
 | 
			
		||||
            semi_token: *semi_token,
 | 
			
		||||
            skip_check_fields: false,
 | 
			
		||||
            names: ParsedStructNames {
 | 
			
		||||
                ident: ident.clone(),
 | 
			
		||||
                type_struct_debug_ident: None,
 | 
			
		||||
                type_struct_ident: format_ident!("__{}__Type", ident),
 | 
			
		||||
                match_variant_ident: None,
 | 
			
		||||
                builder_struct_ident: None,
 | 
			
		||||
                mask_match_variant_ident: None,
 | 
			
		||||
                mask_type_ident: None,
 | 
			
		||||
                mask_type_debug_ident: None,
 | 
			
		||||
                mask_value_ident: None,
 | 
			
		||||
                mask_value_debug_ident: None,
 | 
			
		||||
                mask_builder_struct_ident: None,
 | 
			
		||||
            },
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
    pub(crate) fn write_body(
 | 
			
		||||
        &self,
 | 
			
		||||
        target: Path,
 | 
			
		||||
        names: ParsedStructNames<&Ident, &String>,
 | 
			
		||||
        is_for_mask: bool,
 | 
			
		||||
        tokens: &mut TokenStream,
 | 
			
		||||
    ) {
 | 
			
		||||
        let Self {
 | 
			
		||||
            options,
 | 
			
		||||
            vis,
 | 
			
		||||
            struct_token,
 | 
			
		||||
            generics,
 | 
			
		||||
            fields_kind,
 | 
			
		||||
            fields,
 | 
			
		||||
            semi_token,
 | 
			
		||||
            skip_check_fields,
 | 
			
		||||
            names: _,
 | 
			
		||||
        } = self;
 | 
			
		||||
        let skip_check_fields = *skip_check_fields || is_for_mask;
 | 
			
		||||
        let ParsedStructNames {
 | 
			
		||||
            ident: struct_ident,
 | 
			
		||||
            type_struct_debug_ident,
 | 
			
		||||
            type_struct_ident,
 | 
			
		||||
            match_variant_ident,
 | 
			
		||||
            builder_struct_ident,
 | 
			
		||||
            mask_match_variant_ident: _,
 | 
			
		||||
            mask_type_ident,
 | 
			
		||||
            mask_type_debug_ident: _,
 | 
			
		||||
            mask_value_ident,
 | 
			
		||||
            mask_value_debug_ident,
 | 
			
		||||
            mask_builder_struct_ident: _,
 | 
			
		||||
        } = names;
 | 
			
		||||
        let StructOptions {
 | 
			
		||||
            outline_generated: _,
 | 
			
		||||
            where_,
 | 
			
		||||
            target: _,
 | 
			
		||||
            static_,
 | 
			
		||||
            connect_inexact,
 | 
			
		||||
        } = &options.body;
 | 
			
		||||
        let ValueDeriveGenerics {
 | 
			
		||||
            generics,
 | 
			
		||||
            static_type_generics,
 | 
			
		||||
        } = ValueDeriveGenerics::get(generics.clone(), where_);
 | 
			
		||||
        let (impl_generics, type_generics, where_clause) = generics.split_for_impl();
 | 
			
		||||
        let unskipped_fields = fields
 | 
			
		||||
            .iter()
 | 
			
		||||
            .filter(|field| field.options.body.skip.is_none());
 | 
			
		||||
        let _field_names = Vec::from_iter(fields.iter().map(|field| field.name.clone()));
 | 
			
		||||
        let unskipped_field_names =
 | 
			
		||||
            Vec::from_iter(unskipped_fields.clone().map(|field| field.name.clone()));
 | 
			
		||||
        let unskipped_field_name_strs = Vec::from_iter(
 | 
			
		||||
            unskipped_field_names
 | 
			
		||||
                .iter()
 | 
			
		||||
                .map(|field_name| field_name.to_token_stream().to_string()),
 | 
			
		||||
        );
 | 
			
		||||
        let unskipped_field_vars = Vec::from_iter(
 | 
			
		||||
            unskipped_field_names
 | 
			
		||||
                .iter()
 | 
			
		||||
                .map(|field_name| format_ident!("__v_{}", field_name)),
 | 
			
		||||
        );
 | 
			
		||||
        let unskipped_field_flips = Vec::from_iter(
 | 
			
		||||
            unskipped_fields
 | 
			
		||||
                .clone()
 | 
			
		||||
                .map(|field| field.options.body.flip.is_some()),
 | 
			
		||||
        );
 | 
			
		||||
        let mut any_fields_skipped = false;
 | 
			
		||||
        let type_fields = Vec::from_iter(fields.iter().filter_map(|field| {
 | 
			
		||||
            let ParsedField {
 | 
			
		||||
                options,
 | 
			
		||||
                vis,
 | 
			
		||||
                name,
 | 
			
		||||
                ty,
 | 
			
		||||
            } = field;
 | 
			
		||||
            let FieldOptions { flip: _, skip } = &options.body;
 | 
			
		||||
            if skip.is_some() {
 | 
			
		||||
                any_fields_skipped = true;
 | 
			
		||||
                return None;
 | 
			
		||||
            }
 | 
			
		||||
            let ty = if is_for_mask {
 | 
			
		||||
                parse_quote! { ::fayalite::ty::AsMask<#ty> }
 | 
			
		||||
            } else {
 | 
			
		||||
                ty.to_token_stream()
 | 
			
		||||
            };
 | 
			
		||||
            Some(syn::Field {
 | 
			
		||||
                attrs: vec![],
 | 
			
		||||
                vis: vis.clone(),
 | 
			
		||||
                mutability: FieldMutability::None,
 | 
			
		||||
                ident: match name.clone() {
 | 
			
		||||
                    Member::Named(name) => Some(name),
 | 
			
		||||
                    Member::Unnamed(_) => None,
 | 
			
		||||
                },
 | 
			
		||||
                colon_token: None,
 | 
			
		||||
                ty: parse_quote! { <#ty as ::fayalite::expr::ToExpr>::Type },
 | 
			
		||||
            })
 | 
			
		||||
        }));
 | 
			
		||||
        let field_types = Vec::from_iter(type_fields.iter().map(|field| field.ty.clone()));
 | 
			
		||||
        let match_variant_fields = Vec::from_iter(fields.iter().zip(&type_fields).map(
 | 
			
		||||
            |(parsed_field, type_field)| {
 | 
			
		||||
                let field_ty = &parsed_field.ty;
 | 
			
		||||
                syn::Field {
 | 
			
		||||
                    ty: parse_quote! { ::fayalite::expr::Expr<#field_ty> },
 | 
			
		||||
                    ..type_field.clone()
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
 | 
			
		||||
        let mask_value_fields = Vec::from_iter(fields.iter().zip(&type_fields).map(
 | 
			
		||||
            |(parsed_field, type_field)| {
 | 
			
		||||
                let field_ty = &parsed_field.ty;
 | 
			
		||||
                syn::Field {
 | 
			
		||||
                    ty: parse_quote! { ::fayalite::ty::AsMask<#field_ty> },
 | 
			
		||||
                    ..type_field.clone()
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        ));
 | 
			
		||||
 | 
			
		||||
        let mut type_struct_fields = fields_kind.into_fields(type_fields);
 | 
			
		||||
        let mut match_variant_fields = fields_kind.into_fields(match_variant_fields);
 | 
			
		||||
        let mut mask_value_fields = fields_kind.into_fields(mask_value_fields);
 | 
			
		||||
        let phantom_data_field_name = any_fields_skipped.then(|| {
 | 
			
		||||
            let phantom_data_field_name = Ident::new("__phantom_data", type_struct_ident.span());
 | 
			
		||||
            let member = append_field(
 | 
			
		||||
                &mut type_struct_fields,
 | 
			
		||||
                syn::Field {
 | 
			
		||||
                    attrs: vec![],
 | 
			
		||||
                    vis: vis.clone(),
 | 
			
		||||
                    mutability: FieldMutability::None,
 | 
			
		||||
                    ident: Some(phantom_data_field_name.clone()),
 | 
			
		||||
                    colon_token: None,
 | 
			
		||||
                    ty: parse_quote_spanned! {type_struct_ident.span()=>
 | 
			
		||||
                        ::fayalite::__std::marker::PhantomData<#struct_ident #type_generics>
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
            append_field(
 | 
			
		||||
                &mut match_variant_fields,
 | 
			
		||||
                syn::Field {
 | 
			
		||||
                    attrs: vec![],
 | 
			
		||||
                    vis: Visibility::Inherited,
 | 
			
		||||
                    mutability: FieldMutability::None,
 | 
			
		||||
                    ident: Some(phantom_data_field_name.clone()),
 | 
			
		||||
                    colon_token: None,
 | 
			
		||||
                    ty: parse_quote_spanned! {type_struct_ident.span()=>
 | 
			
		||||
                        ::fayalite::__std::marker::PhantomData<#struct_ident #type_generics>
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
            append_field(
 | 
			
		||||
                &mut mask_value_fields,
 | 
			
		||||
                syn::Field {
 | 
			
		||||
                    attrs: vec![],
 | 
			
		||||
                    vis: Visibility::Inherited,
 | 
			
		||||
                    mutability: FieldMutability::None,
 | 
			
		||||
                    ident: Some(phantom_data_field_name),
 | 
			
		||||
                    colon_token: None,
 | 
			
		||||
                    ty: parse_quote_spanned! {type_struct_ident.span()=>
 | 
			
		||||
                        ::fayalite::__std::marker::PhantomData<#struct_ident #type_generics>
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
            member
 | 
			
		||||
        });
 | 
			
		||||
        let phantom_data_field_name_slice = phantom_data_field_name.as_slice();
 | 
			
		||||
        let type_struct = ItemStruct {
 | 
			
		||||
            attrs: vec![parse_quote! {#[allow(non_camel_case_types)]}],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: type_struct_ident.clone(),
 | 
			
		||||
            generics: generics.clone(),
 | 
			
		||||
            fields: type_struct_fields,
 | 
			
		||||
            semi_token: *semi_token,
 | 
			
		||||
        };
 | 
			
		||||
        type_struct.to_tokens(tokens);
 | 
			
		||||
        let match_variant_struct = ItemStruct {
 | 
			
		||||
            attrs: vec![parse_quote! {#[allow(non_camel_case_types)]}],
 | 
			
		||||
            vis: vis.clone(),
 | 
			
		||||
            struct_token: *struct_token,
 | 
			
		||||
            ident: match_variant_ident.clone(),
 | 
			
		||||
            generics: generics.clone(),
 | 
			
		||||
            fields: match_variant_fields,
 | 
			
		||||
            semi_token: *semi_token,
 | 
			
		||||
        };
 | 
			
		||||
        match_variant_struct.to_tokens(tokens);
 | 
			
		||||
        let mask_type_body = if is_for_mask {
 | 
			
		||||
            quote! {
 | 
			
		||||
                ::fayalite::__std::clone::Clone::clone(self)
 | 
			
		||||
            }
 | 
			
		||||
        } else {
 | 
			
		||||
            let mask_value_struct = ItemStruct {
 | 
			
		||||
                attrs: vec![parse_quote! {#[allow(non_camel_case_types)]}],
 | 
			
		||||
                vis: vis.clone(),
 | 
			
		||||
                struct_token: *struct_token,
 | 
			
		||||
                ident: mask_value_ident.clone(),
 | 
			
		||||
                generics: generics.clone(),
 | 
			
		||||
                fields: mask_value_fields,
 | 
			
		||||
                semi_token: *semi_token,
 | 
			
		||||
            };
 | 
			
		||||
            mask_value_struct.to_tokens(tokens);
 | 
			
		||||
            let debug_mask_value_body = match fields_kind {
 | 
			
		||||
                FieldsKind::Unit => quote! {
 | 
			
		||||
                    f.debug_struct(#mask_value_debug_ident).finish()
 | 
			
		||||
                },
 | 
			
		||||
                FieldsKind::Named(_) => quote! {
 | 
			
		||||
                    f.debug_struct(#mask_value_debug_ident)
 | 
			
		||||
                        #(.field(#unskipped_field_name_strs, &self.#unskipped_field_names))*
 | 
			
		||||
                        .finish()
 | 
			
		||||
                },
 | 
			
		||||
                FieldsKind::Unnamed(_) => quote! {
 | 
			
		||||
                    f.debug_tuple(#mask_value_debug_ident)
 | 
			
		||||
                        #(.field(&self.#unskipped_field_names))*
 | 
			
		||||
                        .finish()
 | 
			
		||||
                },
 | 
			
		||||
            };
 | 
			
		||||
            quote! {
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #impl_generics ::fayalite::__std::fmt::Debug
 | 
			
		||||
                    for #mask_value_ident #type_generics
 | 
			
		||||
                #where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    fn fmt(
 | 
			
		||||
                        &self,
 | 
			
		||||
                        f: &mut ::fayalite::__std::fmt::Formatter<'_>,
 | 
			
		||||
                    ) -> ::fayalite::__std::fmt::Result {
 | 
			
		||||
                        #debug_mask_value_body
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
            quote! {
 | 
			
		||||
                #mask_type_ident {
 | 
			
		||||
                    #(#unskipped_field_names:
 | 
			
		||||
                        ::fayalite::ty::Type::mask_type(&self.#unskipped_field_names),)*
 | 
			
		||||
                    #(#phantom_data_field_name_slice: ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        let debug_type_body = match fields_kind {
 | 
			
		||||
            FieldsKind::Unit => quote! {
 | 
			
		||||
                f.debug_struct(#type_struct_debug_ident).finish()
 | 
			
		||||
            },
 | 
			
		||||
            FieldsKind::Named(_) => quote! {
 | 
			
		||||
                f.debug_struct(#type_struct_debug_ident)
 | 
			
		||||
                    #(.field(#unskipped_field_name_strs, &self.#unskipped_field_names))*
 | 
			
		||||
                    .finish()
 | 
			
		||||
            },
 | 
			
		||||
            FieldsKind::Unnamed(_) => quote! {
 | 
			
		||||
                f.debug_tuple(#type_struct_debug_ident)
 | 
			
		||||
                    #(.field(&self.#unskipped_field_names))*
 | 
			
		||||
                    .finish()
 | 
			
		||||
            },
 | 
			
		||||
        };
 | 
			
		||||
        for the_struct_ident in [&type_struct_ident, match_variant_ident]
 | 
			
		||||
            .into_iter()
 | 
			
		||||
            .chain(is_for_mask.then_some(mask_value_ident))
 | 
			
		||||
        {
 | 
			
		||||
            derive_clone_hash_eq_partialeq_for_struct(
 | 
			
		||||
                the_struct_ident,
 | 
			
		||||
                &generics,
 | 
			
		||||
                &Vec::from_iter(
 | 
			
		||||
                    unskipped_field_names
 | 
			
		||||
                        .iter()
 | 
			
		||||
                        .cloned()
 | 
			
		||||
                        .chain(phantom_data_field_name.clone()),
 | 
			
		||||
                ),
 | 
			
		||||
            )
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let check_v = format_ident!("__v");
 | 
			
		||||
        let field_checks = Vec::from_iter(fields.iter().map(|ParsedField { ty, name, .. }| {
 | 
			
		||||
            quote_spanned! {ty.span()=>
 | 
			
		||||
                __check_field(#check_v.#name);
 | 
			
		||||
            }
 | 
			
		||||
        }));
 | 
			
		||||
        if static_.is_some() {
 | 
			
		||||
            let (impl_generics, type_generics, where_clause) =
 | 
			
		||||
                static_type_generics.split_for_impl();
 | 
			
		||||
            quote! {
 | 
			
		||||
                #[automatically_derived]
 | 
			
		||||
                impl #impl_generics ::fayalite::ty::StaticType for #type_struct_ident #type_generics
 | 
			
		||||
                #where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    fn static_type() -> Self {
 | 
			
		||||
                        Self {
 | 
			
		||||
                            #(#unskipped_field_names: ::fayalite::ty::StaticType::static_type(),)*
 | 
			
		||||
                            #(#phantom_data_field_name_slice:
 | 
			
		||||
                                ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        if !skip_check_fields {
 | 
			
		||||
            quote! {
 | 
			
		||||
                fn __check_field<T: ::fayalite::ty::Value>(_v: T)
 | 
			
		||||
                where
 | 
			
		||||
                    <T as ::fayalite::expr::ToExpr>::Type: ::fayalite::ty::Type<Value = T>,
 | 
			
		||||
                {}
 | 
			
		||||
                fn __check_fields #impl_generics(#check_v: #target #type_generics)
 | 
			
		||||
                #where_clause
 | 
			
		||||
                {
 | 
			
		||||
                    #(#field_checks)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        }
 | 
			
		||||
        let mut builder = Builder::new(builder_struct_ident.clone(), vis.clone());
 | 
			
		||||
        for field in unskipped_fields.clone() {
 | 
			
		||||
            builder.insert_field(
 | 
			
		||||
                field.name.clone(),
 | 
			
		||||
                |v| {
 | 
			
		||||
                    parse_quote_spanned! {v.span()=>
 | 
			
		||||
                        ::fayalite::expr::ToExpr::to_expr(&#v)
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
                |t| {
 | 
			
		||||
                    parse_quote_spanned! {t.span()=>
 | 
			
		||||
                        ::fayalite::expr::Expr<<
 | 
			
		||||
                            <#t as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                            as ::fayalite::ty::Type
 | 
			
		||||
                        >::Value>
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
                |t| {
 | 
			
		||||
                    parse_quote_spanned! {t.span()=>
 | 
			
		||||
                        where
 | 
			
		||||
                            #t: ::fayalite::expr::ToExpr,
 | 
			
		||||
                    }
 | 
			
		||||
                },
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
        let builder = builder.finish_filling_in_fields();
 | 
			
		||||
        builder.to_tokens(tokens);
 | 
			
		||||
        let build_type_fields =
 | 
			
		||||
            Vec::from_iter(unskipped_fields.clone().map(|ParsedField { name, .. }| {
 | 
			
		||||
                let builder_field_name = &builder.get_field(name).unwrap().1.builder_field_name;
 | 
			
		||||
                quote_spanned! {struct_ident.span()=>
 | 
			
		||||
                    #name: ::fayalite::expr::ToExpr::ty(&#builder_field_name)
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
        let build_expr_fields =
 | 
			
		||||
            Vec::from_iter(unskipped_fields.clone().map(|ParsedField { name, .. }| {
 | 
			
		||||
                let builder_field_name = &builder.get_field(name).unwrap().1.builder_field_name;
 | 
			
		||||
                quote_spanned! {struct_ident.span()=>
 | 
			
		||||
                    #builder_field_name.to_canonical_dyn()
 | 
			
		||||
                }
 | 
			
		||||
            }));
 | 
			
		||||
        let build_specified_fields = unskipped_fields.clone().map(
 | 
			
		||||
            |ParsedField {
 | 
			
		||||
                 options: _,
 | 
			
		||||
                 vis: _,
 | 
			
		||||
                 name,
 | 
			
		||||
                 ty,
 | 
			
		||||
             }| {
 | 
			
		||||
                let ty = if is_for_mask {
 | 
			
		||||
                    parse_quote_spanned! {name.span()=>
 | 
			
		||||
                        ::fayalite::expr::Expr<::fayalite::ty::AsMask<#ty>>
 | 
			
		||||
                    }
 | 
			
		||||
                } else {
 | 
			
		||||
                    parse_quote_spanned! {name.span()=>
 | 
			
		||||
                        ::fayalite::expr::Expr<#ty>
 | 
			
		||||
                    }
 | 
			
		||||
                };
 | 
			
		||||
                (name.clone(), ty)
 | 
			
		||||
            },
 | 
			
		||||
        );
 | 
			
		||||
        let build_body = parse_quote_spanned! {struct_ident.span()=>
 | 
			
		||||
            {
 | 
			
		||||
                ::fayalite::expr::ToExpr::to_expr(
 | 
			
		||||
                    &::fayalite::expr::ops::BundleLiteral::new_unchecked(
 | 
			
		||||
                        ::fayalite::intern::Intern::intern(&[#(
 | 
			
		||||
                            #build_expr_fields,
 | 
			
		||||
                        )*][..]),
 | 
			
		||||
                        #type_struct_ident {
 | 
			
		||||
                            #(#build_type_fields,)*
 | 
			
		||||
                            #(#phantom_data_field_name_slice:
 | 
			
		||||
                                ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                        },
 | 
			
		||||
                    ),
 | 
			
		||||
                )
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        builder
 | 
			
		||||
            .make_build_method(
 | 
			
		||||
                &Ident::new("build", struct_ident.span()),
 | 
			
		||||
                build_specified_fields,
 | 
			
		||||
                &generics,
 | 
			
		||||
                &parse_quote_spanned! {struct_ident.span()=>
 | 
			
		||||
                    #type_struct_ident #type_generics
 | 
			
		||||
                },
 | 
			
		||||
                &parse_quote_spanned! {struct_ident.span()=>
 | 
			
		||||
                    ::fayalite::expr::Expr<#target #type_generics>
 | 
			
		||||
                },
 | 
			
		||||
                build_body,
 | 
			
		||||
            )
 | 
			
		||||
            .to_tokens(tokens);
 | 
			
		||||
        make_connect_impl(
 | 
			
		||||
            *connect_inexact,
 | 
			
		||||
            &generics,
 | 
			
		||||
            &type_struct_ident,
 | 
			
		||||
            unskipped_fields.clone().map(|field| {
 | 
			
		||||
                let ty = &field.ty;
 | 
			
		||||
                parse_quote_spanned! {field.name.span()=>
 | 
			
		||||
                    <#ty as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                }
 | 
			
		||||
            }),
 | 
			
		||||
        )
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
        let empty_builder_ty = builder.ty([], Some(&parse_quote! { Self }), false);
 | 
			
		||||
        quote! {
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::__std::fmt::Debug for #type_struct_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn fmt(
 | 
			
		||||
                    &self,
 | 
			
		||||
                    f: &mut ::fayalite::__std::fmt::Formatter<'_>,
 | 
			
		||||
                ) -> ::fayalite::__std::fmt::Result {
 | 
			
		||||
                    #debug_type_body
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::Type for #type_struct_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type CanonicalType = ::fayalite::bundle::DynBundleType;
 | 
			
		||||
                type Value = #target #type_generics;
 | 
			
		||||
                type CanonicalValue = ::fayalite::bundle::DynBundle;
 | 
			
		||||
                type MaskType = #mask_type_ident #type_generics;
 | 
			
		||||
                type MaskValue = #mask_value_ident #type_generics;
 | 
			
		||||
                type MatchVariant = #match_variant_ident #type_generics;
 | 
			
		||||
                type MatchActiveScope = ();
 | 
			
		||||
                type MatchVariantAndInactiveScope = ::fayalite::ty::MatchVariantWithoutScope<
 | 
			
		||||
                    #match_variant_ident #type_generics,
 | 
			
		||||
                >;
 | 
			
		||||
                type MatchVariantsIter = ::fayalite::__std::iter::Once<
 | 
			
		||||
                    <Self as ::fayalite::ty::Type>::MatchVariantAndInactiveScope,
 | 
			
		||||
                >;
 | 
			
		||||
                #[allow(unused_variables)]
 | 
			
		||||
                fn match_variants<IO: ::fayalite::bundle::BundleValue>(
 | 
			
		||||
                    this: ::fayalite::expr::Expr<<Self as ::fayalite::ty::Type>::Value>,
 | 
			
		||||
                    module_builder: &mut ::fayalite::module::ModuleBuilder<
 | 
			
		||||
                        IO,
 | 
			
		||||
                        ::fayalite::module::NormalModule,
 | 
			
		||||
                    >,
 | 
			
		||||
                    source_location: ::fayalite::source_location::SourceLocation,
 | 
			
		||||
                ) -> <Self as ::fayalite::ty::Type>::MatchVariantsIter
 | 
			
		||||
                where
 | 
			
		||||
                    <IO as ::fayalite::expr::ToExpr>::Type:
 | 
			
		||||
                        ::fayalite::bundle::BundleType<Value = IO>,
 | 
			
		||||
                {
 | 
			
		||||
                    ::fayalite::__std::iter::once(::fayalite::ty::MatchVariantWithoutScope(
 | 
			
		||||
                        #match_variant_ident {
 | 
			
		||||
                            #(#unskipped_field_names: this.field(#unskipped_field_name_strs),)*
 | 
			
		||||
                            #(#phantom_data_field_name_slice:
 | 
			
		||||
                                ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                        },
 | 
			
		||||
                    ))
 | 
			
		||||
                }
 | 
			
		||||
                fn mask_type(&self) -> <Self as ::fayalite::ty::Type>::MaskType {
 | 
			
		||||
                    #mask_type_body
 | 
			
		||||
                }
 | 
			
		||||
                fn canonical(&self) -> <Self as ::fayalite::ty::Type>::CanonicalType {
 | 
			
		||||
                    let fields = ::fayalite::bundle::BundleType::fields(self);
 | 
			
		||||
                    ::fayalite::bundle::DynBundleType::new(fields)
 | 
			
		||||
                }
 | 
			
		||||
                fn source_location(&self) -> ::fayalite::source_location::SourceLocation {
 | 
			
		||||
                    ::fayalite::source_location::SourceLocation::caller()
 | 
			
		||||
                }
 | 
			
		||||
                fn type_enum(&self) -> ::fayalite::ty::TypeEnum {
 | 
			
		||||
                    ::fayalite::ty::TypeEnum::BundleType(::fayalite::ty::Type::canonical(self))
 | 
			
		||||
                }
 | 
			
		||||
                fn from_canonical_type(t: <Self as ::fayalite::ty::Type>::CanonicalType) -> Self {
 | 
			
		||||
                    let [#(#unskipped_field_vars),*] = *::fayalite::bundle::BundleType::fields(&t)
 | 
			
		||||
                    else {
 | 
			
		||||
                        ::fayalite::__std::panic!("wrong number of fields");
 | 
			
		||||
                    };
 | 
			
		||||
                    Self {
 | 
			
		||||
                        #(#unskipped_field_names: #unskipped_field_vars.from_canonical_type_helper(
 | 
			
		||||
                            #unskipped_field_name_strs,
 | 
			
		||||
                            #unskipped_field_flips,
 | 
			
		||||
                        ),)*
 | 
			
		||||
                        #(#phantom_data_field_name_slice: ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::TypeWithDeref for #type_struct_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                #[allow(unused_variables)]
 | 
			
		||||
                fn expr_deref(this: &::fayalite::expr::Expr<<Self as ::fayalite::ty::Type>::Value>)
 | 
			
		||||
                    -> &<Self as ::fayalite::ty::Type>::MatchVariant
 | 
			
		||||
                {
 | 
			
		||||
                    ::fayalite::intern::Interned::<_>::into_inner(
 | 
			
		||||
                        ::fayalite::intern::Intern::intern_sized(#match_variant_ident {
 | 
			
		||||
                            #(#unskipped_field_names: this.field(#unskipped_field_name_strs),)*
 | 
			
		||||
                            #(#phantom_data_field_name_slice:
 | 
			
		||||
                                ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                        }),
 | 
			
		||||
                    )
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::bundle::BundleType for #type_struct_ident #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Builder = #empty_builder_ty;
 | 
			
		||||
                fn builder() -> <Self as ::fayalite::bundle::BundleType>::Builder {
 | 
			
		||||
                    #empty_builder_ty::new()
 | 
			
		||||
                }
 | 
			
		||||
                fn fields(&self) -> ::fayalite::intern::Interned<
 | 
			
		||||
                        [::fayalite::bundle::FieldType<::fayalite::intern::Interned<
 | 
			
		||||
                            dyn ::fayalite::ty::DynCanonicalType,
 | 
			
		||||
                        >>],
 | 
			
		||||
                    >
 | 
			
		||||
                {
 | 
			
		||||
                    ::fayalite::intern::Intern::intern(&[#(
 | 
			
		||||
                        ::fayalite::bundle::FieldType {
 | 
			
		||||
                            name: ::fayalite::intern::Intern::intern(#unskipped_field_name_strs),
 | 
			
		||||
                            flipped: #unskipped_field_flips,
 | 
			
		||||
                            ty: ::fayalite::ty::DynType::canonical_dyn(
 | 
			
		||||
                                &self.#unskipped_field_names,
 | 
			
		||||
                            ),
 | 
			
		||||
                        },
 | 
			
		||||
                    )*][..])
 | 
			
		||||
                }
 | 
			
		||||
                fn fields_hint() -> ::fayalite::bundle::FieldsHint {
 | 
			
		||||
                    ::fayalite::bundle::FieldsHint::new([#(
 | 
			
		||||
                        ::fayalite::bundle::FieldType {
 | 
			
		||||
                            name: ::fayalite::intern::Intern::intern(#unskipped_field_name_strs),
 | 
			
		||||
                            flipped: #unskipped_field_flips,
 | 
			
		||||
                            ty: ::fayalite::bundle::TypeHint::<#field_types>::intern_dyn(),
 | 
			
		||||
                        },
 | 
			
		||||
                    )*], false)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::expr::ToExpr for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                type Type = #type_struct_ident #type_generics;
 | 
			
		||||
                fn ty(&self) -> <Self as ::fayalite::expr::ToExpr>::Type {
 | 
			
		||||
                    #type_struct_ident {
 | 
			
		||||
                        #(#unskipped_field_names: ::fayalite::expr::ToExpr::ty(
 | 
			
		||||
                            &self.#unskipped_field_names,
 | 
			
		||||
                        ),)*
 | 
			
		||||
                        #(#phantom_data_field_name_slice: ::fayalite::__std::marker::PhantomData,)*
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                fn to_expr(&self) -> ::fayalite::expr::Expr<Self> {
 | 
			
		||||
                    ::fayalite::expr::Expr::from_value(self)
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::ty::Value for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
                fn to_canonical(&self) -> <
 | 
			
		||||
                        <Self as ::fayalite::expr::ToExpr>::Type
 | 
			
		||||
                        as ::fayalite::ty::Type
 | 
			
		||||
                    >::CanonicalValue
 | 
			
		||||
                {
 | 
			
		||||
                    let ty = ::fayalite::ty::Type::canonical(&::fayalite::expr::ToExpr::ty(self));
 | 
			
		||||
                    ::fayalite::bundle::DynBundle::new(ty, ::fayalite::__std::sync::Arc::new([
 | 
			
		||||
                        #(::fayalite::ty::DynValueTrait::to_canonical_dyn(
 | 
			
		||||
                            &self.#unskipped_field_names,
 | 
			
		||||
                        ),)*
 | 
			
		||||
                    ]))
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            #[automatically_derived]
 | 
			
		||||
            impl #impl_generics ::fayalite::bundle::BundleValue for #target #type_generics
 | 
			
		||||
            #where_clause
 | 
			
		||||
            {
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        .to_tokens(tokens);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToTokens for ParsedStruct {
 | 
			
		||||
    fn to_tokens(&self, tokens: &mut TokenStream) {
 | 
			
		||||
        let ParsedStructNames {
 | 
			
		||||
            ident: struct_ident,
 | 
			
		||||
            type_struct_debug_ident,
 | 
			
		||||
            type_struct_ident,
 | 
			
		||||
            match_variant_ident,
 | 
			
		||||
            builder_struct_ident,
 | 
			
		||||
            mask_match_variant_ident,
 | 
			
		||||
            mask_type_ident,
 | 
			
		||||
            mask_type_debug_ident,
 | 
			
		||||
            mask_value_ident,
 | 
			
		||||
            mask_value_debug_ident,
 | 
			
		||||
            mask_builder_struct_ident,
 | 
			
		||||
        } = &self.names;
 | 
			
		||||
        macro_rules! unwrap_or_set {
 | 
			
		||||
            ($(let $var:ident =? $fallback_value:expr;)*) => {
 | 
			
		||||
                $(let $var = $var.clone().unwrap_or_else(|| $fallback_value);)*
 | 
			
		||||
            };
 | 
			
		||||
        }
 | 
			
		||||
        unwrap_or_set! {
 | 
			
		||||
            let type_struct_debug_ident =? format!("{struct_ident}::Type");
 | 
			
		||||
            let match_variant_ident =? format_ident!("__{}__MatchVariant", struct_ident);
 | 
			
		||||
            let builder_struct_ident =? format_ident!("__{}__Builder", struct_ident);
 | 
			
		||||
            let mask_match_variant_ident =? format_ident!("__AsMask__{}__MatchVariant", struct_ident);
 | 
			
		||||
            let mask_type_ident =? format_ident!("__AsMask__{}__Type", struct_ident);
 | 
			
		||||
            let mask_type_debug_ident =? format!("AsMask<{struct_ident}>::Type");
 | 
			
		||||
            let mask_value_ident =? format_ident!("__AsMask__{}", struct_ident);
 | 
			
		||||
            let mask_value_debug_ident =? format!("AsMask<{struct_ident}>");
 | 
			
		||||
            let mask_builder_struct_ident =? format_ident!("__AsMask__{}__Builder", struct_ident);
 | 
			
		||||
        }
 | 
			
		||||
        let target = get_target(&self.options.body.target, struct_ident);
 | 
			
		||||
        let names = ParsedStructNames {
 | 
			
		||||
            ident: struct_ident.clone(),
 | 
			
		||||
            type_struct_debug_ident: &type_struct_debug_ident,
 | 
			
		||||
            type_struct_ident: type_struct_ident.clone(),
 | 
			
		||||
            match_variant_ident: &match_variant_ident,
 | 
			
		||||
            builder_struct_ident: &builder_struct_ident,
 | 
			
		||||
            mask_match_variant_ident: &mask_match_variant_ident,
 | 
			
		||||
            mask_type_ident: &mask_type_ident,
 | 
			
		||||
            mask_type_debug_ident: &mask_type_debug_ident,
 | 
			
		||||
            mask_value_ident: &mask_value_ident,
 | 
			
		||||
            mask_value_debug_ident: &mask_value_debug_ident,
 | 
			
		||||
            mask_builder_struct_ident: &mask_builder_struct_ident,
 | 
			
		||||
        };
 | 
			
		||||
        self.write_body(target, names, false, tokens);
 | 
			
		||||
        let mask_names = ParsedStructNames {
 | 
			
		||||
            ident: mask_value_ident.clone(),
 | 
			
		||||
            type_struct_debug_ident: &mask_type_debug_ident,
 | 
			
		||||
            type_struct_ident: mask_type_ident.clone(),
 | 
			
		||||
            match_variant_ident: &mask_match_variant_ident,
 | 
			
		||||
            builder_struct_ident: &mask_builder_struct_ident,
 | 
			
		||||
            mask_match_variant_ident: &mask_match_variant_ident,
 | 
			
		||||
            mask_type_ident: &mask_type_ident,
 | 
			
		||||
            mask_type_debug_ident: &mask_type_debug_ident,
 | 
			
		||||
            mask_value_ident: &mask_value_ident,
 | 
			
		||||
            mask_value_debug_ident: &mask_value_debug_ident,
 | 
			
		||||
            mask_builder_struct_ident: &mask_builder_struct_ident,
 | 
			
		||||
        };
 | 
			
		||||
        self.write_body(mask_value_ident.clone().into(), mask_names, true, tokens);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) fn value_derive_struct(mut item: ItemStruct) -> syn::Result<TokenStream> {
 | 
			
		||||
    let item = ParsedStruct::parse(&mut item)?;
 | 
			
		||||
    let outline_generated = item.options.body.outline_generated;
 | 
			
		||||
    let mut contents = quote! {
 | 
			
		||||
        const _: () = {
 | 
			
		||||
            #item
 | 
			
		||||
        };
 | 
			
		||||
    };
 | 
			
		||||
    if outline_generated.is_some() {
 | 
			
		||||
        contents = crate::outline_generated(contents, "value-struct-");
 | 
			
		||||
    }
 | 
			
		||||
    Ok(contents)
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -2,7 +2,7 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
//! proc macros for `fayalite`
 | 
			
		||||
//!
 | 
			
		||||
//! see `fayalite::hdl_module` and `fayalite::ty::Value` for docs
 | 
			
		||||
//! see `fayalite::hdl_module` and `fayalite::hdl` for docs
 | 
			
		||||
 | 
			
		||||
// intentionally not documented here, see `fayalite::hdl_module` for docs
 | 
			
		||||
#[proc_macro_attribute]
 | 
			
		||||
| 
						 | 
				
			
			@ -10,16 +10,19 @@ pub fn hdl_module(
 | 
			
		|||
    attr: proc_macro::TokenStream,
 | 
			
		||||
    item: proc_macro::TokenStream,
 | 
			
		||||
) -> proc_macro::TokenStream {
 | 
			
		||||
    match fayalite_proc_macros_impl::module(attr.into(), item.into()) {
 | 
			
		||||
    match fayalite_proc_macros_impl::hdl_module(attr.into(), item.into()) {
 | 
			
		||||
        Ok(retval) => retval.into(),
 | 
			
		||||
        Err(err) => err.into_compile_error().into(),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// intentionally not documented here, see `fayalite::ty::Value` for docs
 | 
			
		||||
#[proc_macro_derive(Value, attributes(hdl))]
 | 
			
		||||
pub fn value_derive(item: proc_macro::TokenStream) -> proc_macro::TokenStream {
 | 
			
		||||
    match fayalite_proc_macros_impl::value_derive(item.into()) {
 | 
			
		||||
// intentionally not documented here, see `fayalite::hdl` for docs
 | 
			
		||||
#[proc_macro_attribute]
 | 
			
		||||
pub fn hdl(
 | 
			
		||||
    attr: proc_macro::TokenStream,
 | 
			
		||||
    item: proc_macro::TokenStream,
 | 
			
		||||
) -> proc_macro::TokenStream {
 | 
			
		||||
    match fayalite_proc_macros_impl::hdl_attr(attr.into(), item.into()) {
 | 
			
		||||
        Ok(retval) => retval.into(),
 | 
			
		||||
        Err(err) => err.into_compile_error().into(),
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -4,6 +4,7 @@ use fayalite_visit_gen::parse_and_generate;
 | 
			
		|||
use std::{env, fs, path::Path};
 | 
			
		||||
 | 
			
		||||
fn main() {
 | 
			
		||||
    println!("cargo::rustc-check-cfg=cfg(todo)");
 | 
			
		||||
    let path = "visit_types.json";
 | 
			
		||||
    println!("cargo::rerun-if-changed={path}");
 | 
			
		||||
    println!("cargo::rerun-if-changed=build.rs");
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,11 +1,5 @@
 | 
			
		|||
use clap::Parser;
 | 
			
		||||
use fayalite::{
 | 
			
		||||
    cli,
 | 
			
		||||
    clock::{Clock, ClockDomain},
 | 
			
		||||
    hdl_module,
 | 
			
		||||
    int::{DynUInt, DynUIntType, IntCmp, IntTypeTrait, UInt},
 | 
			
		||||
    reset::{SyncReset, ToReset},
 | 
			
		||||
};
 | 
			
		||||
use fayalite::{cli, prelude::*};
 | 
			
		||||
 | 
			
		||||
#[hdl_module]
 | 
			
		||||
fn blinky(clock_frequency: u64) {
 | 
			
		||||
| 
						 | 
				
			
			@ -19,21 +13,21 @@ fn blinky(clock_frequency: u64) {
 | 
			
		|||
        rst: rst.to_reset(),
 | 
			
		||||
    };
 | 
			
		||||
    let max_value = clock_frequency / 2 - 1;
 | 
			
		||||
    let int_ty = DynUIntType::range_inclusive(0..=max_value);
 | 
			
		||||
    let int_ty = UInt::range_inclusive(0..=max_value);
 | 
			
		||||
    #[hdl]
 | 
			
		||||
    let counter: DynUInt = m.reg_builder().clock_domain(cd).reset(int_ty.literal(0));
 | 
			
		||||
    let counter: UInt = reg_builder().clock_domain(cd).reset(0u8.cast_to(int_ty));
 | 
			
		||||
    #[hdl]
 | 
			
		||||
    let output_reg: UInt<1> = m.reg_builder().clock_domain(cd).reset_default();
 | 
			
		||||
    let output_reg: Bool = reg_builder().clock_domain(cd).reset(false);
 | 
			
		||||
    #[hdl]
 | 
			
		||||
    if counter.cmp_eq(max_value) {
 | 
			
		||||
        m.connect_any(counter, 0u8);
 | 
			
		||||
        m.connect(output_reg, !output_reg);
 | 
			
		||||
        connect_any(counter, 0u8);
 | 
			
		||||
        connect(output_reg, !output_reg);
 | 
			
		||||
    } else {
 | 
			
		||||
        m.connect_any(counter, counter + 1_hdl_u1);
 | 
			
		||||
        connect_any(counter, counter + 1_hdl_u1);
 | 
			
		||||
    }
 | 
			
		||||
    #[hdl]
 | 
			
		||||
    let led: UInt<1> = m.output();
 | 
			
		||||
    m.connect(led, output_reg);
 | 
			
		||||
    let led: Bool = m.output();
 | 
			
		||||
    connect(led, output_reg);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Parser)]
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -10,7 +10,7 @@
 | 
			
		|||
//! function to add inputs/outputs and other components to that module.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! #
 | 
			
		||||
//! #[hdl_module]
 | 
			
		||||
//! pub fn example_module() {
 | 
			
		||||
| 
						 | 
				
			
			@ -18,7 +18,7 @@
 | 
			
		|||
//!     let an_input: UInt<10> = m.input(); // create an input that is a 10-bit unsigned integer
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     let some_output: UInt<10> = m.output();
 | 
			
		||||
//!     m.connect(some_output, an_input); // assigns the value of `an_input` to `some_output`
 | 
			
		||||
//!     connect(some_output, an_input); // assigns the value of `an_input` to `some_output`
 | 
			
		||||
//! }
 | 
			
		||||
//! ```
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -11,8 +11,6 @@
 | 
			
		|||
//!  * [`parameter_raw_verilog()`][`ModuleBuilder::parameter_raw_verilog`]
 | 
			
		||||
//!  * [`parameter()`][`ModuleBuilder::parameter`]
 | 
			
		||||
//!
 | 
			
		||||
//! These use the [`ExternModule`][`crate::module::ExternModule`] tag type.
 | 
			
		||||
//!
 | 
			
		||||
//! [inputs/outputs]: crate::_docs::modules::module_bodies::hdl_let_statements::inputs_outputs
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,24 +1,24 @@
 | 
			
		|||
//! # `#[hdl]` Array Expressions
 | 
			
		||||
//!
 | 
			
		||||
//! `#[hdl]` can be used on Array Expressions to construct an [`Array<[T; N]>`][Array] expression:
 | 
			
		||||
//! `#[hdl]` can be used on Array Expressions to construct an [`Array<[T; N]>`][type@Array] expression:
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, array::Array};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let v: UInt<8> = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let w: Array<[UInt<8>; 4]> = m.wire();
 | 
			
		||||
//! m.connect(
 | 
			
		||||
//! let w: Array<UInt<8>, 4> = wire();
 | 
			
		||||
//! connect(
 | 
			
		||||
//!     w,
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     [4_hdl_u8, v, 3_hdl_u8, (v + 7_hdl_u8).cast()] // you can make an array like this
 | 
			
		||||
//!     [4_hdl_u8, v, 3_hdl_u8, (v + 7_hdl_u8).cast_to_static()] // you can make an array like this
 | 
			
		||||
//! );
 | 
			
		||||
//! m.connect(
 | 
			
		||||
//! connect(
 | 
			
		||||
//!     w,
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     [(v + 1_hdl_u8).cast(); 4] // or you can make an array repeat like this
 | 
			
		||||
//!     [(v + 1_hdl_u8).cast_to_static(); 4] // or you can make an array repeat like this
 | 
			
		||||
//! );
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -3,8 +3,7 @@
 | 
			
		|||
//! `#[hdl] if` statements behave similarly to Rust `if` statements, except they end up as muxes
 | 
			
		||||
//! and stuff in the final hardware instead of being run when the fayalite module is being created.
 | 
			
		||||
//!
 | 
			
		||||
//! The condition of an `#[hdl] if` statement must have type [`UInt<1>`] or [`DynUInt`] with
 | 
			
		||||
//! `width() == 1` or be an [expression][Expr] of one of those types.
 | 
			
		||||
//! The condition of an `#[hdl] if` statement must have type [`Expr<Bool>`][Bool].
 | 
			
		||||
//!
 | 
			
		||||
//! `#[hdl] if` statements' bodies must evaluate to type `()` for now.
 | 
			
		||||
//!
 | 
			
		||||
| 
						 | 
				
			
			@ -14,7 +13,4 @@
 | 
			
		|||
//! [match]: super::hdl_match_statements
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
use crate::{
 | 
			
		||||
    expr::Expr,
 | 
			
		||||
    int::{DynUInt, UInt},
 | 
			
		||||
};
 | 
			
		||||
use crate::int::Bool;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -6,14 +6,14 @@
 | 
			
		|||
//! so you should read it.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, expr::Expr, array::Array};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_input: UInt<10> = m.input();
 | 
			
		||||
//! let _: Expr<UInt<10>> = my_input; // my_input has type Expr<UInt<10>>
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_output: Array<[UInt<10>; 3]> = m.output();
 | 
			
		||||
//! let my_output: Array<UInt<10>, 3> = m.output();
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -4,15 +4,15 @@
 | 
			
		|||
//! you can create them like so:
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, array::Array};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_instance = m.instance(some_module());
 | 
			
		||||
//! let my_instance = instance(some_module());
 | 
			
		||||
//! // now you can use `my_instance`'s inputs/outputs like so:
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let v: UInt<3> = m.input();
 | 
			
		||||
//! m.connect(my_instance.a, v);
 | 
			
		||||
//! connect(my_instance.a, v);
 | 
			
		||||
//! #[hdl_module]
 | 
			
		||||
//! fn some_module() {
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -7,12 +7,12 @@
 | 
			
		|||
//!
 | 
			
		||||
//! There are several different ways to create a memory:
 | 
			
		||||
//!
 | 
			
		||||
//! ## using [`ModuleBuilder::memory()`]
 | 
			
		||||
//! ## using [`memory()`]
 | 
			
		||||
//!
 | 
			
		||||
//! This way you have to set the [`depth`][`MemBuilder::depth`] separately.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, clock::ClockDomain};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! // first, we need some IO
 | 
			
		||||
| 
						 | 
				
			
			@ -25,45 +25,45 @@
 | 
			
		|||
//!
 | 
			
		||||
//! // now create the memory
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let mut my_memory = m.memory();
 | 
			
		||||
//! let mut my_memory = memory();
 | 
			
		||||
//! my_memory.depth(256); // the memory has 256 elements
 | 
			
		||||
//!
 | 
			
		||||
//! let read_port = my_memory.new_read_port();
 | 
			
		||||
//!
 | 
			
		||||
//! // connect up the read port
 | 
			
		||||
//! m.connect_any(read_port.addr, read_addr);
 | 
			
		||||
//! m.connect(read_port.en, 1_hdl_u1);
 | 
			
		||||
//! m.connect(read_port.clk, cd.clk);
 | 
			
		||||
//! m.connect(read_data, read_port.data);
 | 
			
		||||
//! connect_any(read_port.addr, read_addr);
 | 
			
		||||
//! connect(read_port.en, true);
 | 
			
		||||
//! connect(read_port.clk, cd.clk);
 | 
			
		||||
//! connect(read_data, read_port.data);
 | 
			
		||||
//!
 | 
			
		||||
//! // we need more IO for the write port
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let write_addr: UInt<8> = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let do_write: UInt<1> = m.input();
 | 
			
		||||
//! let do_write: Bool = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let write_data: UInt<8> = m.input();
 | 
			
		||||
//!
 | 
			
		||||
//! let write_port = my_memory.new_write_port();
 | 
			
		||||
//!
 | 
			
		||||
//! m.connect_any(write_port.addr, write_addr);
 | 
			
		||||
//! m.connect(write_port.en, do_write);
 | 
			
		||||
//! m.connect(write_port.clk, cd.clk);
 | 
			
		||||
//! m.connect(write_port.data, write_port.data);
 | 
			
		||||
//! m.connect(write_port.mask, 1_hdl_u1);
 | 
			
		||||
//! connect_any(write_port.addr, write_addr);
 | 
			
		||||
//! connect(write_port.en, do_write);
 | 
			
		||||
//! connect(write_port.clk, cd.clk);
 | 
			
		||||
//! connect(write_port.data, write_port.data);
 | 
			
		||||
//! connect(write_port.mask, true);
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
//! ## using [`ModuleBuilder::memory_array()`]
 | 
			
		||||
//! ## using [`memory_array()`]
 | 
			
		||||
//!
 | 
			
		||||
//! this allows you to specify the memory's underlying array type directly.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, memory::MemBuilder};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let mut my_memory: MemBuilder<[UInt<8>; 256]> = m.memory_array();
 | 
			
		||||
//! let mut my_memory: MemBuilder<UInt<8>, ConstUsize<256>> = memory_array();
 | 
			
		||||
//!
 | 
			
		||||
//! let read_port = my_memory.new_read_port();
 | 
			
		||||
//! // ...
 | 
			
		||||
| 
						 | 
				
			
			@ -72,25 +72,22 @@
 | 
			
		|||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
//! ## using [`ModuleBuilder::memory_with_init()`]
 | 
			
		||||
//! ## using [`memory_with_init()`]
 | 
			
		||||
//!
 | 
			
		||||
//! This allows you to deduce the memory's array type from the data used to initialize the memory.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! # #[hdl]
 | 
			
		||||
//! # let read_addr: UInt<2> = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let mut my_memory = m.memory_with_init(
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     [0x12_hdl_u8, 0x34_hdl_u8, 0x56_hdl_u8, 0x78_hdl_u8],
 | 
			
		||||
//! );
 | 
			
		||||
//! let mut my_memory = memory_with_init([0x12_hdl_u8, 0x34_hdl_u8, 0x56_hdl_u8, 0x78_hdl_u8]);
 | 
			
		||||
//!
 | 
			
		||||
//! let read_port = my_memory.new_read_port();
 | 
			
		||||
//! // note that `read_addr` is `UInt<2>` since the memory only has 4 elements
 | 
			
		||||
//! m.connect_any(read_port.addr, read_addr);
 | 
			
		||||
//! connect_any(read_port.addr, read_addr);
 | 
			
		||||
//! // ...
 | 
			
		||||
//! let write_port = my_memory.new_write_port();
 | 
			
		||||
//! // ...
 | 
			
		||||
| 
						 | 
				
			
			@ -98,4 +95,4 @@
 | 
			
		|||
//! ```
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
use crate::{memory::MemBuilder, module::ModuleBuilder};
 | 
			
		||||
use crate::prelude::*;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -8,19 +8,19 @@
 | 
			
		|||
//! Registers follow [connection semantics], which are unlike assignments in software, so you should read it.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, array::Array, clock::ClockDomain};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! # #[hdl]
 | 
			
		||||
//! # let v: UInt<1> = m.input();
 | 
			
		||||
//! # let v: Bool = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let cd: ClockDomain = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_register: UInt<8> = m.reg_builder().clock_domain(cd).reset(8_hdl_u8);
 | 
			
		||||
//! let my_register: UInt<8> = reg_builder().clock_domain(cd).reset(8_hdl_u8);
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! if v {
 | 
			
		||||
//!     // my_register is only changed when both `v` is set and `cd`'s clock edge occurs.
 | 
			
		||||
//!     m.connect(my_register, 0x45_hdl_u8);
 | 
			
		||||
//!     connect(my_register, 0x45_hdl_u8);
 | 
			
		||||
//! }
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,26 +2,26 @@
 | 
			
		|||
//!
 | 
			
		||||
//! Wires are kinda like variables, but unlike registers,
 | 
			
		||||
//! they have no memory (they're combinatorial).
 | 
			
		||||
//! You must [connect][`ModuleBuilder::connect`] to all wires, so they have a defined value.
 | 
			
		||||
//! You must [connect] to all wires, so they have a defined value.
 | 
			
		||||
//!
 | 
			
		||||
//! Wires create a Rust variable with type [`Expr<T>`] where `T` is the type of the wire.
 | 
			
		||||
//!
 | 
			
		||||
//! Wires follow [connection semantics], which are unlike assignments in software, so you should read it.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, array::Array, clock::ClockDomain};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! # #[hdl]
 | 
			
		||||
//! # let v: UInt<1> = m.input();
 | 
			
		||||
//! # let v: Bool = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_wire: UInt<8> = m.wire();
 | 
			
		||||
//! let my_wire: UInt<8> = wire();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! if v {
 | 
			
		||||
//!     m.connect(my_wire, 0x45_hdl_u8);
 | 
			
		||||
//!     connect(my_wire, 0x45_hdl_u8);
 | 
			
		||||
//! } else {
 | 
			
		||||
//!     // wires must be connected to under all conditions
 | 
			
		||||
//!     m.connect(my_wire, 0x23_hdl_u8);
 | 
			
		||||
//!     connect(my_wire, 0x23_hdl_u8);
 | 
			
		||||
//! }
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
| 
						 | 
				
			
			@ -29,4 +29,4 @@
 | 
			
		|||
//! [connection semantics]: crate::_docs::semantics::connection_semantics
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
use crate::{expr::Expr, module::ModuleBuilder};
 | 
			
		||||
use crate::prelude::*;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,8 +2,7 @@
 | 
			
		|||
//!
 | 
			
		||||
//! You can have integer literals with an arbitrary number of bits like so:
 | 
			
		||||
//!
 | 
			
		||||
//! `_hdl`-suffixed literals have type [`Expr<UInt<N>>`] or [`Expr<SInt<N>>`]
 | 
			
		||||
//! ... which are basically just [`UInt<N>`] or [`SInt<N>`] converted to an expression.
 | 
			
		||||
//! `_hdl`-suffixed literals have type [`Expr<UInt<N>>`] or [`Expr<SInt<N>>`].
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # #[fayalite::hdl_module]
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,27 +2,24 @@
 | 
			
		|||
//!
 | 
			
		||||
//! Note: Structs are also known as [Bundles] when used in Fayalite, the Bundle name comes from [FIRRTL].
 | 
			
		||||
//!
 | 
			
		||||
//! [Bundles]: crate::bundle::BundleValue
 | 
			
		||||
//! [Bundles]: crate::bundle::BundleType
 | 
			
		||||
//! [FIRRTL]: https://github.com/chipsalliance/firrtl-spec
 | 
			
		||||
//!
 | 
			
		||||
//! `#[hdl]` can be used on Struct/Variant Expressions to construct a value of that
 | 
			
		||||
//! struct/variant's type. They can also be used on tuples.
 | 
			
		||||
//! `#[hdl]` can be used on Struct Expressions to construct a value of that
 | 
			
		||||
//! struct's type. They can also be used on tuples.
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt, array::Array, ty::Value};
 | 
			
		||||
//! #[derive(Value, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
//! #[hdl(static)]
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! pub struct MyStruct {
 | 
			
		||||
//!     pub a: UInt<8>,
 | 
			
		||||
//!     pub b: UInt<16>,
 | 
			
		||||
//! }
 | 
			
		||||
//!
 | 
			
		||||
//! #[derive(Value, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! pub enum MyEnum {
 | 
			
		||||
//!     A,
 | 
			
		||||
//!     B {
 | 
			
		||||
//!         v: UInt<32>,
 | 
			
		||||
//!     },
 | 
			
		||||
//!     B(UInt<32>),
 | 
			
		||||
//! }
 | 
			
		||||
//!
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
| 
						 | 
				
			
			@ -30,8 +27,8 @@
 | 
			
		|||
//! #[hdl]
 | 
			
		||||
//! let v: UInt<8> = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_struct: MyStruct = m.wire();
 | 
			
		||||
//! m.connect(
 | 
			
		||||
//! let my_struct: MyStruct = wire();
 | 
			
		||||
//! connect(
 | 
			
		||||
//!     my_struct,
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     MyStruct {
 | 
			
		||||
| 
						 | 
				
			
			@ -40,15 +37,14 @@
 | 
			
		|||
//!     },
 | 
			
		||||
//! );
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let my_enum: MyEnum = m.wire();
 | 
			
		||||
//! m.connect(
 | 
			
		||||
//! let my_enum: MyEnum = wire();
 | 
			
		||||
//! connect(
 | 
			
		||||
//!     my_enum,
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     MyEnum::B { v: 12345678_hdl_u32 },
 | 
			
		||||
//!     MyEnum.B(12345678_hdl_u32),
 | 
			
		||||
//! );
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let some_tuple: (UInt<4>, UInt<12>) = m.wire();
 | 
			
		||||
//! m.connect(
 | 
			
		||||
//! let some_tuple: (UInt<4>, UInt<12>) = wire();
 | 
			
		||||
//! connect(
 | 
			
		||||
//!     some_tuple,
 | 
			
		||||
//!     #[hdl]
 | 
			
		||||
//!     (12_hdl_u4, 3421_hdl_u12),
 | 
			
		||||
| 
						 | 
				
			
			@ -57,4 +53,4 @@
 | 
			
		|||
//! ```
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
use crate::array::Array;
 | 
			
		||||
use crate::prelude::*;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,6 +1,4 @@
 | 
			
		|||
//! # Normal Modules
 | 
			
		||||
//!
 | 
			
		||||
//! These use the [`NormalModule`][`crate::module::NormalModule`] tag type.
 | 
			
		||||
//!
 | 
			
		||||
//! See also: [Extern Modules][`super::extern_module`]
 | 
			
		||||
//! See also: [Module Bodies][`super::module_bodies`]
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -20,62 +20,60 @@
 | 
			
		|||
//! Connection Semantics Example:
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let a: UInt<8> = m.wire();
 | 
			
		||||
//! let a: UInt<8> = wire();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let b: UInt<8> = m.output();
 | 
			
		||||
//!
 | 
			
		||||
//! // doesn't actually affect anything, since `a` is completely overwritten later
 | 
			
		||||
//! m.connect(a, 5_hdl_u8);
 | 
			
		||||
//! connect(a, 5_hdl_u8);
 | 
			
		||||
//!
 | 
			
		||||
//! // here `a` has value `7` since the last connection assigns
 | 
			
		||||
//! // `7` to `a`, so `b` has value `7` too.
 | 
			
		||||
//! m.connect(b, a);
 | 
			
		||||
//! connect(b, a);
 | 
			
		||||
//!
 | 
			
		||||
//! // this is the last `connect` to `a`, so this `connect` determines `a`'s value
 | 
			
		||||
//! m.connect(a, 7_hdl_u8);
 | 
			
		||||
//! connect(a, 7_hdl_u8);
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
//! # Conditional Connection Semantics
 | 
			
		||||
//!
 | 
			
		||||
//! ```
 | 
			
		||||
//! # use fayalite::{hdl_module, int::UInt};
 | 
			
		||||
//! # use fayalite::prelude::*;
 | 
			
		||||
//! # #[hdl_module]
 | 
			
		||||
//! # fn module() {
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let cond: UInt<1> = m.input();
 | 
			
		||||
//! let cond: Bool = m.input();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let a: UInt<8> = m.wire();
 | 
			
		||||
//! let a: UInt<8> = wire();
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! let b: UInt<8> = m.output();
 | 
			
		||||
//!
 | 
			
		||||
//! // this is the last `connect` to `a` when `cond` is `0`
 | 
			
		||||
//! m.connect(a, 5_hdl_u8);
 | 
			
		||||
//! connect(a, 5_hdl_u8);
 | 
			
		||||
//!
 | 
			
		||||
//! // here `a` has value `7` if `cond` is `1` since the last connection assigns
 | 
			
		||||
//! // `7` to `a`, so `b` has value `7` too, otherwise `a` (and therefore `b`)
 | 
			
		||||
//! // have value `5` since then the connection assigning `7` is in a
 | 
			
		||||
//! // conditional block where the condition doesn't hold.
 | 
			
		||||
//! m.connect(b, a);
 | 
			
		||||
//! connect(b, a);
 | 
			
		||||
//!
 | 
			
		||||
//! #[hdl]
 | 
			
		||||
//! if cond {
 | 
			
		||||
//!     // this is the last `connect` to `a` when `cond` is `1`
 | 
			
		||||
//!     m.connect(a, 7_hdl_u8);
 | 
			
		||||
//!     connect(a, 7_hdl_u8);
 | 
			
		||||
//! }
 | 
			
		||||
//! # }
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
//! [conditional block]: self#conditional-connection-semantics
 | 
			
		||||
//! [`connect()`]: ModuleBuilder::connect
 | 
			
		||||
//! [`connect_any()`]: ModuleBuilder::connect_any
 | 
			
		||||
//! [wire]: crate::_docs::modules::module_bodies::hdl_let_statements::wires
 | 
			
		||||
//! [if]: crate::_docs::modules::module_bodies::hdl_if_statements
 | 
			
		||||
//! [FIRRTL]: https://github.com/chipsalliance/firrtl-spec
 | 
			
		||||
 | 
			
		||||
#[allow(unused)]
 | 
			
		||||
use crate::module::ModuleBuilder;
 | 
			
		||||
use crate::prelude::*;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,7 +1,7 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    expr::Target,
 | 
			
		||||
    expr::target::Target,
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
};
 | 
			
		||||
use serde::{Deserialize, Serialize};
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,671 +1,234 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
 | 
			
		||||
use crate::{
 | 
			
		||||
    bundle::{BundleType, BundleValue},
 | 
			
		||||
    expr::{
 | 
			
		||||
        ops::{ArrayIndex, ArrayLiteral, ExprIndex},
 | 
			
		||||
        Expr, ToExpr,
 | 
			
		||||
    },
 | 
			
		||||
    intern::{Intern, Interned, InternedCompare, Memoize},
 | 
			
		||||
    module::{
 | 
			
		||||
        transform::visit::{Fold, Folder, Visit, Visitor},
 | 
			
		||||
        ModuleBuilder, NormalModule,
 | 
			
		||||
    },
 | 
			
		||||
    expr::{ops::ArrayIndex, Expr, ToExpr},
 | 
			
		||||
    int::{DynSize, KnownSize, Size},
 | 
			
		||||
    intern::{Intern, Interned, LazyInterned},
 | 
			
		||||
    module::transform::visit::{Fold, Folder, Visit, Visitor},
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{
 | 
			
		||||
        CanonicalType, CanonicalTypeKind, CanonicalValue, Connect, DynCanonicalType,
 | 
			
		||||
        DynCanonicalValue, DynType, DynValueTrait, MatchVariantWithoutScope, StaticType,
 | 
			
		||||
        StaticValue, Type, TypeEnum, Value, ValueEnum,
 | 
			
		||||
        CanonicalType, MatchVariantWithoutScope, StaticType, Type, TypeProperties, TypeWithDeref,
 | 
			
		||||
    },
 | 
			
		||||
    util::{ConstBool, GenericConstBool, MakeMutSlice},
 | 
			
		||||
};
 | 
			
		||||
use bitvec::{slice::BitSlice, vec::BitVec};
 | 
			
		||||
use std::{
 | 
			
		||||
    any::Any,
 | 
			
		||||
    borrow::{Borrow, BorrowMut},
 | 
			
		||||
    fmt,
 | 
			
		||||
    hash::Hash,
 | 
			
		||||
    marker::PhantomData,
 | 
			
		||||
    ops::IndexMut,
 | 
			
		||||
    sync::Arc,
 | 
			
		||||
    util::ConstUsize,
 | 
			
		||||
};
 | 
			
		||||
use std::ops::Index;
 | 
			
		||||
 | 
			
		||||
mod sealed {
 | 
			
		||||
    pub trait Sealed {}
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct ArrayType<T: Type = CanonicalType, Len: Size = DynSize> {
 | 
			
		||||
    element: LazyInterned<T>,
 | 
			
		||||
    len: Len::SizeType,
 | 
			
		||||
    type_properties: TypeProperties,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait ValueArrayOrSlice:
 | 
			
		||||
    sealed::Sealed
 | 
			
		||||
    + BorrowMut<[<Self as ValueArrayOrSlice>::Element]>
 | 
			
		||||
    + AsRef<[<Self as ValueArrayOrSlice>::Element]>
 | 
			
		||||
    + AsMut<[<Self as ValueArrayOrSlice>::Element]>
 | 
			
		||||
    + Hash
 | 
			
		||||
    + fmt::Debug
 | 
			
		||||
    + Eq
 | 
			
		||||
    + Send
 | 
			
		||||
    + Sync
 | 
			
		||||
    + 'static
 | 
			
		||||
    + IndexMut<usize, Output = <Self as ValueArrayOrSlice>::Element>
 | 
			
		||||
    + ToOwned
 | 
			
		||||
    + InternedCompare
 | 
			
		||||
{
 | 
			
		||||
    type Element: Value<Type = <Self as ValueArrayOrSlice>::ElementType>;
 | 
			
		||||
    type ElementType: Type<Value = <Self as ValueArrayOrSlice>::Element>;
 | 
			
		||||
    type LenType: 'static + Copy + Ord + fmt::Debug + Hash + Send + Sync;
 | 
			
		||||
    type Match: 'static
 | 
			
		||||
        + Clone
 | 
			
		||||
        + Eq
 | 
			
		||||
        + fmt::Debug
 | 
			
		||||
        + Hash
 | 
			
		||||
        + Send
 | 
			
		||||
        + Sync
 | 
			
		||||
        + BorrowMut<[Expr<Self::Element>]>;
 | 
			
		||||
    type MaskVA: ValueArrayOrSlice<
 | 
			
		||||
            Element = <Self::ElementType as Type>::MaskValue,
 | 
			
		||||
            ElementType = <Self::ElementType as Type>::MaskType,
 | 
			
		||||
            LenType = Self::LenType,
 | 
			
		||||
            MaskVA = Self::MaskVA,
 | 
			
		||||
        > + ?Sized;
 | 
			
		||||
    type IsStaticLen: GenericConstBool;
 | 
			
		||||
    const FIXED_LEN_TYPE: Option<Self::LenType>;
 | 
			
		||||
    fn make_match(array: Expr<Array<Self>>) -> Self::Match;
 | 
			
		||||
    fn len_from_len_type(v: Self::LenType) -> usize;
 | 
			
		||||
    #[allow(clippy::result_unit_err)]
 | 
			
		||||
    fn try_len_type_from_len(v: usize) -> Result<Self::LenType, ()>;
 | 
			
		||||
    fn len_type(&self) -> Self::LenType;
 | 
			
		||||
    fn len(&self) -> usize;
 | 
			
		||||
    fn is_empty(&self) -> bool;
 | 
			
		||||
    fn iter(&self) -> std::slice::Iter<Self::Element> {
 | 
			
		||||
        Borrow::<[_]>::borrow(self).iter()
 | 
			
		||||
    }
 | 
			
		||||
    fn clone_to_arc(&self) -> Arc<Self>;
 | 
			
		||||
    fn arc_make_mut(v: &mut Arc<Self>) -> &mut Self;
 | 
			
		||||
    fn arc_to_arc_slice(self: Arc<Self>) -> Arc<[Self::Element]>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T> sealed::Sealed for [T] {}
 | 
			
		||||
 | 
			
		||||
impl<V: Value> ValueArrayOrSlice for [V]
 | 
			
		||||
where
 | 
			
		||||
    V::Type: Type<Value = V>,
 | 
			
		||||
{
 | 
			
		||||
    type Element = V;
 | 
			
		||||
    type ElementType = V::Type;
 | 
			
		||||
    type LenType = usize;
 | 
			
		||||
    type Match = Box<[Expr<V>]>;
 | 
			
		||||
    type MaskVA = [<Self::ElementType as Type>::MaskValue];
 | 
			
		||||
    type IsStaticLen = ConstBool<false>;
 | 
			
		||||
    const FIXED_LEN_TYPE: Option<Self::LenType> = None;
 | 
			
		||||
 | 
			
		||||
    fn make_match(array: Expr<Array<Self>>) -> Self::Match {
 | 
			
		||||
        (0..array.canonical_type().len())
 | 
			
		||||
            .map(|index| ArrayIndex::<V::Type>::new_unchecked(array.canonical(), index).to_expr())
 | 
			
		||||
            .collect()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn len_from_len_type(v: Self::LenType) -> usize {
 | 
			
		||||
        v
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn try_len_type_from_len(v: usize) -> Result<Self::LenType, ()> {
 | 
			
		||||
        Ok(v)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn len_type(&self) -> Self::LenType {
 | 
			
		||||
        self.len()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn len(&self) -> usize {
 | 
			
		||||
        <[_]>::len(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn is_empty(&self) -> bool {
 | 
			
		||||
        <[_]>::is_empty(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn clone_to_arc(&self) -> Arc<Self> {
 | 
			
		||||
        Arc::from(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn arc_make_mut(v: &mut Arc<Self>) -> &mut Self {
 | 
			
		||||
        MakeMutSlice::make_mut_slice(v)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn arc_to_arc_slice(self: Arc<Self>) -> Arc<[Self::Element]> {
 | 
			
		||||
        self
 | 
			
		||||
impl<T: Type, Len: Size> std::fmt::Debug for ArrayType<T, Len> {
 | 
			
		||||
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 | 
			
		||||
        write!(f, "Array<{:?}, {}>", self.element, self.len())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T, const N: usize> sealed::Sealed for [T; N] {}
 | 
			
		||||
pub type Array<
 | 
			
		||||
    T = CanonicalType,
 | 
			
		||||
    const LEN: usize = { <DynSize as crate::util::GenericConstUsize>::VALUE },
 | 
			
		||||
> = ArrayType<T, ConstUsize<LEN>>;
 | 
			
		||||
 | 
			
		||||
impl<V: Value, const N: usize> ValueArrayOrSlice for [V; N]
 | 
			
		||||
where
 | 
			
		||||
    V::Type: Type<Value = V>,
 | 
			
		||||
{
 | 
			
		||||
    type Element = V;
 | 
			
		||||
    type ElementType = V::Type;
 | 
			
		||||
    type LenType = ();
 | 
			
		||||
    type Match = [Expr<V>; N];
 | 
			
		||||
    type MaskVA = [<Self::ElementType as Type>::MaskValue; N];
 | 
			
		||||
    type IsStaticLen = ConstBool<true>;
 | 
			
		||||
    const FIXED_LEN_TYPE: Option<Self::LenType> = Some(());
 | 
			
		||||
#[allow(non_upper_case_globals)]
 | 
			
		||||
pub const Array: ArrayWithoutGenerics = ArrayWithoutGenerics;
 | 
			
		||||
#[allow(non_upper_case_globals)]
 | 
			
		||||
pub const ArrayType: ArrayWithoutGenerics = ArrayWithoutGenerics;
 | 
			
		||||
 | 
			
		||||
    fn make_match(array: Expr<Array<Self>>) -> Self::Match {
 | 
			
		||||
        std::array::from_fn(|index| {
 | 
			
		||||
            ArrayIndex::<V::Type>::new_unchecked(array.canonical(), index).to_expr()
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn len_from_len_type(_v: Self::LenType) -> usize {
 | 
			
		||||
        N
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn try_len_type_from_len(v: usize) -> Result<Self::LenType, ()> {
 | 
			
		||||
        if v == N {
 | 
			
		||||
            Ok(())
 | 
			
		||||
        } else {
 | 
			
		||||
            Err(())
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn len_type(&self) -> Self::LenType {}
 | 
			
		||||
 | 
			
		||||
    fn len(&self) -> usize {
 | 
			
		||||
        N
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn is_empty(&self) -> bool {
 | 
			
		||||
        N == 0
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn clone_to_arc(&self) -> Arc<Self> {
 | 
			
		||||
        Arc::new(self.clone())
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn arc_make_mut(v: &mut Arc<Self>) -> &mut Self {
 | 
			
		||||
        Arc::make_mut(v)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn arc_to_arc_slice(self: Arc<Self>) -> Arc<[Self::Element]> {
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct ArrayType<VA: ValueArrayOrSlice + ?Sized> {
 | 
			
		||||
    element: VA::ElementType,
 | 
			
		||||
    len: VA::LenType,
 | 
			
		||||
    bit_width: usize,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait ArrayTypeTrait:
 | 
			
		||||
    Type<
 | 
			
		||||
        CanonicalType = ArrayType<[DynCanonicalValue]>,
 | 
			
		||||
        Value = Array<<Self as ArrayTypeTrait>::ValueArrayOrSlice>,
 | 
			
		||||
        CanonicalValue = Array<[DynCanonicalValue]>,
 | 
			
		||||
        MaskType = ArrayType<
 | 
			
		||||
            <<Self as ArrayTypeTrait>::ValueArrayOrSlice as ValueArrayOrSlice>::MaskVA,
 | 
			
		||||
        >,
 | 
			
		||||
    > + From<ArrayType<<Self as ArrayTypeTrait>::ValueArrayOrSlice>>
 | 
			
		||||
    + Into<ArrayType<<Self as ArrayTypeTrait>::ValueArrayOrSlice>>
 | 
			
		||||
    + BorrowMut<ArrayType<<Self as ArrayTypeTrait>::ValueArrayOrSlice>>
 | 
			
		||||
    + sealed::Sealed
 | 
			
		||||
    + Connect<Self>
 | 
			
		||||
{
 | 
			
		||||
    type ValueArrayOrSlice: ValueArrayOrSlice<Element = Self::Element, ElementType = Self::ElementType>
 | 
			
		||||
        + ?Sized;
 | 
			
		||||
    type Element: Value<Type = Self::ElementType>;
 | 
			
		||||
    type ElementType: Type<Value = Self::Element>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> sealed::Sealed for ArrayType<VA> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> ArrayTypeTrait for ArrayType<VA> {
 | 
			
		||||
    type ValueArrayOrSlice = VA;
 | 
			
		||||
    type Element = VA::Element;
 | 
			
		||||
    type ElementType = VA::ElementType;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Clone for ArrayType<VA> {
 | 
			
		||||
    fn clone(&self) -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            element: self.element.clone(),
 | 
			
		||||
            len: self.len,
 | 
			
		||||
            bit_width: self.bit_width,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Copy for ArrayType<VA> where VA::ElementType: Copy {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> ArrayType<VA> {
 | 
			
		||||
    pub fn element(&self) -> &VA::ElementType {
 | 
			
		||||
        &self.element
 | 
			
		||||
    }
 | 
			
		||||
    pub fn len(&self) -> usize {
 | 
			
		||||
        VA::len_from_len_type(self.len)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_empty(&self) -> bool {
 | 
			
		||||
        self.len() == 0
 | 
			
		||||
    }
 | 
			
		||||
    pub fn bit_width(&self) -> usize {
 | 
			
		||||
        self.bit_width
 | 
			
		||||
    }
 | 
			
		||||
    pub fn into_slice_type(self) -> ArrayType<[VA::Element]> {
 | 
			
		||||
        ArrayType {
 | 
			
		||||
            len: self.len(),
 | 
			
		||||
            element: self.element,
 | 
			
		||||
            bit_width: self.bit_width,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_with_len(element: VA::ElementType, len: usize) -> Self {
 | 
			
		||||
        Self::new_with_len_type(
 | 
			
		||||
            element,
 | 
			
		||||
            VA::try_len_type_from_len(len).expect("length should match"),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_with_len_type(element: VA::ElementType, len: VA::LenType) -> Self {
 | 
			
		||||
        let Some(bit_width) = VA::len_from_len_type(len).checked_mul(element.bit_width()) else {
 | 
			
		||||
            panic!("array is too big: bit-width overflowed");
 | 
			
		||||
impl<T: Type, Len: Size> ArrayType<T, Len> {
 | 
			
		||||
    const fn make_type_properties(element: TypeProperties, len: usize) -> TypeProperties {
 | 
			
		||||
        let TypeProperties {
 | 
			
		||||
            is_passive,
 | 
			
		||||
            is_storable,
 | 
			
		||||
            is_castable_from_bits,
 | 
			
		||||
            bit_width,
 | 
			
		||||
        } = element;
 | 
			
		||||
        let Some(bit_width) = bit_width.checked_mul(len) else {
 | 
			
		||||
            panic!("array too big");
 | 
			
		||||
        };
 | 
			
		||||
        ArrayType {
 | 
			
		||||
            element,
 | 
			
		||||
            len,
 | 
			
		||||
        TypeProperties {
 | 
			
		||||
            is_passive,
 | 
			
		||||
            is_storable,
 | 
			
		||||
            is_castable_from_bits,
 | 
			
		||||
            bit_width,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn new(element: T, len: Len::SizeType) -> Self {
 | 
			
		||||
        let type_properties =
 | 
			
		||||
            Self::make_type_properties(element.canonical().type_properties(), Len::as_usize(len));
 | 
			
		||||
        Self {
 | 
			
		||||
            element: LazyInterned::Interned(element.intern_sized()),
 | 
			
		||||
            len,
 | 
			
		||||
            type_properties,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn element(&self) -> T {
 | 
			
		||||
        *self.element
 | 
			
		||||
    }
 | 
			
		||||
    pub fn len(self) -> usize {
 | 
			
		||||
        Len::as_usize(self.len)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_empty(self) -> bool {
 | 
			
		||||
        self.len() == 0
 | 
			
		||||
    }
 | 
			
		||||
    pub fn type_properties(self) -> TypeProperties {
 | 
			
		||||
        self.type_properties
 | 
			
		||||
    }
 | 
			
		||||
    pub fn as_dyn_array(self) -> Array {
 | 
			
		||||
        Array::new_dyn(self.element().canonical(), self.len())
 | 
			
		||||
    }
 | 
			
		||||
    pub fn can_connect<T2: Type, Len2: Size>(self, rhs: ArrayType<T2, Len2>) -> bool {
 | 
			
		||||
        self.len() == rhs.len()
 | 
			
		||||
            && self
 | 
			
		||||
                .element()
 | 
			
		||||
                .canonical()
 | 
			
		||||
                .can_connect(rhs.element().canonical())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized, State: ?Sized + Folder> Fold<State> for ArrayType<VA>
 | 
			
		||||
where
 | 
			
		||||
    VA::ElementType: Fold<State>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: Type, Len: KnownSize> ArrayType<T, Len> {
 | 
			
		||||
    pub fn new_static(element: T) -> Self {
 | 
			
		||||
        Self::new(element, Len::SizeType::default())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: StaticType, Len: KnownSize> StaticType for ArrayType<T, Len> {
 | 
			
		||||
    const TYPE: Self = Self {
 | 
			
		||||
        element: LazyInterned::new_lazy(&|| T::TYPE.intern_sized()),
 | 
			
		||||
        len: Len::SIZE,
 | 
			
		||||
        type_properties: Self::TYPE_PROPERTIES,
 | 
			
		||||
    };
 | 
			
		||||
    const MASK_TYPE: Self::MaskType = ArrayType::<T::MaskType, Len> {
 | 
			
		||||
        element: LazyInterned::new_lazy(&|| T::MASK_TYPE.intern_sized()),
 | 
			
		||||
        len: Len::SIZE,
 | 
			
		||||
        type_properties: Self::MASK_TYPE_PROPERTIES,
 | 
			
		||||
    };
 | 
			
		||||
    const TYPE_PROPERTIES: TypeProperties =
 | 
			
		||||
        Self::make_type_properties(T::TYPE_PROPERTIES, Len::VALUE);
 | 
			
		||||
    const MASK_TYPE_PROPERTIES: TypeProperties =
 | 
			
		||||
        Self::make_type_properties(T::MASK_TYPE_PROPERTIES, Len::VALUE);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> Array<T> {
 | 
			
		||||
    pub fn new_dyn(element: T, len: usize) -> Self {
 | 
			
		||||
        Self::new(element, len)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type + Fold<State>, Len: Size, State: Folder + ?Sized> Fold<State> for ArrayType<T, Len> {
 | 
			
		||||
    fn fold(self, state: &mut State) -> Result<Self, State::Error> {
 | 
			
		||||
        state.fold_array_type(self)
 | 
			
		||||
    }
 | 
			
		||||
    fn default_fold(self, state: &mut State) -> Result<Self, State::Error> {
 | 
			
		||||
        Ok(Self::new_with_len_type(self.element.fold(state)?, self.len))
 | 
			
		||||
 | 
			
		||||
    fn default_fold(self, state: &mut State) -> Result<Self, <State as Folder>::Error> {
 | 
			
		||||
        Ok(ArrayType::new(self.element().fold(state)?, self.len))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized, State: ?Sized + Visitor> Visit<State> for ArrayType<VA>
 | 
			
		||||
where
 | 
			
		||||
    VA::ElementType: Visit<State>,
 | 
			
		||||
impl<T: Type + Visit<State>, Len: Size, State: Visitor + ?Sized> Visit<State>
 | 
			
		||||
    for ArrayType<T, Len>
 | 
			
		||||
{
 | 
			
		||||
    fn visit(&self, state: &mut State) -> Result<(), State::Error> {
 | 
			
		||||
        state.visit_array_type(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn default_visit(&self, state: &mut State) -> Result<(), State::Error> {
 | 
			
		||||
        self.element.visit(state)
 | 
			
		||||
        self.element().visit(state)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<V: Value<Type: Type<Value = V>>, const N: usize> ArrayType<[V; N]> {
 | 
			
		||||
    pub fn new_array(element: V::Type) -> Self {
 | 
			
		||||
        ArrayType::new_with_len_type(element, ())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<V: StaticValue, const N: usize> StaticType for ArrayType<[V; N]> {
 | 
			
		||||
    fn static_type() -> Self {
 | 
			
		||||
        Self::new_array(StaticType::static_type())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<V: Value<Type: Type<Value = V>>> ArrayType<[V]> {
 | 
			
		||||
    pub fn new_slice(element: V::Type, len: usize) -> Self {
 | 
			
		||||
        ArrayType::new_with_len_type(element, len)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Type for ArrayType<VA> {
 | 
			
		||||
    type CanonicalType = ArrayType<[DynCanonicalValue]>;
 | 
			
		||||
    type Value = Array<VA>;
 | 
			
		||||
    type CanonicalValue = Array<[DynCanonicalValue]>;
 | 
			
		||||
    type MaskType = ArrayType<VA::MaskVA>;
 | 
			
		||||
    type MaskValue = Array<VA::MaskVA>;
 | 
			
		||||
    type MatchVariant = VA::Match;
 | 
			
		||||
impl<T: Type, Len: Size> Type for ArrayType<T, Len> {
 | 
			
		||||
    type BaseType = Array;
 | 
			
		||||
    type MaskType = ArrayType<T::MaskType, Len>;
 | 
			
		||||
    type MatchVariant = Len::ArrayMatch<T>;
 | 
			
		||||
    type MatchActiveScope = ();
 | 
			
		||||
    type MatchVariantAndInactiveScope = MatchVariantWithoutScope<VA::Match>;
 | 
			
		||||
    type MatchVariantAndInactiveScope = MatchVariantWithoutScope<Len::ArrayMatch<T>>;
 | 
			
		||||
    type MatchVariantsIter = std::iter::Once<Self::MatchVariantAndInactiveScope>;
 | 
			
		||||
 | 
			
		||||
    fn match_variants<IO: BundleValue>(
 | 
			
		||||
        this: Expr<Self::Value>,
 | 
			
		||||
        module_builder: &mut ModuleBuilder<IO, NormalModule>,
 | 
			
		||||
    fn match_variants(
 | 
			
		||||
        this: Expr<Self>,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Self::MatchVariantsIter
 | 
			
		||||
    where
 | 
			
		||||
        IO::Type: BundleType<Value = IO>,
 | 
			
		||||
    {
 | 
			
		||||
        let _ = module_builder;
 | 
			
		||||
    ) -> Self::MatchVariantsIter {
 | 
			
		||||
        let base = Expr::as_dyn_array(this);
 | 
			
		||||
        let base_ty = Expr::ty(base);
 | 
			
		||||
        let _ = source_location;
 | 
			
		||||
        std::iter::once(MatchVariantWithoutScope(VA::make_match(this)))
 | 
			
		||||
        let retval = Vec::from_iter((0..base_ty.len()).map(|i| ArrayIndex::new(base, i).to_expr()));
 | 
			
		||||
        std::iter::once(MatchVariantWithoutScope(
 | 
			
		||||
            Len::ArrayMatch::<T>::try_from(retval)
 | 
			
		||||
                .ok()
 | 
			
		||||
                .expect("unreachable"),
 | 
			
		||||
        ))
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        #[derive(Clone, Hash, Eq, PartialEq)]
 | 
			
		||||
        struct ArrayMaskTypeMemoize<T: ArrayTypeTrait>(PhantomData<T>);
 | 
			
		||||
        impl<T: ArrayTypeTrait> Copy for ArrayMaskTypeMemoize<T> {}
 | 
			
		||||
        impl<T: ArrayTypeTrait> Memoize for ArrayMaskTypeMemoize<T> {
 | 
			
		||||
            type Input = ArrayType<T::ValueArrayOrSlice>;
 | 
			
		||||
            type InputOwned = ArrayType<T::ValueArrayOrSlice>;
 | 
			
		||||
            type Output = <ArrayType<T::ValueArrayOrSlice> as Type>::MaskType;
 | 
			
		||||
 | 
			
		||||
            fn inner(self, input: &Self::Input) -> Self::Output {
 | 
			
		||||
                ArrayType::new_with_len_type(input.element.mask_type(), input.len)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        ArrayMaskTypeMemoize::<Self>(PhantomData).get(self)
 | 
			
		||||
        ArrayType::new(self.element().mask_type(), self.len)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        ArrayType {
 | 
			
		||||
            element: self.element.canonical_dyn(),
 | 
			
		||||
            len: self.len(),
 | 
			
		||||
            bit_width: self.bit_width,
 | 
			
		||||
        }
 | 
			
		||||
    fn canonical(&self) -> CanonicalType {
 | 
			
		||||
        CanonicalType::Array(Array::new_dyn(self.element().canonical(), self.len()))
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    fn from_canonical(canonical_type: CanonicalType) -> Self {
 | 
			
		||||
        let CanonicalType::Array(array) = canonical_type else {
 | 
			
		||||
            panic!("expected array");
 | 
			
		||||
        };
 | 
			
		||||
        Self::new(
 | 
			
		||||
            T::from_canonical(array.element()),
 | 
			
		||||
            Len::from_usize(array.len()),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    fn source_location() -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::ArrayType(self.canonical())
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            element: VA::ElementType::from_dyn_canonical_type(t.element),
 | 
			
		||||
            len: VA::try_len_type_from_len(t.len).expect("length should match"),
 | 
			
		||||
            bit_width: t.bit_width,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(<dyn Any>::downcast_ref::<ArrayType<[DynCanonicalValue]>>(
 | 
			
		||||
            this,
 | 
			
		||||
        )?)
 | 
			
		||||
impl<T: Type, Len: Size> TypeWithDeref for ArrayType<T, Len> {
 | 
			
		||||
    fn expr_deref(this: &Expr<Self>) -> &Self::MatchVariant {
 | 
			
		||||
        let base = Expr::as_dyn_array(*this);
 | 
			
		||||
        let base_ty = Expr::ty(base);
 | 
			
		||||
        let retval = Vec::from_iter((0..base_ty.len()).map(|i| ArrayIndex::new(base, i).to_expr()));
 | 
			
		||||
        Interned::<_>::into_inner(Intern::intern_sized(
 | 
			
		||||
            Len::ArrayMatch::<T>::try_from(retval)
 | 
			
		||||
                .ok()
 | 
			
		||||
                .expect("unreachable"),
 | 
			
		||||
        ))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<Lhs: ValueArrayOrSlice + ?Sized, Rhs: ValueArrayOrSlice + ?Sized> Connect<ArrayType<Rhs>>
 | 
			
		||||
    for ArrayType<Lhs>
 | 
			
		||||
{
 | 
			
		||||
}
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, Default)]
 | 
			
		||||
pub struct ArrayWithoutGenerics;
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for ArrayType<[DynCanonicalValue]> {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::ArrayType;
 | 
			
		||||
}
 | 
			
		||||
impl<T: Type> Index<T> for ArrayWithoutGenerics {
 | 
			
		||||
    type Output = ArrayWithoutLen<T>;
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct Array<VA: ValueArrayOrSlice + ?Sized> {
 | 
			
		||||
    element_ty: VA::ElementType,
 | 
			
		||||
    value: Arc<VA>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Clone for Array<VA> {
 | 
			
		||||
    fn clone(&self) -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            element_ty: self.element_ty.clone(),
 | 
			
		||||
            value: self.value.clone(),
 | 
			
		||||
        }
 | 
			
		||||
    fn index(&self, element: T) -> &Self::Output {
 | 
			
		||||
        Interned::<_>::into_inner(Intern::intern_sized(ArrayWithoutLen { element }))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> ToExpr for Array<VA> {
 | 
			
		||||
    type Type = ArrayType<VA>;
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
pub struct ArrayWithoutLen<T: Type> {
 | 
			
		||||
    element: T,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        ArrayType::new_with_len_type(self.element_ty.clone(), self.value.len_type())
 | 
			
		||||
    }
 | 
			
		||||
impl<T: Type> Index<usize> for ArrayWithoutLen<T> {
 | 
			
		||||
    type Output = Array<T>;
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    fn index(&self, len: usize) -> &Self::Output {
 | 
			
		||||
        Interned::<_>::into_inner(Intern::intern_sized(Array::new_dyn(self.element, len)))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Value for Array<VA> {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        Array {
 | 
			
		||||
            element_ty: self.element_ty.canonical_dyn(),
 | 
			
		||||
            value: AsRef::<[_]>::as_ref(&*self.value)
 | 
			
		||||
                .iter()
 | 
			
		||||
                .map(|v| v.to_canonical_dyn())
 | 
			
		||||
                .collect(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        #[derive(Hash, Eq, PartialEq)]
 | 
			
		||||
        struct ArrayToBitsMemoize<VA: ValueArrayOrSlice + ?Sized>(PhantomData<VA>);
 | 
			
		||||
        impl<VA: ValueArrayOrSlice + ?Sized> Clone for ArrayToBitsMemoize<VA> {
 | 
			
		||||
            fn clone(&self) -> Self {
 | 
			
		||||
                *self
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        impl<VA: ValueArrayOrSlice + ?Sized> Copy for ArrayToBitsMemoize<VA> {}
 | 
			
		||||
        impl<VA: ValueArrayOrSlice + ?Sized> Memoize for ArrayToBitsMemoize<VA> {
 | 
			
		||||
            type Input = Array<VA>;
 | 
			
		||||
            type InputOwned = Array<VA>;
 | 
			
		||||
            type Output = Interned<BitSlice>;
 | 
			
		||||
 | 
			
		||||
            fn inner(self, input: &Self::Input) -> Self::Output {
 | 
			
		||||
                let mut bits = BitVec::with_capacity(input.ty().bit_width());
 | 
			
		||||
                for element in AsRef::<[_]>::as_ref(&*input.value).iter() {
 | 
			
		||||
                    bits.extend_from_bitslice(&element.to_bits());
 | 
			
		||||
                }
 | 
			
		||||
                Intern::intern_owned(bits)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        ArrayToBitsMemoize::<VA>(PhantomData).get(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for Array<[DynCanonicalValue]> {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::Array(this.clone())
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        Value::to_bits_impl(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Array<VA> {
 | 
			
		||||
    pub fn element_ty(&self) -> &VA::ElementType {
 | 
			
		||||
        &self.element_ty
 | 
			
		||||
    }
 | 
			
		||||
    pub fn len(&self) -> usize {
 | 
			
		||||
        VA::len_from_len_type(self.value.len_type())
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_empty(&self) -> bool {
 | 
			
		||||
        self.len() == 0
 | 
			
		||||
    }
 | 
			
		||||
    pub fn value(&self) -> &Arc<VA> {
 | 
			
		||||
        &self.value
 | 
			
		||||
    }
 | 
			
		||||
    pub fn set_element(&mut self, index: usize, element: VA::Element) {
 | 
			
		||||
        assert_eq!(self.element_ty, element.ty());
 | 
			
		||||
        VA::arc_make_mut(&mut self.value)[index] = element;
 | 
			
		||||
    }
 | 
			
		||||
    pub fn new(element_ty: VA::ElementType, value: Arc<VA>) -> Self {
 | 
			
		||||
        for element in value.iter() {
 | 
			
		||||
            assert_eq!(element_ty, element.ty());
 | 
			
		||||
        }
 | 
			
		||||
        Self { element_ty, value }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn into_slice(self) -> Array<[VA::Element]> {
 | 
			
		||||
        Array {
 | 
			
		||||
            element_ty: self.element_ty,
 | 
			
		||||
            value: self.value.arc_to_arc_slice(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized, T: Into<Arc<VA>>> From<T> for Array<VA>
 | 
			
		||||
impl<T: Type, const LEN: usize> Index<ConstUsize<LEN>> for ArrayWithoutLen<T>
 | 
			
		||||
where
 | 
			
		||||
    VA::Element: StaticValue,
 | 
			
		||||
    ConstUsize<LEN>: KnownSize,
 | 
			
		||||
{
 | 
			
		||||
    fn from(value: T) -> Self {
 | 
			
		||||
        Self::new(StaticType::static_type(), value.into())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<E: ToExpr<Type = T>, T: StaticType<MaskType: StaticType>> ToExpr for [E] {
 | 
			
		||||
    type Type = ArrayType<[T::Value]>;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        ArrayType::new_with_len_type(StaticType::static_type(), self.len())
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        let elements = Intern::intern_owned(Vec::from_iter(
 | 
			
		||||
            self.iter().map(|v| v.to_expr().to_canonical_dyn()),
 | 
			
		||||
        ));
 | 
			
		||||
        ArrayLiteral::new_unchecked(elements, self.ty()).to_expr()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<E: ToExpr<Type = T>, T: StaticType<MaskType: StaticType>> ToExpr for Vec<E> {
 | 
			
		||||
    type Type = ArrayType<[T::Value]>;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        <[E]>::ty(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        <[E]>::to_expr(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<E: ToExpr<Type = T>, T: StaticType<MaskType: StaticType>, const N: usize> ToExpr for [E; N] {
 | 
			
		||||
    type Type = ArrayType<[T::Value; N]>;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        ArrayType::new_with_len_type(StaticType::static_type(), ())
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        let elements = Intern::intern_owned(Vec::from_iter(
 | 
			
		||||
            self.iter().map(|v| v.to_expr().to_canonical_dyn()),
 | 
			
		||||
        ));
 | 
			
		||||
        ArrayLiteral::new_unchecked(elements, self.ty()).to_expr()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub struct ArrayIntoIter<VA: ValueArrayOrSlice> {
 | 
			
		||||
    array: Arc<VA>,
 | 
			
		||||
    indexes: std::ops::Range<usize>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> Iterator for ArrayIntoIter<VA> {
 | 
			
		||||
    type Item = VA::Element;
 | 
			
		||||
 | 
			
		||||
    fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        Some(self.array[self.indexes.next()?].clone())
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn size_hint(&self) -> (usize, Option<usize>) {
 | 
			
		||||
        self.indexes.size_hint()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> std::iter::FusedIterator for ArrayIntoIter<VA> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> ExactSizeIterator for ArrayIntoIter<VA> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> DoubleEndedIterator for ArrayIntoIter<VA> {
 | 
			
		||||
    fn next_back(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        Some(self.array[self.indexes.next_back()?].clone())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> Array<VA> {
 | 
			
		||||
    pub fn iter(&self) -> std::slice::Iter<'_, VA::Element> {
 | 
			
		||||
        self.value.iter()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<'a, VA: ValueArrayOrSlice> IntoIterator for &'a Array<VA> {
 | 
			
		||||
    type Item = &'a VA::Element;
 | 
			
		||||
    type IntoIter = std::slice::Iter<'a, VA::Element>;
 | 
			
		||||
 | 
			
		||||
    fn into_iter(self) -> Self::IntoIter {
 | 
			
		||||
        self.value.iter()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> IntoIterator for Array<VA> {
 | 
			
		||||
    type Item = VA::Element;
 | 
			
		||||
    type IntoIter = ArrayIntoIter<VA>;
 | 
			
		||||
 | 
			
		||||
    fn into_iter(self) -> Self::IntoIter {
 | 
			
		||||
        ArrayIntoIter {
 | 
			
		||||
            indexes: 0..self.len(),
 | 
			
		||||
            array: self.value,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
pub struct ArrayExprIter<VA: ValueArrayOrSlice> {
 | 
			
		||||
    array: Expr<Array<VA>>,
 | 
			
		||||
    indexes: std::ops::Range<usize>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> Iterator for ArrayExprIter<VA> {
 | 
			
		||||
    type Item = Expr<VA::Element>;
 | 
			
		||||
 | 
			
		||||
    fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        Some(ExprIndex::expr_index(self.array, self.indexes.next()?))
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn size_hint(&self) -> (usize, Option<usize>) {
 | 
			
		||||
        self.indexes.size_hint()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> std::iter::FusedIterator for ArrayExprIter<VA> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> ExactSizeIterator for ArrayExprIter<VA> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> DoubleEndedIterator for ArrayExprIter<VA> {
 | 
			
		||||
    fn next_back(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        Some(ExprIndex::expr_index(self.array, self.indexes.next_back()?))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> IntoIterator for Expr<Array<VA>> {
 | 
			
		||||
    type Item = Expr<VA::Element>;
 | 
			
		||||
    type IntoIter = ArrayExprIter<VA>;
 | 
			
		||||
 | 
			
		||||
    fn into_iter(self) -> Self::IntoIter {
 | 
			
		||||
        self.iter()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> IntoIterator for &'_ Expr<Array<VA>> {
 | 
			
		||||
    type Item = Expr<VA::Element>;
 | 
			
		||||
    type IntoIter = ArrayExprIter<VA>;
 | 
			
		||||
 | 
			
		||||
    fn into_iter(self) -> Self::IntoIter {
 | 
			
		||||
        self.iter()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice> Expr<Array<VA>> {
 | 
			
		||||
    pub fn len(self) -> usize {
 | 
			
		||||
        self.canonical_type().len()
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_empty(self) -> bool {
 | 
			
		||||
        self.canonical_type().is_empty()
 | 
			
		||||
    }
 | 
			
		||||
    pub fn iter(self) -> ArrayExprIter<VA> {
 | 
			
		||||
        ArrayExprIter {
 | 
			
		||||
            indexes: 0..self.len(),
 | 
			
		||||
            array: self,
 | 
			
		||||
        }
 | 
			
		||||
    type Output = Array<T, LEN>;
 | 
			
		||||
 | 
			
		||||
    fn index(&self, len: ConstUsize<LEN>) -> &Self::Output {
 | 
			
		||||
        Interned::<_>::into_inner(Intern::intern_sized(Array::new(self.element, len)))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -1,5 +1,5 @@
 | 
			
		|||
use crate::{
 | 
			
		||||
    bundle::{BundleType, BundleValue, DynBundle},
 | 
			
		||||
    bundle::{Bundle, BundleType},
 | 
			
		||||
    firrtl,
 | 
			
		||||
    intern::Interned,
 | 
			
		||||
    module::Module,
 | 
			
		||||
| 
						 | 
				
			
			@ -82,7 +82,7 @@ impl FirrtlOutput {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
impl FirrtlArgs {
 | 
			
		||||
    fn run_impl(&self, top_module: Module<DynBundle>) -> Result<FirrtlOutput> {
 | 
			
		||||
    fn run_impl(&self, top_module: Module<Bundle>) -> Result<FirrtlOutput> {
 | 
			
		||||
        let firrtl::FileBackend {
 | 
			
		||||
            top_fir_file_stem, ..
 | 
			
		||||
        } = firrtl::export(self.base.to_firrtl_file_backend(), &top_module)?;
 | 
			
		||||
| 
						 | 
				
			
			@ -94,20 +94,14 @@ impl FirrtlArgs {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: BundleValue> RunPhase<Module<T>> for FirrtlArgs
 | 
			
		||||
where
 | 
			
		||||
    T::Type: BundleType<Value = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: BundleType> RunPhase<Module<T>> for FirrtlArgs {
 | 
			
		||||
    type Output = FirrtlOutput;
 | 
			
		||||
    fn run(&self, top_module: Module<T>) -> Result<Self::Output> {
 | 
			
		||||
        self.run_impl(top_module.canonical())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: BundleValue> RunPhase<Interned<Module<T>>> for FirrtlArgs
 | 
			
		||||
where
 | 
			
		||||
    T::Type: BundleType<Value = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: BundleType> RunPhase<Interned<Module<T>>> for FirrtlArgs {
 | 
			
		||||
    type Output = FirrtlOutput;
 | 
			
		||||
    fn run(&self, top_module: Interned<Module<T>>) -> Result<Self::Output> {
 | 
			
		||||
        self.run(*top_module)
 | 
			
		||||
| 
						 | 
				
			
			@ -228,7 +222,7 @@ enum CliCommand {
 | 
			
		|||
/// Use like:
 | 
			
		||||
///
 | 
			
		||||
/// ```no_run
 | 
			
		||||
/// # use fayalite::hdl_module;
 | 
			
		||||
/// # use fayalite::prelude::*;
 | 
			
		||||
/// # #[hdl_module]
 | 
			
		||||
/// # fn my_module() {}
 | 
			
		||||
/// use fayalite::cli;
 | 
			
		||||
| 
						 | 
				
			
			@ -241,7 +235,7 @@ enum CliCommand {
 | 
			
		|||
/// You can also use it with a larger [`clap`]-based CLI like so:
 | 
			
		||||
///
 | 
			
		||||
/// ```no_run
 | 
			
		||||
/// # use fayalite::hdl_module;
 | 
			
		||||
/// # use fayalite::prelude::*;
 | 
			
		||||
/// # #[hdl_module]
 | 
			
		||||
/// # fn my_module() {}
 | 
			
		||||
/// use clap::{Subcommand, Parser};
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,111 +2,61 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    expr::{Expr, ToExpr},
 | 
			
		||||
    int::{UInt, UIntType},
 | 
			
		||||
    intern::Interned,
 | 
			
		||||
    hdl,
 | 
			
		||||
    int::Bool,
 | 
			
		||||
    reset::Reset,
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{
 | 
			
		||||
        impl_match_values_as_self, CanonicalType, CanonicalTypeKind, CanonicalValue, Connect,
 | 
			
		||||
        DynCanonicalType, StaticType, Type, TypeEnum, Value, ValueEnum,
 | 
			
		||||
    },
 | 
			
		||||
    util::interned_bit,
 | 
			
		||||
    ty::{impl_match_variant_as_self, CanonicalType, StaticType, Type, TypeProperties},
 | 
			
		||||
};
 | 
			
		||||
use bitvec::slice::BitSlice;
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
 | 
			
		||||
pub struct ClockType;
 | 
			
		||||
pub struct Clock;
 | 
			
		||||
 | 
			
		||||
impl ClockType {
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
impl Type for Clock {
 | 
			
		||||
    type BaseType = Clock;
 | 
			
		||||
    type MaskType = Bool;
 | 
			
		||||
 | 
			
		||||
impl Type for ClockType {
 | 
			
		||||
    type Value = Clock;
 | 
			
		||||
    type CanonicalType = ClockType;
 | 
			
		||||
    type CanonicalValue = Clock;
 | 
			
		||||
    type MaskType = UIntType<1>;
 | 
			
		||||
    type MaskValue = UInt<1>;
 | 
			
		||||
 | 
			
		||||
    impl_match_values_as_self!();
 | 
			
		||||
    impl_match_variant_as_self!();
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        UIntType::new()
 | 
			
		||||
        Bool
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::Clock(*self)
 | 
			
		||||
    fn canonical(&self) -> CanonicalType {
 | 
			
		||||
        CanonicalType::Clock(*self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        t
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
    fn source_location() -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(this)
 | 
			
		||||
    fn from_canonical(canonical_type: CanonicalType) -> Self {
 | 
			
		||||
        let CanonicalType::Clock(retval) = canonical_type else {
 | 
			
		||||
            panic!("expected Clock");
 | 
			
		||||
        };
 | 
			
		||||
        retval
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Connect<Self> for ClockType {}
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for ClockType {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::Clock;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl StaticType for ClockType {
 | 
			
		||||
    fn static_type() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
impl Clock {
 | 
			
		||||
    pub fn type_properties(self) -> TypeProperties {
 | 
			
		||||
        Self::TYPE_PROPERTIES
 | 
			
		||||
    }
 | 
			
		||||
    pub fn can_connect(self, _rhs: Self) -> bool {
 | 
			
		||||
        true
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug)]
 | 
			
		||||
pub struct Clock(pub bool);
 | 
			
		||||
 | 
			
		||||
impl ToExpr for Clock {
 | 
			
		||||
    type Type = ClockType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        ClockType
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<Self> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Value for Clock {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for Clock {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::Clock(*this)
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, Value)]
 | 
			
		||||
#[hdl(static, outline_generated)]
 | 
			
		||||
pub struct ClockDomain {
 | 
			
		||||
    pub clk: Clock,
 | 
			
		||||
    pub rst: Reset,
 | 
			
		||||
impl StaticType for Clock {
 | 
			
		||||
    const TYPE: Self = Self;
 | 
			
		||||
    const MASK_TYPE: Self::MaskType = Bool;
 | 
			
		||||
    const TYPE_PROPERTIES: TypeProperties = TypeProperties {
 | 
			
		||||
        is_passive: true,
 | 
			
		||||
        is_storable: false,
 | 
			
		||||
        is_castable_from_bits: true,
 | 
			
		||||
        bit_width: 1,
 | 
			
		||||
    };
 | 
			
		||||
    const MASK_TYPE_PROPERTIES: TypeProperties = Bool::TYPE_PROPERTIES;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait ToClock {
 | 
			
		||||
| 
						 | 
				
			
			@ -137,10 +87,10 @@ impl ToClock for Expr<Clock> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToClock for Clock {
 | 
			
		||||
    fn to_clock(&self) -> Expr<Clock> {
 | 
			
		||||
        self.to_expr()
 | 
			
		||||
    }
 | 
			
		||||
#[hdl]
 | 
			
		||||
pub struct ClockDomain {
 | 
			
		||||
    pub clk: Clock,
 | 
			
		||||
    pub rst: Reset,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToClock for bool {
 | 
			
		||||
| 
						 | 
				
			
			@ -148,9 +98,3 @@ impl ToClock for bool {
 | 
			
		|||
        self.to_expr().to_clock()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToClock for UInt<1> {
 | 
			
		||||
    fn to_clock(&self) -> Expr<Clock> {
 | 
			
		||||
        self.to_expr().to_clock()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,42 +1,39 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
#![allow(clippy::type_complexity)]
 | 
			
		||||
 | 
			
		||||
use crate::{
 | 
			
		||||
    bundle::{BundleValue, TypeHintTrait},
 | 
			
		||||
    expr::{ops::VariantAccess, Expr, ToExpr},
 | 
			
		||||
    int::{UInt, UIntType},
 | 
			
		||||
    intern::{Intern, Interned, MemoizeGeneric},
 | 
			
		||||
    hdl,
 | 
			
		||||
    int::Bool,
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    module::{
 | 
			
		||||
        EnumMatchVariantAndInactiveScopeImpl, EnumMatchVariantsIterImpl, ModuleBuilder,
 | 
			
		||||
        NormalModule, Scope,
 | 
			
		||||
        enum_match_variants_helper, EnumMatchVariantAndInactiveScopeImpl,
 | 
			
		||||
        EnumMatchVariantsIterImpl, Scope,
 | 
			
		||||
    },
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{
 | 
			
		||||
        CanonicalType, CanonicalTypeKind, CanonicalValue, Connect, DynCanonicalType,
 | 
			
		||||
        DynCanonicalValue, DynType, MatchVariantAndInactiveScope, Type, TypeEnum, Value, ValueEnum,
 | 
			
		||||
    },
 | 
			
		||||
    ty::{CanonicalType, MatchVariantAndInactiveScope, StaticType, Type, TypeProperties},
 | 
			
		||||
};
 | 
			
		||||
use bitvec::{order::Lsb0, slice::BitSlice, vec::BitVec, view::BitView};
 | 
			
		||||
use hashbrown::HashMap;
 | 
			
		||||
use std::{
 | 
			
		||||
    borrow::Cow,
 | 
			
		||||
    fmt,
 | 
			
		||||
    hash::{Hash, Hasher},
 | 
			
		||||
    iter::FusedIterator,
 | 
			
		||||
    marker::PhantomData,
 | 
			
		||||
};
 | 
			
		||||
use std::{fmt, iter::FusedIterator};
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq)]
 | 
			
		||||
pub struct VariantType<T> {
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
pub struct EnumVariant {
 | 
			
		||||
    pub name: Interned<str>,
 | 
			
		||||
    pub ty: Option<T>,
 | 
			
		||||
    pub ty: Option<CanonicalType>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub struct FmtDebugInEnum<'a, T>(&'a VariantType<T>);
 | 
			
		||||
impl EnumVariant {
 | 
			
		||||
    pub fn fmt_debug_in_enum(self) -> FmtDebugInEnum {
 | 
			
		||||
        FmtDebugInEnum(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: fmt::Debug> fmt::Debug for FmtDebugInEnum<'_, T> {
 | 
			
		||||
#[derive(Copy, Clone)]
 | 
			
		||||
pub struct FmtDebugInEnum(EnumVariant);
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for FmtDebugInEnum {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let VariantType { name, ref ty } = *self.0;
 | 
			
		||||
        let EnumVariant { name, ty } = self.0;
 | 
			
		||||
        if let Some(ty) = ty {
 | 
			
		||||
            write!(f, "{name}({ty:?})")
 | 
			
		||||
        } else {
 | 
			
		||||
| 
						 | 
				
			
			@ -45,76 +42,22 @@ impl<T: fmt::Debug> fmt::Debug for FmtDebugInEnum<'_, T> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: fmt::Debug> fmt::Display for FmtDebugInEnum<'_, T> {
 | 
			
		||||
impl fmt::Display for FmtDebugInEnum {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        fmt::Debug::fmt(self, f)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T> VariantType<T> {
 | 
			
		||||
    pub fn map_opt_ty<U, F: FnOnce(Option<T>) -> Option<U>>(self, f: F) -> VariantType<U> {
 | 
			
		||||
        let Self { name, ty } = self;
 | 
			
		||||
        VariantType { name, ty: f(ty) }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn map_ty<U, F: FnOnce(T) -> U>(self, f: F) -> VariantType<U> {
 | 
			
		||||
        let Self { name, ty } = self;
 | 
			
		||||
        VariantType {
 | 
			
		||||
            name,
 | 
			
		||||
            ty: ty.map(f),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn as_ref_ty(&self) -> VariantType<&T> {
 | 
			
		||||
        VariantType {
 | 
			
		||||
            name: self.name,
 | 
			
		||||
            ty: self.ty.as_ref(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn fmt_debug_in_enum(&self) -> FmtDebugInEnum<T> {
 | 
			
		||||
        FmtDebugInEnum(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> VariantType<T> {
 | 
			
		||||
    pub fn canonical(&self) -> VariantType<T::CanonicalType> {
 | 
			
		||||
        self.as_ref_ty().map_ty(T::canonical)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn to_dyn(&self) -> VariantType<Interned<dyn DynType>> {
 | 
			
		||||
        self.as_ref_ty().map_ty(T::to_dyn)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical_dyn(&self) -> VariantType<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
        self.as_ref_ty().map_ty(T::canonical_dyn)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl VariantType<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
    pub fn from_canonical_type_helper_has_value<T: Type>(self, expected_name: &str) -> T {
 | 
			
		||||
        assert_eq!(&*self.name, expected_name, "variant name doesn't match");
 | 
			
		||||
        let Some(ty) = self.ty else {
 | 
			
		||||
            panic!("variant {expected_name} has no value but a value is expected");
 | 
			
		||||
        };
 | 
			
		||||
        T::from_dyn_canonical_type(ty)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn from_canonical_type_helper_no_value(self, expected_name: &str) {
 | 
			
		||||
        assert_eq!(&*self.name, expected_name, "variant name doesn't match");
 | 
			
		||||
        assert!(
 | 
			
		||||
            self.ty.is_none(),
 | 
			
		||||
            "variant {expected_name} has a value but is expected to have no value"
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Eq)]
 | 
			
		||||
struct DynEnumTypeImpl {
 | 
			
		||||
    variants: Interned<[VariantType<Interned<dyn DynCanonicalType>>]>,
 | 
			
		||||
struct EnumImpl {
 | 
			
		||||
    variants: Interned<[EnumVariant]>,
 | 
			
		||||
    name_indexes: HashMap<Interned<str>, usize>,
 | 
			
		||||
    bit_width: usize,
 | 
			
		||||
    is_storable: bool,
 | 
			
		||||
    is_castable_from_bits: bool,
 | 
			
		||||
    type_properties: TypeProperties,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for DynEnumTypeImpl {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        write!(f, "DynEnumType ")?;
 | 
			
		||||
impl std::fmt::Debug for EnumImpl {
 | 
			
		||||
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 | 
			
		||||
        f.write_str("Enum ")?;
 | 
			
		||||
        f.debug_set()
 | 
			
		||||
            .entries(
 | 
			
		||||
                self.variants
 | 
			
		||||
| 
						 | 
				
			
			@ -125,66 +68,123 @@ impl fmt::Debug for DynEnumTypeImpl {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl PartialEq for DynEnumTypeImpl {
 | 
			
		||||
impl std::hash::Hash for EnumImpl {
 | 
			
		||||
    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
 | 
			
		||||
        self.variants.hash(state);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl PartialEq for EnumImpl {
 | 
			
		||||
    fn eq(&self, other: &Self) -> bool {
 | 
			
		||||
        self.variants == other.variants
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Hash for DynEnumTypeImpl {
 | 
			
		||||
    fn hash<H: Hasher>(&self, state: &mut H) {
 | 
			
		||||
        self.variants.hash(state);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct Enum(Interned<EnumImpl>);
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Hash, PartialEq, Eq)]
 | 
			
		||||
pub struct DynEnumType(Interned<DynEnumTypeImpl>);
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for DynEnumType {
 | 
			
		||||
impl fmt::Debug for Enum {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        self.0.fmt(f)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn discriminant_bit_width_impl(variant_count: usize) -> usize {
 | 
			
		||||
    variant_count
 | 
			
		||||
        .next_power_of_two()
 | 
			
		||||
        .checked_ilog2()
 | 
			
		||||
        .unwrap_or(0) as usize
 | 
			
		||||
const fn discriminant_bit_width_impl(variant_count: usize) -> usize {
 | 
			
		||||
    match variant_count.next_power_of_two().checked_ilog2() {
 | 
			
		||||
        Some(x) => x as usize,
 | 
			
		||||
        None => 0,
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl DynEnumType {
 | 
			
		||||
#[derive(Clone)]
 | 
			
		||||
pub struct EnumTypePropertiesBuilder {
 | 
			
		||||
    type_properties: TypeProperties,
 | 
			
		||||
    variant_count: usize,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl EnumTypePropertiesBuilder {
 | 
			
		||||
    #[must_use]
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            type_properties: TypeProperties {
 | 
			
		||||
                is_passive: true,
 | 
			
		||||
                is_storable: true,
 | 
			
		||||
                is_castable_from_bits: true,
 | 
			
		||||
                bit_width: 0,
 | 
			
		||||
            },
 | 
			
		||||
            variant_count: 0,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub const fn clone(&self) -> Self {
 | 
			
		||||
        Self { ..*self }
 | 
			
		||||
    }
 | 
			
		||||
    #[must_use]
 | 
			
		||||
    pub const fn variant(self, field_props: Option<TypeProperties>) -> Self {
 | 
			
		||||
        let Self {
 | 
			
		||||
            mut type_properties,
 | 
			
		||||
            variant_count,
 | 
			
		||||
        } = self;
 | 
			
		||||
        if let Some(TypeProperties {
 | 
			
		||||
            is_passive,
 | 
			
		||||
            is_storable,
 | 
			
		||||
            is_castable_from_bits,
 | 
			
		||||
            bit_width,
 | 
			
		||||
        }) = field_props
 | 
			
		||||
        {
 | 
			
		||||
            assert!(is_passive, "variant type must be a passive type");
 | 
			
		||||
            type_properties = TypeProperties {
 | 
			
		||||
                is_passive: true,
 | 
			
		||||
                is_storable: type_properties.is_storable & is_storable,
 | 
			
		||||
                is_castable_from_bits: type_properties.is_castable_from_bits
 | 
			
		||||
                    & is_castable_from_bits,
 | 
			
		||||
                bit_width: if type_properties.bit_width < bit_width {
 | 
			
		||||
                    bit_width
 | 
			
		||||
                } else {
 | 
			
		||||
                    type_properties.bit_width
 | 
			
		||||
                },
 | 
			
		||||
            };
 | 
			
		||||
        }
 | 
			
		||||
        Self {
 | 
			
		||||
            type_properties,
 | 
			
		||||
            variant_count: variant_count + 1,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub const fn finish(self) -> TypeProperties {
 | 
			
		||||
        assert!(
 | 
			
		||||
            self.variant_count != 0,
 | 
			
		||||
            "zero-variant enums aren't yet supported: \
 | 
			
		||||
            https://github.com/chipsalliance/firrtl-spec/issues/208",
 | 
			
		||||
        );
 | 
			
		||||
        let Some(bit_width) = self
 | 
			
		||||
            .type_properties
 | 
			
		||||
            .bit_width
 | 
			
		||||
            .checked_add(discriminant_bit_width_impl(self.variant_count))
 | 
			
		||||
        else {
 | 
			
		||||
            panic!("enum is too big: bit-width overflowed");
 | 
			
		||||
        };
 | 
			
		||||
        TypeProperties {
 | 
			
		||||
            bit_width,
 | 
			
		||||
            ..self.type_properties
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Enum {
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new(variants: Interned<[VariantType<Interned<dyn DynCanonicalType>>]>) -> Self {
 | 
			
		||||
        assert!(!variants.is_empty(), "zero-variant enums aren't yet supported: https://github.com/chipsalliance/firrtl-spec/issues/208");
 | 
			
		||||
    pub fn new(variants: Interned<[EnumVariant]>) -> Self {
 | 
			
		||||
        let mut name_indexes = HashMap::with_capacity(variants.len());
 | 
			
		||||
        let mut body_bit_width = 0usize;
 | 
			
		||||
        let mut is_storable = true;
 | 
			
		||||
        let mut is_castable_from_bits = true;
 | 
			
		||||
        for (index, &VariantType { name, ty }) in variants.iter().enumerate() {
 | 
			
		||||
            if let Some(old_index) = name_indexes.insert(name, index) {
 | 
			
		||||
        let mut type_props_builder = EnumTypePropertiesBuilder::new();
 | 
			
		||||
        for (index, EnumVariant { name, ty }) in variants.iter().enumerate() {
 | 
			
		||||
            if let Some(old_index) = name_indexes.insert(*name, index) {
 | 
			
		||||
                panic!("duplicate variant name {name:?}: at both index {old_index} and {index}");
 | 
			
		||||
            }
 | 
			
		||||
            if let Some(ty) = ty {
 | 
			
		||||
                assert!(
 | 
			
		||||
                    ty.is_passive(),
 | 
			
		||||
                    "variant type must be a passive type: {ty:?}"
 | 
			
		||||
                );
 | 
			
		||||
                body_bit_width = body_bit_width.max(ty.bit_width());
 | 
			
		||||
                is_storable &= ty.is_storable();
 | 
			
		||||
                is_castable_from_bits &= ty.is_castable_from_bits();
 | 
			
		||||
            }
 | 
			
		||||
            type_props_builder = type_props_builder.variant(ty.map(CanonicalType::type_properties));
 | 
			
		||||
        }
 | 
			
		||||
        let bit_width = body_bit_width
 | 
			
		||||
            .checked_add(discriminant_bit_width_impl(variants.len()))
 | 
			
		||||
            .unwrap_or_else(|| panic!("enum is too big: bit-width overflowed"));
 | 
			
		||||
        Self(
 | 
			
		||||
            DynEnumTypeImpl {
 | 
			
		||||
            EnumImpl {
 | 
			
		||||
                variants,
 | 
			
		||||
                name_indexes,
 | 
			
		||||
                bit_width,
 | 
			
		||||
                is_storable,
 | 
			
		||||
                is_castable_from_bits,
 | 
			
		||||
                type_properties: type_props_builder.finish(),
 | 
			
		||||
            }
 | 
			
		||||
            .intern_sized(),
 | 
			
		||||
        )
 | 
			
		||||
| 
						 | 
				
			
			@ -192,233 +192,62 @@ impl DynEnumType {
 | 
			
		|||
    pub fn discriminant_bit_width(self) -> usize {
 | 
			
		||||
        discriminant_bit_width_impl(self.variants().len())
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_passive(self) -> bool {
 | 
			
		||||
        true
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_storable(self) -> bool {
 | 
			
		||||
        self.0.is_storable
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_castable_from_bits(self) -> bool {
 | 
			
		||||
        self.0.is_castable_from_bits
 | 
			
		||||
    }
 | 
			
		||||
    pub fn bit_width(self) -> usize {
 | 
			
		||||
        self.0.bit_width
 | 
			
		||||
    pub fn type_properties(self) -> TypeProperties {
 | 
			
		||||
        self.0.type_properties
 | 
			
		||||
    }
 | 
			
		||||
    pub fn name_indexes(&self) -> &HashMap<Interned<str>, usize> {
 | 
			
		||||
        &self.0.name_indexes
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone, Hash, PartialEq, Eq)]
 | 
			
		||||
pub struct DynEnum {
 | 
			
		||||
    ty: DynEnumType,
 | 
			
		||||
    variant_index: usize,
 | 
			
		||||
    variant_value: Option<DynCanonicalValue>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl DynEnum {
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_by_index(
 | 
			
		||||
        ty: DynEnumType,
 | 
			
		||||
        variant_index: usize,
 | 
			
		||||
        variant_value: Option<DynCanonicalValue>,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        let variant = ty.variants()[variant_index];
 | 
			
		||||
        assert_eq!(
 | 
			
		||||
            variant_value.as_ref().map(|v| v.ty()),
 | 
			
		||||
            variant.ty,
 | 
			
		||||
            "variant value doesn't match type"
 | 
			
		||||
        );
 | 
			
		||||
        Self {
 | 
			
		||||
            ty,
 | 
			
		||||
            variant_index,
 | 
			
		||||
            variant_value,
 | 
			
		||||
    pub fn can_connect(self, rhs: Self) -> bool {
 | 
			
		||||
        if self.0.variants.len() != rhs.0.variants.len() {
 | 
			
		||||
            return false;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_by_name(
 | 
			
		||||
        ty: DynEnumType,
 | 
			
		||||
        variant_name: Interned<str>,
 | 
			
		||||
        variant_value: Option<DynCanonicalValue>,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        let variant_index = ty.name_indexes()[&variant_name];
 | 
			
		||||
        Self::new_by_index(ty, variant_index, variant_value)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_index(&self) -> usize {
 | 
			
		||||
        self.variant_index
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_value(&self) -> &Option<DynCanonicalValue> {
 | 
			
		||||
        &self.variant_value
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_with_type(&self) -> VariantType<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
        self.ty.variants()[self.variant_index]
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_name(&self) -> Interned<str> {
 | 
			
		||||
        self.variant_with_type().name
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_type(&self) -> Option<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
        self.variant_with_type().ty
 | 
			
		||||
    }
 | 
			
		||||
    pub fn variant_with_value(&self) -> VariantType<&DynCanonicalValue> {
 | 
			
		||||
        self.variant_with_type()
 | 
			
		||||
            .map_opt_ty(|_| self.variant_value.as_ref())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug)]
 | 
			
		||||
pub struct VariantsHint {
 | 
			
		||||
    pub known_variants: Interned<[VariantType<Interned<dyn TypeHintTrait>>]>,
 | 
			
		||||
    pub more_variants: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl VariantsHint {
 | 
			
		||||
    pub fn new(
 | 
			
		||||
        known_variants: impl IntoIterator<Item = VariantType<Interned<dyn TypeHintTrait>>>,
 | 
			
		||||
        more_variants: bool,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        let known_variants = Intern::intern_owned(Vec::from_iter(known_variants));
 | 
			
		||||
        Self {
 | 
			
		||||
            known_variants,
 | 
			
		||||
            more_variants,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn check_variant(
 | 
			
		||||
        self,
 | 
			
		||||
        index: usize,
 | 
			
		||||
        variant: VariantType<&dyn DynType>,
 | 
			
		||||
    ) -> Result<(), String> {
 | 
			
		||||
        let Some(&known_variant) = self.known_variants.get(index) else {
 | 
			
		||||
            return if self.more_variants {
 | 
			
		||||
                Ok(())
 | 
			
		||||
            } else {
 | 
			
		||||
                Err(format!(
 | 
			
		||||
                    "too many variants: name={:?} index={index}",
 | 
			
		||||
                    variant.name
 | 
			
		||||
                ))
 | 
			
		||||
            };
 | 
			
		||||
        };
 | 
			
		||||
        let VariantType {
 | 
			
		||||
            name: known_name,
 | 
			
		||||
            ty: type_hint,
 | 
			
		||||
        } = known_variant;
 | 
			
		||||
        let VariantType { name, ty } = variant;
 | 
			
		||||
        if name != known_name {
 | 
			
		||||
            Err(format!(
 | 
			
		||||
                "wrong variant name {name:?}, expected {known_name:?}"
 | 
			
		||||
            ))
 | 
			
		||||
        } else {
 | 
			
		||||
            match (ty, type_hint) {
 | 
			
		||||
                (Some(ty), Some(type_hint)) => type_hint.matches(ty),
 | 
			
		||||
                (None, None) => Ok(()),
 | 
			
		||||
                (None, Some(_)) => Err(format!(
 | 
			
		||||
                    "expected variant {name:?} to have type, no type provided"
 | 
			
		||||
                )),
 | 
			
		||||
                (Some(_), None) => Err(format!(
 | 
			
		||||
                    "expected variant {name:?} to have no type, but a type was provided"
 | 
			
		||||
                )),
 | 
			
		||||
        for (
 | 
			
		||||
            &EnumVariant {
 | 
			
		||||
                name: lhs_name,
 | 
			
		||||
                ty: lhs_ty,
 | 
			
		||||
            },
 | 
			
		||||
            &EnumVariant {
 | 
			
		||||
                name: rhs_name,
 | 
			
		||||
                ty: rhs_ty,
 | 
			
		||||
            },
 | 
			
		||||
        ) in self.0.variants.iter().zip(rhs.0.variants.iter())
 | 
			
		||||
        {
 | 
			
		||||
            if lhs_name != rhs_name {
 | 
			
		||||
                return false;
 | 
			
		||||
            }
 | 
			
		||||
            match (lhs_ty, rhs_ty) {
 | 
			
		||||
                (None, None) => {}
 | 
			
		||||
                (None, Some(_)) | (Some(_), None) => return false,
 | 
			
		||||
                (Some(lhs_ty), Some(rhs_ty)) => {
 | 
			
		||||
                    if !lhs_ty.can_connect(rhs_ty) {
 | 
			
		||||
                        return false;
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        true
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait EnumType:
 | 
			
		||||
    Type<
 | 
			
		||||
        CanonicalType = DynEnumType,
 | 
			
		||||
        CanonicalValue = DynEnum,
 | 
			
		||||
        MaskType = UIntType<1>,
 | 
			
		||||
        MaskValue = UInt<1>,
 | 
			
		||||
        MatchActiveScope = Scope,
 | 
			
		||||
        MatchVariantAndInactiveScope = EnumMatchVariantAndInactiveScope<Self>,
 | 
			
		||||
        MatchVariantsIter = EnumMatchVariantsIter<Self>,
 | 
			
		||||
    > + Connect<Self>
 | 
			
		||||
where
 | 
			
		||||
    Self::Value: EnumValue + ToExpr<Type = Self>,
 | 
			
		||||
    BaseType = Enum,
 | 
			
		||||
    MaskType = Bool,
 | 
			
		||||
    MatchActiveScope = Scope,
 | 
			
		||||
    MatchVariantAndInactiveScope = EnumMatchVariantAndInactiveScope<Self>,
 | 
			
		||||
    MatchVariantsIter = EnumMatchVariantsIter<Self>,
 | 
			
		||||
>
 | 
			
		||||
{
 | 
			
		||||
    type Builder;
 | 
			
		||||
    fn variants(&self) -> Interned<[EnumVariant]>;
 | 
			
		||||
    fn match_activate_scope(
 | 
			
		||||
        v: Self::MatchVariantAndInactiveScope,
 | 
			
		||||
    ) -> (Self::MatchVariant, Self::MatchActiveScope);
 | 
			
		||||
    fn builder() -> Self::Builder;
 | 
			
		||||
    fn variants(&self) -> Interned<[VariantType<Interned<dyn DynCanonicalType>>]>;
 | 
			
		||||
    fn variants_hint() -> VariantsHint;
 | 
			
		||||
    #[allow(clippy::result_unit_err)]
 | 
			
		||||
    fn variant_to_bits<VariantValue: ToExpr + Eq + Hash + Send + Sync + 'static + Clone>(
 | 
			
		||||
        &self,
 | 
			
		||||
        variant_index: usize,
 | 
			
		||||
        variant_value: Option<&VariantValue>,
 | 
			
		||||
    ) -> Result<Interned<BitSlice>, ()> {
 | 
			
		||||
        #[derive(Hash, Eq, PartialEq)]
 | 
			
		||||
        struct VariantToBitsMemoize<E, V>(PhantomData<(E, V)>);
 | 
			
		||||
        impl<E, V> Clone for VariantToBitsMemoize<E, V> {
 | 
			
		||||
            fn clone(&self) -> Self {
 | 
			
		||||
                *self
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        impl<E, V> Copy for VariantToBitsMemoize<E, V> {}
 | 
			
		||||
        impl<
 | 
			
		||||
                E: EnumType<Value: EnumValue<Type = E>>,
 | 
			
		||||
                V: ToExpr + Eq + Hash + Send + Sync + 'static + Clone,
 | 
			
		||||
            > MemoizeGeneric for VariantToBitsMemoize<E, V>
 | 
			
		||||
        {
 | 
			
		||||
            type InputRef<'a> = (&'a E, usize, Option<&'a V>);
 | 
			
		||||
            type InputOwned = (E, usize, Option<V>);
 | 
			
		||||
            type InputCow<'a> = (Cow<'a, E>, usize, Option<Cow<'a, V>>);
 | 
			
		||||
            type Output = Result<Interned<BitSlice>, ()>;
 | 
			
		||||
 | 
			
		||||
            fn input_borrow(input: &Self::InputOwned) -> Self::InputRef<'_> {
 | 
			
		||||
                (&input.0, input.1, input.2.as_ref())
 | 
			
		||||
            }
 | 
			
		||||
            fn input_eq(a: Self::InputRef<'_>, b: Self::InputRef<'_>) -> bool {
 | 
			
		||||
                a == b
 | 
			
		||||
            }
 | 
			
		||||
            fn input_cow_into_owned(input: Self::InputCow<'_>) -> Self::InputOwned {
 | 
			
		||||
                (input.0.into_owned(), input.1, input.2.map(Cow::into_owned))
 | 
			
		||||
            }
 | 
			
		||||
            fn input_cow_borrow<'a>(input: &'a Self::InputCow<'_>) -> Self::InputRef<'a> {
 | 
			
		||||
                (&input.0, input.1, input.2.as_deref())
 | 
			
		||||
            }
 | 
			
		||||
            fn input_cow_from_owned<'a>(input: Self::InputOwned) -> Self::InputCow<'a> {
 | 
			
		||||
                (Cow::Owned(input.0), input.1, input.2.map(Cow::Owned))
 | 
			
		||||
            }
 | 
			
		||||
            fn input_cow_from_ref(input: Self::InputRef<'_>) -> Self::InputCow<'_> {
 | 
			
		||||
                (Cow::Borrowed(input.0), input.1, input.2.map(Cow::Borrowed))
 | 
			
		||||
            }
 | 
			
		||||
            fn inner(self, input: Self::InputRef<'_>) -> Self::Output {
 | 
			
		||||
                let (ty, variant_index, variant_value) = input;
 | 
			
		||||
                let ty = ty.canonical();
 | 
			
		||||
                let mut bits = BitVec::with_capacity(ty.bit_width());
 | 
			
		||||
                bits.extend_from_bitslice(
 | 
			
		||||
                    &variant_index.view_bits::<Lsb0>()[..ty.discriminant_bit_width()],
 | 
			
		||||
                );
 | 
			
		||||
                if let Some(variant_value) = variant_value {
 | 
			
		||||
                    bits.extend_from_bitslice(&variant_value.to_expr().to_literal_bits()?);
 | 
			
		||||
                }
 | 
			
		||||
                bits.resize(ty.bit_width(), false);
 | 
			
		||||
                Ok(Intern::intern_owned(bits))
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        VariantToBitsMemoize::<Self, VariantValue>(PhantomData).get((
 | 
			
		||||
            self,
 | 
			
		||||
            variant_index,
 | 
			
		||||
            variant_value,
 | 
			
		||||
        ))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait EnumValue: Value
 | 
			
		||||
where
 | 
			
		||||
    <Self as ToExpr>::Type: EnumType<Value = Self>,
 | 
			
		||||
{
 | 
			
		||||
}
 | 
			
		||||
pub struct EnumMatchVariantAndInactiveScope<T: EnumType>(EnumMatchVariantAndInactiveScopeImpl<T>);
 | 
			
		||||
 | 
			
		||||
pub struct EnumMatchVariantAndInactiveScope<T: EnumType>(EnumMatchVariantAndInactiveScopeImpl<T>)
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>;
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> MatchVariantAndInactiveScope for EnumMatchVariantAndInactiveScope<T>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: EnumType> MatchVariantAndInactiveScope for EnumMatchVariantAndInactiveScope<T> {
 | 
			
		||||
    type MatchVariant = T::MatchVariant;
 | 
			
		||||
    type MatchActiveScope = Scope;
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -427,36 +256,22 @@ where
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> EnumMatchVariantAndInactiveScope<T>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
    pub fn variant_access(&self) -> Interned<VariantAccess<T, Interned<dyn DynCanonicalType>>> {
 | 
			
		||||
impl<T: EnumType> EnumMatchVariantAndInactiveScope<T> {
 | 
			
		||||
    pub fn variant_access(&self) -> VariantAccess {
 | 
			
		||||
        self.0.variant_access()
 | 
			
		||||
    }
 | 
			
		||||
    pub fn activate(
 | 
			
		||||
        self,
 | 
			
		||||
    ) -> (
 | 
			
		||||
        Interned<VariantAccess<T, Interned<dyn DynCanonicalType>>>,
 | 
			
		||||
        Scope,
 | 
			
		||||
    ) {
 | 
			
		||||
    pub fn activate(self) -> (VariantAccess, Scope) {
 | 
			
		||||
        self.0.activate()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone)]
 | 
			
		||||
pub struct EnumMatchVariantsIter<T: EnumType>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
pub struct EnumMatchVariantsIter<T: EnumType> {
 | 
			
		||||
    pub(crate) inner: EnumMatchVariantsIterImpl<T>,
 | 
			
		||||
    pub(crate) variant_index: std::ops::Range<usize>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> Iterator for EnumMatchVariantsIter<T>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: EnumType> Iterator for EnumMatchVariantsIter<T> {
 | 
			
		||||
    type Item = EnumMatchVariantAndInactiveScope<T>;
 | 
			
		||||
 | 
			
		||||
    fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
| 
						 | 
				
			
			@ -470,21 +285,15 @@ where
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> ExactSizeIterator for EnumMatchVariantsIter<T>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: EnumType> ExactSizeIterator for EnumMatchVariantsIter<T> {
 | 
			
		||||
    fn len(&self) -> usize {
 | 
			
		||||
        self.variant_index.len()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> FusedIterator for EnumMatchVariantsIter<T> where T::Value: EnumValue<Type = T> {}
 | 
			
		||||
impl<T: EnumType> FusedIterator for EnumMatchVariantsIter<T> {}
 | 
			
		||||
 | 
			
		||||
impl<T: EnumType> DoubleEndedIterator for EnumMatchVariantsIter<T>
 | 
			
		||||
where
 | 
			
		||||
    T::Value: EnumValue<Type = T>,
 | 
			
		||||
{
 | 
			
		||||
impl<T: EnumType> DoubleEndedIterator for EnumMatchVariantsIter<T> {
 | 
			
		||||
    fn next_back(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        self.variant_index.next_back().map(|variant_index| {
 | 
			
		||||
            EnumMatchVariantAndInactiveScope(self.inner.for_variant_index(variant_index))
 | 
			
		||||
| 
						 | 
				
			
			@ -492,129 +301,66 @@ where
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Type for DynEnumType {
 | 
			
		||||
    type CanonicalType = DynEnumType;
 | 
			
		||||
    type Value = DynEnum;
 | 
			
		||||
    type CanonicalValue = DynEnum;
 | 
			
		||||
    type MaskType = UIntType<1>;
 | 
			
		||||
    type MaskValue = UInt<1>;
 | 
			
		||||
    type MatchVariant = Option<Expr<DynCanonicalValue>>;
 | 
			
		||||
    type MatchActiveScope = Scope;
 | 
			
		||||
    type MatchVariantAndInactiveScope = EnumMatchVariantAndInactiveScope<Self>;
 | 
			
		||||
    type MatchVariantsIter = EnumMatchVariantsIter<Self>;
 | 
			
		||||
 | 
			
		||||
    fn match_variants<IO: BundleValue>(
 | 
			
		||||
        this: Expr<Self::Value>,
 | 
			
		||||
        module_builder: &mut ModuleBuilder<IO, NormalModule>,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Self::MatchVariantsIter
 | 
			
		||||
    where
 | 
			
		||||
        IO::Type: crate::bundle::BundleType<Value = IO>,
 | 
			
		||||
    {
 | 
			
		||||
        module_builder.enum_match_variants_helper(this, source_location)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        UIntType::new()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::EnumType(*self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        t
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Connect<Self> for DynEnumType {}
 | 
			
		||||
 | 
			
		||||
pub struct NoBuilder;
 | 
			
		||||
 | 
			
		||||
impl EnumType for DynEnumType {
 | 
			
		||||
    type Builder = NoBuilder;
 | 
			
		||||
 | 
			
		||||
impl EnumType for Enum {
 | 
			
		||||
    fn match_activate_scope(
 | 
			
		||||
        v: Self::MatchVariantAndInactiveScope,
 | 
			
		||||
    ) -> (Self::MatchVariant, Self::MatchActiveScope) {
 | 
			
		||||
        let (expr, scope) = v.0.activate();
 | 
			
		||||
        (expr.variant_type().ty.map(|_| expr.to_expr()), scope)
 | 
			
		||||
        (expr.variant_type().map(|_| expr.to_expr()), scope)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn builder() -> Self::Builder {
 | 
			
		||||
        NoBuilder
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn variants(&self) -> Interned<[VariantType<Interned<dyn DynCanonicalType>>]> {
 | 
			
		||||
    fn variants(&self) -> Interned<[EnumVariant]> {
 | 
			
		||||
        self.0.variants
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
    fn variants_hint() -> VariantsHint {
 | 
			
		||||
        VariantsHint {
 | 
			
		||||
            known_variants: [][..].intern(),
 | 
			
		||||
            more_variants: true,
 | 
			
		||||
        }
 | 
			
		||||
impl Type for Enum {
 | 
			
		||||
    type BaseType = Enum;
 | 
			
		||||
    type MaskType = Bool;
 | 
			
		||||
    type MatchVariant = Option<Expr<CanonicalType>>;
 | 
			
		||||
    type MatchActiveScope = Scope;
 | 
			
		||||
    type MatchVariantAndInactiveScope = EnumMatchVariantAndInactiveScope<Self>;
 | 
			
		||||
    type MatchVariantsIter = EnumMatchVariantsIter<Self>;
 | 
			
		||||
 | 
			
		||||
    fn match_variants(
 | 
			
		||||
        this: Expr<Self>,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Self::MatchVariantsIter {
 | 
			
		||||
        enum_match_variants_helper(this, source_location)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        Bool
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> CanonicalType {
 | 
			
		||||
        CanonicalType::Enum(*self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    fn from_canonical(canonical_type: CanonicalType) -> Self {
 | 
			
		||||
        let CanonicalType::Enum(retval) = canonical_type else {
 | 
			
		||||
            panic!("expected enum");
 | 
			
		||||
        };
 | 
			
		||||
        retval
 | 
			
		||||
    }
 | 
			
		||||
    fn source_location() -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for DynEnumType {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::EnumType;
 | 
			
		||||
#[hdl]
 | 
			
		||||
pub enum HdlOption<T: Type> {
 | 
			
		||||
    HdlNone,
 | 
			
		||||
    HdlSome(T),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ToExpr for DynEnum {
 | 
			
		||||
    type Type = DynEnumType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        self.ty
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    }
 | 
			
		||||
#[allow(non_snake_case)]
 | 
			
		||||
pub fn HdlNone<T: StaticType>() -> Expr<HdlOption<T>> {
 | 
			
		||||
    HdlOption[T::TYPE].HdlNone()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Value for DynEnum {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        self.clone()
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        this.ty
 | 
			
		||||
            .variant_to_bits(this.variant_index, this.variant_value.as_ref())
 | 
			
		||||
            .unwrap()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl EnumValue for DynEnum {}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for DynEnum {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::Enum(this.clone())
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        this.ty
 | 
			
		||||
            .variant_to_bits(this.variant_index, this.variant_value.as_ref())
 | 
			
		||||
            .unwrap()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
mod impl_option {
 | 
			
		||||
    #[allow(dead_code)]
 | 
			
		||||
    #[derive(crate::ty::Value)]
 | 
			
		||||
    #[hdl(target(std::option::Option), connect_inexact, outline_generated)]
 | 
			
		||||
    pub enum Option<T> {
 | 
			
		||||
        None,
 | 
			
		||||
        Some(T),
 | 
			
		||||
    }
 | 
			
		||||
#[allow(non_snake_case)]
 | 
			
		||||
pub fn HdlSome<T: Type>(value: impl ToExpr<Type = T>) -> Expr<HdlOption<T>> {
 | 
			
		||||
    let value = value.to_expr();
 | 
			
		||||
    HdlOption[Expr::ty(value)].HdlSome(value)
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
							
								
								
									
										432
									
								
								crates/fayalite/src/expr/target.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										432
									
								
								crates/fayalite/src/expr/target.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,432 @@
 | 
			
		|||
use crate::{
 | 
			
		||||
    array::Array,
 | 
			
		||||
    bundle::{Bundle, BundleField},
 | 
			
		||||
    expr::Flow,
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    memory::{DynPortType, MemPort},
 | 
			
		||||
    module::{Instance, ModuleIO, TargetName},
 | 
			
		||||
    reg::Reg,
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
    wire::Wire,
 | 
			
		||||
};
 | 
			
		||||
use std::fmt;
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct TargetPathBundleField {
 | 
			
		||||
    pub name: Interned<str>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetPathBundleField {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        write!(f, ".{}", self.name)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct TargetPathArrayElement {
 | 
			
		||||
    pub index: usize,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetPathArrayElement {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        write!(f, "[{}]", self.index)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct TargetPathDynArrayElement {}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetPathDynArrayElement {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        write!(f, "[<dyn>]")
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub enum TargetPathElement {
 | 
			
		||||
    BundleField(TargetPathBundleField),
 | 
			
		||||
    ArrayElement(TargetPathArrayElement),
 | 
			
		||||
    DynArrayElement(TargetPathDynArrayElement),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<TargetPathBundleField> for TargetPathElement {
 | 
			
		||||
    fn from(value: TargetPathBundleField) -> Self {
 | 
			
		||||
        Self::BundleField(value)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<TargetPathArrayElement> for TargetPathElement {
 | 
			
		||||
    fn from(value: TargetPathArrayElement) -> Self {
 | 
			
		||||
        Self::ArrayElement(value)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<TargetPathDynArrayElement> for TargetPathElement {
 | 
			
		||||
    fn from(value: TargetPathDynArrayElement) -> Self {
 | 
			
		||||
        Self::DynArrayElement(value)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetPathElement {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::BundleField(v) => v.fmt(f),
 | 
			
		||||
            Self::ArrayElement(v) => v.fmt(f),
 | 
			
		||||
            Self::DynArrayElement(v) => v.fmt(f),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl TargetPathElement {
 | 
			
		||||
    pub fn canonical_ty(&self, parent: Interned<Target>) -> CanonicalType {
 | 
			
		||||
        match self {
 | 
			
		||||
            &Self::BundleField(TargetPathBundleField { name }) => {
 | 
			
		||||
                let parent_ty = Bundle::from_canonical(parent.canonical_ty());
 | 
			
		||||
                let field = parent_ty
 | 
			
		||||
                    .field_by_name(name)
 | 
			
		||||
                    .expect("field name is known to be a valid field of parent type");
 | 
			
		||||
                field.ty
 | 
			
		||||
            }
 | 
			
		||||
            &Self::ArrayElement(TargetPathArrayElement { index }) => {
 | 
			
		||||
                let parent_ty = Array::<CanonicalType>::from_canonical(parent.canonical_ty());
 | 
			
		||||
                assert!(index < parent_ty.len());
 | 
			
		||||
                parent_ty.element()
 | 
			
		||||
            }
 | 
			
		||||
            Self::DynArrayElement(_) => {
 | 
			
		||||
                let parent_ty = Array::<CanonicalType>::from_canonical(parent.canonical_ty());
 | 
			
		||||
                parent_ty.element()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn flow(&self, parent: Interned<Target>) -> Flow {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::BundleField(v) => {
 | 
			
		||||
                let parent_ty = Bundle::from_canonical(parent.canonical_ty());
 | 
			
		||||
                let field = parent_ty
 | 
			
		||||
                    .field_by_name(v.name)
 | 
			
		||||
                    .expect("field name is known to be a valid field of parent type");
 | 
			
		||||
                parent.flow().flip_if(field.flipped)
 | 
			
		||||
            }
 | 
			
		||||
            Self::ArrayElement(_) => parent.flow(),
 | 
			
		||||
            Self::DynArrayElement(_) => parent.flow(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_static(&self) -> bool {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::BundleField(_) | Self::ArrayElement(_) => true,
 | 
			
		||||
            Self::DynArrayElement(_) => false,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
macro_rules! impl_target_base {
 | 
			
		||||
    (
 | 
			
		||||
        $(#[$enum_meta:meta])*
 | 
			
		||||
        $enum_vis:vis enum $TargetBase:ident {
 | 
			
		||||
            $(
 | 
			
		||||
                #[is = $is_fn:ident]
 | 
			
		||||
                #[to = $to_fn:ident]
 | 
			
		||||
                $(#[$variant_meta:meta])*
 | 
			
		||||
                $Variant:ident($VariantTy:ty),
 | 
			
		||||
            )*
 | 
			
		||||
        }
 | 
			
		||||
    ) => {
 | 
			
		||||
        $(#[$enum_meta])*
 | 
			
		||||
        $enum_vis enum $TargetBase {
 | 
			
		||||
            $(
 | 
			
		||||
                $(#[$variant_meta])*
 | 
			
		||||
                $Variant($VariantTy),
 | 
			
		||||
            )*
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl fmt::Debug for $TargetBase {
 | 
			
		||||
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
                match self {
 | 
			
		||||
                    $(Self::$Variant(v) => v.fmt(f),)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        $(
 | 
			
		||||
        impl From<$VariantTy> for $TargetBase {
 | 
			
		||||
            fn from(value: $VariantTy) -> Self {
 | 
			
		||||
                Self::$Variant(value)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl From<$VariantTy> for Target {
 | 
			
		||||
            fn from(value: $VariantTy) -> Self {
 | 
			
		||||
                $TargetBase::$Variant(value).into()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        )*
 | 
			
		||||
 | 
			
		||||
        impl $TargetBase {
 | 
			
		||||
            $(
 | 
			
		||||
            $enum_vis fn $is_fn(&self) -> bool {
 | 
			
		||||
                self.$to_fn().is_some()
 | 
			
		||||
            }
 | 
			
		||||
            $enum_vis fn $to_fn(&self) -> Option<&$VariantTy> {
 | 
			
		||||
                if let Self::$Variant(retval) = self {
 | 
			
		||||
                    Some(retval)
 | 
			
		||||
                } else {
 | 
			
		||||
                    None
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            )*
 | 
			
		||||
            $enum_vis fn must_connect_to(&self) -> bool {
 | 
			
		||||
                match self {
 | 
			
		||||
                    $(Self::$Variant(v) => v.must_connect_to(),)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            $enum_vis fn flow(&self) -> Flow {
 | 
			
		||||
                match self {
 | 
			
		||||
                    $(Self::$Variant(v) => v.flow(),)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            $enum_vis fn source_location(&self) -> SourceLocation {
 | 
			
		||||
                match self {
 | 
			
		||||
                    $(Self::$Variant(v) => v.source_location(),)*
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl_target_base! {
 | 
			
		||||
    #[derive(Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
    pub enum TargetBase {
 | 
			
		||||
        #[is = is_module_io]
 | 
			
		||||
        #[to = module_io]
 | 
			
		||||
        ModuleIO(ModuleIO<CanonicalType>),
 | 
			
		||||
        #[is = is_mem_port]
 | 
			
		||||
        #[to = mem_port]
 | 
			
		||||
        MemPort(MemPort<DynPortType>),
 | 
			
		||||
        #[is = is_reg]
 | 
			
		||||
        #[to = reg]
 | 
			
		||||
        Reg(Reg<CanonicalType>),
 | 
			
		||||
        #[is = is_wire]
 | 
			
		||||
        #[to = wire]
 | 
			
		||||
        Wire(Wire<CanonicalType>),
 | 
			
		||||
        #[is = is_instance]
 | 
			
		||||
        #[to = instance]
 | 
			
		||||
        Instance(Instance<Bundle>),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetBase {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        write!(f, "{:?}", self.target_name())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl TargetBase {
 | 
			
		||||
    pub fn target_name(&self) -> TargetName {
 | 
			
		||||
        match self {
 | 
			
		||||
            TargetBase::ModuleIO(v) => TargetName(v.scoped_name(), None),
 | 
			
		||||
            TargetBase::MemPort(v) => TargetName(v.mem_name(), Some(v.port_name())),
 | 
			
		||||
            TargetBase::Reg(v) => TargetName(v.scoped_name(), None),
 | 
			
		||||
            TargetBase::Wire(v) => TargetName(v.scoped_name(), None),
 | 
			
		||||
            TargetBase::Instance(v) => TargetName(v.scoped_name(), None),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical_ty(&self) -> CanonicalType {
 | 
			
		||||
        match self {
 | 
			
		||||
            TargetBase::ModuleIO(v) => v.ty(),
 | 
			
		||||
            TargetBase::MemPort(v) => v.ty().canonical(),
 | 
			
		||||
            TargetBase::Reg(v) => v.ty(),
 | 
			
		||||
            TargetBase::Wire(v) => v.ty(),
 | 
			
		||||
            TargetBase::Instance(v) => v.ty().canonical(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub struct TargetChild {
 | 
			
		||||
    parent: Interned<Target>,
 | 
			
		||||
    path_element: Interned<TargetPathElement>,
 | 
			
		||||
    canonical_ty: CanonicalType,
 | 
			
		||||
    flow: Flow,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for TargetChild {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let Self {
 | 
			
		||||
            parent,
 | 
			
		||||
            path_element,
 | 
			
		||||
            canonical_ty: _,
 | 
			
		||||
            flow: _,
 | 
			
		||||
        } = self;
 | 
			
		||||
        parent.fmt(f)?;
 | 
			
		||||
        fmt::Display::fmt(path_element, f)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for TargetChild {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let Self {
 | 
			
		||||
            parent,
 | 
			
		||||
            path_element,
 | 
			
		||||
            canonical_ty: _,
 | 
			
		||||
            flow: _,
 | 
			
		||||
        } = self;
 | 
			
		||||
        parent.fmt(f)?;
 | 
			
		||||
        path_element.fmt(f)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl TargetChild {
 | 
			
		||||
    pub fn new(parent: Interned<Target>, path_element: Interned<TargetPathElement>) -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            parent,
 | 
			
		||||
            path_element,
 | 
			
		||||
            canonical_ty: path_element.canonical_ty(parent),
 | 
			
		||||
            flow: path_element.flow(parent),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn parent(self) -> Interned<Target> {
 | 
			
		||||
        self.parent
 | 
			
		||||
    }
 | 
			
		||||
    pub fn path_element(self) -> Interned<TargetPathElement> {
 | 
			
		||||
        self.path_element
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical_ty(self) -> CanonicalType {
 | 
			
		||||
        self.canonical_ty
 | 
			
		||||
    }
 | 
			
		||||
    pub fn flow(self) -> Flow {
 | 
			
		||||
        self.flow
 | 
			
		||||
    }
 | 
			
		||||
    pub fn bundle_field(self) -> Option<BundleField> {
 | 
			
		||||
        if let TargetPathElement::BundleField(TargetPathBundleField { name }) = *self.path_element {
 | 
			
		||||
            let parent_ty = Bundle::from_canonical(self.parent.canonical_ty());
 | 
			
		||||
            Some(
 | 
			
		||||
                parent_ty
 | 
			
		||||
                    .field_by_name(name)
 | 
			
		||||
                    .expect("field name known to be a valid field of parent"),
 | 
			
		||||
            )
 | 
			
		||||
        } else {
 | 
			
		||||
            None
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
pub enum Target {
 | 
			
		||||
    Base(Interned<TargetBase>),
 | 
			
		||||
    Child(TargetChild),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<TargetBase> for Target {
 | 
			
		||||
    fn from(value: TargetBase) -> Self {
 | 
			
		||||
        Self::Base(Intern::intern_sized(value))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<TargetChild> for Target {
 | 
			
		||||
    fn from(value: TargetChild) -> Self {
 | 
			
		||||
        Self::Child(value)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl From<Interned<TargetBase>> for Target {
 | 
			
		||||
    fn from(value: Interned<TargetBase>) -> Self {
 | 
			
		||||
        Self::Base(value)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Target {
 | 
			
		||||
    pub fn base(&self) -> Interned<TargetBase> {
 | 
			
		||||
        let mut target = self;
 | 
			
		||||
        loop {
 | 
			
		||||
            match target {
 | 
			
		||||
                Self::Base(v) => break *v,
 | 
			
		||||
                Self::Child(v) => target = &v.parent,
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn child(&self) -> Option<TargetChild> {
 | 
			
		||||
        match *self {
 | 
			
		||||
            Target::Base(_) => None,
 | 
			
		||||
            Target::Child(v) => Some(v),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn is_static(&self) -> bool {
 | 
			
		||||
        let mut target = self;
 | 
			
		||||
        loop {
 | 
			
		||||
            match target {
 | 
			
		||||
                Self::Base(_) => return true,
 | 
			
		||||
                Self::Child(v) if !v.path_element().is_static() => return false,
 | 
			
		||||
                Self::Child(v) => target = &v.parent,
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    #[must_use]
 | 
			
		||||
    pub fn join(&self, path_element: Interned<TargetPathElement>) -> Self {
 | 
			
		||||
        TargetChild::new(self.intern(), path_element).into()
 | 
			
		||||
    }
 | 
			
		||||
    pub fn flow(&self) -> Flow {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::Base(v) => v.flow(),
 | 
			
		||||
            Self::Child(v) => v.flow(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical_ty(&self) -> CanonicalType {
 | 
			
		||||
        match self {
 | 
			
		||||
            Target::Base(v) => v.canonical_ty(),
 | 
			
		||||
            Target::Child(v) => v.canonical_ty(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for Target {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::Base(v) => v.fmt(f),
 | 
			
		||||
            Self::Child(v) => v.fmt(f),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for Target {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::Base(v) => v.fmt(f),
 | 
			
		||||
            Self::Child(v) => v.fmt(f),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait GetTarget {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl GetTarget for bool {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>> {
 | 
			
		||||
        None
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + GetTarget + Send + Sync + 'static> GetTarget for Interned<T> {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>> {
 | 
			
		||||
        T::target(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + GetTarget> GetTarget for &'_ T {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>> {
 | 
			
		||||
        T::target(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + GetTarget> GetTarget for &'_ mut T {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>> {
 | 
			
		||||
        T::target(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + GetTarget> GetTarget for Box<T> {
 | 
			
		||||
    fn target(&self) -> Option<Interned<Target>> {
 | 
			
		||||
        T::target(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -22,6 +22,179 @@ use std::{
 | 
			
		|||
 | 
			
		||||
pub mod type_map;
 | 
			
		||||
 | 
			
		||||
pub trait LazyInternedTrait<T: ?Sized + Send + Sync + 'static>: Send + Sync + Any {
 | 
			
		||||
    fn get(&self) -> Interned<T>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static, F: ?Sized + Fn() -> Interned<T> + Send + Sync + Any>
 | 
			
		||||
    LazyInternedTrait<T> for F
 | 
			
		||||
{
 | 
			
		||||
    fn get(&self) -> Interned<T> {
 | 
			
		||||
        self()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[repr(transparent)]
 | 
			
		||||
pub struct LazyInternedFn<T: ?Sized + Send + Sync + 'static>(pub &'static dyn LazyInternedTrait<T>);
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> Copy for LazyInternedFn<T> {}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> Clone for LazyInternedFn<T> {
 | 
			
		||||
    fn clone(&self) -> Self {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> Hash for LazyInternedFn<T> {
 | 
			
		||||
    fn hash<H: Hasher>(&self, state: &mut H) {
 | 
			
		||||
        self.0.get_ptr_eq_with_type_id().hash(state);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> Eq for LazyInternedFn<T> {}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> PartialEq for LazyInternedFn<T> {
 | 
			
		||||
    fn eq(&self, other: &Self) -> bool {
 | 
			
		||||
        self.0.get_ptr_eq_with_type_id() == other.0.get_ptr_eq_with_type_id()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub enum LazyInterned<T: ?Sized + Send + Sync + 'static> {
 | 
			
		||||
    Interned(Interned<T>),
 | 
			
		||||
    Lazy(LazyInternedFn<T>),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Send + Sync + 'static> LazyInterned<T> {
 | 
			
		||||
    pub const fn new_lazy(v: &'static dyn LazyInternedTrait<T>) -> Self {
 | 
			
		||||
        Self::Lazy(LazyInternedFn(v))
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static> Clone for LazyInterned<T> {
 | 
			
		||||
    fn clone(&self) -> Self {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static> Copy for LazyInterned<T> {}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> Deref for LazyInterned<T> {
 | 
			
		||||
    type Target = T;
 | 
			
		||||
 | 
			
		||||
    fn deref(&self) -> &Self::Target {
 | 
			
		||||
        Interned::into_inner(self.interned())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> Eq for LazyInterned<T> where Interned<T>: Eq {}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> PartialEq for LazyInterned<T>
 | 
			
		||||
where
 | 
			
		||||
    Interned<T>: PartialEq,
 | 
			
		||||
{
 | 
			
		||||
    fn eq(&self, other: &Self) -> bool {
 | 
			
		||||
        self.interned() == other.interned()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> Ord for LazyInterned<T>
 | 
			
		||||
where
 | 
			
		||||
    Interned<T>: Ord,
 | 
			
		||||
{
 | 
			
		||||
    fn cmp(&self, other: &Self) -> Ordering {
 | 
			
		||||
        self.interned().cmp(&other.interned())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> PartialOrd for LazyInterned<T>
 | 
			
		||||
where
 | 
			
		||||
    Interned<T>: PartialOrd,
 | 
			
		||||
{
 | 
			
		||||
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
 | 
			
		||||
        self.interned().partial_cmp(&other.interned())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static + Intern> Hash for LazyInterned<T>
 | 
			
		||||
where
 | 
			
		||||
    Interned<T>: Hash,
 | 
			
		||||
{
 | 
			
		||||
    fn hash<H: Hasher>(&self, state: &mut H) {
 | 
			
		||||
        self.interned().hash(state);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized + Sync + Send + 'static> LazyInterned<T> {
 | 
			
		||||
    pub fn interned(self) -> Interned<T>
 | 
			
		||||
    where
 | 
			
		||||
        T: Intern,
 | 
			
		||||
    {
 | 
			
		||||
        struct MemoizeInterned<T: ?Sized + Intern>(PhantomData<T>);
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> Hash for MemoizeInterned<T> {
 | 
			
		||||
            fn hash<H: Hasher>(&self, _state: &mut H) {}
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> PartialEq for MemoizeInterned<T> {
 | 
			
		||||
            fn eq(&self, _other: &Self) -> bool {
 | 
			
		||||
                true
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> Eq for MemoizeInterned<T> {}
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> Clone for MemoizeInterned<T> {
 | 
			
		||||
            fn clone(&self) -> Self {
 | 
			
		||||
                *self
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> Copy for MemoizeInterned<T> {}
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + Intern> MemoizeGeneric for MemoizeInterned<T> {
 | 
			
		||||
            type InputRef<'a> = LazyInternedFn<T>;
 | 
			
		||||
 | 
			
		||||
            type InputOwned = LazyInternedFn<T>;
 | 
			
		||||
 | 
			
		||||
            type InputCow<'a> = LazyInternedFn<T>;
 | 
			
		||||
 | 
			
		||||
            type Output = Interned<T>;
 | 
			
		||||
 | 
			
		||||
            fn input_eq(a: Self::InputRef<'_>, b: Self::InputRef<'_>) -> bool {
 | 
			
		||||
                a == b
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn input_borrow(input: &Self::InputOwned) -> Self::InputRef<'_> {
 | 
			
		||||
                *input
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn input_cow_into_owned(input: Self::InputCow<'_>) -> Self::InputOwned {
 | 
			
		||||
                input
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn input_cow_borrow<'a>(input: &'a Self::InputCow<'_>) -> Self::InputRef<'a> {
 | 
			
		||||
                *input
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn input_cow_from_owned<'a>(input: Self::InputOwned) -> Self::InputCow<'a> {
 | 
			
		||||
                input
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn input_cow_from_ref(input: Self::InputRef<'_>) -> Self::InputCow<'_> {
 | 
			
		||||
                input
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn inner(self, input: Self::InputRef<'_>) -> Self::Output {
 | 
			
		||||
                input.0.get()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        match self {
 | 
			
		||||
            Self::Interned(retval) => retval,
 | 
			
		||||
            Self::Lazy(retval) => MemoizeInterned(PhantomData).get(retval),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait InternedCompare {
 | 
			
		||||
    type InternedCompareKey: Ord + Hash;
 | 
			
		||||
    fn interned_compare_key_ref(this: &Self) -> Self::InternedCompareKey;
 | 
			
		||||
| 
						 | 
				
			
			@ -471,6 +644,12 @@ pub struct Interned<T: ?Sized + 'static + Send + Sync, C: InternContext = Global
 | 
			
		|||
 | 
			
		||||
macro_rules! forward_fmt_trait {
 | 
			
		||||
    ($Tr:ident) => {
 | 
			
		||||
        impl<T: ?Sized + Intern + fmt::$Tr> fmt::$Tr for LazyInterned<T> {
 | 
			
		||||
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
                T::fmt(&**self, f)
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + 'static + Send + Sync + fmt::$Tr, C: InternContext> fmt::$Tr
 | 
			
		||||
            for Interned<T, C>
 | 
			
		||||
        {
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -12,7 +12,6 @@ extern crate self as fayalite;
 | 
			
		|||
pub use std as __std;
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
#[doc(alias = "hdl")]
 | 
			
		||||
/// The `#[hdl_module]` attribute is applied to a Rust function so that that function creates
 | 
			
		||||
/// a [`Module`][`::fayalite::module::Module`] when called.
 | 
			
		||||
/// In the function body it will implicitly create a
 | 
			
		||||
| 
						 | 
				
			
			@ -21,9 +20,17 @@ pub use std as __std;
 | 
			
		|||
/// See [Fayalite Modules][crate::_docs::modules]
 | 
			
		||||
pub use fayalite_proc_macros::hdl_module;
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use fayalite_proc_macros::hdl;
 | 
			
		||||
 | 
			
		||||
/// struct used as a placeholder when applying defaults
 | 
			
		||||
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Default)]
 | 
			
		||||
pub struct __;
 | 
			
		||||
 | 
			
		||||
#[cfg(feature = "unstable-doc")]
 | 
			
		||||
pub mod _docs;
 | 
			
		||||
 | 
			
		||||
// FIXME: finish
 | 
			
		||||
pub mod annotations;
 | 
			
		||||
pub mod array;
 | 
			
		||||
pub mod bundle;
 | 
			
		||||
| 
						 | 
				
			
			@ -41,5 +48,6 @@ pub mod reset;
 | 
			
		|||
pub mod source_location;
 | 
			
		||||
pub mod ty;
 | 
			
		||||
pub mod util;
 | 
			
		||||
pub mod valueless;
 | 
			
		||||
//pub mod valueless;
 | 
			
		||||
pub mod prelude;
 | 
			
		||||
pub mod wire;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -4,15 +4,16 @@
 | 
			
		|||
#![allow(clippy::multiple_bound_locations)]
 | 
			
		||||
use crate::{
 | 
			
		||||
    annotations::{Annotation, IntoAnnotations, TargetedAnnotation},
 | 
			
		||||
    array::{Array, ArrayType, ArrayTypeTrait, ValueArrayOrSlice},
 | 
			
		||||
    bundle::{BundleType, BundleValue, DynBundle, DynBundleType},
 | 
			
		||||
    clock::{Clock, ClockType},
 | 
			
		||||
    expr::{Expr, ExprEnum, ExprTrait, Flow, ToExpr},
 | 
			
		||||
    int::{DynUInt, DynUIntType, UInt, UIntType},
 | 
			
		||||
    array::{Array, ArrayType},
 | 
			
		||||
    bundle::{Bundle, BundleType},
 | 
			
		||||
    clock::Clock,
 | 
			
		||||
    expr::{Expr, Flow, ToExpr, ToLiteralBits},
 | 
			
		||||
    hdl,
 | 
			
		||||
    int::{Bool, DynSize, Size, UInt, UIntType},
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    module::ScopedNameId,
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{AsMask, DynCanonicalType, DynCanonicalValue, DynType, Type, Value},
 | 
			
		||||
    ty::{AsMask, CanonicalType, Type},
 | 
			
		||||
    util::DebugAsDisplay,
 | 
			
		||||
};
 | 
			
		||||
use bitvec::slice::BitSlice;
 | 
			
		||||
| 
						 | 
				
			
			@ -20,37 +21,37 @@ use std::{
 | 
			
		|||
    cell::RefCell,
 | 
			
		||||
    fmt,
 | 
			
		||||
    hash::{Hash, Hasher},
 | 
			
		||||
    marker::PhantomData,
 | 
			
		||||
    num::NonZeroU32,
 | 
			
		||||
    rc::Rc,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
pub struct ReadStruct<Element> {
 | 
			
		||||
    pub addr: DynUInt,
 | 
			
		||||
    pub en: UInt<1>,
 | 
			
		||||
#[hdl]
 | 
			
		||||
pub struct ReadStruct<Element, AddrWidth: Size> {
 | 
			
		||||
    pub addr: UIntType<AddrWidth>,
 | 
			
		||||
    pub en: Bool,
 | 
			
		||||
    pub clk: Clock,
 | 
			
		||||
    #[hdl(flip)]
 | 
			
		||||
    pub data: Element,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
pub struct WriteStruct<Element: Value> {
 | 
			
		||||
    pub addr: DynUInt,
 | 
			
		||||
    pub en: UInt<1>,
 | 
			
		||||
#[hdl]
 | 
			
		||||
pub struct WriteStruct<Element, AddrWidth: Size> {
 | 
			
		||||
    pub addr: UIntType<AddrWidth>,
 | 
			
		||||
    pub en: Bool,
 | 
			
		||||
    pub clk: Clock,
 | 
			
		||||
    pub data: Element,
 | 
			
		||||
    pub mask: AsMask<Element>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[allow(clippy::multiple_bound_locations)]
 | 
			
		||||
#[derive(Value, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
pub struct ReadWriteStruct<Element: Value> {
 | 
			
		||||
    pub addr: DynUInt,
 | 
			
		||||
    pub en: UInt<1>,
 | 
			
		||||
#[hdl]
 | 
			
		||||
pub struct ReadWriteStruct<Element, AddrWidth: Size> {
 | 
			
		||||
    pub addr: UIntType<AddrWidth>,
 | 
			
		||||
    pub en: Bool,
 | 
			
		||||
    pub clk: Clock,
 | 
			
		||||
    #[hdl(flip)]
 | 
			
		||||
    pub rdata: Element,
 | 
			
		||||
    pub wmode: UInt<1>,
 | 
			
		||||
    pub wmode: Bool,
 | 
			
		||||
    pub wdata: Element,
 | 
			
		||||
    pub wmask: AsMask<Element>,
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -63,11 +64,10 @@ pub trait PortType:
 | 
			
		|||
    sealed::Sealed + Clone + Eq + Hash + fmt::Debug + Send + Sync + 'static
 | 
			
		||||
{
 | 
			
		||||
    type PortKindTy: Copy + Eq + Hash + fmt::Debug + Send + Sync + 'static;
 | 
			
		||||
    type PortType: BundleType<Value = Self::PortValue>;
 | 
			
		||||
    type PortValue: BundleValue<Type = Self::PortType>;
 | 
			
		||||
    type Port: BundleType;
 | 
			
		||||
    fn port_kind(port_kind: Self::PortKindTy) -> PortKind;
 | 
			
		||||
    fn from_port_kind(port_kind: PortKind) -> Self::PortKindTy;
 | 
			
		||||
    fn port_ty(port: &MemPort<Self>) -> Self::PortType;
 | 
			
		||||
    fn port_ty(port: &MemPort<Self>) -> Self::Port;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
| 
						 | 
				
			
			@ -79,8 +79,7 @@ impl sealed::Sealed for DynPortType {}
 | 
			
		|||
 | 
			
		||||
impl PortType for DynPortType {
 | 
			
		||||
    type PortKindTy = PortKind;
 | 
			
		||||
    type PortType = DynBundleType;
 | 
			
		||||
    type PortValue = DynBundle;
 | 
			
		||||
    type Port = Bundle;
 | 
			
		||||
 | 
			
		||||
    fn port_kind(port_kind: Self::PortKindTy) -> PortKind {
 | 
			
		||||
        port_kind
 | 
			
		||||
| 
						 | 
				
			
			@ -90,41 +89,38 @@ impl PortType for DynPortType {
 | 
			
		|||
        port_kind
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn port_ty(port: &MemPort<Self>) -> Self::PortType {
 | 
			
		||||
        match port.port_kind {
 | 
			
		||||
            PortKind::ReadOnly => MemPort::<ReadStruct<DynCanonicalValue>>::from_canonical(*port)
 | 
			
		||||
                .ty()
 | 
			
		||||
                .canonical(),
 | 
			
		||||
            PortKind::WriteOnly => MemPort::<WriteStruct<DynCanonicalValue>>::from_canonical(*port)
 | 
			
		||||
                .ty()
 | 
			
		||||
                .canonical(),
 | 
			
		||||
            PortKind::ReadWrite => {
 | 
			
		||||
                MemPort::<ReadWriteStruct<DynCanonicalValue>>::from_canonical(*port)
 | 
			
		||||
    fn port_ty(port: &MemPort<Self>) -> Self::Port {
 | 
			
		||||
        Bundle::new(match port.port_kind {
 | 
			
		||||
            PortKind::ReadOnly => {
 | 
			
		||||
                MemPort::<ReadStruct<CanonicalType, DynSize>>::from_canonical(*port)
 | 
			
		||||
                    .ty()
 | 
			
		||||
                    .canonical()
 | 
			
		||||
                    .fields()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
            PortKind::WriteOnly => {
 | 
			
		||||
                MemPort::<WriteStruct<CanonicalType, DynSize>>::from_canonical(*port)
 | 
			
		||||
                    .ty()
 | 
			
		||||
                    .fields()
 | 
			
		||||
            }
 | 
			
		||||
            PortKind::ReadWrite => {
 | 
			
		||||
                MemPort::<ReadWriteStruct<CanonicalType, DynSize>>::from_canonical(*port)
 | 
			
		||||
                    .ty()
 | 
			
		||||
                    .fields()
 | 
			
		||||
            }
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub trait PortStruct:
 | 
			
		||||
    BundleValue
 | 
			
		||||
    + sealed::Sealed
 | 
			
		||||
    + PortType<PortKindTy = (), PortType = <Self as ToExpr>::Type, PortValue = Self>
 | 
			
		||||
where
 | 
			
		||||
    Self::Type: BundleType<Value = Self>,
 | 
			
		||||
{
 | 
			
		||||
    type Element: Value<Type = Self::ElementType>;
 | 
			
		||||
    type ElementType: Type<Value = Self::Element>;
 | 
			
		||||
pub trait PortStruct: BundleType + sealed::Sealed + PortType<PortKindTy = (), Port = Self> {
 | 
			
		||||
    type Element: Type;
 | 
			
		||||
    const PORT_KIND: PortKind;
 | 
			
		||||
 | 
			
		||||
    fn addr(this: Expr<Self>) -> Expr<DynUInt>;
 | 
			
		||||
    fn en(this: Expr<Self>) -> Expr<UInt<1>>;
 | 
			
		||||
    fn addr(this: Expr<Self>) -> Expr<UInt>;
 | 
			
		||||
    fn en(this: Expr<Self>) -> Expr<Bool>;
 | 
			
		||||
    fn clk(this: Expr<Self>) -> Expr<Clock>;
 | 
			
		||||
    fn rdata(this: Expr<Self>) -> Option<Expr<Self::Element>>;
 | 
			
		||||
    fn wdata(this: Expr<Self>) -> Option<Expr<Self::Element>>;
 | 
			
		||||
    fn wmask(this: Expr<Self>) -> Option<Expr<AsMask<Self::Element>>>;
 | 
			
		||||
    fn wmode(this: Expr<Self>) -> Expr<UInt<1>>;
 | 
			
		||||
    fn wmode(this: Expr<Self>) -> Expr<Bool>;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
macro_rules! impl_port_struct {
 | 
			
		||||
| 
						 | 
				
			
			@ -137,19 +133,11 @@ macro_rules! impl_port_struct {
 | 
			
		|||
            $($body:tt)*
 | 
			
		||||
        }
 | 
			
		||||
    ) => {
 | 
			
		||||
        impl<$Element: Value> sealed::Sealed for $Struct<$Element>
 | 
			
		||||
        where
 | 
			
		||||
            $Element::Type: Type<Value = $Element>,
 | 
			
		||||
        {
 | 
			
		||||
        }
 | 
			
		||||
        impl<$Element: Type> sealed::Sealed for $Struct<$Element, DynSize> {}
 | 
			
		||||
 | 
			
		||||
        impl<$Element: Value> PortType for $Struct<$Element>
 | 
			
		||||
        where
 | 
			
		||||
            $Element::Type: Type<Value = $Element>,
 | 
			
		||||
        {
 | 
			
		||||
        impl<$Element: Type> PortType for $Struct<$Element, DynSize> {
 | 
			
		||||
            type PortKindTy = ();
 | 
			
		||||
            type PortType = <Self as ToExpr>::Type;
 | 
			
		||||
            type PortValue = Self;
 | 
			
		||||
            type Port = Self;
 | 
			
		||||
 | 
			
		||||
            fn port_kind(_port_kind: Self::PortKindTy) -> PortKind {
 | 
			
		||||
                Self::PORT_KIND
 | 
			
		||||
| 
						 | 
				
			
			@ -164,18 +152,14 @@ macro_rules! impl_port_struct {
 | 
			
		|||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<$Element: Value> PortStruct for $Struct<$Element>
 | 
			
		||||
        where
 | 
			
		||||
            $Element::Type: Type<Value = $Element>,
 | 
			
		||||
        {
 | 
			
		||||
        impl<$Element: Type> PortStruct for $Struct<$Element, DynSize> {
 | 
			
		||||
            type Element = $Element;
 | 
			
		||||
            type ElementType = $Element::Type;
 | 
			
		||||
 | 
			
		||||
            fn addr(this: Expr<Self>) -> Expr<DynUInt> {
 | 
			
		||||
            fn addr(this: Expr<Self>) -> Expr<UInt> {
 | 
			
		||||
                this.addr
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            fn en(this: Expr<Self>) -> Expr<UInt<1>> {
 | 
			
		||||
            fn en(this: Expr<Self>) -> Expr<Bool> {
 | 
			
		||||
                this.en
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -190,14 +174,9 @@ macro_rules! impl_port_struct {
 | 
			
		|||
 | 
			
		||||
impl_port_struct! {
 | 
			
		||||
    impl<Element> _ for ReadStruct {
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> <MemPort<Self> as ToExpr>::Type {
 | 
			
		||||
            type T<V> = <V as ToExpr>::Type;
 | 
			
		||||
            T::<Self> {
 | 
			
		||||
                addr: port.addr_type,
 | 
			
		||||
                en: UIntType::new(),
 | 
			
		||||
                clk: ClockType,
 | 
			
		||||
                data: Element::Type::from_dyn_canonical_type(port.mem_element_type),
 | 
			
		||||
            }
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> Self {
 | 
			
		||||
            let element = Element::from_canonical(port.mem_element_type);
 | 
			
		||||
            ReadStruct[element][port.addr_type.width()]
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const PORT_KIND: PortKind = PortKind::ReadOnly;
 | 
			
		||||
| 
						 | 
				
			
			@ -214,7 +193,7 @@ impl_port_struct! {
 | 
			
		|||
            None
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        fn wmode(_this: Expr<Self>) -> Expr<UInt<1>> {
 | 
			
		||||
        fn wmode(_this: Expr<Self>) -> Expr<Bool> {
 | 
			
		||||
            false.to_expr()
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -222,16 +201,9 @@ impl_port_struct! {
 | 
			
		|||
 | 
			
		||||
impl_port_struct! {
 | 
			
		||||
    impl<Element> _ for WriteStruct {
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> <MemPort<Self> as ToExpr>::Type {
 | 
			
		||||
            type T<V> = <V as ToExpr>::Type;
 | 
			
		||||
            let element_ty = Element::Type::from_dyn_canonical_type(port.mem_element_type);
 | 
			
		||||
            T::<Self> {
 | 
			
		||||
                addr: port.addr_type,
 | 
			
		||||
                en: UIntType::new(),
 | 
			
		||||
                clk: ClockType,
 | 
			
		||||
                mask: element_ty.mask_type(),
 | 
			
		||||
                data: element_ty,
 | 
			
		||||
            }
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> Self {
 | 
			
		||||
            let element = Element::from_canonical(port.mem_element_type);
 | 
			
		||||
            WriteStruct[element][port.addr_type.width()]
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const PORT_KIND: PortKind = PortKind::WriteOnly;
 | 
			
		||||
| 
						 | 
				
			
			@ -248,7 +220,7 @@ impl_port_struct! {
 | 
			
		|||
            Some(this.mask)
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        fn wmode(_this: Expr<Self>) -> Expr<UInt<1>> {
 | 
			
		||||
        fn wmode(_this: Expr<Self>) -> Expr<Bool> {
 | 
			
		||||
            true.to_expr()
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -256,18 +228,9 @@ impl_port_struct! {
 | 
			
		|||
 | 
			
		||||
impl_port_struct! {
 | 
			
		||||
    impl<Element> _ for ReadWriteStruct {
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> <MemPort<Self> as ToExpr>::Type {
 | 
			
		||||
            type T<V> = <V as ToExpr>::Type;
 | 
			
		||||
            let element_ty = Element::Type::from_dyn_canonical_type(port.mem_element_type);
 | 
			
		||||
            T::<Self> {
 | 
			
		||||
                addr: port.addr_type,
 | 
			
		||||
                en: UIntType::new(),
 | 
			
		||||
                clk: ClockType,
 | 
			
		||||
                rdata: element_ty.clone(),
 | 
			
		||||
                wmode: UIntType::new(),
 | 
			
		||||
                wmask: element_ty.mask_type(),
 | 
			
		||||
                wdata: element_ty,
 | 
			
		||||
            }
 | 
			
		||||
        fn port_ty(port: &MemPort<Self>) -> Self {
 | 
			
		||||
            let element = Element::from_canonical(port.mem_element_type);
 | 
			
		||||
            ReadWriteStruct[element][port.addr_type.width()]
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const PORT_KIND: PortKind = PortKind::ReadWrite;
 | 
			
		||||
| 
						 | 
				
			
			@ -284,7 +247,7 @@ impl_port_struct! {
 | 
			
		|||
            Some(this.wmask)
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        fn wmode(this: Expr<Self>) -> Expr<UInt<1>> {
 | 
			
		||||
        fn wmode(this: Expr<Self>) -> Expr<Bool> {
 | 
			
		||||
            this.wmode
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -382,42 +345,31 @@ pub struct MemPort<T: PortType> {
 | 
			
		|||
    source_location: SourceLocation,
 | 
			
		||||
    port_kind: T::PortKindTy,
 | 
			
		||||
    port_index: usize,
 | 
			
		||||
    addr_type: DynUIntType,
 | 
			
		||||
    mem_element_type: Interned<dyn DynCanonicalType>,
 | 
			
		||||
    addr_type: UInt,
 | 
			
		||||
    mem_element_type: CanonicalType,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: PortType> fmt::Debug for MemPort<T> {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let Self {
 | 
			
		||||
            mem_name,
 | 
			
		||||
            source_location: _,
 | 
			
		||||
            port_kind: _,
 | 
			
		||||
            port_index: _,
 | 
			
		||||
            addr_type,
 | 
			
		||||
            mem_element_type,
 | 
			
		||||
        } = self;
 | 
			
		||||
        f.debug_struct("MemPort")
 | 
			
		||||
            .field("mem_name", mem_name)
 | 
			
		||||
            .field("port_name", &self.port_name())
 | 
			
		||||
            .field("addr_type", addr_type)
 | 
			
		||||
            .field("mem_element_type", mem_element_type)
 | 
			
		||||
            .finish_non_exhaustive()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: PortType> ToExpr for MemPort<T> {
 | 
			
		||||
    type Type = T::PortType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        T::port_ty(self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        Expr::new_unchecked(self.expr_enum())
 | 
			
		||||
        f.write_str("MemPort(")?;
 | 
			
		||||
        self.mem_name.fmt(f)?;
 | 
			
		||||
        f.write_str(".")?;
 | 
			
		||||
        self.port_name().fmt(f)?;
 | 
			
		||||
        f.write_str(": ")?;
 | 
			
		||||
        match self.port_kind() {
 | 
			
		||||
            PortKind::ReadOnly => f.write_str("ReadStruct<")?,
 | 
			
		||||
            PortKind::WriteOnly => f.write_str("WriteStruct<")?,
 | 
			
		||||
            PortKind::ReadWrite => f.write_str("ReadWriteStruct<")?,
 | 
			
		||||
        }
 | 
			
		||||
        self.mem_element_type.fmt(f)?;
 | 
			
		||||
        f.write_str(">)")
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: PortType> MemPort<T> {
 | 
			
		||||
    pub fn ty(&self) -> T::Port {
 | 
			
		||||
        T::port_ty(self)
 | 
			
		||||
    }
 | 
			
		||||
    pub fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        self.source_location
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -436,10 +388,10 @@ impl<T: PortType> MemPort<T> {
 | 
			
		|||
            index: self.port_index,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn mem_element_type(&self) -> Interned<dyn DynCanonicalType> {
 | 
			
		||||
    pub fn mem_element_type(&self) -> CanonicalType {
 | 
			
		||||
        self.mem_element_type
 | 
			
		||||
    }
 | 
			
		||||
    pub fn addr_type(&self) -> DynUIntType {
 | 
			
		||||
    pub fn addr_type(&self) -> UInt {
 | 
			
		||||
        self.addr_type
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical(&self) -> MemPort<DynPortType> {
 | 
			
		||||
| 
						 | 
				
			
			@ -463,7 +415,6 @@ impl<T: PortType> MemPort<T> {
 | 
			
		|||
    pub fn from_canonical(port: MemPort<DynPortType>) -> Self
 | 
			
		||||
    where
 | 
			
		||||
        T: PortStruct,
 | 
			
		||||
        T::Type: BundleType<Value = T>,
 | 
			
		||||
    {
 | 
			
		||||
        let MemPort {
 | 
			
		||||
            mem_name,
 | 
			
		||||
| 
						 | 
				
			
			@ -488,8 +439,8 @@ impl<T: PortType> MemPort<T> {
 | 
			
		|||
        mem_name: ScopedNameId,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        port_name: PortName,
 | 
			
		||||
        addr_type: DynUIntType,
 | 
			
		||||
        mem_element_type: Interned<dyn DynCanonicalType>,
 | 
			
		||||
        addr_type: UInt,
 | 
			
		||||
        mem_element_type: CanonicalType,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        assert!(
 | 
			
		||||
            mem_element_type.is_storable(),
 | 
			
		||||
| 
						 | 
				
			
			@ -520,10 +471,10 @@ pub enum ReadUnderWrite {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
 | 
			
		||||
struct MemImpl<T: ArrayTypeTrait, P> {
 | 
			
		||||
struct MemImpl<Element: Type, Len: Size, P> {
 | 
			
		||||
    scoped_name: ScopedNameId,
 | 
			
		||||
    source_location: SourceLocation,
 | 
			
		||||
    array_type: T,
 | 
			
		||||
    array_type: ArrayType<Element, Len>,
 | 
			
		||||
    initial_value: Option<Interned<BitSlice>>,
 | 
			
		||||
    ports: P,
 | 
			
		||||
    read_latency: usize,
 | 
			
		||||
| 
						 | 
				
			
			@ -533,11 +484,11 @@ struct MemImpl<T: ArrayTypeTrait, P> {
 | 
			
		|||
    mem_annotations: Interned<[Annotation]>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub struct Mem<VA: ValueArrayOrSlice + ?Sized>(
 | 
			
		||||
    Interned<MemImpl<ArrayType<VA>, Interned<[Interned<MemPort<DynPortType>>]>>>,
 | 
			
		||||
pub struct Mem<Element: Type = CanonicalType, Len: Size = DynSize>(
 | 
			
		||||
    Interned<MemImpl<Element, Len, Interned<[MemPort<DynPortType>]>>>,
 | 
			
		||||
);
 | 
			
		||||
 | 
			
		||||
struct PortsDebug<'a>(&'a [Interned<MemPort<DynPortType>>]);
 | 
			
		||||
struct PortsDebug<'a>(&'a [MemPort<DynPortType>]);
 | 
			
		||||
 | 
			
		||||
impl fmt::Debug for PortsDebug<'_> {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
| 
						 | 
				
			
			@ -551,7 +502,7 @@ impl fmt::Debug for PortsDebug<'_> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> fmt::Debug for Mem<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> fmt::Debug for Mem<Element, Len> {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let MemImpl {
 | 
			
		||||
            scoped_name,
 | 
			
		||||
| 
						 | 
				
			
			@ -579,37 +530,37 @@ impl<VA: ValueArrayOrSlice + ?Sized> fmt::Debug for Mem<VA> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Hash for Mem<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> Hash for Mem<Element, Len> {
 | 
			
		||||
    fn hash<H: Hasher>(&self, state: &mut H) {
 | 
			
		||||
        self.0.hash(state);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Eq for Mem<VA> {}
 | 
			
		||||
impl<Element: Type, Len: Size> Eq for Mem<Element, Len> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> PartialEq for Mem<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> PartialEq for Mem<Element, Len> {
 | 
			
		||||
    fn eq(&self, other: &Self) -> bool {
 | 
			
		||||
        self.0 == other.0
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Copy for Mem<VA> {}
 | 
			
		||||
impl<Element: Type, Len: Size> Copy for Mem<Element, Len> {}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Clone for Mem<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> Clone for Mem<Element, Len> {
 | 
			
		||||
    fn clone(&self) -> Self {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> Mem<Element, Len> {
 | 
			
		||||
    #[allow(clippy::too_many_arguments)]
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_unchecked(
 | 
			
		||||
        scoped_name: ScopedNameId,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        array_type: ArrayType<VA>,
 | 
			
		||||
        array_type: ArrayType<Element, Len>,
 | 
			
		||||
        initial_value: Option<Interned<BitSlice>>,
 | 
			
		||||
        ports: Interned<[Interned<MemPort<DynPortType>>]>,
 | 
			
		||||
        ports: Interned<[MemPort<DynPortType>]>,
 | 
			
		||||
        read_latency: usize,
 | 
			
		||||
        write_latency: NonZeroU32,
 | 
			
		||||
        read_under_write: ReadUnderWrite,
 | 
			
		||||
| 
						 | 
				
			
			@ -617,14 +568,14 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
        mem_annotations: Interned<[Annotation]>,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        if let Some(initial_value) = initial_value {
 | 
			
		||||
            MemBuilder::<VA>::check_initial_value_bit_slice(
 | 
			
		||||
            MemBuilder::<Element, Len>::check_initial_value_bit_slice(
 | 
			
		||||
                array_type.element(),
 | 
			
		||||
                Some(array_type.len()),
 | 
			
		||||
                initial_value,
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
        let addr_width = memory_addr_width(array_type.len());
 | 
			
		||||
        let expected_mem_element_type = array_type.element().canonical_dyn();
 | 
			
		||||
        let expected_mem_element_type = array_type.element().canonical();
 | 
			
		||||
        assert!(
 | 
			
		||||
            expected_mem_element_type.is_storable(),
 | 
			
		||||
            "memory element type must be a storable type"
 | 
			
		||||
| 
						 | 
				
			
			@ -637,7 +588,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
                port_index,
 | 
			
		||||
                addr_type,
 | 
			
		||||
                mem_element_type,
 | 
			
		||||
            } = **port;
 | 
			
		||||
            } = *port;
 | 
			
		||||
            assert_eq!(mem_name, scoped_name, "memory name must match with ports");
 | 
			
		||||
            assert_eq!(
 | 
			
		||||
                port_index, index,
 | 
			
		||||
| 
						 | 
				
			
			@ -659,7 +610,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
            };
 | 
			
		||||
            assert_eq!(
 | 
			
		||||
                Some(port),
 | 
			
		||||
                ports.get(port.port_index).map(|v| &**v),
 | 
			
		||||
                ports.get(port.port_index),
 | 
			
		||||
                "port on memory must match annotation's target base"
 | 
			
		||||
            );
 | 
			
		||||
        }
 | 
			
		||||
| 
						 | 
				
			
			@ -682,13 +633,13 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
    pub fn source_location(self) -> SourceLocation {
 | 
			
		||||
        self.0.source_location
 | 
			
		||||
    }
 | 
			
		||||
    pub fn array_type(self) -> ArrayType<VA> {
 | 
			
		||||
    pub fn array_type(self) -> ArrayType<Element, Len> {
 | 
			
		||||
        self.0.array_type.clone()
 | 
			
		||||
    }
 | 
			
		||||
    pub fn initial_value(self) -> Option<Interned<BitSlice>> {
 | 
			
		||||
        self.0.initial_value
 | 
			
		||||
    }
 | 
			
		||||
    pub fn ports(self) -> Interned<[Interned<MemPort<DynPortType>>]> {
 | 
			
		||||
    pub fn ports(self) -> Interned<[MemPort<DynPortType>]> {
 | 
			
		||||
        self.0.ports
 | 
			
		||||
    }
 | 
			
		||||
    pub fn read_latency(self) -> usize {
 | 
			
		||||
| 
						 | 
				
			
			@ -706,7 +657,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
    pub fn mem_annotations(self) -> Interned<[Annotation]> {
 | 
			
		||||
        self.0.mem_annotations
 | 
			
		||||
    }
 | 
			
		||||
    pub fn canonical(self) -> Mem<[DynCanonicalValue]> {
 | 
			
		||||
    pub fn canonical(self) -> Mem {
 | 
			
		||||
        let MemImpl {
 | 
			
		||||
            scoped_name,
 | 
			
		||||
            source_location,
 | 
			
		||||
| 
						 | 
				
			
			@ -719,7 +670,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> Mem<VA> {
 | 
			
		|||
            port_annotations,
 | 
			
		||||
            mem_annotations,
 | 
			
		||||
        } = *self.0;
 | 
			
		||||
        let array_type = array_type.canonical();
 | 
			
		||||
        let array_type = array_type.as_dyn_array();
 | 
			
		||||
        Mem(Intern::intern_sized(MemImpl {
 | 
			
		||||
            scoped_name,
 | 
			
		||||
            source_location,
 | 
			
		||||
| 
						 | 
				
			
			@ -751,10 +702,10 @@ impl<T: fmt::Debug> fmt::Debug for MaybeSpecified<T> {
 | 
			
		|||
pub(crate) struct MemBuilderTarget {
 | 
			
		||||
    pub(crate) scoped_name: ScopedNameId,
 | 
			
		||||
    pub(crate) source_location: SourceLocation,
 | 
			
		||||
    pub(crate) mem_element_type: Interned<dyn DynCanonicalType>,
 | 
			
		||||
    pub(crate) mem_element_type: CanonicalType,
 | 
			
		||||
    pub(crate) depth: Option<usize>,
 | 
			
		||||
    pub(crate) initial_value: Option<Interned<BitSlice>>,
 | 
			
		||||
    pub(crate) ports: Vec<Interned<MemPort<DynPortType>>>,
 | 
			
		||||
    pub(crate) ports: Vec<MemPort<DynPortType>>,
 | 
			
		||||
    pub(crate) read_latency: usize,
 | 
			
		||||
    pub(crate) write_latency: NonZeroU32,
 | 
			
		||||
    pub(crate) read_under_write: ReadUnderWrite,
 | 
			
		||||
| 
						 | 
				
			
			@ -763,11 +714,11 @@ pub(crate) struct MemBuilderTarget {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
impl MemBuilderTarget {
 | 
			
		||||
    pub(crate) fn make_memory(&self) -> Option<Mem<[DynCanonicalValue]>> {
 | 
			
		||||
    pub(crate) fn make_memory(&self) -> Option<Mem> {
 | 
			
		||||
        Some(Mem::new_unchecked(
 | 
			
		||||
            self.scoped_name,
 | 
			
		||||
            self.source_location,
 | 
			
		||||
            ArrayType::new_slice(self.mem_element_type, self.depth?),
 | 
			
		||||
            ArrayType::new_dyn(self.mem_element_type, self.depth?),
 | 
			
		||||
            self.initial_value,
 | 
			
		||||
            Intern::intern(&self.ports),
 | 
			
		||||
            self.read_latency,
 | 
			
		||||
| 
						 | 
				
			
			@ -817,16 +768,18 @@ impl fmt::Debug for MemBuilderTarget {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub struct MemBuilder<VA: ValueArrayOrSlice + ?Sized> {
 | 
			
		||||
    mem_element_type: VA::ElementType,
 | 
			
		||||
pub struct MemBuilder<Element: Type, Len: Size = DynSize> {
 | 
			
		||||
    mem_element_type: Element,
 | 
			
		||||
    target: Rc<RefCell<MemBuilderTarget>>,
 | 
			
		||||
    _phantom: PhantomData<Len>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> fmt::Debug for MemBuilder<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> fmt::Debug for MemBuilder<Element, Len> {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        let Self {
 | 
			
		||||
            mem_element_type,
 | 
			
		||||
            target,
 | 
			
		||||
            _phantom: _,
 | 
			
		||||
        } = &self;
 | 
			
		||||
        target.borrow().debug_fmt("MemBuilder", mem_element_type, f)
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -838,15 +791,16 @@ pub fn memory_addr_width(depth: usize) -> usize {
 | 
			
		|||
        .map_or(usize::BITS, usize::ilog2) as usize
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		||||
impl<Element: Type, Len: Size> MemBuilder<Element, Len> {
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    fn check_initial_value_bit_slice(
 | 
			
		||||
        mem_element_type: &VA::ElementType,
 | 
			
		||||
        mem_element_type: Element,
 | 
			
		||||
        depth: Option<usize>,
 | 
			
		||||
        initial_value: Interned<BitSlice>,
 | 
			
		||||
    ) -> Interned<BitSlice> {
 | 
			
		||||
        let element_bit_width = mem_element_type.canonical().bit_width();
 | 
			
		||||
        if let Some(depth) = depth {
 | 
			
		||||
            let expected_len = depth.checked_mul(mem_element_type.bit_width()).expect(
 | 
			
		||||
            let expected_len = depth.checked_mul(element_bit_width).expect(
 | 
			
		||||
                "memory must be small enough that its initializer bit length fits in usize",
 | 
			
		||||
            );
 | 
			
		||||
            assert_eq!(
 | 
			
		||||
| 
						 | 
				
			
			@ -858,7 +812,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
        assert!(
 | 
			
		||||
            initial_value
 | 
			
		||||
                .len()
 | 
			
		||||
                .checked_rem(mem_element_type.bit_width())
 | 
			
		||||
                .checked_rem(element_bit_width)
 | 
			
		||||
                .unwrap_or(initial_value.len())
 | 
			
		||||
                == 0,
 | 
			
		||||
            "Mem's initializer bit length must be a multiple of the element type's bit width",
 | 
			
		||||
| 
						 | 
				
			
			@ -867,14 +821,14 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    fn check_initial_value_expr(
 | 
			
		||||
        mem_element_type: &VA::ElementType,
 | 
			
		||||
        mem_element_type: &Element,
 | 
			
		||||
        depth: Option<usize>,
 | 
			
		||||
        initial_value: Expr<Array<[DynCanonicalValue]>>,
 | 
			
		||||
        initial_value: Expr<Array>,
 | 
			
		||||
    ) -> Interned<BitSlice> {
 | 
			
		||||
        let initial_value_ty = initial_value.canonical_type();
 | 
			
		||||
        let initial_value_ty = Expr::ty(initial_value);
 | 
			
		||||
        assert_eq!(
 | 
			
		||||
            *mem_element_type,
 | 
			
		||||
            <VA::ElementType as DynType>::from_dyn_canonical_type(*initial_value_ty.element()),
 | 
			
		||||
            Element::from_canonical(initial_value_ty.element()),
 | 
			
		||||
            "Mem's element type must match initializer's element type",
 | 
			
		||||
        );
 | 
			
		||||
        if let Some(depth) = depth {
 | 
			
		||||
| 
						 | 
				
			
			@ -889,7 +843,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
        };
 | 
			
		||||
        debug_assert_eq!(
 | 
			
		||||
            retval.len(),
 | 
			
		||||
            initial_value_ty.bit_width(),
 | 
			
		||||
            initial_value_ty.type_properties().bit_width,
 | 
			
		||||
            "initial value produced wrong literal bits length"
 | 
			
		||||
        );
 | 
			
		||||
        retval
 | 
			
		||||
| 
						 | 
				
			
			@ -898,9 +852,9 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
    pub(crate) fn new(
 | 
			
		||||
        scoped_name: ScopedNameId,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        mem_element_type: VA::ElementType,
 | 
			
		||||
        mem_element_type: Element,
 | 
			
		||||
    ) -> (Self, Rc<RefCell<MemBuilderTarget>>) {
 | 
			
		||||
        let canonical_mem_element_type = mem_element_type.canonical_dyn();
 | 
			
		||||
        let canonical_mem_element_type = mem_element_type.canonical();
 | 
			
		||||
        assert!(
 | 
			
		||||
            canonical_mem_element_type.is_storable(),
 | 
			
		||||
            "memory element type must be a storable type"
 | 
			
		||||
| 
						 | 
				
			
			@ -909,7 +863,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
            scoped_name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            mem_element_type: canonical_mem_element_type,
 | 
			
		||||
            depth: VA::FIXED_LEN_TYPE.map(VA::len_from_len_type),
 | 
			
		||||
            depth: Len::KNOWN_VALUE,
 | 
			
		||||
            initial_value: None,
 | 
			
		||||
            ports: vec![],
 | 
			
		||||
            read_latency: 0,
 | 
			
		||||
| 
						 | 
				
			
			@ -922,6 +876,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
            Self {
 | 
			
		||||
                mem_element_type,
 | 
			
		||||
                target: Rc::clone(&target),
 | 
			
		||||
                _phantom: PhantomData,
 | 
			
		||||
            },
 | 
			
		||||
            target,
 | 
			
		||||
        )
 | 
			
		||||
| 
						 | 
				
			
			@ -931,19 +886,19 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
        &mut self,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        port_kind: PortKind,
 | 
			
		||||
    ) -> Interned<MemPort<DynPortType>> {
 | 
			
		||||
    ) -> MemPort<DynPortType> {
 | 
			
		||||
        let mut target = self.target.borrow_mut();
 | 
			
		||||
        let Some(depth) = target.depth else {
 | 
			
		||||
            panic!("MemBuilder::depth must be called before adding ports");
 | 
			
		||||
        };
 | 
			
		||||
        let port = Intern::intern_sized(MemPort {
 | 
			
		||||
        let port = MemPort {
 | 
			
		||||
            mem_name: target.scoped_name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            port_kind,
 | 
			
		||||
            port_index: target.ports.len(),
 | 
			
		||||
            addr_type: DynUIntType::new(memory_addr_width(depth)),
 | 
			
		||||
            addr_type: UInt::new(memory_addr_width(depth)),
 | 
			
		||||
            mem_element_type: target.mem_element_type,
 | 
			
		||||
        });
 | 
			
		||||
        };
 | 
			
		||||
        target.ports.push(port);
 | 
			
		||||
        port
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -952,50 +907,53 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
        &mut self,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        kind: PortKind,
 | 
			
		||||
    ) -> Expr<DynBundle> {
 | 
			
		||||
        Expr::new_unchecked(ExprEnum::MemPort(self.new_port_impl(source_location, kind)))
 | 
			
		||||
    ) -> Expr<Bundle> {
 | 
			
		||||
        self.new_port_impl(source_location, kind).to_expr()
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_port(&mut self, kind: PortKind) -> Expr<DynBundle> {
 | 
			
		||||
    pub fn new_port(&mut self, kind: PortKind) -> Expr<Bundle> {
 | 
			
		||||
        self.new_port_with_loc(SourceLocation::caller(), kind)
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_read_port_with_loc(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Expr<ReadStruct<VA::Element>> {
 | 
			
		||||
        Expr::new_unchecked(ExprEnum::MemPort(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::ReadOnly),
 | 
			
		||||
        ))
 | 
			
		||||
    ) -> Expr<ReadStruct<Element, DynSize>> {
 | 
			
		||||
        Expr::from_bundle(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::ReadOnly)
 | 
			
		||||
                .to_expr(),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_read_port(&mut self) -> Expr<ReadStruct<VA::Element>> {
 | 
			
		||||
    pub fn new_read_port(&mut self) -> Expr<ReadStruct<Element, DynSize>> {
 | 
			
		||||
        self.new_read_port_with_loc(SourceLocation::caller())
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_write_port_with_loc(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Expr<WriteStruct<VA::Element>> {
 | 
			
		||||
        Expr::new_unchecked(ExprEnum::MemPort(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::WriteOnly),
 | 
			
		||||
        ))
 | 
			
		||||
    ) -> Expr<WriteStruct<Element, DynSize>> {
 | 
			
		||||
        Expr::from_bundle(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::WriteOnly)
 | 
			
		||||
                .to_expr(),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_write_port(&mut self) -> Expr<WriteStruct<VA::Element>> {
 | 
			
		||||
    pub fn new_write_port(&mut self) -> Expr<WriteStruct<Element, DynSize>> {
 | 
			
		||||
        self.new_write_port_with_loc(SourceLocation::caller())
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_rw_port_with_loc(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
    ) -> Expr<ReadWriteStruct<VA::Element>> {
 | 
			
		||||
        Expr::new_unchecked(ExprEnum::MemPort(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::ReadWrite),
 | 
			
		||||
        ))
 | 
			
		||||
    ) -> Expr<ReadWriteStruct<Element, DynSize>> {
 | 
			
		||||
        Expr::from_bundle(
 | 
			
		||||
            self.new_port_impl(source_location, PortKind::ReadWrite)
 | 
			
		||||
                .to_expr(),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_rw_port(&mut self) -> Expr<ReadWriteStruct<VA::Element>> {
 | 
			
		||||
    pub fn new_rw_port(&mut self) -> Expr<ReadWriteStruct<Element, DynSize>> {
 | 
			
		||||
        self.new_rw_port_with_loc(SourceLocation::caller())
 | 
			
		||||
    }
 | 
			
		||||
    pub fn scoped_name(&self) -> ScopedNameId {
 | 
			
		||||
| 
						 | 
				
			
			@ -1004,7 +962,7 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
    pub fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        self.target.borrow().source_location
 | 
			
		||||
    }
 | 
			
		||||
    pub fn get_mem_element_type(&self) -> &VA::ElementType {
 | 
			
		||||
    pub fn get_mem_element_type(&self) -> &Element {
 | 
			
		||||
        &self.mem_element_type
 | 
			
		||||
    }
 | 
			
		||||
    #[allow(clippy::result_unit_err)]
 | 
			
		||||
| 
						 | 
				
			
			@ -1027,28 +985,28 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
        target.depth = Some(depth);
 | 
			
		||||
    }
 | 
			
		||||
    #[allow(clippy::result_unit_err)]
 | 
			
		||||
    pub fn get_array_type(&self) -> Result<ArrayType<VA>, ()> {
 | 
			
		||||
        Ok(ArrayType::new_with_len(
 | 
			
		||||
    pub fn get_array_type(&self) -> Result<ArrayType<Element, Len>, ()> {
 | 
			
		||||
        Ok(ArrayType::new(
 | 
			
		||||
            self.mem_element_type.clone(),
 | 
			
		||||
            self.get_depth()?,
 | 
			
		||||
            Len::from_usize(self.get_depth()?),
 | 
			
		||||
        ))
 | 
			
		||||
    }
 | 
			
		||||
    pub fn get_initial_value(&self) -> Option<Interned<BitSlice>> {
 | 
			
		||||
        self.target.borrow().initial_value
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn initial_value(&mut self, initial_value: impl ToExpr<Type = ArrayType<VA>>) {
 | 
			
		||||
    pub fn initial_value(&mut self, initial_value: impl ToExpr<Type = ArrayType<Element, Len>>) {
 | 
			
		||||
        let mut target = self.target.borrow_mut();
 | 
			
		||||
        if target.initial_value.is_some() {
 | 
			
		||||
            panic!("can't set Mem's initial value more than once");
 | 
			
		||||
        }
 | 
			
		||||
        let initial_value = initial_value.to_expr().canonical();
 | 
			
		||||
        let initial_value = Expr::as_dyn_array(initial_value.to_expr());
 | 
			
		||||
        target.initial_value = Some(Self::check_initial_value_expr(
 | 
			
		||||
            &self.mem_element_type,
 | 
			
		||||
            target.depth,
 | 
			
		||||
            initial_value,
 | 
			
		||||
        ));
 | 
			
		||||
        target.depth = Some(initial_value.ty().len());
 | 
			
		||||
        target.depth = Some(Expr::ty(initial_value).len());
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn initial_value_bit_slice(&mut self, initial_value: Interned<BitSlice>) {
 | 
			
		||||
| 
						 | 
				
			
			@ -1057,11 +1015,11 @@ impl<VA: ValueArrayOrSlice + ?Sized> MemBuilder<VA> {
 | 
			
		|||
            panic!("can't set Mem's initial value more than once");
 | 
			
		||||
        }
 | 
			
		||||
        target.initial_value = Some(Self::check_initial_value_bit_slice(
 | 
			
		||||
            &self.mem_element_type,
 | 
			
		||||
            self.mem_element_type,
 | 
			
		||||
            target.depth,
 | 
			
		||||
            initial_value,
 | 
			
		||||
        ));
 | 
			
		||||
        let element_bit_width = self.mem_element_type.bit_width();
 | 
			
		||||
        let element_bit_width = self.mem_element_type.canonical().bit_width();
 | 
			
		||||
        if element_bit_width != 0 {
 | 
			
		||||
            target.depth = Some(initial_value.len() / element_bit_width);
 | 
			
		||||
        }
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -2,10 +2,14 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    array::{Array, ArrayType},
 | 
			
		||||
    bundle::{BundleType, BundleValue, DynBundle, DynBundleType},
 | 
			
		||||
    enum_::{DynEnum, DynEnumType, EnumType, EnumValue, VariantType},
 | 
			
		||||
    expr::{ops, Expr, ExprEnum, ToExpr},
 | 
			
		||||
    int::{DynUInt, DynUIntType, IntCmp},
 | 
			
		||||
    bundle::{Bundle, BundleType},
 | 
			
		||||
    enum_::{Enum, EnumType, EnumVariant},
 | 
			
		||||
    expr::{
 | 
			
		||||
        ops::{self, EnumLiteral},
 | 
			
		||||
        CastBitsTo, CastToBits, Expr, ExprEnum, ToExpr,
 | 
			
		||||
    },
 | 
			
		||||
    hdl,
 | 
			
		||||
    int::{DynSize, IntCmp, Size, UInt, UIntType},
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    memory::{DynPortType, Mem, MemPort},
 | 
			
		||||
    module::{
 | 
			
		||||
| 
						 | 
				
			
			@ -13,7 +17,7 @@ use crate::{
 | 
			
		|||
        Block, Module, NameIdGen, ScopedNameId, Stmt, StmtConnect, StmtIf, StmtMatch, StmtWire,
 | 
			
		||||
    },
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{DynCanonicalType, DynCanonicalValue, Type, TypeEnum, Value, ValueEnum},
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
    wire::Wire,
 | 
			
		||||
};
 | 
			
		||||
use core::fmt;
 | 
			
		||||
| 
						 | 
				
			
			@ -21,7 +25,7 @@ use hashbrown::HashMap;
 | 
			
		|||
 | 
			
		||||
#[derive(Debug)]
 | 
			
		||||
pub enum SimplifyEnumsError {
 | 
			
		||||
    EnumIsNotCastableFromBits { enum_type: DynEnumType },
 | 
			
		||||
    EnumIsNotCastableFromBits { enum_type: Enum },
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl fmt::Display for SimplifyEnumsError {
 | 
			
		||||
| 
						 | 
				
			
			@ -37,26 +41,23 @@ impl fmt::Display for SimplifyEnumsError {
 | 
			
		|||
 | 
			
		||||
impl std::error::Error for SimplifyEnumsError {}
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, Eq, PartialEq, Hash, Debug)]
 | 
			
		||||
struct TagAndBody<T> {
 | 
			
		||||
#[hdl]
 | 
			
		||||
struct TagAndBody<T, BodyWidth: Size> {
 | 
			
		||||
    tag: T,
 | 
			
		||||
    body: DynUInt,
 | 
			
		||||
    body: UIntType<BodyWidth>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type TagAndBodyType<T> = <TagAndBody<T> as ToExpr>::Type;
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug)]
 | 
			
		||||
enum EnumTypeState {
 | 
			
		||||
    TagEnumAndBody(TagAndBodyType<DynEnum>),
 | 
			
		||||
    TagUIntAndBody(TagAndBodyType<DynUInt>),
 | 
			
		||||
    UInt(DynUIntType),
 | 
			
		||||
    TagEnumAndBody(TagAndBody<Enum, DynSize>),
 | 
			
		||||
    TagUIntAndBody(TagAndBody<UInt, DynSize>),
 | 
			
		||||
    UInt(UInt),
 | 
			
		||||
    Unchanged,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
struct State {
 | 
			
		||||
    enum_types: HashMap<DynEnumType, EnumTypeState>,
 | 
			
		||||
    replacement_mem_ports:
 | 
			
		||||
        HashMap<Interned<MemPort<DynPortType>>, Interned<Wire<Interned<dyn DynCanonicalType>>>>,
 | 
			
		||||
    enum_types: HashMap<Enum, EnumTypeState>,
 | 
			
		||||
    replacement_mem_ports: HashMap<MemPort<DynPortType>, Wire<CanonicalType>>,
 | 
			
		||||
    kind: SimplifyEnumsKind,
 | 
			
		||||
    name_id_gen: NameIdGen,
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -64,12 +65,12 @@ struct State {
 | 
			
		|||
impl State {
 | 
			
		||||
    fn get_or_make_enum_type_state(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        enum_type: DynEnumType,
 | 
			
		||||
        enum_type: Enum,
 | 
			
		||||
    ) -> Result<EnumTypeState, SimplifyEnumsError> {
 | 
			
		||||
        if let Some(retval) = self.enum_types.get(&enum_type) {
 | 
			
		||||
            return Ok(retval.clone());
 | 
			
		||||
        }
 | 
			
		||||
        if !enum_type.is_castable_from_bits() {
 | 
			
		||||
        if !enum_type.type_properties().is_castable_from_bits {
 | 
			
		||||
            return Err(SimplifyEnumsError::EnumIsNotCastableFromBits { enum_type });
 | 
			
		||||
        }
 | 
			
		||||
        let has_body = enum_type
 | 
			
		||||
| 
						 | 
				
			
			@ -78,29 +79,29 @@ impl State {
 | 
			
		|||
            .any(|variant| variant.ty.is_some());
 | 
			
		||||
        let retval = match (self.kind, has_body) {
 | 
			
		||||
            (SimplifyEnumsKind::SimplifyToEnumsWithNoBody, true) => {
 | 
			
		||||
                EnumTypeState::TagEnumAndBody(TagAndBodyType::<DynEnum> {
 | 
			
		||||
                    tag: DynEnumType::new(Interned::from_iter(enum_type.variants().iter().map(
 | 
			
		||||
                        |v| VariantType {
 | 
			
		||||
                EnumTypeState::TagEnumAndBody(TagAndBody {
 | 
			
		||||
                    tag: Enum::new(Interned::from_iter(enum_type.variants().iter().map(|v| {
 | 
			
		||||
                        EnumVariant {
 | 
			
		||||
                            name: v.name,
 | 
			
		||||
                            ty: None,
 | 
			
		||||
                        },
 | 
			
		||||
                    ))),
 | 
			
		||||
                    body: DynUIntType::new(
 | 
			
		||||
                        enum_type.bit_width() - enum_type.discriminant_bit_width(),
 | 
			
		||||
                        }
 | 
			
		||||
                    }))),
 | 
			
		||||
                    body: UInt::new_dyn(
 | 
			
		||||
                        enum_type.type_properties().bit_width - enum_type.discriminant_bit_width(),
 | 
			
		||||
                    ),
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
            (SimplifyEnumsKind::SimplifyToEnumsWithNoBody, false) => EnumTypeState::Unchanged,
 | 
			
		||||
            (SimplifyEnumsKind::ReplaceWithBundleOfUInts, _) => {
 | 
			
		||||
                EnumTypeState::TagUIntAndBody(TagAndBodyType::<DynUInt> {
 | 
			
		||||
                    tag: DynUIntType::new(enum_type.discriminant_bit_width()),
 | 
			
		||||
                    body: DynUIntType::new(
 | 
			
		||||
                        enum_type.bit_width() - enum_type.discriminant_bit_width(),
 | 
			
		||||
                EnumTypeState::TagUIntAndBody(TagAndBody {
 | 
			
		||||
                    tag: UInt::new_dyn(enum_type.discriminant_bit_width()),
 | 
			
		||||
                    body: UInt::new_dyn(
 | 
			
		||||
                        enum_type.type_properties().bit_width - enum_type.discriminant_bit_width(),
 | 
			
		||||
                    ),
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
            (SimplifyEnumsKind::ReplaceWithUInt, _) => {
 | 
			
		||||
                EnumTypeState::UInt(DynUIntType::new(enum_type.bit_width()))
 | 
			
		||||
                EnumTypeState::UInt(UInt::new_dyn(enum_type.type_properties().bit_width))
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
        self.enum_types.insert(enum_type, retval.clone());
 | 
			
		||||
| 
						 | 
				
			
			@ -108,21 +109,14 @@ impl State {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn value_to_uint<T: Value>(value: Option<&T>, target_ty: DynUIntType) -> DynUInt {
 | 
			
		||||
    let Some(value) = value else {
 | 
			
		||||
        return DynUInt::with_type(target_ty, 0u8);
 | 
			
		||||
    };
 | 
			
		||||
    DynUInt::from_bit_slice(&value.to_bits())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn connect_port(
 | 
			
		||||
    stmts: &mut Vec<Stmt>,
 | 
			
		||||
    lhs: Expr<DynCanonicalValue>,
 | 
			
		||||
    rhs: Expr<DynCanonicalValue>,
 | 
			
		||||
    lhs: Expr<CanonicalType>,
 | 
			
		||||
    rhs: Expr<CanonicalType>,
 | 
			
		||||
    source_location: SourceLocation,
 | 
			
		||||
) {
 | 
			
		||||
    println!("connect_port: lhs={lhs:?} rhs={rhs:?}");
 | 
			
		||||
    if lhs.canonical_type() == rhs.canonical_type() {
 | 
			
		||||
    println!("connect_port:\nlhs={lhs:?}\nrhs={rhs:?}");
 | 
			
		||||
    if Expr::ty(lhs) == Expr::ty(rhs) {
 | 
			
		||||
        stmts.push(
 | 
			
		||||
            dbg!(StmtConnect {
 | 
			
		||||
                lhs,
 | 
			
		||||
| 
						 | 
				
			
			@ -133,55 +127,53 @@ fn connect_port(
 | 
			
		|||
        );
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
    match (
 | 
			
		||||
        lhs.canonical_type().type_enum(),
 | 
			
		||||
        rhs.canonical_type().type_enum(),
 | 
			
		||||
    ) {
 | 
			
		||||
        (TypeEnum::BundleType(lhs_type), TypeEnum::UInt(_)) => {
 | 
			
		||||
            let lhs = lhs.with_type::<DynBundle>();
 | 
			
		||||
    match (Expr::ty(lhs), Expr::ty(rhs)) {
 | 
			
		||||
        (CanonicalType::Bundle(lhs_type), CanonicalType::UInt(_) | CanonicalType::Bool(_)) => {
 | 
			
		||||
            let lhs = Expr::<Bundle>::from_canonical(lhs);
 | 
			
		||||
            for field in lhs_type.fields() {
 | 
			
		||||
                assert!(!field.flipped);
 | 
			
		||||
                connect_port(stmts, lhs.field(&field.name), rhs, source_location);
 | 
			
		||||
                connect_port(stmts, Expr::field(lhs, &field.name), rhs, source_location);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        (TypeEnum::UInt(_), TypeEnum::BundleType(rhs_type)) => {
 | 
			
		||||
            let rhs = rhs.with_type::<DynBundle>();
 | 
			
		||||
        (CanonicalType::UInt(_) | CanonicalType::Bool(_), CanonicalType::Bundle(rhs_type)) => {
 | 
			
		||||
            let rhs = Expr::<Bundle>::from_canonical(rhs);
 | 
			
		||||
            for field in rhs_type.fields() {
 | 
			
		||||
                assert!(!field.flipped);
 | 
			
		||||
                connect_port(stmts, lhs, rhs.field(&field.name), source_location);
 | 
			
		||||
                connect_port(stmts, lhs, Expr::field(rhs, &field.name), source_location);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        (TypeEnum::BundleType(lhs_type), TypeEnum::BundleType(_)) => {
 | 
			
		||||
            let lhs = lhs.with_type::<DynBundle>();
 | 
			
		||||
            let rhs = rhs.with_type::<DynBundle>();
 | 
			
		||||
        (CanonicalType::Bundle(lhs_type), CanonicalType::Bundle(_)) => {
 | 
			
		||||
            let lhs = Expr::<Bundle>::from_canonical(lhs);
 | 
			
		||||
            let rhs = Expr::<Bundle>::from_canonical(rhs);
 | 
			
		||||
            for field in lhs_type.fields() {
 | 
			
		||||
                let (lhs_field, rhs_field) = if field.flipped {
 | 
			
		||||
                    (rhs.field(&field.name), lhs.field(&field.name))
 | 
			
		||||
                    (Expr::field(rhs, &field.name), Expr::field(lhs, &field.name))
 | 
			
		||||
                } else {
 | 
			
		||||
                    (lhs.field(&field.name), rhs.field(&field.name))
 | 
			
		||||
                    (Expr::field(lhs, &field.name), Expr::field(rhs, &field.name))
 | 
			
		||||
                };
 | 
			
		||||
                connect_port(stmts, lhs_field, rhs_field, source_location);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        (TypeEnum::ArrayType(lhs_type), TypeEnum::ArrayType(_)) => {
 | 
			
		||||
            let lhs = lhs.with_type::<Array<[DynCanonicalValue]>>();
 | 
			
		||||
            let rhs = rhs.with_type::<Array<[DynCanonicalValue]>>();
 | 
			
		||||
        (CanonicalType::Array(lhs_type), CanonicalType::Array(_)) => {
 | 
			
		||||
            let lhs = Expr::<Array>::from_canonical(lhs);
 | 
			
		||||
            let rhs = Expr::<Array>::from_canonical(rhs);
 | 
			
		||||
            for index in 0..lhs_type.len() {
 | 
			
		||||
                connect_port(stmts, lhs[index], rhs[index], source_location);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        (TypeEnum::BundleType(_), _)
 | 
			
		||||
        | (TypeEnum::EnumType(_), _)
 | 
			
		||||
        | (TypeEnum::ArrayType(_), _)
 | 
			
		||||
        | (TypeEnum::UInt(_), _)
 | 
			
		||||
        | (TypeEnum::SInt(_), _)
 | 
			
		||||
        | (TypeEnum::Clock(_), _)
 | 
			
		||||
        | (TypeEnum::AsyncReset(_), _)
 | 
			
		||||
        | (TypeEnum::SyncReset(_), _)
 | 
			
		||||
        | (TypeEnum::Reset(_), _) => unreachable!(
 | 
			
		||||
        (CanonicalType::Bundle(_), _)
 | 
			
		||||
        | (CanonicalType::Enum(_), _)
 | 
			
		||||
        | (CanonicalType::Array(_), _)
 | 
			
		||||
        | (CanonicalType::UInt(_), _)
 | 
			
		||||
        | (CanonicalType::SInt(_), _)
 | 
			
		||||
        | (CanonicalType::Bool(_), _)
 | 
			
		||||
        | (CanonicalType::Clock(_), _)
 | 
			
		||||
        | (CanonicalType::AsyncReset(_), _)
 | 
			
		||||
        | (CanonicalType::SyncReset(_), _)
 | 
			
		||||
        | (CanonicalType::Reset(_), _) => unreachable!(
 | 
			
		||||
            "trying to connect memory ports:\n{:?}\n{:?}",
 | 
			
		||||
            lhs.canonical_type().type_enum(),
 | 
			
		||||
            rhs.canonical_type().type_enum(),
 | 
			
		||||
            Expr::ty(lhs),
 | 
			
		||||
            Expr::ty(rhs),
 | 
			
		||||
        ),
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -189,18 +181,11 @@ fn connect_port(
 | 
			
		|||
impl Folder for State {
 | 
			
		||||
    type Error = SimplifyEnumsError;
 | 
			
		||||
 | 
			
		||||
    fn fold_dyn_enum(&mut self, _v: DynEnum) -> Result<DynEnum, Self::Error> {
 | 
			
		||||
    fn fold_enum(&mut self, _v: Enum) -> Result<Enum, Self::Error> {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_dyn_enum_type(&mut self, _v: DynEnumType) -> Result<DynEnumType, Self::Error> {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_module<T: BundleValue>(&mut self, v: Module<T>) -> Result<Module<T>, Self::Error>
 | 
			
		||||
    where
 | 
			
		||||
        T::Type: BundleType<Value = T>,
 | 
			
		||||
    {
 | 
			
		||||
    fn fold_module<T: BundleType>(&mut self, v: Module<T>) -> Result<Module<T>, Self::Error> {
 | 
			
		||||
        let old_name_id_gen =
 | 
			
		||||
            std::mem::replace(&mut self.name_id_gen, NameIdGen::for_module(v.canonical()));
 | 
			
		||||
        let retval = Fold::default_fold(v, self);
 | 
			
		||||
| 
						 | 
				
			
			@ -211,98 +196,86 @@ impl Folder for State {
 | 
			
		|||
    fn fold_expr_enum(&mut self, op: ExprEnum) -> Result<ExprEnum, Self::Error> {
 | 
			
		||||
        match op {
 | 
			
		||||
            ExprEnum::EnumLiteral(op) => Ok(match self.get_or_make_enum_type_state(op.ty())? {
 | 
			
		||||
                EnumTypeState::TagEnumAndBody(TagAndBodyType::<DynEnum> { tag, body }) => {
 | 
			
		||||
                    TagAndBodyType::<DynEnum>::builder()
 | 
			
		||||
                        .field_tag(DynEnum::new_by_index(tag, op.variant_index(), None))
 | 
			
		||||
                EnumTypeState::TagEnumAndBody(TagAndBody { tag, body }) => *Expr::expr_enum(
 | 
			
		||||
                    <TagAndBody<Enum, DynSize> as BundleType>::Builder::default()
 | 
			
		||||
                        .field_tag(EnumLiteral::new_by_index(tag, op.variant_index(), None))
 | 
			
		||||
                        .field_body(match op.variant_value() {
 | 
			
		||||
                            Some(variant_value) => variant_value.fold(self)?.cast_to_bits(),
 | 
			
		||||
                            None => DynUInt::with_type(body, 0u8).to_expr(),
 | 
			
		||||
                            None => body.zero().to_expr(),
 | 
			
		||||
                        })
 | 
			
		||||
                        .build()
 | 
			
		||||
                        .expr_enum()
 | 
			
		||||
                }
 | 
			
		||||
                EnumTypeState::TagUIntAndBody(TagAndBodyType::<DynUInt> { tag, body }) => {
 | 
			
		||||
                    TagAndBodyType::<DynUInt>::builder()
 | 
			
		||||
                        .field_tag(DynUInt::with_type(tag, op.variant_index()))
 | 
			
		||||
                        .field_body(match op.variant_value() {
 | 
			
		||||
                            Some(variant_value) => variant_value.fold(self)?.cast_to_bits(),
 | 
			
		||||
                            None => DynUInt::with_type(body, 0u8).to_expr(),
 | 
			
		||||
                        })
 | 
			
		||||
                        .build()
 | 
			
		||||
                        .expr_enum()
 | 
			
		||||
                }
 | 
			
		||||
                EnumTypeState::UInt(_) => TagAndBodyType::<DynUInt>::builder()
 | 
			
		||||
                    .field_tag(DynUInt::with_type(
 | 
			
		||||
                        DynUIntType::new(op.ty().discriminant_bit_width()),
 | 
			
		||||
                        op.variant_index(),
 | 
			
		||||
                    ))
 | 
			
		||||
                    .field_body(match op.variant_value() {
 | 
			
		||||
                        Some(variant_value) => variant_value.fold(self)?.cast_to_bits(),
 | 
			
		||||
                        None => DynUInt::with_type(
 | 
			
		||||
                            DynUIntType::new(
 | 
			
		||||
                                op.ty().bit_width() - op.ty().discriminant_bit_width(),
 | 
			
		||||
                            ),
 | 
			
		||||
                            0u8,
 | 
			
		||||
                        )
 | 
			
		||||
                        .to_expr(),
 | 
			
		||||
                    })
 | 
			
		||||
                    .build()
 | 
			
		||||
                    .cast_to_bits()
 | 
			
		||||
                    .expr_enum(),
 | 
			
		||||
                EnumTypeState::Unchanged => ExprEnum::EnumLiteral(
 | 
			
		||||
                    ops::EnumLiteral::new_unchecked(
 | 
			
		||||
                        op.variant_value().map(|v| v.fold(self)).transpose()?,
 | 
			
		||||
                        op.variant_index(),
 | 
			
		||||
                        op.ty(),
 | 
			
		||||
                    )
 | 
			
		||||
                    .intern_sized(),
 | 
			
		||||
                ),
 | 
			
		||||
                EnumTypeState::TagUIntAndBody(TagAndBody { tag, body }) => *Expr::expr_enum(
 | 
			
		||||
                    <TagAndBody<UInt, DynSize> as BundleType>::Builder::default()
 | 
			
		||||
                        .field_tag(tag.from_int_wrapping(op.variant_index()))
 | 
			
		||||
                        .field_body(match op.variant_value() {
 | 
			
		||||
                            Some(variant_value) => variant_value.fold(self)?.cast_to_bits(),
 | 
			
		||||
                            None => body.zero().to_expr(),
 | 
			
		||||
                        })
 | 
			
		||||
                        .to_expr(),
 | 
			
		||||
                ),
 | 
			
		||||
                EnumTypeState::UInt(_) => *Expr::expr_enum(
 | 
			
		||||
                    <TagAndBody<UInt, DynSize> as BundleType>::Builder::default()
 | 
			
		||||
                        .field_tag(
 | 
			
		||||
                            UIntType::new(op.ty().discriminant_bit_width())
 | 
			
		||||
                                .from_int_wrapping(op.variant_index()),
 | 
			
		||||
                        )
 | 
			
		||||
                        .field_body(match op.variant_value() {
 | 
			
		||||
                            Some(variant_value) => variant_value.fold(self)?.cast_to_bits(),
 | 
			
		||||
                            None => UIntType::new(
 | 
			
		||||
                                op.ty().type_properties().bit_width
 | 
			
		||||
                                    - op.ty().discriminant_bit_width(),
 | 
			
		||||
                            )
 | 
			
		||||
                            .zero()
 | 
			
		||||
                            .to_expr(),
 | 
			
		||||
                        })
 | 
			
		||||
                        .cast_to_bits(),
 | 
			
		||||
                ),
 | 
			
		||||
                EnumTypeState::Unchanged => ExprEnum::EnumLiteral(ops::EnumLiteral::new_by_index(
 | 
			
		||||
                    op.ty(),
 | 
			
		||||
                    op.variant_index(),
 | 
			
		||||
                    op.variant_value().map(|v| v.fold(self)).transpose()?,
 | 
			
		||||
                )),
 | 
			
		||||
            }),
 | 
			
		||||
            ExprEnum::VariantAccess(op) => {
 | 
			
		||||
                Ok(match self.get_or_make_enum_type_state(op.base().ty())? {
 | 
			
		||||
            ExprEnum::VariantAccess(op) => Ok(
 | 
			
		||||
                match self.get_or_make_enum_type_state(Expr::ty(op.base()))? {
 | 
			
		||||
                    EnumTypeState::TagEnumAndBody(_) | EnumTypeState::TagUIntAndBody(_) => {
 | 
			
		||||
                        match op.variant_type().ty {
 | 
			
		||||
                            Some(_) => op
 | 
			
		||||
                                .base()
 | 
			
		||||
                                .expr_enum()
 | 
			
		||||
                                .fold(self)?
 | 
			
		||||
                                .to_expr()
 | 
			
		||||
                                .with_type::<TagAndBody<DynCanonicalValue>>()
 | 
			
		||||
                                .body[..op.ty().bit_width()]
 | 
			
		||||
                                .cast_bits_to::<DynCanonicalValue>(op.ty())
 | 
			
		||||
                                .expr_enum(),
 | 
			
		||||
                            None => ().to_expr().expr_enum(),
 | 
			
		||||
                        match op.variant_type() {
 | 
			
		||||
                            Some(variant_type) => *Expr::expr_enum(
 | 
			
		||||
                                Expr::<TagAndBody<CanonicalType, DynSize>>::from_canonical(
 | 
			
		||||
                                    (*Expr::expr_enum(op.base())).fold(self)?.to_expr(),
 | 
			
		||||
                                )
 | 
			
		||||
                                .body[..variant_type.bit_width()]
 | 
			
		||||
                                    .cast_bits_to(variant_type),
 | 
			
		||||
                            ),
 | 
			
		||||
                            None => *Expr::expr_enum(().to_expr()),
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    EnumTypeState::UInt(_) => match op.variant_type().ty {
 | 
			
		||||
                        Some(_) => {
 | 
			
		||||
                            let base_int = op
 | 
			
		||||
                                .base()
 | 
			
		||||
                                .expr_enum()
 | 
			
		||||
                                .fold(self)?
 | 
			
		||||
                                .to_expr()
 | 
			
		||||
                                .with_type::<DynUInt>();
 | 
			
		||||
                    EnumTypeState::UInt(_) => match op.variant_type() {
 | 
			
		||||
                        Some(variant_type) => {
 | 
			
		||||
                            let base_int = Expr::<UInt>::from_canonical(
 | 
			
		||||
                                (*Expr::expr_enum(op.base())).fold(self)?.to_expr(),
 | 
			
		||||
                            );
 | 
			
		||||
                            dbg!(base_int);
 | 
			
		||||
                            let base_ty = op.base().ty();
 | 
			
		||||
                            let ty_bit_width = op.ty().bit_width();
 | 
			
		||||
                            base_int[base_ty.discriminant_bit_width()..][..ty_bit_width]
 | 
			
		||||
                                .cast_bits_to::<DynCanonicalValue>(op.ty())
 | 
			
		||||
                                .expr_enum()
 | 
			
		||||
                        }
 | 
			
		||||
                        None => ().to_expr().expr_enum(),
 | 
			
		||||
                    },
 | 
			
		||||
                    EnumTypeState::Unchanged => match op.variant_type().ty {
 | 
			
		||||
                        Some(_) => ExprEnum::VariantAccess(
 | 
			
		||||
                            ops::VariantAccess::new_unchecked(
 | 
			
		||||
                                op.base().fold(self)?,
 | 
			
		||||
                                op.variant_index(),
 | 
			
		||||
                            let base_ty = Expr::ty(op.base());
 | 
			
		||||
                            let variant_type_bit_width = variant_type.bit_width();
 | 
			
		||||
                            *Expr::expr_enum(
 | 
			
		||||
                                base_int[base_ty.discriminant_bit_width()..]
 | 
			
		||||
                                    [..variant_type_bit_width]
 | 
			
		||||
                                    .cast_bits_to(variant_type),
 | 
			
		||||
                            )
 | 
			
		||||
                            .intern_sized(),
 | 
			
		||||
                        ),
 | 
			
		||||
                        None => ().to_expr().expr_enum(),
 | 
			
		||||
                        }
 | 
			
		||||
                        None => *Expr::expr_enum(().to_expr()),
 | 
			
		||||
                    },
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
                    EnumTypeState::Unchanged => match op.variant_type() {
 | 
			
		||||
                        Some(_) => ExprEnum::VariantAccess(ops::VariantAccess::new_by_index(
 | 
			
		||||
                            op.base().fold(self)?,
 | 
			
		||||
                            op.variant_index(),
 | 
			
		||||
                        )),
 | 
			
		||||
                        None => *Expr::expr_enum(().to_expr()),
 | 
			
		||||
                    },
 | 
			
		||||
                },
 | 
			
		||||
            ),
 | 
			
		||||
            ExprEnum::MemPort(mem_port) => Ok(
 | 
			
		||||
                if let Some(&wire) = self.replacement_mem_ports.get(&mem_port) {
 | 
			
		||||
                    ExprEnum::Wire(wire)
 | 
			
		||||
| 
						 | 
				
			
			@ -310,24 +283,34 @@ impl Folder for State {
 | 
			
		|||
                    ExprEnum::MemPort(mem_port.fold(self)?)
 | 
			
		||||
                },
 | 
			
		||||
            ),
 | 
			
		||||
            ExprEnum::Literal(_)
 | 
			
		||||
            | ExprEnum::ArrayLiteral(_)
 | 
			
		||||
            ExprEnum::UIntLiteral(_)
 | 
			
		||||
            | ExprEnum::SIntLiteral(_)
 | 
			
		||||
            | ExprEnum::BoolLiteral(_)
 | 
			
		||||
            | ExprEnum::BundleLiteral(_)
 | 
			
		||||
            | ExprEnum::ArrayLiteral(_)
 | 
			
		||||
            | ExprEnum::NotU(_)
 | 
			
		||||
            | ExprEnum::NotS(_)
 | 
			
		||||
            | ExprEnum::NotB(_)
 | 
			
		||||
            | ExprEnum::Neg(_)
 | 
			
		||||
            | ExprEnum::BitAndU(_)
 | 
			
		||||
            | ExprEnum::BitAndS(_)
 | 
			
		||||
            | ExprEnum::BitAndB(_)
 | 
			
		||||
            | ExprEnum::BitOrU(_)
 | 
			
		||||
            | ExprEnum::BitOrS(_)
 | 
			
		||||
            | ExprEnum::BitOrB(_)
 | 
			
		||||
            | ExprEnum::BitXorU(_)
 | 
			
		||||
            | ExprEnum::BitXorS(_)
 | 
			
		||||
            | ExprEnum::BitXorB(_)
 | 
			
		||||
            | ExprEnum::AddU(_)
 | 
			
		||||
            | ExprEnum::AddS(_)
 | 
			
		||||
            | ExprEnum::SubU(_)
 | 
			
		||||
            | ExprEnum::SubS(_)
 | 
			
		||||
            | ExprEnum::MulU(_)
 | 
			
		||||
            | ExprEnum::MulS(_)
 | 
			
		||||
            | ExprEnum::DivU(_)
 | 
			
		||||
            | ExprEnum::DivS(_)
 | 
			
		||||
            | ExprEnum::RemU(_)
 | 
			
		||||
            | ExprEnum::RemS(_)
 | 
			
		||||
            | ExprEnum::DynShlU(_)
 | 
			
		||||
            | ExprEnum::DynShlS(_)
 | 
			
		||||
            | ExprEnum::DynShrU(_)
 | 
			
		||||
| 
						 | 
				
			
			@ -336,41 +319,68 @@ impl Folder for State {
 | 
			
		|||
            | ExprEnum::FixedShlS(_)
 | 
			
		||||
            | ExprEnum::FixedShrU(_)
 | 
			
		||||
            | ExprEnum::FixedShrS(_)
 | 
			
		||||
            | ExprEnum::CmpLtB(_)
 | 
			
		||||
            | ExprEnum::CmpLeB(_)
 | 
			
		||||
            | ExprEnum::CmpGtB(_)
 | 
			
		||||
            | ExprEnum::CmpGeB(_)
 | 
			
		||||
            | ExprEnum::CmpEqB(_)
 | 
			
		||||
            | ExprEnum::CmpNeB(_)
 | 
			
		||||
            | ExprEnum::CmpLtU(_)
 | 
			
		||||
            | ExprEnum::CmpLtS(_)
 | 
			
		||||
            | ExprEnum::CmpLeU(_)
 | 
			
		||||
            | ExprEnum::CmpLeS(_)
 | 
			
		||||
            | ExprEnum::CmpGtU(_)
 | 
			
		||||
            | ExprEnum::CmpGtS(_)
 | 
			
		||||
            | ExprEnum::CmpGeU(_)
 | 
			
		||||
            | ExprEnum::CmpGeS(_)
 | 
			
		||||
            | ExprEnum::CmpEqU(_)
 | 
			
		||||
            | ExprEnum::CmpEqS(_)
 | 
			
		||||
            | ExprEnum::CmpNeU(_)
 | 
			
		||||
            | ExprEnum::CmpLtS(_)
 | 
			
		||||
            | ExprEnum::CmpLeS(_)
 | 
			
		||||
            | ExprEnum::CmpGtS(_)
 | 
			
		||||
            | ExprEnum::CmpGeS(_)
 | 
			
		||||
            | ExprEnum::CmpEqS(_)
 | 
			
		||||
            | ExprEnum::CmpNeS(_)
 | 
			
		||||
            | ExprEnum::CastUIntToUInt(_)
 | 
			
		||||
            | ExprEnum::CastUIntToSInt(_)
 | 
			
		||||
            | ExprEnum::CastSIntToUInt(_)
 | 
			
		||||
            | ExprEnum::CastSIntToSInt(_)
 | 
			
		||||
            | ExprEnum::SliceUInt(_)
 | 
			
		||||
            | ExprEnum::SliceSInt(_)
 | 
			
		||||
            | ExprEnum::ReduceBitAnd(_)
 | 
			
		||||
            | ExprEnum::ReduceBitOr(_)
 | 
			
		||||
            | ExprEnum::ReduceBitXor(_)
 | 
			
		||||
            | ExprEnum::CastBoolToUInt(_)
 | 
			
		||||
            | ExprEnum::CastBoolToSInt(_)
 | 
			
		||||
            | ExprEnum::CastUIntToBool(_)
 | 
			
		||||
            | ExprEnum::CastSIntToBool(_)
 | 
			
		||||
            | ExprEnum::CastBoolToSyncReset(_)
 | 
			
		||||
            | ExprEnum::CastUIntToSyncReset(_)
 | 
			
		||||
            | ExprEnum::CastSIntToSyncReset(_)
 | 
			
		||||
            | ExprEnum::CastBoolToAsyncReset(_)
 | 
			
		||||
            | ExprEnum::CastUIntToAsyncReset(_)
 | 
			
		||||
            | ExprEnum::CastSIntToAsyncReset(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToBool(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToUInt(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToSInt(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToReset(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToBool(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToUInt(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToSInt(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToReset(_)
 | 
			
		||||
            | ExprEnum::CastResetToBool(_)
 | 
			
		||||
            | ExprEnum::CastResetToUInt(_)
 | 
			
		||||
            | ExprEnum::CastResetToSInt(_)
 | 
			
		||||
            | ExprEnum::CastBoolToClock(_)
 | 
			
		||||
            | ExprEnum::CastUIntToClock(_)
 | 
			
		||||
            | ExprEnum::CastSIntToClock(_)
 | 
			
		||||
            | ExprEnum::CastClockToBool(_)
 | 
			
		||||
            | ExprEnum::CastClockToUInt(_)
 | 
			
		||||
            | ExprEnum::CastClockToSInt(_)
 | 
			
		||||
            | ExprEnum::FieldAccess(_)
 | 
			
		||||
            | ExprEnum::ArrayIndex(_)
 | 
			
		||||
            | ExprEnum::DynArrayIndex(_)
 | 
			
		||||
            | ExprEnum::ReduceBitAndU(_)
 | 
			
		||||
            | ExprEnum::ReduceBitAndS(_)
 | 
			
		||||
            | ExprEnum::ReduceBitOrU(_)
 | 
			
		||||
            | ExprEnum::ReduceBitOrS(_)
 | 
			
		||||
            | ExprEnum::ReduceBitXorU(_)
 | 
			
		||||
            | ExprEnum::ReduceBitXorS(_)
 | 
			
		||||
            | ExprEnum::SliceUInt(_)
 | 
			
		||||
            | ExprEnum::SliceSInt(_)
 | 
			
		||||
            | ExprEnum::CastToBits(_)
 | 
			
		||||
            | ExprEnum::CastBitsTo(_)
 | 
			
		||||
            | ExprEnum::CastBitToClock(_)
 | 
			
		||||
            | ExprEnum::CastBitToSyncReset(_)
 | 
			
		||||
            | ExprEnum::CastBitToAsyncReset(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToReset(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToReset(_)
 | 
			
		||||
            | ExprEnum::CastClockToBit(_)
 | 
			
		||||
            | ExprEnum::CastSyncResetToBit(_)
 | 
			
		||||
            | ExprEnum::CastAsyncResetToBit(_)
 | 
			
		||||
            | ExprEnum::CastResetToBit(_)
 | 
			
		||||
            | ExprEnum::ModuleIO(_)
 | 
			
		||||
            | ExprEnum::Instance(_)
 | 
			
		||||
            | ExprEnum::Wire(_)
 | 
			
		||||
| 
						 | 
				
			
			@ -382,7 +392,7 @@ impl Folder for State {
 | 
			
		|||
        let mut memories = vec![];
 | 
			
		||||
        let mut stmts = vec![];
 | 
			
		||||
        for memory in block.memories {
 | 
			
		||||
            let old_element_ty = *memory.array_type().element();
 | 
			
		||||
            let old_element_ty = memory.array_type().element();
 | 
			
		||||
            let new_element_ty = memory.array_type().element().fold(self)?;
 | 
			
		||||
            if new_element_ty != old_element_ty {
 | 
			
		||||
                let mut new_ports = vec![];
 | 
			
		||||
| 
						 | 
				
			
			@ -394,7 +404,7 @@ impl Folder for State {
 | 
			
		|||
                        port.addr_type(),
 | 
			
		||||
                        new_element_ty,
 | 
			
		||||
                    );
 | 
			
		||||
                    new_ports.push(new_port.intern());
 | 
			
		||||
                    new_ports.push(new_port);
 | 
			
		||||
                    let new_port_ty = new_port.ty();
 | 
			
		||||
                    let mut wire_ty_fields = Vec::from_iter(new_port_ty.fields());
 | 
			
		||||
                    if let Some(wmask_name) = new_port.port_kind().wmask_name() {
 | 
			
		||||
| 
						 | 
				
			
			@ -404,7 +414,7 @@ impl Folder for State {
 | 
			
		|||
                            .unwrap();
 | 
			
		||||
                        wire_ty_fields[index].ty = port.ty().fields()[index].ty;
 | 
			
		||||
                    }
 | 
			
		||||
                    let wire_ty = DynBundleType::new(Intern::intern_owned(wire_ty_fields));
 | 
			
		||||
                    let wire_ty = Bundle::new(Intern::intern_owned(wire_ty_fields));
 | 
			
		||||
                    if wire_ty == new_port_ty {
 | 
			
		||||
                        continue;
 | 
			
		||||
                    }
 | 
			
		||||
| 
						 | 
				
			
			@ -419,23 +429,22 @@ impl Folder for State {
 | 
			
		|||
                    stmts.push(
 | 
			
		||||
                        StmtWire {
 | 
			
		||||
                            annotations: Default::default(),
 | 
			
		||||
                            wire: wire.to_dyn_canonical_wire(),
 | 
			
		||||
                            wire: wire.canonical(),
 | 
			
		||||
                        }
 | 
			
		||||
                        .into(),
 | 
			
		||||
                    );
 | 
			
		||||
                    connect_port(
 | 
			
		||||
                        &mut stmts,
 | 
			
		||||
                        new_port.to_expr().to_canonical_dyn(),
 | 
			
		||||
                        wire.to_expr().to_canonical_dyn(),
 | 
			
		||||
                        Expr::canonical(new_port.to_expr()),
 | 
			
		||||
                        Expr::canonical(wire.to_expr()),
 | 
			
		||||
                        port.source_location(),
 | 
			
		||||
                    );
 | 
			
		||||
                    self.replacement_mem_ports
 | 
			
		||||
                        .insert(port, wire.to_dyn_canonical_wire().intern_sized());
 | 
			
		||||
                    self.replacement_mem_ports.insert(port, wire.canonical());
 | 
			
		||||
                }
 | 
			
		||||
                memories.push(Mem::new_unchecked(
 | 
			
		||||
                    memory.scoped_name(),
 | 
			
		||||
                    memory.source_location(),
 | 
			
		||||
                    ArrayType::new_slice(new_element_ty, memory.array_type().len()),
 | 
			
		||||
                    ArrayType::new_dyn(new_element_ty, memory.array_type().len()),
 | 
			
		||||
                    memory.initial_value(),
 | 
			
		||||
                    Intern::intern_owned(new_ports),
 | 
			
		||||
                    memory.read_latency(),
 | 
			
		||||
| 
						 | 
				
			
			@ -458,7 +467,7 @@ impl Folder for State {
 | 
			
		|||
    fn fold_stmt(&mut self, stmt: Stmt) -> Result<Stmt, Self::Error> {
 | 
			
		||||
        fn match_int_tag(
 | 
			
		||||
            state: &mut State,
 | 
			
		||||
            int_tag_expr: Expr<DynUInt>,
 | 
			
		||||
            int_tag_expr: Expr<UInt>,
 | 
			
		||||
            source_location: SourceLocation,
 | 
			
		||||
            blocks: Interned<[Block]>,
 | 
			
		||||
        ) -> Result<StmtIf, SimplifyEnumsError> {
 | 
			
		||||
| 
						 | 
				
			
			@ -473,16 +482,15 @@ impl Folder for State {
 | 
			
		|||
                });
 | 
			
		||||
            };
 | 
			
		||||
            let mut retval = StmtIf {
 | 
			
		||||
                cond: int_tag_expr.cmp_eq(DynUInt::with_type(
 | 
			
		||||
                    int_tag_expr.ty(),
 | 
			
		||||
                    next_to_last_variant_index,
 | 
			
		||||
                )),
 | 
			
		||||
                cond: int_tag_expr
 | 
			
		||||
                    .cmp_eq(Expr::ty(int_tag_expr).from_int_wrapping(next_to_last_variant_index)),
 | 
			
		||||
                source_location,
 | 
			
		||||
                blocks: [next_to_last_block.fold(state)?, last_block.fold(state)?],
 | 
			
		||||
            };
 | 
			
		||||
            for (variant_index, block) in blocks_iter.rev() {
 | 
			
		||||
                retval = StmtIf {
 | 
			
		||||
                    cond: int_tag_expr.cmp_eq(DynUInt::with_type(int_tag_expr.ty(), variant_index)),
 | 
			
		||||
                    cond: int_tag_expr
 | 
			
		||||
                        .cmp_eq(Expr::ty(int_tag_expr).from_int_wrapping(variant_index)),
 | 
			
		||||
                    source_location,
 | 
			
		||||
                    blocks: [
 | 
			
		||||
                        block.fold(state)?,
 | 
			
		||||
| 
						 | 
				
			
			@ -500,33 +508,27 @@ impl Folder for State {
 | 
			
		|||
                expr,
 | 
			
		||||
                source_location,
 | 
			
		||||
                blocks,
 | 
			
		||||
            }) => match self.get_or_make_enum_type_state(expr.ty())? {
 | 
			
		||||
            }) => match self.get_or_make_enum_type_state(Expr::ty(expr))? {
 | 
			
		||||
                EnumTypeState::TagEnumAndBody(_) => Ok(StmtMatch {
 | 
			
		||||
                    expr: expr
 | 
			
		||||
                        .expr_enum()
 | 
			
		||||
                        .fold(self)?
 | 
			
		||||
                        .to_expr()
 | 
			
		||||
                        .with_type::<TagAndBody<DynEnum>>()
 | 
			
		||||
                        .tag,
 | 
			
		||||
                    expr: Expr::<TagAndBody<Enum, DynSize>>::from_canonical(
 | 
			
		||||
                        Expr::canonical(expr).fold(self)?,
 | 
			
		||||
                    )
 | 
			
		||||
                    .tag,
 | 
			
		||||
                    source_location,
 | 
			
		||||
                    blocks: blocks.fold(self)?,
 | 
			
		||||
                }
 | 
			
		||||
                .into()),
 | 
			
		||||
                EnumTypeState::TagUIntAndBody(_) => {
 | 
			
		||||
                    let int_tag_expr = expr
 | 
			
		||||
                        .expr_enum()
 | 
			
		||||
                        .fold(self)?
 | 
			
		||||
                        .to_expr()
 | 
			
		||||
                        .with_type::<TagAndBody<DynUInt>>()
 | 
			
		||||
                        .tag;
 | 
			
		||||
                    let int_tag_expr = Expr::<TagAndBody<UInt, DynSize>>::from_canonical(
 | 
			
		||||
                        Expr::canonical(expr).fold(self)?,
 | 
			
		||||
                    )
 | 
			
		||||
                    .tag;
 | 
			
		||||
                    Ok(match_int_tag(self, int_tag_expr, source_location, blocks)?.into())
 | 
			
		||||
                }
 | 
			
		||||
                EnumTypeState::UInt(_) => {
 | 
			
		||||
                    let int_tag_expr = expr
 | 
			
		||||
                        .expr_enum()
 | 
			
		||||
                        .fold(self)?
 | 
			
		||||
                        .to_expr()
 | 
			
		||||
                        .with_type::<DynUInt>()[..expr.ty().discriminant_bit_width()];
 | 
			
		||||
                    let int_tag_expr =
 | 
			
		||||
                        Expr::<UInt>::from_canonical(Expr::canonical(expr).fold(self)?)
 | 
			
		||||
                            [..Expr::ty(expr).discriminant_bit_width()];
 | 
			
		||||
                    Ok(match_int_tag(self, int_tag_expr, source_location, blocks)?.into())
 | 
			
		||||
                }
 | 
			
		||||
                EnumTypeState::Unchanged => Ok(StmtMatch {
 | 
			
		||||
| 
						 | 
				
			
			@ -544,92 +546,49 @@ impl Folder for State {
 | 
			
		|||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_type_enum(&mut self, type_enum: TypeEnum) -> Result<TypeEnum, Self::Error> {
 | 
			
		||||
        match type_enum {
 | 
			
		||||
            TypeEnum::EnumType(enum_type) => {
 | 
			
		||||
    fn fold_canonical_type(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        canonical_type: CanonicalType,
 | 
			
		||||
    ) -> Result<CanonicalType, Self::Error> {
 | 
			
		||||
        match canonical_type {
 | 
			
		||||
            CanonicalType::Enum(enum_type) => {
 | 
			
		||||
                Ok(match self.get_or_make_enum_type_state(enum_type)? {
 | 
			
		||||
                    EnumTypeState::TagEnumAndBody(ty) => TypeEnum::BundleType(ty.canonical()),
 | 
			
		||||
                    EnumTypeState::TagUIntAndBody(ty) => TypeEnum::BundleType(ty.canonical()),
 | 
			
		||||
                    EnumTypeState::UInt(ty) => TypeEnum::UInt(ty),
 | 
			
		||||
                    EnumTypeState::Unchanged => TypeEnum::EnumType(enum_type),
 | 
			
		||||
                    EnumTypeState::TagEnumAndBody(ty) => ty.canonical(),
 | 
			
		||||
                    EnumTypeState::TagUIntAndBody(ty) => ty.canonical(),
 | 
			
		||||
                    EnumTypeState::UInt(ty) => ty.canonical(),
 | 
			
		||||
                    EnumTypeState::Unchanged => enum_type.canonical(),
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
            TypeEnum::BundleType(_)
 | 
			
		||||
            | TypeEnum::ArrayType(_)
 | 
			
		||||
            | TypeEnum::UInt(_)
 | 
			
		||||
            | TypeEnum::SInt(_)
 | 
			
		||||
            | TypeEnum::Clock(_)
 | 
			
		||||
            | TypeEnum::AsyncReset(_)
 | 
			
		||||
            | TypeEnum::SyncReset(_)
 | 
			
		||||
            | TypeEnum::Reset(_) => type_enum.default_fold(self),
 | 
			
		||||
            CanonicalType::Bundle(_)
 | 
			
		||||
            | CanonicalType::Array(_)
 | 
			
		||||
            | CanonicalType::UInt(_)
 | 
			
		||||
            | CanonicalType::SInt(_)
 | 
			
		||||
            | CanonicalType::Bool(_)
 | 
			
		||||
            | CanonicalType::Clock(_)
 | 
			
		||||
            | CanonicalType::AsyncReset(_)
 | 
			
		||||
            | CanonicalType::SyncReset(_)
 | 
			
		||||
            | CanonicalType::Reset(_) => canonical_type.default_fold(self),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_value_enum(&mut self, value_enum: ValueEnum) -> Result<ValueEnum, Self::Error> {
 | 
			
		||||
        match value_enum {
 | 
			
		||||
            ValueEnum::Enum(enum_value) => {
 | 
			
		||||
                Ok(match self.get_or_make_enum_type_state(enum_value.ty())? {
 | 
			
		||||
                    EnumTypeState::TagEnumAndBody(TagAndBodyType::<DynEnum> {
 | 
			
		||||
                        tag: tag_ty,
 | 
			
		||||
                        body: body_ty,
 | 
			
		||||
                    }) => ValueEnum::Bundle(
 | 
			
		||||
                        TagAndBody {
 | 
			
		||||
                            tag: DynEnum::new_by_index(tag_ty, enum_value.variant_index(), None),
 | 
			
		||||
                            body: value_to_uint(enum_value.variant_value().as_ref(), body_ty),
 | 
			
		||||
                        }
 | 
			
		||||
                        .to_canonical(),
 | 
			
		||||
                    ),
 | 
			
		||||
                    EnumTypeState::TagUIntAndBody(TagAndBodyType::<DynUInt> {
 | 
			
		||||
                        tag: tag_ty,
 | 
			
		||||
                        body: body_ty,
 | 
			
		||||
                    }) => ValueEnum::Bundle(
 | 
			
		||||
                        TagAndBody {
 | 
			
		||||
                            tag: DynUInt::with_type(tag_ty, enum_value.variant_index()),
 | 
			
		||||
                            body: value_to_uint(enum_value.variant_value().as_ref(), body_ty),
 | 
			
		||||
                        }
 | 
			
		||||
                        .to_canonical(),
 | 
			
		||||
                    ),
 | 
			
		||||
                    EnumTypeState::UInt(target_ty) => {
 | 
			
		||||
                        ValueEnum::UInt(value_to_uint(Some(&enum_value), target_ty))
 | 
			
		||||
                    }
 | 
			
		||||
                    EnumTypeState::Unchanged => ValueEnum::Enum(enum_value),
 | 
			
		||||
                })
 | 
			
		||||
            }
 | 
			
		||||
            ValueEnum::Bundle(_)
 | 
			
		||||
            | ValueEnum::Array(_)
 | 
			
		||||
            | ValueEnum::UInt(_)
 | 
			
		||||
            | ValueEnum::SInt(_)
 | 
			
		||||
            | ValueEnum::Clock(_)
 | 
			
		||||
            | ValueEnum::AsyncReset(_)
 | 
			
		||||
            | ValueEnum::SyncReset(_)
 | 
			
		||||
            | ValueEnum::Reset(_) => value_enum.default_fold(self),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_variant_type<T>(&mut self, _v: VariantType<T>) -> Result<VariantType<T>, Self::Error> {
 | 
			
		||||
    fn fold_enum_variant(&mut self, _v: EnumVariant) -> Result<EnumVariant, Self::Error> {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_enum_literal<EnumTy>(
 | 
			
		||||
    fn fold_enum_literal<T: EnumType>(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        _v: ops::EnumLiteral<EnumTy>,
 | 
			
		||||
    ) -> Result<ops::EnumLiteral<EnumTy>, Self::Error>
 | 
			
		||||
        _v: ops::EnumLiteral<T>,
 | 
			
		||||
    ) -> Result<ops::EnumLiteral<T>, Self::Error>
 | 
			
		||||
    where
 | 
			
		||||
        EnumTy: EnumType,
 | 
			
		||||
        EnumTy::Value: EnumValue<Type = EnumTy>,
 | 
			
		||||
        T: Fold<Self>,
 | 
			
		||||
    {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_variant_access<T, VariantTy>(
 | 
			
		||||
    fn fold_variant_access<VariantType: Type>(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        _v: ops::VariantAccess<T, VariantTy>,
 | 
			
		||||
    ) -> Result<ops::VariantAccess<T, VariantTy>, Self::Error>
 | 
			
		||||
    where
 | 
			
		||||
        T: EnumType,
 | 
			
		||||
        T::Value: EnumValue,
 | 
			
		||||
        VariantTy: Type,
 | 
			
		||||
    {
 | 
			
		||||
        _v: ops::VariantAccess<VariantType>,
 | 
			
		||||
    ) -> Result<ops::VariantAccess<VariantType>, Self::Error> {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -642,9 +601,9 @@ pub enum SimplifyEnumsKind {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
pub fn simplify_enums(
 | 
			
		||||
    module: Interned<Module<DynBundle>>,
 | 
			
		||||
    module: Interned<Module<Bundle>>,
 | 
			
		||||
    kind: SimplifyEnumsKind,
 | 
			
		||||
) -> Result<Interned<Module<DynBundle>>, SimplifyEnumsError> {
 | 
			
		||||
) -> Result<Interned<Module<Bundle>>, SimplifyEnumsError> {
 | 
			
		||||
    module.fold(&mut State {
 | 
			
		||||
        enum_types: HashMap::new(),
 | 
			
		||||
        replacement_mem_ports: HashMap::new(),
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,10 +2,10 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    annotations::TargetedAnnotation,
 | 
			
		||||
    array::{Array, ArrayType, ValueArrayOrSlice},
 | 
			
		||||
    bundle::{BundleType, BundleValue, DynBundle},
 | 
			
		||||
    expr::{Expr, ExprEnum, ToExpr},
 | 
			
		||||
    int::{DynSInt, DynSIntType, DynUInt, DynUIntType},
 | 
			
		||||
    array::Array,
 | 
			
		||||
    bundle::{Bundle, BundleType},
 | 
			
		||||
    expr::{CastBitsTo, CastToBits, Expr, ExprEnum, ToExpr},
 | 
			
		||||
    int::{Bool, SInt, Size, UInt},
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    memory::{Mem, MemPort, PortType},
 | 
			
		||||
    module::{
 | 
			
		||||
| 
						 | 
				
			
			@ -13,7 +13,7 @@ use crate::{
 | 
			
		|||
        Block, Module, NameId, NameIdGen, ScopedNameId, Stmt, StmtConnect, StmtWire,
 | 
			
		||||
    },
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{DynCanonicalValue, DynType, Type, TypeEnum},
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
    util::MakeMutSlice,
 | 
			
		||||
    wire::Wire,
 | 
			
		||||
};
 | 
			
		||||
| 
						 | 
				
			
			@ -28,26 +28,31 @@ use std::{
 | 
			
		|||
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
 | 
			
		||||
enum SingleType {
 | 
			
		||||
    UInt(DynUIntType),
 | 
			
		||||
    SInt(DynSIntType),
 | 
			
		||||
    UIntArray(ArrayType<[DynUInt]>),
 | 
			
		||||
    SIntArray(ArrayType<[DynSInt]>),
 | 
			
		||||
    UInt(UInt),
 | 
			
		||||
    SInt(SInt),
 | 
			
		||||
    Bool(Bool),
 | 
			
		||||
    UIntArray(Array<UInt>),
 | 
			
		||||
    SIntArray(Array<SInt>),
 | 
			
		||||
    BoolArray(Array<Bool>),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl SingleType {
 | 
			
		||||
    fn is_array_type(self, array_type: ArrayType<[DynCanonicalValue]>) -> bool {
 | 
			
		||||
    fn is_array_type(self, array_type: Array) -> bool {
 | 
			
		||||
        match self {
 | 
			
		||||
            SingleType::UInt(_) | SingleType::SInt(_) => false,
 | 
			
		||||
            SingleType::UIntArray(ty) => ty.canonical() == array_type,
 | 
			
		||||
            SingleType::SIntArray(ty) => ty.canonical() == array_type,
 | 
			
		||||
            SingleType::UInt(_) | SingleType::SInt(_) | SingleType::Bool(_) => false,
 | 
			
		||||
            SingleType::UIntArray(ty) => ty.as_dyn_array() == array_type,
 | 
			
		||||
            SingleType::SIntArray(ty) => ty.as_dyn_array() == array_type,
 | 
			
		||||
            SingleType::BoolArray(ty) => ty.as_dyn_array() == array_type,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    fn array_len(self) -> usize {
 | 
			
		||||
        match self {
 | 
			
		||||
            SingleType::UInt(_ty) => 1,
 | 
			
		||||
            SingleType::SInt(_ty) => 1,
 | 
			
		||||
            SingleType::Bool(_ty) => 1,
 | 
			
		||||
            SingleType::UIntArray(ty) => ty.len(),
 | 
			
		||||
            SingleType::SIntArray(ty) => ty.len(),
 | 
			
		||||
            SingleType::BoolArray(ty) => ty.len(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -58,7 +63,7 @@ enum MemSplit {
 | 
			
		|||
        fields: Rc<[MemSplit]>,
 | 
			
		||||
    },
 | 
			
		||||
    Single {
 | 
			
		||||
        output_mem: Option<Mem<[DynCanonicalValue]>>,
 | 
			
		||||
        output_mem: Option<Mem>,
 | 
			
		||||
        element_type: SingleType,
 | 
			
		||||
        unchanged_element_type: bool,
 | 
			
		||||
    },
 | 
			
		||||
| 
						 | 
				
			
			@ -83,17 +88,17 @@ impl MemSplit {
 | 
			
		|||
            MemSplit::Array { elements: _ } => self,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    fn new(element_type: TypeEnum) -> Self {
 | 
			
		||||
    fn new(element_type: CanonicalType) -> Self {
 | 
			
		||||
        match element_type {
 | 
			
		||||
            TypeEnum::BundleType(bundle_ty) => MemSplit::Bundle {
 | 
			
		||||
            CanonicalType::Bundle(bundle_ty) => MemSplit::Bundle {
 | 
			
		||||
                fields: bundle_ty
 | 
			
		||||
                    .fields()
 | 
			
		||||
                    .into_iter()
 | 
			
		||||
                    .map(|field| Self::new(field.ty.type_enum()).mark_changed_element_type())
 | 
			
		||||
                    .map(|field| Self::new(field.ty).mark_changed_element_type())
 | 
			
		||||
                    .collect(),
 | 
			
		||||
            },
 | 
			
		||||
            TypeEnum::ArrayType(ty) => {
 | 
			
		||||
                let element = MemSplit::new(ty.element().type_enum());
 | 
			
		||||
            CanonicalType::Array(ty) => {
 | 
			
		||||
                let element = MemSplit::new(ty.element());
 | 
			
		||||
                if let Self::Single {
 | 
			
		||||
                    output_mem: _,
 | 
			
		||||
                    element_type,
 | 
			
		||||
| 
						 | 
				
			
			@ -103,7 +108,7 @@ impl MemSplit {
 | 
			
		|||
                    match element_type {
 | 
			
		||||
                        SingleType::UInt(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::UIntArray(ArrayType::new_slice(
 | 
			
		||||
                            element_type: SingleType::UIntArray(Array::new_dyn(
 | 
			
		||||
                                element_type,
 | 
			
		||||
                                ty.len(),
 | 
			
		||||
                            )),
 | 
			
		||||
| 
						 | 
				
			
			@ -111,7 +116,15 @@ impl MemSplit {
 | 
			
		|||
                        },
 | 
			
		||||
                        SingleType::SInt(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::SIntArray(ArrayType::new_slice(
 | 
			
		||||
                            element_type: SingleType::SIntArray(Array::new_dyn(
 | 
			
		||||
                                element_type,
 | 
			
		||||
                                ty.len(),
 | 
			
		||||
                            )),
 | 
			
		||||
                            unchanged_element_type,
 | 
			
		||||
                        },
 | 
			
		||||
                        SingleType::Bool(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::BoolArray(Array::new_dyn(
 | 
			
		||||
                                element_type,
 | 
			
		||||
                                ty.len(),
 | 
			
		||||
                            )),
 | 
			
		||||
| 
						 | 
				
			
			@ -119,8 +132,8 @@ impl MemSplit {
 | 
			
		|||
                        },
 | 
			
		||||
                        SingleType::UIntArray(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::UIntArray(ArrayType::new_slice(
 | 
			
		||||
                                *element_type.element(),
 | 
			
		||||
                            element_type: SingleType::UIntArray(Array::new_dyn(
 | 
			
		||||
                                element_type.element(),
 | 
			
		||||
                                ty.len()
 | 
			
		||||
                                    .checked_mul(element_type.len())
 | 
			
		||||
                                    .expect("memory element type can't be too big"),
 | 
			
		||||
| 
						 | 
				
			
			@ -129,8 +142,18 @@ impl MemSplit {
 | 
			
		|||
                        },
 | 
			
		||||
                        SingleType::SIntArray(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::SIntArray(ArrayType::new_slice(
 | 
			
		||||
                                *element_type.element(),
 | 
			
		||||
                            element_type: SingleType::SIntArray(Array::new_dyn(
 | 
			
		||||
                                element_type.element(),
 | 
			
		||||
                                ty.len()
 | 
			
		||||
                                    .checked_mul(element_type.len())
 | 
			
		||||
                                    .expect("memory element type can't be too big"),
 | 
			
		||||
                            )),
 | 
			
		||||
                            unchanged_element_type: false,
 | 
			
		||||
                        },
 | 
			
		||||
                        SingleType::BoolArray(element_type) => Self::Single {
 | 
			
		||||
                            output_mem: None,
 | 
			
		||||
                            element_type: SingleType::BoolArray(Array::new_dyn(
 | 
			
		||||
                                element_type.element(),
 | 
			
		||||
                                ty.len()
 | 
			
		||||
                                    .checked_mul(element_type.len())
 | 
			
		||||
                                    .expect("memory element type can't be too big"),
 | 
			
		||||
| 
						 | 
				
			
			@ -145,25 +168,30 @@ impl MemSplit {
 | 
			
		|||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            TypeEnum::UInt(ty) => Self::Single {
 | 
			
		||||
            CanonicalType::UInt(ty) => Self::Single {
 | 
			
		||||
                output_mem: None,
 | 
			
		||||
                element_type: SingleType::UInt(ty),
 | 
			
		||||
                unchanged_element_type: true,
 | 
			
		||||
            },
 | 
			
		||||
            TypeEnum::SInt(ty) => Self::Single {
 | 
			
		||||
            CanonicalType::SInt(ty) => Self::Single {
 | 
			
		||||
                output_mem: None,
 | 
			
		||||
                element_type: SingleType::SInt(ty),
 | 
			
		||||
                unchanged_element_type: true,
 | 
			
		||||
            },
 | 
			
		||||
            TypeEnum::EnumType(ty) => Self::Single {
 | 
			
		||||
            CanonicalType::Bool(ty) => Self::Single {
 | 
			
		||||
                output_mem: None,
 | 
			
		||||
                element_type: SingleType::UInt(DynUIntType::new(ty.bit_width())),
 | 
			
		||||
                element_type: SingleType::Bool(ty),
 | 
			
		||||
                unchanged_element_type: true,
 | 
			
		||||
            },
 | 
			
		||||
            CanonicalType::Enum(ty) => Self::Single {
 | 
			
		||||
                output_mem: None,
 | 
			
		||||
                element_type: SingleType::UInt(UInt::new_dyn(ty.type_properties().bit_width)),
 | 
			
		||||
                unchanged_element_type: false,
 | 
			
		||||
            },
 | 
			
		||||
            TypeEnum::Clock(_)
 | 
			
		||||
            | TypeEnum::AsyncReset(_)
 | 
			
		||||
            | TypeEnum::SyncReset(_)
 | 
			
		||||
            | TypeEnum::Reset(_) => unreachable!("memory element type is a storable type"),
 | 
			
		||||
            CanonicalType::Clock(_)
 | 
			
		||||
            | CanonicalType::AsyncReset(_)
 | 
			
		||||
            | CanonicalType::SyncReset(_)
 | 
			
		||||
            | CanonicalType::Reset(_) => unreachable!("memory element type is a storable type"),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -173,9 +201,9 @@ struct MemState {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
struct SplitState<'a> {
 | 
			
		||||
    wire_rdata: Box<[Option<Expr<DynCanonicalValue>>]>,
 | 
			
		||||
    wire_wdata: Box<[Option<Expr<DynCanonicalValue>>]>,
 | 
			
		||||
    wire_wmask: Box<[Option<Expr<DynCanonicalValue>>]>,
 | 
			
		||||
    wire_rdata: Box<[Option<Expr<CanonicalType>>]>,
 | 
			
		||||
    wire_wdata: Box<[Option<Expr<CanonicalType>>]>,
 | 
			
		||||
    wire_wmask: Box<[Option<Expr<CanonicalType>>]>,
 | 
			
		||||
    initial_value: Option<Box<[&'a BitSlice]>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -223,7 +251,7 @@ impl<'a> SplitStateStack<'a> {
 | 
			
		|||
    }
 | 
			
		||||
    fn push_map(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        mut wire_map: impl FnMut(Expr<DynCanonicalValue>) -> Expr<DynCanonicalValue>,
 | 
			
		||||
        mut wire_map: impl FnMut(Expr<CanonicalType>) -> Expr<CanonicalType>,
 | 
			
		||||
        mut initial_value_element_map: impl FnMut(&BitSlice) -> &BitSlice,
 | 
			
		||||
    ) {
 | 
			
		||||
        let top_index = self.top_index + 1;
 | 
			
		||||
| 
						 | 
				
			
			@ -260,10 +288,10 @@ impl<'a> SplitStateStack<'a> {
 | 
			
		|||
 | 
			
		||||
struct SplitMemState<'a, 'b> {
 | 
			
		||||
    module_state: &'a mut ModuleState,
 | 
			
		||||
    input_mem: Mem<[DynCanonicalValue]>,
 | 
			
		||||
    output_mems: &'a mut Vec<Mem<[DynCanonicalValue]>>,
 | 
			
		||||
    input_mem: Mem,
 | 
			
		||||
    output_mems: &'a mut Vec<Mem>,
 | 
			
		||||
    output_stmts: &'a mut Vec<Stmt>,
 | 
			
		||||
    element_type: TypeEnum,
 | 
			
		||||
    element_type: CanonicalType,
 | 
			
		||||
    split: &'a mut MemSplit,
 | 
			
		||||
    mem_name_path: &'a mut String,
 | 
			
		||||
    split_state_stack: &'a mut SplitStateStack<'b>,
 | 
			
		||||
| 
						 | 
				
			
			@ -275,7 +303,7 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
        let outer_mem_name_path_len = self.mem_name_path.len();
 | 
			
		||||
        match self.split {
 | 
			
		||||
            MemSplit::Bundle { fields } => {
 | 
			
		||||
                let TypeEnum::BundleType(bundle_type) = self.element_type else {
 | 
			
		||||
                let CanonicalType::Bundle(bundle_type) = self.element_type else {
 | 
			
		||||
                    unreachable!();
 | 
			
		||||
                };
 | 
			
		||||
                for ((field, field_offset), split) in bundle_type
 | 
			
		||||
| 
						 | 
				
			
			@ -289,7 +317,9 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
                    self.mem_name_path.push_str(&field.name);
 | 
			
		||||
                    let field_ty_bit_width = field.ty.bit_width();
 | 
			
		||||
                    self.split_state_stack.push_map(
 | 
			
		||||
                        |e: Expr<DynCanonicalValue>| e.with_type::<DynBundle>().field(&field.name),
 | 
			
		||||
                        |e: Expr<CanonicalType>| {
 | 
			
		||||
                            Expr::field(Expr::<Bundle>::from_canonical(e), &field.name)
 | 
			
		||||
                        },
 | 
			
		||||
                        |initial_value_element| {
 | 
			
		||||
                            &initial_value_element[field_offset..][..field_ty_bit_width]
 | 
			
		||||
                        },
 | 
			
		||||
| 
						 | 
				
			
			@ -299,7 +329,7 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
                        input_mem: self.input_mem,
 | 
			
		||||
                        output_mems: self.output_mems,
 | 
			
		||||
                        output_stmts: self.output_stmts,
 | 
			
		||||
                        element_type: field.ty.type_enum(),
 | 
			
		||||
                        element_type: field.ty,
 | 
			
		||||
                        split,
 | 
			
		||||
                        mem_name_path: self.mem_name_path,
 | 
			
		||||
                        split_state_stack: self.split_state_stack,
 | 
			
		||||
| 
						 | 
				
			
			@ -328,7 +358,7 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
                    .zip(self.mem_state.replacement_ports.iter())
 | 
			
		||||
                {
 | 
			
		||||
                    let port_expr = port.to_expr();
 | 
			
		||||
                    let wire_expr = Expr::<DynBundle>::new_unchecked(*wire);
 | 
			
		||||
                    let wire_expr = Expr::<Bundle>::from_canonical(wire.to_expr());
 | 
			
		||||
                    for name in [
 | 
			
		||||
                        Some("addr"),
 | 
			
		||||
                        Some("clk"),
 | 
			
		||||
| 
						 | 
				
			
			@ -340,8 +370,8 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
                        };
 | 
			
		||||
                        self.output_stmts.push(
 | 
			
		||||
                            StmtConnect {
 | 
			
		||||
                                lhs: port_expr.field(name),
 | 
			
		||||
                                rhs: wire_expr.field(name),
 | 
			
		||||
                                lhs: Expr::field(port_expr, name),
 | 
			
		||||
                                rhs: Expr::field(wire_expr, name),
 | 
			
		||||
                                source_location: port.source_location(),
 | 
			
		||||
                            }
 | 
			
		||||
                            .into(),
 | 
			
		||||
| 
						 | 
				
			
			@ -352,18 +382,16 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
                self.output_mems.push(new_mem);
 | 
			
		||||
            }
 | 
			
		||||
            MemSplit::Array { elements } => {
 | 
			
		||||
                let TypeEnum::ArrayType(array_type) = self.element_type else {
 | 
			
		||||
                let CanonicalType::Array(array_type) = self.element_type else {
 | 
			
		||||
                    unreachable!();
 | 
			
		||||
                };
 | 
			
		||||
                let element_type = array_type.element().type_enum();
 | 
			
		||||
                let element_bit_width = array_type.element().bit_width();
 | 
			
		||||
                let element_type = array_type.element();
 | 
			
		||||
                let element_bit_width = element_type.bit_width();
 | 
			
		||||
                for (index, split) in elements.make_mut_slice().iter_mut().enumerate() {
 | 
			
		||||
                    self.mem_name_path.truncate(outer_mem_name_path_len);
 | 
			
		||||
                    write!(self.mem_name_path, "_{index}").unwrap();
 | 
			
		||||
                    self.split_state_stack.push_map(
 | 
			
		||||
                        |e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                            e.with_type::<Array<[DynCanonicalValue]>>()[index]
 | 
			
		||||
                        },
 | 
			
		||||
                        |e| Expr::<Array>::from_canonical(e)[index],
 | 
			
		||||
                        |initial_value_element| {
 | 
			
		||||
                            &initial_value_element[index * element_bit_width..][..element_bit_width]
 | 
			
		||||
                        },
 | 
			
		||||
| 
						 | 
				
			
			@ -388,7 +416,7 @@ impl SplitMemState<'_, '_> {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
struct ModuleState {
 | 
			
		||||
    output_module: Option<Interned<Module<DynBundle>>>,
 | 
			
		||||
    output_module: Option<Interned<Module<Bundle>>>,
 | 
			
		||||
    name_id_gen: NameIdGen,
 | 
			
		||||
    memories: HashMap<ScopedNameId, MemState>,
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -397,18 +425,18 @@ impl ModuleState {
 | 
			
		|||
    #[allow(clippy::too_many_arguments)]
 | 
			
		||||
    fn connect_split_mem_port_arrays(
 | 
			
		||||
        output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
        input_array_types: &[ArrayType<[DynCanonicalValue]>],
 | 
			
		||||
        input_array_types: &[Array],
 | 
			
		||||
        memory_element_array_range_start: usize,
 | 
			
		||||
        memory_element_array_range_len: usize,
 | 
			
		||||
        wire_rdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        wire_wdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        wire_wmask: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        port_rdata: Option<Expr<Array<[DynCanonicalValue]>>>,
 | 
			
		||||
        port_wdata: Option<Expr<Array<[DynCanonicalValue]>>>,
 | 
			
		||||
        port_wmask: Option<Expr<Array<[DynCanonicalValue]>>>,
 | 
			
		||||
        connect_rdata: impl Copy + Fn(&mut Vec<Stmt>, Expr<DynCanonicalValue>, Expr<DynCanonicalValue>),
 | 
			
		||||
        connect_wdata: impl Copy + Fn(&mut Vec<Stmt>, Expr<DynCanonicalValue>, Expr<DynCanonicalValue>),
 | 
			
		||||
        connect_wmask: impl Copy + Fn(&mut Vec<Stmt>, Expr<DynCanonicalValue>, Expr<DynCanonicalValue>),
 | 
			
		||||
        wire_rdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        wire_wdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        wire_wmask: Option<Expr<CanonicalType>>,
 | 
			
		||||
        port_rdata: Option<Expr<Array>>,
 | 
			
		||||
        port_wdata: Option<Expr<Array>>,
 | 
			
		||||
        port_wmask: Option<Expr<Array>>,
 | 
			
		||||
        connect_rdata: impl Copy + Fn(&mut Vec<Stmt>, Expr<CanonicalType>, Expr<CanonicalType>),
 | 
			
		||||
        connect_wdata: impl Copy + Fn(&mut Vec<Stmt>, Expr<CanonicalType>, Expr<CanonicalType>),
 | 
			
		||||
        connect_wmask: impl Copy + Fn(&mut Vec<Stmt>, Expr<CanonicalType>, Expr<CanonicalType>),
 | 
			
		||||
    ) {
 | 
			
		||||
        let Some((input_array_type, input_array_types_rest)) = input_array_types.split_first()
 | 
			
		||||
        else {
 | 
			
		||||
| 
						 | 
				
			
			@ -430,8 +458,7 @@ impl ModuleState {
 | 
			
		|||
        assert_eq!(memory_element_array_range_len % input_array_type.len(), 0);
 | 
			
		||||
        let chunk_size = memory_element_array_range_len / input_array_type.len();
 | 
			
		||||
        for index in 0..input_array_type.len() {
 | 
			
		||||
            let map =
 | 
			
		||||
                |e: Expr<DynCanonicalValue>| e.with_type::<Array<[DynCanonicalValue]>>()[index];
 | 
			
		||||
            let map = |e| Expr::<Array>::from_canonical(e)[index];
 | 
			
		||||
            let wire_rdata = wire_rdata.map(map);
 | 
			
		||||
            let wire_wdata = wire_wdata.map(map);
 | 
			
		||||
            let wire_wmask = wire_wmask.map(map);
 | 
			
		||||
| 
						 | 
				
			
			@ -456,20 +483,20 @@ impl ModuleState {
 | 
			
		|||
    fn connect_split_mem_port(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
        mut input_element_type: TypeEnum,
 | 
			
		||||
        mut input_element_type: CanonicalType,
 | 
			
		||||
        single_type: SingleType,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        wire_rdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        wire_wdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        wire_wmask: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        port_rdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        port_wdata: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        port_wmask: Option<Expr<DynCanonicalValue>>,
 | 
			
		||||
        wire_rdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        wire_wdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        wire_wmask: Option<Expr<CanonicalType>>,
 | 
			
		||||
        port_rdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        port_wdata: Option<Expr<CanonicalType>>,
 | 
			
		||||
        port_wmask: Option<Expr<CanonicalType>>,
 | 
			
		||||
    ) {
 | 
			
		||||
        let mut input_array_types = vec![];
 | 
			
		||||
        let connect_read = |output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
                            wire_read: Expr<DynCanonicalValue>,
 | 
			
		||||
                            port_read: Expr<DynCanonicalValue>| {
 | 
			
		||||
                            wire_read: Expr<CanonicalType>,
 | 
			
		||||
                            port_read: Expr<CanonicalType>| {
 | 
			
		||||
            output_stmts.push(
 | 
			
		||||
                StmtConnect {
 | 
			
		||||
                    lhs: wire_read,
 | 
			
		||||
| 
						 | 
				
			
			@ -480,8 +507,8 @@ impl ModuleState {
 | 
			
		|||
            );
 | 
			
		||||
        };
 | 
			
		||||
        let connect_write = |output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
                             wire_write: Expr<DynCanonicalValue>,
 | 
			
		||||
                             port_write: Expr<DynCanonicalValue>| {
 | 
			
		||||
                             wire_write: Expr<CanonicalType>,
 | 
			
		||||
                             port_write: Expr<CanonicalType>| {
 | 
			
		||||
            output_stmts.push(
 | 
			
		||||
                StmtConnect {
 | 
			
		||||
                    lhs: port_write,
 | 
			
		||||
| 
						 | 
				
			
			@ -491,32 +518,29 @@ impl ModuleState {
 | 
			
		|||
                .into(),
 | 
			
		||||
            );
 | 
			
		||||
        };
 | 
			
		||||
        let connect_read_enum =
 | 
			
		||||
            |output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
             wire_read: Expr<DynCanonicalValue>,
 | 
			
		||||
             port_read: Expr<DynCanonicalValue>| {
 | 
			
		||||
                connect_read(
 | 
			
		||||
                    output_stmts,
 | 
			
		||||
                    wire_read,
 | 
			
		||||
                    port_read
 | 
			
		||||
                        .with_type::<DynUInt>()
 | 
			
		||||
                        .cast_bits_to(wire_read.ty()),
 | 
			
		||||
                );
 | 
			
		||||
            };
 | 
			
		||||
        let connect_read_enum = |output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
                                 wire_read: Expr<CanonicalType>,
 | 
			
		||||
                                 port_read: Expr<CanonicalType>| {
 | 
			
		||||
            connect_read(
 | 
			
		||||
                output_stmts,
 | 
			
		||||
                wire_read,
 | 
			
		||||
                Expr::<UInt>::from_canonical(port_read).cast_bits_to(Expr::ty(wire_read)),
 | 
			
		||||
            );
 | 
			
		||||
        };
 | 
			
		||||
        let connect_write_enum =
 | 
			
		||||
            |output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
             wire_write: Expr<DynCanonicalValue>,
 | 
			
		||||
             port_write: Expr<DynCanonicalValue>| {
 | 
			
		||||
             wire_write: Expr<CanonicalType>,
 | 
			
		||||
             port_write: Expr<CanonicalType>| {
 | 
			
		||||
                connect_write(
 | 
			
		||||
                    output_stmts,
 | 
			
		||||
                    wire_write.cast_to_bits().to_canonical_dyn(),
 | 
			
		||||
                    Expr::canonical(wire_write.cast_to_bits()),
 | 
			
		||||
                    port_write,
 | 
			
		||||
                );
 | 
			
		||||
            };
 | 
			
		||||
        loop {
 | 
			
		||||
            match input_element_type {
 | 
			
		||||
                TypeEnum::BundleType(_) => unreachable!("bundle types are always split"),
 | 
			
		||||
                TypeEnum::EnumType(_)
 | 
			
		||||
                CanonicalType::Bundle(_) => unreachable!("bundle types are always split"),
 | 
			
		||||
                CanonicalType::Enum(_)
 | 
			
		||||
                    if input_array_types
 | 
			
		||||
                        .first()
 | 
			
		||||
                        .map(|&v| single_type.is_array_type(v))
 | 
			
		||||
| 
						 | 
				
			
			@ -532,7 +556,7 @@ impl ModuleState {
 | 
			
		|||
                        connect_write(output_stmts, wire_wmask, port_wmask);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                TypeEnum::EnumType(_) => Self::connect_split_mem_port_arrays(
 | 
			
		||||
                CanonicalType::Enum(_) => Self::connect_split_mem_port_arrays(
 | 
			
		||||
                    output_stmts,
 | 
			
		||||
                    &input_array_types,
 | 
			
		||||
                    0,
 | 
			
		||||
| 
						 | 
				
			
			@ -540,25 +564,19 @@ impl ModuleState {
 | 
			
		|||
                    wire_rdata,
 | 
			
		||||
                    wire_wdata,
 | 
			
		||||
                    wire_wmask,
 | 
			
		||||
                    port_rdata.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    port_wdata.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    port_wmask.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    port_rdata.map(Expr::from_canonical),
 | 
			
		||||
                    port_wdata.map(Expr::from_canonical),
 | 
			
		||||
                    port_wmask.map(Expr::from_canonical),
 | 
			
		||||
                    connect_read_enum,
 | 
			
		||||
                    connect_write_enum,
 | 
			
		||||
                    connect_write_enum,
 | 
			
		||||
                ),
 | 
			
		||||
                TypeEnum::ArrayType(array_type) => {
 | 
			
		||||
                CanonicalType::Array(array_type) => {
 | 
			
		||||
                    input_array_types.push(array_type);
 | 
			
		||||
                    input_element_type = array_type.element().type_enum();
 | 
			
		||||
                    input_element_type = array_type.element();
 | 
			
		||||
                    continue;
 | 
			
		||||
                }
 | 
			
		||||
                TypeEnum::UInt(_) | TypeEnum::SInt(_)
 | 
			
		||||
                CanonicalType::UInt(_) | CanonicalType::SInt(_) | CanonicalType::Bool(_)
 | 
			
		||||
                    if input_array_types
 | 
			
		||||
                        .first()
 | 
			
		||||
                        .map(|&v| single_type.is_array_type(v))
 | 
			
		||||
| 
						 | 
				
			
			@ -574,59 +592,56 @@ impl ModuleState {
 | 
			
		|||
                        connect_write(output_stmts, wire_wmask, port_wmask);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                TypeEnum::UInt(_) | TypeEnum::SInt(_) => Self::connect_split_mem_port_arrays(
 | 
			
		||||
                    output_stmts,
 | 
			
		||||
                    &input_array_types,
 | 
			
		||||
                    0,
 | 
			
		||||
                    single_type.array_len(),
 | 
			
		||||
                    wire_rdata,
 | 
			
		||||
                    wire_wdata,
 | 
			
		||||
                    wire_wmask,
 | 
			
		||||
                    port_rdata.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    port_wdata.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    port_wmask.map(|e: Expr<DynCanonicalValue>| {
 | 
			
		||||
                        e.with_type::<Array<[DynCanonicalValue]>>()
 | 
			
		||||
                    }),
 | 
			
		||||
                    connect_read,
 | 
			
		||||
                    connect_write,
 | 
			
		||||
                    connect_write,
 | 
			
		||||
                ),
 | 
			
		||||
                TypeEnum::Clock(_)
 | 
			
		||||
                | TypeEnum::AsyncReset(_)
 | 
			
		||||
                | TypeEnum::SyncReset(_)
 | 
			
		||||
                | TypeEnum::Reset(_) => unreachable!("memory element type is a storable type"),
 | 
			
		||||
                CanonicalType::UInt(_) | CanonicalType::SInt(_) | CanonicalType::Bool(_) => {
 | 
			
		||||
                    Self::connect_split_mem_port_arrays(
 | 
			
		||||
                        output_stmts,
 | 
			
		||||
                        &input_array_types,
 | 
			
		||||
                        0,
 | 
			
		||||
                        single_type.array_len(),
 | 
			
		||||
                        wire_rdata,
 | 
			
		||||
                        wire_wdata,
 | 
			
		||||
                        wire_wmask,
 | 
			
		||||
                        port_rdata.map(Expr::from_canonical),
 | 
			
		||||
                        port_wdata.map(Expr::from_canonical),
 | 
			
		||||
                        port_wmask.map(Expr::from_canonical),
 | 
			
		||||
                        connect_read,
 | 
			
		||||
                        connect_write,
 | 
			
		||||
                        connect_write,
 | 
			
		||||
                    )
 | 
			
		||||
                }
 | 
			
		||||
                CanonicalType::Clock(_)
 | 
			
		||||
                | CanonicalType::AsyncReset(_)
 | 
			
		||||
                | CanonicalType::SyncReset(_)
 | 
			
		||||
                | CanonicalType::Reset(_) => unreachable!("memory element type is a storable type"),
 | 
			
		||||
            }
 | 
			
		||||
            break;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    fn create_split_mem(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        input_mem: Mem<[DynCanonicalValue]>,
 | 
			
		||||
        input_mem: Mem,
 | 
			
		||||
        output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
        input_element_type: TypeEnum,
 | 
			
		||||
        input_element_type: CanonicalType,
 | 
			
		||||
        single_type: SingleType,
 | 
			
		||||
        mem_name_path: &str,
 | 
			
		||||
        split_state: &SplitState<'_>,
 | 
			
		||||
    ) -> Mem<[DynCanonicalValue]> {
 | 
			
		||||
    ) -> Mem {
 | 
			
		||||
        let mem_name = self.name_id_gen.gen(Intern::intern_owned(format!(
 | 
			
		||||
            "{}{mem_name_path}",
 | 
			
		||||
            input_mem.scoped_name().1 .0
 | 
			
		||||
        )));
 | 
			
		||||
        let mem_name = ScopedNameId(input_mem.scoped_name().0, mem_name);
 | 
			
		||||
        let output_element_type = match single_type {
 | 
			
		||||
            SingleType::UInt(ty) => ty.canonical_dyn(),
 | 
			
		||||
            SingleType::SInt(ty) => ty.canonical_dyn(),
 | 
			
		||||
            SingleType::UIntArray(ty) => ty.canonical_dyn(),
 | 
			
		||||
            SingleType::SIntArray(ty) => ty.canonical_dyn(),
 | 
			
		||||
            SingleType::UInt(ty) => ty.canonical(),
 | 
			
		||||
            SingleType::SInt(ty) => ty.canonical(),
 | 
			
		||||
            SingleType::Bool(ty) => ty.canonical(),
 | 
			
		||||
            SingleType::UIntArray(ty) => ty.canonical(),
 | 
			
		||||
            SingleType::SIntArray(ty) => ty.canonical(),
 | 
			
		||||
            SingleType::BoolArray(ty) => ty.canonical(),
 | 
			
		||||
        };
 | 
			
		||||
        let output_array_type =
 | 
			
		||||
            ArrayType::new_slice(output_element_type, input_mem.array_type().len());
 | 
			
		||||
        let output_array_type = Array::new_dyn(output_element_type, input_mem.array_type().len());
 | 
			
		||||
        let initial_value = split_state.initial_value.as_ref().map(|initial_value| {
 | 
			
		||||
            let mut bits = BitVec::with_capacity(output_array_type.bit_width());
 | 
			
		||||
            let mut bits = BitVec::with_capacity(output_array_type.type_properties().bit_width);
 | 
			
		||||
            for element in initial_value.iter() {
 | 
			
		||||
                bits.extend_from_bitslice(element);
 | 
			
		||||
            }
 | 
			
		||||
| 
						 | 
				
			
			@ -643,7 +658,6 @@ impl ModuleState {
 | 
			
		|||
                    port.addr_type(),
 | 
			
		||||
                    output_element_type,
 | 
			
		||||
                )
 | 
			
		||||
                .intern_sized()
 | 
			
		||||
            })
 | 
			
		||||
            .collect();
 | 
			
		||||
        let output_mem = Mem::new_unchecked(
 | 
			
		||||
| 
						 | 
				
			
			@ -676,15 +690,15 @@ impl ModuleState {
 | 
			
		|||
            let port_rdata = port
 | 
			
		||||
                .port_kind()
 | 
			
		||||
                .rdata_name()
 | 
			
		||||
                .map(|name| port_expr.field(name));
 | 
			
		||||
                .map(|name| Expr::field(port_expr, name));
 | 
			
		||||
            let port_wdata = port
 | 
			
		||||
                .port_kind()
 | 
			
		||||
                .wdata_name()
 | 
			
		||||
                .map(|name| port_expr.field(name));
 | 
			
		||||
                .map(|name| Expr::field(port_expr, name));
 | 
			
		||||
            let port_wmask = port
 | 
			
		||||
                .port_kind()
 | 
			
		||||
                .wmask_name()
 | 
			
		||||
                .map(|name| port_expr.field(name));
 | 
			
		||||
                .map(|name| Expr::field(port_expr, name));
 | 
			
		||||
            self.connect_split_mem_port(
 | 
			
		||||
                output_stmts,
 | 
			
		||||
                input_element_type,
 | 
			
		||||
| 
						 | 
				
			
			@ -702,11 +716,11 @@ impl ModuleState {
 | 
			
		|||
    }
 | 
			
		||||
    fn process_mem(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        input_mem: Mem<[DynCanonicalValue]>,
 | 
			
		||||
        output_mems: &mut Vec<Mem<[DynCanonicalValue]>>,
 | 
			
		||||
        input_mem: Mem,
 | 
			
		||||
        output_mems: &mut Vec<Mem>,
 | 
			
		||||
        output_stmts: &mut Vec<Stmt>,
 | 
			
		||||
    ) {
 | 
			
		||||
        let element_type = input_mem.array_type().element().type_enum();
 | 
			
		||||
        let element_type = input_mem.array_type().element();
 | 
			
		||||
        let mut split = MemSplit::new(element_type);
 | 
			
		||||
        let mem_state = match split {
 | 
			
		||||
            MemSplit::Single {
 | 
			
		||||
| 
						 | 
				
			
			@ -748,7 +762,7 @@ impl ModuleState {
 | 
			
		|||
                        port_ty,
 | 
			
		||||
                    );
 | 
			
		||||
                    let wire_expr = wire.to_expr();
 | 
			
		||||
                    let canonical_wire = wire.to_dyn_canonical_wire();
 | 
			
		||||
                    let canonical_wire = wire.canonical();
 | 
			
		||||
                    output_stmts.push(
 | 
			
		||||
                        StmtWire {
 | 
			
		||||
                            annotations: Default::default(),
 | 
			
		||||
| 
						 | 
				
			
			@ -756,21 +770,21 @@ impl ModuleState {
 | 
			
		|||
                        }
 | 
			
		||||
                        .into(),
 | 
			
		||||
                    );
 | 
			
		||||
                    replacement_ports.push(ExprEnum::Wire(canonical_wire.intern_sized()));
 | 
			
		||||
                    replacement_ports.push(ExprEnum::Wire(canonical_wire));
 | 
			
		||||
                    wire_port_rdata.push(
 | 
			
		||||
                        port.port_kind()
 | 
			
		||||
                            .rdata_name()
 | 
			
		||||
                            .map(|name| wire_expr.field(name)),
 | 
			
		||||
                            .map(|name| Expr::field(wire_expr, name)),
 | 
			
		||||
                    );
 | 
			
		||||
                    wire_port_wdata.push(
 | 
			
		||||
                        port.port_kind()
 | 
			
		||||
                            .wdata_name()
 | 
			
		||||
                            .map(|name| wire_expr.field(name)),
 | 
			
		||||
                            .map(|name| Expr::field(wire_expr, name)),
 | 
			
		||||
                    );
 | 
			
		||||
                    wire_port_wmask.push(
 | 
			
		||||
                        port.port_kind()
 | 
			
		||||
                            .wmask_name()
 | 
			
		||||
                            .map(|name| wire_expr.field(name)),
 | 
			
		||||
                            .map(|name| Expr::field(wire_expr, name)),
 | 
			
		||||
                    );
 | 
			
		||||
                }
 | 
			
		||||
                let mem_state = MemState {
 | 
			
		||||
| 
						 | 
				
			
			@ -815,8 +829,8 @@ impl ModuleState {
 | 
			
		|||
 | 
			
		||||
#[derive(Default)]
 | 
			
		||||
struct State {
 | 
			
		||||
    modules: HashMap<Interned<Module<DynBundle>>, ModuleState>,
 | 
			
		||||
    current_module: Option<Interned<Module<DynBundle>>>,
 | 
			
		||||
    modules: HashMap<Interned<Module<Bundle>>, ModuleState>,
 | 
			
		||||
    current_module: Option<Interned<Module<Bundle>>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl State {
 | 
			
		||||
| 
						 | 
				
			
			@ -828,11 +842,11 @@ impl State {
 | 
			
		|||
 | 
			
		||||
struct PushedState<'a> {
 | 
			
		||||
    state: &'a mut State,
 | 
			
		||||
    old_module: Option<Interned<Module<DynBundle>>>,
 | 
			
		||||
    old_module: Option<Interned<Module<Bundle>>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<'a> PushedState<'a> {
 | 
			
		||||
    fn push_module(state: &'a mut State, module: Interned<Module<DynBundle>>) -> Self {
 | 
			
		||||
    fn push_module(state: &'a mut State, module: Interned<Module<Bundle>>) -> Self {
 | 
			
		||||
        let old_module = state.current_module.replace(module);
 | 
			
		||||
        Self { state, old_module }
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -860,10 +874,7 @@ impl DerefMut for PushedState<'_> {
 | 
			
		|||
impl Folder for State {
 | 
			
		||||
    type Error = Infallible;
 | 
			
		||||
 | 
			
		||||
    fn fold_module<T: BundleValue>(&mut self, v: Module<T>) -> Result<Module<T>, Self::Error>
 | 
			
		||||
    where
 | 
			
		||||
        T::Type: BundleType<Value = T>,
 | 
			
		||||
    {
 | 
			
		||||
    fn fold_module<T: BundleType>(&mut self, v: Module<T>) -> Result<Module<T>, Self::Error> {
 | 
			
		||||
        let module: Interned<_> = v.canonical().intern_sized();
 | 
			
		||||
        if let Some(module_state) = self.modules.get(&module) {
 | 
			
		||||
            return Ok(Module::from_canonical(
 | 
			
		||||
| 
						 | 
				
			
			@ -886,10 +897,10 @@ impl Folder for State {
 | 
			
		|||
        Ok(Module::from_canonical(*module))
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn fold_mem<VA: ValueArrayOrSlice + ?Sized>(
 | 
			
		||||
    fn fold_mem<Element: Type, Len: Size>(
 | 
			
		||||
        &mut self,
 | 
			
		||||
        _v: Mem<VA>,
 | 
			
		||||
    ) -> Result<Mem<VA>, Self::Error> {
 | 
			
		||||
        _v: Mem<Element, Len>,
 | 
			
		||||
    ) -> Result<Mem<Element, Len>, Self::Error> {
 | 
			
		||||
        unreachable!()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -933,7 +944,7 @@ impl Folder for State {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn simplify_memories(module: Interned<Module<DynBundle>>) -> Interned<Module<DynBundle>> {
 | 
			
		||||
pub fn simplify_memories(module: Interned<Module<Bundle>>) -> Interned<Module<Bundle>> {
 | 
			
		||||
    module
 | 
			
		||||
        .fold(&mut State::default())
 | 
			
		||||
        .unwrap_or_else(|v| match v {})
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -3,18 +3,19 @@
 | 
			
		|||
#![allow(clippy::multiple_bound_locations)]
 | 
			
		||||
use crate::{
 | 
			
		||||
    annotations::{Annotation, CustomFirrtlAnnotation, TargetedAnnotation},
 | 
			
		||||
    array::{Array, ArrayType, ArrayTypeTrait, ValueArrayOrSlice},
 | 
			
		||||
    bundle::{BundleType, BundleValue, DynBundle, DynBundleType, FieldType},
 | 
			
		||||
    clock::{Clock, ClockType},
 | 
			
		||||
    enum_::{DynEnum, DynEnumType, EnumType, EnumValue, VariantType},
 | 
			
		||||
    array::ArrayType,
 | 
			
		||||
    bundle::{Bundle, BundleField, BundleType},
 | 
			
		||||
    clock::Clock,
 | 
			
		||||
    enum_::{Enum, EnumType, EnumVariant},
 | 
			
		||||
    expr::{
 | 
			
		||||
        ops, Expr, ExprEnum, Literal, Target, TargetBase, TargetChild, TargetPathArrayElement,
 | 
			
		||||
        TargetPathBundleField, TargetPathDynArrayElement, TargetPathElement, ToExpr,
 | 
			
		||||
    },
 | 
			
		||||
    int::{
 | 
			
		||||
        DynInt, DynIntType, DynSInt, DynSIntType, DynUInt, DynUIntType, StaticOrDynIntType, IntType,
 | 
			
		||||
        IntTypeTrait,
 | 
			
		||||
        ops,
 | 
			
		||||
        target::{
 | 
			
		||||
            Target, TargetBase, TargetChild, TargetPathArrayElement, TargetPathBundleField,
 | 
			
		||||
            TargetPathDynArrayElement, TargetPathElement,
 | 
			
		||||
        },
 | 
			
		||||
        Expr, ExprEnum,
 | 
			
		||||
    },
 | 
			
		||||
    int::{Bool, SIntType, SIntValue, Size, UIntType, UIntValue},
 | 
			
		||||
    intern::{Intern, Interned},
 | 
			
		||||
    memory::{Mem, MemPort, PortKind, PortName, PortType, ReadUnderWrite},
 | 
			
		||||
    module::{
 | 
			
		||||
| 
						 | 
				
			
			@ -24,10 +25,9 @@ use crate::{
 | 
			
		|||
        StmtMatch, StmtReg, StmtWire,
 | 
			
		||||
    },
 | 
			
		||||
    reg::Reg,
 | 
			
		||||
    reset::{AsyncReset, AsyncResetType, Reset, ResetType, SyncReset, SyncResetType},
 | 
			
		||||
    reset::{AsyncReset, Reset, SyncReset},
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{DynCanonicalType, DynCanonicalValue, DynType, Type, TypeEnum, Value, ValueEnum},
 | 
			
		||||
    util::{ConstBool, GenericConstBool},
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
    wire::Wire,
 | 
			
		||||
};
 | 
			
		||||
use num_bigint::{BigInt, BigUint};
 | 
			
		||||
| 
						 | 
				
			
			@ -473,7 +473,4 @@ impl<T: ?Sized + Visit<State>, State: ?Sized + Visitor> Visit<State> for &'_ mut
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type InternedDynType = Interned<dyn DynType>;
 | 
			
		||||
type InternedDynCanonicalType = Interned<dyn DynCanonicalType>;
 | 
			
		||||
 | 
			
		||||
include!(concat!(env!("OUT_DIR"), "/visit.rs"));
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										22
									
								
								crates/fayalite/src/prelude.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										22
									
								
								crates/fayalite/src/prelude.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,22 @@
 | 
			
		|||
pub use crate::{
 | 
			
		||||
    annotations::Annotation,
 | 
			
		||||
    array::{Array, ArrayType},
 | 
			
		||||
    cli::Cli,
 | 
			
		||||
    clock::{Clock, ClockDomain, ToClock},
 | 
			
		||||
    enum_::{HdlNone, HdlOption, HdlSome},
 | 
			
		||||
    expr::{CastBitsTo, CastTo, CastToBits, Expr, ReduceBits, ToExpr},
 | 
			
		||||
    hdl, hdl_module,
 | 
			
		||||
    int::{Bool, DynSize, IntCmp, KnownSize, SInt, SIntType, Size, UInt, UIntType},
 | 
			
		||||
    memory::{Mem, MemBuilder, ReadUnderWrite},
 | 
			
		||||
    module::{
 | 
			
		||||
        annotate, connect, connect_any, instance, memory, memory_array, memory_with_init,
 | 
			
		||||
        reg_builder, wire, Instance, Module, ModuleBuilder,
 | 
			
		||||
    },
 | 
			
		||||
    reg::Reg,
 | 
			
		||||
    reset::{AsyncReset, Reset, SyncReset, ToAsyncReset, ToReset, ToSyncReset},
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{AsMask, CanonicalType, Type},
 | 
			
		||||
    util::{ConstUsize, GenericConstUsize},
 | 
			
		||||
    wire::Wire,
 | 
			
		||||
    __,
 | 
			
		||||
};
 | 
			
		||||
| 
						 | 
				
			
			@ -2,21 +2,21 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    clock::ClockDomain,
 | 
			
		||||
    expr::{Expr, ExprTrait, Flow, ToExpr},
 | 
			
		||||
    expr::{Expr, Flow},
 | 
			
		||||
    intern::Interned,
 | 
			
		||||
    module::{NameId, ScopedNameId},
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{DynCanonicalType, DynType, Type},
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
};
 | 
			
		||||
use std::fmt;
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Eq, PartialEq, Hash)]
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash)]
 | 
			
		||||
pub struct Reg<T: Type> {
 | 
			
		||||
    name: ScopedNameId,
 | 
			
		||||
    source_location: SourceLocation,
 | 
			
		||||
    ty: T,
 | 
			
		||||
    clock_domain: Expr<ClockDomain>,
 | 
			
		||||
    init: Option<Expr<T::Value>>,
 | 
			
		||||
    init: Option<Expr<T>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type + fmt::Debug> fmt::Debug for Reg<T> {
 | 
			
		||||
| 
						 | 
				
			
			@ -37,20 +37,8 @@ impl<T: Type + fmt::Debug> fmt::Debug for Reg<T> {
 | 
			
		|||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> ToExpr for Reg<T> {
 | 
			
		||||
    type Type = T;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        self.ty.clone()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        Expr::new_unchecked(self.expr_enum())
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> Reg<T> {
 | 
			
		||||
    pub fn canonical(&self) -> Reg<T::CanonicalType> {
 | 
			
		||||
    pub fn canonical(&self) -> Reg<CanonicalType> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
| 
						 | 
				
			
			@ -66,49 +54,20 @@ impl<T: Type> Reg<T> {
 | 
			
		|||
            init: init.map(Expr::canonical),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn to_dyn_reg(&self) -> Reg<Interned<dyn DynType>> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ref ty,
 | 
			
		||||
            clock_domain,
 | 
			
		||||
            init,
 | 
			
		||||
        } = *self;
 | 
			
		||||
        Reg {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ty: ty.to_dyn(),
 | 
			
		||||
            clock_domain,
 | 
			
		||||
            init: init.map(Expr::to_dyn),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn to_dyn_canonical_reg(&self) -> Reg<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ref ty,
 | 
			
		||||
            clock_domain,
 | 
			
		||||
            init,
 | 
			
		||||
        } = *self;
 | 
			
		||||
        Reg {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ty: ty.canonical_dyn(),
 | 
			
		||||
            clock_domain,
 | 
			
		||||
            init: init.map(Expr::to_canonical_dyn),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    #[track_caller]
 | 
			
		||||
    pub fn new_unchecked(
 | 
			
		||||
        scoped_name: ScopedNameId,
 | 
			
		||||
        source_location: SourceLocation,
 | 
			
		||||
        ty: T,
 | 
			
		||||
        clock_domain: Expr<ClockDomain>,
 | 
			
		||||
        init: Option<Expr<T::Value>>,
 | 
			
		||||
        init: Option<Expr<T>>,
 | 
			
		||||
    ) -> Self {
 | 
			
		||||
        assert!(ty.is_storable(), "register type must be a storable type");
 | 
			
		||||
        assert!(
 | 
			
		||||
            ty.canonical().is_storable(),
 | 
			
		||||
            "register type must be a storable type"
 | 
			
		||||
        );
 | 
			
		||||
        if let Some(init) = init {
 | 
			
		||||
            assert_eq!(ty, init.ty(), "register's type must match init type");
 | 
			
		||||
            assert_eq!(ty, Expr::ty(init), "register's type must match init type");
 | 
			
		||||
        }
 | 
			
		||||
        Self {
 | 
			
		||||
            name: scoped_name,
 | 
			
		||||
| 
						 | 
				
			
			@ -118,6 +77,9 @@ impl<T: Type> Reg<T> {
 | 
			
		|||
            init,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn ty(&self) -> T {
 | 
			
		||||
        self.ty
 | 
			
		||||
    }
 | 
			
		||||
    pub fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        self.source_location
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -139,7 +101,7 @@ impl<T: Type> Reg<T> {
 | 
			
		|||
    pub fn clock_domain(&self) -> Expr<ClockDomain> {
 | 
			
		||||
        self.clock_domain
 | 
			
		||||
    }
 | 
			
		||||
    pub fn init(&self) -> Option<Expr<T::Value>> {
 | 
			
		||||
    pub fn init(&self) -> Option<Expr<T>> {
 | 
			
		||||
        self.init
 | 
			
		||||
    }
 | 
			
		||||
    pub fn flow(&self) -> Flow {
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -2,358 +2,119 @@
 | 
			
		|||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    expr::{Expr, ToExpr},
 | 
			
		||||
    int::{UInt, UIntType},
 | 
			
		||||
    intern::Interned,
 | 
			
		||||
    int::Bool,
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{
 | 
			
		||||
        impl_match_values_as_self, CanonicalType, CanonicalTypeKind, CanonicalValue, Connect,
 | 
			
		||||
        DynCanonicalType, StaticType, Type, TypeEnum, Value, ValueEnum,
 | 
			
		||||
    },
 | 
			
		||||
    util::interned_bit,
 | 
			
		||||
    ty::{impl_match_variant_as_self, CanonicalType, StaticType, Type, TypeProperties},
 | 
			
		||||
};
 | 
			
		||||
use bitvec::slice::BitSlice;
 | 
			
		||||
 | 
			
		||||
pub trait ResetTypeTrait: CanonicalType + StaticType<MaskType = UIntType<1>> {}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
 | 
			
		||||
pub struct AsyncResetType;
 | 
			
		||||
 | 
			
		||||
impl AsyncResetType {
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
mod sealed {
 | 
			
		||||
    pub trait ResetTypeSealed {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Type for AsyncResetType {
 | 
			
		||||
    type Value = AsyncReset;
 | 
			
		||||
    type CanonicalType = AsyncResetType;
 | 
			
		||||
    type CanonicalValue = AsyncReset;
 | 
			
		||||
    type MaskType = UIntType<1>;
 | 
			
		||||
    type MaskValue = UInt<1>;
 | 
			
		||||
pub trait ResetType: StaticType<MaskType = Bool> + sealed::ResetTypeSealed {}
 | 
			
		||||
 | 
			
		||||
    impl_match_values_as_self!();
 | 
			
		||||
macro_rules! reset_type {
 | 
			
		||||
    ($name:ident, $Trait:ident::$trait_fn:ident, $is_castable_from_bits:literal) => {
 | 
			
		||||
        #[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
 | 
			
		||||
        pub struct $name;
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        UIntType::new()
 | 
			
		||||
    }
 | 
			
		||||
        impl Type for $name {
 | 
			
		||||
            type BaseType = $name;
 | 
			
		||||
            type MaskType = Bool;
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
            impl_match_variant_as_self!();
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
            fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
                Bool
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::AsyncReset(*self)
 | 
			
		||||
    }
 | 
			
		||||
            fn canonical(&self) -> CanonicalType {
 | 
			
		||||
                CanonicalType::$name(*self)
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        t
 | 
			
		||||
    }
 | 
			
		||||
            fn source_location() -> SourceLocation {
 | 
			
		||||
                SourceLocation::builtin()
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Connect<Self> for AsyncResetType {}
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for AsyncResetType {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::AsyncReset;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl StaticType for AsyncResetType {
 | 
			
		||||
    fn static_type() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ResetTypeTrait for AsyncResetType {}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug)]
 | 
			
		||||
pub struct AsyncReset(pub bool);
 | 
			
		||||
 | 
			
		||||
impl ToExpr for AsyncReset {
 | 
			
		||||
    type Type = AsyncResetType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        AsyncResetType
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<Self> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Value for AsyncReset {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for AsyncReset {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::AsyncReset(*this)
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
 | 
			
		||||
pub struct SyncResetType;
 | 
			
		||||
 | 
			
		||||
impl SyncResetType {
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Type for SyncResetType {
 | 
			
		||||
    type CanonicalType = SyncResetType;
 | 
			
		||||
    type Value = SyncReset;
 | 
			
		||||
    type CanonicalValue = SyncReset;
 | 
			
		||||
    type MaskType = UIntType<1>;
 | 
			
		||||
    type MaskValue = UInt<1>;
 | 
			
		||||
 | 
			
		||||
    impl_match_values_as_self!();
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        UIntType::new()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::SyncReset(*self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        t
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Connect<Self> for SyncResetType {}
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for SyncResetType {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::SyncReset;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl StaticType for SyncResetType {
 | 
			
		||||
    fn static_type() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ResetTypeTrait for SyncResetType {}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
 | 
			
		||||
pub struct SyncReset(pub bool);
 | 
			
		||||
 | 
			
		||||
impl ToExpr for SyncReset {
 | 
			
		||||
    type Type = SyncResetType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        SyncResetType
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<Self> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Value for SyncReset {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for SyncReset {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::SyncReset(*this)
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        interned_bit(this.0)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, Default)]
 | 
			
		||||
pub struct ResetType;
 | 
			
		||||
 | 
			
		||||
impl ResetType {
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Type for ResetType {
 | 
			
		||||
    type Value = Reset;
 | 
			
		||||
    type CanonicalType = ResetType;
 | 
			
		||||
    type CanonicalValue = Reset;
 | 
			
		||||
    type MaskType = UIntType<1>;
 | 
			
		||||
    type MaskValue = UInt<1>;
 | 
			
		||||
 | 
			
		||||
    impl_match_values_as_self!();
 | 
			
		||||
 | 
			
		||||
    fn mask_type(&self) -> Self::MaskType {
 | 
			
		||||
        UIntType::new()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn canonical(&self) -> Self::CanonicalType {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn source_location(&self) -> SourceLocation {
 | 
			
		||||
        SourceLocation::builtin()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn type_enum(&self) -> TypeEnum {
 | 
			
		||||
        TypeEnum::Reset(*self)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn from_canonical_type(t: Self::CanonicalType) -> Self {
 | 
			
		||||
        t
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn as_dyn_canonical_type_impl(this: &Self) -> Option<&dyn DynCanonicalType> {
 | 
			
		||||
        Some(this)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Connect<Self> for ResetType {}
 | 
			
		||||
 | 
			
		||||
impl CanonicalType for ResetType {
 | 
			
		||||
    const CANONICAL_TYPE_KIND: CanonicalTypeKind = CanonicalTypeKind::Reset;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl StaticType for ResetType {
 | 
			
		||||
    fn static_type() -> Self {
 | 
			
		||||
        Self
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ResetTypeTrait for ResetType {}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash)]
 | 
			
		||||
pub enum Reset {}
 | 
			
		||||
 | 
			
		||||
impl ToExpr for Reset {
 | 
			
		||||
    type Type = ResetType;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        match *self {}
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<Self> {
 | 
			
		||||
        Expr::from_value(self)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Value for Reset {
 | 
			
		||||
    fn to_canonical(&self) -> <Self::Type as Type>::CanonicalValue {
 | 
			
		||||
        *self
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        match *this {}
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl CanonicalValue for Reset {
 | 
			
		||||
    fn value_enum_impl(this: &Self) -> ValueEnum {
 | 
			
		||||
        ValueEnum::Reset(*this)
 | 
			
		||||
    }
 | 
			
		||||
    fn to_bits_impl(this: &Self) -> Interned<BitSlice> {
 | 
			
		||||
        match *this {}
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
macro_rules! make_to_reset {
 | 
			
		||||
    (
 | 
			
		||||
        $(#[from_value($from_value_ty:ty)])*
 | 
			
		||||
        $vis:vis trait $Trait:ident {
 | 
			
		||||
            fn $fn:ident(&self) -> Expr<$T:ty>;
 | 
			
		||||
            fn from_canonical(canonical_type: CanonicalType) -> Self {
 | 
			
		||||
                let CanonicalType::$name(retval) = canonical_type else {
 | 
			
		||||
                    panic!("expected {}", stringify!($name));
 | 
			
		||||
                };
 | 
			
		||||
                retval
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    ) => {
 | 
			
		||||
        $vis trait $Trait {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T>;
 | 
			
		||||
 | 
			
		||||
        impl $name {
 | 
			
		||||
            pub fn type_properties(self) -> TypeProperties {
 | 
			
		||||
                Self::TYPE_PROPERTIES
 | 
			
		||||
            }
 | 
			
		||||
            pub fn can_connect(self, _rhs: Self) -> bool {
 | 
			
		||||
                true
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl StaticType for $name {
 | 
			
		||||
            const TYPE: Self = Self;
 | 
			
		||||
            const MASK_TYPE: Self::MaskType = Bool;
 | 
			
		||||
            const TYPE_PROPERTIES: TypeProperties = TypeProperties {
 | 
			
		||||
                is_passive: true,
 | 
			
		||||
                is_storable: false,
 | 
			
		||||
                is_castable_from_bits: $is_castable_from_bits,
 | 
			
		||||
                bit_width: 1,
 | 
			
		||||
            };
 | 
			
		||||
            const MASK_TYPE_PROPERTIES: TypeProperties = Bool::TYPE_PROPERTIES;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl sealed::ResetTypeSealed for $name {}
 | 
			
		||||
 | 
			
		||||
        impl ResetType for $name {}
 | 
			
		||||
 | 
			
		||||
        pub trait $Trait {
 | 
			
		||||
            fn $trait_fn(&self) -> Expr<$name>;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + $Trait> $Trait for &'_ T {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
                (**self).$fn()
 | 
			
		||||
            fn $trait_fn(&self) -> Expr<$name> {
 | 
			
		||||
                (**self).$trait_fn()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + $Trait> $Trait for &'_ mut T {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
                (**self).$fn()
 | 
			
		||||
            fn $trait_fn(&self) -> Expr<$name> {
 | 
			
		||||
                (**self).$trait_fn()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl<T: ?Sized + $Trait> $Trait for Box<T> {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
                (**self).$fn()
 | 
			
		||||
            fn $trait_fn(&self) -> Expr<$name> {
 | 
			
		||||
                (**self).$trait_fn()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl $Trait for Expr<$T> {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
        impl $Trait for Expr<$name> {
 | 
			
		||||
            fn $trait_fn(&self) -> Expr<$name> {
 | 
			
		||||
                *self
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        impl $Trait for $T {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
                self.to_expr()
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        $(impl $Trait for $from_value_ty {
 | 
			
		||||
            fn $fn(&self) -> Expr<$T> {
 | 
			
		||||
                self.to_expr().$fn()
 | 
			
		||||
            }
 | 
			
		||||
        })*
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
make_to_reset! {
 | 
			
		||||
    #[from_value(SyncReset)]
 | 
			
		||||
    #[from_value(AsyncReset)]
 | 
			
		||||
    pub trait ToReset {
 | 
			
		||||
        fn to_reset(&self) -> Expr<Reset>;
 | 
			
		||||
reset_type!(AsyncReset, ToAsyncReset::to_async_reset, true);
 | 
			
		||||
reset_type!(SyncReset, ToSyncReset::to_sync_reset, true);
 | 
			
		||||
reset_type!(
 | 
			
		||||
    Reset,
 | 
			
		||||
    ToReset::to_reset,
 | 
			
		||||
    false // Reset is not castable from bits because we don't know if it's async or sync
 | 
			
		||||
);
 | 
			
		||||
 | 
			
		||||
impl ToSyncReset for bool {
 | 
			
		||||
    fn to_sync_reset(&self) -> Expr<SyncReset> {
 | 
			
		||||
        self.to_expr().to_sync_reset()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
make_to_reset! {
 | 
			
		||||
    #[from_value(bool)]
 | 
			
		||||
    #[from_value(UInt<1>)]
 | 
			
		||||
    pub trait ToAsyncReset {
 | 
			
		||||
        fn to_async_reset(&self) -> Expr<AsyncReset>;
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
make_to_reset! {
 | 
			
		||||
    #[from_value(bool)]
 | 
			
		||||
    #[from_value(UInt<1>)]
 | 
			
		||||
    pub trait ToSyncReset {
 | 
			
		||||
        fn to_sync_reset(&self) -> Expr<SyncReset>;
 | 
			
		||||
impl ToAsyncReset for bool {
 | 
			
		||||
    fn to_async_reset(&self) -> Expr<AsyncReset> {
 | 
			
		||||
        self.to_expr().to_async_reset()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -3,10 +3,14 @@
 | 
			
		|||
 | 
			
		||||
mod const_bool;
 | 
			
		||||
mod const_cmp;
 | 
			
		||||
mod const_usize;
 | 
			
		||||
mod misc;
 | 
			
		||||
mod scoped_ref;
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use const_bool::{ConstBool, ConstBoolDispatch, ConstBoolDispatchTag, GenericConstBool};
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use const_usize::{ConstUsize, GenericConstUsize};
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use const_cmp::{
 | 
			
		||||
| 
						 | 
				
			
			@ -14,6 +18,9 @@ pub use const_cmp::{
 | 
			
		|||
    const_usize_cmp,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use scoped_ref::ScopedRef;
 | 
			
		||||
 | 
			
		||||
#[doc(inline)]
 | 
			
		||||
pub use misc::{
 | 
			
		||||
    interned_bit, iter_eq_by, BitSliceWriteWithBase, DebugAsDisplay, DebugAsRawString, MakeMutSlice,
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										29
									
								
								crates/fayalite/src/util/const_usize.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								crates/fayalite/src/util/const_usize.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,29 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use std::{fmt::Debug, hash::Hash};
 | 
			
		||||
 | 
			
		||||
mod sealed {
 | 
			
		||||
    pub trait Sealed {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// the only implementation is `ConstUsize<Self::VALUE>`
 | 
			
		||||
pub trait GenericConstUsize:
 | 
			
		||||
    sealed::Sealed + Copy + Ord + Hash + Default + Debug + 'static + Send + Sync
 | 
			
		||||
{
 | 
			
		||||
    const VALUE: usize;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Default)]
 | 
			
		||||
pub struct ConstUsize<const VALUE: usize>;
 | 
			
		||||
 | 
			
		||||
impl<const VALUE: usize> Debug for ConstUsize<VALUE> {
 | 
			
		||||
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 | 
			
		||||
        f.debug_tuple("ConstUsize").field(&Self::VALUE).finish()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<const VALUE: usize> sealed::Sealed for ConstUsize<VALUE> {}
 | 
			
		||||
 | 
			
		||||
impl<const VALUE: usize> GenericConstUsize for ConstUsize<VALUE> {
 | 
			
		||||
    const VALUE: usize = VALUE;
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										106
									
								
								crates/fayalite/src/util/scoped_ref.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										106
									
								
								crates/fayalite/src/util/scoped_ref.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,106 @@
 | 
			
		|||
mod safety_boundary {
 | 
			
		||||
    use std::{cell::Cell, ptr::NonNull};
 | 
			
		||||
 | 
			
		||||
    pub(super) struct Impl<T: ?Sized>(Cell<Option<NonNull<T>>>);
 | 
			
		||||
 | 
			
		||||
    impl<T: ?Sized> Impl<T> {
 | 
			
		||||
        #[inline]
 | 
			
		||||
        pub(super) const fn new() -> Self {
 | 
			
		||||
            Self(Cell::new(None))
 | 
			
		||||
        }
 | 
			
		||||
        #[inline]
 | 
			
		||||
        pub(super) const fn opt_static_ref(v: Option<&'static T>) -> Self {
 | 
			
		||||
            Self(Cell::new(if let Some(v) = v {
 | 
			
		||||
                // SAFETY: v is a valid reference for lifetime 'static
 | 
			
		||||
                unsafe { Some(NonNull::new_unchecked(v as *const T as *mut T)) }
 | 
			
		||||
            } else {
 | 
			
		||||
                None
 | 
			
		||||
            }))
 | 
			
		||||
        }
 | 
			
		||||
        /// set `self` to `value` for the duration of the `f()` call.
 | 
			
		||||
        #[inline]
 | 
			
		||||
        pub(super) fn set_opt<F: FnOnce() -> R, R>(&self, value: Option<&T>, f: F) -> R {
 | 
			
		||||
            struct ResetOnDrop<'a, T: ?Sized> {
 | 
			
		||||
                this: &'a Impl<T>,
 | 
			
		||||
                old: Option<NonNull<T>>,
 | 
			
		||||
            }
 | 
			
		||||
            impl<T: ?Sized> Drop for ResetOnDrop<'_, T> {
 | 
			
		||||
                fn drop(&mut self) {
 | 
			
		||||
                    self.this.0.set(self.old);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            // reset to old value before exiting this function ensuring `self`
 | 
			
		||||
            // is not set to `value` when its lifetime is expired
 | 
			
		||||
            let _reset_on_drop = ResetOnDrop {
 | 
			
		||||
                this: self,
 | 
			
		||||
                old: self.0.replace(value.map(NonNull::from)),
 | 
			
		||||
            };
 | 
			
		||||
            f()
 | 
			
		||||
        }
 | 
			
		||||
        #[inline]
 | 
			
		||||
        pub(super) fn get_ptr(&self) -> Option<NonNull<T>> {
 | 
			
		||||
            self.0.get()
 | 
			
		||||
        }
 | 
			
		||||
        /// get the reference in `self` for the duration of the `f(...)` call.
 | 
			
		||||
        #[inline]
 | 
			
		||||
        pub(super) fn with_opt<F: for<'a> FnOnce(Option<&'a T>) -> R, R>(&self, f: F) -> R {
 | 
			
		||||
            // SAFETY:
 | 
			
		||||
            // `self.0` is only `Some` when inside some `set_opt` call or when set
 | 
			
		||||
            // to some `&'static T`, which ensures that the pointer is live and valid.
 | 
			
		||||
            //
 | 
			
		||||
            // the reference we give away has its lifetime scoped to this
 | 
			
		||||
            // function call which ensures that it won't escape
 | 
			
		||||
            unsafe { f(self.0.get().map(|v| &*v.as_ptr())) }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// holds a `Cell<Option<&'scoped T>>` where `'scoped` is erased. This is useful for holding references in TLS.
 | 
			
		||||
pub struct ScopedRef<T: ?Sized>(safety_boundary::Impl<T>);
 | 
			
		||||
 | 
			
		||||
impl<T: ?Sized> ScopedRef<T> {
 | 
			
		||||
    /// create a new empty [`ScopedRef`]
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub const fn new() -> Self {
 | 
			
		||||
        Self(safety_boundary::Impl::new())
 | 
			
		||||
    }
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub const fn opt_static_ref(v: Option<&'static T>) -> Self {
 | 
			
		||||
        Self(safety_boundary::Impl::opt_static_ref(v))
 | 
			
		||||
    }
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub const fn static_ref(v: &'static T) -> Self {
 | 
			
		||||
        Self::opt_static_ref(Some(v))
 | 
			
		||||
    }
 | 
			
		||||
    /// set `self` to `value` for the duration of the `f()` call.
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn set_opt<F: FnOnce() -> R, R>(&self, value: Option<&T>, f: F) -> R {
 | 
			
		||||
        self.0.set_opt(value, f)
 | 
			
		||||
    }
 | 
			
		||||
    /// set `self` to `value` for the duration of the `f()` call.
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn set<F: FnOnce() -> R, R>(&self, value: &T, f: F) -> R {
 | 
			
		||||
        self.0.set_opt(Some(value), f)
 | 
			
		||||
    }
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn is_some(&self) -> bool {
 | 
			
		||||
        self.0.get_ptr().is_some()
 | 
			
		||||
    }
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn is_none(&self) -> bool {
 | 
			
		||||
        self.0.get_ptr().is_none()
 | 
			
		||||
    }
 | 
			
		||||
    /// get the reference in `self` for the duration of the `f(...)` call. panics if no reference is set.
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn with<F: FnOnce(&T) -> R, R>(&self, f: F) -> R {
 | 
			
		||||
        self.0.with_opt(
 | 
			
		||||
            #[inline]
 | 
			
		||||
            |v| f(v.expect("called ScopedRef::with on an empty ScopedRef")),
 | 
			
		||||
        )
 | 
			
		||||
    }
 | 
			
		||||
    /// get the reference in `self` for the duration of the `f(...)` call.
 | 
			
		||||
    #[inline]
 | 
			
		||||
    pub fn with_opt<F: FnOnce(Option<&T>) -> R, R>(&self, f: F) -> R {
 | 
			
		||||
        self.0.with_opt(f)
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -1,15 +1,15 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use crate::{
 | 
			
		||||
    expr::{Expr, ExprTrait, Flow, ToExpr},
 | 
			
		||||
    expr::Flow,
 | 
			
		||||
    intern::Interned,
 | 
			
		||||
    module::{NameId, ScopedNameId},
 | 
			
		||||
    source_location::SourceLocation,
 | 
			
		||||
    ty::{DynCanonicalType, DynType, Type},
 | 
			
		||||
    ty::{CanonicalType, Type},
 | 
			
		||||
};
 | 
			
		||||
use std::fmt;
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Eq, PartialEq, Hash)]
 | 
			
		||||
#[derive(Copy, Clone, Eq, PartialEq, Hash)]
 | 
			
		||||
pub struct Wire<T: Type> {
 | 
			
		||||
    name: ScopedNameId,
 | 
			
		||||
    source_location: SourceLocation,
 | 
			
		||||
| 
						 | 
				
			
			@ -18,27 +18,14 @@ pub struct Wire<T: Type> {
 | 
			
		|||
 | 
			
		||||
impl<T: Type + fmt::Debug> fmt::Debug for Wire<T> {
 | 
			
		||||
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
 | 
			
		||||
        f.debug_struct("Wire")
 | 
			
		||||
            .field("name", &self.name)
 | 
			
		||||
            .field("ty", &self.ty)
 | 
			
		||||
            .finish_non_exhaustive()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> ToExpr for Wire<T> {
 | 
			
		||||
    type Type = T;
 | 
			
		||||
 | 
			
		||||
    fn ty(&self) -> Self::Type {
 | 
			
		||||
        self.ty.clone()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn to_expr(&self) -> Expr<<Self::Type as Type>::Value> {
 | 
			
		||||
        Expr::new_unchecked(self.expr_enum())
 | 
			
		||||
        write!(f, "Wire({:?}: ", self.name)?;
 | 
			
		||||
        self.ty.fmt(f)?;
 | 
			
		||||
        f.write_str(")")
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T: Type> Wire<T> {
 | 
			
		||||
    pub fn canonical(&self) -> Wire<T::CanonicalType> {
 | 
			
		||||
    pub fn canonical(&self) -> Wire<CanonicalType> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
| 
						 | 
				
			
			@ -50,29 +37,8 @@ impl<T: Type> Wire<T> {
 | 
			
		|||
            ty: ty.canonical(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn to_dyn_wire(&self) -> Wire<Interned<dyn DynType>> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ref ty,
 | 
			
		||||
        } = *self;
 | 
			
		||||
        Wire {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ty: ty.to_dyn(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    pub fn to_dyn_canonical_wire(&self) -> Wire<Interned<dyn DynCanonicalType>> {
 | 
			
		||||
        let Self {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ref ty,
 | 
			
		||||
        } = *self;
 | 
			
		||||
        Wire {
 | 
			
		||||
            name,
 | 
			
		||||
            source_location,
 | 
			
		||||
            ty: ty.canonical_dyn(),
 | 
			
		||||
        }
 | 
			
		||||
    pub fn ty(&self) -> T {
 | 
			
		||||
        self.ty
 | 
			
		||||
    }
 | 
			
		||||
    pub fn new_unchecked(
 | 
			
		||||
        scoped_name: ScopedNameId,
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										26
									
								
								crates/fayalite/tests/hdl_types.rs
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								crates/fayalite/tests/hdl_types.rs
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,26 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use fayalite::{
 | 
			
		||||
    array::ArrayType,
 | 
			
		||||
    hdl,
 | 
			
		||||
    int::{IntType, Size, UInt},
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[hdl(outline_generated)]
 | 
			
		||||
pub struct S<T: IntType, Len: Size> {
 | 
			
		||||
    pub a: T,
 | 
			
		||||
    b: UInt<3>,
 | 
			
		||||
    pub(crate) c: ArrayType<UInt<1>, Len>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[hdl(outline_generated)]
 | 
			
		||||
pub enum E<T> {
 | 
			
		||||
    A,
 | 
			
		||||
    B(UInt<3>),
 | 
			
		||||
    C(T),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[hdl(outline_generated)]
 | 
			
		||||
pub struct S2<T = ()> {
 | 
			
		||||
    pub v: E<T>,
 | 
			
		||||
}
 | 
			
		||||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							| 
						 | 
				
			
			@ -1,8 +1,8 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use fayalite::ty::Value;
 | 
			
		||||
use fayalite::hdl;
 | 
			
		||||
 | 
			
		||||
#[derive(Value)]
 | 
			
		||||
#[hdl]
 | 
			
		||||
union U {
 | 
			
		||||
    a: (),
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										7
									
								
								crates/fayalite/tests/ui/hdl_types.stderr
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								crates/fayalite/tests/ui/hdl_types.stderr
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,7 @@
 | 
			
		|||
error: top-level #[hdl] can only be used on structs or enums
 | 
			
		||||
 --> tests/ui/hdl_types.rs:5:1
 | 
			
		||||
  |
 | 
			
		||||
5 | #[hdl]
 | 
			
		||||
  | ^^^^^^
 | 
			
		||||
  |
 | 
			
		||||
  = note: this error originates in the attribute macro `hdl` (in Nightly builds, run with -Z macro-backtrace for more info)
 | 
			
		||||
| 
						 | 
				
			
			@ -1,7 +0,0 @@
 | 
			
		|||
error: derive(Value) can only be used on structs or enums
 | 
			
		||||
 --> tests/ui/value_derive.rs:5:10
 | 
			
		||||
  |
 | 
			
		||||
5 | #[derive(Value)]
 | 
			
		||||
  |          ^^^^^
 | 
			
		||||
  |
 | 
			
		||||
  = note: this error originates in the derive macro `Value` (in Nightly builds, run with -Z macro-backtrace for more info)
 | 
			
		||||
| 
						 | 
				
			
			@ -1,32 +0,0 @@
 | 
			
		|||
// SPDX-License-Identifier: LGPL-3.0-or-later
 | 
			
		||||
// See Notices.txt for copyright information
 | 
			
		||||
use fayalite::{
 | 
			
		||||
    int::UInt,
 | 
			
		||||
    ty::{StaticValue, Value},
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, Hash, PartialEq, Eq, Debug)]
 | 
			
		||||
#[hdl(outline_generated)]
 | 
			
		||||
pub struct S<T> {
 | 
			
		||||
    pub a: T,
 | 
			
		||||
    b: UInt<3>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, Hash, PartialEq, Eq, Debug)]
 | 
			
		||||
#[hdl(outline_generated)]
 | 
			
		||||
pub enum E<T> {
 | 
			
		||||
    A,
 | 
			
		||||
    B {},
 | 
			
		||||
    C(),
 | 
			
		||||
    D(UInt<3>),
 | 
			
		||||
    E { a: UInt<3> },
 | 
			
		||||
    F(UInt<3>, UInt<3>),
 | 
			
		||||
    G(T),
 | 
			
		||||
    H(T, UInt<1>),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Value, Clone, Hash, PartialEq, Eq, Debug)]
 | 
			
		||||
#[hdl(outline_generated, static, where(T: StaticValue))]
 | 
			
		||||
pub struct S2<T> {
 | 
			
		||||
    pub v: E<T>,
 | 
			
		||||
}
 | 
			
		||||
										
											
												File diff suppressed because it is too large
												Load diff
											
										
									
								
							
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue