Implemented computation of eb distribution

Author: bwbush

analysis/markov/Linleios/Evolve.lean

@@ -58,3 +58,12 @@ def simulate (env : Environment) (start : Probabilities) (ε : Float) : Nat →
 
 def totalProbability (states : Probabilities) : Probability :=
   states.values.sum
+
+def ebDistribution : Probabilities → HashMap Nat Probability :=
+  HashMap.fold
+    (
+      fun acc state p =>
+        HashMap.mergeWith (fun _ => Add.add) acc
+          $ singleton ⟨ state.ebCount , p ⟩
+    )
+    ∅

analysis/markov/Main.lean

@@ -4,12 +4,14 @@ import Linleios
 
 def env : Environment := makeEnvironment 0.05 0.95 0.90 600 0.75 1 4 7
 
-def s0 : Probabilities := default
-def sn := simulate env s0 1e-6 2
-def pn := totalProbability sn
+def sn := simulate env default 1e-6 10
 
 def main : IO Unit :=
   do
     let print {α : Type} [Repr α] (x : α) : IO Unit := IO.println (reprPrec x 0).pretty
+    IO.println ""
     print sn
-    print $ 1 - pn
+    IO.println ""
+    print $ ebDistribution sn
+    IO.println ""
+    print $ 1 - totalProbability sn