use proc_macro2::TokenStream as TokenStream2; use quote::quote; use rtfm_syntax::{ast::App, Context}; use crate::{ analyze::Analysis, check::Extra, codegen::{locals, module, resources_struct, util}, }; pub fn codegen( app: &App, analysis: &Analysis, extra: &Extra, ) -> ( // const_app_software_tasks -- free queues, buffers and `${task}Resources` constructors Vec, // root_software_tasks -- items that must be placed in the root of the crate: // - `${task}Locals` structs // - `${task}Resources` structs // - `${task}` modules Vec, // user_software_tasks -- the `#[task]` functions written by the user Vec, ) { let mut const_app = vec![]; let mut root = vec![]; let mut user_tasks = vec![]; for (name, task) in &app.software_tasks { let receiver = task.args.core; let inputs = &task.inputs; let (_, _, _, input_ty) = util::regroup_inputs(inputs); let cap = task.args.capacity; let cap_lit = util::capacity_literal(cap); let cap_ty = util::capacity_typenum(cap, true); // create free queues and inputs / instants buffers if let Some(free_queues) = analysis.free_queues.get(name) { for (&sender, &ceiling) in free_queues { let cfg_sender = util::cfg_core(sender, app.args.cores); let fq = util::fq_ident(name, sender); let (loc, fq_ty, fq_expr) = if receiver == sender { ( cfg_sender.clone(), quote!(rtfm::export::SCFQ<#cap_ty>), quote!(rtfm::export::Queue(unsafe { rtfm::export::iQueue::u8_sc() })), ) } else { let shared = if cfg!(feature = "heterogeneous") { Some(quote!(#[rtfm::export::shared])) } else { None }; ( shared, quote!(rtfm::export::MCFQ<#cap_ty>), quote!(rtfm::export::Queue(rtfm::export::iQueue::u8())), ) }; let loc = &loc; const_app.push(quote!( /// Queue version of a free-list that keeps track of empty slots in /// the following buffers #loc static mut #fq: #fq_ty = #fq_expr; )); // Generate a resource proxy if needed if let Some(ceiling) = ceiling { const_app.push(quote!( #cfg_sender struct #fq<'a> { priority: &'a rtfm::export::Priority, } )); const_app.push(util::impl_mutex( extra, &[], cfg_sender.as_ref(), false, &fq, fq_ty, ceiling, quote!(&mut #fq), )); } let ref elems = (0..cap) .map(|_| quote!(core::mem::MaybeUninit::uninit())) .collect::>(); if app.uses_schedule(receiver) { let m = extra.monotonic(); let instants = util::instants_ident(name, sender); const_app.push(quote!( #loc /// Buffer that holds the instants associated to the inputs of a task static mut #instants: [core::mem::MaybeUninit<<#m as rtfm::Monotonic>::Instant>; #cap_lit] = [#(#elems,)*]; )); } let inputs = util::inputs_ident(name, sender); const_app.push(quote!( #loc /// Buffer that holds the inputs of a task static mut #inputs: [core::mem::MaybeUninit<#input_ty>; #cap_lit] = [#(#elems,)*]; )); } } // `${task}Resources` let mut needs_lt = false; if !task.args.resources.is_empty() { let (item, constructor) = resources_struct::codegen( Context::SoftwareTask(name), task.args.priority, &mut needs_lt, app, analysis, ); root.push(item); const_app.push(constructor); } // `${task}Locals` let mut locals_pat = None; if !task.locals.is_empty() { let (struct_, pat) = locals::codegen(Context::SoftwareTask(name), &task.locals, app); locals_pat = Some(pat); root.push(struct_); } let cfg_receiver = util::cfg_core(receiver, app.args.cores); let context = &task.context; let attrs = &task.attrs; let cfgs = &task.cfgs; let stmts = &task.stmts; user_tasks.push(quote!( #(#attrs)* #(#cfgs)* #cfg_receiver #[allow(non_snake_case)] fn #name(#(#locals_pat,)* #context: #name::Context #(,#inputs)*) { use rtfm::Mutex as _; #(#stmts)* } )); root.push(module::codegen( Context::SoftwareTask(name), needs_lt, app, extra, )); } (const_app, root, user_tasks) } ption value='jarred/ast-again'>jarred/ast-again Unnamed repository; edit this file 'description' to name the repository.
aboutsummaryrefslogtreecommitdiff
AgeCommit message (Expand)AuthorFilesLines
2022-06-23Fix buffer test which was failing for correct reasonsGravatar Jarred Sumner 1-304/+304
2022-06-23Update MakefileGravatar Jarred Sumner 1-2/+4
2022-06-23Update MakefileGravatar Jarred Sumner 1-1/+1
2022-06-22Delete some filesGravatar Jarred Sumner 3-4/+0
2022-06-22Update libbacktraceGravatar Jarred Sumner 1-0/+0
2022-06-22change the directory structurejarred/renameGravatar Jarred Sumner 919-1727/+1958
2022-06-22maybe improve CPP build timeGravatar Jarred Sumner 1-24/+37
2022-06-22Move builtins to src/javascript/jsc/builtinsGravatar Jarred Sumner 72-6/+2285
2022-06-22Tweak test runner outputGravatar Jarred Sumner 1-90/+44
2022-06-22Cleanup some testsGravatar Jarred Sumner 2-8/+46
2022-06-22Update Dockerfile.baseGravatar Jarred Sumner 1-1/+1
2022-06-22Update WebKitGravatar Jarred Sumner 1-0/+0
2022-06-22cleanup websocket testGravatar Jarred Sumner 1-3/+6
2022-06-22Fix `WebSocket` when HTTP server is not runningGravatar Jarred Sumner 14-38/+103
2022-06-22Update build-idGravatar Jarred Sumner 1-1/+1
2022-06-22cleanupGravatar Jarred Sumner 6-719/+3
2022-06-22Update index.d.tsGravatar Jarred Sumner 1-0/+1
2022-06-22types for `bun:jsc`Gravatar Jarred Sumner 2-1/+37
2022-06-22Slightly customize the `events` polyfill so it uses ESMGravatar Jarred Sumner 1-1/+522
2022-06-22Fix memory bugs in escapeHTML & arrayBufferToStringGravatar Jarred Sumner 1-65/+61