task_local and lock_free analysis (take 1)

2 files changed, 193 insertions, 0 deletions

diff --git a/macros/src/custom_local.rs b/macros/src/custom_local.rs
new file mode 100644
index 00000000..33817905
--- /dev/null
+++ b/macros/src/custom_local.rs
@@ -0,0 +1,178 @@
+use syn::parse;
+//use syn::Ident;
+//use proc_macro2::{Ident, Span};
+use proc_macro2::Ident;
+use rtfm_syntax::{
+    analyze::{Analysis, Ownership},
+    ast::App,
+};
+use syn::Error;
+//     ast::{App, CustomArg},
+
+type Idents<'a> = Vec<&'a Ident>;
+
+// Assign an `extern` interrupt to each priority level
+pub fn app(app: &App, _analysis: &Analysis) -> parse::Result<()> {
+    // collect task local resources
+    let task_local: Idents = app
+        .resources
+        .iter()
+        .filter(|(_, r)| r.properties.task_local)
+        .map(|(i, _)| i)
+        .chain(
+            app.late_resources
+                .iter()
+                .filter(|(_, r)| r.properties.task_local)
+                .map(|(i, _)| i),
+        )
+        .collect();
+
+    let lock_free: Idents = app
+        .resources
+        .iter()
+        .filter(|(_, r)| r.properties.lock_free)
+        .map(|(i, _)| i)
+        .chain(
+            app.late_resources
+                .iter()
+                .filter(|(_, r)| r.properties.lock_free)
+                .map(|(i, _)| i),
+        )
+        .collect();
+
+    // collect all tasks into a vector
+    type Task = String;
+
+    let all_tasks: Vec<(Task, Idents)> = app
+        .idles
+        .iter()
+        .map(|(core, ht)| {
+            (
+                format!("Idle (core {})", core),
+                ht.args.resources.iter().map(|(v, _)| v).collect::<Vec<_>>(),
+            )
+        })
+        .chain(app.software_tasks.iter().map(|(name, ht)| {
+            (
+                name.to_string(),
+                ht.args.resources.iter().map(|(v, _)| v).collect::<Vec<_>>(),
+            )
+        }))
+        .chain(app.hardware_tasks.iter().map(|(name, ht)| {
+            (
+                name.to_string(),
+                ht.args.resources.iter().map(|(v, _)| v).collect::<Vec<_>>(),
+            )
+        }))
+        .collect();
+
+    // check that task_local resources is only used once
+    let mut error = vec![];
+    for task_local_id in task_local.iter() {
+        let mut used = vec![];
+        for (task, tr) in all_tasks.iter() {
+            for r in tr {
+                if task_local_id == r {
+                    used.push((task, r));
+                }
+            }
+        }
+        if used.len() > 1 {
+            error.push(Error::new(
+                task_local_id.span(),
+                format!(
+                    "task local resource {:?} is used by multiple tasks",
+                    task_local_id.to_string()
+                ),
+            ));
+
+            used.iter().for_each(|(task, resource)| {
+                error.push(Error::new(
+                    resource.span(),
+                    format!(
+                        "task local resource {:?} is used by task {:?}",
+                        resource.to_string(),
+                        task
+                    ),
+                ))
+            });
+        }
+    }
+
+    // filter out contended resources
+    let contended: Vec<(&Ident, &Ownership)> = _analysis
+        .ownerships
+        .iter()
+        .filter(|(_id, own)| match own {
+            Ownership::Contended { .. } => true,
+            _ => false,
+        })
+        .collect();
+
+    // filter out lock_free contended resources
+    let lock_free_violation: Vec<&(&Ident, &Ownership)> = contended
+        .iter()
+        .filter(|(cont_id, _)| lock_free.iter().any(|lf_id| cont_id == lf_id))
+        .collect();
+
+    // report contention error
+    lock_free_violation.iter().for_each(|(lf_err_id, _)| {
+        error.push(Error::new(
+            lf_err_id.span(),
+            format!(
+                "lock_free resource {:?} is contended by higher priority task",
+                lf_err_id.to_string()
+            ),
+        ))
+    });
+
+    // collect errors
+    if error.is_empty() {
+        Ok(())
+    } else {
+        let mut err = error.iter().next().unwrap().clone();
+        error.iter().for_each(|e| err.combine(e.clone()));
+        Err(err)
+    }
+
+    //     for tl in task_local {
+    //         println!("tl {:?}", tl);
+    //  //       let mut first_use = None;
+    //         for i in _analysis.ownerships.iter() {
+    //             println!("\nown: {:?}", i);
+    //     }
+
+    //     // println!("analysis {:?}", _analysis.locations);
+
+    // for i in _analysis.locations.iter() {
+    //     println!("\nloc: {:?}", i);
+    // }
+
+    // ErrorMessage {
+    //     // Span is implemented as an index into a thread-local interner to keep the
+    //     // size small. It is not safe to access from a different thread. We want
+    //     // errors to be Send and Sync to play nicely with the Failure crate, so pin
+    //     // the span we're given to its original thread and assume it is
+    //     // Span::call_site if accessed from any other thread.
+    //     start_span: ThreadBound<Span>,
+    //     end_span: ThreadBound<Span>,
+    //     message: String,
+    // }
+    //        (app.name, "here");
+    // let span = app.name.span();
+    // let start = span.start();
+    //    Err(vec![ErrorMessage { start_span: app.name.}]);
+
+    // let mut task_locals = Vec::new();
+    // println!("-- app:late_resources");
+
+    // println!(
+    //     "task_locals {:?}",
+    //     app.late_resources.filter(|r| r.task_local)
+    // );
+
+    // println!("-- resources");
+    // for i in app.resources.iter() {
+    //     println!("res: {:?}", i);
+    // }
+}
diff --git a/macros/src/lib.rs b/macros/src/lib.rs
index e659559e..2c81ede0 100644
--- a/macros/src/lib.rs
+++ b/macros/src/lib.rs
@@ -10,6 +10,7 @@ use rtic_syntax::Settings;
 mod analyze;
 mod check;
 mod codegen;
+mod custom_local;
 #[cfg(test)]
 mod tests;
 
@@ -214,13 +215,27 @@ pub fn app(args: TokenStream, input: TokenStream) -> TokenStream {
         Ok(x) => x,
     };
 
+    match custom_local::app(&app, &analysis) {
+        Err(e) => return e.to_compile_error().into(),
+        Ok(_) => {}
+    }
+
     let extra = match check::app(&app, &analysis) {
         Err(e) => return e.to_compile_error().into(),
         Ok(x) => x,
     };
 
+    // println!("extra {:?}", extra);
+
     let analysis = analyze::app(analysis, &app);
 
+    // println!("after analysis, extra {:?}", extra);
+
+    // match custom_local::app(&app, &analysis) {
+    //     Err(e) => return e.to_compile_error().into(),
+    //     Ok(_) => {}
+    // }
+
     let ts = codegen::app(&app, &analysis, &extra);
 
     // Try to write the expanded code to disk