Add Subspace::complement

SpectralSequences · Jul 6, 2023 · 413ac5b · 413ac5b
1 parent ac82650
commit 413ac5b
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/ext/crates/fp/src/matrix/subspace.rs b/ext/crates/fp/src/matrix/subspace.rs
@@ -134,6 +134,24 @@ impl Subspace {
         }
     }
 
+    /// Returns a complementary space to `self`.
+    pub fn complement(&self) -> Subspace {
+        let dimension = self.ambient_dimension() - self.dimension();
+        let mut complement = Subspace::new(self.prime(), dimension, self.ambient_dimension());
+        complement.add_vectors(|mut row| {
+            let mut i = 0;
+            while i < self.ambient_dimension() {
+                i += 1;
+                if !self.pivots().contains(&(i as isize - 1)) {
+                    row.set_entry(i - 1, 1);
+                    return Some(());
+                }
+            }
+            None
+        });
+        complement
+    }
+
     pub fn contains(&self, vector: Slice) -> bool {
         let mut vector: FpVector = vector.into_owned();
         self.reduce(vector.as_slice_mut());