Added C3 algorithm

Author: Sdoba16

src/driver/c3_linearization.rs

@@ -0,0 +1,231 @@
+// TODO: Remove this once the code is actively used.
+#![allow(dead_code)]
+use std::{collections::HashMap, fmt};
+
+use crate::driver::DependencyGraph;
+
+impl DependencyGraph {
+    fn c3_linearize(&self) -> Result<Vec<usize>, C3Error> {
+        self.linearize_module(0)
+    }
+
+    fn linearize_module(&self, root: usize) -> Result<Vec<usize>, C3Error> {
+        let mut memo = HashMap::new();
+        let mut visiting = Vec::new();
+
+        self.linearize_rec(root, &mut memo, &mut visiting)
+    }
+
+    /// Core recursive DFS for C3 linearization.
+    ///
+    /// - **Memoization (`memo`):** Prevents exponential blowup in diamond dependencies by calculating shared modules exactly once.
+    /// - **Cycle Detection (`visiting`):** Prevents infinite loops by returning an error if the current path repeats a module (e.g., A -> B -> A).
+    ///
+    /// # Returns
+    /// - `Ok(Vec<usize>)`: The deterministic, top-down ordered sequence of dependencies for the module.
+    /// - `Err(C3Error)`: If a circular dependency or unresolvable conflict is detected.
+    fn linearize_rec(
+        &self,
+        module: usize,
+        memo: &mut HashMap<usize, Vec<usize>>,
+        visiting: &mut Vec<usize>,
+    ) -> Result<Vec<usize>, C3Error> {
+        if let Some(result) = memo.get(&module) {
+            return Ok(result.clone());
+        }
+
+        if let Some(cycle_start) = visiting.iter().position(|&m| m == module) {
+            return Err(C3Error::CycleDetected(
+                visiting[cycle_start..]
+                    .iter()
+                    .map(|&id| self.modules[id].source.str_name())
+                    .collect(),
+            ));
+        }
+
+        visiting.push(module);
+
+        let parents = self
+            .dependencies
+            .get(&module)
+            .map_or(&[] as &[usize], |v| v.as_slice());
+
+        let mut seqs: Vec<Vec<usize>> = Vec::with_capacity(parents.len() + 1);
+
+        for &parent in parents {
+            seqs.push(self.linearize_rec(parent, memo, visiting)?);
+        }
+
+        let mut result = vec![module];
+        let merged = c3_merge(seqs).map_err(|conflicts| C3Error::InconsistentLinearization {
+            module: self.modules[module].source.str_name(),
+            conflicts: self.format_conflict_names(conflicts),
+        })?;
+
+        result.extend(merged);
+
+        visiting.pop();
+        memo.insert(module, result.clone());
+
+        Ok(result)
+    }
+
+    /// Helper to convert raw usize conflict sequences into readable module names
+    fn format_conflict_names(&self, conflicts: Vec<Vec<usize>>) -> Vec<Vec<String>> {
+        conflicts
+            .into_iter()
+            .map(|seq| {
+                seq.into_iter()
+                    .map(|id| self.modules[id].source.str_name())
+                    .collect()
+            })
+            .collect()
+    }
+}
+
+/// Merges a list of sequences (parent linearizations) into a single sequence.
+/// The algorithm ensures that the local precedence order of each sequence is preserved.
+fn c3_merge(seqs: Vec<Vec<usize>>) -> Result<Vec<usize>, Vec<Vec<usize>>> {
+    // Convert to a collection of slices. This allows us to "remove" the head
+    // by simply advancing the slice pointer by 1, which is entirely zero-cost.
+    let mut slices: Vec<&[usize]> = seqs.iter().map(AsRef::as_ref).collect();
+    let mut result = Vec::new();
+
+    loop {
+        slices.retain(|s| !s.is_empty());
+        if slices.is_empty() {
+            return Ok(result);
+        }
+
+        let candidate = slices
+            .iter()
+            .map(|s| s[0])
+            .find(|&head| !slices.iter().any(|s| s[1..].contains(&head)));
+
+        let Some(head) = candidate else {
+            let conflicts = slices.into_iter().map(|s| s.to_vec()).collect();
+            return Err(conflicts);
+        };
+
+        result.push(head);
+        for seq in &mut slices {
+            if seq.first() == Some(&head) {
+                *seq = &seq[1..];
+            }
+        }
+    }
+}
+
+#[derive(Debug)]
+enum C3Error {
+    CycleDetected(Vec<String>),
+    /// Error for inconsistent MRO.
+    /// This can happen if the dependency graph has a shape that makes the
+    /// order of parent classes ambiguous.
+    /// Example: A depends on B and C, and B also depends on C.
+    /// The linearization of A is A + merge(linearization(B), linearization(C), [B, C]).
+    /// If B appears before C in one parent's linearization but C appears before B
+    /// in another's, the merge will fail.
+    InconsistentLinearization {
+        module: String,
+        conflicts: Vec<Vec<String>>,
+    },
+}
+
+impl fmt::Display for C3Error {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            C3Error::CycleDetected(cycle) => {
+                write!(f, "Circular dependency detected: {:?}", cycle.join(" -> "))
+            }
+            C3Error::InconsistentLinearization { module, conflicts } => {
+                writeln!(f, "Inconsistent resolution order for module '{}'", module)?;
+                writeln!(
+                    f,
+                    "The compiler could not resolve the following conflicting import constraints:"
+                )?;
+
+                // Loop through the matrix and print each conflicting sequence
+                for conflict in conflicts {
+                    writeln!(f, "  [{}]", conflict.join(", "))?;
+                }
+
+                write!(
+                    f,
+                    "Try reordering your `use` statements to avoid cross-wiring."
+                )
+            }
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::driver::tests::setup_graph;
+
+    use super::*;
+
+    #[test]
+    fn test_c3_simple_import() {
+        let (graph, ids, _dir) = setup_graph(
+            "c3_simple_import",
+            vec![
+                ("main.simf", "use lib::math::some_func;"),
+                ("libs/lib/math.simf", ""),
+            ],
+        );
+
+        let order = graph.c3_linearize().unwrap();
+
+        let root_id = ids["main"];
+        let math_id = ids["math"];
+
+        // Assuming linearization order: Dependent (Root) -> Dependency (Math)
+        assert_eq!(order, vec![root_id, math_id]);
+    }
+
+    #[test]
+    fn test_c3_diamond_dependency_deduplication() {
+        // Setup:
+        // root -> imports A, B
+        // A -> imports Common
+        // B -> imports Common
+        // Expected: Common loaded ONLY ONCE.
+
+        let (graph, ids, _dir) = setup_graph(
+            "c3",
+            vec![
+                ("main.simf", "use lib::A::foo; use lib::B::bar;"),
+                ("libs/lib/A.simf", "use lib::Common::dummy1;"),
+                ("libs/lib/B.simf", "use lib::Common::dummy2;"),
+                ("libs/lib/Common.simf", ""),
+            ],
+        );
+
+        let order = graph.c3_linearize().unwrap();
+
+        // Verify order using IDs from the helper map
+        let main_id = ids["main"];
+        let a_id = ids["A"];
+        let b_id = ids["B"];
+        let common_id = ids["Common"];
+
+        // The current list is ordered top-down. The calling function will reverse this to process 'common' first and 'main' last.
+        assert_eq!(order, vec![main_id, a_id, b_id, common_id]);
+    }
+
+    #[test]
+    fn test_c3_detects_cycle() {
+        let (graph, _, _dir) = setup_graph(
+            "c3",
+            vec![
+                ("main.simf", "use lib::A::entry;"),
+                ("libs/lib/A.simf", "use lib::B::func;"),
+                ("libs/lib/B.simf", "use lib::A::func;"),
+            ],
+        );
+
+        let order = graph.c3_linearize();
+        matches!(order, Err(C3Error::CycleDetected(_)));
+    }
+}

src/driver/mod.rs

@@ -1,3 +1,5 @@
+mod c3_linearization;
+
 use std::collections::{HashMap, HashSet, VecDeque};
 use std::path::PathBuf;
 use std::sync::Arc;
@@ -46,6 +48,10 @@ impl CanonSourceFile {
         &self.name
     }
 
+    pub fn str_name(&self) -> String {
+        self.name.as_path().display().to_string()
+    }
+
     pub fn content(&self) -> Arc<str> {
         self.content.clone()
     }