Implemented state for Tally (private)

input-output-hk · Oct 19, 2020 · 5c01f89 · 5c01f89
1 parent 63f4dd0
commit 5c01f89
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/chain-vote/src/lib.rs b/chain-vote/src/lib.rs
@@ -155,6 +155,12 @@ impl Tally {
         (TallyState { r2s }, TallyDecryptShare { r1s: dshares })
     }
 
+    pub fn state(&self) -> TallyState {
+        TallyState {
+            r2s: self.r.iter().map(|r| r.elements().1.clone()).collect(),
+        }
+    }
+
     pub fn to_bytes(&self) -> Vec<u8> {
         use std::io::Write;
         let mut bytes: Vec<u8> = Vec::with_capacity(CIPHERTEXT_BYTES_LEN * self.r.len());