add way of saving the encryption

README.md

@@ -17,7 +17,7 @@ cargo run [input_file.txt] [output_file.bin]
 
 `Old file size: 2026200 (in bytes)`
 
-`After file size:1107887 (in bytes)`
+`After file size: 1107887 (in bytes)`
 
 # Notes:
 This can surely be optimized. It was one of my first projects in Rust, so I am sure I will look back in agony at how poorly it was done. However, that is a problem for future me.

out.txt

@@ -0,0 +1,4 @@
+A 0
+B 10
+C 110
+D 111

src/huffman.rs

@@ -153,6 +153,7 @@ fn convert_to_code_str(original_msg: &String,map: &BTreeMap<char,Vec<bool>>) ->V
     ret
 }
 
+
 fn get_hash_char_freq(msg:String) -> BTreeMap<char,usize> {
     /*
     Return a hashmap of characters and their respective frequencies.

src/main.rs

@@ -1,3 +1,4 @@
+use std::collections::BTreeMap;
 use std::fs::File;
 use std::fs;
 use std::io::{Read, Result,Write};
@@ -61,6 +62,28 @@ fn write_str_to_file(filename: &str, data: &Vec<bool>) -> std::io::Result<()>{
         file.write(&[bit_buffer])?;
     }
 
+    file.flush();
+
+    Ok(())
+}
+
+fn print_hash_to_file(map: &BTreeMap<char,Vec<bool>>, filename: &str) -> std::io::Result<()>{
+    let mut file = File::create(filename)?;
+
+    for (key,value) in map{
+        file.write(&[key.to_owned() as u8])?;
+        file.write(&[' ' as u8])?;
+        for bool_val in value{
+            if *bool_val{
+                file.write(&['1' as u8])?;
+            }else{
+                file.write(&['0' as u8])?;
+            }
+        }
+        file.write(&['\n' as u8])?;
+    }
+    file.flush()?;
+
     Ok(())
 }
 
@@ -117,4 +140,6 @@ fn main() {
     println!("Old file size: {prior_file_size} (in bytes)");
     println!("After file size: {after_file_size} (in bytes)");
 
+    print_hash_to_file(&hash_code,"out.txt");
+
 }