Initial commit.

src/variable.rs

@@ -0,0 +1,38 @@
+use crate::{Gaussian, N_INF};
+
+#[derive(Debug)]
+pub struct TeamVariable {
+    pub prior: Gaussian,
+    pub likelihood_lose: Gaussian,
+    pub likelihood_win: Gaussian,
+    pub likelihood_draw: Gaussian,
+}
+
+impl TeamVariable {
+    pub fn p(&self) -> Gaussian {
+        self.prior * self.likelihood_lose * self.likelihood_win * self.likelihood_draw
+    }
+
+    pub fn posterior_win(&self) -> Gaussian {
+        self.prior * self.likelihood_lose * self.likelihood_draw
+    }
+
+    pub fn posterior_lose(&self) -> Gaussian {
+        self.prior * self.likelihood_win * self.likelihood_draw
+    }
+
+    pub fn likelihood(&self) -> Gaussian {
+        self.likelihood_win * self.likelihood_lose * self.likelihood_draw
+    }
+}
+
+impl Default for TeamVariable {
+    fn default() -> Self {
+        Self {
+            prior: N_INF,
+            likelihood_lose: N_INF,
+            likelihood_win: N_INF,
+            likelihood_draw: N_INF,
+        }
+    }
+}