1 files changed, 22 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 39966e6..6be1aba 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -6,6 +6,7 @@ use std::fmt;
 use std::iter;
 use std::iter::FromIterator;
 use std::iter::IntoIterator;
+use std::iter::Iterator;
 
 pub type Intcode = BigInt;
 
@@ -57,6 +58,18 @@ impl fmt::Debug for IntcodeProgram {
     }
 }
 
+pub fn intcode_to_bool(i: &Intcode) -> bool {
+    *i != Intcode::from(0)
+}
+
+pub fn bool_to_intcode(i: bool) -> Intcode {
+    if i {
+        Intcode::from(1)
+    } else {
+        Intcode::from(0)
+    }
+}
+
 impl IntcodeProgram {
     pub fn with_noun_verb_input(&self, noun: Intcode, verb: Intcode) -> IntcodeProgram {
         self.with_memory_set(1.into(), noun)
@@ -444,3 +457,12 @@ mod tests {
         assert_eq!(program.output, i32_vec_to_intcode_vec(quine));
     }
 }
+
+pub fn sort_by_key<T, K: Ord>(
+    iter: impl IntoIterator<Item = T>,
+    f: impl FnMut(&T) -> K,
+) -> impl Iterator<Item = T> {
+    let mut tmp: Vec<T> = iter.into_iter().collect();
+    tmp.sort_by_key(f);
+    tmp.into_iter()
+}