From 9e445b3583c15c7701f3167eaa8dfe4afd541691 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20Tj=C3=A4der?= <henrik@tjaders.com>
Date: Sat, 4 Feb 2023 16:47:17 +0100
Subject: Move rtic macros to repo root, tune xtask

---
 rtic-macros/src/syntax/optimize.rs | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 rtic-macros/src/syntax/optimize.rs

(limited to 'rtic-macros/src/syntax/optimize.rs')

diff --git a/rtic-macros/src/syntax/optimize.rs b/rtic-macros/src/syntax/optimize.rs
new file mode 100644
index 0000000..e83ba31
--- /dev/null
+++ b/rtic-macros/src/syntax/optimize.rs
@@ -0,0 +1,36 @@
+use std::collections::{BTreeSet, HashMap};
+
+use crate::syntax::ast::App;
+
+pub fn app(app: &mut App, settings: &Settings) {
+    // "compress" priorities
+    // If the user specified, for example, task priorities of "1, 3, 6",
+    // compress them into "1, 2, 3" as to leave no gaps
+    if settings.optimize_priorities {
+        // all task priorities ordered in ascending order
+        let priorities = app
+            .hardware_tasks
+            .values()
+            .map(|task| Some(task.args.priority))
+            .chain(
+                app.software_tasks
+                    .values()
+                    .map(|task| Some(task.args.priority)),
+            )
+            .collect::<BTreeSet<_>>();
+
+        let map = priorities
+            .iter()
+            .cloned()
+            .zip(1..)
+            .collect::<HashMap<_, _>>();
+
+        for task in app.hardware_tasks.values_mut() {
+            task.args.priority = map[&Some(task.args.priority)];
+        }
+
+        for task in app.software_tasks.values_mut() {
+            task.args.priority = map[&Some(task.args.priority)];
+        }
+    }
+}
-- 
cgit v1.2.3