Factor linear/Lotka-Voltera ODE analyses through PolynomialSystem

packages/catlog/src/simulate/ode/mod.rs

@@ -158,13 +158,7 @@ pub(crate) fn textplot_mapped_ode_result<Sys>(
 }
 
 pub mod kuramoto;
-#[allow(non_snake_case)]
-pub mod linear_ode;
-#[allow(non_snake_case)]
-pub mod lotka_volterra;
 pub mod polynomial;
 
 pub use kuramoto::*;
-pub use linear_ode::*;
-pub use lotka_volterra::*;
 pub use polynomial::*;

packages/catlog/src/simulate/ode/polynomial.rs

@@ -6,6 +6,7 @@ use std::ops::{Add, Neg};
 
 use derivative::Derivative;
 use indexmap::IndexMap;
+use itertools::Itertools;
 use nalgebra::DVector;
 use num_traits::{One, Pow, Zero};
 
@@ -115,6 +116,21 @@ where
             })
             .collect()
     }
+
+    /// Converts to a single aligned LaTeX environment.
+    pub fn to_latex_string(&self) -> String
+    where
+        Var: Display,
+        Coef: Display + PartialEq + One + Neg<Output = Coef>,
+        Exp: Display + PartialEq + One,
+    {
+        let equations = self
+            .to_latex_equations()
+            .into_iter()
+            .map(|p| p.lhs + " &= " + &p.rhs)
+            .join("\\\\\n");
+        format!("$$\n\\begin{{align*}}\n{}\n\\end{{align*}}\n$$\n", equations)
+    }
 }
 
 #[derive(Debug, PartialEq, Eq)]
@@ -184,6 +200,7 @@ where
 ///
 /// Such a system is ready for use in numerical solvers: the coefficients are
 /// floating point numbers and the variables are consecutive integer indices.
+#[derive(PartialEq, Debug)]
 pub struct NumericalPolynomialSystem<Exp> {
     /// Components of the vector field.
     pub components: Vec<Polynomial<usize, f32, Exp>>,