From 7614b96fe45240dafe91ae549e712b560e2d4c10 Mon Sep 17 00:00:00 2001
From: Emil Fresk <emil.fresk@gmail.com>
Date: Sat, 31 Dec 2022 14:45:13 +0100
Subject: RTIC v2: Initial commit

rtic-syntax is now part of RTIC repository
---
 macros/src/syntax/optimize.rs | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 macros/src/syntax/optimize.rs

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

diff --git a/macros/src/syntax/optimize.rs b/macros/src/syntax/optimize.rs
new file mode 100644
index 00000000..87a6258d
--- /dev/null
+++ b/macros/src/syntax/optimize.rs
@@ -0,0 +1,36 @@
+use std::collections::{BTreeSet, HashMap};
+
+use crate::syntax::{ast::App, Settings};
+
+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